single_dic_workflow.h

Go to the documentation of this file.

 
#ifndef SINGLEDIC_SINGLE_DIC_WORKFLOW_H
#define SINGLEDIC_SINGLE_DIC_WORKFLOW_H
 
#include "singledic/singledic_config.h"
 
#include <opencv2/opencv.hpp>
#include <string>
#include <vector>
 
namespace singledic {
 
struct FrameResult {
    int grid_width = 0;           
    int grid_height = 0;          
    std::vector<double> u;        
    std::vector<double> v;        
    std::vector<double> corrcoef; 
    bool valid = false;           
};
 
struct SingleDicResult {
    cv::Size image_size;             
    cv::Mat roi_mask;                
    int subset_spacing = 1;          
    std::vector<int> seed_pw;        
    std::vector<FrameResult> frames; 
    bool ok = false;                 
    std::string message;             
};
 
class SingleDicWorkflow {
public:
    explicit SingleDicWorkflow(const SinglediConfig& config);
 
    SingleDicResult run();
 
private:
    SinglediConfig cfg_;
 
    // --- pipeline stages (each mirrors a MATLAB step) ----------------------
 
    bool importFrames(std::vector<cv::Mat>& frames) const;
 
    std::vector<cv::Mat> saturate(const std::vector<cv::Mat>& frames) const;
 
    std::vector<cv::Mat> maybeFilter(const std::vector<cv::Mat>& frames,
                                     const cv::Mat& roi_mask) const;
 
    cv::Mat loadOrCreateRoi(const cv::Mat& reference) const;
 
    std::vector<int> loadOrCreateSeed(const cv::Mat& roi_mask) const;
 
    bool track(const std::vector<cv::Mat>& frames, const cv::Mat& roi_mask,
               const std::vector<int>& seed_pw, SingleDicResult& out) const;
 
    // --- output writers ----------------------------------------------------
 
    bool writeCsv(const SingleDicResult& res) const;
 
    bool writeMat(const SingleDicResult& res) const;
};
 
} // namespace singledic
 
#endif // SINGLEDIC_SINGLE_DIC_WORKFLOW_H