MCEventHeader.cxx

Go to the documentation of this file.

// Copyright 2019-2020 CERN and copyright holders of ALICE O2.
// See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
// All rights not expressly granted are reserved.
//
// This software is distributed under the terms of the GNU General Public
// License v3 (GPL Version 3), copied verbatim in the file "COPYING".
//
// In applying this license CERN does not waive the privileges and immunities
// granted to it by virtue of its status as an Intergovernmental Organization
// or submit itself to any jurisdiction.
 
 
#include "SimulationDataFormat/MCEventHeader.h"
#include "FairRootManager.h"
#include <TFile.h>
#include <TTree.h>
#include <iostream>
 
namespace o2
{
namespace dataformats
{
 
/*****************************************************************/
/*****************************************************************/
 
void MCEventHeader::Reset()
{
  FairMCEventHeader::Reset();
 
  clearInfo();
  mEmbeddingFileName.clear();
  mEmbeddingEventIndex = 0;
}
 
void MCEventHeader::extractFileFromKinematics(std::string_view kinefilename, std::string_view targetfilename)
{
  auto oldfile = TFile::Open(kinefilename.data());
  auto kinetree = (TTree*)oldfile->Get("o2sim");
  // deactivate all branches
  kinetree->SetBranchStatus("*", 0);
  // activate the header branch
  kinetree->SetBranchStatus("MCEventHeader*", 1);
  // create a new file + a clone of old tree header. Do not copy events
  auto newfile = TFile::Open(targetfilename.data(), "RECREATE");
  auto newtree = kinetree->CloneTree(0);
  // here we copy the branches
  newtree->CopyEntries(kinetree, kinetree->GetEntries());
  newtree->SetEntries(kinetree->GetEntries());
  // flush to disk
  newtree->Write();
  newfile->Close();
  delete newfile;
 
  // clean
  if (oldfile) {
    oldfile->Close();
    delete oldfile;
  }
}
 
void MCEventHeader::print() const
{
  // print some used parts from FairMCEventHeader
  std::cout << "RunID " << fRunId << "\n";
  std::cout << "EventID " << fEventId << "\n";
  std::cout << "Vertex-X " << fX << "\n";
  std::cout << "Vertex-Y " << fY << "\n";
  std::cout << "Vertex-Z " << fZ << "\n";
  std::cout << "Vertex-T " << fT << "\n";
  std::cout << "Impact-B " << fB << "\n";
  std::cout << "NPrim " << fNPrim << "\n";
  // print meta-fields
  printInfo();
}
 
/*****************************************************************/
/*****************************************************************/
 
} /* namespace dataformats */
} /* namespace o2 */
 
ClassImp(o2::dataformats::MCEventHeader);