TRestDetectorGarfieldDriftProcess.cxx

 
#include "TRestDetectorGarfieldDriftProcess.h"
 
#include <TGeoBBox.h>
#include <TRandom3.h>
 
ClassImp(TRestDetectorGarfieldDriftProcess);
 
#if defined USE_Garfield
using namespace Garfield;
#include <ComponentConstant.hh>
 
#include "TGDMLParse.h"
#endif
 
#include <stdio.h>
 
using namespace std;
 
const double cmTomm = 10.;
 
#if defined USE_Garfield
 
TRestDetectorGarfieldDriftProcess::TRestDetectorGarfieldDriftProcess() : fRandom(0), fGfSensor(0) {
    Initialize();
}
 
// __________________________________________________________
//     TODO : Perhaps this constructor should be removed
//            since we will always load the config from TRestRun
//            when we use AddProcess. It would be necessary only if we use the
//            process stand alone but even then we could just call LoadConfig
//            __________________________________________________________
 
TRestDetectorGarfieldDriftProcess::TRestDetectorGarfieldDriftProcess(const char* configFilename)
    : fRandom(0), fGfSensor(0) {
    Initialize();
 
    if (LoadConfigFromFile(configFilename) == -1) LoadDefaultConfig();
 
    PrintMetadata();
 
    if (fReadout == nullptr) fReadout = new TRestDetectorReadout(configFilename);
}
 
// TRestDetectorGarfieldDriftProcess destructor
TRestDetectorGarfieldDriftProcess::~TRestDetectorGarfieldDriftProcess() {
    delete fReadout;
    delete fGeometry;
    delete fOutputHitsEvent;
}
 
void TRestDetectorGarfieldDriftProcess::LoadDefaultConfig() {
    SetName("garfieldDriftProcess-Default");
    SetTitle("Default config");
 
    cout << "TRestDetectorGarfieldDriftProcess metadata not found. Loading default values" << endl;
 
    fDriftPotential = 1000;
    fGasPressure = 10;
}
 
void TRestDetectorGarfieldDriftProcess::LoadConfig(const string& configFilename, const string& name) {
    if (LoadConfigFromFile(configFilename, name)) LoadDefaultConfig();
 
    if (fDriftPotential == PARAMETER_NOT_FOUND_DBL) {
        fDriftPotential = 1000;  // V
    }
}
 
#endif
 
void TRestDetectorGarfieldDriftProcess::Initialize() {
    SetSectionName(ClassName());
    SetLibraryVersion(LIBRARY_VERSION);
 
    fRandom = new TRandom3(0);
    fInputHitsEvent = nullptr;
    fOutputHitsEvent = new TRestDetectorHitsEvent();
 
#if defined USE_Garfield
    fReadout = nullptr;
    fGas = nullptr;
    fGeometry = nullptr;
    fPEReduction = 1.;
    fStopDistance = 2;  // default distance from readout to stop drift set to 2mm
#endif
}
 
#if defined USE_Garfield
void TRestDetectorGarfieldDriftProcess::InitProcess() {
    // Function to be executed once at the beginning of process
    // (before starting the process of the events)
 
    // Start by calling the InitProcess function of the abstract class.
    // Comment this if you don't want it.
    // TRestEventProcess::InitProcess();
 
    // Getting gas data
    fGas = GetMetadata<TRestDetectorGas>();
    if (fGas != nullptr) {
        if (fGasPressure <= 0)
            fGasPressure = fGas->GetPressure();
        else
            fGas->SetPressure(fGasPressure);
 
    } else {
        cout << "REST_WARNING. No TRestDetectorGas found in TRestRun." << endl;
    }
 
    // Getting readout data
    fReadout = GetMetadata<TRestDetectorReadout>();
    if (fReadout == nullptr) {
        cout << "REST ERROR: Readout has not been initialized" << endl;
        exit(-1);
    }
 
    // Getting geometry data
    if (fGDML_Filename == "") {
        cout << "REST ERROR no geometry GDML file name parameter (gdml_file) given "
                "in TRestDetectorGarfieldDriftProcess "
             << endl;
        exit(-1);
    }
 
    TGeoVolume* geovol = nullptr;
    fGeometry = new TRestDetectorGeometry();
    fGeometry->InitGfGeometry();
 
    geovol = TGDMLParse::StartGDML(fGDML_Filename);
    if (!geovol) {
        cout << "REST ERROR when TRestDetectorGeometry read GDML file " << fGDML_Filename << endl;
        exit(-1);
    }
 
    fGeometry->SetTopVolume(geovol);
    fGeometry->CloseGeometry();
    fGeometry->DefaultColors();
    //         fGeometry->UpdateElements();
 
    cout << "TRestDetectorGarfieldDriftProcess  GetVerboseLevel : "
         << static_cast<int>(this->GetVerboseLevel()) << endl;
 
    // analyze GDML geometry to find major elements (gas volume, electrodes,
    // readout)
    TObjArray* thenodes = geovol->GetNodes();
    for (TIter it = thenodes->begin(); it != thenodes->end(); ++it) {
        TGeoNode* itnode = (TGeoNode*)(*it);
        if (GetVerboseLevel() >= TRestStringOutput::REST_Verbose_Level::REST_Info) {
            cout << "****** itnode " << itnode->GetName() << endl;
            itnode->PrintCandidates();
        }
        cout << "****** itnode " << itnode->GetName() << endl;
        itnode->PrintCandidates();
 
        TGeoVolume* itvol = itnode->GetVolume();
        if (GetVerboseLevel() >= TRestStringOutput::REST_Verbose_Level::REST_Info) {
            cout << "  *  *  itvolume " << itvol->GetName() << endl;
            itvol->Print();
        }
        cout << "  *  *  itvolume " << itvol->GetName() << endl;
        itvol->Print();
 
        TGeoMedium* itmed = itvol->GetMedium();
        if (GetVerboseLevel() >= TRestStringOutput::REST_Verbose_Level::REST_Info)
            cout << "  *  *  itmed " << itmed->GetName() << endl;
 
        // gas volume
        if (fGas->GetGDMLMaterialRef() == itmed->GetName()) {
            fGeometry->SetGfGeoMedium(itmed->GetName(), fGas);
            if (GetVerboseLevel() >= TRestStringOutput::REST_Verbose_Level::REST_Info) {
                cout << "  -> gas volume   SetMedium itmed " << itmed->GetName() << "  fGas "
                     << fGas->GetGasMedium()->GetName() << endl;
                fGas->GetGasMedium()->PrintGas();
            }
            cout << "  -> gas volume   SetMedium itmed " << itmed->GetName() << "  fGas "
                 << fGas->GetGasMedium()->GetName() << endl;
            fGas->GetGasMedium()->PrintGas();
        }
 
        // drift electrode (it should be called a cathode, not anode, no ?)
        if ((strncmp(itvol->GetName(), "anodeVol", 8) == 0) ||
            (strncmp(itvol->GetName(), "cathodeVol", 10) == 0)) {
            fGeometry->SetDriftElecNode(itnode);
            if (GetVerboseLevel() >= TRestStringOutput::REST_Verbose_Level::REST_Info)
                cout << "  -> cathode volume " << endl;
            cout << "  -> cathode volume " << endl;
        }
 
        // micromegas readout electrode
        if ((strncmp(itvol->GetName(), "micromegasVol", 13) == 0)) {
            fGeometry->AddReadoutElecNode(itnode);
            if (GetVerboseLevel() >= TRestStringOutput::REST_Verbose_Level::REST_Info)
                cout << "  -> readout volume " << endl;
            cout << "  -> readout volume " << endl;
        }
    }
 
    // For the moment we set only constant electric field in Garfield
    double matrixZpos = 0, planeZpos = 0, planeZvec = 0;
    cout << "fGeometry " << fGeometry << " nb node " << fGeometry->GetNReadoutElecNodes() << endl;
    for (int ii = 0; ii < fGeometry->GetNReadoutElecNodes(); ii++) {
        bool rdPlaneFound = false;
 
        TGeoNode* readoutnode = fGeometry->GetReadoutElecNode(ii);
        cout << "ii " << ii << " readoutnode " << readoutnode << " visible " << readoutnode->IsVisible()
             << endl;
        if (!readoutnode) continue;
        cout << "readoutnode visible " << readoutnode->IsVisible() << endl;
        readoutnode->ls();
        readoutnode->PrintCandidates();
        TGeoMatrix* readoutmatrix = readoutnode->GetMatrix();
        if (!readoutmatrix) continue;
        cout << "readoutmatrix " << endl;
        readoutmatrix->Print();
        matrixZpos = (readoutmatrix->IsTranslation() ? 10 * readoutmatrix->GetTranslation()[2]
                                                     : 0);  // converted to mm
        for (int jj = 0; jj < fReadout->GetNumberOfReadoutPlanes(); jj++) {
            TRestDetectorReadoutPlane* readoutplane = fReadout->GetReadoutPlane(jj);
            planeZpos = readoutplane->GetPosition().Z();
            cout << "    jj " << jj << " matrixZpos " << matrixZpos << " planeZpos " << planeZpos << endl;
            if (fabs(planeZpos - matrixZpos) > 1)
                continue;  // we search for fReadout entry at same Z position
            rdPlaneFound = true;
            planeZvec = readoutplane->GetNormal().z();
            cout << "     planeZvec " << planeZvec << endl;
        }
 
        if (rdPlaneFound) {
            double field = fDriftPotential /
                           (fGeometry->GetDriftElecNode()->GetMatrix()->GetTranslation()[2] - planeZpos);
            ComponentConstant* cmp = new ComponentConstant();
            //         cmp->SetElectricField(0, 0, fElectricField);  // assuming V/cm
            cmp->SetElectricField(0, 0, field);           // assuming V/cm
            cmp->SetPotential(0, 0, planeZpos / 10., 0);  // potential at readout level set to 0
            cmp->SetWeightingField(0, 0, planeZvec, "m");
            fGeometry->AddGfComponent(cmp);
            cout << "add component field " << field << " planeZpos " << planeZpos << " planeZvec "
                 << planeZvec << " drift pos "
                 << fGeometry->GetDriftElecNode()->GetMatrix()->GetTranslation()[2] << endl;
        }
    }
 
    // create Sensor corresponding to readout plane
    // only the first component used, so 1 sensor only for the moment
    if (fGeometry->GetNbOfGfComponent() > 0) {
        fGfSensor = new Sensor();
        fGfSensor->AddComponent(fGeometry->GetGfComponent(0));
        fGfSensor->AddElectrode(fGeometry->GetGfComponent(0), "m");
        fGfSensor->SetTimeWindow(0., 0.1, 500);
 
        double xmin = -1e9, xmax = 1e9, ymin = -1e9, ymax = 1e9;
        TGeoShape* readoutshape = fGeometry->GetReadoutElecNode(0)->GetVolume()->GetShape();
        TGeoMatrix* readoutmatrix = fGeometry->GetReadoutElecNode(0)->GetMatrix();
        double xmid = 10. * readoutmatrix->GetTranslation()[0],
               ymid = 10. * readoutmatrix->GetTranslation()[1];
        if (readoutshape->InheritsFrom("TGeoBBox")) {
            // keep drifting electrons 10cm around the readout volume in x and y
            TGeoBBox* readoutbox = (TGeoBBox*)readoutshape;
            xmin = xmid - 10. * readoutbox->GetDX() - 100;
            xmax = xmid + 10. * readoutbox->GetDX() + 100;
            ymin = ymid - 10. * readoutbox->GetDY() - 100;
            ymax = ymid + 10. * readoutbox->GetDY() + 100;
        }
        double driftelecZpos = fGeometry->GetDriftElecNode()
                                   ->GetMatrix()
                                   ->GetTranslation()[2];  // hope that all these objects are really there...
        // drift area defined from bouding box of readout shape
        fGfSensor->SetArea(xmin / 10., ymin / 10., planeZpos / 10. + fStopDistance, xmax / 10., ymax / 10.,
                           driftelecZpos);  // drift stops fStopDistance above the readout plane
        fGeometry->AddGfSensor(fGfSensor);
        printf("  GfSensor area x- %lf y- %lf z- %lf x+ %lf y+ %lf z+ %lf\n", xmin / 10., ymin / 10.,
               planeZpos / 10. + 0.2, xmax / 10., ymax / 10., driftelecZpos);
 
        fGfDriftMethod = new DRIFT_METHOD();
        //       fGfDriftMethod->EnableDebugging();
        //       fGfDriftMethod->EnableVerbose();
        fGfDriftMethod->SetSensor(fGfSensor);
        //       fGfDriftMethod->SetTimeSteps(0.2);
        fGfDriftMethod->SetCollisionSteps(100);
    }
 
    if (!fGfDriftMethod || !fGfSensor) {
        cout << "REST ERROR Garfield not fully initialized in "
                "TRestDetectorGarfieldDriftProcess::InitProcess "
             << endl;
        cout << "  fGfDriftMethod " << fGfDriftMethod << " fGfSensor " << fGfSensor << " GetNbOfGfComponent "
             << fGeometry->GetNbOfGfComponent() << endl;
        cout << " matrixZpos " << matrixZpos << " planeZpos " << planeZpos << " planeZvec " << planeZvec
             << endl;
        exit(-1);
    }
 
    // for (register double ix=-3000; ix<3000; ix+=10) {
    // printf("  ix %lf medium %p is_inside %d", ix,
    // fGeometry->GetGfGeometry()->GetMedium(ix/10.,5,0),
    //               fGeometry->GetGfGeometry()->IsInside(ix/10.,5,0));
    //   cout << " moduleId " <<
    //   fReadout->GetReadoutPlane(0)->isInsideDriftVolume( ix, 50, 0 );
    // fGeometry->SetCurrentPoint(ix/10., 5,0);
    //   if (fGeometry->IsOutside()) cout << " is outside " << endl; else cout <<
    //   " is inside" << endl; TGeoNode* cnode =fGeometry ->GetCurrentNode();
    // TGeoNode* cnode2= fGeometry->FindNode(ix/10., 5,0);
    //   std::string name(cnode->GetMedium()->GetMaterial()->GetName());
    //   register Medium* tmed = fGeometry->GetGfMedium(ix/10., 5,0);
    //   cout << "    node " << cnode->GetName() << " material " << name;
    //   cout << " node2 " << cnode2->GetName() << " material " <<
    //   cnode2->GetMedium()->GetMaterial()->GetName(); if (tmed) cout << " medium
    //   " << tmed->GetName(); cout << endl;
    // }
}
 
//------------------------------------------------------------------------------
 
Int_t TRestDetectorGarfieldDriftProcess::FindModule(Int_t readoutPlane, Double_t x, Double_t y) {
    // TODO verify this
    TRestDetectorReadoutPlane* plane = fReadout->GetReadoutPlane(readoutPlane);
    for (size_t md = 0; md < plane->GetNumberOfModules(); md++) {
        if ((*plane)[md].IsInside({x, y})) {
            return md;
        }
    }
 
    return -1;
}
 
#endif
 
TRestEvent* TRestDetectorGarfieldDriftProcess::ProcessEvent(TRestEvent* inputEvent) {
#if defined USE_Garfield
    fInputHitsEvent = (TRestDetectorHitsEvent*)inputEvent;
 
    double x, y, z, energy;
    double xi, yi, zi, ti, xf, yf, zf, tf, energyf;
    int status;
 
    if (GetVerboseLevel() >= TRestStringOutput::REST_Verbose_Level::REST_Debug) {
        cout << "Number of hits : " << fInputHitsEvent->GetNumberOfHits() << endl;
        cout << "--------------------------" << endl;
        fInputHitsEvent->PrintEvent(20);
    }
 
    for (unsigned int hit = 0; hit < fInputHitsEvent->GetNumberOfHits(); hit++) {
        x = fInputHitsEvent->GetX(hit);
        y = fInputHitsEvent->GetY(hit);
        z = fInputHitsEvent->GetZ(hit);
        energy = fInputHitsEvent->GetEnergy(hit);
        //         if (hit%10==0) printf("hit: x %lf y %lf z %lf Energy %lf\n", x,
        //         y, z, energy); cout  <<  "hit: x "  <<  x  <<  " y "  <<  y  <<
        //         " z "  <<  z  << " Energy " << energy <<  endl;
 
        // assuming energy in keV, W factor in eV
        Medium* tmed = fGeometry->GetGfMedium(x, y, z);
        if (!tmed) {
            cout << "no medium found at x " << x << " y " << y << " z " << z << endl;
            continue;
            //         } else {
            //           cout << "  medium " << tmed->GetName() << " W " <<
            //           tmed->GetW() << " at x " << x << " y " << y << " z " << z <<
            //           endl;
        }
        double Wfactor = tmed->GetW();
        double nb_electrons_real = energy * 1000. / Wfactor / fPEReduction;
        unsigned int nb_electrons =
            fRandom->Poisson(nb_electrons_real);  // we need an int number of electrons to drift...
        //         if (nb_electrons>0) printf("hit: x %lf y %lf z %lf Energy %lf\n",
        //         x, y, z, energy); if (nb_electrons>0) cout  <<  "   : energy " <<
        //         energy << " nb_electrons_real " << nb_electrons_real << "
        //         nb_electrons " << nb_electrons << endl;
 
        // lets drift all those electrons
        for (unsigned int iel = 0; iel < nb_electrons; iel++) {
            // drift an electron
            fGfDriftMethod->DriftElectron(x / 10., y / 10., z / 10., 0);
            //           fGfDriftMethod->GetEndPoint(xf, yf, zf, tf, status);
            if (fGfDriftMethod->GetNumberOfElectronEndpoints() == 0) {
                cout << " TRestDetectorGarfieldDriftProcess::ProcessEvent:  no electron end "
                        "point ???"
                     << endl;
                continue;
            }
 
            // create an hit for the drifted electron
            fGfDriftMethod->GetElectronEndpoint(0, xi, yi, zi, ti, xf, yf, zf, tf,
                                                status);  // works with AvalancheMC
            // xi *= 10; yi *= 10; zi *= 10; // cm to mm
            xf *= cmTomm;
            yf *= cmTomm;
            zf *= cmTomm;  // cm to mm
 
            tf /= 1000.;                               // ns to us
            energyf = fPEReduction * Wfactor / 1000.;  // back to keV
            fOutputHitsEvent->AddHit(xf, yf, zf, energyf, tf);
 
            //           if (nb_electrons>0) cout  <<  "   : Nendpoint " <<
            //           fGfDriftMethod->GetNumberOfElectronEndpoints() << " xi "  <<
            //           xi  <<  " yi "  <<  yi  <<  " zi "  <<  zi  << " ti " << ti
            //                 << " xf "  <<  xf  <<  " yf "  <<  yf  <<  " zf "  <<
            //                 zf  << " tf " << tf << " status " << status ;
        }
    }
 
    if (fOutputHitsEvent->GetNumberOfHits() == 0) return nullptr;
 
    // fSignalEvent->PrintEvent();
 
    if (GetVerboseLevel() >= TRestStringOutput::REST_Verbose_Level::REST_Debug) {
        cout << "TRestDetectorElectronDiffusionProcess. Hits added : " << fOutputHitsEvent->GetNumberOfHits()
             << endl;
        cout << "TRestDetectorElectronDiffusionProcess. Hits total energy : "
             << fOutputHitsEvent->GetTotalEnergy() << endl;
        cout << " fTimedHitsEvent " << fOutputHitsEvent << " class " << fOutputHitsEvent->ClassName() << endl;
        fOutputHitsEvent->PrintEvent(20);
    }
 
    return fOutputHitsEvent;
#else
    fInputHitsEvent = (TRestDetectorHitsEvent*)inputEvent;
    fOutputHitsEvent = fInputHitsEvent;
    RESTDebug << "nullptr process" << RESTendl;
    return inputEvent;
 
#endif
}
 
#if defined USE_Garfield
 
void TRestDetectorGarfieldDriftProcess::EndProcess() {
    // Function to be executed once at the end of the process
    // (after all events have been processed)
 
    // Start by calling the EndProcess function of the abstract class.
    // Comment this if you don't want it.
    // TRestEventProcess::EndProcess();
}
 
void TRestDetectorGarfieldDriftProcess::InitFromConfigFile() {
    fGasPressure = StringToDouble(GetParameter("gasPressure", "-1"));
    //     fElectricField = GetDblParameterWithUnits( "electricField" );
    fDriftPotential = GetDblParameterWithUnits("driftPotential");
    fPEReduction = StringToDouble(GetParameter("primaryElectronReduction", "1"));
    fStopDistance = StringToDouble(GetParameter("stopDistance", "2"));
 
    // TODO : Still units must be implemented for velocity quantities
 
    fGDML_Filename = GetParameter("geometryPath", "") + GetParameter("gdml_file", "");
}
 
#endif