WAPPArchive.h

//-*-C++-*-
/***************************************************************************
 *
 *   Copyright (C) 2008 by Paul Demorest
 *   Licensed under the Academic Free License version 2.1
 *
 ***************************************************************************/
 
#ifndef __WAPPArchive_h
#define __WAPPArchive_h
 
#include "Pulsar/BasicArchive.h"
#include "Pulsar/Agent.h"
 
#include "polyco.h"
#include "wapp_headers.h"
 
// Default to DCT if FFTW is present
#define WAPP_USE_FFTW_DCT 1
 
#if WAPP_USE_FFTW_DCT
#  ifdef HAVE_CONFIG_H
#    include <config.h>
#  endif
#  if HAVE_FFTW3
#    include <fftw3.h>
#  else
#    warning "WAPP_USE_FFTW_DCT requires FFTW3, falling back to general R2C"
#    undef WAPP_USE_FFTW_DCT
#    define WAPP_USE_FFTW_DCT 0 
#  endif
#endif
 
namespace Pulsar {


  class WAPPArchive : public BasicArchive {
 
  public:
    
    WAPPArchive ();

    WAPPArchive (const WAPPArchive& archive);

    ~WAPPArchive ();

    const WAPPArchive& operator = (const WAPPArchive& archive);
    
    WAPPArchive (const Archive& archive);

    void copy (const Archive& archive);

    WAPPArchive* clone () const;
 
  protected:

    virtual void load_header (const char* filename);

    void load_polycos();

    virtual Integration*
    load_Integration (const char* filename, unsigned subint);

    void load_extensions();

    bool can_unload () const { return false; }

    virtual void unload_file (const char* filename) const;
 
    // Advocates the use of the ExampleArchive plugin
    class Agent;

    friend class Archive::Advocate<WAPPArchive>;
 
  private:

    void init ();

    polyco hdr_polyco;

    uint32_t wapp_hdr_version;
    uint32_t wapp_hdr_size;
    size_t wapp_ascii_hdr_size; // Includes terminating null byte
    size_t wapp_file_size;

    char *rawhdr;

    WAPP_HEADER *hdr;

    int raw_data_is_lags;

#if WAPP_USE_FFTW_DCT
    fftwf_plan fplan;
    float *fftbuf;
#endif
 
  };
 
 
  // Advocates the use of the ExampleArchive plugin
  class WAPPArchive::Agent : public Archive::Advocate<WAPPArchive> {
    
  public:
    
    Agent () { } 
    
    bool advocate (const char* filename);
    
    std::string get_name () { return "WAPP"; }
    
    std::string get_description ();
    
  };
 
}
 
#endif