first commit

nodes/osd/vp_osd_node_v3.cpp

@@ -0,0 +1,82 @@
+#include <opencv2/imgproc.hpp>
+
+#include "vp_osd_node_v3.h"
+
+namespace vp_nodes {
+    
+    vp_osd_node_v3::vp_osd_node_v3(std::string node_name, std::string font, bool show_bbox): 
+                                vp_node(node_name), show_bbox(show_bbox) {
+        if (!font.empty()) {
+            ft2 = cv::freetype::createFreeType2();
+            ft2->loadFontData(font, 0);   
+        }       
+        this->initialized();
+    }
+    
+    vp_osd_node_v3::~vp_osd_node_v3() {
+        deinitialized();
+    }
+
+    std::shared_ptr<vp_objects::vp_meta> vp_osd_node_v3::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) {
+            if (show_bbox) {
+                // track_id
+                auto id = std::to_string(i->track_id);
+                auto labels_to_display = i->primary_label;
+
+                // tracked
+                if (i->track_id != -1) {
+                    labels_to_display = "#" + id + " " + labels_to_display;
+                }
+                
+                for (auto& label : i->secondary_labels) {
+                    labels_to_display += "|" + label;
+                }
+                
+                // 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);
+                }
+            }
+            /* mask area */
+            if (!i->mask.empty()) {
+                // Resize the mask, threshold, color and apply it on the image
+                cv::resize(i->mask, i->mask, cv::Size(i->width, i->height));
+                cv::Mat mask = (i->mask > mask_threshold);
+                cv::Mat coloredRoi = (0.3 * cv::Scalar(255, 255, 0) + 0.7 * meta->osd_frame(cv::Rect(i->x, i->y, i->width, i->height)));
+                coloredRoi.convertTo(coloredRoi, CV_8UC3);
+            
+                // Draw the contours on the image
+                vector<cv::Mat> contours;
+                cv::Mat hierarchy;
+                mask.convertTo(mask, CV_8U);
+                cv::findContours(mask, contours, hierarchy, cv::RETR_CCOMP, cv::CHAIN_APPROX_SIMPLE);
+                cv::drawContours(coloredRoi, contours, -1, cv::Scalar(255, 255, 0), 5, cv::LINE_8, hierarchy, 100);
+                coloredRoi.copyTo(meta->osd_frame(cv::Rect(i->x, i->y, i->width, i->height)), mask);  
+            }                    
+        }
+        return meta;
+    }
+}