"""
@file: generate_list_cedar.py
@time: 2018/6/20 15:03
@desc:Create the paris list of CEDAR Database

"""

import copy
import os
import sys

import imageio
import numpy as np

num_genuine = 24
num_forged = 24


# 生成数组l的全部组合（长度k）
def combine(l, k):
    answers = []
    one = [0] * k

    def next_c(li=0, ni=0):
        if ni == k:
            answers.append(copy.copy(one))
            return
        for lj in range(li, len(l)):
            one[ni] = l[lj]
            next_c(lj + 1, ni + 1)

    next_c()
    return answers


# 生成两个数组间的全部组合
def combine_2list(list1, list2):
    answers = []
    for i1 in list1:
        for i2 in list2:
            answers.append([i1, i2])
    return answers


def generate_list(train_size, listfile_name):
    signers_list = list(range(1, 56))
    list_file_train = open(listfile_name + '_train.txt', 'w')
    list_file_test = open(listfile_name + '_val.txt', 'w')

    train_indexs = np.arange(0, len(signers_list), 1)
    np.random.shuffle(train_indexs)
    train_indexs = train_indexs[:train_size]

    for i, signer in enumerate(signers_list):
        list_file = list_file_train if i in train_indexs else list_file_test
        genuine_genuine_suf = combine(list(range(1, num_genuine + 1)), 2)
        for item in genuine_genuine_suf:
            genuine0 = "%s%d_%d%s" % ('full_org/original_', int(signer), item[0], '.png')
            genuine1 = "%s%d_%d%s" % ('full_org/original_', int(signer), item[1], '.png')
            line = genuine0 + ' ' + genuine1 + ' 1\n'
            list_file.write(line)

        genuine_forged_suf = combine_2list(list(range(1, num_genuine + 1)), list(range(1, num_forged + 1)))
        for item in genuine_forged_suf:
            genuine = "%s%d_%d%s" % ('full_org/original_', int(signer), item[0], '.png')
            forged = "%s%d_%d%s" % ('full_forg/forgeries_', int(signer), item[1], '.png')
            line = genuine + ' ' + forged + ' 0\n'
            list_file.write(line)

    list_file_train.close()
    list_file_test.close()


def tif_to_jpg(tif_dir, jpg_dir):
    for root, dirs, files in os.walk(tif_dir):
        to_dir = root.replace(tif_dir, jpg_dir)
        if not os.path.exists(to_dir):
            os.mkdir(to_dir)
        for file in files:
            if not file.endswith('.tif'):
                continue
            image = imageio.imread(os.path.join(root, file))
            jpg_file = file.replace('.tif', '.jpg')
            imageio.imwrite(os.path.join(to_dir, jpg_file), image)


def main(argv=None):
    if argv is None:
        argv = sys.argv

    generate_list(50, '../experiments/data_list/cedar')
    # generate_list(100, '../experiments/data_list/bhsig260_Hindi')


if __name__ == "__main__":
    sys.exit(main())