SplineSmooth.h

//-*-C++-*-
/***************************************************************************
 *
 *   Copyright (C) 2020 by Willem van Straten
 *   Licensed under the Academic Free License version 2.1
 *
 ***************************************************************************/
 
#ifndef __Pulsar_SplineSmooth_h
#define __Pulsar_SplineSmooth_h
 
#include "ReferenceAble.h"
#include "Estimate.h"
 
namespace Pulsar {
 
 
  class SplineSmooth : public Reference::Able
  {
  public:
 
    SplineSmooth();
 
    SplineSmooth (const std::string& json_string);
    
    ~SplineSmooth();
 
    void set_alpha (double _alpha) { alpha = _alpha; }
    double get_alpha () const { return alpha; }
 
    void unload (const std::string&) const;
    void load (const std::string&);
 
  protected:
    
    double evaluate (const std::vector<double>& xval);
 
    template<typename T>
    void new_spline (const std::vector<T>& data_x,
                     const std::vector< Estimate<double> >& data_y);
 
  private:
    
    double alpha;
 
    class Handle;
    Handle* handle;
  };
  
  class SplineSmooth1D : public SplineSmooth
  {
    public:
 
      void fit (const std::vector< double >& data_x,
                     const std::vector< Estimate<double> >& data_y);
 
      double evaluate (double x);
  };
 
  class SplineSmooth2D : public SplineSmooth
  {
    public:
 
    SplineSmooth2D () {}
    
    SplineSmooth2D (const std::string& json) : SplineSmooth (json) {}
 
    void fit (const std::vector< std::pair<double,double> >& data_x,
              const std::vector< Estimate<double> >& data_y);
    
    double evaluate ( const std::pair<double,double>& );
  };
 
 
  class CrossValidatedSmooth2D
  {
    bool logarithmic;             // smoothing factors on logarithmic scale
    
    unsigned npartition;          // m=40 in Clark (1977)
    double validation_fraction;   // 0.1 in Clark (1977)
    
    SplineSmooth2D* spline;       // the 2-D spline implementation
 
    double iqr_threshold;         // Tukey's fence used to detect outliers
    unsigned nflagged_iqr;
 
    double gof_step_threshold;
    unsigned nflagged_gof;
    
    std::vector<double> gof_tot;
    std::vector<unsigned> gof_count;
 
    std::string gof_filename;
    std::ofstream* gof_out;
 
  public:
    
    CrossValidatedSmooth2D ();
 
    void set_gof_filename (const std::string& name) { gof_filename = name; }
    
    void set_spline (SplineSmooth2D* _spline) { spline = _spline; }
 
 
    void set_npartition (unsigned m) { npartition = m; }
    unsigned get_npartition () const { return npartition; }
 
 
    void set_validation_fraction (double f) { validation_fraction = f; }
    double get_validation_fraction () const { return validation_fraction; }
 
 
    void set_iqr_threshold (double t) { iqr_threshold = t; }
    double get_iqr_threshold () const { return iqr_threshold; }
 
 
    void set_gof_step_threshold (double s) { gof_step_threshold = s; }
    double get_gof_step_threshold () const { return gof_step_threshold; }
 
    void fit ( std::vector< std::pair<double,double> >& data_x,
               std::vector< Estimate<double> >& data_y );
 
    void remove_iqr_outliers
    ( std::vector< std::pair<double,double> >& x,
      std::vector< Estimate<double> >& y );
 
    unsigned get_nflagged_iqr () const { return nflagged_iqr; }
    
    void find_optimal_smoothing_factor
    ( const std::vector< std::pair<double,double> >& dat_x,
      const std::vector< Estimate<double> >& dat_y );
    
    void remove_gof_outliers
    ( std::vector< std::pair<double,double> >& x,
      std::vector< Estimate<double> >& y );
 
    unsigned get_nflagged_gof () const { return nflagged_gof; }
 
    double get_mean_gof (const std::vector< std::pair<double,double> >& data_x,
                         const std::vector< Estimate<double> >& data_y);
 
    double get_mean_gof (double log_10_alpha,
                         const std::vector< std::pair<double,double> >& data_x,
                         const std::vector< Estimate<double> >& data_y);
    
  };
 
  class BootstrapUncertainty2D
  {
    unsigned nsample;
    SplineSmooth2D* spline;      // the spline implementation
  
  public:
    
    BootstrapUncertainty2D ();
 
    void set_spline (SplineSmooth2D* _spline) { spline = _spline; }
  
    void get_uncertainty (const std::vector< std::pair<double,double> >& data_x,
                          std::vector< Estimate<double> >& data_y);
 
  };
  
}
 
#endif