CDCRecoTrackFilterModule.cc

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2010 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Dong Van Thanh, ...                                      *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#include "cdc/modules/cdcRecoTrackFilter/CDCRecoTrackFilterModule.h"
#include <framework/datastore/StoreArray.h>
#include <tracking/dataobjects/RecoHitInformation.h>
#include <tracking/dataobjects/RecoTrack.h>
#include <boost/foreach.hpp>
 
using namespace std;
using namespace Belle2;
using namespace CDC;
 
REG_MODULE(CDCRecoTrackFilter)
 
CDCRecoTrackFilterModule::CDCRecoTrackFilterModule() : Module()
{
  setPropertyFlags(c_ParallelProcessingCertified);  // specify this flag if you need parallel processing
  setDescription("use this module to exclude Layers in fitting, after TrackFinding");
  addParam("RecoTracksColName", m_recoTrackArrayName, "Name of collection to hold Belle2::RecoTrack", std::string(""));
  addParam("ExcludeSLayer", m_excludeSLayer, "Super layers (0-8) not used in the fitting", std::vector<unsigned short> {});
  addParam("ExcludeICLayer", m_excludeICLayer, "layers (0-55) not used in the fitting", std::vector<unsigned short> {});
}
 
CDCRecoTrackFilterModule::~CDCRecoTrackFilterModule()
{
}
 
void CDCRecoTrackFilterModule::initialize()
{
  StoreArray<RecoTrack> recoTracks(m_recoTrackArrayName);
  m_recoTrackArrayName = recoTracks.getName();
 
}
 
void CDCRecoTrackFilterModule::beginRun()
{
}
 
void CDCRecoTrackFilterModule::event()
{
  const StoreArray<Belle2::RecoTrack> recoTracks(m_recoTrackArrayName);
 
  // Loop over Recotracks
  int nTr = recoTracks.getEntries();
  for (int i = 0; i < nTr; ++i) {
    const RecoTrack* track = recoTracks[i];
    BOOST_FOREACH(const RecoHitInformation::UsedCDCHit * cdchit, track->getCDCHitList()) {
      WireID wireid(cdchit->getID());
      unsigned short slay = wireid.getISuperLayer();
      unsigned short iclay = wireid.getICLayer();
      for (unsigned short j = 0; j < m_excludeSLayer.size(); ++j) {
        if (slay == m_excludeSLayer.at(j)) {
          track->getRecoHitInformation(cdchit)->setUseInFit(false);
        }
      }
 
      for (unsigned short j = 0; j < m_excludeICLayer.size(); ++j) {
        if (iclay == m_excludeICLayer.at(j)) {
          track->getRecoHitInformation(cdchit)->setUseInFit(false);
        }
      }
 
    }//end of track (Boost_foreach)
  }//end RecoTrack array
}//End Event
void CDCRecoTrackFilterModule::endRun()
{
}
 
void CDCRecoTrackFilterModule::terminate()
{
}