35 lines
1.1 KiB
C++
35 lines
1.1 KiB
C++
|
|
#pragma once
|
|
|
|
#ifdef VP_WITH_LLM
|
|
#include <opencv2/imgproc.hpp>
|
|
#include <opencv2/freetype.hpp>
|
|
|
|
#include "../vp_node.h"
|
|
|
|
namespace vp_nodes {
|
|
// on screen display(short as osd) node.
|
|
// mainly used to display description(output from LLM) on frame.
|
|
class vp_mllm_osd_node: public vp_node
|
|
{
|
|
private:
|
|
// leave a gap at the bottom of osd frame
|
|
int gap_height = 112;
|
|
int padding = 5;
|
|
|
|
// support chinese font
|
|
cv::Ptr<cv::freetype::FreeType2> ft2;
|
|
std::vector<std::string> utf8_split(const std::string& text);
|
|
void draw_text_in_rect(cv::Mat& img,
|
|
const std::string& text,
|
|
const cv::Rect& rect,
|
|
int fontHeight,
|
|
cv::Scalar color);
|
|
protected:
|
|
virtual std::shared_ptr<vp_objects::vp_meta> handle_frame_meta(std::shared_ptr<vp_objects::vp_frame_meta> meta) override;
|
|
public:
|
|
vp_mllm_osd_node(std::string node_name, std::string font);
|
|
~vp_mllm_osd_node();
|
|
};
|
|
}
|
|
#endif |