create_rootgeom_SiBT_Run8.C

Go to the documentation of this file.

/*
 * Use this macro to create the ROOT geometry of Silicon Beam Tracker (SiBT)
 * for the RUN-8 configuration (3 stations)
 *
 * Author: Baranov D.
 * Created": 14.11.2021
 *
 * WARNING: all units is in cm!!!
 *
 */
 
#include "TGeoMatrix.h"
#include "TGeoVolume.h"
#include "TString.h"
 
// Set Parameters of SiBT --------------------------------------------------------
const Int_t NStations = 3;     // stations in the detector
const Int_t NMaxModules = 1;   // max. number of modules in a station
 
const Double_t XStationPositions[NStations] = {0.115 /*0*/, -0.295 /*1*/, 0.177 /*2*/};     // geometry center
const Double_t YStationPositions[NStations] = {+0.089 /*0*/, -0.046 /*1*/, -0.012 /*2*/};   // geometry center
const Double_t ZStationPositions[NStations] = {-283.427 /*0*/, -183.225 /*1*/,
                                               -84.378 /*2*/};   // side which is the nearest to the beam
// where -283.427 = -283.097 - 0.33; -183.225 = -183.115 - 0.11; -84.378 = -84.068 - 0.31; (S.Pyaydin's scheme)
 
//(X-Y-Z)Shifts of modules in each station
const Double_t XModuleShifts[NStations][NMaxModules] = {
    {0.0},   // station 0
    {0.0},   // station 1
    {0.0}    // station 2
};
const Double_t YModuleShifts[NStations][NMaxModules] = {
    {0.0},   // station 0
    {0.0},   // station 1
    {0.0}    // station 2
};
const Double_t ZModuleShifts[NStations][NMaxModules] = {
    {0.0},   // station 0
    {0.0},   // station 1
    {0.0}    // station 2
};
 
// Rotations of stations around Z (deg.)
const Double_t ZStationRotations[NStations][NMaxModules] = {
    {-0.028},    // station 0
    {-29.817},   // station 1
    {-59.949}    // station 2
};
 
// Sizes of elements (cm) -------------------------------------------------------
 
// Sensor sizes
const Double_t XSensSize = 6.08;    // cm
const Double_t YSensSize = 6.08;    // cm
const Double_t ZSensSize = 0.018;   // cm
//------------------------------------------------------------------------------
 
// GeoManager
TGeoManager* gGeoMan = NULL;
 
// media
TGeoMedium* pMedAir = 0;
TGeoMedium* pMedCarbon = 0;
TGeoMedium* pMedFiberGlass = 0;
TGeoMedium* pMedSilicon = 0;
 
class FairGeoMedia;
class FairGeoBuilder;
 
TGeoVolume* CreateStation(TString station_name);
TGeoVolume* CreateModule(TString module_name);
 
void DefineRequiredMedia(FairGeoMedia* geoMedia, FairGeoBuilder* geoBuild)
{
 
    // air medium
    FairGeoMedium* mAir = geoMedia->getMedium("air");
    if (!mAir)
        Fatal("Main", "FairMedium air not found");
    geoBuild->createMedium(mAir);
    pMedAir = gGeoManager->GetMedium("air");
    if (!pMedAir)
        Fatal("Main", "Medium air not found");
 
    // carbon medium
    FairGeoMedium* mCarbon = geoMedia->getMedium("carbon");
    if (!mCarbon)
        Fatal("Main", "FairMedium carbon not found");
    geoBuild->createMedium(mCarbon);
    pMedCarbon = gGeoManager->GetMedium("carbon");
    if (!pMedCarbon)
        Fatal("Main", "Medium carbon not found");
 
    // fiberglass medium
    FairGeoMedium* mFiberGlass = geoMedia->getMedium("fiberglass");
    if (!mFiberGlass)
        Fatal("Main", "FairMedium fiberglass not found");
    geoBuild->createMedium(mFiberGlass);
    pMedFiberGlass = gGeoManager->GetMedium("fiberglass");
    if (!pMedFiberGlass)
        Fatal("Main", "Medium fiberglass not found");
 
    // silicon medium
    FairGeoMedium* mSilicon = geoMedia->getMedium("silicon");
    if (!mSilicon)
        Fatal("Main", "FairMedium silicon not found");
    geoBuild->createMedium(mSilicon);
    pMedSilicon = gGeoManager->GetMedium("silicon");
    if (!pMedSilicon)
        Fatal("Main", "Medium silicon not found");
}
 
void create_rootgeom_SiBT_Run8()
{
 
    // ----  set working directory  --------------------------------------------
    TString gPath = gSystem->Getenv("VMCWORKDIR");
 
    // -------   Geometry file name (output)   ----------------------------------
    const TString geoDetectorName = "SiBT";
    const TString geoDetectorVersion = "Run8";
    const TString geoFileName = gPath + "/geometry/" + geoDetectorName + "_" + geoDetectorVersion + ".root";
 
    // ----  global geometry parameters  ---------------------------------------
    FairGeoLoader* geoLoad = new FairGeoLoader("TGeo", "FairGeoLoader");
    FairGeoInterface* geoFace = geoLoad->getGeoInterface();
 
    // -------   Load media from media file   ----------------------------------
    TString medFile = gPath + "/geometry/media.geo";
    geoFace->setMediaFile(medFile);
    geoFace->readMedia();
    FairGeoMedia* geoMedia = geoFace->getMedia();
    FairGeoBuilder* geoBuild = geoLoad->getGeoBuilder();
 
    DefineRequiredMedia(geoMedia, geoBuild);
    // -------------------------------------------------------------------------
 
    // --------------   Create geometry and top volume  ------------------------
    gGeoMan = (TGeoManager*)gROOT->FindObject("FAIRGeom");
    gGeoMan->SetName(geoDetectorName + "_geom");
    TGeoVolume* top = new TGeoVolumeAssembly("TOP");
    top->SetMedium(pMedAir);
    gGeoMan->SetTopVolume(top);
    // gGeoMan->SetTopVisible(1);
    //  -------------------------------------------------------------------------
 
    // Define TOP Geometry
    TGeoVolume* SiBT = new TGeoVolumeAssembly(geoDetectorName);
    SiBT->SetMedium(pMedAir);
 
    for (Int_t istation = 0; istation < NStations; ++istation) {
        // for(Int_t istation = 0; istation < 3; ++istation) {
 
        Int_t stationNum = istation;   // station number
 
        TGeoVolume* station = CreateStation(TString("station") + TString::Itoa(stationNum, 10));
 
        TGeoVolume* module0 = CreateModule(TString("module0_") + station->GetName());
 
        TGeoCombiTrans* module0_transform = new TGeoCombiTrans();
        module0_transform->RotateZ(ZStationRotations[stationNum][0]);
        module0_transform->SetTranslation(XModuleShifts[stationNum][0], YModuleShifts[stationNum][0],
                                          ZModuleShifts[stationNum][0] + 0.5 * ZSensSize);
 
        TGeoCombiTrans* station_transform = new TGeoCombiTrans();
        station_transform->SetTranslation(XStationPositions[stationNum], YStationPositions[stationNum],
                                          ZStationPositions[stationNum]);
 
        station->AddNode(module0, 0, new TGeoCombiTrans(*module0_transform));
 
        SiBT->AddNode(station, 0, station_transform);
    }
 
    top->AddNode(SiBT, 0);
    top->SetVisContainers(kTRUE);
 
    // ---------------   Finish   -----------------------------------------------
    gGeoMan->CloseGeometry();
    gGeoMan->CheckOverlaps(0.0001);
    gGeoMan->PrintOverlaps();
    gGeoMan->Test();
    // gGeoMan->RestoreMasterVolume();
 
    TFile* geoFile = new TFile(geoFileName, "RECREATE");
    top->Write();
    geoFile->Close();
    top->Draw("ogl");
}
 
TGeoVolume* CreateStation(TString station_name)
{
 
    TGeoVolume* stationA = new TGeoVolumeAssembly(station_name);
    stationA->SetMedium(pMedAir);
 
    return stationA;
}
 
TGeoVolume* CreateModule(TString module_name)
{
 
    // shapes
    TGeoShape* moduleS = new TGeoBBox(TString("moduleS") += TString("_") + module_name, XSensSize * 0.5,
                                      YSensSize * 0.5, ZSensSize * 0.5);
 
    // volumes
    TGeoVolume* moduleV = new TGeoVolume(TString("SiBTSensor") += TString("_") + module_name, moduleS);
 
    // medium
    if (pMedSilicon) {
        moduleV->SetMedium(pMedSilicon);
    } else
        Fatal("Main", "Invalid medium for moduleV volumes!");
 
    // volume visual property
    moduleV->SetLineColor(TColor::GetColor("#ff47ca"));
    moduleV->SetTransparency(30);
 
    return moduleV;
}
//------------------------------------------------------------------------------