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