BeamBackgroundFileParserV2.h

#ifndef BeamBackgroundFileParserV2_h
#define BeamBackgroundFileParserV2_h

#include "IBeamBackgroundFileParser.h"
#include "TFile.h"
#include "TTree.h"
#include <fstream>
#include <filesystem>
#include <random>

class BeamBackgroundFileParserV2: public IBeamBackgroundFileParser {
public:
    BeamBackgroundFileParserV2(const std::string& dirname, const std::string& treename, double beam_energy);

    bool load(IBeamBackgroundFileParser::BeamBackgroundData&) { return 0; }
    bool load(IBeamBackgroundFileParser::BeamBackgroundData&, int iEntry);
    bool SampleParticleNum(int&, int&);
    //int totalEnteries() { return m_readTree->GetEntries(); }

private:
    std::unique_ptr<TFile> m_inputFile;
    TTree* m_readTree;
    double m_beam_energy;

    float x, y, z, px, py, dp, pid;

};

#endif