# coding=utf-8
import os

from PIL import Image

import my_img_process as mip
import numpy as np
from util import fileutil


def show_bws(dir_name, min_brightness=128, n1=1, n2=100, line_width=4):
    if not dir_name:
        return

    fs = os.listdir(dir_name)
    mats = []

    count = 0
    for f in fs:
        count += 1
        if count < n1:
            continue

        f_name = dir_name + f
        print f
        with Image.open(f_name) as img:
            mat = mip.img_to_binary_mat(
                img,
                minumum_brightness=min_brightness,
                brighter=False)

            mip.remove_singular_noise(
                mat, np.array(([1, 1, 1], [1, 2, 1], [1, 1, 1])))
            mats.append(mat)

        if count >= n2:
            break

    mip.show_mats(mats, line_width=line_width)


def splitchars(splitter, src_dir, tgt_dir, required_char_num=None):
    files = os.listdir(src_dir)
    startd = 1
    for f in files:
        print f
        fname = os.path.join(src_dir, f)
        fcontent = fileutil.read_file_content(fname)
        mats = splitter.split(fcontent)
        if required_char_num and len(mats) < required_char_num:
            raise Exception('char num less than required')
        mip.save_mats_as_imgs(mats, tgt_dir, 'chars_', startd)
        startd += len(mats)

if __name__ == '__main__':
    dir_name = 'F:/shanghai_si_captchas/'
    show_bws(dir_name, 195, 101, 112, 4)