-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathvideo_processor.py
More file actions
592 lines (482 loc) · 30 KB
/
Copy pathvideo_processor.py
File metadata and controls
592 lines (482 loc) · 30 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
# video_processor.py
#
# Autor: Sebastian Pachla <seeb@seeb.net.pl>
# Wersja: 4.8.0.9
# Data: 2025-09-10
#
# Opis zmian względem wersji 4.8.0.8:
# - Dodano automatyczną weryfikację zgodności liczby klatek na zegarze z klatkażem wideo
# - Zaimplementowano logikę ignorowania wszystkich klatek z FF>00 gdy wykryto niezgodność
# - Dodano mechanizm wykrywania i raportowania problemów z wyświetlaczem zegara
#
# Moduł odpowiedzialny za przetwarzanie wideo, włącznie z dodawaniem kodu czasowego
# (Poprawiona logika tworzenia obiektu Timecode ze stringa HH:MM:SS:FF)
#
# Opis:
# Ten moduł zawiera klasę VideoProcessor do przetwarzania plików wideo.
# Główne funkcje obejmują:
# 1. Odczytywanie kodu QR GoPro z nagrań wideo w celu uzyskania dokładnego czasu rozpoczęcia.
# 2. Generowanie sygnału Linear Timecode (LTC) jako pliku audio WAV, opartego na odczytanym czasie i klatkażu wideo.
# 3. Łączenie wygenerowanego pliku audio z oryginalnym wideo jako dodatkowej ścieżki audio za pomocą FFmpeg.
# Rozwiązanie omija brak filtra 'smpteh' w standardowych kompilacjach FFmpeg.
#
# Zależności:
# - FFmpeg (musi być zainstalowany i dostępny w PATH)
# - ffprobe (część pakietu FFmpeg, musi być zainstalowany i dostępny w PATH)
# - OpenCV (cv2) (instalacja: `pip install opencv-python`)
# - pyzbar (instalacja: `pip install pyzbar`)
# - pytesseract (instalacja: `pip install pytesseract`)
# - pytz (biblioteka Python, instalacja: `pip install pytz`)
# - numpy (instalacja: `pip install numpy`)
# - scipy (instalacja: `pip install scipy`)
# - timecode (biblioteka Python, najprawdopodobniej zainstalowana globalnie, np. `pip install timecode`)
# - timecode_tools (repozytorium sklonowane do external_libs/, używamy tylko tools.py z tego)
import sys
import os
import datetime
import subprocess
import cv2
from pyzbar import pyzbar
import re
from fractions import Fraction
import pytz
import numpy as np
import traceback
import numbers
import pytesseract
# Import dla zapisu pliku WAV
from scipy.io.wavfile import write as write_wav
# Dodaj ścieżkę do katalogu 'external_libs'
# W pliku 'main.py' dodano obsługę 'external_libs', więc nie jest to już potrzebne
# tutaj, ale zostawiamy zakomentowane na wszelki wypadek.
# current_script_dir = os.path.dirname(os.path.abspath(__file__))
# external_libs_path = os.path.join(current_script_dir, 'external_libs')
# if external_libs_path not in sys.path:
# sys.path.insert(0, external_libs_path)
# --- KLUCZOWE ZMIANY W IMPORCIE ---
try:
# Import Timecode z systemowej biblioteki (tej, którą masz zainstalowaną)
from timecode import Timecode
# Import ltc_encode i cint z 'external_libs/timecode_tools/tools.py'
from timecode_tools.tools import ltc_encode, cint
print("Pomyślnie załadowano Timecode (zewnętrzny) i ltc_encode/cint (z timecode_tools).")
except ImportError as e:
print(f"Błąd: Nie można załadować wymaganych modułów. Upewnij się, że 'timecode' (pip install timecode) jest zainstalowany i 'external_libs/timecode_tools/tools.py' jest dostępne. Błąd: {e}")
traceback.print_exc()
sys.exit(1) # Zakończ program z błędem, ponieważ to jest krytyczny import
__version__ = "4.8.0.9" # Zaktualizowany numer wersji
print(f"Ładowanie video_processor.py - Wersja: {__version__}")
# --- PARAMETRY UŻYTKOWNIKA ---
# Ilość sekund wideo, które mają być skanowane w poszukiwaniu kodu QR lub wizualnego timecode'u.
SCAN_DURATION_SECONDS = 10
# Możesz zmienić tę wartość, aby dostosować czas skanowania.
# --- FUNKCJA parse_gopro_qr_timecode - PRZYWRÓCONA ---
def parse_gopro_qr_timecode(qr_data: str) -> datetime.datetime:
"""
Parsuje dane kodu QR GoPro do obiektu datetime.
Oczekiwany format: oT<YYMMDDHHMMSS.mmm>oTD<DST_FLAG>oTZ<TIMEZONE_OFFSET>oTI<UTC_OFFSET>
Przykład: oT250618091541.679oTD1oTZ2oTI0
"""
if not qr_data.startswith('oT'):
raise ValueError("Nieprawidłowy format kodu QR GoPro: nie zaczyna się od 'oT'")
timestamp_match = re.search(r'oT(\d{12}\.\d{3})', qr_data)
if not timestamp_match:
raise ValueError("Nieprawidłowy format kodu QR GoPro: nie znaleziono części znacznika czasu")
timestamp_str = timestamp_match.group(1)
year_str = timestamp_str[0:2]
current_year = datetime.datetime.now().year
year_prefix = (current_year // 100) * 100
# Prosta heurystyka dla roku: jeśli XX w YYMMDD jest większe niż bieżący rok + 10 lat,
# zakładamy, że to poprzednie stulecie (np. 99 dla 1999, jeśli jest rok 2000-2009)
if int(year_str) > (current_year % 100) + 10:
year_prefix -= 100
full_year = year_prefix + int(year_str)
month = int(timestamp_str[2:4])
day = int(timestamp_str[4:6])
hour = int(timestamp_str[6:8])
minute = int(timestamp_str[8:10])
second = int(timestamp_str[10:12])
millisecond = int(timestamp_str[13:16])
# Tworzymy naive datetime
naive_dt = datetime.datetime(full_year, month, day, hour, minute, second, millisecond * 1000)
# Parsowanie offsetu UTC
utc_offset_match = re.search(r'oTI(-?\d+)', qr_data)
if not utc_offset_match:
utc_offset_hours = 0 # Domyślnie 0, jeśli nie ma offsetu
else:
utc_offset_hours = int(utc_offset_match.group(1))
# Tworzenie obiektu timezone z offsetu
offset_timedelta = datetime.timedelta(hours=utc_offset_hours)
tz_info = datetime.timezone(offset_timedelta)
# Lokalizacja naive_dt do strefy czasowej z QR i konwersja na UTC
localized_dt = naive_dt.replace(tzinfo=tz_info)
utc_dt = localized_dt.astimezone(datetime.timezone.utc)
return utc_dt
def generate_ltc_audio_file(start_time_utc: datetime.datetime, duration_seconds: float, fps: float, output_path: str):
"""
Generuje plik WAV zawierający sygnał LTC.
Używa klasy Timecode z zewnętrznej biblioteki 'timecode' i funkcji ltc_encode z 'timecode_tools/tools.py'.
"""
sample_rate = 48000
bits = 16 # Domyślnie 16-bitowe audio, jak w standardach LTC
print(f"DEBUG (LTC Gen): start_time_utc: {start_time_utc} (type: {type(start_time_utc)})")
print(f"DEBUG (LTC Gen): duration_seconds: {duration_seconds} (type: <class 'float'>)")
print(f"DEBUG (LTC Gen): fps: {fps} (type: <class 'float'>)")
try:
if not isinstance(duration_seconds, (int, float)):
raise TypeError(f"duration_seconds musi być liczbą, otrzymano {type(duration_seconds)}: {duration_seconds}")
if not isinstance(fps, (int, float)):
raise TypeError(f"fps musi być liczbą, otrzymano {type(fps)}: {fps}")
if not isinstance(start_time_utc, datetime.datetime):
raise TypeError(f"start_time_utc musi być obiektem datetime.datetime, otrzymano {type(start_time_utc)}: {start_time_utc}")
print(f"DEBUG (LTC Gen): Wartości po konwersji/sprawdzeniu: duration_seconds={duration_seconds}, fps={fps}")
# Utworzenie obiektu Timecode
# Biblioteka Timecode (v1.4.1) nie posiada metody from_datetime().
# Musimy ręcznie sformatować datetime na string HH:MM:SS:FF dla konstruktora Timecode(fps, start_string).
# Obliczanie całkowitej liczby klatek od północy UTC do start_time_utc
# Upewniamy się, że czas jest "naiwny" (bez strefy czasowej) do obliczeń
naive_start_time = start_time_utc.replace(tzinfo=None)
# Oblicz sekundy od północy
total_seconds_from_midnight = (naive_start_time - naive_start_time.replace(hour=0, minute=0, second=0, microsecond=0)).total_seconds()
# Dodaj klatki wynikające z milisekund. Zaokrąglamy w dół, aby nie przekroczyć bieżącej klatki.
frames_from_seconds = int(total_seconds_from_midnight * fps)
# Rozłożenie całkowitej liczby klatek na HH:MM:SS:FF
frames_per_hour = int(fps * 3600)
frames_per_minute = int(fps * 60)
frames_per_second = int(fps)
hours = frames_from_seconds // frames_per_hour
remaining_frames = frames_from_seconds % frames_per_hour
minutes = remaining_frames // frames_per_minute
remaining_frames %= frames_per_minute
seconds = remaining_frames // frames_per_second
frames = remaining_frames % frames_per_second
# Upewniamy się, że nie ma wartości ujemnych dla klatek (może się zdarzyć przy bardzo małych floatach)
frames = max(0, frames)
start_time_code_string = f"{int(hours):02d}:{int(minutes):02d}:{int(seconds):02d}:{int(frames):02d}"
print(f"DEBUG (LTC Gen): Tworzenie obiektu Timecode. Start timecode string: {start_time_code_string}, fps: {fps}")
tc_start = Timecode(fps, start_time_code_string) # Używamy konstruktora Timecode(fps, start_string)
print(f"DEBUG (LTC Gen): Obiekt Timecode startowy utworzony: {tc_start}")
total_frames_to_generate_ltc = int(duration_seconds * fps)
print(f"DEBUG (LTC Gen): Całkowita liczba klatek do wygenerowania LTC: {total_frames_to_generate_ltc}")
# Generowanie danych LTC dla każdej klatki
ltc_frames_data = []
current_tc = tc_start
# Generujemy o jedną klatkę więcej niż total_frames_to_generate_ltc, aby upewnić się,
# że pokrywamy pełny czas trwania i uniknąć niedomiaru w przypadku zaokrągleń
for _ in range(total_frames_to_generate_ltc + 1):
ltc_frames_data.append(ltc_encode(current_tc, as_string=True))
current_tc.next()
# Łączenie danych binarnych LTC
ltc_bitstream = ''.join(ltc_frames_data)
# Generowanie sygnału "Double Pulse" na podstawie bitstreamu LTC
double_pulse_data = ''
next_is_up = True
for bit_char in ltc_bitstream:
if bit_char == '0':
if next_is_up:
double_pulse_data += '11'
else:
double_pulse_data += '00'
next_is_up = not next_is_up
else:
double_pulse_data += '10' if next_is_up else '01'
# Konwersja sygnału "Double Pulse" na dane PCM
total_samples = int(sample_rate * duration_seconds)
# Obliczamy liczbę próbek na "bit" w double_pulse_data
# Jeśli double_pulse_data jest puste (mało prawdopodobne, ale dla bezpieczeństwa)
if not double_pulse_data:
print("Ostrzeżenie: double_pulse_data jest puste, nie można wygenerować audio LTC.")
return False
samples_per_bit_in_double_pulse = total_samples / len(double_pulse_data)
# Stworzenie tablicy NumPy na dane audio
audio_data = np.zeros(total_samples, dtype=np.int16) # np.int16 dla 16-bitowych próbek
on_val = 32767 # Max wartość dla int16
off_val = -32768 # Min wartość dla int16
for i in range(total_samples):
# Obliczenie, który bit z double_pulse_data odpowiada bieżącej próbce
# Używamy min, aby zapobiec wyjściu poza zakres przy ostatniej próbce
double_pulse_index = min(int(i / samples_per_bit_in_double_pulse), len(double_pulse_data) - 1)
if double_pulse_data[double_pulse_index] == '1':
audio_data[i] = on_val
else:
audio_data[i] = off_val
# Zapisanie danych audio do pliku WAV
write_wav(output_path, sample_rate, audio_data)
print(f"Wygenerowano tymczasowy plik audio (LTC): {output_path}")
return True
except Exception as e:
print(f"Błąd podczas generowania pliku LTC audio {output_path}: {e}")
traceback.print_exc()
return False
class VideoProcessor:
def __init__(self, output_base_dir: str, input_base_dir: str):
self.output_base_dir = output_base_dir
self.input_base_dir = input_base_dir
if not os.path.exists(self.output_base_dir):
os.makedirs(self.output_base_dir)
def _get_video_info(self, video_path: str) -> tuple[float, float]:
"""Pobiera czas trwania wideo i klatkaż za pomocą ffprobe."""
try:
probe_duration_cmd = [
'ffprobe', '-v', 'error', '-show_entries', 'format=duration',
'-of', 'default=noprint_wrappers=1:nokey=1', video_path
]
duration_result = subprocess.run(probe_duration_cmd, capture_output=True, text=True, check=True)
duration_str = duration_result.stdout.strip()
if not duration_str:
raise ValueError(f"FFprobe nie zwrócił czasu trwania dla {video_path}. Wyjście: '{duration_result.stdout.strip()}' Błąd: '{duration_result.stderr.strip()}'")
duration_seconds = float(duration_str)
probe_fps_cmd = [
'ffprobe', '-v', 'error', '-select_streams', 'v:0',
'-show_entries', 'stream=avg_frame_rate',
'-of', 'default=noprint_wrappers=1:nokey=1', video_path
]
fps_result = subprocess.run(probe_fps_cmd, capture_output=True, text=True, check=True)
fps_output = fps_result.stdout.strip()
fps_matches = re.findall(r'^\d+(?:/\d+)?$', fps_output, re.MULTILINE)
if not fps_matches:
raise ValueError(f"Nie udało się sparsować klatkażu z wyjścia ffprobe dla {video_path}. Wyjście: '{fps_output}' Błąd: '{fps_result.stderr.strip()}'")
fps_str = fps_matches[0]
frame_rate = float(Fraction(fps_str))
print(f"DEBUG: _get_video_info dla {video_path} zwróciło: duration={duration_seconds}, fps={frame_rate}")
return duration_seconds, frame_rate
except (subprocess.CalledProcessError, ValueError, ZeroDivisionError) as e:
raise ValueError(f"Błąd podczas pobierania informacji wideo dla {video_path}: {e}") from e
def _read_qr_from_video(self, video_path: str, frame_rate: float) -> tuple[datetime.datetime | None, int]:
"""Odczytuje pierwszy prawidłowy kod QR z początku wideo."""
cap = cv2.VideoCapture(video_path)
if not cap.isOpened():
print(f"Błąd: Nie można otworzyć pliku wideo {video_path}")
return None, -1
first_qr_time = None
first_qr_frame_index = -1
frame_index = 0
# Obliczanie maksymalnej liczby klatek do przeskanowania na podstawie parametru użytkownika
max_frames_to_scan = int(frame_rate * SCAN_DURATION_SECONDS)
# Przynajmniej 50 klatek, żeby nie przegapić QR (dodatkowe zabezpieczenie)
if max_frames_to_scan < 50:
max_frames_to_scan = 50
while frame_index < max_frames_to_scan:
ret, frame = cap.read()
# NOWA, LEPSZA POPRAWKA: Sprawdzenie poprawności klatki po odczycie
if not ret or frame is None or frame.size == 0 or frame.shape[0] == 0 or frame.shape[1] == 0:
print(f"Ostrzeżenie: Nie można odczytać prawidłowej klatki {frame_index} dla {video_path}. Koniec wideo lub błąd.")
break
decoded_objects = pyzbar.decode(frame)
for obj in decoded_objects:
try:
qr_data = obj.data.decode('utf-8')
# Upewnij się, że parse_gopro_qr_timecode jest zdefiniowane
qr_timestamp = parse_gopro_qr_timecode(qr_data)
first_qr_time = qr_timestamp
first_qr_frame_index = frame_index
cap.release()
return first_qr_time, first_qr_frame_index
except ValueError as e:
# print(f"Ostrzeżenie: Nieprawidłowy kod QR: {e}") # Można włączyć dla debugowania
continue
frame_index += 1
cap.release()
print(f"Nie znaleziono prawidłowego kodu QR w pierwszych {max_frames_to_scan} klatkach {video_path}.")
return None, -1
def _read_ocr_timecode_from_video(self, video_path: str, frame_rate: float) -> tuple[datetime.datetime | None, int]:
"""
Odczytuje wizualny timecode (np. z planszy lub wyświetlacza) z wideo za pomocą OCR.
Weryfikuje zgodność liczby klatek na zegarze z klatkażem wideo.
Jeśli wykryto niezgodność, ufa tylko klatkom z FF=00.
"""
print(f"Szukam wizualnego timecode'u w {video_path}...")
cap = cv2.VideoCapture(video_path)
if not cap.isOpened():
print(f"Błąd: Nie można otworzyć pliku wideo {video_path}")
return None, -1
frame_index = 0
max_frames_to_scan = int(frame_rate * SCAN_DURATION_SECONDS)
if max_frames_to_scan < 50:
max_frames_to_scan = 50
# Zmienne do przechowywania potencjalnego timecode'u (gdy nie mamy jeszcze FF=00)
potential_time = None
potential_frame_index = -1
# Zmienne do weryfikacji zgodności liczby klatek
expected_max_frame = int(frame_rate) - 1 # Oczekiwana maksymalna wartość FF (np. 49 dla 50fps)
detected_frames = set() # Zbiór wykrytych numerów klatek (FF)
trust_only_zero_frames = False # Flaga ustawiana na True, gdy wykryto niezgodność
# Licznik wykrytych timecode'ów do analizy
timecode_count = 0
max_timecodes_to_analyze = 100 # Maksymalna liczba timecode'ów do analizy przed podjęciem decyzji
# OCR może wymagać różnych konfiguracji w zależności od jakości obrazu.
tesseract_configs = [
'--psm 6 -c tessedit_char_whitelist=0123456789:.',
'--psm 7 -c tessedit_char_whitelist=0123456789:.',
'--psm 8 -c tessedit_char_whitelist=0123456789:.'
]
while frame_index < max_frames_to_scan:
ret, frame = cap.read()
if not ret or frame is None or frame.size == 0 or frame.shape[0] == 0 or frame.shape[1] == 0:
print(f"Ostrzeżenie: Nie można odczytać prawidłowej klatki {frame_index} dla {video_path}. Koniec wideo lub błąd.")
break
# Konwersja klatki na skalę szarości dla lepszej wydajności OCR
gray_frame = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
# W pętli po kilku konfiguracjach, aby zwiększyć szansę na sukces
for config in tesseract_configs:
try:
ocr_text = pytesseract.image_to_string(gray_frame, config=config)
# Oczyszczenie tekstu z białych znaków i nadmiarowych linii
cleaned_text = ocr_text.strip().replace(" ", "").replace("\n", "")
# Wyszukiwanie wzorca HH:MM:SS:FF
timecode_match = re.search(r'\d{2}:\d{2}:\d{2}:\d{2}', cleaned_text)
if timecode_match:
timecode_string = timecode_match.group(0)
print(f"Znaleziono wizualny timecode w klatce {frame_index}: {timecode_string}")
# Sprawdzamy, czy timecode zawiera FF=00
frame_number_match = re.search(r'\d{2}:\d{2}:\d{2}:(\d{2})', timecode_string)
if frame_number_match:
frame_number = int(frame_number_match.group(1))
# Dodajemy wykryty numer klatki do zbioru (do analizy)
detected_frames.add(frame_number)
timecode_count += 1
# Jeśli zebraliśmy wystarczająco dużo próbek, analizujemy zgodność
if timecode_count >= max_timecodes_to_analyze:
max_detected_frame = max(detected_frames)
# Sprawdzamy, czy maksymalna wykryta wartość FF jest zgodna z klatkażem
if max_detected_frame > expected_max_frame:
print(f"OSTRZEŻENIE: Wykryto niezgodność liczby klatek! Maksymalna wykryta wartość FF: {max_detected_frame}, oczekiwana: {expected_max_frame}")
trust_only_zero_frames = True
print(f"Przełączam na tryb zaufania tylko klatkom z FF=00")
elif len(detected_frames) < expected_max_frame * 0.5: # Wykryto mniej niż połowę oczekiwanych wartości FF
print(f"OSTRZEŻENIE: Wykryto zbyt mało różnych wartości FF: {len(detected_frames)}/{expected_max_frame+1}")
trust_only_zero_frames = True
print(f"Przełączam na tryb zaufania tylko klatkom z FF=00")
try:
# Parsujemy timecode
hour = int(timecode_string[0:2])
minute = int(timecode_string[3:5])
second = int(timecode_string[6:8])
# Jeśli wykryto niezgodność lub FF!=00, ignorujemy część ułamkową sekundy
microseconds = 0
if not trust_only_zero_frames and frame_number != 0:
# Obliczamy część ułamkową sekundy na podstawie numeru klatki
frame_fraction = frame_number / frame_rate
microseconds = int(frame_fraction * 1_000_000)
# Tworzenie obiektu datetime
now = datetime.datetime.now()
naive_dt = datetime.datetime.combine(now.date(), datetime.time(hour, minute, second, microsecond=microseconds))
calculated_time = pytz.utc.localize(naive_dt)
# Jeśli wykryto niezgodność, ufamy tylko klatkom z FF=00
if trust_only_zero_frames and frame_number != 0:
# Zapisujemy potencjalny wynik, ale szukamy dalej klatki FF=00
if potential_time is None:
potential_time = calculated_time
potential_frame_index = frame_index
print(f"Zapisano potencjalny timecode z klatki {frame_index}, ale szukam dalej klatki z FF=00")
else:
# Znaleźliśmy odpowiednią klatkę - zwracamy wynik
cap.release()
return calculated_time, frame_index
except ValueError as e:
print(f"Ostrzeżenie: Nie udało się sparsować timecode '{timecode_string}': {e}")
continue
except Exception as e:
# Błąd OCR nie powinien zatrzymywać programu
print(f"Ostrzeżenie: Wystąpił błąd podczas OCR na klatce {frame_index}: {e}")
continue
frame_index += 1
cap.release()
# Jeśli nie znaleźliśmy odpowiedniej klatki, ale mamy potencjalny wynik
if potential_time is not None:
print(f"Nie znaleziono idealnej klatki, używam najlepszego dopasowania z klatki {potential_frame_index}")
return potential_time, potential_frame_index
print(f"Nie znaleziono wizualnego timecode'u w pierwszych {max_frames_to_scan} klatkach {video_path}.")
return None, -1
def _add_ltc_track_to_video(self, video_path: str, start_datetime_utc: datetime.datetime, frame_rate: float, duration_seconds: float) -> bool:
"""Dodaje ścieżkę audio z sygnałem (LTC) do wideo za pomocą ffmpeg."""
relative_to_input = os.path.relpath(video_path, start=self.input_base_dir)
output_sub_dir = os.path.join(self.output_base_dir, os.path.dirname(relative_to_input))
os.makedirs(output_sub_dir, exist_ok=True)
base_name, ext = os.path.splitext(os.path.basename(video_path))
output_filename = f"{base_name}_LTC{ext}"
output_path = os.path.join(output_sub_dir, output_filename)
//----//
temp_ltc_audio_file = os.path.join(os.path.dirname(output_path), f"temp_ltc_{base_name}.wav")
if start_datetime_utc is None or duration_seconds is None or frame_rate is None:
print(f"Błąd (w _add_ltc_track_to_video): Brak wymaganych danych (czas rozpoczęcia, czas trwania lub klatkaż) do wygenerowania audio dla {video_path}. Pomijam generowanie audio.")
return False
if not isinstance(duration_seconds, (int, float)) or not isinstance(frame_rate, (int, float)):
print(f"Błąd (w _add_ltc_track_to_video): Czas trwania ({duration_seconds}) lub klatkaż ({frame_rate}) nie jest liczbą dla {video_path}. Pomijam generowanie audio.")
return False
if not generate_ltc_audio_file(start_datetime_utc, duration_seconds, frame_rate, temp_ltc_audio_file):
return False
# Komenda FFmpeg do dodawania ścieżki audio
command = [
'ffmpeg',
'-i', video_path,
'-i', temp_ltc_audio_file,
'-map', '0:v:0', # Mapuje pierwszą ścieżkę wideo z wejścia 0 (oryginalne wideo)
'-map', '0:a?', # Mapuje wszystkie istniejące ścieżki audio z wejścia 0, jeśli są
'-map', '1:a:0', # Mapuje pierwszą ścieżkę audio z wejścia 1 (nasz nowo wygenerowany LTC)
'-c:v', 'copy', # Kopiuje strumień wideo bez rekompresji
'-c:a', 'copy', # Kopiuje istniejące strumienie audio bez rekompresji
'-c:a:1', 'pcm_s16le', # Koduje NOWĄ (drugą, jeśli była oryginalna) ścieżkę audio (nasz LTC) jako PCM 16-bit Little-Endian
# UWAGA: Index `1` dla `-c:a:1` oznacza, że ta opcja będzie dotyczyć mapowanego strumienia audio z indeksu 1 (czyli `1:a:0`)
'-shortest', # Kończy kodowanie, gdy najkrótszy strumień się skończy
'-y', # Nadpisuje plik wyjściowy bez pytania
'-metadata', f"creation_time={start_datetime_utc.isoformat(timespec='milliseconds').replace('+00:00', 'Z')}", # Dodaje metadane czasu utworzenia
output_path
]
print(f"FFmpeg command (final): {' '.join(command)}")
try:
# Użycie subprocess.run z przekierowaniem wyjścia dla lepszego debugowania
result = subprocess.run(command, check=True, capture_output=True, text=True, encoding='utf-8')
print(f"Pomyślnie dodano sygnał audio (LTC) do {video_path}. Plik zapisano jako {output_path}")
if result.stdout:
print("FFmpeg stdout (fragment):\n", result.stdout[-500:]) # Ostatnie 500 znaków
if result.stderr:
print("FFmpeg stderr (fragment):\n", result.stderr[-500:]) # Ostatnie 500 znaków
return True
except subprocess.CalledProcessError as e:
print(f"Błąd FFmpeg podczas dodawania audio do {video_path}: {e}")
print("FFmpeg stdout (pełny):\n", e.stdout)
print("FFmpeg stderr (pełny):\n", e.stderr)
return False
except FileNotFoundError:
print("Błąd: ffmpeg nie znaleziono. Upewnij się, że jest zainstalowany i dostępny w PATH.")
return False
except Exception as e:
print(f"Wystąpił nieoczekiwany błąd podczas wywołania FFmpeg dla {video_path}: {e}")
traceback.print_exc()
return False
finally:
if os.path.exists(temp_ltc_audio_file):
os.remove(temp_ltc_audio_file)
print(f"Usunięto tymczasowy plik audio (LTC): {temp_ltc_audio_file}")
def process_video(self, video_path: str) -> bool:
"""
Przetwarza pojedynczy plik wideo, aby dodać ścieżkę audio (LTC).
Najpierw szuka kodu QR, a w przypadku niepowodzenia próbuje odczytać timecode za pomocą OCR.
"""
print(f"Przetwarzanie: {video_path}")
try:
duration_seconds, frame_rate = self._get_video_info(video_path)
# KROK 1: Próba odczytu kodu QR
calculated_start_time_utc, qr_frame_index = self._read_qr_from_video(video_path, frame_rate)
if calculated_start_time_utc:
print(f"Znaleziono QR kod w klatce {qr_frame_index}: {calculated_start_time_utc.isoformat()}")
print(f"Obliczony czas rozpoczęcia wideo (UTC): {calculated_start_time_utc}")
return self._add_ltc_track_to_video(video_path, calculated_start_time_utc, frame_rate, duration_seconds)
else:
# KROK 2: Jeśli QR nie znaleziono, próba odczytu wizualnego timecode'u przez OCR
print("Próba odczytu wizualnego timecode'u...")
calculated_start_time_utc, ocr_frame_index = self._read_ocr_timecode_from_video(video_path, frame_rate)
if calculated_start_time_utc:
print(f"Znaleziono wizualny timecode w klatce {ocr_frame_index}: {calculated_start_time_utc.isoformat()}")
return self._add_ltc_track_to_video(video_path, calculated_start_time_utc, frame_rate, duration_seconds)
else:
print(f"Pomijanie {video_path}: Nie znaleziono ani kodu QR, ani wizualnego timecode'u.")
return False
except ValueError as e:
print(f"Wystąpił błąd podczas pobierania informacji o wideo dla {video_path}: {e}")
return False
except Exception as e:
print(f"Wystąpił nieoczekiwany błąd podczas przetwarzania {video_path}: {e}")
traceback.print_exc()
return False