RecoResonance.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2018 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributor: Wouter Hulsbergen, Francesco Tenchini, Jo-Frederik Krohn  *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
 
#include <analysis/VertexFitting/TreeFitter/RecoComposite.h>
 
namespace TreeFitter {
 
  class RecoResonance : public RecoComposite {
 
  public:
 
    RecoResonance(Belle2::Particle* bc, const ParticleBase* mother);
 
    virtual ErrCode initParticleWithMother(FitParams& fitparams) override;
 
    virtual ErrCode initMotherlessParticle(FitParams& fitparams) override;
 
    virtual ~RecoResonance() {};
 
    virtual int dim() const override { return hasEnergy() ? 4 : 3;} // (px,py,pz,(E))
 
    virtual ErrCode projectConstraint(Constraint::Type, const FitParams&, Projection&) const override;
 
    virtual int type() const override { return kRecoResonance; }
 
    virtual int posIndex() const override { return mother()->posIndex(); }
 
    virtual int momIndex() const override { return index(); }
 
    virtual int tauIndex() const override { return -1; }
 
    virtual std::string parname(int index) const override;
 
    virtual void addToConstraintList(constraintlist& list, int depth) const override
    {
      list.push_back(Constraint(this, Constraint::resonance, depth, dimM()));
    }
 
  };
 
}