nearest_neighbors.h

#ifndef __NEAREST_NEIGHBORS__
#define __NEAREST_NEIGHBORS__

#include <eigen/Eigen/Dense>
#include <deal.II/lac/la_parallel_vector.h>
#include "min_max_scaler.h"

namespace PHiLiP {
namespace ProperOrthogonalDecomposition {
using Eigen::MatrixXd;
using Eigen::VectorXd;
using Eigen::RowVectorXd;

class NearestNeighbors
{
public:
    NearestNeighbors();

    void update_snapshots(const MatrixXd& snapshot_parameters, dealii::LinearAlgebra::distributed::Vector<double> snapshot);

    MatrixXd kPairwiseNearestNeighborsMidpoint();

    MatrixXd kNearestNeighborsMidpoint(const RowVectorXd& point);

    //Given a point, return the index of nearest snapshot_parameter
    dealii::LinearAlgebra::distributed::Vector<double> nearestNeighborMidpointSolution(const RowVectorXd& point);

    MatrixXd snapshot_params;

    MatrixXd scaled_snapshot_params;

    MinMaxScaler scaler;

    std::vector<dealii::LinearAlgebra::distributed::Vector<double>> snapshots;

};

}
}

#endif