235 lines
7.4 KiB
C++
235 lines
7.4 KiB
C++
#ifndef ClassDetGeo_H
|
|
#define ClassDetGeo_H
|
|
|
|
#include <stdio.h> /// for FILE
|
|
#include <cstdlib>
|
|
#include <string>
|
|
#include <vector>
|
|
#include <unistd.h>
|
|
|
|
#include "TMath.h"
|
|
#include "TString.h"
|
|
#include "TMacro.h"
|
|
|
|
#include "AnalysisLib.h"
|
|
|
|
struct Array{
|
|
|
|
bool enable;
|
|
|
|
double detPerpDist; /// distance from axis
|
|
double detWidth; /// width
|
|
double detLength; /// length
|
|
double blocker;
|
|
double firstPos; /// meter
|
|
double eSigma; /// intrinsic energy resolution MeV
|
|
double zSigma; /// intrinsic position resolution mm
|
|
bool detFaceOut; ///detector_facing_Out_or_In
|
|
std::vector<double> pos; /// near position in meter
|
|
int nDet, mDet; /// nDet = number of different pos, mDet, number of same pos
|
|
std::vector<double> detPos; ///absolute position of detector
|
|
|
|
double zMin, zMax;
|
|
|
|
void DeduceAbsolutePos(){
|
|
|
|
nDet = pos.size();
|
|
detPos.clear();
|
|
|
|
for(int id = 0; id < nDet; id++){
|
|
if( firstPos > 0 ) detPos.push_back(firstPos + pos[id]);
|
|
if( firstPos < 0 ) detPos.push_back(firstPos - pos[nDet - 1 - id]);
|
|
// printf("%d | %f, %f \n", id, pos[id], detPos[id]);
|
|
}
|
|
|
|
zMin = TMath::Min(detPos.front(), detPos.back()) - (firstPos < 0 ? detLength : 0);
|
|
zMax = TMath::Max(detPos.front(), detPos.back()) + (firstPos > 0 ? detLength : 0);
|
|
}
|
|
|
|
void PrintArray() const{
|
|
for(int i = 0; i < nDet ; i++){
|
|
if( firstPos > 0 ){
|
|
printf("%d, %8.2f mm - %8.2f mm \n", i, detPos[i], detPos[i] + detLength);
|
|
}else{
|
|
printf("%d, %8.2f mm - %8.2f mm \n", i, detPos[i] - detLength , detPos[i]);
|
|
}
|
|
}
|
|
|
|
printf(" Blocker Position: %8.2f mm \n", firstPos > 0 ? firstPos - blocker : firstPos + blocker );
|
|
printf(" First Position: %8.2f mm \n", firstPos);
|
|
printf(" number of det : %d x %d \n", mDet, nDet);
|
|
printf(" detector facing : %s\n", detFaceOut ? "Out" : "In");
|
|
printf(" energy resol.: %f MeV\n", eSigma);
|
|
printf(" pos-Z resol.: %f mm \n", zSigma);
|
|
}
|
|
|
|
};
|
|
|
|
class DetGeo {
|
|
|
|
public:
|
|
DetGeo(){};
|
|
DetGeo(TString detGeoTxt){ LoadDetectorGeo(detGeoTxt, false);}
|
|
DetGeo(TMacro * macro){ LoadDetectorGeo(macro, false);}
|
|
~DetGeo(){};
|
|
|
|
double Bfield; /// T
|
|
int BfieldSign ; /// sign of B-field
|
|
double BfieldTheta; /// rad, 0 = z-axis, pi/2 = y axis, pi = -z axis
|
|
double bore; /// bore , mm
|
|
|
|
double recoilPos; /// recoil, downstream
|
|
double recoilInnerRadius; /// radius recoil inner
|
|
double recoilOuterRadius; /// radius recoil outter
|
|
|
|
double recoilPos1, recoilPos2; /// imaginary recoils
|
|
|
|
double elumPos1, elumPos2; /// imaginary elum, only sensitive to light recoil
|
|
|
|
//===================1st array
|
|
Array array[2];
|
|
|
|
double zMin, zMax; /// range of detectors
|
|
bool isCoincidentWithRecoil;
|
|
|
|
bool LoadDetectorGeo(TString fileName, bool verbose = true);
|
|
bool LoadDetectorGeo(TMacro * macro, bool verbose = true);
|
|
|
|
void Print( bool printAll = true) const;
|
|
|
|
private:
|
|
|
|
};
|
|
|
|
inline bool DetGeo::LoadDetectorGeo(TString fileName, bool verbose){
|
|
|
|
TMacro * haha = new TMacro();
|
|
if( haha->ReadFile(fileName) > 0 ) {
|
|
if( LoadDetectorGeo(haha, verbose) ){
|
|
delete haha;
|
|
return true;
|
|
}else{
|
|
delete haha;
|
|
return false;
|
|
}
|
|
}else{
|
|
delete haha;
|
|
return false;
|
|
}
|
|
}
|
|
|
|
///Using TMacro to load the detectorGeo frist,
|
|
///this indrect method is good for loading detectorGeo from TMacro in root file
|
|
inline bool DetGeo::LoadDetectorGeo(TMacro * macro, bool verbose){
|
|
|
|
if( macro == NULL ) return false;
|
|
|
|
TList * haha = macro->GetListOfLines();
|
|
int numLine = (haha)->GetSize();
|
|
|
|
for( int i = 0; i < 2 ; i++) array[i].pos.clear();
|
|
|
|
int detFlag = 0;
|
|
int detLine = 0;
|
|
|
|
for( int i = 0 ; i < numLine; i++){
|
|
|
|
std::string line = macro->GetListOfLines()->At(i)->GetName();
|
|
|
|
if( AnalysisLib::isEmptyOrSpaces(line) ) continue;
|
|
|
|
std::vector<std::string> str = AnalysisLib::SplitStr(line, " ");
|
|
|
|
// printf("%3d | %s\n", i, str[0].c_str());
|
|
|
|
if( str[0].find("####") != std::string::npos ) break;
|
|
if( str[0].find("#===") != std::string::npos ) {
|
|
detFlag ++;
|
|
detLine = 0;
|
|
continue;;
|
|
}
|
|
|
|
if( detFlag == 0 ){
|
|
if ( detLine == 0 ) {
|
|
Bfield = atof(str[0].c_str());
|
|
BfieldSign = Bfield > 0 ? 1: -1;
|
|
}
|
|
if ( detLine == 1 ) BfieldTheta = atof(str[0].c_str());
|
|
if ( detLine == 2 ) bore = atof(str[0].c_str());
|
|
if ( detLine == 3 ) recoilPos = atof(str[0].c_str());
|
|
if ( detLine == 4 ) recoilInnerRadius = atof(str[0].c_str());
|
|
if ( detLine == 5 ) recoilOuterRadius = atof(str[0].c_str());
|
|
if ( detLine == 6 ) isCoincidentWithRecoil = str[0] == "false" ? false: true;
|
|
if ( detLine == 7 ) recoilPos1 = atof(str[0].c_str());
|
|
if ( detLine == 8 ) recoilPos2 = atof(str[0].c_str());
|
|
if ( detLine == 9 ) elumPos1 = atof(str[0].c_str());
|
|
if ( detLine == 10 ) elumPos2 = atof(str[0].c_str());
|
|
}
|
|
|
|
if( detFlag > 0){
|
|
unsigned short ID = detFlag - 1;
|
|
if ( detLine == 0 ) array[ID].enable = str[0] == "true" ? true : false;
|
|
if ( detLine == 1 ) array[ID].detPerpDist = atof(str[0].c_str());
|
|
if ( detLine == 2 ) array[ID].detWidth = atof(str[0].c_str());
|
|
if ( detLine == 3 ) array[ID].detLength = atof(str[0].c_str());
|
|
if ( detLine == 4 ) array[ID].blocker = atof(str[0].c_str());
|
|
if ( detLine == 5 ) array[ID].firstPos = atof(str[0].c_str());
|
|
if ( detLine == 6 ) array[ID].eSigma = atof(str[0].c_str());
|
|
if ( detLine == 7 ) array[ID].zSigma = atof(str[0].c_str());
|
|
if ( detLine == 8 ) array[ID].detFaceOut = str[0] == "Out" ? true : false;
|
|
if ( detLine == 9 ) array[ID].mDet = atoi(str[0].c_str());
|
|
if ( detLine >= 10 ) array[ID].pos.push_back(atof(str[0].c_str()));
|
|
}
|
|
|
|
detLine ++;
|
|
}
|
|
|
|
zMin = 99999;
|
|
zMax = -99999;
|
|
|
|
for( int i = 0; i < 2; i ++ ){
|
|
array[i].DeduceAbsolutePos();
|
|
if (array[i].enable ) {
|
|
double zmax = TMath::Max(array[i].zMin, array[i].zMax);
|
|
double zmin = TMath::Min(array[i].zMin, array[i].zMax);
|
|
if( zmax > zMax ) zMax = zmax;
|
|
if( zmin < zMin ) zMin = zmin;
|
|
}
|
|
}
|
|
|
|
if( verbose ) Print(false);
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
inline void DetGeo::Print(bool printAll) const{
|
|
|
|
printf("=====================================================\n");
|
|
printf(" B-field: %8.2f T, Theta : %6.2f deg \n", Bfield, BfieldTheta);
|
|
if( BfieldTheta != 0.0 ) {
|
|
printf(" +---- field angle != 0 is not supported!!! \n");
|
|
}
|
|
printf(" Recoil detector pos: %8.2f mm, radius: %6.2f - %6.2f mm \n", recoilPos, recoilInnerRadius, recoilOuterRadius);
|
|
|
|
for( int i = 0; i < 2 ; i++){
|
|
|
|
if( printAll || array[i].enable ) {
|
|
printf("-----------------------------------%d-th Detector Position \n", i);
|
|
array[i].PrintArray();
|
|
}
|
|
|
|
}
|
|
|
|
if( elumPos1 != 0 || elumPos2 != 0 || recoilPos1 != 0 || recoilPos2 != 0){
|
|
printf("=================================== Auxillary/Imaginary Detectors\n");
|
|
}
|
|
if( elumPos1 != 0 ) printf(" Elum 1 pos.: %f mm \n", elumPos1);
|
|
if( elumPos2 != 0 ) printf(" Elum 2 pos.: %f mm \n", elumPos2);
|
|
if( recoilPos1 != 0 ) printf(" Recoil 1 pos.: %f mm \n", recoilPos1);
|
|
if( recoilPos2 != 0 ) printf(" Recoil 2 pos.: %f mm \n", recoilPos2);
|
|
printf("=====================================================\n");
|
|
|
|
}
|
|
|
|
#endif |