ExactSot.h

Go to the documentation of this file.

#pragma once
 
#include <deal.II/base/point.h>
#include <deal.II/lac/vector.h>
#include "SemiDiscreteOT/utils/utils.h"
#include <geogram/basic/common.h>
#include <geogram/basic/logger.h>
#include <geogram/basic/command_line.h>
#include <geogram/basic/command_line_args.h>
#include <geogram/basic/stopwatch.h>
#include <geogram/basic/file_system.h>
#include <geogram/basic/process.h>
#include <geogram/basic/progress.h>
#include <geogram/mesh/mesh.h>
#include <geogram/mesh/mesh_io.h>
#include <geogram/mesh/mesh_tetrahedralize.h>
#include <geogram/voronoi/CVT.h>
#include <exploragram/optimal_transport/optimal_transport_3d.h>
#include <exploragram/optimal_transport/sampling.h>
#include <memory>
#include <fstream>
#include <stdexcept>
#include <iostream>
#include <string>
#include <vector>
 
// Forward declarations for Geogram types
namespace GEO {
    class Mesh;
    class OptimalTransportMap3d;
}
 
class ExactSot {
public:
    ExactSot();
 
    ~ExactSot();
 
    // Delete copy operations due to unique GEO::Mesh ownership
    ExactSot(const ExactSot&) = delete;
    ExactSot& operator=(const ExactSot&) = delete;
 
    bool set_source_mesh(const std::string& filename);
 
    bool set_target_points(const std::string& filename, const std::string& io_coding);
 
    void set_parameters(unsigned int max_iterations = 1000,
                       double epsilon = 0.01);
 
    bool run();
 
    std::vector<double> get_potential() const;
 
    std::vector<dealii::Point<3>> get_target_points() const;
 
    bool save_results(const std::string& potential_file,
                     const std::string& points_file,
                     const std::string& io_coding = "txt") const;
 
private:
    // Helper function to load volume mesh
    bool load_volume_mesh(const std::string& filename, GEO::Mesh& mesh);
 
    // Member variables
    std::unique_ptr<GEO::Mesh> source_mesh;
    GEO::vector<double> potential;
    std::vector<dealii::Point<3>> target_points;
    
    // Parameters
    unsigned int max_iterations_;
    double epsilon_;
};