import os
import cv2
import numpy as np


def joint(or_path, tar_path, size):
  determination = tar_path
  if not os.path.exists(determination):
    os.makedirs(determination)

  path = or_path
  folders = os.listdir(path)
  folders_name = []
  folders.sort(key = lambda x: int(x.split('.')[0]))
  for folder in folders:
    folders_name.append(path + "\\" + str(folder))
  foldler_len = len(folders_name)
  joint = []
  for i in range(size):
    joint.append(cv2.imread(folders_name[i]))
  index = 0
  for i in range(len(folders_name)):
    if i < size:
      continue
    image = cv2.imread(folders_name[i])
    joint[index] = np.hstack((joint[index], image))
    index += 1
    if index > size - 1:
      index = 0
  for i in range(1, len(joint)):
    joint[0] = np.vstack((joint[0], joint[i]))
  cv2.imwrite(tar_path + "/all.jpg", joint[0])
  print(joint[0].shape)


if __name__ == "__main__":
  joint(r"", r"", size)#原图像文件夹路径、新图像存放路径、单方向图像数