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:
        explicit 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);
        explicit 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_ */