#pragma once

#include <map>
#include <algorithm>
#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_person_gathering_node : public vp_node
    {
    private:
        // channel -> status of targets (id -> num of hit frames)
        std::map<int, std::map<int, int>> all_person_gathering_checking_status;

        // channel -> jam status of channel (jam or not)
        std::map<int, bool> all_person_gathering_results;

        // channel -> last frame index to notify jam
        std::map<int, int> all_last_notifys;

        // record params
        bool need_record_image;
        bool need_record_video;

        bool isNearby(int p1_width, int p1_x, int p1_y, int p2_width, int p2_x, int p2_y, float factor);

        float person_gathering_distance_factor = 4.0f;

        // jam checking logic parameters which may be configed by constructor passed in by user
        const int check_interval_frames = 1;
        const int check_min_hit_frames = 1 * 1; // 25 fps * 2 seconds

        int min_gathering_person = 3;

        const int check_notify_interval = 10; // interval time (seconds) to notify (ensure not frequently)
    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_person_gathering_node(std::string node_name,
                                    int min_gathering_person,
                                    bool need_record_image = true,
                                    bool need_record_video = true);
        ~vp_ba_person_gathering_node();
        std::string to_string() override;
    };
}