IterativeBaseline.h

//-*-C++-*-
/***************************************************************************
 *
 *   Copyright (C) 2005 by Willem van Straten
 *   Licensed under the Academic Free License version 2.1
 *
 ***************************************************************************/
 
// psrchive/More/General/Pulsar/IterativeBaseline.h
 
#ifndef __Pulsar_IterativeBaseline_h
#define __Pulsar_IterativeBaseline_h
 
#include "Pulsar/BaselineEstimator.h"
 
namespace Pulsar {
 
  class PhaseWeightSmooth;

  class IterativeBaseline : public BaselineEstimator {
 
  public:

    IterativeBaseline ();

    ~IterativeBaseline ();

    void set_initial_baseline (BaselineEstimator*);
    BaselineEstimator* get_initial_baseline () const;

    virtual void set_threshold (float sigma);
    float get_threshold () const { return threshold; }

    void set_max_iterations (unsigned iterations);
 
    void set_run_postprocessing (bool f) { run_postprocessing = false; }
    bool get_run_postprocessing () const { return run_postprocessing; }
    
  protected:

    void calculate (PhaseWeight* weight);

    virtual void get_bounds (PhaseWeight*, float& lower, float& upper) = 0;
 
    virtual void postprocess (PhaseWeight* weight, const Profile* profile);
    
    float threshold;

    unsigned max_iterations;

    bool run_postprocessing;
    
    Reference::To<BaselineEstimator> initial_baseline;

    bool get_initial_bounds () const { return initial_bounds; }
 
  private:

    bool initial_bounds;
 
  };
 
}
 
#endif