PositionMapEvaluator.hpp

Go to the documentation of this file.

#pragma once
#include <memory>
#include <wmtk/Types.hpp>
#include <wmtk/image/Sampling.hpp>
 
namespace wmtk::image {
class Image;
class SamplingAnalyticFunction;
} // namespace wmtk::image
 
namespace wmtk::function::utils {
class PositionMapEvaluator
{
protected:
    std::unique_ptr<wmtk::image::Sampling> m_sampling;
 
public:
    PositionMapEvaluator();
    ~PositionMapEvaluator();
    PositionMapEvaluator(PositionMapEvaluator&&); // move assignment operator
    PositionMapEvaluator& operator=(PositionMapEvaluator&&); // move assignment operator
 
    PositionMapEvaluator(const image::Image& image);
 
    PositionMapEvaluator(
        const wmtk::image::SamplingAnalyticFunction::FunctionType type,
        const double a,
        const double b,
        const double c);
 
 
    // Dont forget to update this if we change autodiff tyeps (add declarations in the cpp)
    template <typename T>
    Vector3<T> uv_to_pos(const Vector2<T>& uv) const
{
    return Vector3<T>(uv.x(), uv.y(), m_sampling->sample(uv.x(), uv.y()));
}
};
 
} // namespace wmtk::function::utils