wangrong
2025-03-19 291afb9c81aae1d15d9e3edcea618d0a2e1fc5a6
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
#pragma once
 
#include <map>
#include "../vp_node.h"
#include "../../objects/shapes/vp_point.h"
#include "../../objects/shapes/vp_line.h"
#include "../../objects/vp_image_record_control_meta.h"
#include "../../objects/vp_video_record_control_meta.h"
 
namespace vp_nodes {
    // behaviour analysis node for stop (support multi channels)
    class vp_ba_stop_node: public vp_node 
    {
    private:
        // channel -> vertexs of region, 1 channel supports only 1 region at most (can be 0, which means no stop check on this channel)
        std::map<int, std::vector<vp_objects::vp_point>> all_stop_regions;
 
        // channel -> status of targets (id -> num of hit frames)
        std::map<int, std::map<int, int>> all_stop_checking_status;
 
        // record params
        bool need_record_image;
        bool need_record_video;
 
        // check if point inside of polygon
        bool point_in_poly(vp_objects::vp_point p, std::vector<vp_objects::vp_point> region);
 
        // stop checking logic parameters which may be configed by constructor passed in by user
        const int check_interval_frames = 70;
        const int check_min_hit_frames = 20/3 * 60;  // 25 fps * 2 seconds
        const int check_max_distance = 15; //The closer the camera, the larger the pixel distance.
    protected:
        virtual std::shared_ptr<vp_objects::vp_meta> handle_frame_meta(std::shared_ptr<vp_objects::vp_frame_meta> meta) override;
    public:
        vp_ba_stop_node(std::string node_name, 
                            std::map<int, std::vector<vp_objects::vp_point>> stop_regions,
                            bool need_record_image = true,
                            bool need_record_video = true);
        ~vp_ba_stop_node();
        std::string to_string() override;
    };
}