dic_structures.h

Go to the documentation of this file.

 
#ifndef DIC_STRUCTURES_H
#define DIC_STRUCTURES_H
 
#include <opencv2/opencv.hpp>
#include <vector>
#include <string>
#include <map>
#include "strain_computation.h"  // For FrameDeformationResult
 
namespace cppxdic {
 
struct StepAnalysisInfo {
    bool enabled;           // high_strain_analysis
    std::string type;       // seed_propagation type
    bool auto_update;       // auto_ref_change
    int step;              // step_ref_change
    
    StepAnalysisInfo() : 
        enabled(true), type("seed"), 
        auto_update(true), step(10) {}
};
 
struct DICInfo {
    std::vector<double> cutoff_corrcoef;  // Array of cutoff values (301 elements)
    double cutoff_diffnorm;
    int cutoff_iteration;
    std::vector<std::string> imgcorr;  // Cell array of image names (length 2)
    int lenscoef;
    double pixtounits;  // Pixels to units conversion
    int radius;
    int spacing;
    StepAnalysisInfo stepanalysis;  // Step analysis parameters
    bool subsettrunc;
    int total_threads;
    std::string type;
    std::string units;
};
 
struct Points2D {
    std::vector<double> x;  // X coordinates
    std::vector<double> y;  // Y coordinates
};
 
struct Point3DFrame {
    std::vector<double> x;  // X coordinates
    std::vector<double> y;  // Y coordinates
    std::vector<double> z;  // Z coordinates
};
 
struct DispData {
    std::vector<std::vector<double>> DispVec;  // Displacement vectors per frame
    std::vector<std::vector<double>> DispMgn;  // Displacement magnitudes per frame
};
 
struct CalibrationData {
    // 2x2 arrays for stereo pairs (rows=cameras, cols=pairs)
    std::vector<std::vector<std::string>> DLT_paths;  // 2x2 cell array of paths
    std::vector<std::vector<std::vector<double>>> DLT_params;  // 2x2 cell array of parameter vectors
};
 
struct DIC2DPairResults {
    int nCamRef;        // Reference camera number
    int nCamDef;        // Deformed camera number
    int nImages;        // Number of images
    std::vector<int> pairOrder;  // Trial-level stitch order metadata
    bool pairForced = false;     // Trial-level forced-stitch metadata
    cv::Mat ROImask;    // ROI mask
    DICInfo ncorrInfo;  // DIC parameters
    std::vector<Points2D> Points;  // Points per frame (cell array in MATLAB: each cell is Nx2 array)
    std::vector<std::vector<double>> CorCoeffVec;  // Correlation coefficients per frame (cell array: each cell is Nx1 array)
    std::vector<int> Faces;  // Triangular mesh faces (Nx3 flattened)
    std::vector<double> FaceColors;  // Face color values
};
 
struct DistortionData {
    // 2x2 arrays for stereo pairs (rows=cameras, cols=pairs)
    std::vector<std::vector<std::string>> distortion_models;  // 2x2 cell array of model names
    std::vector<std::vector<std::string>> distortion_paths;   // 2x2 cell array of paths
};
 
struct DIC3DpairResults {
    std::vector<int> cameraPairInd;  // [cam1, cam2] - Camera pair indices
    
    // Calibration for this pair (2x1 cell arrays)
    std::vector<std::string> DLTpath;  // {path_cam1, path_cam2}
    std::vector<std::vector<double>> DLTparameters;  // {L1, L2}
    
    // Distortion for this pair (2x1 cell arrays)
    std::vector<std::string> distortionModel;  // {model_cam1, model_cam2}
    std::vector<std::string> distortionPath;  // {path_cam1, path_cam2}
    
    // Geometry
    std::vector<int> Faces;  // Triangle connectivity (3 x nFaces flattened)
    std::vector<double> FaceColors;  // Face colors (1 x nFaces)
    
    // Per-frame data (cell arrays nImages x 1)
    std::vector<Point3DFrame> Points3D;  // 3D points per frame
    DispData Disp;  // DispVec, DispMgn per frame
    std::vector<std::vector<double>> FaceCentroids;  // Face centroids per frame (nFrames x (nFaces*3))
    std::vector<std::vector<double>> corrComb;  // Combined correlation per frame
    std::vector<std::vector<double>> FaceCorrComb;  // Face correlation per frame
};
 
struct DIC3Dcombined {
    std::vector<int> pairIndices;  // Camera pair indices
    std::vector<Point3DFrame> Points3D;  // 3D points per frame (cell array in MATLAB)
    std::vector<int> Faces;  // Triangular mesh faces (Nx3 flattened)
    std::vector<double> FaceColors;  // Face colors
    std::vector<std::vector<double>> corrComb;  // Combined correlation coefficients per frame
    std::vector<std::vector<double>> FaceCorrComb;  // Face correlation coefficients per frame
    std::vector<std::vector<double>> FaceCentroids;  // Triangle centroids per frame (nFrames x nFaces x 3)
    DispData Disp;  // Displacement data
    CalibrationData calibration;  // Calibration info
    DistortionData distortion;  // Distortion model and paths
    
    // Multi-pair stitching metadata
    std::vector<int> FacePairInds;  // Which stereo pair each face belongs to (1-indexed)
    std::vector<int> PointPairInds; // Which stereo pair each point belongs to (1-indexed)
    
    // Original DIC 2D info (cell array nPairs x 1)
    std::vector<DIC2DPairResults> DIC2Dinfo;
    
    // Individual pair results before stitching (cell array 1 x nPairs)
    std::vector<DIC3DpairResults> AllPairsResults;
    
    // Binary serialization methods
    void saveBinary(const std::string& filepath) const;
    static DIC3Dcombined loadBinary(const std::string& filepath);
};
 
struct DeformGradient {
    std::vector<double> F11, F12, F13;
    std::vector<double> F21, F22, F23;
    std::vector<double> F31, F32, F33;
};
 
struct StrainTensor {
    std::vector<double> E11, E12, E13;
    std::vector<double> E21, E22, E23;
    std::vector<double> E31, E32, E33;
};
 
struct DeformData {
    std::vector<DeformGradient> F;  // Deformation gradients per frame
    std::vector<StrainTensor> strain;  // Strain tensors per frame (Green-Lagrange)
    std::vector<std::vector<double>> princStrain;  // Principal strains per frame
    std::vector<std::vector<double>> maxShearStrain;  // Max shear strain per frame
};
 
struct RBMData {
    std::vector<std::vector<double>> RotMat;   // Rotation matrices per frame (9 values each)
    std::vector<std::vector<double>> TransVec; // Translation vectors per frame (3 values each)
};
 
struct DIC3DPPresults : DIC3Dcombined {
    DeformData Deform;         // Deformation and strain data (with RBM) - simplified format
    DeformData Deform_ARBM;    // Deformation and strain data after RBM removal
    RBMData RBM;               // Rigid body motion transformation data
    std::vector<std::vector<double>> Points3D_ARBM_x;  // 3D points after RBM per frame
    std::vector<std::vector<double>> Points3D_ARBM_y;
    std::vector<std::vector<double>> Points3D_ARBM_z;
    std::vector<std::vector<double>> FaceIsoInd;  // Face isotropy index per frame
    std::string deftype;  // "cum" (cumulative) or "rate"
    size_t n_frames;  // Number of frames processed
    
    // Full deformation result (all 36 fields) for MAT file writing
    FrameDeformationResult deform_full;
};
 
} // namespace cppxdic
 
#endif // DIC_STRUCTURES_H