FITSFile.h

//-*-C++-*-
/***************************************************************************
 *
 *   Copyright (C) 2008 by Jonathan Khoo
 *   Licensed under the Academic Free License version 2.1
 *
 ***************************************************************************/
 
#ifndef __FITSFile_h
#define __FITSFile_h
 
#include "fitsio.h"
 
#include "Pulsar/Archive.h"
 
#include "dsp/File.h"
#include "dsp/BitSeries.h"
 
namespace dsp
{
 
  class FITSUnpacker;
 
  class FITSFile : public File
  {
    public:
      FITSFile(const char* filename = 0);
 
      bool is_valid(const char* filename) const;
 
      void close ();
 
      void reopen ();
 
      void add_extensions (Extensions*);
      
      unsigned get_samples_in_row() const { return samples_in_row; }
 
      unsigned get_bytes_per_row() { return bytes_per_row; }
 
      class Extension;
 
    protected:
      friend class FITSUnpacker;
 
      virtual void open_file(const char* filename);
 
      virtual int64_t load_bytes(unsigned char* buffer, uint64_t bytes);
 
      int64_t seek_bytes (uint64_t bytes);
 
      void set_samples_in_row (unsigned n) { samples_in_row = n; }
 
      void set_bytes_per_row(const unsigned bytes) { bytes_per_row = bytes; }
 
      void set_data_colnum(const int colnum) { data_colnum = colnum; }
 
      int get_data_colnum() const { return data_colnum; }
 
      Reference::To<Pulsar::Archive> archive;
 
      int data_colnum;
 
      int scl_colnum;
 
      int offs_colnum;
 
      fitsfile* fp;
 
      unsigned samples_in_row;
 
      unsigned bytes_per_row;
 
      uint64_t current_byte;
 
      unsigned current_row;
 
      std::vector<float> dat_scl;
 
      std::vector<float> dat_offs;
 
      unsigned number_of_rows;
      void set_number_of_rows (unsigned N) { number_of_rows = N; }
      unsigned get_number_of_rows () const { return number_of_rows; }
 
      int signint;
 
      float zero_off;
  };
 
  class FITSFile::Extension : public BitSeries::Extension
  {
    public:
 
      class Row
      {
      public:
 
        std::vector<float> dat_scl;
 
        std::vector<float> dat_offs;
 
        uint64_t nsamp;
      };
 
      Extension () { zero_off = 0.0; }
 
      float zero_off;
 
      std::vector<Row> rows;
  };
 
}
 
#endif