mirror of
https://github.com/gwm17/Mask.git
synced 2024-11-13 14:08:49 -05:00
76 lines
1.6 KiB
C++
76 lines
1.6 KiB
C++
#ifndef MASKFILE_H
|
|
#define MASKFILE_H
|
|
|
|
#include <string>
|
|
#include <fstream>
|
|
#include <vector>
|
|
|
|
#include "Nucleus.h"
|
|
|
|
namespace Mask {
|
|
|
|
struct MaskFileHeader {
|
|
int rxn_type = -1;
|
|
int nsamples = -1;
|
|
};
|
|
|
|
struct MaskFileData {
|
|
std::vector<double> E, KE, p, theta, phi; //ordered: target, (if not decay)projectile, ejectile, residual, break1...
|
|
std::vector<int> Z, A;
|
|
std::vector<bool> detect_flag;
|
|
bool eof = false; //flag on end of file
|
|
};
|
|
|
|
class MaskFile {
|
|
public:
|
|
enum class FileType {
|
|
read,
|
|
write,
|
|
append,
|
|
none
|
|
};
|
|
|
|
MaskFile();
|
|
MaskFile(const std::string& name, MaskFile::FileType type);
|
|
bool Open(const std::string& name, MaskFile::FileType type);
|
|
inline bool IsOpen() { return file.is_open(); }
|
|
void Close();
|
|
|
|
void WriteHeader(int rxn_type, int nsamples);
|
|
void WriteData(std::vector<Nucleus>& data);
|
|
void WriteData(MaskFileData& data);
|
|
MaskFileHeader ReadHeader();
|
|
MaskFileData ReadData();
|
|
|
|
|
|
|
|
private:
|
|
|
|
FileType file_type;
|
|
std::string filename;
|
|
unsigned int buffer_position;
|
|
unsigned int buffer_end;
|
|
unsigned int data_size;
|
|
int m_rxn_type;
|
|
int buffersize_bytes;
|
|
|
|
std::fstream file;
|
|
|
|
std::vector<char> data_buffer;
|
|
|
|
static constexpr int onestep_rxn_n = 2;
|
|
static constexpr int twostep_rxn_n = 4;
|
|
static constexpr int threestep_rxn_n = 6;
|
|
|
|
static constexpr int buffersize = 10000; //number of events
|
|
static constexpr int width = 0;
|
|
static constexpr int precision = 3;
|
|
|
|
static constexpr std::size_t double_size = sizeof(double);
|
|
static constexpr std::size_t int_size = sizeof(int);
|
|
static constexpr std::size_t bool_size = sizeof(bool);
|
|
};
|
|
|
|
};
|
|
|
|
#endif |