DADABuffer.h

//-*-C++-*-
/***************************************************************************
 *
 *   Copyright (C) 2002 - 2024 by Willem van Straten
 *   Licensed under the Academic Free License version 2.1
 *
 ***************************************************************************/
 
#ifndef __DADABuffer_h
#define __DADABuffer_h
 
#include "dsp/File.h"
#include "dada_hdu.h"
 
namespace dsp {
 
 
  class DADABuffer : public File
  {
 
  public:
 
    DADABuffer ();
 
    ~DADABuffer ();
 
 
    bool is_valid (const char* filename) const;
 
    virtual void open_file (const char* filename);
 
    virtual void open_key (const std::string& key);
 
    void load_header ();
 
 
    virtual void reopen ();
 
    virtual void close ();
 
    virtual void seek (int64_t offset, int whence = 0);
 
    virtual void set_block_size (uint64_t _size);
 
    virtual bool eod() const override;
 
    virtual void set_info (Observation* obs) { info = obs; }
 
    void rewind ();
 
  protected:
    void disconnect ();
 
    virtual int64_t load_bytes (unsigned char* buffer, uint64_t bytes);
 
    virtual int64_t seek_bytes (uint64_t bytes);
 
    virtual void set_total_samples ();
 
    dada_hdu_t* hdu = nullptr;
 
    bool passive = false;
 
    unsigned byte_resolution = 1;
 
  private:
 
    /*
      The following methods and attributes are used only if CUDA is enabled
    */
 
    int64_t load_bytes_device (unsigned char* device_memory, uint64_t bytes, void * device_handle);
 
    void set_require_registered_memory() { require_registered_memory = true; };
 
    unsigned zero_input = 0;
 
    void* zeroed_buffer = nullptr;
 
    uint64_t zeroed_buffer_size = 0;
 
    bool require_registered_memory = false;
 
    bool registered_memory = false;
 
  };
 
}
 
#endif // !defined(__DADABuffer_h)