raw_buffer.h

#pragma once

using namespace std;

/*
  The RawBuffer stores data from a raw file in pinned memory.

  Its format is row-major:
    input[block][antenna][coarse-channel][time-within-block][polarization][real or imag]
*/
class RawBuffer {
 public:
  const int num_blocks;
  const int num_antennas;
  const int num_coarse_channels;
  const int timesteps_per_block;
  const int npol;
  
  int8_t* data;
  size_t size;
  
  RawBuffer(int num_blocks, int num_antennas, int num_coarse_channels,
            int timesteps_per_block, int npol);

  ~RawBuffer();

  // Returns a pointer to where data for the given block should start
  char* blockPointer(int block) const;

  void set(int block, int antenna, int coarse_channel,
           int timestep, int pol, bool imag, int8_t value);
};

// Helper function to determine memory size needed
size_t rawBufferSize(int num_blocks, int num_antennas, int num_coarse_channels,
                     int timesteps_per_block, int npol);