mirror of
https://github.com/gwm17/Mask.git
synced 2025-03-14 23:48:50 -04:00
83 lines
2.4 KiB
C
83 lines
2.4 KiB
C
|
#ifndef STRIP_DETECTOR_H
|
||
|
|
#define STRIP_DETECTOR_H
|
||
|
|
|
||
|
|
// +z is along beam axis
|
||
|
|
// +y is vertically "downward" in the lab frame
|
||
|
|
|
||
|
|
//angles must be in radians, but distances can be whatever
|
||
|
|
//PROVIDED all input distances are the same
|
||
|
|
|
||
|
|
//Front strips from largest y to smallest y
|
||
|
|
|
||
|
|
//Back strips from lowest z to highest z
|
||
|
|
|
||
|
|
#include <cmath>
|
||
|
|
#include <vector>
|
||
|
|
|
||
|
|
#include "Math/Point3D.h"
|
||
|
|
#include "Math/Vector3D.h"
|
||
|
|
#include "Math/RotationZ.h"
|
||
|
|
#include "Mask/RandomGenerator.h"
|
||
|
|
|
||
|
|
struct StripHit
|
||
|
|
{
|
||
|
|
int front_strip_index=-1;
|
||
|
|
int back_strip_index=-1;
|
||
|
|
double front_ratio=0.0;
|
||
|
|
};
|
||
|
|
|
||
|
|
class StripDetector
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
|
||
|
|
StripDetector(int ns, double len, double wid, double cphi, double cz, double crho);
|
||
|
|
~StripDetector();
|
||
|
|
const ROOT::Math::XYZPoint& GetFrontStripCoordinates(int stripch, int corner) const { return m_frontStripCoords[stripch][corner]; }
|
||
|
|
const ROOT::Math::XYZPoint& GetBackStripCoordinates(int stripch, int corner) const { return m_backStripCoords[stripch][corner]; }
|
||
|
|
const ROOT::Math::XYZPoint& GetRotatedFrontStripCoordinates(int stripch, int corner) const
|
||
|
|
{
|
||
|
|
return m_rotFrontStripCoords[stripch][corner];
|
||
|
|
}
|
||
|
|
const ROOT::Math::XYZPoint& GetRotatedBackStripCoordinates(int stripch, int corner) const
|
||
|
|
{
|
||
|
|
return m_rotBackStripCoords[stripch][corner];
|
||
|
|
}
|
||
|
|
ROOT::Math::XYZVector GetNormRotated() const { return m_zRotation * m_norm; }
|
||
|
|
|
||
|
|
void SetPixelSmearing(bool isSmearing) { m_isSmearing = isSmearing; }
|
||
|
|
|
||
|
|
ROOT::Math::XYZPoint GetHitCoordinates(int front_stripch, double front_strip_ratio);
|
||
|
|
StripHit GetChannelRatio(double theta, double phi);
|
||
|
|
|
||
|
|
private:
|
||
|
|
bool ValidChannel(int f) { return ((f >= 0 && f < m_nStrips) ? true : false); };
|
||
|
|
bool ValidRatio(double r) { return ((r >= -1 && r <= 1) ? true : false); };
|
||
|
|
void CalculateCorners();
|
||
|
|
|
||
|
|
int m_nStrips;
|
||
|
|
static constexpr int s_nCorners = 4;
|
||
|
|
|
||
|
|
double m_stripLength; //common to all strips, hence total
|
||
|
|
double m_totalWidth;
|
||
|
|
double m_frontStripWidth; //assuming equal widths
|
||
|
|
double m_backStripLength; //assuming equal widths
|
||
|
|
|
||
|
|
double m_centerPhi; //assuming det centered above x-axis (corresponds to zero phi)
|
||
|
|
double m_centerZ;
|
||
|
|
double m_centerRho; //perpendicular radius from axis
|
||
|
|
|
||
|
|
std::vector<std::vector<ROOT::Math::XYZPoint>> m_frontStripCoords, m_backStripCoords;
|
||
|
|
std::vector<std::vector<ROOT::Math::XYZPoint>> m_rotFrontStripCoords, m_rotBackStripCoords;
|
||
|
|
|
||
|
|
ROOT::Math::XYZVector m_norm;
|
||
|
|
|
||
|
|
ROOT::Math::RotationZ m_zRotation;
|
||
|
|
|
||
|
|
std::uniform_real_distribution<double> m_uniformFraction;
|
||
|
|
|
||
|
|
bool m_isSmearing;
|
||
|
|
|
||
|
|
};
|
||
|
|
|
||
|
|
#endif
|