FairDetector.cxx

/********************************************************************************
 * Copyright (C) 2014-2024 GSI Helmholtzzentrum fuer Schwerionenforschung GmbH  *
 *                                                                              *
 *              This software is distributed under the terms of the             *
 *              GNU Lesser General Public Licence (LGPL) version 3,             *
 *                  copied verbatim in the file "LICENSE"                       *
 ********************************************************************************/
// -------------------------------------------------------------------------
// -----                    FairDetector source file                   -----
// -----            Created 06/01/04  by M. Al-Turany/ D. Bertini      -----
// -------------------------------------------------------------------------

#include "FairDetector.h"

#include "FairGeoNode.h"   // for FairGeoNode
#include "FairRootManager.h"
#include "FairRunSim.h"
#include "FairVolume.h"   // for FairVolume

#include <TFolder.h>       // for TFolder
#include <TGeoManager.h>   // for gGeoManager
#include <TList.h>         // for TList
#include <TObject.h>       // for TObject
#include <TROOT.h>         // for TROOT, gROOT
#include <TRefArray.h>     // for TRefArray
#include <TString.h>       // for TString
#include <TVirtualMC.h>    // for TVirtualMC
#include <fairlogger/Logger.h>

FairDetector::FairDetector(const char* Name, Bool_t Active, Int_t DetId)
    : FairModule(Name, "FAIR Detector", Active)
    , fDetId(DetId)
{
    flGeoPar = new TList();
    TString lname(GetName());
    lname += "GeoPar";
    flGeoPar->SetName(lname.Data());
    fGeoSaved = kFALSE;
}

FairDetector::FairDetector(const FairDetector& rhs)
    : FairModule(rhs)
    , fDetId(rhs.fDetId)
{}

FairDetector::~FairDetector() { delete flGeoPar; }

FairDetector& FairDetector::operator=(const FairDetector& rhs)
{
    // check assignment to self
    if (this == &rhs)
        return *this;

    // base class assignment
    FairModule::operator=(rhs);

    // assignment operator
    fDetId = rhs.fDetId;

    return *this;
}

FairDetector::FairDetector()
    : fDetId(0)
{}

// -------------------------------------------------------------------------

void FairDetector::DefineSensitiveVolumes()
{
    LOG(info) << "FairDetector::DefineSensitiveVolumes";
    TObjArray* volumes = gGeoManager->GetListOfVolumes();
    TIter next(volumes);
    TGeoVolume* volume;
    while ((volume = static_cast<TGeoVolume*>(next()))) {
        if (IsSensitive(volume->GetName())) {
            LOG(debug) << "Sensitive Volume " << volume->GetName();
            AddSensitiveVolume(volume);
        }
    }
}

// -------------------------------------------------------------------------

void FairDetector::Initialize()
{
    // Registers hits collection in Root manager;
    // sets sensitive volumes.
    // ---

    // Define sensitive volumes if in MT
    if (gMC->IsMT()) {
        std::cout << "Define sensitive volume " << std::endl;
        DefineSensitiveVolumes();
    }

    GetRunSim().UpdateSensitiveVolumesForModule(*this);

    // Initialize cached pointer to MC (on master in sequential mode)
    fMC = TVirtualMC::GetMC();
}

void FairDetector::SaveGeoParams()
{
    if (!fGeoSaved) {
        LOG(info) << "Detector: " << GetName() << " Geometry parameters saved ... ";
        TFolder* mf = dynamic_cast<TFolder*>(gROOT->FindObjectAny(FairRootManager::GetFolderName()));
        TFolder* stsf = nullptr;
        if (mf) {
            stsf = dynamic_cast<TFolder*>(mf->FindObjectAny(GetName()));
        }
        if (stsf) {
            TFolder* newf = stsf->AddFolder("Parameters", "Detector parameters", nullptr);
            newf->Add(flGeoPar);
        }
        fGeoSaved = kTRUE;
    }
}