FairRKPropagator.h

Go to the documentation of this file.

/********************************************************************************
 *    Copyright (C) 2014 GSI Helmholtzzentrum fuer Schwerionenforschung GmbH    *
 *                                                                              *
 *              This software is distributed under the terms of the             *
 *              GNU Lesser General Public Licence (LGPL) version 3,             *
 *                  copied verbatim in the file "LICENSE"                       *
 ********************************************************************************/
#ifndef RKPropagator
#define RKPropagator
#include "FairPropagator.h"   // for FairPropagator, PCAOutputStruct

#include <Rtypes.h>     // for THashConsistencyHolder, ClassDef
#include <TVector3.h>   // for TVector3
#include <iosfwd>       // for string
class FairField;        // lines 20-20
class FairTrackParH;
class FairTrackParP;   // lines 21-21
class TBuffer;
class TClass;
class TMemberInspector;

enum PropagationFlag
{
    NONE,
    TOPLANE,
    TOVOLUME,
    TOLENGTH
};

class FairRKPropagator : public FairPropagator
{
  private:
    FairRKPropagator(const FairRKPropagator&);              // Not implemented
    FairRKPropagator& operator=(const FairRKPropagator&);   // Not implemented
    double fMaxStep;
    FairField* fMagField;

    PropagationFlag fPropagationFlag;
    TVector3 fDefPlaneV0;
    TVector3 fDefPlaneV1;
    TVector3 fDefPlaneV2;

    int fPCAPropagationType;
    int fPCAPropagationDir;
    FairTrackParP* fPCAPropagationPar;

    double GetChargeFromPDG(int pdg);
    double CalculatePointToWireDistance(TVector3 point, TVector3 wire1, TVector3 wire2, TVector3& vwi);

  public:
    double Step(double Charge, double* vecRKIn, double* vecOut);
    double OneStepRungeKutta(double charge, double step, double* vect, double* vout);
    FairRKPropagator(FairField* field);
    void Propagate(double Charge, double* vecRKIn, double* Pos);

    void PropagateToPlane(double Charge, double* vecRKIn, double* vec1, double* vec2, double* vec3, double* vecOut);

    /* ===========================================================================================================================================
     */

    virtual bool SetDestinationPlane(const TVector3& v0, const TVector3& v1, const TVector3& v2);

    virtual bool SetOriginPlane(const TVector3& v0, const TVector3& v1);

    virtual bool SetDestinationVolume(std::string volName, int copyNo, int option);

    virtual bool SetDestinationLength(float length);

    virtual bool SetPropagateOnlyParameters() { return true; }

    virtual bool Propagate(FairTrackParH* TStart, FairTrackParH* TEnd, int PDG);
    virtual bool Propagate(FairTrackParP* TStart, FairTrackParH* TEnd, int PDG);
    virtual bool Propagate(FairTrackParP* TStart, FairTrackParP* TEnd, int PDG);
    virtual bool Propagate(FairTrackParH* TStart, FairTrackParP* TEnd, int PDG);
    virtual bool Propagate(float* x1, float* p1, float* x2, float* p2, int PDG);

    virtual bool SetPCAPropagation(int pca, int dir = 1, FairTrackParP* par = nullptr);

    virtual PCAOutputStruct FindPCA(int PCA,
                                    int PDGCode,
                                    TVector3 Point,
                                    TVector3 Wire1,
                                    TVector3 Wire2,
                                    double MaxDistance);

    virtual ~FairRKPropagator();
    ClassDef(FairRKPropagator, 2);
};

#endif