CgemGeoAlign.h

Go to the documentation of this file.

#ifndef CGEMGEOALIGN_H
#define CGEMGEOALIGN_H
 
#include "CgemGeomSvc/StraightLine.h"
#include "CLHEP/Geometry/Point3D.h"
 
#include "CLHEP/Vector/ThreeVector.h"
#include "CLHEP/Vector/LorentzVector.h"//
#include "CLHEP/Vector/TwoVector.h"
 
using CLHEP::Hep3Vector;
#include <string>
 
class ToyCgem
{
    public:
    ToyCgem(int layer_vir, double r, double x0, double y0, double z0, double rx, double ry, double rz);
    Hep3Vector getdir();
    double GetX0() {return X0;}
    double GetY0() {return Y0;}
    double GetZ0() {return Z0;}
    double GetRX() {return RX;}
    double GetRY() {return RY;}
    double GetRZ() {return RZ;}
    double GetR() {return R;}
 
    private:
    int Layer;
    double R;
    //double Length;
    double X0;
    double Y0;
    double Z0;
    double RX;
    double RY;
    double RZ;
 
};
 
class ToyRay
{
    public:
    ToyRay(double dr, double phi, double dz, double tanl);
    Hep3Vector getdir();
    double GetDrho() {return Drho;}
    double GetPhi() {return Phi;}
    double GetDz() {return Dz;}
    double GetTanL() {return TanL;}
 
    //bool getinter(int layer, Hep3Vector& posUp, Hep3Vector& posDown);
 
    private:
    double Drho;
    double Phi;
    double Dz;
    double TanL;
};
 
 
 
class CgemGeoAlign {
    public:
        CgemGeoAlign();
        ~CgemGeoAlign() {};
 
        void initAlignPar(std::string alignFile);
        void setLayerR(int layer_geo, double r) { m_r[layer_geo] = r; }
 
        double getDx(int layer_vir) { return m_dx[layer_vir]; }
        double getDy(int layer_vir) { return m_dy[layer_vir]; }
        double getDz(int layer_vir) { return m_dz[layer_vir]; }
        double getRx(int layer_vir) { return m_rx[layer_vir]; }
        double getRy(int layer_vir) { return m_ry[layer_vir]; }
        double getRz(int layer_vir) { return m_rz[layer_vir]; }
        
        double getDx(int layer_geo, int sheet_geo) { return m_dx[layer_geo*2+sheet_geo]; }
        double getDy(int layer_geo, int sheet_geo) { return m_dy[layer_geo*2+sheet_geo]; }
        double getDz(int layer_geo, int sheet_geo) { return m_dz[layer_geo*2+sheet_geo]; }
        double getRx(int layer_geo, int sheet_geo) { return m_rx[layer_geo*2+sheet_geo]; }
        double getRy(int layer_geo, int sheet_geo) { return m_ry[layer_geo*2+sheet_geo]; }
        double getRz(int layer_geo, int sheet_geo) { return m_rz[layer_geo*2+sheet_geo]; }
 
        /* modification of alignment parameters (used by alignment algorithm) */
        void setDx(int layer_vir, double v) { m_dx[layer_vir] = v; }
        void setDy(int layer_vir, double v) { m_dy[layer_vir] = v; }
        void setDz(int layer_vir, double v) { m_dz[layer_vir] = v; }
        void setRx(int layer_vir, double v) { m_rx[layer_vir] = v; }
        void setRy(int layer_vir, double v) { m_ry[layer_vir] = v; }
        void setRz(int layer_vir, double v) { m_rz[layer_vir] = v; }
 
        void setDx(int layer_geo, int sheet_geo, double v) { m_dx[layer_geo*2+sheet_geo] = v; }
        void setDy(int layer_geo, int sheet_geo, double v) { m_dy[layer_geo*2+sheet_geo] = v; }
        void setDz(int layer_geo, int sheet_geo, double v) { m_dz[layer_geo*2+sheet_geo] = v; }
        void setRx(int layer_geo, int sheet_geo, double v) { m_rx[layer_geo*2+sheet_geo] = v; }
        void setRy(int layer_geo, int sheet_geo, double v) { m_ry[layer_geo*2+sheet_geo] = v; }
        void setRz(int layer_geo, int sheet_geo, double v) { m_rz[layer_geo*2+sheet_geo] = v; }
        /* resest alignment parameters to original values (used by alignment) */
        void resetAlignPar();
 
        StraightLine StraightLineConversion(int layer_vir, StraightLine lineOriginal);
 
        /* written by Jingyi */
        void StraightLineConversion_v1(int layer_vir, double lineOriginal[], double lineConverted[]);
 
        void HelixConversion(int layer_vir, double helixOriginal[], double helixConverted[]);
 
        HepPoint3D point_transform(int layer_vir, HepPoint3D pos);
        HepPoint3D point_transform(int layer_geo, int sheet_geo,  HepPoint3D pos);
 
        void point_transform(double& x, double& y, double& z, double shift_x,
                double shift_y, double shift_z, double rotation_z);
 
        // HepPoint3D point_invTransform(int layer_vir, HepPoint3D pos);
        // HepPoint3D point_invTransform_New(int layer_vir, HepPoint3D pos);
        // HepPoint3D point_invTransform_New(int layer_geo, int sheet_geo, HepPoint3D pos);
        HepPoint3D point_invTransform(int layer_vir, HepPoint3D pos);
        HepPoint3D point_invTransform(int layer_geo, int sheet_geo, HepPoint3D pos);
        
        void GetLocPos(int layer_vir, Hep3Vector GposUp, Hep3Vector GposDown, HepPoint3D& LposUp, HepPoint3D& LposDown);
        void GetLocPos(int layer_vir, Hep3Vector Gpos,  HepPoint3D& Lpos);
        void GetLocPos(int layer_geo, int sheet_geo, Hep3Vector GposUp, Hep3Vector GposDown, HepPoint3D& LposUp, HepPoint3D& LposDown);
        void GetLocPos(int layer_geo, int sheet_geo, Hep3Vector Gpos,  HepPoint3D& Lpos);
 
        bool getinter(ToyRay ARay, int layer_vir, Hep3Vector& posUp, Hep3Vector& posDown);
        bool getinter(ToyRay ARay, int layer_geo, int sheet_geo, Hep3Vector& posUp, Hep3Vector& posDown);
        bool getinter(ToyRay ARay, int layer_vir, Hep3Vector& pos);
        bool getinter(ToyRay ARay, int layer_geo, int sheet_geo, Hep3Vector& pos);
 
    private:
        double m_r[3];
 
        double m_dx[6];
        double m_dy[6];
        double m_dz[6];
        double m_rx[6];
        double m_ry[6];
        double m_rz[6];
 
        double m_dxOrig[6];
        double m_dyOrig[6];
        double m_dzOrig[6];
        double m_rxOrig[6];
        double m_ryOrig[6];
        double m_rzOrig[6];
 
};
 
#endif