controlbasis.hpp

Go to the documentation of this file.

#include "defs.hpp"
#include <stdio.h>
#include <stdlib.h>
#include <math.h>
#include <vector>
#include <cassert>
#include "util.hpp"
#pragma once
 
 
class ControlBasis {
    protected:
        int nparams; 
        double tstart; 
        double tstop; 
        int skip; 
        ControlType controltype; 
        bool enforceZeroBoundary; 
 
    public: 
        ControlBasis();
 
        ControlBasis(int nparams_, double tstart, double tstop, bool enforceZeroBoundary);
 
        virtual ~ControlBasis();
 
        int getNparams() {return nparams; };
 
        double getTstart() {return tstart; };
 
        double getTstop() {return tstop; };
 
        ControlType getType() {return controltype;};
 
        void setSkip(int skip_) {skip = skip_;};
 
        int getSkip(){return skip;};
 
        virtual int getNSplines() {return 0;};
 
        virtual double computeVariation(std::vector<double>& /*params*/, int /*carrierfreqID*/){return 0.0;};
 
        virtual void computeVariation_diff(double* /*grad*/, std::vector<double>& /*params*/, double /*var_bar*/, int /*carrierfreqID*/){};
 
        virtual void enforceBoundary(double* /*x*/, int /*carrier_id*/) {};
 
        virtual void evaluate(const double t, const std::vector<double>& coeff, int carrier_freq_id, double* Blt1, double*Blt2) = 0;
 
        virtual void derivative(const double t, const std::vector<double>& coeff, double* coeff_diff, const double valbar1, const double valbar2, int carrier_freq_id)= 0;
};
 
class BSpline2nd : public ControlBasis {
    protected:
        int nsplines; 
        double dtknot; 
        double *tcenter; 
        double width; 
 
        double basisfunction(int id, double t);
 
    public:
        BSpline2nd(int nsplines, double tstart, double tstop, bool enforceZeroBoundary);
 
        ~BSpline2nd();
        
        int getNSplines() {return nsplines;};
 
        void enforceBoundary(double* x, int carrier_id);
 
        void evaluate(const double t, const std::vector<double>& coeff, int carrier_freq_id, double* Blt1_ptr, double* Blt2_ptr);
 
        void derivative(const double t, const std::vector<double>& coeff, double* coeff_diff, const double valbar1, const double valbar2, int carrier_freq_id);
};
 
class BSpline2ndAmplitude : public ControlBasis {
    protected:
        int nsplines; 
        double dtknot; 
        double *tcenter; 
        double width; 
        double scaling; 
 
        double basisfunction(int id, double t);
 
    public:
        BSpline2ndAmplitude(int nsplines, double scaling, double tstart, double tstop, bool enforceZeroBoundary);
 
        ~BSpline2ndAmplitude();
 
        int getNSplines() {return nsplines;};
 
        void enforceBoundary(double* x, int carrier_id);
 
        void evaluate(const double t, const std::vector<double>& coeff, int carrier_freq_id, double* Blt1_ptr, double* Blt2_ptr);
 
        void derivative(const double t, const std::vector<double>& coeff, double* coeff_diff, const double valbar1, const double valbar2, int carrier_freq_id);
};
 
class BSpline0 : public ControlBasis {
    protected:
        int nsplines; 
        double dtknot; 
        double width; 
 
    public:
        BSpline0(int nsplines, double tstart, double tstop, bool enforceZeroBoundary);
        ~BSpline0();
 
        int getNSplines() {return nsplines;};
 
        void enforceBoundary(double* x, int carrier_id);
 
        double computeVariation(std::vector<double>& params, int carrierfreqID);
 
        virtual void computeVariation_diff(double* grad, std::vector<double>&params, double var_bar, int carrierfreqID);
 
        void evaluate(const double t, const std::vector<double>& coeff, int carrier_freq_id, double* Blt1_ptr, double* Blt2_ptr);
 
        void derivative(const double t, const std::vector<double>& coeff, double* coeff_diff, const double valbar1, const double valbar2, int carrier_freq_id);
};