StoreArrayMerger.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 <tracking/modules/vxdCDCTrackMerger/StoreArrayMerger.h>
 
#include <tracking/trackFindingCDC/utilities/StringManipulation.h>
#include <tracking/trackFindingCDC/utilities/Algorithms.h>
 
#include <framework/core/ModuleParamList.templateDetails.h>
 
using namespace Belle2;
 
void StoreArrayMerger::exposeParameters(ModuleParamList* moduleParamList, const std::string& prefix)
{
  Super::exposeParameters(moduleParamList, prefix);
 
  // CDC input tracks
  moduleParamList->addParameter(TrackFindingCDC::prefixed(prefix, "CDCRecoTrackStoreArrayName"), m_param_cdcRecoTrackStoreArrayName,
                                "StoreArray name of the CDC Track Store Array", m_param_cdcRecoTrackStoreArrayName);
  // VXD input tracks
  moduleParamList->addParameter(TrackFindingCDC::prefixed(prefix, "VXDRecoTrackStoreArrayName"), m_param_vxdRecoTrackStoreArrayName,
                                "StoreArray name of the VXD Track Store Array", m_param_vxdRecoTrackStoreArrayName);
}
 
void StoreArrayMerger::initialize()
{
  m_cdcRecoTracks.isRequired(m_param_cdcRecoTrackStoreArrayName);
  m_vxdRecoTracks.isRequired(m_param_vxdRecoTrackStoreArrayName);
 
  m_cdcRecoTracks.registerRelationTo(m_vxdRecoTracks);
  m_vxdRecoTracks.registerRelationTo(m_cdcRecoTracks);
 
  Super::initialize();
}
 
void StoreArrayMerger::apply(std::vector<RecoTrack*>& cdcRecoTrackVector, std::vector<RecoTrack*>& vxdRecoTrackVector)
{
  cdcRecoTrackVector.reserve(m_cdcRecoTracks.getEntries());
  vxdRecoTrackVector.reserve(m_vxdRecoTracks.getEntries());
 
  for (RecoTrack& recoTrack : m_cdcRecoTracks) {
    cdcRecoTrackVector.push_back(&recoTrack);
  }
 
  for (RecoTrack& recoTrack : m_vxdRecoTracks) {
    vxdRecoTrackVector.push_back(&recoTrack);
  }
}
 
void StoreArrayMerger::removeCDCRecoTracksWithPartner(std::vector<RecoTrack*>& tracks)
{
  removeRecoTracksWithPartner(tracks, m_param_vxdRecoTrackStoreArrayName);
}
 
void StoreArrayMerger::removeVXDRecoTracksWithPartner(std::vector<RecoTrack*>& tracks)
{
  removeRecoTracksWithPartner(tracks, m_param_cdcRecoTrackStoreArrayName);
}
 
void StoreArrayMerger::removeRecoTracksWithPartner(std::vector<RecoTrack*>& tracks, const std::string& partnerStoreArrayName)
{
  const auto& trackHasAlreadyRelations = [&partnerStoreArrayName](const RecoTrack * recoTrack) {
    return recoTrack->getRelated<RecoTrack>(partnerStoreArrayName) != nullptr;
  };
 
  TrackFindingCDC::erase_remove_if(tracks, trackHasAlreadyRelations);
}