MomentumConstraint.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * See https://github.com/tferber/OrcaKinfit, forked from                 *
 * https://github.com/iLCSoft/MarlinKinfit                                *
 *                                                                        *
 * Further information about the fit engine and the user interface        *
 * provided in MarlinKinfit can be found at                               *
 * https://www.desy.de/~blist/kinfit/doc/html/                            *
 * and in the LCNotes LC-TOOL-2009-001 and LC-TOOL-2009-004 available     *
 * from http://www-flc.desy.de/lcnotes/                                   *
 *                                                                        *
 * Adopted by: Torben Ferber (torben.ferber@desy.de) (TF)                 *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#ifndef __MOMENTUMCONSTRAINT_H
#define __MOMENTUMCONSTRAINT_H
 
#include "analysis/OrcaKinFit/ParticleConstraint.h"
 
namespace Belle2 {
  namespace OrcaKinFit {
 
    class ParticleFitObject;
 
//  Class PConstraint:
 
    class MomentumConstraint : public ParticleConstraint {
    public:
      MomentumConstraint(double efact_ = 0,     
                         double pxfact_ = 0,   
                         double pyfact_ = 0,   
                         double pzfact_ = 0,   
                         double value_ = 0     
                        );
      virtual ~MomentumConstraint();
      virtual double getValue() const override;
      virtual void getDerivatives(int idim,      
                                  double der[]   
                                 ) const override;
 
      // not used at all
      //    virtual void addToGlobalDerMatrix (double lambda, int idim, double *M) const;
 
      virtual void invalidateCache() const override;
 
      virtual int getVarBasis() const override;
 
    protected:
      void updateCache() const;
 
 
      double efact;
      double pxfact;
      double pyfact;
      double pzfact;
      double value;
 
      mutable bool cachevalid;
      mutable int  nparams;
 
      virtual bool secondDerivatives(int i,                         
                                     int j,                        
                                     double* derivatives           
                                    ) const override;
      virtual bool firstDerivatives(int i,                         
                                    double* derivatives           
                                   ) const override;
 
      enum { VAR_BASIS = 0 }; // this means that the constraint knows about E,px,py,pz
 
    };
 
  }// end OrcaKinFit namespace
} // end Belle2 namespace
 
#endif // __MOMENTUMCONSTRAINT_H