import cv2
import argparse
import os
import re

'''
读取txt文件信息，对图片进行标注，并合成视频
txt格式:
img0001 2 (45,56,60,80) (52,86,60,90)
img0001 1 (45,56,60,80)
...
'''

pathIn = "00009"  # 需要合成的图片文件夹
pathOut = 'shop001.avi'  # 输出视频文件保存路径
result = open('009/groundTruth.txt', 'r')  # 读取txt文件
array_of_img = []

# 遍历pathIn文件夹内所有图片，并保存到数组中
for filename in os.listdir(pathIn):
    img = cv2.imread(pathIn + "/" + filename)
    array_of_img.append(img)

# 设置输出视频参数
fps = 10
size = (360, 240)
videoWriter = cv2.VideoWriter(pathOut, cv2.VideoWriter_fourcc('M', 'J', 'P', 'G'), fps, size)

# 遍历遍历数组里面的图片，根据txt文件信息进行标注后，保存成视频
k = 0
for line in result:
    kk = re.compile(r'[(](.*?)[)]')  # 正则表达式，读取"()"内的文本
    arr = kk.findall(line)  # 逐行读取
    img = array_of_img[k]
    for j in arr:
        rect = j.split(' ')  # 将一段长字符串按空格区分成若干子字符串
        c_x = int(rect[0])
        c_y = int(rect[1])
        bw = int(rect[2])
        bh = int(rect[3])
        cv2.rectangle(img, (c_x, c_y), (bw, bh), (0, 255, 0), 1)
    videoWriter.write(img)
    cv2.imshow("image", img)
    cv2.waitKey(100)
    k += 1