96 lines
4.3 KiB
C++
96 lines
4.3 KiB
C++
|
|
#include <opencv2/imgproc.hpp>
|
|
#include "vp_osd_node_v2.h"
|
|
#include "../../utils/vp_utils.h"
|
|
|
|
|
|
namespace vp_nodes {
|
|
|
|
vp_osd_node_v2::vp_osd_node_v2(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_osd_node_v2::~vp_osd_node_v2() {
|
|
deinitialized();
|
|
}
|
|
|
|
std::shared_ptr<vp_objects::vp_meta> vp_osd_node_v2::handle_frame_meta(std::shared_ptr<vp_objects::vp_frame_meta> meta) {
|
|
// operations on osd_frame
|
|
if (meta->osd_frame.empty()) {
|
|
// add a gap at the bottom of osd frame
|
|
meta->osd_frame = cv::Mat(meta->frame.rows + gap_height + padding * 2, meta->frame.cols, meta->frame.type(), cv::Scalar(128, 128, 128));
|
|
|
|
// initialize by copying frame to osd frame
|
|
auto roi = meta->osd_frame(cv::Rect(0, 0, meta->frame.cols, meta->frame.rows));
|
|
meta->frame.copyTo(roi);
|
|
}
|
|
auto& canvas = meta->osd_frame;
|
|
|
|
auto base_left = padding;
|
|
// scan targets
|
|
for (int i = 0; i < meta->targets.size(); i++) {
|
|
auto& target = meta->targets[i];
|
|
|
|
// scan sub targets
|
|
for (int j = 0; j < target->sub_targets.size(); j++) {
|
|
auto& sub_target = target->sub_targets[j];
|
|
cv::rectangle(canvas, cv::Rect(sub_target->x, sub_target->y, sub_target->width, sub_target->height), cv::Scalar(255, 255, 255), 2);
|
|
|
|
auto roi = canvas(cv::Rect(base_left, meta->osd_frame.rows - padding - gap_height, gap_height, gap_height));
|
|
// white background
|
|
roi = cv::Scalar(255, 255, 255);
|
|
|
|
auto ori = canvas(cv::Rect(sub_target->x, sub_target->y, sub_target->width, sub_target->height));
|
|
cv::Mat tmp = ori.clone();
|
|
int offset_w = 0, offset_h = 0;
|
|
if (tmp.rows > tmp.cols) {
|
|
cv::resize(tmp, tmp, cv::Size(int(float(gap_height) / tmp.rows * tmp.cols) , gap_height));
|
|
offset_w = (gap_height - tmp.cols) / 2;
|
|
offset_h = 0;
|
|
}
|
|
else {
|
|
cv::resize(tmp, tmp, cv::Size(gap_height, int(float(gap_height) / tmp.cols * tmp.rows)));
|
|
offset_h = (gap_height - tmp.rows) / 2;
|
|
offset_w = 0;
|
|
}
|
|
|
|
// copy sub target to the bottom of screen
|
|
roi = canvas(cv::Rect(base_left + offset_w, meta->osd_frame.rows - padding - gap_height + offset_h, tmp.cols, tmp.rows));
|
|
tmp.copyTo(roi);
|
|
|
|
// line from target to sub target
|
|
cv::line(canvas, cv::Point(target->x + target->width / 2, target->y + target->height), cv::Point(base_left + gap_height / 2, meta->osd_frame.rows - padding - gap_height), cv::Scalar(255, 0, 0), 3, cv::LINE_AA);
|
|
|
|
// label of sub target
|
|
auto sub_label = vp_utils::string_split(sub_target->label, '_');
|
|
|
|
// !!!
|
|
// for plate sub target (color_text)
|
|
if (sub_label.size() == 2) {
|
|
assert(ft2 != nullptr);
|
|
ft2->putText(canvas, sub_label[1], cv::Point(base_left + 10, meta->osd_frame.rows - padding - 5), 36, cv::Scalar(0), cv::FILLED, cv::LINE_AA, true);
|
|
}
|
|
base_left += gap_height + padding;
|
|
}
|
|
|
|
// target
|
|
auto labels_to_display = target->primary_label;
|
|
for (auto& label : target->secondary_labels) {
|
|
labels_to_display += "/" + label;
|
|
}
|
|
|
|
cv::rectangle(canvas, cv::Rect(target->x, target->y, target->width, target->height), cv::Scalar(255, 255, 0), 3);
|
|
if (ft2 != nullptr) {
|
|
ft2->putText(canvas, labels_to_display, cv::Point(target->x, target->y), 25, cv::Scalar(255, 0, 255), cv::FILLED, cv::LINE_AA, true);
|
|
}
|
|
else {
|
|
cv::putText(canvas, labels_to_display, cv::Point(target->x, target->y), 1, 1, cv::Scalar(255, 0, 255));
|
|
}
|
|
}
|
|
return meta;
|
|
}
|
|
} |