EngineBase.h

#pragma once
#include <iostream>
#include <map>
#include <deque>
#include <opencv2/imgcodecs.hpp>
#include "CvUtils.h"
#include "FaceDrawer.h"
#include "FaceDetectorYunet.h"
#include "FaceMesh.h"
#include "LandmarkFlowFilter.h"
 
namespace
{
    // Only available under Emscripten include
#ifdef __EMSCRIPTEN__
#include <emscripten/emscripten.h>
#include <emscripten/val.h>
 
    void log(const std::string& message) {
        emscripten::val console = emscripten::val::global("console");
        console.call<void>("log", message);
    }
#else
    void log(const std::string& message) {
        // Alternative logging on other platforms, e.g. writing to console
        std::cout << message << std::endl;
    }
#endif
 
};
 
class EngineBase
{
public:
    EngineBase() {}
 
    virtual ~EngineBase() {};
 
    virtual void init(std::string resourcePath);
 
    virtual void process(cv::Mat& input, cv::Mat& output) = 0;
 
    void addSigno(cv::Mat& image);
 
    void drawHistoryChart(cv::Mat& frame, const cv::Rect& haloRect,
                          const std::deque<std::map<std::string, double>>& history,
                          const std::map<std::string, cv::Scalar>& colors);
 
protected:
 
    cv::Mat preProcessImage(cv::Mat& image);
 
    cv::Mat postProcessImage(cv::Mat& image, cv::Mat& workingFrame);
 
    std::string faceVerticesPath;
 
    core::face::FaceDetectorYunet detector;
 
    visualization::FaceDrawer faceDrawer;
 
    core::face::FaceMesh faceMesh;
 
    core::filter::LandmarkFlowFilter filter;
 
    cv::Rect imageRect;
 
    float padding_ratio = 0.3;
    int top = 0;
    int bottom = 0;
    int left = 0;
    int right = 0;
 
    static constexpr size_t HISTORY_LENGTH = 50;
    std::map<int, std::deque<std::map<std::string, double>>> mHistory;
};