package com.walker.openocr.vehicle;

import com.walker.openocr.ValueParser;
import com.walker.openocr.table.CellObject;
import com.walker.openocr.table.TableConfig;
import com.walker.openocr.table.TableObject;
import com.walker.openocr.util.TextUtils;

import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * 行驶证识别结果对象定义。
 * @author 时克英
 * @date 2022-09-27
 */
public class VehicleLicenseTable extends TableObject {

    // 属性id 与 解析器关联
    private Map<String, ValueParser> idParserMap = new HashMap<>(8);

    private Map<String, CellObject> tableData = new HashMap<>();

    public VehicleLicenseTable(TableConfig tableConfig, List<ValueParser> valueParserList) {
        super(tableConfig);
        if(valueParserList == null || valueParserList.size() == 0){
            logger.warn("valueParserList 没有设置，无法更好解析行驶证属性参数");
        } else {
            for(ValueParser valueParser : valueParserList){
                idParserMap.put(valueParser.getName(), valueParser);
            }
        }
    }

    @Override
    public void printRowCache(){
        super.printRowCache();
        logger.debug("~~~~~~~~~~~~~~~~~ table data ~~~~~~~~~~~~~~~~~");
        for(CellObject other : this.tableData.values()){
            logger.debug(other.toString());
        }
    }

    @Override
    public Map<String, CellObject> getTableDataMap(){
        return this.tableData;
    }

    @Override
    public void calculateValue(){
        super.calculateValue();

        //
//        Map<String, CellObject> map = new HashMap<>();
        if(this.rowCache.size() > 0){
            for (List<CellObject> list : this.rowCache.values()){
                for(CellObject co : list){
                    if(co.isConfigurable()){
                        tableData.put(co.getId(), co);
                    }
                }
            }
        }

        // 把没有解析到的属性值，通过简单数据推断后设置到属性中。
        this.predictValue("hao_pai_hao_ma");
        this.predictValue("che_liang_lei_xing");
        this.predictValue("zhu_zhi");
        this.predictValue("che_liang_shi_bie_dai_ma");
    }

    private void predictValue(String id){
        ValueParser valueParser = this.idParserMap.get(id);
        CellObject one = tableData.get(id);
        if(one != null && !TextUtils.isEmpty(one.getValue())){
            if(valueParser != null){
                if(valueParser.isTypeValue(one.getValue())){
                    return;
                }
            }
        } else if(one != null && one.getValue() == null){
            if(valueParser != null){
                if(valueParser.isTypeValue(one.getValue())){
                    return;
                }
            }
        } else if(one == null){
            // 没有解析到该属性
            one = new CellObject();
            one.setId(id);
            tableData.put(id, one);
        }

        String predictValue = this.parseValue(id, valueParser);
        if(predictValue != null){
            if(!one.isConfigurable()){
                logger.debug(predictValue + " 没有配置项，创建一个: " + id);
                one.setCellConfigItem(this.getRecognizeConfig().getCellConfigItem(id));
            }
            one.setValue(predictValue);
            one.setId(id);
        }
    }

    private String parseValue(String id, ValueParser valueParser){
//        ValueParser valueParser = this.idParserMap.get(id);
        if(valueParser == null){
            return null;
        }
        Object result = null;
        for (List<CellObject> list : this.rowCache.values()){
            for(CellObject co : list){
                result = valueParser.getValue(co.getSource().getText());
                if(result != null){
                    logger.debug("找到成功解析的属性值:" + co.getId() + ", " + co.getSource().getText());
                    return result.toString();
                }
            }
        }
        // 从其他属性搜索
        if(result == null){
            for(CellObject co : this.otherCellObjectList){
                result = valueParser.getValue(co.getSource().getText());
                if(result != null){
                    logger.debug("找到'otherCellObjectList'解析的属性值:" + co.getId() + ", " + co.getSource().getText());
                    return result.toString();
                }
            }
        }
        return null;
    }
}