VSDM.h

/*
 *  Copyright (c) 2000-2022 Inria
 *  All rights reserved.
 *
 *  Redistribution and use in source and binary forms, with or without
 *  modification, are permitted provided that the following conditions are met:
 *
 *  * Redistributions of source code must retain the above copyright notice,
 *  this list of conditions and the following disclaimer.
 *  * Redistributions in binary form must reproduce the above copyright notice,
 *  this list of conditions and the following disclaimer in the documentation
 *  and/or other materials provided with the distribution.
 *  * Neither the name of the ALICE Project-Team nor the names of its
 *  contributors may be used to endorse or promote products derived from this
 *  software without specific prior written permission.
 *
 *  THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 *  AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 *  IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 *  ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
 *  LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 *  CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 *  SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 *  INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 *  CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 *  ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 *  POSSIBILITY OF SUCH DAMAGE.
 *
 *  Contact: Bruno Levy
 *
 *     https://www.inria.fr/fr/bruno-levy
 *
 *     Inria,
 *     Domaine de Voluceau,
 *     78150 Le Chesnay - Rocquencourt
 *     FRANCE
 *
 */
 
#ifndef EXPLORAGRAM_OPTIMAL_TRANSPORT_VSDM
#define EXPLORAGRAM_OPTIMAL_TRANSPORT_VSDM
 
#include <exploragram/basic/common.h>
#include <geogram/delaunay/delaunay.h>
#include <geogram/voronoi/RVD.h>
#include <geogram/numerics/optimizer.h>
#include <geogram/NL/nl_matrix.h>
 
namespace GEO {
    class Mesh;
    class ProgressTask;
 
    class EXPLORAGRAM_API VSDM {
    public:
    VSDM(Mesh* S, Mesh* T);
 
    virtual ~VSDM();
 
    void set_affinity(double x) {
        affinity_ = x;
    }
 
    void optimize(index_t nb_iter);
 
    void set_progress(ProgressTask* progress) {
        progress_ = progress;
    }
 
    void set_subdivision_surface(Mesh* mesh, index_t nb_subdiv);
 
    static void funcgrad_CB(index_t n, double* x, double& f, double* g);
 
    static void newiteration_CB(
        index_t n, const double* x, double f, const double* g, double gnorm
    );
 
    protected:
    static VSDM* instance() {
        return instance_;
    }
 
    virtual void funcgrad(index_t n, double* x, double& f, double* g);
 
    virtual void newiteration();
 
    void add_funcgrad_affinity(index_t n, double* x, double& f, double* g);
 
    static void compute_graph_Laplacian(Mesh* S, NLSparseMatrix* L);
 
    protected:
    Delaunay_var delaunay_;
    RestrictedVoronoiDiagram_var RVD_;
    Mesh* S_;
    Mesh* T_;
    double affinity_;
    double affinity_scaling_;
    ProgressTask* progress_;
    static VSDM* instance_;
    Optimizer_var optimizer_;
    index_t nb_iter_;
    index_t cur_iter_;
 
    NLSparseMatrix L_;
    vector<double> temp_V1_;
    vector<double> temp_V2_;
 
    NLMatrix subd_matrix_;
    Mesh* subd_;
    vector<double> subd_g_;
    };
 
}
 
#endif