cnstream/modules/InfineFilter/include/InfineFilter.h

/*
 * @Description: 
 * @Version: 1.0
 * @Autor: lishengyin
 * @Date: 2022-03-22 14:00:35
 * @LastEditors: lishengyin
 * @LastEditTime: 2022-08-01 16:34:15
 */
#ifndef __INFINEFILTER_H_
#define __INFINEFILTER_H_


#include <memory>
#include <string>
#include <vector>
#include <map>
#include <list>
#include <queue>

#include "cnstream_frame.hpp"
#include "cnstream_module.hpp"
#include "easyinfer/model_loader.h"
#include "easytrack/easy_track.h"
#include "cnstream_frame_va.hpp"

#include "opencv2/highgui/highgui.hpp"
#include "opencv2/imgproc/imgproc.hpp"
#include "opencv2/dnn.hpp"
#include "opencv2/opencv.hpp"
#include "opencv2/highgui.hpp"
#include "opencv2/imgcodecs/imgcodecs.hpp"

#include <rapidjson/document.h>
#include <rapidjson/rapidjson.h>
#include <rapidjson/stringbuffer.h>
#include <rapidjson/writer.h>
#include <ctime>
#include <iostream>
#include <time.h>
#include <cstdlib>
#include <dirent.h>
#include <sys/types.h>
#include <sys/stat.h>

using namespace std;


namespace cnstream{
    // Pointer for frame info
    using CNFrameInfoPtr = std::shared_ptr<CNFrameInfo>;
    /// Pointer for infer object
    using CNInferObjectPtr = std::shared_ptr<CNInferObject>;

    class InfineFilter : public cnstream::Module, public cnstream::ModuleCreator<InfineFilter> {
        using super = cnstream::Module;
        
    public:
        explicit InfineFilter(const std::string &name) : super(name) {}
        bool Open(cnstream::ModuleParamSet paramSet) override;

        void Close() override { std::cout << this->GetName() << " Close called" << std::endl; }

        int Process(std::shared_ptr<cnstream::CNFrameInfo> data) override;

        bool filter(cv::Mat& image, CNInferObjsPtr objs_holder);

        bool AlienFilter(CNInferObjsPtr& objs_holder);

        bool proportionalFilter(cv::Mat& image, CNInferObjsPtr objs_holder);
        
        bool nightFilter(cv::Mat& image, CNInferObjsPtr objs_holder);

        bool getNight();

        std::string getTime();

    private:
        InfineFilter(const InfineFilter &) = delete;
        InfineFilter &operator=(InfineFilter const &) = delete;
        
        vector<std::string> labels;
        vector<std::string> AlienLabels;
        
        bool saveFilterResult = true;
        bool Night_Filter = true;
        std::string files = "./output/";
        double Proportion_th = 0.2;
        double Night_th = 0.2;
        int start = 21;
        int end = 6;
    };
};



#endif