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