RobustStepFinder.h

//-*-C++-*-
/***************************************************************************
 *
 *   Copyright (C) 2021 by Willem van Straten
 *   Licensed under the Academic Free License version 2.1
 *
 ***************************************************************************/
 
// psrchive/More/Polarimetry/Pulsar/RobustStepFinder.h
 
#ifndef __Pulsar_RobustStepFinder_H
#define __Pulsar_RobustStepFinder_H
 
#include "Pulsar/SystemCalibratorStepFinder.h"
 
namespace Pulsar
{
  class RobustStepFinder : public SystemCalibrator::StepFinder
  {
    float step_threshold;

    unsigned depth;

    SystemCalibrator* calibrator;

    std::vector<unsigned> psr_before;
    
    std::vector<unsigned> psr_after;

    std::vector<unsigned> cal_before;
    
    std::vector<unsigned> cal_after;

    std::vector<unsigned> compare;
    
    template<typename Container>
    void count_consistent (const Container& container,
                           std::vector<unsigned>& before,
                           std::vector<unsigned>& after,
                           bool wedge);
 
    template<typename Container>
    void remove_inconsistent (Container& container,
                              std::vector<unsigned>& before,
                              std::vector<unsigned>& after);

    void find_steps (std::vector<unsigned>& steps,
                     const std::vector<unsigned>& before,
                     const std::vector<unsigned>& after);
 
    template<class Container>
    void add_steps (std::vector<MJD>& steps,
                    std::vector<unsigned>& step_sub,
                    Container& data,
                    bool align_to_cal = 0);
 
    // adjust epochs and ids to align to calibrator observations
    bool align_to_cal (MJD& i_epoch, std::string& i_id,
                       MJD& j_epoch, std::string& j_id);
      

 
    void find_steps_pulsar (std::vector<MJD>& steps);
    void find_steps_calibrator (std::vector<MJD>& steps);

    void insert_steps (std::vector<MJD>& steps,
                       Calibration::VariableBackend* xform);
 
    void remove_extra_calibrators ();
    void remove_outliers ();

    void insert_steps ();
 
  public:

    RobustStepFinder (float threshold = 3.0)
    {
      step_threshold = threshold;
      depth = 3;
    }
 
    void set_step_threshold (float val) { step_threshold = val; }
    
    void process (SystemCalibrator*);
 
  };
}
 
#endif