flandre/test/legacy/main.py
2025-05-11 15:50:08 +08:00

76 lines
2.2 KiB
Python

import base64
import threading
import time
from pathlib import Path
import cv2
import numpy as np
from tqdm import tqdm
import zmq
if __name__ == '__main__':
# r1 = base64.b64encode(Path('/home/lambda/Pictures/drawing.png').read_bytes()).decode()
# r1 = base64.b64encode(Path('/home/lambda/Pictures/remilia3.png').read_bytes()).decode()
# requests.post('http://localhost:12345/sendpic/test', json=dict(value='data:image/png;base64, ' + r1))
# for p in tqdm(Path('/home/lambda/Videos/pngs/New Folder/').glob('*.png')):
# r1 = base64.b64encode(p.read_bytes()).decode()
# requests.post('http://localhost:12345/sendpic/test', json=dict(value='data:image/png;base64, ' + r1))
# arr = [cv2.imread(str(img)).tobytes() for img in Path('/home/lambda/Videos/pngs/New Folder/').glob('*.png')]
arr = []
for img in tqdm(list(Path('/home/lambda/Videos/pngs/New Folder/').glob('*.png'))):
img = cv2.imread(str(img))
# img = cv2.resize(img, (1920 // 2, 1080 // 2))
img = img.reshape(1080, 1920, 3)
z = np.zeros((1080, 1920, 4), dtype=np.uint8)
z[:, :, :3] = img
img = z
img = cv2.cvtColor(img, cv2.COLOR_BGRA2RGBA)
arr.append(img.tobytes())
# img = cv2.resize(img, (1920 // 4, 1080 // 4,))
# while True:
# for p in Path('/home/lambda/Videos/pngs/New Folder/').glob('*.png'):
# img = cv2.imread(str(p))
# # print(img.shape)
# input()
# socket.send(img.tobytes())
# while True:
# for b in arr:
# input()
# socket.send(b)
lii = [0, True]
def t(li):
context = zmq.Context()
socket = context.socket(zmq.PUSH)
socket.connect("tcp://localhost:5555")
while True:
for i, b in enumerate(arr):
while True:
socket.send(arr[li[0]])
time.sleep(1 / 120)
if li[1]:
li[0] = i
break
threading.Thread(target=t, args=(lii,)).start()
# while True:
# for i, b in enumerate(arr):
# # input()
# lii[0] = i
# time.sleep(1 / 60)
while True:
input()
lii[1] = not lii[1]