ParticleFitObject.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 __PARTICLEFITOBJECT_H
#define __PARTICLEFITOBJECT_H
 
#include "analysis/OrcaKinFit/BaseFitObject.h"
#include "analysis/OrcaKinFit/FourVector.h"
 
 
// Class ParticleFitObject
 
namespace Belle2 {
  namespace OrcaKinFit {
 
 
    class ParticleFitObject: public BaseFitObject {
    public:
      ParticleFitObject();
 
 
      ParticleFitObject(const ParticleFitObject& rhs               
                       );
 
      ParticleFitObject& operator= (const ParticleFitObject& rhs   
                                   );
 
      virtual ~ParticleFitObject();
 
      virtual ParticleFitObject& assign(const BaseFitObject& source    
                                       ) override;
 
      virtual bool setMass(double mass_);
      virtual double getMass() const;
 
      virtual std::ostream& print4Vector(std::ostream& os    
                                        ) const;
 
      virtual FourVector getFourMomentum() const;
 
      virtual double getE() const;
      virtual double getPx() const;
      virtual double getPy() const;
      virtual double getPz() const;
 
      virtual double getP() const;
      virtual double getP2() const;
      virtual double getPt() const;
      virtual double getPt2() const;
 
      virtual double getDPx(int ilocal               
                           ) const = 0;
      virtual double getDPy(int ilocal               
                           ) const = 0;
      virtual double getDPz(int ilocal               
                           ) const = 0;
      virtual double getDE(int ilocal                
                          ) const = 0;
 
      virtual void getDerivatives(double der[], int idim) const override;
 
      virtual void addToGlobalChi2DerMatrixNum(double* M,    
                                               int idim,    
                                               double eps   
                                              );
 
      virtual void addToGlobalChi2DerVectorNum(double* y,     
                                               int idim,     
                                               double eps    
                                              );
 
      virtual std::ostream& print(std::ostream& os     
                                 ) const override;
 
      void test1stDerivatives();
      void test2ndDerivatives();
 
      double num1stDerivative(int ilocal,   
                              double eps   
                             );
 
      double num2ndDerivative(int ilocal1,  
                              double eps1, 
                              int ilocal2, 
                              double eps2  
                             );
 
      virtual double getChi2() const override;
 
    protected:
      double mass;
 
      mutable FourVector fourMomentum;
 
      // this is to flag phi angle, for example
      double paramCycl[BaseDefs::MAXPAR];
 
    };
 
  }// end OrcaKinFit namespace
} // end Belle2 namespace
 
#endif // __PARTICLEFITOBJECT_H