import glob
import os
import shutil
import cv2

def merge_images(input_folder, output_folder):
    if not os.path.exists(output_folder):
        os.mkdir(output_folder)

    count = 0
    for root, _, filenames in os.walk(input_folder):
        for fn in filenames:
            if fn.endswith('.jpg'):
                count += 1
                shutil.copy(os.path.join(root, fn), os.path.join(output_folder, 'c_{:010d}.jpg'.format(count)))


if __name__ == '__main__':

    merge_images('p/beach', 'output')

    for f in glob.glob(os.path.join('/home/logic/test/data/1xx_Beach', 'c_*.jpg')):
        os.remove(f)

    for f in glob.glob('output/*.jpg'):
        img = cv2.imread(f)
        if img is None:
            continue
        shutil.copy(f, '/home/logic/test/data/1xx_Beach')
    for f in glob.glob('output/*.jpg'):
        os.remove(f)
    print('beach done')

    merge_images('p/building', 'output')
    for f in glob.glob('output/*.jpg'):
        img = cv2.imread(f)
        if img is None:
            continue
        shutil.copy(f, '/home/logic/test/data/2xx_Building')
    for f in glob.glob('output/*.jpg'):
        os.remove(f)
    print('building done')
    
    merge_images('p/bus', 'output')
    for f in glob.glob('output/*.jpg'):
        img = cv2.imread(f)
        if img is None:
            continue
        shutil.copy(f, '/home/logic/test/data/3xx_Bus')
    for f in glob.glob('output/*.jpg'):
        os.remove(f)
    print('bus done')
    
    merge_images('p/dinosaur', 'output')
    for f in glob.glob('output/*.jpg'):
        img = cv2.imread(f)
        if img is None:
            continue
        shutil.copy(f, '/home/logic/test/data/4xx_Dinosaur')
    for f in glob.glob('output/*.jpg'):
        os.remove(f)
    print('dinosaur done')
    
    merge_images('p/elephant', 'output')
    for f in glob.glob('output/*.jpg'):
        img = cv2.imread(f)
        if img is None:
            continue
        shutil.copy(f, '/home/logic/test/data/5xx_Elephant')
    for f in glob.glob('output/*.jpg'):
        os.remove(f)
    print('elephant done')
    
    merge_images('p/horse', 'output')
    for f in glob.glob('output/*.jpg'):
        img = cv2.imread(f)
        if img is None:
            continue
        shutil.copy(f, '/home/logic/test/data/7xx_Horse')
    for f in glob.glob('output/*.jpg'):
        os.remove(f)
    print('horse done')
    
    merge_images('p/mountain', 'output')
    for f in glob.glob('output/*.jpg'):
        img = cv2.imread(f)
        if img is None:
            continue
        shutil.copy(f, '/home/logic/test/data/8xx_Mountain')
    for f in glob.glob('output/*.jpg'):
        os.remove(f)
    print('mountain done')
    
    merge_images('p/dish', 'output')
    for f in glob.glob('output/*.jpg'):
        img = cv2.imread(f)
        if img is None:
            continue
        shutil.copy(f, '/home/logic/test/data/9xx_Dish')
    for f in glob.glob('output/*.jpg'):
        os.remove(f)
    print('dish done')

    print('done')