package com.walker.openocr.table;
|
|
import com.walker.infrastructure.utils.StringUtils;
|
import com.walker.openocr.Constants;
|
import com.walker.openocr.MultipleLine;
|
import com.walker.openocr.util.JsonUtils;
|
import org.slf4j.Logger;
|
import org.slf4j.LoggerFactory;
|
|
import java.util.List;
|
import java.util.Map;
|
|
public abstract class AbstractTableConfigLoader implements TableConfigLoader{
|
|
protected final transient Logger logger = LoggerFactory.getLogger(this.getClass());
|
|
private static final String NAME_TABLE_TITLE = "table_title";
|
private static final String NAME_TABLE_TYPE_KEYS = "table_type_keys";
|
private static final String NAME_TABLE_TYPE_KEYS_NOT = "table_type_keys_not";
|
private static final String NAME_TABLE_SETTING = "table_setting";
|
private static final String NAME_REMOVE_COLUMNS = "remove_columns";
|
|
private static final String KEY_CELL_TOLERANCE = "cell_tolerance";
|
private static final String KEY_TITLE_TOLERANCE = "title_tolerance";
|
private static final String KEY_MULTI_LINE_TOLERANCE = "multi_line_tolerance";
|
private static final String KEY_CELL_NAME = "name";
|
private static final String KEY_MULTI_LINE = "multi_line";
|
private static final String KEY_NONE_CESS = "none_cell";
|
private static final String KEY_END_FLAG = "end_flag";
|
private static final String KEY_ALIGN = "align";
|
private static final String KEY_DATA_TYPE = "data_type";
|
private static final String KEY_ORDER_NUM = "order_num";
|
private static final String KEY_FULL_ROW = "full_row";
|
private static final String KEY_MIN_VALUE_SIZE = "min_value_size";
|
|
|
@Override
|
public TableConfig load(Object option) {
|
List<String> configContent = this.doLoadContent(option);
|
if(configContent == null || configContent.size() == 0){
|
logger.error("未加载到配置项内容:" + option);
|
return null;
|
}
|
|
TableConfig tableConfig = new TableConfig();
|
|
try {
|
String[] keyValue = null;
|
for (String line : configContent) {
|
// 忽略注释行
|
if (line.startsWith(Constants.VALUE_SHARP)) {
|
continue;
|
}
|
keyValue = line.split(Constants.EQUALS);
|
if (keyValue.length != 2) {
|
// throw new IllegalArgumentException("解析表格配置行错误，等号左右必须都有值:" + line);
|
if(keyValue.length == 1){
|
logger.warn("该属性没有配置任何值:" + line);
|
continue;
|
}
|
throw new IllegalArgumentException("解析表格配置行错误，等号左右必须都有值，配置不正确:" + line);
|
}
|
|
logger.debug("加载key:" + keyValue[0] + ", value=" + keyValue[1]);
|
|
if (keyValue[0].equals(NAME_TABLE_TYPE_KEYS)) {
|
this.processTableTypeKeys(keyValue[1], tableConfig);
|
|
} else if(keyValue[0].equals(NAME_TABLE_TYPE_KEYS_NOT)){
|
// 2023-11-01 添加表格标题可排除的关键词
|
this.processTableTypeKeysNot(keyValue[1], tableConfig);
|
|
} else if (keyValue[0].equals(NAME_TABLE_SETTING)) {
|
this.processTableSetting(keyValue[1], tableConfig);
|
|
} else if (keyValue[0].equals(NAME_REMOVE_COLUMNS)) {
|
this.processRemoveColumns(keyValue[1], tableConfig);
|
|
} else if (keyValue[0].equals(NAME_TABLE_TITLE)) {
|
tableConfig.setTableTitle(keyValue[1]);
|
|
} else {
|
this.processCellConfig(keyValue[0], keyValue[1], tableConfig);
|
}
|
}
|
logger.info("加载完成 TableConfig:" + option);
|
} catch (Exception ex){
|
logger.error("解析表格配置出现错误:" + option, ex);
|
}
|
return tableConfig;
|
}
|
|
private void processCellConfig(String key, String value, TableConfig tableConfig) throws Exception{
|
Map<String, Object> map = JsonUtils.jsonStringToObject(value, Map.class);
|
if(map == null || map.size() == 0){
|
return;
|
}
|
CellConfigItem item = new CellConfigItem();
|
item.setId(key);
|
|
Object name = map.get(KEY_CELL_NAME);
|
if(name != null){
|
item.setName(name.toString());
|
}
|
Object orderNum = map.get(KEY_ORDER_NUM);
|
if(orderNum != null){
|
item.setOrderNum(Integer.parseInt(orderNum.toString()));
|
}
|
Object align = map.get(KEY_ALIGN);
|
if(align != null){
|
item.setAlign(align.toString());
|
}
|
Object multipleLine = map.get(KEY_MULTI_LINE);
|
if(multipleLine != null){
|
item.setMultipleLine(MultipleLine.getType(multipleLine.toString()));
|
}
|
Object dataType = map.get(KEY_DATA_TYPE);
|
if(dataType != null){
|
item.setDataType(dataType.toString());
|
}
|
Object noneCell = map.get(KEY_NONE_CESS);
|
if(noneCell != null){
|
item.setNoneCell(Boolean.parseBoolean(noneCell.toString()));
|
}
|
Object endFlag = map.get(KEY_END_FLAG);
|
if(endFlag != null){
|
item.setEndFlag(endFlag.toString());
|
}
|
Object fullRow = map.get(KEY_FULL_ROW);
|
if(fullRow != null){
|
item.setFullRow(Boolean.parseBoolean(fullRow.toString()));
|
}
|
Object minValueSize = map.get(KEY_MIN_VALUE_SIZE);
|
if(minValueSize != null){
|
item.setMinValueSize(Integer.parseInt(minValueSize.toString()));
|
}
|
tableConfig.addCellConfig(item);
|
}
|
|
private void processRemoveColumns(String text, TableConfig tableConfig){
|
String[] names = text.split(Constants.VALUE_SPLIT);
|
tableConfig.setRemoveColumnsName(names);
|
}
|
|
private void processTableSetting(String text, TableConfig tableConfig) throws Exception{
|
Map<String, Object> map = JsonUtils.jsonStringToObject(text, Map.class);
|
if(map == null || map.size() == 0){
|
throw new IllegalArgumentException("未找到表格配置项:" + NAME_TABLE_SETTING);
|
}
|
Object cellTolerance = map.get(KEY_CELL_TOLERANCE);
|
if(cellTolerance != null){
|
tableConfig.setCellTolerance(Integer.parseInt(cellTolerance.toString()));
|
}
|
Object titleTolerance = map.get(KEY_TITLE_TOLERANCE);
|
if(titleTolerance != null){
|
tableConfig.setTitleTolerance(Float.parseFloat(titleTolerance.toString()));
|
}
|
Object multiLineTolerance = map.get(KEY_MULTI_LINE_TOLERANCE);
|
if(multiLineTolerance != null){
|
tableConfig.setMultiLineTolerance(Integer.parseInt(multiLineTolerance.toString()));
|
}
|
}
|
|
private void processTableTypeKeys(String text, TableConfig tableConfig){
|
tableConfig.setTableTypeKeys(text);
|
}
|
|
private void processTableTypeKeysNot(String text, TableConfig tableConfig){
|
if(StringUtils.isNotEmpty(text)){
|
List<String> list = StringUtils.asList(text.split(StringUtils.DEFAULT_SPLIT_SEPARATOR));
|
tableConfig.setTableTitleNotKey(list);
|
}
|
}
|
|
protected abstract List<String> doLoadContent(Object option);
|
}
|
