CRNCubicSpline.h

Go to the documentation of this file.

/* Copyright 2011-2016 CoReNum, INSA-Lyon, Université Paris Descartes, ENS-Lyon
 * 
 * This file is part of libcrn.
 * 
 * libcrn is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 * 
 * libcrn is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Lesser General Public License for more details.
 * 
 * You should have received a copy of the GNU Lesser General Public License
 * along with libcrn.  If not, see <http://www.gnu.org/licenses/>.
 * 
 * file: CRNCubicSpline.h
 * \author Yann LEYDIER
 */

#ifndef CRNCubicSpline_HEADER
#define CRNCubicSpline_HEADER

#include <CRNMath/CRNCartesian2DFunction.h>
#include <CRNGeometry/CRNPoint2DDouble.h>
#include <CRNException.h>

namespace crn
{
    class CubicSpline: public Cartesian2DFunction
    {
        public:
            enum class Extrapolation { LINEAR, CUBIC };

            template<typename Iter> CubicSpline(Iter beg, Iter en):
                data(std::distance(beg, en)),
                extrapolation(Extrapolation::CUBIC)
            {
                if (data.size() < 3)
                    throw crn::ExceptionDimension("There must be at least 3 samples to create a spline");
                size_t cnt = 0;
                for (Iter tmp = beg; tmp != en; ++tmp)
                    data[cnt++] = makePoint(*tmp);
                computeCoeffs();
            }
            CubicSpline(const CubicSpline&) = default;
            CubicSpline(CubicSpline&&) = default;
            virtual ~CubicSpline() override {}
            CubicSpline& operator=(const CubicSpline&) = default;
            CubicSpline& operator=(CubicSpline&&) = default;

            void SetExtrapolationMode(Extrapolation ex) noexcept { extrapolation = ex; }

            virtual double operator[](double x) const override;
            using Cartesian2DFunction::operator[];

            const std::vector<crn::Point2DDouble>& GetData() const noexcept { return data; }

        private:
            template<typename T> Point2DDouble makePoint(const crn::Point2D<T> &p)
            { return crn::Point2DDouble(double(p.X), double(p.Y)); }
            template<typename T> Point2DDouble makePoint(const crn::Point2D<T> *p)
            { return crn::Point2DDouble(double(p->X), double(p->Y)); }
            template<typename T, typename Y> Point2DDouble makePoint(const std::pair<T, Y> &p)
            { return crn::Point2DDouble(double(p.first), double(p.second)); }
            void computeCoeffs();

            std::vector<double> z; 
            std::vector<crn::Point2DDouble> data; 
            Extrapolation extrapolation; 
            CRN_DECLARE_CLASS_CONSTRUCTOR(CubicSpline)
    };
    template<> struct IsClonable<CubicSpline> : public std::true_type {};

    CRN_ALIAS_SMART_PTR(CubicSpline)
}

#endif