package com.walker.openocr;

import com.walker.openocr.idcard.IdCardConfig;
import com.walker.openocr.idcard.IdCardObject;
import com.walker.openocr.idcard.IdCardTextResolver;
import com.walker.openocr.table.TableConfig;
import com.walker.openocr.table.TableConfigLoader;
import com.walker.openocr.table.TableObject;
import com.walker.openocr.table.TableTextResolver;
import com.walker.openocr.table.config.FileConfigLoader;
import com.walker.openocr.util.FileReaderUtils;
import com.walker.openocr.util.TableObjectUtils;
import com.walker.openocr.util.TextUtils;
import com.walker.openocr.vehicle.CarTypeValueParser;
import com.walker.openocr.vehicle.IdValueParser;
import com.walker.openocr.vehicle.PlateValueParser;
import com.walker.openocr.vehicle.VehicleLicenseResolver;
import org.junit.Test;

import java.util.ArrayList;
import java.util.List;

public class TableDataTest {

    private String folder = "D:/dev_tools/workspace_2022/walker-dependencies/walker-ml-openocr/src/test/resources";

//    @Test
    public void testValueParser(){
//        String input = "云A853L5";
//        String input = "豫VL120";
//        PlateValueParser parser = new PlateValueParser();
//        System.out.println("pser value = " + parser.getValue(input));

//        String input = "LSVWY4180KN246094";
//        String input = "LSVAK29C8L2969237";
//        IdValueParser idValueParser = new IdValueParser();
//        System.out.println(idValueParser.getValue(input));

//        String input = "小型轿车";
        String input = "车辆类型小型新车";
        CarTypeValueParser carTypeValueParser = new CarTypeValueParser();
        System.out.println(carTypeValueParser.getValue(input));
    }

//    @Test
    public void testVehicleLicense(){
//        String filePath = folder + "/demo_vehicle_01.txt";
        String filePath = folder + "/demo_vehicle_02.txt";
        List<TextBlock> textBlocks = this.acquireTextBlockList(filePath);

        FileConfigLoader vehicleConfig = new FileConfigLoader();
        TableConfig tableConfig = vehicleConfig.load(folder + "/vehicle_license_config.txt");

        List<TableConfig> tableConfigs = new ArrayList<>(4);
        tableConfigs.add(tableConfig);

        VehicleLicenseResolver vehicleLicenseResolver = new VehicleLicenseResolver();
        vehicleLicenseResolver.resolve(textBlocks, tableConfigs);
    }

//    @Test
    public void testFirstMoneyNumber(){
        String input = "总金额为12，006,599.01";
//        String input = "保险费合计（人民币大写)：壹佰贰拾元整 (Y：120.00元)其中救助基金(1.00%）";
//        String input = "RMB284.32元（不含税保费：268.23元，税额：16.09元）（大写）人民币贰佰捌拾";
        String money = TextUtils.parseFirstMoneyNumber(input);
        System.out.println(money);
    }

//    @Test
    public void testStringDateDouble(){
        String input = "自2022年3月14日00：00时起至2023年3月13日24：00时止";
//        String input = "保险期间自2022年08月25日00时00分起至2023年08月24日24时00分止";
        String[] doubleDateNumber = TextUtils.parseDoubleDateWithNumber(input);
        System.out.println(doubleDateNumber[0] + ", " + doubleDateNumber[1]);
    }

//    @Test
    public void TestIdCard(){
//        String filePath = folder + "/idcard_01.txt";
        String filePath = folder + "/idcard_02.txt";
        List<TextBlock> textBlocks = this.acquireTextBlockList(filePath);

        List<IdCardConfig> tableConfigs = new ArrayList<>(4);
        tableConfigs.add(new IdCardConfig());

        IdCardTextResolver idCardTextResolver = new IdCardTextResolver();
        IdCardObject idCardObject = idCardTextResolver.resolve(textBlocks, tableConfigs);
        System.out.println(idCardObject);
    }

//    @Test
    public void TestLoadFile(){
//        TableConfigLoader dadiTableConfigTest = new FileConfigLoader();
//        TableConfig tableConfig = dadiTableConfigTest.load(folder + "/table_config_dadi.txt");
//        System.out.println("table title = " + tableConfig.getTableTitle());

        String filePath = folder + "/demo_table_dadi.txt";
//        String filePath = folder + "/demo_table_pinan.txt";
        List<TextBlock> textBlocks = this.acquireTextBlockList(filePath);
//        for(TextBlock textBlock : textBlocks){
//            System.out.println(textBlock);
//        }

        TableConfig dadiTableConfig = TestTableConfigUtils.generateDadiConfig();
        TableConfig pinganTableConfig = TestTableConfigUtils.generatePinganConfig();
        List<TableConfig> tableConfigs = new ArrayList<>(4);
        tableConfigs.add(dadiTableConfig);
        tableConfigs.add(pinganTableConfig);

        TextResolver<TableObject, TableConfig> textResolver = new TableTextResolver();
        textResolver.resolve(textBlocks, tableConfigs);

    }

    private List<TextBlock> acquireTextBlockList(String filePath){
        List<String> lines = FileReaderUtils.getFileLines(filePath);
        List<TextBlock> textBlocks = new ArrayList<>(32);
        if(lines != null && lines.size() > 0){
            int blockSize = lines.size() / 4;
            System.out.println("block size = " + blockSize);
            int j = 0;
            String text = null;
            String start = null;
            String end = null;
            String score = null;
            TextBlock textBlock = null;
            for(int i=0; i<lines.size(); i++){
                if(j == 0){
                    text = lines.get(i);
                    j++;
                    continue;
                }
                if(j == 1){
                    start = lines.get(i);
                    j++;
                    continue;
                }
                if(j == 2){
                    end = lines.get(i);
                    j++;
                    continue;
                }
                if(j == 3){
                    score = lines.get(i);
                    textBlock = this.acquireOneTextBlock(text, start, end, score);
                    textBlocks.add(textBlock);
                    j = 0;
                    continue;
                }
            }
        }
        return textBlocks;
    }

    private TextBlock acquireOneTextBlock(String text, String start, String end, String score){
        text = TableObjectUtils.formatText(text);
        TextBlock textBlock = new TextBlock();
        textBlock.setText(text);
        textBlock.setStartPosition(this.acquirePosition(start));
        textBlock.setEndPosition(this.acquirePosition(end));
        textBlock.setScore(Float.parseFloat(score));
        return textBlock;
    }

    private float[] acquirePosition(String position){
        position = position.replaceFirst(LEFT_BRACE, "").replaceFirst(RIGHT_BRACE, "");
        String[] xyStr = position.split(",");
        return new float[]{Float.parseFloat(xyStr[0].trim()), Float.parseFloat(xyStr[1].trim())};
    }

    private static final String LEFT_BRACE = "\\[";
    private static final String RIGHT_BRACE = "]";
}