visualization.h

Go to the documentation of this file.

 
#ifndef VISUALIZATION_H
#define VISUALIZATION_H
 
#include "dic_structures.h"
#include "config.h"
#include <opencv2/opencv.hpp>
#include <string>
#include <vector>
#include <map>
 
namespace cppxdic {
 
enum class ExportFormat {
    VTK,    // VTK Legacy format (for ParaView, VisIt)
    PLY,    // PLY format (for MeshLab, CloudCompare)
    CSV     // CSV format (for custom analysis)
};
 
struct VisData {
    std::vector<Point3DFrame> Points3D;              // 3D points per frame
    std::vector<int> Faces;                       // Triangle faces (Nx3 flattened)
    std::vector<std::vector<double>> FaceColors; // Face colors per frame
    std::map<std::string, std::vector<std::vector<double>>> FaceScalars;  // Scalar fields per frame
    std::map<std::string, std::vector<std::vector<double>>> FaceVectors;  // Vector fields per frame
    size_t n_frames;
};
 
class Visualization {
public:
    Visualization(const Config& config);
    ~Visualization() = default;
 
    DIC3DPPresults loadFromBinaryCache(const std::string& filepath);
 
    void applyTemporalFilter(DIC3DPPresults& results);
 
    std::vector<double> applySpatialSmooth(
        const std::vector<double>& face_data,
        const std::vector<int>& faces);
 
    VisData prepareVisualizationData(const DIC3DPPresults& results);
 
    void exportData(const VisData& vis_data, 
                    const std::string& output_path, 
                    int frame_idx = -1);
 
    void exportFrameVTK(const VisData& vis_data, 
                        const std::string& filepath, 
                        int frame_idx);
 
    void exportFramePLY(const VisData& vis_data, 
                        const std::string& filepath, 
                        int frame_idx);
 
    void exportFrameCSV(const VisData& vis_data, 
                        const std::string& filepath, 
                        int frame_idx);
 
    void generateVideo(const VisData& vis_data,
                       const std::string& output_path,
                       const std::string& field_name = "");
 
    void generateSummaryStats(const DIC3DPPresults& results, 
                              const std::string& output_path);
 
    void printTrialInfo(const DIC3DPPresults& results);
 
private:
    Config config_;
    ExportFormat export_format_;
 
    ExportFormat parseExportFormat(const std::string& format_str);
 
    std::vector<std::vector<double>> extractScalarField(
        const DIC3DPPresults& results, 
        const std::string& field_name);
 
    void applyCorrelationFilter(std::vector<std::vector<double>>& face_data,
                                 const std::vector<std::vector<double>>& corr_data);
 
    void applyGapFilter(std::vector<std::vector<double>>& face_data,
                        const std::vector<int>& FacePairInds);
 
    void computePrincipalStrains(const DIC3DPPresults& results,
                                  std::vector<std::vector<double>>& Epc1,
                                  std::vector<std::vector<double>>& Epc2);
 
    std::vector<std::vector<double>> butterworthFilter(
        const std::vector<std::vector<double>>& data,
        double cutoff_freq,
        double sample_freq);
 
    cv::Mat renderFrame(const VisData& vis_data, int frame_idx,
                        const std::string& field_name,
                        double data_min, double data_max,
                        int width, int height);
 
    static cv::Vec3b applyJetColormap(double val);
};
 
} // namespace cppxdic
 
#endif // VISUALIZATION_H