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
|
#!/usr/bin/env python3
import os
import re
import subprocess
import tempfile
import sys
from datetime import datetime, timedelta
from argparse import ArgumentParser
input_fmt = '%Y-%m-%d-%H:%M:%S.%f'
output_fmt = '%Y-%m-%d-%H:%M:%S'
# declare types
File = dict
FileList = list[File]
def get_files(source_directory: str) -> FileList:
files = []
for f in os.listdir(source_directory):
# 2022-06-15-00:02:40.187877.jpg
m = re.match(r'(^\d{4}-\d{2}-\d{2}-\d{2}:\d{2}:\d{2}\.\d{1,6})\.jpg$', f)
if not m:
continue
files.append({
'filename': os.path.join(source_directory, f),
'time': datetime.strptime(m.group(1), input_fmt),
})
files.sort(key=lambda f: f['time'].timestamp())
return files
def group_files(files: FileList) -> list[FileList]:
groups = []
group_idx = None
for file in files:
if group_idx is None or \
not groups[group_idx] or \
file['time'] - groups[group_idx][-1]['time'] <= timedelta(seconds=10):
if group_idx is None:
groups.append([])
group_idx = 0
else:
group_idx += 1
groups.append([])
groups[group_idx].append(file)
return groups
def merge(groups: list[FileList],
output_directory: str,
delete_source_files=False,
cedrus=False) -> None:
for g in groups:
success = False
fd = tempfile.NamedTemporaryFile(delete=False)
try:
n = len(g)
last_frame_dur = 0
for i in range(n):
file = g[i]
fd.write(f'file \'{file["filename"]}\'\n'.encode())
if i < n-1:
last_frame_dur = g[i+1]['time'].timestamp()-file['time'].timestamp()
fd.write(f'duration {last_frame_dur}\n'.encode())
fd.close()
print(f'temp concat file: {fd.name}')
start = g[0]['time'].strftime(output_fmt)
stop = g[-1]['time'].strftime(output_fmt)
fn = f'{start}_{stop}_merged.mp4'
output = os.path.join(output_directory, fn)
if cedrus:
ffmpeg = '/home/user/.local/bin/ffmpeg-cedrus'
env = dict(os.environ)
env['LD_LIBRARY_PATH'] = '/usr/local/lib'
args = ['-c:v', 'cedrus264',
'-pix_fmt', 'nv12']
else:
ffmpeg = 'ffmpeg'
env = {}
args = ['-c:v', 'libx264',
'-preset', 'veryslow',
# '-crf', '23',
# '-vb', '448k',
'-pix_fmt', 'yuv420p',
'-filter:v', 'fps=2']
cmd = [ffmpeg, '-y',
'-f', 'concat',
'-safe', '0',
'-i', fd.name,
'-map_metadata', '-1',
*args,
output]
p = subprocess.run(cmd, capture_output=False, env=env)
if p.returncode != 0:
print(f'error: ffmpeg returned {p.returncode}')
else:
success = True
finally:
os.unlink(fd.name)
if success and delete_source_files:
for file in g:
os.unlink(file['filename'])
def print_groups(groups):
for g in groups:
g1 = g[0]
g2 = g[len(g)-1]
print(str(g1['time'])+' .. '+str(g2['time']))
if __name__ == '__main__':
parser = ArgumentParser()
parser.add_argument('--input-directory', '-i', type=str, required=True,
help='Directory with files')
parser.add_argument('--output-directory', '-o', type=str, required=True,
help='Output directory')
parser.add_argument('-D', '--delete-source-files', action='store_true')
parser.add_argument('--cedrus', action='store_true')
# parser.add_argument('--vbr', action='store_true',
# help='Re-encode using VBR (-q:a 4)')
arg = parser.parse_args()
# if arg.cedrus and not os.getegid() == 0:
# raise RuntimeError("Must be run as root.")
files = get_files(os.path.realpath(arg.input_directory))
if not len(files):
print(f"No jpeg files found in {arg.input_directory}.")
sys.exit()
groups = group_files(files)
# print_groups(groups)
merge(groups,
os.path.realpath(arg.output_directory),
delete_source_files=arg.delete_source_files,
cedrus=arg.cedrus)
|
