RelationVarSet.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2015 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Oliver Frost                                             *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
#pragma once
 
#include <tracking/trackFindingCDC/varsets/BaseVarSet.h>
 
#include <tracking/trackFindingCDC/utilities/Relation.h>
#include <tracking/trackFindingCDC/utilities/MayBePtr.h>
 
#include <vector>
#include <string>
#include <cassert>
 
namespace Belle2 {
  namespace TrackFindingCDC {
    template <class ABaseVarSet>
    class RelationVarSet : public BaseVarSet<Relation<const typename ABaseVarSet::Object> > {
 
      using Super = BaseVarSet<Relation<const typename ABaseVarSet::Object> >;
 
    public:
      using BaseObject = typename ABaseVarSet::Object;
 
    public:
      void initialize() override
      {
        this->addProcessingSignalListener(&m_firstVarSet);
        this->addProcessingSignalListener(&m_secondVarSet);
        Super::initialize();
      }
 
      bool extract(const Relation<const BaseObject>* obj) override
      {
        assert(obj);
        bool firstExtracted = m_firstVarSet.extract(obj->first);
        bool secondExtracted = m_secondVarSet.extract(obj->second);
        return firstExtracted and secondExtracted;
      }
 
      bool extract(const Relation<const BaseObject>& obj)
      {
        return extract(&obj);
      }
 
      std::vector<Named<Float_t*>> getNamedVariables(const std::string& prefix) override
      {
        std::vector<Named<Float_t*> > result = m_firstVarSet.getNamedVariables(prefix + m_firstPrefix);
        std::vector<Named<Float_t*> > extend = m_secondVarSet.getNamedVariables(prefix + m_secondPrefix);
        result.insert(result.end(), extend.begin(), extend.end());
        return result;
      }
 
      MayBePtr<Float_t> find(const std::string& varName) override
      {
        // it is hard to do this with string::compare as cppcheck recommends
        // cppcheck-suppress stlIfStrFind
        if (0 == varName.find(m_firstPrefix)) {
          std::string varNameWithoutPrefix = varName.substr(m_firstPrefix.size());
          MayBePtr<Float_t> found = m_firstVarSet.find(varNameWithoutPrefix);
          if (found) return found;
        }
 
        // it is hard to do this with string::compare as cppcheck recommends
        // cppcheck-suppress stlIfStrFind
        if (0 == varName.find(m_secondPrefix)) {
          std::string varNameWithoutPrefix = varName.substr(m_secondPrefix.size());
          MayBePtr<Float_t> found = m_secondVarSet.find(varNameWithoutPrefix);
          if (found) return found;
        }
 
        return nullptr;
      }
 
    private:
      std::string m_firstPrefix = "first_";
 
      ABaseVarSet m_firstVarSet;
 
      std::string m_secondPrefix = "second_";
 
      ABaseVarSet m_secondVarSet;
 
    };
  }
}