2.7.18

shikeyin

2024-01-11 65da8373531677b1c37a98f53eaa30c892f35e5a

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
package com.iplatform.openocr;
 
import com.walker.infrastructure.utils.JsonUtils;
import com.walker.infrastructure.utils.StringUtils;
import com.walker.openocr.TextBlock;
import org.springframework.http.ResponseEntity;
import org.springframework.web.client.RestTemplate;
 
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
 
public abstract class RestTemplateOcrEngine extends AbstractOcrEngine{
//   @Override
//    protected List<TableConfig> loadTableConfigs() {
//        return null;
//    }
 
    private RestTemplate restTemplate = new RestTemplate();
 
    @Override
    public List<TextBlock> recognize(String id, String imgBase64) {
        if(StringUtils.isEmpty(id) || StringUtils.isEmpty(imgBase64)){
            logger.error("调用ocr服务，请求数据为空: id | img");
            return null;
        }
        Request request = new Request();
        request.setId(id);
        request.setImg(imgBase64);
 
        ResponseEntity<String> entity = restTemplate.postForEntity(this.getRemoteUrl(), request, String.class);
        if(entity != null){
//            List<?> list = entity.getBody();
            String jsonData = entity.getBody();
            logger.debug(jsonData);
            if(StringUtils.isEmpty(jsonData)){
                return null;
            }
 
            List<TextBlock> data = new ArrayList<>();
 
            try {
//                logger.debug(jsonData);
                List<Object> list = JsonUtils.jsonStringToObject(jsonData, List.class);
                for(Object obj : list){
                    data.add(this.transferToTextBlock((Map<String, Object>) obj));
                }
                logger.debug("++++++ 请求返回 TextBlock: " + data.size());
                return data;
 
            } catch (Exception e) {
                logger.error("解析json结果错误:" + jsonData, e);
                return data;
            }
        }
        return null;
    }
 
    private TextBlock transferToTextBlock(Map<String, Object> map){
        TextBlock textBlock = new TextBlock();
        textBlock.setText(map.get("name").toString());
        textBlock.setScore(Float.parseFloat(map.get("score").toString()));
        textBlock.setStartPosition(new float[]{Float.parseFloat(map.get("x1").toString()), Float.parseFloat(map.get("y1").toString())});
        textBlock.setEndPosition(new float[]{Float.parseFloat(map.get("x2").toString()), Float.parseFloat(map.get("y2").toString())});
        return textBlock;
    }
}