java-insagent-pc/src/main/java/com/ydtech/utils/baidu/MultiObjectDetect.java

package com.ydtech.utils.baidu;


import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;
import com.ydtech.core.page.HttpResult;
import com.ydtech.modules.ins.model.CardUserInfo;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.core.env.Environment;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;
import java.net.URLEncoder;
import java.util.HashMap;
import java.util.Map;
import java.util.Optional;

/**
 * @ClassName MultiObjectDetect
 * @Description: TODO
 * @Author
 * @Date 2021/6/25
 **/
@Component
public class MultiObjectDetect {


    @Autowired
    private Environment environment;

    @Autowired
    private static Environment staticEnvironment;

    @PostConstruct
    public void init() {
        staticEnvironment = environment;
    }


    public static HttpResult multiObjectDetect(String imgData) {
        Map<String, Object> ma = new HashMap<>();

        // 请求url
        String url = staticEnvironment.getProperty("baidu.vehicle");
        try {
            // 本地文件路径
            imgData = imgData.substring(imgData.indexOf(",") + 1);

            String imgParam = URLEncoder.encode(imgData, "UTF-8");

            String param = "image=" + imgParam;
            // 注意这里仅为了简化编码每一次请求都去获取access_token，线上环境access_token有过期时间， 客户端可自行缓存，过期后重新获取。
            String accessToken = AuthService.getAuth();

            String result = HttpUtil.post(url, accessToken, param);
            JSONObject ja = JSON.parseObject(result);

            String wordsResult = ja.getString("words_result");

            CardUserInfo cardUserInfo = new CardUserInfo();
            JSONObject jb = JSON.parseObject(wordsResult);
            cardUserInfo.setVIN(JSON.parseObject(jb.getString("车辆识别代号")).getString("words").toString());
            cardUserInfo.setIssueDate(JSON.parseObject(jb.getString("发证日期")).getString("words"));
            String hpzl = JSON.parseObject(jb.getString("车辆类型")).getString("words");
            if (hpzl != null && hpzl.equals("小型轿车")) {
                cardUserInfo.setPlateType("02");
            } else {
                cardUserInfo.setPlateType("01");

            }
            cardUserInfo.setMotorUsageTypeCode("8A");
            cardUserInfo.setMotorTypeCode("A0");
            if (!StringUtils.isEmpty(jb.getString("注册日期"))) {
                cardUserInfo.setRegisterDate(JSON.parseObject(jb.getString("注册日期")).getString("words"));
            } else if (!StringUtils.isEmpty(jb.getString("注册登记日期"))) {
                cardUserInfo.setRegisterDate(JSON.parseObject(jb.getString("注册登记日期")).getString("words"));
            }

            cardUserInfo.setEngine(JSON.parseObject(jb.getString("发动机号码")).getString("words"));
            cardUserInfo.setPlateNo(JSON.parseObject(jb.getString("号牌号码")).getString("words"));
            cardUserInfo.setBackOcrID(JSON.parseObject(jb.getString("品牌型号")).getString("words"));

            ma.put("carInfo", cardUserInfo);

            return HttpResult.ok("200", ma);
        } catch (Exception e) {
            e.printStackTrace();
        }
        return HttpResult.ok("201", ma);
    }


    public static JSONObject request(String images, String vehicleLicenseSide) {
        // 请求url
        String url = staticEnvironment.getProperty("baidu.vehicle");
        try {

            images = images.substring(images.indexOf(",") + 1);
            String imgParam = URLEncoder.encode(images, "UTF-8");
            String param = "image=" + imgParam + "&vehicle_license_side=" + vehicleLicenseSide;

            // 注意这里仅为了简化编码每一次请求都去获取access_token，线上环境access_token有过期时间， 客户端可自行缓存，过期后重新获取。
            String accessToken = AuthService.getAuth();

            String result = HttpUtil.post(url, accessToken, param);
            JSONObject ja = JSON.parseObject(result);
            String wordsResult = ja.getString("words_result");

            return JSON.parseObject(wordsResult);
        } catch (Exception e) {
            e.printStackTrace();
        }
        return new JSONObject();
    }

    public static void drivingPermitFront(String images, CardUserInfo cardUserInfo) {
        if (StringUtils.isEmpty(images)) {
            return;
        }

        try {
            JSONObject jb = request(images, "front");
            cardUserInfo.setVIN(JSON.parseObject(jb.getString("车辆识别代号")).getString("words").toString());
            cardUserInfo.setIssueDate(JSON.parseObject(jb.getString("发证日期")).getString("words"));
            //先获取使用性质
            String useNature = JSON.parseObject(jb.getString("使用性质")).getString("words");
            String hpzl = JSON.parseObject(jb.getString("车辆类型")).getString("words");
            String s = Optional.ofNullable(hpzl).orElse("");

            if(s.contains("货车")) {
                cardUserInfo.setPlateType("A1");
                if("货运".equals(useNature)){
                    cardUserInfo.setVehicleUse("04");
                }else if("非营运".equals(useNature)){
                    cardUserInfo.setVehicleUse("08");
                }
                cardUserInfo.setCategory("H31");
            }else {
                cardUserInfo.setPlateType("A0");
                cardUserInfo.setCategory("K33");
                if("非营运".equals(useNature)){
                    cardUserInfo.setVehicleUse("05");
                }
            }

            //判断使用性质
            cardUserInfo.setUseNature(JSON.parseObject(jb.getString("使用性质")).getString("words"));
            cardUserInfo.setMotorUsageTypeCode("8A");
            cardUserInfo.setMotorTypeCode("A0");
            if (!StringUtils.isEmpty(jb.getString("注册日期"))) {
                cardUserInfo.setRegisterDate(JSON.parseObject(jb.getString("注册日期")).getString("words"));
            } else if (!StringUtils.isEmpty(jb.getString("注册登记日期"))) {
                cardUserInfo.setRegisterDate(JSON.parseObject(jb.getString("注册登记日期")).getString("words"));
            }

            cardUserInfo.setEngine(JSON.parseObject(jb.getString("发动机号码")).getString("words"));
            cardUserInfo.setPlateNo(JSON.parseObject(jb.getString("号牌号码")).getString("words"));
            cardUserInfo.setBackOcrID(JSON.parseObject(jb.getString("品牌型号")).getString("words"));
            cardUserInfo.setCarOwner(JSON.parseObject(jb.getString("所有人")).getString("words"));
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    public static void conversionFront(CardUserInfo cardUserInfo, CardUserInfo cardUserInfoNew) {
        cardUserInfo.setVIN(cardUserInfoNew.getVIN());
        cardUserInfo.setIssueDate(cardUserInfoNew.getIssueDate());
        cardUserInfo.setCategory(cardUserInfoNew.getCategory());
        cardUserInfo.setPlateType(cardUserInfoNew.getPlateType());
        cardUserInfo.setVehicleUse(cardUserInfoNew.getVehicleUse());
        cardUserInfo.setUseNature(cardUserInfoNew.getUseNature());
        cardUserInfo.setMotorUsageTypeCode(cardUserInfoNew.getMotorUsageTypeCode());
        cardUserInfo.setMotorTypeCode(cardUserInfoNew.getMotorTypeCode());
        cardUserInfo.setRegisterDate(cardUserInfoNew.getRegisterDate());
        cardUserInfo.setEngine(cardUserInfoNew.getEngine());
        cardUserInfo.setPlateNo(cardUserInfoNew.getPlateNo());
        cardUserInfo.setBackOcrID(cardUserInfoNew.getBackOcrID());
        cardUserInfo.setCarOwner(cardUserInfoNew.getCarOwner());
    }


    /**
     * 行驶证背面
     *
     * @param images
     * @return
     */
    public static void drivingPermitBack(String images, CardUserInfo cardUserInfo) {
        if (StringUtils.isEmpty(images)) {
            return;
        }

        try {
            JSONObject jb = request(images, "back");
            cardUserInfo.setGrossMass(JSON.parseObject(jb.getString("总质量")).getString("words").replace("kg", ""));
            cardUserInfo.setApprovedPassengersCapacity(JSON.parseObject(jb.getString("核定载人数")).getString("words").replace("人", ""));
            cardUserInfo.setUnladenMass(JSON.parseObject(jb.getString("整备质量")).getString("words").replace("kg", ""));
            cardUserInfo.setLimitLoad(JSON.parseObject(jb.getString("核定载质量")).getString("words").replace("kg", ""));
        } catch (Exception e) {
            e.printStackTrace();
        }

    }

    public static void conversionBack(CardUserInfo cardUserInfo, CardUserInfo cardUserInfoNew) {
        cardUserInfo.setGrossMass(cardUserInfoNew.getGrossMass());
        cardUserInfo.setApprovedPassengersCapacity(cardUserInfoNew.getApprovedPassengersCapacity());
        cardUserInfo.setUnladenMass(cardUserInfoNew.getUnladenMass());
        cardUserInfo.setLimitLoad(cardUserInfoNew.getLimitLoad());
    }

}