MilleData.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
#pragma once
 
#include <calibration/core/MergeableNamed.h>
 
#include <genfit/MilleBinary.h>
#include <genfit/GblTrajectory.h>
 
#include <string>
#include <vector>
 
namespace Belle2 {
  class MilleData : public MergeableNamed {
  public:
    MilleData(bool doublePrecision = false, bool absFilePaths = false) : MergeableNamed(), m_doublePrecision(doublePrecision),
      m_absFilePaths(absFilePaths) {};
    virtual ~MilleData() { close(); }
 
    virtual void merge(const MergeableNamed* other) override;
    virtual void clear() override { m_files.clear(); m_numRecords = 0; }
    void open(std::string filename);
    bool isOpen() { return !!m_binary; }
    void fill(gbl::GblTrajectory& trajectory);
    void close();
    const std::vector<std::string>& getFiles() const { return m_files; }
    MilleData& operator=(const MilleData& other);
    MilleData(const MilleData& other) : MergeableNamed(other), m_doublePrecision(other.m_doublePrecision),
      m_absFilePaths(other.m_absFilePaths), m_files(other.m_files),
      m_binary(nullptr),
      m_numRecords(other.m_numRecords) {}
 
    virtual MilleData* Clone(const char* newname = "") const override;
 
    int getNumRecords() {return m_numRecords;}
    bool hasDoublePrecision() {return m_doublePrecision;}
 
    void addFile(const std::string& path)
    {
      m_files.push_back(path);
    }
 
  private:
    bool m_doublePrecision{false};
    bool m_absFilePaths{false};
    std::vector<std::string> m_files{}; 
    gbl::MilleBinary* m_binary{nullptr}; 
    int m_numRecords{0};
 
    ClassDefOverride(MilleData, 2) 
  };
}