|
#include "vp_ba_stop_osd_node.h"
|
|
namespace vp_nodes {
|
|
vp_ba_stop_osd_node::vp_ba_stop_osd_node(std::string node_name, std::string font): vp_node(node_name) {
|
if (!font.empty()) {
|
ft2 = cv::freetype::createFreeType2();
|
ft2->loadFontData(font, 0);
|
}
|
this->initialized();
|
}
|
|
vp_ba_stop_osd_node::~vp_ba_stop_osd_node() {
|
deinitialized();
|
}
|
|
std::shared_ptr<vp_objects::vp_meta> vp_ba_stop_osd_node::handle_frame_meta(std::shared_ptr<vp_objects::vp_frame_meta> meta) {
|
// operations on osd_frame
|
if (meta->osd_frame.empty()) {
|
meta->osd_frame = meta->frame.clone();
|
}
|
|
auto& canvas = meta->osd_frame;
|
// scan targets
|
for (auto& i : meta->targets) {
|
// track_id
|
auto id = std::to_string(i->track_id);
|
auto labels_to_display = i->primary_label;
|
auto primary_score_ = std::to_string(i->primary_score);
|
|
// tracked
|
if (i->track_id != -1) {
|
labels_to_display = "#" + id + " " + labels_to_display;
|
}
|
|
for (auto& label : i->secondary_labels) {
|
labels_to_display += "|" + label + "|" + primary_score_;
|
}
|
|
// draw tracks if size>=2
|
if (i->tracks.size() >= 2) {
|
for (int n = 0; n < (i->tracks.size() - 1); n++) {
|
auto p1 = i->tracks[n].track_point();
|
auto p2 = i->tracks[n + 1].track_point();
|
cv::line(canvas, cv::Point(p1.x, p1.y), cv::Point(p2.x, p2.y), cv::Scalar(0, 255, 255), 1, cv::LINE_AA);
|
}
|
}
|
|
cv::rectangle(canvas, cv::Rect(i->x, i->y, i->width, i->height), cv::Scalar(255, 255, 0), 2);
|
if (ft2 != nullptr) {
|
ft2->putText(canvas, labels_to_display, cv::Point(i->x, i->y), 20, cv::Scalar(255, 0, 255), cv::FILLED, cv::LINE_AA, true);
|
}
|
else {
|
//cv::putText(canvas, labels_to_display, cv::Point(i->x, i->y), 1, 1, cv::Scalar(255, 0, 255));
|
int baseline = 0;
|
auto size = cv::getTextSize(labels_to_display, 1, 1, 1, &baseline);
|
vp_utils::put_text_at_center_of_rect(canvas, labels_to_display, cv::Rect(i->x, i->y - size.height, size.width, size.height), true, 1, 1, cv::Scalar(), cv::Scalar(179, 52, 255), cv::Scalar(179, 52, 255));
|
}
|
|
// scan sub targets
|
for (auto& sub_target: i->sub_targets) {
|
cv::rectangle(canvas, cv::Rect(sub_target->x, sub_target->y, sub_target->width, sub_target->height), cv::Scalar(255));
|
if (ft2 != nullptr) {
|
ft2->putText(canvas, sub_target->label, cv::Point(sub_target->x, sub_target->y), 20, cv::Scalar(0, 0, 255), cv::FILLED, cv::LINE_AA, true);
|
}
|
else {
|
cv::putText(canvas, sub_target->label, cv::Point(sub_target->x, sub_target->y), 1, 1, cv::Scalar(0, 0, 255));
|
}
|
}
|
}
|
|
/* stop draw for current channel */
|
auto& region = all_stop_regions[meta->channel_index];
|
auto& total_stop = all_total_stop[meta->channel_index];
|
auto& stop_count = all_stop_count[meta->channel_index];
|
|
// scan ba results and ONLY deal with stop and unstop
|
for (auto& i : meta->ba_results) {
|
if (i->type == vp_objects::vp_ba_type::STOP) {
|
region = i->involve_region_in_frame;
|
stop_count++;
|
total_stop.push_back(i->involve_target_ids_in_frame[0]); // only 1 target
|
}
|
if (i->type == vp_objects::vp_ba_type::UNSTOP) {
|
region = i->involve_region_in_frame;
|
for (auto j = total_stop.begin(); j < total_stop.end();) {
|
if (*j == i->involve_target_ids_in_frame[0]) {
|
j = total_stop.erase(j);
|
break;
|
}
|
j++;
|
}
|
}
|
}
|
|
// draw stop data
|
auto poly_vertexs = [&]() {
|
std::vector<cv::Point> vertexs;
|
for(auto& p: region) {
|
vertexs.push_back(cv::Point(p.x, p.y));
|
}
|
return vertexs;
|
};
|
cv::polylines(canvas, poly_vertexs(), true, cv::Scalar(0, 255, 0), 2, cv::LINE_AA);
|
// cv::putText(canvas, vp_utils::string_format("total stop targets: [%d]", stop_count), cv::Point(20, 20), 1, 2, cv::Scalar(0, 0, 255), 2);
|
|
// stop targets, highlight
|
auto stop_targets = meta->get_targets_by_ids(total_stop);
|
for (auto& t: stop_targets) {
|
cv::rectangle(canvas, cv::Rect(t->x, t->y, t->width, t->height), cv::Scalar(0, 0, 255), 2, cv::LINE_AA);
|
}
|
return meta;
|
}
|
}
|
