TouschekTURTLEInputModule.cc

/**************************************************************************
 * 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.                  *
 **************************************************************************/
 
#include <generators/modules/TouschekTURTLEInputModule.h>
 
#include <framework/datastore/StoreArray.h>
 
#include <framework/gearbox/Gearbox.h>
#include <framework/gearbox/GearDir.h>
#include <framework/gearbox/Unit.h>
#include <framework/utilities/FileSystem.h>
 
#include <generators/touschek/TouschekReaderTURTLE.h>
 
#include <fstream>
 
using namespace std;
using namespace Belle2;
 
//-----------------------------------------------------------------
//                 Register the Module
//-----------------------------------------------------------------
REG_MODULE(TouschekTURTLEInput);
 
//-----------------------------------------------------------------
//                 Implementation
//-----------------------------------------------------------------
 
TouschekTURTLEInputModule::TouschekTURTLEInputModule() : Module()
{
  //Set module properties
  setDescription("Reads the Touschek data from a TURTLE file and stores it into the MCParticle collection.");
  setPropertyFlags(c_Input);
 
  //Parameter definition
  addParam("FilenameHER",  m_filenameHER, "The filename of the HER TURTLE input file.");
  addParam("FilenameLER",  m_filenameLER, "The filename of the LER TURTLE input file.");
  addParam("ReadHER", m_readHER, "Set to false to skip reading the HER data.", true);
  addParam("ReadLER", m_readLER, "Set to false to skip reading the LER data.", true);
  addParam("zCoordinate", m_zPos,
           "Indicates the z-coordinate of the TURTLE particles in the file. (default is 50cm, 400cm is also implemented.)", 50.0);
  addParam("MaxParticles", m_maxParticles,
           "The maximum number of particles per event that should be read. -1 means all of the particles are read.", -1);
 
  //Create and initialize member variables
  m_herPipePartMatrix = new TGeoHMatrix("TouschekPlaneHER");
  m_lerPipePartMatrix = new TGeoHMatrix("TouschekPlaneLER");
  m_readerHER = new TouschekReaderTURTLE(m_herPipePartMatrix, 11);  //HER: electrons
  m_readerLER = new TouschekReaderTURTLE(m_lerPipePartMatrix, -11); //LER: positrons
}
 
 
TouschekTURTLEInputModule::~TouschekTURTLEInputModule()
{
  delete m_readerHER;
  delete m_readerLER;
}
 
 
void TouschekTURTLEInputModule::initialize()
{
  //Check if the Gearbox is ready for reading
  if (!Gearbox::getInstance().isOpen()) {
    B2FATAL("The Touschek Turtle input module requires a valid Gearbox. Please make sure you have the Gearbox module added to your path.");
  }
 
  //Initialize MCParticle collection
  StoreArray<MCParticle> mcparticle;
  mcparticle.registerInDataStore();
 
  //Check parameters
  if ((m_readHER) && (!FileSystem::fileExists(m_filenameHER))) {
    B2ERROR("Parameter <FilenameHER>: Could not open the file. The filename " << m_filenameHER << " does not exist !");
  } else m_readerHER->open(m_filenameHER);
 
  if ((m_readLER) && (!FileSystem::fileExists(m_filenameLER))) {
    B2ERROR("Parameter <FilenameLER>: Could not open the file. The filename " << m_filenameLER << " does not exist !");
  } else m_readerLER->open(m_filenameLER);
 
  //Get the transformation from local Touschek plane space to global geant4 space
  //For the HER
  GearDir irDir = Gearbox::getInstance().getDetectorComponent("Cryostat");
  double angleher = irDir.getAngle("AngleHER");
  double angleler = irDir.getAngle("AngleLER");
  m_herPipePartMatrix->RotateY(angleher / Unit::deg);
  m_lerPipePartMatrix->RotateY(angleler / Unit::deg);
}
 
 
void TouschekTURTLEInputModule::event()
{
  try {
    int readHERParticles = 0;
    int readLERParticles = 0;
 
    //Read the data
    MCParticleGraph mpg;
    if (m_readHER) readHERParticles = m_readerHER->getParticles(m_maxParticles, mpg); //HER: electrons
    if (m_readLER) readLERParticles = m_readerLER->getParticles(m_maxParticles, mpg); //LER: positrons
 
    if ((readHERParticles > 0) || (readLERParticles > 0)) {
      //Generate MCParticle list
      mpg.generateList("", MCParticleGraph::c_setDecayInfo | MCParticleGraph::c_checkCyclic);
 
      B2INFO("Read " << readHERParticles << " e- particles (HER).");
      B2INFO("Read " << readLERParticles << " e+ particles (LER).");
    }
  } catch (runtime_error& exc) {
    B2ERROR(exc.what());
  }
}