FrequencyIntegrate.h

//-*-C++-*-
/***************************************************************************
 *
 *   Copyright (C) 2005 by Willem van Straten
 *   Licensed under the Academic Free License version 2.1
 *
 ***************************************************************************/
 
// psrchive/More/General/Pulsar/FrequencyIntegrate.h
 
#ifndef __Pulsar_FrequencyIntegrate_h
#define __Pulsar_FrequencyIntegrate_h
 
#include "Pulsar/Integrate.h"
#include "Pulsar/EvenlySpaced.h"
#include "Pulsar/EvenlyWeighted.h"
#include "Pulsar/NonlinearlySpaced.h"
 
#include "Pulsar/Integration.h"
 
namespace Pulsar {
 
  class Integration;


  class FrequencyIntegrate : public Integrate<Integration> {
 
  public:

    FrequencyIntegrate ();

    void transform (Integration*);

    void set_dedisperse (bool);
    bool get_dedisperse () const;

    void set_defaraday (bool flag);
    bool get_defaraday () const;

    class EvenlySpaced;

    class EvenlyWeighted;

    class WavelengthSquaredSpaced;
    
  protected:
 
    bool dedisperse;
    bool defaraday;
 
  };
 
  class FrequencyIntegrate::EvenlySpaced :
    public Integrate<Integration>::EvenlySpaced
  {
    unsigned get_size (const Integration* sub)
    { return sub->get_nchan(); }
  };
 
  class FrequencyIntegrate::EvenlyWeighted : 
    public Integrate<Integration>::EvenlyWeighted
  {
    unsigned get_size (const Integration* sub)
    { return sub->get_nchan(); }
 
    double get_weight (const Integration* sub, unsigned ichan)
    { return sub->get_weight (ichan); }
  };
 
  class FrequencyIntegrate::WavelengthSquaredSpaced : 
    public Integrate<Integration>::NonlinearlySpaced
  {
    unsigned get_size (const Integration* sub)
    { return sub->get_nchan(); }
 
    double get_value (const Integration* sub, unsigned ichan)
    { double freq = sub->get_centre_frequency (ichan); return 1.0 / (freq*freq); }
  };
  
}
 
#endif