OutputFileRoot.hh

//
// This file is part of MARLEY (Model of Argon Reaction Low Energy Yields)
//
// MARLEY is free software: you can redistribute it and/or modify it under the
// terms of version 3 of the GNU General Public License as published by the
// Free Software Foundation.
//
// For the full text of the license please see COPYING or
// visit http://opensource.org/licenses/GPL-3.0
//
// Please respect the MCnet academic usage guidelines. See GUIDELINES
// or visit https://www.montecarlonet.org/GUIDELINES for details.
 
#pragma once
 
// Standard library includes
#include <memory>
 
// ROOT includes
#include "TFile.h"
#include "TTree.h"
 
// MARLEY includes
#include "marley/OutputFile.hh"
 
namespace HepMC3 {
  struct GenEventData;
  struct GenRunInfoData;
};
 
namespace marley {

  class OutputFileRoot : public OutputFile {
 
    public:
 
      OutputFileRoot( const JSON& output_config );
 
      virtual ~OutputFileRoot();
 
      virtual bool resume( std::unique_ptr<marley::Generator>& gen,
        long& num_previous_events ) override;
 
      inline virtual int_fast64_t bytes_written() override
        { return out_tfile_->GetBytesWritten(); }
 
      virtual void write_event( HepMC3::GenEvent* event ) override;
 
    protected:

      virtual void open();

      void clear_event_data();

      std::unique_ptr< TFile > out_tfile_;

      // This is a bare pointer, but ROOT will associate it with out_tfile_, so
      // we don't want to delete it ourselves or let a smart pointer do it.
      TTree* out_tree_ = nullptr;

      std::unique_ptr< HepMC3::GenEventData > temp_event_data_;

      HepMC3::GenEventData* temp_event_data_ptr_ = nullptr;

      std::unique_ptr< HepMC3::GenRunInfoData > temp_run_info_data_;

      std::shared_ptr< HepMC3::GenRunInfo > run_info_;
  };
 
}