AGDDController.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#include "AGDDControl/AGDDController.h"
 
#include "AGDDKernel/AGDDBuilder.h"
#include "AGDDControl/IAGDDParser.h"
#include "AGDDControl/XercesParser.h"
#include "AGDDControl/AGDD2GeoModelBuilder.h"
#include "AGDDKernel/AGDDVolume.h"
#include "AGDDKernel/AGDDPositioner.h"
#include "AGDDKernel/AliasStore.h"
#include "AGDDModel/AGDDParameterStore.h"
#include "AGDDModel/AGDDMaterialStore.h"
#include "GeoModelInterfaces/IGeoModelSvc.h"
#include "GeoModelUtilities/GeoModelExperiment.h"
#include "GeoModelKernel/GeoVDetectorManager.h"
#include "GeoModelKernel/GeoPhysVol.h"
#include "GeoModelKernel/GeoVPhysVol.h"
#include "StoreGate/StoreGateSvc.h"
 
#include <map>
#include <vector>
#include <string>
#include <fstream>
#include <ctime>
#include <TString.h> // for Form
#include <iostream>
 
AGDDController::~AGDDController()
{
    if (m_theParser) delete m_theParser;
    if (m_theBuilder) delete m_theBuilder;
}
 
AGDDController::AGDDController()
  : m_theBuilder(0),m_locked(false),m_disableSections(false),
    m_printLevel(0)
{
    m_theParser=new XercesParser (m_xs);
    m_theBuilder=new AGDD2GeoModelBuilder (m_ds, m_vs, m_ss, m_as, m_ms);
}
 
void AGDDController::SetBuilder(AGDDBuilder *b) 
{
    m_theBuilder=b;
}
void AGDDController::SetParser(IAGDDParser *b) 
{
    m_theParser=b;
}
AGDDBuilder* AGDDController::GetBuilder() 
{
    return m_theBuilder;
}
IAGDDParser* AGDDController::GetParser() 
{
    return m_theParser;
}
void AGDDController::AddFile(const std::string& fName) 
{
    m_filesToParse.push_back(fName);
}
void AGDDController::AddSection(const std::string& section) 
{
    m_sectionsToBuild.push_back(section);
}
void AGDDController::AddVolume(const std::string& section) 
{
    m_volumesToBuild.push_back(section);
}
 
 
void AGDDController::ParseFiles()
{
        if (!m_theParser) m_theParser=new XercesParser(m_xs);
    for (unsigned int i=0;i<m_filesToParse.size();i++) {
        if (!m_theParser->ParseFileAndNavigate(*this, m_filesToParse[i])) throw std::runtime_error(Form("File: %s, Line: %d\nAGDDController::ParseFiles() - Could parse file %s.", __FILE__, __LINE__, m_filesToParse[i].c_str()));
    }
}
 
void AGDDController::BuildSections()
{
    for (unsigned int i=0;i<m_sectionsToBuild.size();i++)
    {
        if (m_printLevel) std::cout<< " -----> now building section "<<m_sectionsToBuild[i]<<std::endl;
        m_theBuilder->BuildFromSection(m_sectionsToBuild[i]);
    } 
}
 
void AGDDController::BuildVolumes()
{
    for (unsigned int i=0;i<m_volumesToBuild.size();i++)
        m_theBuilder->BuildFromVolume(m_volumesToBuild[i]);
}
 
void AGDDController::BuildAll()
{
        BuildVolumes();
        BuildSections();
}
 
void AGDDController::PrintSections() const
{
    m_ss.PrintAllSections();
}
 
void AGDDController::ParseString(const std::string& s)
{
       if (!m_theParser) m_theParser=new XercesParser(m_xs);
       m_theParser->ParseStringAndNavigate(*this, s);
}
 
bool AGDDController::WriteAGDDtoDBFile(const std::string& s)
{
    if (!m_theParser)
    {
        std::cout<< "  -----> asking for a dump without defined parser makes no sense - crashing!" <<std::endl;
        return false;
    }
    else {
        m_theParser->WriteToFile(s);
        return true;
    }
}
 
void AGDDController::UseGeoModelDetector ATLAS_NOT_THREAD_SAFE (const std::string& name)
{
    StoreGateSvc* pDetStore=0;
    ISvcLocator* svcLocator = Gaudi::svcLocator();
    StatusCode sc=svcLocator->service("DetectorStore",pDetStore);
    if (sc.isFailure()) 
    {
        std::cout<<"AGDDController could not get at the detector store!"<<std::endl;
    }
    const GeoModelExperiment* theExpt = nullptr;
    sc=pDetStore->retrieve( theExpt,"ATLAS");
    if (sc.isFailure()) 
    {
        std::cout<<"AGDDController could not get GeoModelExperiment!"<<std::endl;
    }
    else
    {
        const GeoVDetectorManager *theManager=theExpt->getManager(name);
        if (theManager->getNumTreeTops()>1) std::cout<<"AGDDController: more than one treetop!!!"<<std::endl;
        PVConstLink pv=theManager->getTreeTop(0);
        GeoVPhysVol* ppv=const_cast<GeoVPhysVol*>(&(*pv));
        ((AGDD2GeoModelBuilder*)(m_theBuilder))->SetMotherVolume((GeoPhysVol*)ppv);
    }
 
}
 
#if 0
void AGDDController::PrintVolumeHierarchy(const std::string& name, int ilevel)
{
    AGDDVolume *vol=m_vs.GetVolume(name);
    int currentLevel=ilevel+1;
    for (int i=0;i<ilevel;i++) std::cout<<"    ";
    std::cout<<"|  "<<name<<std::endl;
    if (!vol->NrOfDaughter()) return;
    std::vector<std::string> nameVec;
    std::vector<std::string>::iterator it;
    for (int i=0;i<vol->NrOfDaughter();i++)
    {
        std::string nameV=vol->GetDaughter(i)->GetVolume()->GetName();
        it=find(nameVec.begin(),nameVec.end(),nameV);
        if (it!=nameVec.end()) continue;
        nameVec.push_back(nameV);
        PrintVolumeHierarchy(nameV,currentLevel);
    }
}
#endif
 
void AGDDController::Clean()
{
    delete m_theParser;
    m_theParser=0;
    
    m_filesToParse.clear();
    m_sectionsToBuild.clear();
    m_volumesToBuild.clear();
    m_structuresToBuild.clear();
    
    m_ss.Clean();
    m_vs.Clean();
}
 
XMLHandlerStore& AGDDController::GetHandlerStore()
{
  return m_xs;
}
 
 
AGDDVolumeStore& AGDDController::GetVolumeStore()
{
  return m_vs;
}
 
 
AGDDColorStore& AGDDController::GetColorStore()
{
  return m_cs;
}
 
 
AGDDSectionStore& AGDDController::GetSectionStore()
{
  return m_ss;
}
 
 
AGDDDetectorStore& AGDDController::GetDetectorStore()
{
  return m_ds;
}
 
 
AGDDPositionerStore& AGDDController::GetPositionerStore()
{
  return m_ps;
}
 
 
AGDDMaterialStore& AGDDController::GetMaterialStore()
{
  return m_ms;
}
 
 
AGDDParameterStore& AGDDController::GetParameterStore()
{
  return m_prs;
}
 
 
AliasStore& AGDDController::GetAliasStore()
{
  return m_as;
}
 
 
AGDD::ExpressionEvaluator& AGDDController::Evaluator()
{
  return m_eval;
}