HDVF/main_per_hdvf.cpp

// Persistent HDVF computation (command line version)
// --------
// Computes a "perfect" persistent HDVF from a given filtration
//      and provides a batch mode to specify arguments
// For help: per_hdvf -h
// --------
// A. Bac
// --------
 
#include <iostream>
#include <chrono>
#include <type_traits>
#include <typeinfo>
#include <CGAL/HDVF/Zp.h>
#include <CGAL/HDVF/Simplex.h>
#include <CGAL/HDVF/Simplicial_chain_complex.h>
#include <CGAL/HDVF/Cubical_chain_complex.h>
#include <CGAL/HDVF/Geometric_chain_complex_tools.h>
#include <CGAL/HDVF/Hdvf_persistence.h>
#include <CGAL/OSM/OSM.h>
#include <CGAL/HDVF/hdvf_tools.h>
#include <CGAL/HDVF/Mesh_object_io.h>
#include <CGAL/HDVF/Cub_object_io.h>
#include "arguments.h"
 
// ------- A ring
// For Z/nZ other than Z (ie. n=0) and Z/2Z, uncomment and set the following define properly
 
//#define SCALAR 5
 
// ------- Filtration
 
// For a "custom" filtration, please use the following filtration function and set the
//#define CUST_FILTRATION
 
// Degree type definition
#ifndef CUST_FILTRATION
using DegType = double ;
#else // Custom filtration
using DegType = double ;
#endif
 
// Custom filtration
#ifdef CUST_FILTRATION
 
// Example returning x+z value for each vertex
template<typename ComplexType>
std::function<DegType(int)>  deg_fun (const ComplexType& complex)
{
    std::function<DegType(int)> deg_fun_f = [&complex](int i)
    {
        const std::vector<double> Xi(complex.get_vertex_coords(i)) ;
        return Xi.at(0)+Xi.at(2) ;
    } ;
    return deg_fun_f ;
}
 
#endif
 
template <typename MeshType, typename ComplexType>
void mesh_complex_output(const MeshType& mesh, const ComplexType& complex, const Options& options)
{
    if (options.with_output)
    {
        // Mesh
        std::cout << "----> mesh informations" << std::endl ;
        mesh.print_infos() ;
 
        // Complex
        std::cout << "----> complex informations" << std::endl ;
        complex.print_complex();
    }
}
 
template<typename CoefficientType, typename ComplexType, typename DegType, typename FiltrationType >
CGAL::HDVF::Hdvf_persistence<CoefficientType, ComplexType, DegType, FiltrationType>& per_HDVF_comput (const ComplexType& complex, const FiltrationType& f, const Options &options)
{
    typedef CGAL::HDVF::Hdvf_persistence<CoefficientType, ComplexType, DegType, FiltrationType> HDVFType ;
    HDVFType& hdvf(*(new HDVFType(complex, f, options.HDVF_opt, options.with_vtk_export)));
 
    std::cout << "----> START computing persistent homology" << std::endl ;
    hdvf.compute_perfect_hdvf() ;
    std::cout << "------> END computing persistent homology" << std::endl ;
 
    if (options.with_output)
    {
        std::cout << "----> perHDVF" << std::endl ;
        hdvf.print_hdvf_persistence_info(std::cout);
        std::cout << "----> reduction" << std::endl ;
        hdvf.print_reduction() ;
        std::cout << "----> persistent diagram" << std::endl ;
        std::cout << hdvf ;
    }
    if (options.with_export)
    {
        std::cout << "----> exporting..." << std::endl ;
        std::string file(options.outfile_root+"_reduction.txt") ;
        std::ofstream out ( file, std::ios::out | std::ios::trunc);
 
        if ( not out . good () ) {
            std::cerr << "hdvf: with_export. Fatal Error:\n  " << file << " not found.\n";
            throw std::runtime_error("File Parsing Error: File not found");
        }
        out << "----> perHDVF" << std::endl ;
        hdvf.print_hdvf_persistence_info(out);
        out << "----> reduction" << std::endl ;
        hdvf.print_reduction(out) ;
 
        out.close() ;
 
        std::string file_per(options.outfile_root+"_per.txt") ;
        std::ofstream out_per ( file_per, std::ios::out | std::ios::trunc);
 
        if ( not out_per . good () ) {
            std::cerr << "hdvf: with_export. Fatal Error:\n  " << file << " not found.\n";
            throw std::runtime_error("File Parsing Error: File not found");
        }
        out_per << "----> persistent diagram" << std::endl ;
        out_per << hdvf ;
 
        out_per.close() ;
    }
    return hdvf ;
}
 
template<typename CoefficientType, typename ComplexType, typename DegType, typename FiltrationType>
FiltrationType& build_filtration(const ComplexType& complex, const Options& options)
{
    typedef FiltrationType Filtration;
    std::function<DegType(size_t)>  deg_function ;
#ifndef CUST_FILTRATION
    // Standard lower star filtration along x,y or z
    if (options.star_filtr == StarFiltrStd::FiltrX)
        deg_function = (deg_fun(complex, CGAL::HDVF::f_x)) ;
    else if (options.star_filtr == StarFiltrStd::FiltrY)
        deg_function = (deg_fun(complex, CGAL::HDVF::f_y)) ;
    else if (options.star_filtr == StarFiltrStd::FiltrZ)
        deg_function = (deg_fun(complex, CGAL::HDVF::f_z)) ;
    else
    {
        std::cout << "Unknown lower start filtration" << std::endl ;
        throw("Unknown lower start filtration") ;
    }
#else
    // Use deg_fun defined above
    deg_function = (deg_fun(complex)) ;
#endif
 
    std::cout << "----> START building filtration" << std::endl ;
    Filtration& f = *(new Filtration(complex, deg_function)) ;
    std::cout << "------> END building filtration" << std::endl ;
    return f ;
}
 
 
template <typename CoefficientType>
void main_code (const Options &options)
{
    if (options.in_format == InputFormat::SIMP)
    {
//        using ComplexType = AbstractSimpComplex<CoefficientType>  ;
//        using HDVFType = HDVF<CoefficientType, ComplexType> ;
//
//        // MeshObject
//        MeshObject mesh ;
//        mesh.read_simp(options.in_file) ;
//
//        // Complex
//        ComplexType complex(mesh);
//
//        mesh_complex_output<MeshObject, ComplexType>(mesh, complex, options) ;
//
//        // HDVF computation, export, output
//        HDVFType hdvf(HDVF_comput<CoefficientType,ComplexType>(complex, options)) ;
//
//        // Export to vtk
//        // None for SIMP format
        std::cout << "not yet..." << std::endl ;
        throw("not yet") ;
    }
    else if (options.in_format == InputFormat::OFF)
    {
        using ComplexType = CGAL::HDVF::Simplicial_chain_complex<CoefficientType> ;
        using FiltrationType = CGAL::HDVF::Filtration_lower_star<CoefficientType, ComplexType, DegType> ;
        using HDVFType = CGAL::HDVF::Hdvf_persistence<CoefficientType, ComplexType,DegType, FiltrationType> ;
 
        // MeshObject
        CGAL::HDVF::Mesh_object_io mesh ;
        mesh.read_off(options.in_file) ;
 
        // Complex
        ComplexType complex(mesh, mesh.get_nodes());
 
        // Build filtration
        FiltrationType& f(build_filtration<CoefficientType, ComplexType, DegType, FiltrationType>(complex, options)) ;
 
        mesh_complex_output<CGAL::HDVF::Mesh_object_io, ComplexType>(mesh, complex, options) ;
 
        // HDVF computation, export, output
        HDVFType& hdvf(per_HDVF_comput<CoefficientType,ComplexType,DegType, FiltrationType>(complex,f, options)) ;
 
        // Export to vtk
        if (options.with_vtk_export)
        {
            std::cout << "----> exporting to vtk" << std::endl ;
            hdvf_persistence_geometric_chain_complex_output_vtk(hdvf, complex, options.outfile_root, options.co_faces) ;
        }
    }
    // CubComplex
    else if ((options.in_format == InputFormat::PGM) || (options.in_format == InputFormat::CUB))
    {
        using ComplexType = CGAL::HDVF::Cubical_chain_complex<CoefficientType> ;
        using FiltrationType = CGAL::HDVF::Filtration_lower_star<CoefficientType, ComplexType, DegType> ;
        using HDVFType = CGAL::HDVF::Hdvf_persistence<CoefficientType, ComplexType, DegType, FiltrationType> ;
 
        CGAL::HDVF::Cub_object_io mesh ;
        typename ComplexType::typeComplexCube primal_dual(ComplexType::PRIMAL) ;
        if (options.primal)
        {
            if (options.in_format == InputFormat::PGM)
                mesh.read_pgm(options.in_file, true) ; // Read with Khalimsky coordinates (for primal)
            else
                mesh.read_cub(options.in_file, true) ; // Read with Khalimsky coordinates (for primal)
        }
        else // dual
        {
            if (options.in_format == InputFormat::PGM)
                mesh.read_pgm(options.in_file, false) ; // Read with pixel coordinates (for dual)
            else
                mesh.read_cub(options.in_file, false) ; // Read with pixel coordinates (for dual)
            primal_dual = ComplexType::DUAL ;
        }
 
        // Complex
        ComplexType complex(mesh, primal_dual);
 
        mesh_complex_output<CGAL::HDVF::Cub_object_io, ComplexType>(mesh, complex, options) ;
 
        // Build filtration
        FiltrationType& f(build_filtration<CoefficientType, ComplexType, DegType, FiltrationType>(complex, options)) ;
 
        // HDVF computation, export, output
        HDVFType& hdvf(per_HDVF_comput<CoefficientType,ComplexType,DegType,FiltrationType>(complex,f, options)) ;
 
        // Export to vtk
        if (options.with_vtk_export)
        {
            std::cout << "----> exporting to vtk" << std::endl ;
            hdvf_persistence_geometric_chain_complex_output_vtk(hdvf, complex, options.outfile_root, options.co_faces) ;
        }
    }
}
 
// Main
 
int main(int argc, char **argv)
{
    if (argc <= 2)
        usage() ;
    else
    {
        for (int i=0;i<argc; ++i)
            std::cout << "arg " << i << " : " << argv[i] << std::endl ;
 
        Options options(read_arguments_hdvf(argc, argv)) ;
        std::cout << "options:" << std::endl << options ;
 
        // ----- Definition of the CoefficientType
#ifndef SCALAR
        if (options.scalar == 0)
        {
            using CoefficientType = int ;
            main_code<CoefficientType>(options) ;
        }
        else if (options.scalar == 2)
        {
            using CoefficientType = CGAL::HDVF::Zp<2,int8_t> ;
            main_code<CoefficientType>(options) ;
        }
        else
        {
            std::cerr << "Z" << options.scalar << " not instantiated, use the #define at line 27" << std::endl ;
        }
#else
        typedef CGAL::HDVF::Zp<SCALAR> CoefficientType;
#endif
    }
 
    return 0 ;
}