Mower.h

//-*-C++-*-
/***************************************************************************
 *
 *   Copyright (C) 2012 by Willem van Straten
 *   Licensed under the Academic Free License version 2.1
 *
 ***************************************************************************/
 
// psrchive/More/RFIMitigation/Pulsar/MADLM.h
 
#ifndef __Pulsar_MADLM_h
#define __Pulsar_MADLM_h
 
#include "Pulsar/Transformation.h"
#include "Pulsar/Integration.h"
 
#include "TextInterface.h"
#include "Functor.h"
 
namespace Pulsar {
 
 
  class Mower : public Transformation<Integration> {
 
  public:
 
    Mower ();
 
    virtual TextInterface::Parser* get_interface ();
 
    // Text interface to the Mower class
    class Interface : public TextInterface::To<Mower> {
    public:
      Interface (Mower* = 0);
    };
 
    void transform (Integration*);
 
    virtual void set_threshold (float madm);
    float get_threshold () const;
 
    virtual void set_median_smoothing (float turns);
    float get_median_smoothing () const;
 
    virtual void set_broadband (bool);
 
    virtual void set_prune (const PhaseWeight* prune_mask);
    const PhaseWeight* get_prune () const;
 
    virtual void add_precondition( Functor< bool(Profile*,PhaseWeight*) > );
 
  protected:
    
    Reference::To<PhaseWeight> mowed;
 
    Reference::To<const PhaseWeight> prune;
 
    virtual bool build_mask (Profile* profile);
 
    virtual void compute (PhaseWeight* mask, const Profile* difference) = 0;
 
    float median_smoothing_turns;
 
    float cutoff_threshold;
 
    bool broadband;
 
    std::vector< Functor< bool(Profile*,PhaseWeight*) > > precondition;
 
  };
 
}
 
#endif