Skip to content
Snippets Groups Projects
Commit 1b1c10ab authored by Riccardo Maria Bianchi's avatar Riccardo Maria Bianchi :sunny: Committed by Melissa Yexley
Browse files

Add the possibility to filter over DetManagers

parent f56c65d5
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
DetectorDescription/GeoModel/GeoModelStandalone/GeoExporter/src/GeoExporter.cxx
+ 120
− 40
View file @ 1b1c10ab
......@@ -27,7 +27,8 @@
#include "GeoModelDBManager/GMDBManager.h"
#include "GeoModelWrite/WriteGeoModel.h" //DumpGeoModelActionLocal.h"
#include "GeoModelWrite/WriteGeoModel.h"
// #include "GeoModelHelpers/defineWorld.h" //TODO: Use this as soon as we have the latest GeoModel in Athena main
#include <QCoreApplication>
#include <QtCore/QStringList>
......@@ -43,6 +44,45 @@
#include <boost/range/irange.hpp>
// Units
#include "GeoModelKernel/Units.h"
#define UNITS \
GeoModelKernelUnits // so we will get, e.g., 'GeoModelKernelUnits::cm'
//TODO: replace this with GeoModelHelpers/defineWorld.h
//_____________________________________________________________________________________
GeoPhysVol* createTheWorld()
{
// Define the units
#define gr UNITS::gram
#define mole UNITS::mole
#define cm3 UNITS::cm3
// Define the chemical elements
GeoElement* Nitrogen = new GeoElement ("Nitrogen" ,"N" , 7.0 , 14.0067 *gr/mole);
GeoElement* Oxygen = new GeoElement ("Oxygen" ,"O" , 8.0 , 15.9995 *gr/mole);
GeoElement* Argon = new GeoElement ("Argon" ,"Ar" , 18.0 , 39.948 *gr/mole);
GeoElement* Hydrogen = new GeoElement ("Hydrogen" ,"H" , 1.0 , 1.00797 *gr/mole);
// Define the materials
double densityOfAir=0.001214 *gr/cm3;
GeoMaterial *air = new GeoMaterial("Air", densityOfAir);
air->add(Nitrogen , 0.7494);
air->add(Oxygen, 0.2369);
air->add(Argon, 0.0129);
air->add(Hydrogen, 0.0008);
air->lock();
//-----------------------------------------------------------------------------------//
// create the world volume container and
// get the 'world' volume, i.e. the root volume of the GeoModel tree
std::cout << "Creating the 'world' volume, i.e. the root volume of the GeoModel tree..." << std::endl;
const GeoBox* worldBox = new GeoBox(1000*UNITS::cm, 1000*UNITS::cm, 1000*UNITS::cm);
const GeoLogVol* worldLog = new GeoLogVol("WorldLog", worldBox, air);
GeoPhysVol* world = new GeoPhysVol(worldLog);
return world;
}
//____________________________________________________________________
......@@ -135,45 +175,64 @@ void GeoExporter::init()
else {
VP1Msg::message("User's settings - DetDescrTag: " + user_detdescrtag);
}
// -- get 'force' option
bool forceOverride = environment.value("DUMPGEOFORCEOVERRIDE");
// -- get 'forceOverwrite' option
bool user_forceOverwrite = ( environment.value("DUMPGEOFORCEOVERWRITE")=="1" ? true : false );
VP1Msg::message("User's settings - forceOverwrite option: " + QString::number(user_forceOverwrite) );
// -- get sub-systems settings
bool user_noid = environment.value("DUMPGEO_NOID").toInt();
bool user_nocalo = environment.value("DUMPGEO_NOCALO").toInt();
bool user_nomuon = environment.value("DUMPGEO_NOMUON").toInt();
QString user_subsystems_filters = "" + QString(((user_noid) ? "-noID" : "")) + QString(((user_nocalo) ? "-noCalo" : "")) + QString(((user_nomuon) ? "-noMuon" : ""));
// -- get GeoModel Treetop filter // FIXME: check and update this!!
QString user_treetopfilter = environment.value("DUMPGEOTREETOPFILTER");
VP1Msg::message("User's settings - GeoModel Treetop filter: " + user_treetopfilter);
// -- get Overwrite flag
auto getOWFlag = [&environment]() -> bool { bool okConv; int flag = environment.value("DUMPGEOOVERWRITE").toInt(&okConv, 10); return (okConv && 1==flag) ? true : false; }; // convert the "0"/"1" env var string to bool with a lambda
bool user_overwrite = getOWFlag();
VP1Msg::message("User's settings - Overwrite filter: " + QString::number(user_overwrite) );
// Get list of Treetop names from the TREETOPFILTER
QString user_filterTreeTops = environment.value("DUMPGEOFILTERTREETOPS");
VP1Msg::message("User's settings - GeoModel TreeTops filter: " + user_filterTreeTops);
// -- get GeoModel Detector Managers filter // FIXME: check and update this!!
QString user_filterDetManagers = environment.value("DUMPGEOFILTERDETMANAGERS");
VP1Msg::message("User's settings - GeoModel DetectorManagers filter: " + user_filterDetManagers);
// // -- get Overwrite flag
// auto getOWFlag = [&environment]() -> bool { bool okConv; int flag = environment.value("DUMPGEOFORCEOVERWRITE").toInt(&okConv, 10); return (okConv && 1==flag) ? true : false; }; // convert the "0"/"1" env var string to bool with a lambda
// bool user_overwrite = getOWFlag();
// VP1Msg::message("User's settings - Overwrite filter: " + QString::number(user_overwrite) );
// Get list of TreeTops from the TREETOPFILTER
QStringList user_treetopslist;
if ( ! user_treetopfilter.isEmpty() )
user_treetopslist = user_treetopfilter.split(',');
if ( ! user_filterTreeTops.isEmpty() ) {
user_treetopslist = user_filterTreeTops.split(',');
}
// Get list of DetectorManagers from the TREETOPFILTER
QStringList user_detmanagerslist;
if ( ! user_filterDetManagers.isEmpty() ) {
user_detmanagerslist = user_filterDetManagers.split(',');
}
GeoPhysVol* volTop = createTheWorld();
/*
// Debug: get list of managers
std::cout << "List of GeoModel managers: " << std::endl;
if ( !(user_detmanagerslist.empty()) ) {
// Get list of managers
std::cout << "\nList of GeoModel managers: " << std::endl;
std::vector<std::string> managersList = theExpt->getListOfManagers();
for (auto const& mm : managersList)
if ( !(managersList.empty()) ) {
for (auto const& mm : managersList)
{
const GeoVDetectorManager* manager = theExpt->getManager(mm);
std::string detManName = manager->getName();
std::cout << "\n\t DetectorManager: " << detManName << std::endl;
unsigned int nTreetops = manager->getNumTreeTops();
std::cout << mm << " - n.Treetops: " << nTreetops << std::endl;
std::cout << mm << "\t - n.Treetops: " << nTreetops << std::endl;
if ( nTreetops > 0 && user_treetopslist.contains(QString::fromStdString(mm)) ) {
if ( nTreetops > 0 && user_detmanagerslist.contains(QString::fromStdString(detManName)) ) {
for(unsigned int i=0; i < nTreetops; ++i) {
PVConstLink treetop(manager->getTreeTop(i));
// mainTopPhysVol->add( const_cast<GeoVPhysVol*>(&(*treetop)));
const GeoVPhysVol* vol = &(*treetop);
volTop->add(const_cast<GeoVPhysVol*>(vol));
std::string volName = vol->getLogVol()->getName();
std::cout << "\t\t treetop: " << volName << std::endl;
// DEBUG: dive into the Treetop
if (mm=="BeamPipe") {
if ("BeamPipe"==detManName) {
GeoVolumeCursor av(treetop);
while (!av.atEnd()) {
std::cout << i << " - name: " << av.getName() << "\n";
......@@ -183,34 +242,50 @@ void GeoExporter::init()
} // end for
} // end if
} // end for
*/
}
}
// if ( !(user_treetopslist.empty()) ) {
std::cout << "\nLooping over top volumes in the GeoModel tree (children of the 'World' volume)..." << std::endl;
GeoVolumeCursor av(world);
while (!av.atEnd()) {
std::string volname = av.getName();
std::cout << "\t* volume name:" << volname << std::endl ;
av.next(); // increment volume cursor.
}
// }
std::cout << "Creating the SQLite DB file..." << std::endl;
QString path = "";
if (user_detdescrtag.isEmpty())
path = "geometry.db";
else {
//qDebug() << "user_treetopslist:" << user_treetopslist;
if ( user_treetopslist.isEmpty() )
path = "geometry-" + user_detdescrtag + user_subsystems_filters + ".db";
else
path = "geometry-" + user_detdescrtag + "-" + user_treetopslist.join("-") + ".db"; // FIXME: check and update!
QString fileName = "geometry";
if ( !(user_detdescrtag.isEmpty()) ) {
fileName = "geometry-" + user_detdescrtag;
}
if ( !(user_treetopslist.isEmpty()) ) {
fileName = fileName + "-" + user_treetopslist.join("-");
}
if ( !(user_detmanagerslist.isEmpty()) ) {
fileName = fileName + "-" + user_detmanagerslist.join("-");
}
if ( !(user_subsystems_filters.isEmpty()) ) {
fileName = fileName + "-" + user_subsystems_filters + ".db";
}
fileName = fileName + ".db";
// check if path exists and if yes: Is it a file and no directory?
bool fileExists = QFileInfo::exists(path) && QFileInfo(path).isFile();
// check if fileName exists and if yes: Is it a file and no directory?
bool fileExists = QFileInfo::exists(fileName) && QFileInfo(fileName).isFile();
if (fileExists) {
if (user_overwrite) {
VP1Msg::message("Removing the existing dump file ("+path+")...");
QFile file (path);
if (user_forceOverwrite) {
VP1Msg::message("Removing the existing dump file ("+fileName+")...");
QFile file (fileName);
file.remove();
} else if ( !user_overwrite ) {
VP1Msg::messageWarningAllRed("The output file ("+path+") is already present in the current folder, but you don't use the '-f' flag to overwrite it. The program will be stopped. Please remove or move the existing file to another folder, or use the '-f' flag to replace it.");
} else if ( !user_forceOverwrite ) {
VP1Msg::messageWarningAllRed("The output file ("+fileName+") is already present in the current folder, but you don't use the '-f' flag to overwrite it. The program will be stopped. Please remove or move the existing file to another folder, or use the '-f' flag to replace it.");
throw "existing output file";
}
}
// open the DB connection
GMDBManager db(path.toStdString());
GMDBManager db(fileName.toStdString());
// check the DB connection
if (db.checkIsDBOpen())
......@@ -223,7 +298,12 @@ void GeoExporter::init()
std::cout << "Dumping the GeoModel geometry to the DB file..." << std::endl;
// Dump the tree volumes into a DB
GeoModelIO::WriteGeoModel dumpGeoModelGraph(db); // init the GeoModel node action
world->exec(&dumpGeoModelGraph); // visit all GeoModel nodes
// visit all GeoModel nodes
if (!(user_detmanagerslist.empty()) || !(user_treetopslist.empty())) {
volTop->exec(&dumpGeoModelGraph);
} else {
world->exec(&dumpGeoModelGraph);
}
std::cout << "Saving the GeoModel tree to the DB." << std::endl;
dumpGeoModelGraph.saveToDB(); // save to the SQlite DB file
std::cout << "DONE. Geometry saved." <<std::endl;
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment