BmnVSP.cxx

Go to the documentation of this file.

#include "BmnVSP.h"
 
#include "BmnVSPPoint.h"
#include "CbmStack.h"
#include "FairGeoMedia.h"
#include "FairGeoRootBuilder.h"
#include "FairRootManager.h"
#include "FairVolume.h"
#include "TFile.h"
#include "TGDMLParse.h"
#include "TGeoManager.h"
#include "TParticle.h"
#include "TParticlePDG.h"
#include "TRegexp.h"
#include "TVirtualMC.h"
 
#include <iostream>
 
using namespace std;
 
BmnVSP::BmnVSP()
    : FairDetector("VSP", kTRUE)
{
    fPointCollection = new TClonesArray("BmnVSPPoint");
    fPosIndex = 0;
    fVerboseLevel = 1;
    ResetParameters();
}
 
BmnVSP::BmnVSP(const char* name, Bool_t active)
    : FairDetector(name, active)
{
    fPointCollection = new TClonesArray("BmnVSPPoint");
    fPosIndex = 0;
    fVerboseLevel = 1;
    ResetParameters();
}
 
BmnVSP::~BmnVSP()
{
    if (fPointCollection) {
        fPointCollection->Delete();
        delete fPointCollection;
    }
}
 
//------------------------------------------------------------------------------
 
Bool_t BmnVSP::ProcessHits(FairVolume* vol)
{
 
    // Set parameters at entrance of volume. Reset ELoss.
    if (gMC->IsTrackEntering()) {
 
        ResetParameters();
        fELoss = 0.;
        fTime = gMC->TrackTime() * 1.0e09;
        fLength = gMC->TrackLength();
        fIsPrimary = 0;
        fCharge = -1;
        fPdgId = 0;
 
        TLorentzVector PosIn;
        gMC->TrackPosition(PosIn);
        fPosIn.SetXYZ(PosIn.X(), PosIn.Y(), PosIn.Z());
 
        TLorentzVector MomIn;
        gMC->TrackMomentum(MomIn);
        fMomIn.SetXYZ(MomIn.X(), MomIn.Y(), MomIn.Z());
 
        TParticle* part = 0;
        part = gMC->GetStack()->GetCurrentTrack();
        if (part) {
            fIsPrimary = (Int_t)part->IsPrimary();
            fCharge = (Int_t)part->GetPDG()->Charge();
            fPdgId = (Int_t)part->GetPdgCode();
        }
 
        fVolumeID = vol->getMCid();
        fTrackID = gMC->GetStack()->GetCurrentTrackNumber();
    }
 
    // Sum energy loss for all steps in the active volume
    fELoss += gMC->Edep();
 
    // Create BmnVSPPoint at EXIT of active volume;
    if ((gMC->IsTrackExiting() || gMC->IsTrackStop() || gMC->IsTrackDisappeared()) && fELoss > 0) {
 
        TLorentzVector PosOut;
        gMC->TrackPosition(PosOut);
        fPosOut.SetXYZ(PosOut.X(), PosOut.Y(), PosOut.Z());
 
        TLorentzVector MomOut;
        gMC->TrackMomentum(MomOut);
        fMomOut.SetXYZ(MomOut.X(), MomOut.Y(), MomOut.Z());
 
        // correction step to avoid the seg. violation error due to invalid memory access
        TVector3 diff_pos = fPosIn - fPosOut;
 
        if (diff_pos.Mag() < 0.001)
            return kFALSE;   // ignore points produced by tracks with zero length inside the current sens. volume
        if (fMomOut.Mag() == 0)
            return kFALSE;   // ignore points produced by tracks with zero momentum inside the current sens. volume
 
        TVector3 corr_step = fMomOut;
        corr_step.SetMag(0.001);   // 10 um
        TVector3 pos = fPosOut;
        fPosOut = pos - corr_step;
        gGeoManager->FindNode(fPosOut[0], fPosOut[1], fPosOut[2]);
 
        if (gGeoManager->GetCurrentNode()->GetMotherVolume() == 0)
            return kFALSE;   // check if the current node has its mother vol.
 
        // Determine station and module numbers for the current hit ------------
        Int_t stationNum = -1;   // current station number (default)
        Int_t moduleNum = -1;    // current module number (default)
 
        TGeoVolume* motherVolume = gGeoManager->GetCurrentNode()->GetMotherVolume();
        TString moduleVolumeName = motherVolume->GetName();
 
        TRegexp expr = "^moduleV_module[0-9]+_station[0-9]+_VSP$";
        if (moduleVolumeName.Contains(expr)) {
            TRegexp mod_expr = "module[0-9]+";
            TRegexp stat_expr = "station[0-9]+";
            moduleNum = TString(TString(moduleVolumeName(mod_expr))(TRegexp("[0-9]+"))).Atoi();
            stationNum = TString(TString(moduleVolumeName(stat_expr))(TRegexp("[0-9]+"))).Atoi();
        }
 
        if (stationNum == -1 || moduleNum == -1)
            return kFALSE;
 
        BmnVSPPoint* p = AddHit(fTrackID, fVolumeID, fPosIn, fPosOut, fMomIn, fMomOut, fTime, fLength, fELoss,
                                fIsPrimary, fCharge, fPdgId);
 
        // if (stationNum == -1) {
        //     cout << "BmnVSP.cxx" << endl;
        //     cout << moduleVolumeName << endl;
        //     cout << "stationNum = " << stationNum << "\n";
        //     cout << "moduleNum = " << moduleNum << "\n";
        //     p->Print(nullptr);
        //     cout << "\n";
        // }
 
        p->SetStation(stationNum);
        p->SetModule(moduleNum);
 
        ((CbmStack*)gMC->GetStack())->AddPoint(kVSP);
    }
 
    return kTRUE;
}
 
void BmnVSP::EndOfEvent()
{
    if (fVerboseLevel)
        Print("");
    fPointCollection->Clear();
    fPosIndex = 0;
}
 
void BmnVSP::Register()
{
    FairRootManager::Instance()->Register("VSPPoint", "VSP", fPointCollection, kTRUE);
}
 
TClonesArray* BmnVSP::GetCollection(Int_t iColl) const
{
    if (iColl == 0)
        return fPointCollection;
    return NULL;
}
 
void BmnVSP::Print(Option_t*) const
{
    Int_t nHits = fPointCollection->GetEntriesFast();
    cout << "-I- BmnVSP: " << nHits << " points registered in this event." << endl;
 
    if (fVerboseLevel > 1) {
        for (Int_t i = 0; i < nHits; i++) {
            (*fPointCollection)[i]->Print();
        }
    }
}
 
void BmnVSP::Reset()
{
    fPointCollection->Clear();
    ResetParameters();
}
 
void BmnVSP::CopyClones(TClonesArray* cl1, TClonesArray* cl2, Int_t offset)
{
    Int_t nEntries = cl1->GetEntriesFast();
    cout << "-I- BmnVSP: " << nEntries << " entries to add." << endl;
    TClonesArray& clref = *cl2;
    BmnVSPPoint* oldpoint = NULL;
 
    for (Int_t i = 0; i < nEntries; i++) {
        oldpoint = (BmnVSPPoint*)cl1->At(i);
        Int_t index = oldpoint->GetTrackID() + offset;
        oldpoint->SetTrackID(index);
        new (clref[fPosIndex]) BmnVSPPoint(*oldpoint);
        fPosIndex++;
    }
 
    cout << "-I- BmnVSP: " << cl2->GetEntriesFast() << " merged entries." << endl;
}
 
void BmnVSP::ConstructGeometry()
{
    TString fileName = GetGeometryFileName();
 
    if (fileName.EndsWith(".root")) {
        LOG(info) << "Constructing VSP geometry from ROOT file " << fileName.Data();
        ConstructRootGeometry();
    }
 
    else if (fileName.EndsWith(".gdml"))
    {
        LOG(info) << "Constructing VSP geometry from GDML file " << fileName.Data();
        ConstructGDMLGeometry(nullptr);
    }
 
    else
    {
        LOG(fatal) << "Geometry format of VSP file " << fileName.Data() << " not supported.";
    }
}
 
// -----   ConstructGDMLGeometry   ---------------------------------------------
void BmnVSP::ConstructGDMLGeometry(TGeoMatrix*)
{
    TFile* old = gFile;
    TGDMLParse parser;
    TGeoVolume* gdmlTop;
 
    // Before importing GDML
    Int_t maxInd = gGeoManager->GetListOfMedia()->GetEntries() - 1;
 
    gdmlTop = parser.GDMLReadFile(GetGeometryFileName());
 
    // Cheating - reassigning media indices after GDML import (need to fix this in TGDMLParse class!!!)
    //   for (Int_t i=0; i<gGeoManager->GetListOfMedia()->GetEntries(); i++)
    //      gGeoManager->GetListOfMedia()->At(i)->Dump();
    // After importing GDML
    Int_t j = gGeoManager->GetListOfMedia()->GetEntries() - 1;
    Int_t curId;
    TGeoMedium* m;
    do {
        m = (TGeoMedium*)gGeoManager->GetListOfMedia()->At(j);
        curId = m->GetId();
        m->SetId(curId + maxInd);
        j--;
    } while (curId > 1);
 
    //   LOG(debug) << "====================================================================";
    //   for (Int_t i=0; i<gGeoManager->GetListOfMedia()->GetEntries(); i++)
    //       gGeoManager->GetListOfMedia()->At(i)->Dump();
 
    Int_t newMaxInd = gGeoManager->GetListOfMedia()->GetEntries() - 1;
 
    gGeoManager->GetTopVolume()->AddNode(gdmlTop, 1, 0);
    ExpandNodeForGdml(gGeoManager->GetTopVolume()->GetNode(gGeoManager->GetTopVolume()->GetNdaughters() - 1));
 
    for (Int_t k = maxInd + 1; k < newMaxInd + 1; k++) {
        TGeoMedium* medToDel = (TGeoMedium*)(gGeoManager->GetListOfMedia()->At(maxInd + 1));
        LOG(debug) << "    removing media " << medToDel->GetName() << " with id " << medToDel->GetId() << " (k=" << k
                   << ")";
        gGeoManager->GetListOfMedia()->Remove(medToDel);
    }
    gGeoManager->SetAllIndex();
 
    gFile = old;
}
 
void BmnVSP::ExpandNodeForGdml(TGeoNode* node)
{
    LOG(debug) << "----------------------------------------- ExpandNodeForGdml for node " << node->GetName();
 
    TGeoVolume* curVol = node->GetVolume();
 
    LOG(debug) << "    volume: " << curVol->GetName();
 
    if (curVol->IsAssembly()) {
        LOG(debug) << "    skipping volume-assembly";
    } else {
        TGeoMedium* curMed = curVol->GetMedium();
        TGeoMaterial* curMat = curVol->GetMaterial();
        TGeoMedium* curMedInGeoManager = gGeoManager->GetMedium(curMed->GetName());
        TGeoMaterial* curMatOfMedInGeoManager = curMedInGeoManager->GetMaterial();
        TGeoMaterial* curMatInGeoManager = gGeoManager->GetMaterial(curMat->GetName());
 
        // Current medium and material assigned to the volume from GDML
        LOG(debug2) << "    curMed\t\t\t\t" << curMed << "\t" << curMed->GetName() << "\t" << curMed->GetId();
        LOG(debug2) << "    curMat\t\t\t\t" << curMat << "\t" << curMat->GetName() << "\t" << curMat->GetIndex();
 
        // Medium and material found in the gGeoManager - either the pre-loaded one or one from GDML
        LOG(debug2) << "    curMedInGeoManager\t\t" << curMedInGeoManager << "\t" << curMedInGeoManager->GetName()
                    << "\t" << curMedInGeoManager->GetId();
        LOG(debug2) << "    curMatOfMedInGeoManager\t\t" << curMatOfMedInGeoManager << "\t"
                    << curMatOfMedInGeoManager->GetName() << "\t" << curMatOfMedInGeoManager->GetIndex();
        LOG(debug2) << "    curMatInGeoManager\t\t" << curMatInGeoManager << "\t" << curMatInGeoManager->GetName()
                    << "\t" << curMatInGeoManager->GetIndex();
 
        TString matName = curMat->GetName();
        TString medName = curMed->GetName();
 
        if (curMed->GetId() != curMedInGeoManager->GetId()) {
            if (fFixedMedia.find(medName) == fFixedMedia.end()) {
                LOG(debug) << "    Medium needs to be fixed";
                fFixedMedia[medName] = curMedInGeoManager;
                Int_t ind = curMat->GetIndex();
                gGeoManager->RemoveMaterial(ind);
                LOG(debug) << "    removing material " << curMat->GetName() << " with index " << ind;
                for (Int_t i = ind; i < gGeoManager->GetListOfMaterials()->GetEntries(); i++) {
                    TGeoMaterial* m = (TGeoMaterial*)gGeoManager->GetListOfMaterials()->At(i);
                    m->SetIndex(m->GetIndex() - 1);
                }
 
                LOG(debug) << "    Medium fixed";
            } else {
                LOG(debug) << "    Already fixed medium found in the list    ";
            }
        } else {
            if (fFixedMedia.find(medName) == fFixedMedia.end()) {
                LOG(debug) << "    There is no correct medium in the memory yet";
 
                FairGeoLoader* geoLoad = FairGeoLoader::Instance();
                FairGeoInterface* geoFace = geoLoad->getGeoInterface();
                FairGeoMedia* geoMediaBase = geoFace->getMedia();
                FairGeoBuilder* geobuild = geoLoad->getGeoBuilder();
 
                FairGeoMedium* curMedInGeo = geoMediaBase->getMedium(medName);
                if (curMedInGeo == 0) {
                    LOG(fatal) << "    Media not found in Geo file: " << medName;
                } else {
                    LOG(debug) << "    Found media in Geo file" << medName;
                    /*Int_t nmed = */ geobuild->createMedium(curMedInGeo);
                    fFixedMedia[medName] = (TGeoMedium*)gGeoManager->GetListOfMedia()->Last();
                    gGeoManager->RemoveMaterial(curMatOfMedInGeoManager->GetIndex());
                    LOG(debug) << "    removing material " << curMatOfMedInGeoManager->GetName() << " with index "
                               << curMatOfMedInGeoManager->GetIndex();
                    for (Int_t i = curMatOfMedInGeoManager->GetIndex();
                         i < gGeoManager->GetListOfMaterials()->GetEntries(); i++)
                    {
                        TGeoMaterial* m = (TGeoMaterial*)gGeoManager->GetListOfMaterials()->At(i);
                        m->SetIndex(m->GetIndex() - 1);
                    }
                }
 
                if (curMedInGeo->getSensitivityFlag()) {
                    LOG(debug) << "    Adding sensitive  " << curVol->GetName();
                    AddSensitiveVolume(curVol);
                }
            } else {
                LOG(debug) << "    Already fixed medium found in the list";
                LOG(debug) << "!!! Sensitivity: " << fFixedMedia[medName]->GetParam(0);
                if (fFixedMedia[medName]->GetParam(0) == 1) {
                    LOG(debug) << "    Adding sensitive  " << curVol->GetName();
                    AddSensitiveVolume(curVol);
                }
            }
        }
 
        curVol->SetMedium(fFixedMedia[medName]);
        gGeoManager->SetAllIndex();
 
        // gGeoManager->GetListOfMaterials()->Print();
        //  gGeoManager->GetListOfMedia()->Print();
    }
 
    if (curVol->GetNdaughters() != 0) {
        TObjArray* NodeChildList = curVol->GetNodes();
        TGeoNode* curNodeChild;
        for (Int_t j = 0; j < NodeChildList->GetEntriesFast(); j++) {
            curNodeChild = (TGeoNode*)NodeChildList->At(j);
            ExpandNodeForGdml(curNodeChild);
        }
    }
}
 
//--- Check if Sensitive -------------------------------------------------------
Bool_t BmnVSP::CheckIfSensitive(std::string name)
{
    TString tsname = name;
    if (tsname.Contains("sensor")) {
        return kTRUE;
    }
    return kFALSE;
}
//------------------------------------------------------------------------------
 
//------------------------------------------------------------------------------
BmnVSPPoint* BmnVSP::AddHit(Int_t trackID,
                            Int_t detID,
                            TVector3 posIn,
                            TVector3 posOut,
                            TVector3 momIn,
                            TVector3 momOut,
                            Double_t time,
                            Double_t length,
                            Double_t eLoss,
                            Int_t isPrimary,
                            Double_t charge,
                            Int_t pdgId)
{
 
    TClonesArray& clref = *fPointCollection;
    Int_t size = clref.GetEntriesFast();
    // std::cout << "ELoss: " << eLoss << "\n";
    return new (clref[size])
        BmnVSPPoint(trackID, detID, posIn, posOut, momIn, momOut, time, length, eLoss, isPrimary, charge, pdgId);
}
//------------------------------------------------------------------------------