RecoResonance.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 * External Contributor: Wouter Hulsbergen                                *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
 
#pragma once
 
#include <analysis/VertexFitting/TreeFitter/Composite.h>
 
namespace TreeFitter {

  class RecoResonance : public Composite {
 
  public:

    RecoResonance(Belle2::Particle* bc, const ParticleBase* mother, const ConstraintConfiguration& config);

    virtual ErrCode initParticleWithMother(FitParams& fitparams) override;

    virtual ErrCode initMotherlessParticle(FitParams& fitparams) override;

    virtual ~RecoResonance() {};

    virtual int dim() const override
    {
      if (particle()->hasExtraInfo("treeFitterTreatMeAsInvisible")
          && particle()->getExtraInfo("treeFitterTreatMeAsInvisible") == 1) {
        // invisible particle shouldn't contribute degrees of freedom to fit
        // set dimension to "measurement" dimension to cancel them out
        return dimM() ;
      }
      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;
 
  private:
    bool m_massconstraint;
 
 
  };
 
}