mirror of
https://github.com/slsdetectorgroup/slsDetectorPackage.git
synced 2025-04-19 18:40:01 +02:00
274 lines
6.8 KiB
C++
274 lines
6.8 KiB
C++
// SPDX-License-Identifier: LGPL-3.0-or-other
|
|
// Copyright (C) 2021 Contributors to the SLS Detector Package
|
|
#ifndef ETA3_INTERPOLATION_BASE_H
|
|
#define ETA3_INTERPOLATION_BASE_H
|
|
|
|
#ifdef MYROOT1
|
|
#include <TObject.h>
|
|
#include <TTree.h>
|
|
#include <TH2D.h>
|
|
#include <TH2F.h>
|
|
#endif
|
|
|
|
#include "etaInterpolationBase.h"
|
|
|
|
class eta3InterpolationBase : public virtual etaInterpolationBase {
|
|
|
|
public:
|
|
eta3InterpolationBase(int nx=400, int ny=400, int ns=25, int nsy=25, int nb=-1, int nby=-1, double emin=1, double emax=0) : etaInterpolationBase(nx, ny, ns, nsy, nb, nby, emin, emax) {
|
|
// cout << "e3ib " << nb << " " << emin << " " << emax << endl;
|
|
/* if (nbeta<=0) { */
|
|
/* nbeta=nSubPixels*10; */
|
|
/* } */
|
|
|
|
// cout << nbeta << " " << etamin << " " << etamax << endl;
|
|
};
|
|
|
|
eta3InterpolationBase(eta3InterpolationBase *orig): etaInterpolationBase(orig){ };
|
|
|
|
/* virtual eta3InterpolationBase* Clone()=0; */
|
|
|
|
|
|
|
|
|
|
// virtual void prepareInterpolation(int &ok){};
|
|
|
|
|
|
//////////////////////////////////////////////////////////////////////////////
|
|
//////////// /*It return position hit for the event in input */ //////////////
|
|
virtual void getInterpolatedPosition(int x, int y, int *data, double &int_x, double &int_y)
|
|
{
|
|
double tot;
|
|
double etax,etay;
|
|
|
|
int corner=calcEta3(data,etax,etay, tot);
|
|
|
|
getInterpolatedPosition(x,y,etax,etay,corner,int_x,int_y);
|
|
|
|
return;
|
|
};
|
|
|
|
|
|
virtual void getInterpolatedPosition(int x, int y, double *data, double &int_x, double &int_y)
|
|
{
|
|
//double sDum[2][2];
|
|
double tot;
|
|
double etax,etay;
|
|
|
|
int corner=calcEta3(data,etax,etay, tot);
|
|
|
|
getInterpolatedPosition(x,y,etax,etay,corner,int_x,int_y);
|
|
|
|
return;
|
|
};
|
|
|
|
|
|
virtual void getInterpolatedPosition(int x, int y, double totquad,int quad,double *cl,double &int_x, double &int_y) {
|
|
|
|
|
|
double etax, etay;
|
|
if (nSubPixelsX>2 || nSubPixelsY>2 ) {
|
|
calcEta3(cl,etax,etay, totquad);
|
|
}
|
|
return getInterpolatedPosition(x,y,etax, etay,quad,int_x,int_y);
|
|
|
|
}
|
|
|
|
|
|
|
|
virtual void getInterpolatedPosition(int x, int y, double totquad,int quad,int *cl,double &int_x, double &int_y) {
|
|
|
|
|
|
double etax, etay;
|
|
if (nSubPixelsX>2 || nSubPixelsY>2 ) {
|
|
calcEta3(cl,etax,etay, totquad);
|
|
}
|
|
return getInterpolatedPosition(x,y,etax, etay,quad,int_x,int_y);
|
|
|
|
}
|
|
|
|
|
|
|
|
virtual void getInterpolatedPosition(int x, int y, double etax, double etay, int corner, double &int_x, double &int_y)
|
|
{
|
|
|
|
|
|
double xpos_eta=0,ypos_eta=0;
|
|
int ex,ey;
|
|
|
|
if (nSubPixelsX>2 || nSubPixelsY>2 ) {
|
|
|
|
#ifdef MYROOT1
|
|
xpos_eta=(hhx->GetBinContent(hhx->GetXaxis()->FindBin(etax),hhy->GetYaxis()->FindBin(etay)))/((double)nSubPixelsX);
|
|
ypos_eta=(hhy->GetBinContent(hhx->GetXaxis()->FindBin(etax),hhy->GetYaxis()->FindBin(etay)))/((double)nSubPixelsY);
|
|
#endif
|
|
#ifndef MYROOT1
|
|
ex=(etax-etamin)/etastepX;
|
|
ey=(etay-etamin)/etastepY;
|
|
if (ex<0) {
|
|
/* cout << etax << " " << etamin << " "; */
|
|
/* cout << "3x*"<< ex << endl; */
|
|
ex=0;
|
|
}
|
|
if (ex>=nbetaX) {
|
|
/* cout << etax << " " << etamin << " "; */
|
|
/* cout << "3x?"<< ex << endl; */
|
|
ex=nbetaX-1;
|
|
}
|
|
if (ey<0) {
|
|
/* cout << etay << " " << etamin << " "; */
|
|
/* cout << "3y*"<< ey << endl; */
|
|
ey=0;
|
|
}
|
|
if (ey>=nbetaY) {
|
|
/* cout << etay << " " << etamin << " "; */
|
|
/* cout << "3y?"<< ey << endl; */
|
|
ey=nbetaY-1;
|
|
|
|
}
|
|
xpos_eta=(((double)hhx[(ey*nbetaX+ex)]));///((double)nSubPixels);
|
|
ypos_eta=(((double)hhy[(ey*nbetaX+ex)]));///((double)nSubPixels);
|
|
|
|
#endif
|
|
|
|
} else {
|
|
switch (corner) {
|
|
case BOTTOM_LEFT:
|
|
xpos_eta=-0.25;
|
|
ypos_eta=-0.25;
|
|
break;
|
|
case BOTTOM_RIGHT:
|
|
xpos_eta=0.25;
|
|
ypos_eta=-0.25;
|
|
break;
|
|
case TOP_LEFT:
|
|
xpos_eta=-0.25;
|
|
ypos_eta=0.25;
|
|
break;
|
|
case TOP_RIGHT:
|
|
xpos_eta=0.25;
|
|
ypos_eta=0.25;
|
|
break;
|
|
default:
|
|
xpos_eta=0;
|
|
ypos_eta=0;
|
|
}
|
|
|
|
}
|
|
int_x=((double)x) + xpos_eta;
|
|
int_y=((double)y) + ypos_eta;
|
|
// int_x=5. + xpos_eta;
|
|
// int_y=5. + ypos_eta;
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/* ///////////////////////////////////////////////////////////////////////////////////////////////// */
|
|
/* virtual void getPositionETA3(int x, int y, double *data, double &int_x, double &int_y) */
|
|
/* { */
|
|
/* double sDum[2][2]; */
|
|
/* double tot, totquad; */
|
|
/* double eta3x,eta3y; */
|
|
/* double ex,ey; */
|
|
|
|
/* calcQuad(data, tot, totquad, sDum); */
|
|
/* calcEta3(data,eta3x, eta3y,tot); */
|
|
|
|
/* double xpos_eta,ypos_eta; */
|
|
|
|
/* #ifdef MYROOT1 */
|
|
/* xpos_eta=((hhx->GetBinContent(hhx->GetXaxis()->FindBin(eta3x),hhy->GetYaxis()->FindBin(eta3y))))/((double)nSubPixels); */
|
|
/* ypos_eta=((hhy->GetBinContent(hhx->GetXaxis()->FindBin(eta3x),hhy->GetYaxis()->FindBin(eta3y))))/((double)nSubPixels); */
|
|
|
|
/* #endif */
|
|
/* #ifndef MYROOT1 */
|
|
/* ex=(eta3x-etamin)/etastep; */
|
|
/* ey=(eta3y-etamin)/etastep; */
|
|
|
|
/* if (ex<0) ex=0; */
|
|
/* if (ex>=nbeta) ex=nbeta-1; */
|
|
/* if (ey<0) ey=0; */
|
|
/* if (ey>=nbeta) ey=nbeta-1; */
|
|
|
|
/* xpos_eta=(((double)hhx[(int)(ey*nbeta+ex)]))/((double)nSubPixels); */
|
|
/* ypos_eta=(((double)hhy[(int)(ey*nbeta+ex)]))/((double)nSubPixels); */
|
|
/* #endif */
|
|
|
|
/* int_x=((double)x) + xpos_eta; */
|
|
/* int_y=((double)y) + ypos_eta; */
|
|
|
|
/* return; */
|
|
/* }; */
|
|
|
|
virtual int addToFlatField(double totquad,int quad,int *cl,double &etax, double &etay) {
|
|
|
|
calcEta3(cl, etax, etay, totquad);
|
|
return addToFlatField(etax,etay);
|
|
}
|
|
|
|
virtual int addToFlatField(double totquad,int quad,double *cl,double &etax, double &etay) {
|
|
|
|
|
|
calcEta3(cl, etax, etay, totquad);
|
|
return addToFlatField(etax,etay);
|
|
}
|
|
|
|
|
|
|
|
//////////////////////////////////////////////////////////////////////////////////////
|
|
virtual int addToFlatField(double *cluster, double &etax, double &etay){
|
|
double totquad;
|
|
calcEta3(cluster, etax, etay, totquad);
|
|
return addToFlatField(etax,etay);
|
|
|
|
};
|
|
|
|
virtual int addToFlatField(int *cluster, double &etax, double &etay){
|
|
|
|
double totquad;
|
|
|
|
calcEta3(cluster, etax, etay, totquad);
|
|
return addToFlatField(etax,etay);
|
|
|
|
};
|
|
|
|
|
|
virtual int addToFlatField(double etax, double etay){
|
|
#ifdef MYROOT1
|
|
heta->Fill(etax,etay);
|
|
#endif
|
|
#ifndef MYROOT1
|
|
int ex,ey;
|
|
ex=(etax-etamin)/etastepX;
|
|
ey=(etay-etamin)/etastepY;
|
|
if (ey<nbetaY && ex<nbetaX && ex>=0 && ey>=0)
|
|
heta[ey*nbetaX+ex]++;
|
|
#endif
|
|
return 0;
|
|
};
|
|
|
|
|
|
/* protected: */
|
|
|
|
/* #ifdef MYROOT1 */
|
|
/* TH2D *heta; */
|
|
/* TH2D *hhx; */
|
|
/* TH2D *hhy; */
|
|
/* #endif */
|
|
/* #ifndef MYROOT1 */
|
|
/* int *heta; */
|
|
/* float *hhx; */
|
|
/* float *hhy; */
|
|
/* #endif */
|
|
/* int nbeta; */
|
|
/* double etamin, etamax, etastep; */
|
|
|
|
};
|
|
|
|
#endif
|