GblTrajectory.h

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file GblTrajectory.h

/*
 * GblTrajectory.h
 *
 *  Created on: Aug 18, 2011
 *      Author: kleinwrt
 */

#ifndef GBLTRAJECTORY_H_
#define GBLTRAJECTORY_H_

#include "GblPoint.h"
#include "GblData.h"
#include "GblPoint.h"
#include "BorderedBandMatrix.h"
#include "MilleBinary.h"
#include "TMatrixDSymEigen.h"

namespace gbl {


class GblTrajectory {
public:
  GblTrajectory(const std::vector<GblPoint> &aPointList, bool flagCurv = true,
      bool flagU1dir = true, bool flagU2dir = true);
  GblTrajectory(const std::vector<GblPoint> &aPointList, unsigned int aLabel,
      const TMatrixDSym &aSeed, bool flagCurv = true, bool flagU1dir =
          true, bool flagU2dir = true);
  GblTrajectory(
      const std::vector<std::pair<std::vector<GblPoint>, TMatrixD> > &aPointaAndTransList);
  GblTrajectory(
      const std::vector<std::pair<std::vector<GblPoint>, TMatrixD> > &aPointaAndTransList,
      const TMatrixD &extDerivatives, const TVectorD &extMeasurements,
      const TVectorD &extPrecisions);
  GblTrajectory(
      const std::vector<std::pair<std::vector<GblPoint>, TMatrixD> > &aPointaAndTransList,
      const TMatrixD &extDerivatives, const TVectorD &extMeasurements,
      const TMatrixDSym &extPrecisions);
  virtual ~GblTrajectory();
  bool isValid() const;
  unsigned int getNumPoints() const;
  unsigned int getResults(int aSignedLabel, TVectorD &localPar,
      TMatrixDSym &localCov) const;
  unsigned int getMeasResults(unsigned int aLabel, unsigned int &numRes,
      TVectorD &aResiduals, TVectorD &aMeasErrors, TVectorD &aResErrors,
      TVectorD &aDownWeights);
  unsigned int getScatResults(unsigned int aLabel, unsigned int &numRes,
      TVectorD &aResiduals, TVectorD &aMeasErrors, TVectorD &aResErrors,
      TVectorD &aDownWeights);
  unsigned int getLabels(std::vector<unsigned int> &aLabelList);
  unsigned int getLabels(std::vector<std::vector<unsigned int> > &aLabelList);
  unsigned int fit(double &Chi2, int &Ndf, double &lostWeight,
      std::string optionList = "");
  void milleOut(MilleBinary &aMille);
  void printTrajectory(unsigned int level = 0);
  void printPoints(unsigned int level = 0);
  void printData();
        std::vector<GblData> getData() {return theData;}

private:
  unsigned int numAllPoints; 
  std::vector<unsigned int> numPoints; 
  unsigned int numTrajectories; 
  unsigned int numOffsets; 
  unsigned int numInnerTrans; 
  unsigned int numCurvature; 
  unsigned int numParameters; 
  unsigned int numLocals; 
  unsigned int numMeasurements; 
  unsigned int externalPoint; 
  bool constructOK; 
  bool fitOK; 
  std::vector<unsigned int> theDimension; 
  std::vector<std::vector<GblPoint> > thePoints; 
  std::vector<GblData> theData; 
  std::vector<unsigned int> measDataIndex; 
  std::vector<unsigned int> scatDataIndex; 
  TMatrixDSym externalSeed; 
  std::vector<TMatrixD> innerTransformations; 
  // composed trajectory (from common external parameters)
  TMatrixD externalDerivatives; // Derivatives for external measurements of composed trajectory
  TVectorD externalMeasurements; // Residuals for external measurements of composed trajectory
  TVectorD externalPrecisions; // Precisions for external measurements of composed trajectory
  VVector theVector; 
  BorderedBandMatrix theMatrix; 

  std::pair<std::vector<unsigned int>, TMatrixD> getJacobian(
      int aSignedLabel) const;
  void getFitToLocalJacobian(std::vector<unsigned int> &anIndex,
      SMatrix55 &aJacobian, const GblPoint &aPoint, unsigned int measDim,
      unsigned int nJacobian = 1) const;
  void getFitToKinkJacobian(std::vector<unsigned int> &anIndex,
      SMatrix27 &aJacobian, const GblPoint &aPoint) const;
  void construct();
  void defineOffsets();
  void calcJacobians();
  void prepare();
  void buildLinearEquationSystem();
  void predict();
  double downWeight(unsigned int aMethod);
  void getResAndErr(unsigned int aData, double &aResidual,
      double &aMeadsError, double &aResError, double &aDownWeight);
};
}
#endif /* GBLTRAJECTORY_H_ */