/home/bes3soft/bes3soft/Boss/7.0.2/dist/7.0.2/DetectorDescription/ROOTGeo/ROOTGeo-00-00-15/src/MucROOTGeo.cxx

Go to the documentation of this file.

//$id$
/*
 *    2004/11/29   Zhengyun You      Peking University
 *                 Muc Geometry General for EventDisplay
 *
 *    2004/12/11   Zhengyun You      Peking University
 *                 named from MucGeo to MucROOTGeo
 *                 inherit from class SubDetectorROOTGeo
 */

using namespace std;

#include <string>
#include <fstream>
#include <iostream>
#include <sstream>
#include <vector>
#include <iomanip>

#include <TGeoManager.h>

#include "ROOTGeo/MucROOTGeo.h"

const int MucROOTGeo::m_kSeg[m_kPart]  = {4, 8, 4};
const int MucROOTGeo::m_kAbsorber[m_kPart] = {9, 9, 9};
const int MucROOTGeo::m_kGap[m_kPart]      = {8, 9, 8};
const int MucROOTGeo::m_kPanel[m_kPart]    = {4, 3, 4};

MucROOTGeo::MucROOTGeo()
{ 
  // Default constructor.
  for (int part = 0; part < m_kPart; part++) {
    for (int seg = 0; seg < m_kSegMax; seg++) {
      for (int gap = 0; gap < m_kGapMax; gap++) {
        m_NodeGap[part][seg][gap] = 0;
        m_PhysicalGap[part][seg][gap] = 0;
        //      m_NodeBox[part][seg][gap] = 0;
        //      m_PhysicalBox[part][seg][gap] = 0;
        for (int strip = 0; strip < m_kStripMax; strip++) {
          m_NodeStrip[part][seg][gap][strip] = 0;
          m_PhysicalStrip[part][seg][gap][strip] = 0;
        }
      }
    }
  }

  string GdmlManagementPath = getenv("GDMLMANAGEMENTROOT");
  //cout << "MucGeoGeneral::InitFromXML(), GdmlManagementPath not found" << endl;
  string GdmlFile = GdmlManagementPath + string("/dat/Muc.gdml");
  //string GdmlFile = string("Muc.gdml");
  cout << "GdmlFile " << GdmlFile << endl;
  InitFromGdml( GdmlFile.c_str(), "Muc" );
}

MucROOTGeo::~MucROOTGeo()
{ }

void
MucROOTGeo::InitFromGdml( const char *gdmlFile, const char *setupName )
{
  ReadGdml(gdmlFile, setupName);
  SetNode();
  //SetDefaultVis();

  m_ROOTGeoInit = 1;  
}

void 
MucROOTGeo::SetNode()
{
  m_Muc = GetTopVolume();
  if(!m_Muc) {
    m_Muc = GetLogicalVolume("logicalMuc");
  }
  if(!m_Muc) std::cout << "m_Muc = 0" << std::endl;

  int gapCount = 0;
  for (int part = 0; part < m_kPart; part++) {
    for (int seg = 0; seg < m_kSeg[part]; seg++) {  
      for (int gap = 0; gap < m_kGap[part]; gap++) {
        m_StripNum[part][seg][gap] = GetVolumeStripPlane(part, seg, gap)->GetNdaughters();
        //std::cout << part << " " << seg << " " << gap << " " << m_StripNum[part][seg][gap] <<" "<<GetVolumeStripPlane(part, seg, gap)->GetName()<< std::endl;

        std::stringstream osnameGap;
        osnameGap << "pv_" << "l" << "Muc" << "P" << part << "S" << seg << "G" << gap << "_" << gapCount; 
        m_NodeGap[part][seg][gap] = GetNode( osnameGap.str() );
        gapCount++;

        //cout<<"in MucROOTGeo "<<m_NodeGap[part][seg][gap]->GetName()<<" "<<m_NodeGap[part][seg][gap]->GetDaughter(0)->GetName()<<endl;

        std::stringstream osnameBox;
        osnameBox << "pv_" << "l" << "Muc" << "P" << part << "S" << seg << "G" << gap << "Al" << "_" << "0"; 
        //      m_NodeAluminumBox[part][seg][gap] = GetNode( osnameBox.str() );
        TGeoNode *temp = GetNode( osnameBox.str() );

        int segment = 0; if(part==1&&seg==2) segment = 2;
        std::stringstream osnameStripPlane;
        osnameStripPlane << "pv_" << "l" << "Muc" << "P" << part << "S" << segment << "G" << gap << "SP" << "_" << "0";
        m_NodeStripPlane[part][seg][gap] = GetNode( osnameStripPlane.str() );
        
        for (int strip = 0; strip < m_StripNum[part][seg][gap]; strip++) {
          std::stringstream osnameStrip;
          if(strip < 10) {
            osnameStrip << "pv_" << "l" << "Muc" << "P" << part << "S" << segment << "G" << gap << "s" << "00" << strip << "_" << strip; 
          }
          else if(strip < 100) {
            osnameStrip << "pv_" << "l" << "Muc" << "P" << part << "S" << segment << "G" << gap << "s" << "0" << strip << "_" << strip; 
          }
          else {
            osnameStrip << "pv_" << "l" << "Muc" << "P" << part << "S" << segment << "G" << gap << "s" << strip << "_" << strip; 
          }
          
          m_NodeStrip[part][seg][gap][strip] = GetNode( osnameStrip.str() );
        }
      }
    }
  }
}

void 
MucROOTGeo::SetDefaultVis()
{
  //std::cout << "begin of set defaultvis" << std::endl;
  int mucColor        = 2;
  int absorberColor   = 2;
  int gapColor        = 8;
  int gasChamberColor = 8;
  int bakeliteColor   = 8;
  int stripColor      = 6;

  m_Muc->SetLineColor(mucColor);
  m_Muc->SetVisibility(0);
  
  for (int part = 0; part < m_kPart; part++) {
    for (int seg = 0; seg < m_kSeg[part]; seg++) {
      for (int absorber = 0; absorber < m_kAbsorber[part]; absorber++) {
        GetVolumeAbsorber(part, seg, absorber)->SetLineColor(absorberColor);
        GetVolumeAbsorber(part, seg, absorber)->SetVisibility(0);
        if (part != 1) {
          for (int panel = 0; panel < m_kPanel[part]; panel++) {
            GetVolumeAbsorberPanel(part, seg, absorber, panel)->SetLineColor(absorberColor);
            GetVolumeAbsorberPanel(part, seg, absorber, panel)->SetVisibility(0);
          }
        }
      }
      
      for (int gap = 0; gap < m_kGap[part]; gap++) {
        GetVolumeGap(part, seg, gap)->SetLineColor(gapColor);
        GetVolumeGap(part, seg, gap)->SetVisibility(0);
        GetVolumeStripPlane(part, seg, gap)->SetLineColor(gapColor);
        GetVolumeStripPlane(part, seg, gap)->SetVisibility(0);
 
        for (int strip = 0; strip < m_StripNum[part][seg][gap]; strip++) {
          GetVolumeStrip(part, seg, gap, strip)->SetLineColor(stripColor);
          GetVolumeStrip(part, seg, gap, strip)->SetVisibility(1);
        }
        
        for (int panel = 0; panel < m_kPanel[part]; panel++) {
          for(int bakelite = 0; bakelite < m_kBakelite; bakelite++) {
            GetVolumeBakelite(part, seg, gap, panel, bakelite)->SetLineColor(bakeliteColor);
            GetVolumeBakelite(part, seg, gap, panel, bakelite)->SetVisibility(0);
          }
          for (int gasChamber = 0; gasChamber < m_kGasChamber; gasChamber++) {
            GetVolumeGasChamber(part, seg, gap, panel, gasChamber)->SetLineColor(gasChamberColor);
            GetVolumeGasChamber(part, seg, gap, panel, gasChamber)->SetVisibility(0);
          }
        }
      }
    }
  }

  for (int part = 0; part < m_kPart; part++) {
    for (int seg = 0; seg < m_kSeg[part]; seg++) {
      for (int gap = 0; gap < m_kGap[part]; gap++) {
        GetGap(part, seg, gap)->SetVisibility(0);
        for (int strip = 0; strip < m_StripNum[part][seg][gap]; strip++) {
          GetStrip(part, seg, gap, strip)->SetVisibility(0);
        }
      }
    }
  }

  //std::cout << "end of set defaultvis" << std::endl;
}

void
MucROOTGeo::SetAllVisible()
{
  for (int part = 0; part < m_kPart; part++) {
    for (int seg = 0; seg < m_kSeg[part]; seg++) {
      for (int absorber = 0; absorber < m_kAbsorber[part]; absorber++) {
        if(part == 1) {
          GetVolumeAbsorber(part, seg, absorber)->SetVisibility(1);
        }
        else {
          GetVolumeAbsorber(part, seg, absorber)->SetVisibility(0);
          for (int panel = 0; panel < m_kPanel[part]; panel++) {
            GetVolumeAbsorberPanel(part, seg, absorber, panel)->SetVisibility(1);
          }
        }
      }
        
      for (int gap = 0; gap < m_kGap[part]; gap++) {
        GetVolumeGap(part, seg, gap)->SetVisibility(0);
        for (int strip = 0; strip < m_StripNum[part][seg][gap]; strip++) {
          GetVolumeStrip(part, seg, gap, strip)->SetVisibility(0);
        }
        
        for (int panel = 0; panel < m_kPanel[part]; panel++) {
          for(int bakelite = 0; bakelite < m_kBakelite; bakelite++) {
            GetVolumeBakelite(part, seg, gap, panel, bakelite)->SetVisibility(1);
          }
          for (int gasChamber = 0; gasChamber < m_kGasChamber; gasChamber++) {
            GetVolumeGasChamber(part, seg, gap, panel, gasChamber)->SetVisibility(1);
          }
        }
      }
    }
  }

  for (int part = 0; part < m_kPart; part++) {
    for (int seg = 0; seg < m_kSeg[part]; seg++) {
      for (int gap = 0; gap < m_kGap[part]; gap++) {
        GetGap(part, seg, gap)->SetVisibility(1);
        for (int strip = 0; strip < m_StripNum[part][seg][gap]; strip++) {
          GetStrip(part, seg, gap, strip)->SetVisibility(0);
        }
      }
    }
  }
}

void
MucROOTGeo::SetQuarterVisible()
{
  for (int part = 0; part < m_kPart; part++) {
    for (int seg = 0; seg < m_kSeg[part]; seg++) {
      for (int absorber = 0; absorber < m_kAbsorber[part]; absorber++) {
        if(part == 1) {
          if(seg > 2) GetVolumeAbsorber(part, seg, absorber)->SetVisibility(1);
        }
        else {
          GetVolumeAbsorber(part, seg, absorber)->SetVisibility(0);
          for (int panel = 0; panel < m_kPanel[part]; panel++) {
            if(seg > 0 && seg < 3) GetVolumeAbsorberPanel(part, seg, absorber, panel)->SetVisibility(1);
          }
        }
      }
      
      for (int gap = 0; gap < m_kGap[part]; gap++) {
        if (part == 1 && seg > 2) {
          GetVolumeGap(part, seg, gap)->SetVisibility(1);
          GetGap(part, seg, gap)->SetVisibility(1);
        }
        else {
          GetVolumeGap(part, seg, gap)->SetVisibility(0);
          GetGap(part, seg, gap)->SetVisibility(0);
        }
        for (int strip = 0; strip < m_StripNum[part][seg][gap]; strip++) {
          GetVolumeStrip(part, seg, gap, strip)->SetVisibility(0);
        }
        
        for (int panel = 0; panel < m_kPanel[part]; panel++) {
          for (int bakelite = 0; bakelite < m_kBakelite; bakelite++) {
            if ((part != 1 && seg > 0 && seg < 3) || (part == 1 && seg > 2)) GetVolumeBakelite(part, seg, gap, panel, bakelite)->SetVisibility(1);
          }
          for (int gasChamber = 0; gasChamber < m_kGasChamber; gasChamber++) {
            if ((part != 1 && seg > 0 && seg < 3) || (part == 1 && seg > 2)) GetVolumeGasChamber(part, seg, gap, panel, gasChamber)->SetVisibility(1);
          }
        }
      }
    }
  }

}

void 
MucROOTGeo::SetPhysicalNode()
{
  int stripColor = 2;

  if (gGeoManager == 0) std::cout << "Create gGeoManager first" << std::endl; 
  TGeoNode *bes = gGeoManager->GetTopNode();
  TGeoNode *muc = bes->GetDaughter(m_childNo);
  
  for (int part = 0; part < m_kPart; part++) {
    for (int seg = 0; seg < m_kSeg[part]; seg++) {
      for (int gap = 0; gap < m_kGap[part]; gap++) {
        TGeoNode *nodeGap = GetGap(part, seg, gap);
        m_PhysicalGap[part][seg][gap] = gGeoManager->MakePhysicalNode( TString("/") + bes->GetName() + 
                                                                       TString("/") + muc->GetName() +
                                                                       TString("/") + nodeGap->GetName() +
                                                                       TString("/") + nodeGap->GetDaughter(0)->GetName());
        m_PhysicalGap[part][seg][gap]->SetVisibility(0);
        m_PhysicalGap[part][seg][gap]->SetIsVolAtt(kFALSE);

        TGeoNode *nodeStripPlane = GetStripPlane(part, seg, gap);
        //std::cout << nodeGap->GetName() << "  " << nodeStripPlane->GetName() << endl;
        
        //TGeoNode *nodeBox = GetAluminumBox(part, seg, gap);
//      m_PhysicalAluminumBox[part][seg][gap] = gGeoManager->MakePhysicalNode( TString("/") + bes->GetName() + 
//                                                                       TString("/") + muc->GetName() +
//                                                                       TString("/") + nodeGap->GetName() +
//                                                                       TString("/") + nodeGap->GetDaughter(0)->GetName() );


        for (int strip = 0; strip < m_StripNum[part][seg][gap]; strip++) {
          TGeoNode *nodeStrip = GetStrip(part, seg, gap, strip);
          m_PhysicalStrip[part][seg][gap][strip] = gGeoManager->MakePhysicalNode( TString("/") + bes->GetName() + 
                                                                                  TString("/") + muc->GetName() +
                                                                                  TString("/") + nodeGap->GetName() +
                                                                                  TString("/") + nodeGap->GetDaughter(0)->GetName() +
                                                                                  TString("/") + nodeStripPlane->GetName() +
                                                                                  TString("/") + nodeStrip->GetName() );
          m_PhysicalStrip[part][seg][gap][strip]->SetVisibility(0);
          m_PhysicalStrip[part][seg][gap][strip]->SetIsVolAtt(kFALSE);
          m_PhysicalStrip[part][seg][gap][strip]->SetLineColor(stripColor);
        }
      }
    }
  }
}

int
MucROOTGeo::GetPartNum()
{
  int part = m_kPart;
  return part;
}

int 
MucROOTGeo::GetSegNum(int part)
{
  int seg = m_kSeg[part];
  return seg;
}
  
int 
MucROOTGeo::GetGapNum(int part)
{
  int gap = m_kGap[part];
  return gap;
}

int 
MucROOTGeo::GetStripNum(int part, int seg, int gap)
{
  int strip = m_StripNum[part][seg][gap];
  return strip;
}

TGeoVolume*
MucROOTGeo::GetVolumeAbsorber( int part, int seg, int absorber )
{
  std::stringstream osname;
  osname << "l" << "Muc" << "P" << part << "S" << seg << "Ab" << absorber;
  
  return GetLogicalVolume( osname.str() );
}
  
float
MucROOTGeo::GetAbsorberThickness( int part, int seg, int absorber )
{
  float thickness = 0.0;
  TGeoVolume *vol = GetVolumeAbsorber(part, seg, absorber);
  if (vol) {
    thickness = 2.0*((TGeoBBox*)vol->GetShape())->GetDZ();
  }
 
  return thickness;
}

TGeoVolume*
MucROOTGeo::GetVolumeAbsorberPanel( int part, int seg, int absorber, int panel )
{
  std::stringstream osname;
  osname << "l" << "Muc" << "P" << part << "S" << seg << "Ab" << absorber << "Pn" << panel;
  
  return GetLogicalVolume( osname.str() );
}
  
TGeoVolume*
MucROOTGeo::GetVolumeGap( int part, int seg, int gap )
{
  std::stringstream osname;
  osname << "l" << "Muc" << "P" << part << "S" << seg << "G" << gap;
  
  return GetLogicalVolume( osname.str() );
}

TGeoVolume*
MucROOTGeo::GetVolumeAluminumBox( int part, int seg, int gap )
{
  std::stringstream osname;
  osname <<"l" << "Muc" << "P" << part << "S" << seg << "G" << gap <<"Al";
  
  return GetLogicalVolume( osname.str() );
}

TGeoVolume*
MucROOTGeo::GetVolumeStripPlane( int part, int seg, int gap )
{
  int segment = 0; if(part==1&&seg==2) segment = 2;
  std::stringstream osname;
  osname << "l" << "Muc" << "P" << part << "S" << segment << "G" << gap << "SP";

  return GetLogicalVolume( osname.str() );
}
  
TGeoVolume*
MucROOTGeo::GetVolumeStrip( int part, int seg, int gap, int strip )
{
  std::stringstream osname;
  if(strip < 10) {
    osname << "l" << "Muc" << "P" << part << "S" << "0" << "G" << gap << "s" << "00" << strip; 
  }
  else if(strip<100){
    osname << "l" << "Muc" << "P" << part << "S" << "0" << "G" << gap << "s" << "0" << strip; 
  }
  else {
    osname << "l" << "Muc" << "P" << part << "S" << "0" << "G" << gap << "s" << strip; 
  }
  
  return GetLogicalVolume( osname.str() );
}
  
TGeoVolume* 
MucROOTGeo::GetVolumeGasChamber( int part, int seg, int gap, int panel, int gasChamber )
{
  std::stringstream osname;
  osname << "l" << "Muc" << "P" << part << "S" << "0" << "G" << gap << "R" << gasChamber << "Pn" << panel << "C";

  return GetLogicalVolume( osname.str() );
}

TGeoVolume* 
MucROOTGeo::GetVolumeBakelite( int part, int seg, int gap, int panel, int bakelite )
{
  std::stringstream osname;
  osname << "l" << "Muc" << "P" << part << "S" << "0" << "G" << gap << "R" << panel << "B" << bakelite;
  
  return GetLogicalVolume( osname.str() );
}

TGeoNode*
MucROOTGeo::GetGap( int part, int seg, int gap )
{
  if (m_NodeGap[part][seg][gap] != 0) {
    return m_NodeGap[part][seg][gap];
  }
  else {
    std::cout << "Node: " << "Part" << part << "Seg" << seg << "Gap" << gap << " not found" << std::endl;
    return 0;
  } 
}

// TGeoNode*
// MucROOTGeo::GetAluminumBox( int part, int seg, int gap )
// {
//   if (m_NodeAluminumBox[part][seg][gap] != 0) {
//     return m_NodeAluminumBox[part][seg][gap];
//   }
//   else {
//     std::cout << "Node: " << "Part" << part << "Seg" << seg << "Gap" << gap << "AluminumBox" << " not found" << std::endl;
//     return 0;
//   } 
// }


TGeoNode*
MucROOTGeo::GetStripPlane( int part, int seg, int gap )
{
  if (m_NodeStripPlane[part][seg][gap] != 0) {
    return m_NodeStripPlane[part][seg][gap];
  }
  else {
    std::cout << "Node: " << "Part" << part << "Seg" << seg << "Gap" << gap << "StripPlane" << " not found" << std::endl;
    return 0;
  }
}
    
TGeoNode*
MucROOTGeo::GetStrip( int part, int seg, int gap, int strip )
{
  if (m_NodeStrip[part][seg][gap][strip] != 0) {
    return m_NodeStrip[part][seg][gap][strip];
  }
  else {
    std::cout << "Node: " << "Part" << part << "Seg" << "0" << "Gap" << gap << "Strip" << strip << " not found" << std::endl;
    return 0;
  } 
}

TGeoPhysicalNode*
MucROOTGeo::GetPhysicalGap( int part, int seg, int gap )
{
  if (m_PhysicalGap[part][seg][gap] != 0) {
    return m_PhysicalGap[part][seg][gap];
  }
  else {
    std::cout << "PhysicalNode: " << "Part" << part << "Seg" << seg << "Gap" << gap << " not found" << std::endl;
    return 0;
  } 
}

// TGeoPhysicalNode*
// MucROOTGeo::GetPhysicalAluminumBox( int part, int seg, int gap )
// {
//   if (m_PhysicalAluminumBox[part][seg][gap] != 0) {
//     return m_PhysicalAluminumBox[part][seg][gap];
//   }
//   else {
//     std::cout << "PhysicalNode: " << "Part" << part << "Seg" << seg << "Gap" << gap << "AluminumBox" << " not found" << std::endl;
//     return 0;
//   } 
// }

TGeoPhysicalNode*
MucROOTGeo::GetPhysicalStrip( int part, int seg, int gap, int strip )
{
  if (m_PhysicalStrip[part][seg][gap][strip] != 0) {
    return m_PhysicalStrip[part][seg][gap][strip];
  }
  else {
    std::cout << "PhysicalNode: " << "Part" << part << "Seg" << seg << "Gap" << gap << "Strip" << strip << " not found" << std::endl;
    return 0;
  } 
}

---