shipProject/src/libTaskModel/HandlerSAR.cpp at master · hbkooo/shipProject · GitHub

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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
//
// Created by hbk on 19-4-19.
//
#include "HandlerSAR.h"
#include <glog/logging.h>
#include "libAlgorithm/sar_detect.h"
#include "libTaskModel/Queue.h"
#include "common/configParams.h"
#include "common/error.h"

using namespace std;
using namespace MQ;
using namespace detect;

HandlerSAR::HandlerSAR()
{

}

HandlerSAR::HandlerSAR(const std::string &path, const dev::ConfigParams *cp_)
        : HandlerBase(cp_->apiKey), cp(cp_)
{
    // GLOG 输出控制
    FLAGS_minloglevel = 3;

    string prototxt = path + "sar.prototxt";
    string model = path + "sar.caffemodel";

    sarDetector = std::make_shared<SarDetector>(prototxt, model);

}

HandlerSAR::~HandlerSAR()
{

}

std::string HandlerSAR::Detectsar(std::string const &request, const dev::TIMEPOINTS &timePoints)
{
    writeLog(LOG_DEBUG_LEVEL, "HandlerSAR", "DetectSAR begin" );
    string interfaceName("/sar");
    Json::Value value;
    string responseBody = std::move(validate(interfaceName, request, value));
    if(!responseBody.empty())
    {
        return responseBody;
    }
    Json::Value a;
    Json::Value resp;
    if(!value.isMember("image_base64") || !value["image_base64"].isString())
    {
        responseBody = HandlerBase::fillErrorResponse(interfaceName, resp, FINDER_PARAMETERS_ERROR, "Json参数错误，缺少参数或类型不对!");
        return responseBody;
    }
    cv::Mat image;
    if (!base64ToMat(value["image_base64"].asString(), image))
    {
        responseBody = HandlerBase::fillErrorResponse(interfaceName, resp, ERROR_IMAGE_BASE64,"图片base64编码错误!");
        return responseBody;
    }

    std::vector<std::vector<float>> locates;
    GetSARs(image,locates);
    resp["result_nums"] = locates.size();
    for(size_t i =0; i < locates.size(); ++i)
    {
        Json::Value result;
        result["label"] = locates[i][0];
        result["score"] = locates[i][1];
        result["xmin"] = locates[i][2];
        result["ymin"] = locates[i][3];
        result["xmax"] = locates[i][4];
        result["ymax"] = locates[i][5];
        resp["result"].append(result);
    }
    std::cout << "detect over and detct number is " << locates.size() << "\n\n";
    return HandlerBase::fillResponse(interfaceName, resp, FINDER_SAR_SUCCESS, timePoints);
}

void HandlerSAR::GetSARs(const cv::Mat &image, std::vector<std::vector<float>> &locates)
{
    writeLog(LOG_DEBUG_LEVEL, "HandlerSAR", " GetDetectSAR begin");
    // format : [label, score, xmin, ymin, xmax, ymax]
    locates = sarDetector->Detect(image);

}

std::string HandlerSAR::handle(const Message &message_) {
#ifdef USE_LICENSE
    if (!super_switch) {
        Json::Value resp;
        return HandlerBase::fillErrorResponse("/sar", resp, FINDER_NOT_AUTHORIZED, "license is error ... ");
    }
#endif

    switch (message_.msgType) {
        case INTERFACEID_SAR:
            return Detectsar(message_.str, message_.timePoints);
        default:
            std::cout << "HandlerSAR unkonown msgType: " << message_.msgType << std::endl;
            std::string response = "unkonown message!!!!";
            return response;
    }
}