package com.walker.openocr.table;

import com.walker.openocr.util.TableConfigUtils;

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

public class TableConfig {

    // 表格类型定义的关键词集合，用于确定表格类型，String[]里面为完全匹配（and），List中为任意一个匹配（or）
    private List<String[]> tableTypeKeys = new ArrayList<>(8);

    // 单元格计算时，容差像素
    private int cellTolerance = 5;

    // 标题匹配时，容错百分比，如：姓名 --> 名，则也算匹配上
//    private float titleTolerance = 0.5f;
    private float titleTolerance = 1.0f;

    // 要移除的单元格名称集合
    private String[] removeColumnsName = null;

    // 单元格配置参数集合，一个为一格
    private List<CellConfigItem> cellConfigItems = new ArrayList<>(32);

    // 字段列配置属性集合，一个为一列
    private List<ColumnConfigItem> columnConfigItems = new ArrayList<>(8);

    // 多行文本，每行之间容差距离（匹配2行单元格）
    private int multiLineTolerance = 30;

    // 表头名称，用户配置
    private String tableTitle;

    public String getTableTitle() {
        return tableTitle;
    }

    public void setTableTitle(String tableTitle) {
        this.tableTitle = tableTitle;
    }

    public int getMultiLineTolerance() {
        return multiLineTolerance;
    }

    public void setMultiLineTolerance(int multiLineTolerance) {
        this.multiLineTolerance = multiLineTolerance;
    }

    public int getCellTolerance() {
        return cellTolerance;
    }

    public void setCellTolerance(int cellTolerance) {
        this.cellTolerance = cellTolerance;
    }

    public float getTitleTolerance() {
        return titleTolerance;
    }

    public void setTitleTolerance(float titleTolerance) {
        this.titleTolerance = titleTolerance;
    }

    public String[] getRemoveColumnsName() {
        return removeColumnsName;
    }

    public void setRemoveColumnsName(String[] removeColumnsName) {
        this.removeColumnsName = removeColumnsName;
    }

    public List<String[]> getTableTypeKeys() {
        return tableTypeKeys;
    }

    /**
     * 设置表格分类关键词。
     * @param text 如：大地,保险,机动车
     */
    public void setTableTypeKeys(String text){
        if(text == null || text.equals("")){
            throw new IllegalArgumentException("tableTypeKeys is required!");
        }
        String[] array = text.split("\\|");
        if(array == null || array.length == 0){
            throw new IllegalArgumentException("分隔字符串错误：tableConfig参数错误：" + text);
        }

        String[] keys = null;
        for (String one : array){
//            System.out.println(one);
            keys = one.split(",");
            this.tableTypeKeys.add(keys);
        }
    }

    public void addCellConfig(CellConfigItem item){
        if(item == null){
            return;
        }
        if(this.cellConfigItems.contains(item)){
            return;
        }
        this.cellConfigItems.add(item);
    }

    public void addColumnConfig(ColumnConfigItem item){
        if(item == null){
            return;
        }
        if(this.columnConfigItems.contains(item)){
            return;
        }
        this.columnConfigItems.add(item);
    }

    /**
     * 根据单元格配置ID，返回配置项。
     * @param id
     * @return
     * @date 2022-09-30
     */
    public CellConfigItem getCellConfigItem(String id){
        if(this.cellConfigItems.size() == 0){
            return null;
        }
        for(CellConfigItem item : this.cellConfigItems){
            if(item.getId().equals(id)){
                return item;
            }
        }
        return null;
    }

    public CellConfigItem getCellConfig(String name){
        if(this.cellConfigItems.size() == 0){
            return null;
        }
        for(CellConfigItem item : this.cellConfigItems){
            if(item.getName().equals(name)){
                return item;
            }
        }
        return null;
    }

    public ColumnConfigItem getColumnConfig(String name){
        if(this.columnConfigItems.size() == 0){
            return null;
        }
        for(ColumnConfigItem item : this.columnConfigItems){
            if(item.getName().equals(name)){
                return item;
            }
        }
        return null;
    }

    @Override
    public String toString(){
        StringBuilder sb = new StringBuilder();
        for(String[] keys : this.tableTypeKeys){
            for(String s : keys){
                sb.append(s).append(",");
            }
        }
        return sb.toString();
    }

    /**
     * 查找是否存在匹配的配置对象（单元格或列对象）
     * @param text 给定的识别文本
     * @return
     */
    public ConfigItem findConfigItem(String text){
        List<CellConfigItem> matchedList = new ArrayList<>(2);
        for(CellConfigItem item : this.cellConfigItems){
            if(TableConfigUtils.isCellConfigItem(item, text, this)){
//                return item;
                matchedList.add(item);  // 因为可能存在多个匹配的，所以暂缓返回，最终统一比较最匹配的
            }
        }
        if(matchedList.size() == 1){
            return matchedList.get(0);
        }

//        if(text.equals("无责任死亡伤残赔偿限额18000元")){
//            System.out.println("++++++++++");
//        }
        // 如：配置项（死亡伤残赔偿限额）
        // “死亡伤残赔偿限额180000元” 和 “无责任死亡伤残赔偿限额18000元” 都匹配该项
        if(matchedList.size() > 1){
            float[] scores = new float[matchedList.size()];
            for(int i=0; i<matchedList.size(); i++){
                scores[i] = TableConfigUtils.getSimpleSimilarScore(matchedList.get(i).getName(), text);
            }
            // 获取最大分值的
            float maxScore = 0;
            int maxIndex = 0;
            for (int j=0; j<scores.length; j++){
                if(scores[j] > maxScore){
                    maxScore = scores[j];
                    maxIndex = j;
                }
            }
            System.out.println("发现多个配置匹配，找到最高评分:" + maxScore + ", config=" + matchedList.get(maxIndex).getName());
            return matchedList.get(maxIndex);
        }

//        for(ColumnConfigItem item : this.columnConfigItems){
//
//        }
        return null;
    }

    public List<String> getTableTitleNotKey() {
        return tableTitleNotKey;
    }

    /**
     * 设置表格（类型）标题不能包含的关键词。
     * <pre>
     *     1) 因为有些加油小票，会存在大类、小类关键词，如：中石油，中油联合，这种情况包含："中国石油"都会匹配上，油联合就没法弄了。
     * </pre>
     * @param tableTitleNotKey
     * @date 2023-11-01
     */
    public void setTableTitleNotKey(List<String> tableTitleNotKey) {
        this.tableTitleNotKey = tableTitleNotKey;
    }

    private List<String> tableTitleNotKey = new ArrayList<>(4);
}