minicap解析.zip

# coding:utf-8 import socket import threading from Banner import Banner from queue import Queue import datetime import cv2 as cv import numpy as np from itsdangerous import want_bytes class MinicapStream: __instance = None __mutex = threading.Lock() def __init__(self): # 定义IP self.IP = "127.0.0.1" # 定义端口 self.PORT = 1717 # 进程的pid self.Pid = 0 # 存放banner头部信息 self.banner = Banner() self.minicapSocket = None self.ReadImageStreamTask = None self.push = None self.client = None self.picture = Queue() def __new__(cls, *args, **kwargs): if not cls.__instance: cls.__instance = super().__new__(cls, *args, **kwargs) return cls.__instance def get_d(self): print(self.picture.qzise()) def run(self): # 创建socket连接并启动 # 定义socket类型 self.ReadImageStreamTask = threading.Thread(target=self.ReadImageStream).start() def ReadImageStream(self): readBannerBytes = 0 bannerLength = 2 readFrameBytes = 0 frameBodylength = 0 dataBody = b"" if self.client == None: self.client = socket.socket() self.client.connect((self.IP, self.PORT)) while True: # 获取当前系统时间 now_time = datetime.datetime.now() time1_str = datetime.datetime.strftime(now_time, '%Y-%m-%d %H:%M:%S') # print(time1_str) data = self.client.recv(4096) # print(data) length = len(data) if not length: continue cursor = 0 while cursor < length: # just do it if readBannerBytes < bannerLength: # byte10 = data[cursor] & 0xff if readBannerBytes == 0: self.banner.Version = want_bytes(data[cursor]) elif readBannerBytes == 1: bannerLength = want_bytes(data[cursor]) self.banner.Length = bannerLength elif readBannerBytes in [2, 3, 4, 5]: self.banner.Pid += (want_bytes(data[cursor]) << ((readBannerBytes - 2) * 8)) >> 0 elif readBannerBytes in [6, 7, 8, 9]: self.banner.RealWidth += (want_bytes(data[cursor]) << ((readBannerBytes - 6) * 8)) >> 0 elif readBannerBytes in [10, 11, 12, 13]: self.banner.RealHeight += (want_bytes(data[cursor]) << ((readBannerBytes - 10) * 8)) >> 0 elif readBannerBytes in [14, 15, 16, 17]: self.banner.VirtualWidth += (want_bytes(data[cursor]) << ((readBannerBytes - 14) * 8)) >> 0 elif readBannerBytes in [18, 19, 20, 21]: self.banner.VirtualHeight += (want_bytes(data[cursor]) << ((readBannerBytes - 18) * 8)) >> 0 elif readBannerBytes == 22: self.banner.Orientation = want_bytes(data[cursor]) * 90 elif readBannerBytes == 23: self.banner.Quirks = want_bytes(data[cursor]) cursor += 1 readBannerBytes += 1 if readBannerBytes == bannerLength: print(self.banner.toString()) elif readFrameBytes < 4: frameBodylength = frameBodylength + ((want_bytes(data[cursor]) << (readFrameBytes * 8)) >> 0) cursor += 1 readFrameBytes += 1 else: if length - cursor >= frameBodylength: dataBody = dataBody + data[cursor:(cursor + frameBodylength)] if want_bytes(dataBody[0]) != 0xFF or want_bytes(dataBody[1]) != 0xD8: break self.picture.put(dataBody) now_time = datetime.datetime.now() time1_str = datetime.datetime.strftime(now_time, '%Y%m%d%H%M%S') self.save_file('d:/minicapimg/'+time1_str+'.png', dataBody) cursor += frameBodylength frameBodylength = 0 readFrameBytes = 0 dataBody = b"" else: dataBody = dataBody + data[cursor:length] frameBodylength -= length - cursor readFrameBytes += length - cursor cursor = length def save_file(self, file_name, data): file = open(file_name, "wb") file.write(data) file.flush() file.close() if __name__ == '__main__': a = MinicapStream() a.run()