BsplineEvaluator2D.h

Go to the documentation of this file.

#pragma once
 
#include <vector>
 
#include "warpxm/warpxm_config.h"
#include "lib/spline/spline.h"
#include "lib/wmudgfunctions.h"
 
class BsplineEvaluator2D
{
private:
    std::vector<real> _tx, _ty;              
    int _nx, _ny;                            
    std::vector<real> _c;                    
    int _kx, _ky;                            
    std::vector<real> _lx, _ly;              
    std::vector<real> _hx, _hy;              
    std::vector<std::vector<real>> _wx, _wy; 
    int _mx, _my;                            
public:
    BsplineEvaluator2D(const std::string& filename)
    {
        std::string file = WmUDGFunctions::readFile(filename);
 
        _tx = WmUDGFunctions::parseStringForVector<real>(file, "knotX");
        _ty = WmUDGFunctions::parseStringForVector<real>(file, "knotY");
        _c = WmUDGFunctions::parseStringForVector<real>(file, "coefs");
        _kx = WmUDGFunctions::parseStringForValue<int>(file, "degreeX");
        _ky = WmUDGFunctions::parseStringForValue<int>(file, "degreeY");
        _nx = _tx.size();
        _ny = _ty.size();
        _hx.resize(_kx+1);
        _hy.resize(_ky+1);
 
        // Default is set to having only one evaluated point
        _lx.resize(1);
        _ly.resize(1);
        _wx.resize(1);
        _wx[0].resize(_kx+1);
        _wy.resize(1);
        _wy[0].resize(_ky+1);
        _mx = 1;
        _my = 1;
    }
 
    void eval_spline_2d(const std::vector<real>& x,
                        const std::vector<real>& y,
                        std::vector<std::vector<real>>& spline_value)
    {
        if (x.size() != _mx)
        {
            throw WxExcept("Not allowed to have more than one evaluated point.");
        }
        if (y.size() != _my)
        {
            throw WxExcept("Not allowed to have more than one evaluated point.");
        }
        
        wxm::spline::eval_spline2d(_tx, _ty, _nx, _ny, _c, _kx, _ky, 
                                   _lx, _ly, _hx, _hy, _wx, _wy, 
                                   _mx, _my, x, y, spline_value);
    }
 
}; // class SplineEvaluator