bspline.h

Go to the documentation of this file.

/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *\
 *                                                                           *
 *                       / /   / /    __    \ \  / /                         *
 *                      / /__ / /   / _ \    \ \/ /                          *
 *                     /  ___  /   | |/_/    / /\ \                          *
 *                    / /   / /    \_\      / /  \ \                         *
 *                                                                           *
 *                         Jakub Benda (c) 2014                              *
 *                     Charles University in Prague                          *
 *                                                                           *
\* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */

#ifndef HEX_BSPLINE
#define HEX_BSPLINE

#include "arrays.h"
#include "complex.h"
#include "misc.h"

class Bspline
{
    public:
    
        Bspline (int order, rArrayView const & rknots, double th, rArrayView const & cknots);
        
        Complex bspline(int i, int iknot, int k, Complex r) const;
        
        Complex dspline(int i, int iknot, int k, Complex r) const;
        
        void B(int i, int iknot, int n, const Complex* x, Complex* y) const;
        
        void dB(int i, int iknot, int n, const Complex* x, Complex* y) const;
        
        inline Complex rotate (double r) const
        {
            return (r <= R0_) ? r : R0_ + (r - R0_) * rotation_;
        };
        
        inline double unrotate (Complex z) const
        {
            return (z.imag() == 0.) ? z.real() : R0_ + ((z - R0_) / rotation_).real();
        };
        
        cArray zip (const cArrayView coeff, const rArrayView grid) const;
        
        cArray zip (const cArrayView coeff, const rArrayView xgrid, const rArrayView ygrid) const;
        
        void writeVTK (std::ofstream & out, const cArrayView coeff, const rArrayView xgrid, const rArrayView ygrid) const;
        
        int knot (Complex x) const;
        
        Complex eval(const cArrayView coeff, double x) const;
        
        Complex eval(const cArrayView coeff, double x, double y) const;
        
        // getters
        
        inline Complex const & t (int i) const { return *(t_ + i); }
        
        inline int Nspline() const { return Nspline_; }
        
        inline int Nknot() const { return Nknot_; }
        
        inline int Nreknot() const { return Nreknot_; }
        
        inline int order() const { return order_; }
        
        inline double R0() const { return R0_; };
        
        inline double Rmax() const { return Rmax_; };
        
        inline double ECStheta() const { return theta_; }
        
        inline rArray const & rknots() const { return rknots_; }
        
        inline rArray const & cknots() const { return cknots_; }
    
    private:
    
        rArray rknots_;
        
        rArray cknots_;
        
        Complex * restrict t_;
        
        double theta_;
        
        Complex rotation_;
        
        double R0_;
        
        double Rmax_;
        
        int Nknot_;
        
        int Nreknot_;
        
        int Nspline_;
        
        int Nintval_;
        
        int order_;
};

#endif