MVAFacetFilter.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2015 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Nils Braun, Oliver Frost                                 *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
#pragma once
 
#include <tracking/trackFindingCDC/filters/facet/BaseFacetFilter.h>
 
#include <tracking/trackFindingCDC/filters/facet/BasicFacetVarSet.h>
#include <tracking/trackFindingCDC/filters/facet/Chi2FacetFilter.h>
 
#include <tracking/trackFindingCDC/filters/base/MVAFilter.dcl.h>
 
#include <tracking/trackFindingCDC/filters/base/FilterVarSet.dcl.h>
 
#include <tracking/trackFindingCDC/varsets/VariadicUnionVarSet.h>
 
namespace Belle2 {
  namespace TrackFindingCDC {
    class CDCFacet;
 
    using MVAFacetVarSet = VariadicUnionVarSet<BasicFacetVarSet,
          FilterVarSet<Chi2FacetFilter>>;
 
    class MVAFacetFilter: public MVA<BaseFacetFilter> {
 
    public:
      using Super = MVA<BaseFacetFilter>;
 
    public:
      MVAFacetFilter();
 
      Weight predict(const CDCFacet& facet) final;
    };
  }
}