BestMatchedTruthVarSet.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2019 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Michael Eliachevitch, Nils Braun                         *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
#pragma once
 
#include <tracking/trackFindingCDC/varsets/VarSet.h>
#include <tracking/trackFindingCDC/varsets/VarNames.h>
 
#include <framework/dataobjects/EventMetaData.h>
#include <framework/datastore/StoreObjPtr.h>
 
namespace Belle2 {
  namespace TrackFindingCDC {
    class CDCTrack;
 
    constexpr static char const* const bestMatchedTruthVarNames[] = {
      // Truth: PR Track is not a fake and the best matched track (not a clone)
      "truth",
      // Other truth variables are mostly to be helpful during debugging.
      "truth_track_is_fake",
      "truth_track_is_matched",
      "truth_matched_hits",
      "truth_track_is_clone",
      "truth_first_nloops",
      "truth_event_id",
      "truth_MCTrackID",
    };
 
    struct BestMatchedTruthVarNames : public VarNames<CDCTrack> {
 
      static const size_t nVars = size(bestMatchedTruthVarNames);
 
      static constexpr char const* getName(int iName)
      {
        return bestMatchedTruthVarNames[iName];
      }
    };
 
    class BestMatchedTruthVarSet : public VarSet<BestMatchedTruthVarNames> {
 
    public:
      void initialize() final;
 
      void beginEvent() final;
 
      bool extract(const CDCTrack* ptrCDCTrack) override;
 
    private:
      using Super = VarSet<BestMatchedTruthVarNames>;
 
      StoreObjPtr<EventMetaData> m_eventMetaData;
    };
  }
}