MatcherInterface.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2016 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Nils Braun                                               *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
#pragma once
 
#include <tracking/trackFindingCDC/findlets/base/Findlet.h>
#include <tracking/trackFindingCDC/utilities/WeightedRelation.h>
 
#include <cmath>
 
namespace Belle2 {
  namespace TrackFindingCDC {
    template <class ACollectorItem, class ACollectionItem>
    class MatcherInterface :
      public Findlet<ACollectorItem,
      const ACollectionItem,
      WeightedRelation<ACollectorItem, const ACollectionItem>> {
    public:
      using WeightedRelationItem = WeightedRelation<ACollectorItem, const ACollectionItem>;
 
      void apply(std::vector<ACollectorItem>& collectorItems, const std::vector<ACollectionItem>& collectionItems,
                 std::vector<WeightedRelationItem>& weightedRelations) override
      {
        for (ACollectorItem& collectorItem : collectorItems) {
          match(collectorItem, collectionItems, weightedRelations);
        }
 
        std::sort(weightedRelations.begin(), weightedRelations.end());
      }
 
    protected:
      virtual void match(ACollectorItem& collectorItem, const std::vector<ACollectionItem>& collectionItems,
                         std::vector<WeightedRelationItem>& relationsForCollector)
      {
        for (const ACollectionItem& collectionItem : collectionItems) {
          Weight weight = match(collectorItem, collectionItem);
          relationsForCollector.emplace_back(&collectorItem, weight, &collectionItem);
        }
      };
 
      virtual Weight match(ACollectorItem& collectorItem __attribute__((unused)),
                           const ACollectionItem& collectionItem __attribute__((unused)))
      {
        return NAN;
      }
    };
  }
}