ARICHMCParticlesModule.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.                  *
 **************************************************************************/
 
// Own header.
#include <arich/modules/arichMCParticles/ARICHMCParticlesModule.h>
 
// framework - DataStore
#include <framework/datastore/DataStore.h>
 
// framework aux
#include <framework/gearbox/Const.h>
#include <framework/logging/Logger.h>
 
#include <mdst/dataobjects/Track.h>
#include <tracking/dataobjects/ExtHit.h>
#include <mdst/dataobjects/MCParticle.h>
 
namespace Belle2 {
  //-----------------------------------------------------------------
  //-----------------------------------------------------------------
 
  REG_MODULE(ARICHMCParticles);
 
  //-----------------------------------------------------------------
  //                 Implementation
  //-----------------------------------------------------------------
 
  ARICHMCParticlesModule::ARICHMCParticlesModule()
  {
    // set module description (e.g. insert text)
    setDescription("Creates collection of MCParticles related to tracks that hit ARICH.");
    // Set property flags
    setPropertyFlags(c_ParallelProcessingCertified);
 
  }
 
  ARICHMCParticlesModule::~ARICHMCParticlesModule()
  {
  }
 
  void ARICHMCParticlesModule::initialize()
 
  {
    // Dependecies check
    m_tracks.isRequired();
    m_extHits.isRequired();
 
    m_arichMCPs.registerInDataStore();
    m_tracks.registerRelationTo(m_arichMCPs);
 
  }
 
  void ARICHMCParticlesModule::event()
  {
    const Const::EDetector arich = Const::EDetector::ARICH; // arich
    const Const::ChargedStable particleHypothesis = Const::pion;
    const int pdgCode = abs(particleHypothesis.getPDGCode());
 
    for (int itrk = 0; itrk < m_tracks.getEntries(); ++itrk) {
 
      const Track* track = m_tracks[itrk];
      const TrackFitResult* fitResult = track->getTrackFitResultWithClosestMass(particleHypothesis);
      if (!fitResult) {
        B2ERROR("No TrackFitResult for " << particleHypothesis.getPDGCode());
        continue;
      }
 
      const MCParticle* particle = track->getRelated<MCParticle>();
      if (!particle) continue;
 
      RelationVector<ExtHit> extHits = DataStore::getRelationsWithObj<ExtHit>(track);
 
      for (unsigned i = 0; i < extHits.size(); i++) {
        const ExtHit* extHit = extHits[i];
        if (abs(extHit->getPdgCode()) != pdgCode or
            extHit->getDetectorID() != arich or
            extHit->getStatus() != EXT_EXIT or // particles registered at the EXIT of the Al plate
            extHit->getMomentum().Z() < 0.0) continue; // track passes in backward
        if (extHit->getCopyID() == 6789) {
          //MCParticle arichMCP = *particle;
          MCParticle* arichP = m_arichMCPs.appendNew(*particle);
          track->addRelationTo(arichP);
 
        }
 
      }
    }
 
  }
 
} // end Belle2 namespace