GblTrajectory.h

Go to the documentation of this file.

/*
 * 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);
    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);
    void getLabels(std::vector<unsigned int> &aLabelList);
    void 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();

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_ */