He3tubeModule.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 <beast/he3tube/modules/He3tubeModule.h>
#include <beast/he3tube/dataobjects/He3tubeSimHit.h>
#include <mdst/dataobjects/MCParticle.h>
#include <framework/datastore/StoreArray.h>
#include <framework/datastore/RelationArray.h>
#include <framework/datastore/RelationIndex.h>
#include <framework/logging/Logger.h>
 
using namespace std;
using namespace Belle2;
using namespace Belle2::he3tube;
 
//We have to register the module to the Framework. The "Module" part of the
//class name will be appended automatically so every module hast to be named
//XxxModule
REG_MODULE(He3tube);
 
 
He3tubeModule::He3tubeModule() : Module(), m_intParameter(0), m_doubleParameter(0), m_stringParameter("")
{
  setDescription("Creates Helium-3 tube - sub-detector of BEASTII");
 
  //We can define parameters which can be set from the steering file. The arguments are:
  // name, reference to the variable where the value will be stored, description, default value
  //If the default value is omitted the user has to specify this parameter, otherwise an error is produced
  addParam("intParameter", m_intParameter,
           "Useless parameter of type integer", 0);
  addParam("doubleParameter", m_doubleParameter,
           "Useless parameter of type double", 0.0);
  addParam("stringParameter", m_stringParameter,
           "Useless parameter of type string", string(""));
  addParam("doubleListParameter", m_doubleListParameter,
           "Useless parameter of type vector<double>", vector<double>(3, 0));
 
  //Valid parameter types are int, double, string, bool and vectors of any of those
}
 
void He3tubeModule::initialize()
{
  B2INFO("He3tube: Initialize");
  //Here you can do some stuff before processing starts. If you want to
  //write to some collections of the DataStore you have to register these
  //here by using StoreArray<T>::registerPersistent() for collections which
  //should be written to the output file by default or
  //StoreArray<T>::registerTransient() for collections which will not be
  //saved by default. If one just wants to access collections one should
  //check if they were registered by using the isRequired member
 
  StoreArray<MCParticle>   mcParticles;
  StoreArray<He3tubeSimHit>  simHits;
  RelationArray relMCSimHit(mcParticles, simHits);
  if (!(mcParticles.isRequired() && simHits.isRequired() && relMCSimHit.isRequired())) {
    //Fatal is not necessary here as the storeArrays should just look
    //empty if not registered but let's make sure everything is present
    B2FATAL("Not all collections found, exiting processing");
  }
}
 
void He3tubeModule::beginRun()
{
  B2INFO("He3tube: Begin of new run");
  //Here comes the initialisation specific to each run
}
 
void He3tubeModule::event()
{
  B2INFO("He3tube: Event is being processed");
  //Here comes the actual event processing
 
  StoreArray<MCParticle>   mcParticles;
  StoreArray<He3tubeSimHit>  simHits;
 
  //RelationIndex is a readonly, bidirectional index for a Relation so that one
  //can easily use the RelationArray without looping over it manually.
  RelationIndex<MCParticle, He3tubeSimHit> relMCSimHit(mcParticles, simHits);
 
  //Lets loop over all created He3tubeSimHits:
  //int nSimHits = simHits.getEntries();
  //for (int i = 0; i < nSimHits; ++i) {
  //He3tubeSimHit& hit = *simHits[i];
  //Find all MCParticles which point to that SimHit and the corresponding weight
  //RelationIndex<MCParticle, He3tubeSimHit>::range_to range = relMCSimHit.getElementsTo(hit);
  //for (; range.first != range.second; ++range.first) {
  //And Print something about the relation
  //const RelationIndex<MCParticle, He3tubeSimHit>::Element& relation = *range.first;
  //B2INFO("He3tubeSimHit #" << i << " has an energy deposition of " << hit.getEnergyDep()
  //<< " and is related to MCParticle #" << relation.indexFrom
  //<< " which has an PDG code of " << relation.from->getPDG());
  //}
  //}
 
  //Now let's do it the other way round:
  int nMCParticles = mcParticles.getEntries();
  for (int i = 0; i < nMCParticles; ++i) {
    MCParticle& mcp = *mcParticles[i];
    // Find all He3tubeSimHits which point from that MCParticle.
    typedef RelationIndex<MCParticle, He3tubeSimHit>::Element relMCSimHit_Element;
    for (const relMCSimHit_Element& relation : relMCSimHit.getElementsFrom(mcp)) {
      B2INFO("MCParticle #" << i << " created the AwesomSimHit #" << relation.indexTo
             << " which has an energy deposition of " << relation.to->getEnergyDep());
    }
  }
}
 
void He3tubeModule::endRun()
{
  B2INFO("He3tube: End of run");
  //Here cleanup after each run
}
 
 
void He3tubeModule::terminate()
{
  B2INFO("He3tube: Terminate");
  //Here final cleanup
}