yukari/flandre
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update legacy file process

Browse Source
This commit is contained in:
flandre 2025-04-26 18:28:03 +08:00
parent 1e23ce4721
commit 06589d7f33
2 changed files with 146 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
flandre/pyqt/FFmpegReceiver.py
View File

@ -18,7 +18,7 @@ class FFmpegReceiver(QtCore.QObject):
'-loglevel', 'quiet',
'-flags', 'low_delay',
'-fflags', 'nobuffer',
'-i', 'rtsp://admin:ab12ab12@11.6.2.4:554/h264/ch1/main/av_stream',
'-i', 'rtsp://admin:ab12ab12@11.6.2.5:554/h264/ch1/main/av_stream',
'-pix_fmt', 'rgb24',
'-f', 'rawvideo',
'-'

146
test/process_leagacy.py
View File

@ -1,6 +1,9 @@
import hashlib
import shutil
from pathlib import Path
from flandre.utils.RfMeta import RfFrameMeta, RfSequenceMeta
from flandre.utils.RfSequence import RfSequence
from flandre.utils.archive import to_zip
@ -30,6 +33,7 @@ def f1():
to_zip(arg, tempdst, Path('/mnt/16T/private_dataset/us/steel-top,U=30,M=PWI,S=(256 1502).zip'))
def f2():
tempdst = Path('/mnt/16T/private_dataset/ustemp')
src = Path('/mnt/16T/private_dataset/New Folder/T1,U=30,M=FMC,S=(256 256 1502)/')
@ -43,5 +47,145 @@ def f2():
arr.sort(key=lambda item: (item[1].robot_y, item[1].robot_x))
print(arr)
WRITE_TMP = Path('/run/media/lambda/040fb2b8-1584-4683-bac8-fec3b264167d/write_cache/')
def proc(src: Path | str, dst: Path | str, fx, tmp=WRITE_TMP):
src = Path(src)
dst = Path(dst)
assert not dst.exists()
shutil.rmtree(tmp)
tmp.mkdir(parents=True)
arr = fx(src)
print(arr)
# to_zip(arr, tmp, dst)
def f_ok(src: Path) -> list[tuple[Path, RfFrameMeta, list[tuple[Path, str]]]]:
arr: list[tuple[Path, RfFrameMeta, list[tuple[Path, str]]]] = []
for p in src.glob('*'):
m = RfFrameMeta.from_name(p.stem)
arr.append((p, m, []))
arr.sort(key=lambda item: item[1].sequence_id)
return arr
def f_x(src: Path) -> list[tuple[Path, RfFrameMeta, list[tuple[Path, str]]]]:
arr: list[tuple[Path, RfFrameMeta, list[tuple[Path, str]]]] = []
for p in src.glob('*'):
m = RfFrameMeta.from_name(p.stem)
arr.append((p, m, []))
arr.sort(key=lambda item: item[1].robot_x)
for i, a in enumerate(arr):
a[1].sequence_id = i
a[1].encoder = 0
return arr
def f_xy(src: Path) -> list[tuple[Path, RfFrameMeta, list[tuple[Path, str]]]]:
arr: list[tuple[Path, RfFrameMeta, list[tuple[Path, str]]]] = []
for p in src.glob('*'):
m = RfFrameMeta.from_name(p.stem)
arr.append((p, m, []))
arr.sort(key=lambda item: (item[1].robot_y, item[1].robot_x))
for i, a in enumerate(arr):
a[1].sequence_id = i
a[1].encoder = 0
return arr
def f_legacy4(src: Path, t, xx=100, yy=100) -> list[tuple[Path, RfFrameMeta, list[tuple[Path, str]]]]:
arr = []
for p in src.glob(f'*{t}.bin'):
x, y, _ = p.name.split('_')
r = RfFrameMeta(encoder=0, robot_x=int(x) * xx, robot_y=int(y) * yy)
arr.append((p, r))
arr.sort(key=lambda item: (item[1].robot_y, item[1].robot_x))
arg: list[tuple[Path, RfFrameMeta, list[tuple[Path, str]]]] = []
for i, item in enumerate(arr):
file, meta = item
meta.sequence_id = i
pic = file.with_suffix('.png')
farr = []
if pic.exists():
farr.append((pic, '.png'))
arg.append((file, meta, farr))
return arg
def f_legacy4_pwi(src: Path) -> list[tuple[Path, RfFrameMeta, list[tuple[Path, str]]]]:
return f_legacy4(src, 'pwi')
def f_legacy4_tfm(src: Path) -> list[tuple[Path, RfFrameMeta, list[tuple[Path, str]]]]:
return f_legacy4(src, 'fmc')
def f_legacy4_pwi_y10(src: Path) -> list[tuple[Path, RfFrameMeta, list[tuple[Path, str]]]]:
return f_legacy4(src, 'pwi', yy=10)
def f_legacy4_tfm_y10(src: Path) -> list[tuple[Path, RfFrameMeta, list[tuple[Path, str]]]]:
return f_legacy4(src, 'fmc', yy=10)
def f_cap(src: Path) -> list[tuple[Path, RfFrameMeta, list[tuple[Path, str]]]]:
arr = []
for p in src.glob(f'*parallel.bin'):
y, _ = p.name.split('_')
r = RfFrameMeta(encoder=0, robot_y=int(y) * 100)
arr.append((p, r))
arr.sort(key=lambda item: item[1].robot_y)
arg: list[tuple[Path, RfFrameMeta, list[tuple[Path, str]]]] = []
for i, item in enumerate(arr):
file, meta = item
meta.sequence_id = i
arg.append((file, meta, []))
return arg
def f_legacy_unknown(src: Path, t) -> list[tuple[Path, RfFrameMeta, list[tuple[Path, str]]]]:
arr = []
for p in src.glob(f'*{t}.bin'):
x, y, _ = p.name.split('_')
r = RfFrameMeta(encoder=0, robot_x=int(x) * xx, robot_y=int(y) * yy)
arr.append((p, r))
arr.sort(key=lambda item: (item[1].robot_y, item[1].robot_x))
arg: list[tuple[Path, RfFrameMeta, list[tuple[Path, str]]]] = []
for i, item in enumerate(arr):
file, meta = item
meta.sequence_id = i
pic = file.with_suffix('.png')
farr = []
if pic.exists():
farr.append((pic, '.png'))
arg.append((file, meta, farr))
return arg
if __name__ == '__main__':
f2()
b2b = hashlib.blake2b(
Path('/run/media/lambda/040fb2b8-1584-4683-bac8-fec3b264167d/c1/remilia/cap1/34_4040_fmc.bin').read_bytes(),
digest_size=4).hexdigest()
print(b2b)
print('--')
for f in Path('/mnt/16T/private_dataset/us/').glob('*.zip'):
m = RfSequenceMeta.from_path(f)
if m.mode == RfSequenceMeta.RfSequenceMode.TFM and m.us == 30:
for f in RfSequence(f).frames:
print(m.commit, f.meta.sequence_id, f.meta.encoder, f.meta.robot_x, f.meta.robot_y, f.meta.blake2b)
# RfSequence()
# proc(
# '/run/media/lambda/040fb2b8-1584-4683-bac8-fec3b264167d/c2/Desktop/New folder/4d/cap1',
# '/mnt/16T/private_dataset/us/cap1,U=30,M=PWI,S=(256 1502).zip',
# f_legacy_unknown,
# )