colored_point_handler.h

#ifndef COLORED_POINT_HANDLER_H
#define COLORED_POINT_HANDLER_H
 
// standart headers
#include <iostream>
 
// OpenCV headers
#include <opencv2/opencv.hpp>
 
// my headers
#include "VVT-V2/vibrating_point.h"
#include "VVT-V2/histogram.h"
#include "VVT-V2/output_to_csv.h"
 
using namespace cv;
 
class ColoredPointHandler : public Histogram, public VibratingPoint, public OutputToCsv
{
    enum coloring_mode
    {
        DEFAULT,
        COLORING_BASED_ON_FREQUENCY,
        COLORING_BASED_ON_AMPLITUDE,
    };
 
public:
    ColoredPointHandler(Point2f init_coordinates, double sampling_rate, int point_id, float radius_resize_factor, std::string output_csv_filename);
    ~ColoredPointHandler();
    void Draw(Mat& frame) override;
    void SetColoringMode(int coloring_mode);
private:
    void DrawHistogram() override;
    void DrawPoint(Mat& frame);
    void DrawInteractionRectangle(Mat& frame);
 
private:
    void UpdatePointColor() override;
 
private:
    // Текущий режим работы (цвета на основе амплитуд/частот)
    int coloring_mode_;
    float radius_resize_factor_;
    int point_id_;
    Scalar point_color_;
 
};
 
#endif