#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;
|
std::ofstream record_file; // File stream for recording
|
|
// 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
|
|
void open_record_file();
|
void close_record_file();
|
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;
|
};
|
}
|
