TrackConstraint.h

/**************************************************************************
 * basf2 (Belle II Analysis Software Framework)                           *
 * Author: The Belle II Collaboration                                     *
 *                                                                        *
 * 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/                                   *
 *                                                                        *
 * See git log for contributors and copyright holders.                    *
 * This file is licensed under LGPL-3.0, see LICENSE.md.                  *
 **************************************************************************/
 
#ifndef __TrackConstraint_H
#define __TrackConstraint_H
 
#include<vector>
#include<cassert>
#include "analysis/OrcaKinFit/BaseConstraint.h"
 
namespace Belle2 {
  namespace OrcaKinFit {
 
    class TrackFitObject;
 
//  Class TrackConstraint:
 
    class TrackConstraint: public BaseConstraint {
    public:
      inline TrackConstraint();
      virtual ~TrackConstraint() {};
 
      virtual void setFOList(std::vector <TrackFitObject*>* fitobjects_ 
                            )
      {
        for (int i = 0; i < (int) fitobjects_->size(); i++) {
          fitobjects.push_back((*fitobjects_)[i]);
          flags.push_back(1);
          sign.push_back(i == 0 ? 1 : -1);
        }
      };
      virtual void addToFOList(TrackFitObject& fitobject, int flag = 1
                              )
      {
        fitobjects.push_back(&fitobject);
        flags.push_back(flag);
        sign.push_back(sign.size() == 0 ? 1 : -1);
      };
      virtual double getValue() const = 0;
 
      virtual void getDerivatives(int idim, double der[]) const = 0;
      virtual void add1stDerivativesToMatrix(double* M,      
                                             int idim        
                                            ) const
      {assert(false);}
      virtual void add2ndDerivativesToMatrix(double* M,       
                                             int idim,       
                                             double lambda   
                                            ) const
      {assert(false);}
 
      virtual void addToGlobalChi2DerVector(double* y,    
                                            int idim,    
                                            double lambda //< The lambda value
                                           ) const
      {assert(false);};
 
      virtual int  getGlobalNum() const {return globalNum;}
      virtual void setGlobalNum(int iglobal) {globalNum = iglobal;}
 
      virtual void invalidateCache() const {};
 
    protected:
      typedef std::vector <TrackFitObject*> FitObjectContainer;
      typedef FitObjectContainer::iterator FitObjectIterator;
      typedef FitObjectContainer::const_iterator ConstFitObjectIterator;
      FitObjectContainer fitobjects;
      std::vector <double> derivatives;
      std::vector <int> flags;
      std::vector <double> sign;
 
      int globalNum;
 
    };
 
    TrackConstraint::TrackConstraint()
    {
      invalidateCache();
    }
 
  }// end OrcaKinFit namespace
} // end Belle2 namespace
 
#endif // __TrackConstraint_H