new command line tool called gmcat

GMCAT/CMakeLists.txt

+
+set(MYLIB_VERSION_MAJOR 1)
+set(MYLIB_VERSION_MINOR 0)
+set(MYLIB_VERSION_PATCH 0)
+
+set( CMAKE_BUILD_TYPE DEBUG )
+set(CMAKE_CXX_FLAGS "-fPIC -O0 -g -gdwarf-2" )
+
+project ( "gmcat" VERSION ${MYLIB_VERSION_MAJOR}.${MYLIB_VERSION_MINOR}.${MYLIB_VERSION_PATCH} LANGUAGES CXX )
+
+find_package( GeoModelCore REQUIRED )
+find_package( GeoModelIO REQUIRED )
+find_package( Eigen3 REQUIRED )
+file( GLOB SOURCES src/*.cxx )
+
+include_directories ("${PROJECT_SOURCE_DIR}")
+include_directories ("${PROJECT_SOURCE_DIR}/src")
+add_executable ( gmcat ${SOURCES} ${HEADERS}  )
+
+# External dependencies:
+
+target_link_libraries (gmcat GeoModelCore::GeoModelKernel GeoModelIO::GeoModelRead GeoModelIO::GeoModelWrite )
+install(TARGETS gmcat DESTINATION bin)
+
+
+
+

GMCAT/src/gmcat.cxx

+#include "GeoModelKernel/GeoVGeometryPlugin.h"
+#include "GeoModelDBManager/GMDBManager.h"
+#include "GeoModelRead/ReadGeoModel.h"
+#include "GeoModelWrite/WriteGeoModel.h"
+
+#include "GeoModelKernel/GeoGeometryPluginLoader.h"
+#include "GeoModelKernel/GeoVolumeCursor.h"
+#include "GeoModelKernel/GeoMaterial.h"
+#include "GeoModelKernel/GeoElement.h"
+#include "GeoModelKernel/GeoPVLink.h"
+
+
+#include "GeoModelKernel/GeoBox.h"
+#include "GeoModelKernel/GeoCountVolAction.h"
+#include "GeoModelKernel/GeoAccessVolumeAction.h"
+#include "GeoModelKernel/GeoNameTag.h"
+
+#include <iostream>
+#include <string>
+#include <vector>
+#ifdef __APPLE__
+const std::string shared_obj_extension=".dylib";
+#else
+const std::string shared_obj_extension=".so";
+#endif
+
+#define SYSTEM_OF_UNITS GeoModelKernelUnits // --> 'GeoModelKernelUnits::cm'
+
+int main(int argc, char ** argv) {
+  //
+  // Usage message:
+  //
+  
+  std::string gmcat= argv[0];
+  std::string usage= "usage: " + gmcat + " [plugin1"+shared_obj_extension
+    + "] [plugin2" + shared_obj_extension
+    + "] ...[file1.db] [file2.db].. -o outputFile]"; 
+  //
+  // Print usage message if no args given:
+  // 
+  if (argc==1) {
+    std::cerr << usage << std::endl;
+    return 0;
+  }
+  //
+  // Parse the command line:
+  // 
+  std::vector<std::string> inputFiles;
+  std::vector<std::string> inputPlugins;
+  std::string outputFile;
+  for (int argi=1;argi<argc;argi++) {
+    std::string argument=argv[argi];
+    if (argument.find("-o")!=std::string::npos) {
+      argi++;
+      if (argi>=argc) {
+	std::cerr << usage << std::endl;
+	return 1;
+      }
+      outputFile=argv[argi];
+    }
+    else if (argument.find(shared_obj_extension)!=std::string::npos) {
+      inputPlugins.push_back(argument);
+    }
+    else if (argument.find(".db")!=std::string::npos) {
+      inputFiles.push_back(argument);
+    }
+    else {
+      std::cerr << "Unrecognized argument " << argument << std::endl;
+      std::cerr << usage << std::endl;
+      return 2;
+    }
+  }
+
+  // if (((inputPlugins.size()+inputFiles.size())<1) || outputFile=="") {
+  //   std::cout << usage << std::endl;
+  //   return 3;
+  // }
+  //
+  // Create a huge world volume made of Air:
+  //
+  const double  gr =   SYSTEM_OF_UNITS::gram;
+  const double  mole = SYSTEM_OF_UNITS::mole;
+  const double  cm3 =  SYSTEM_OF_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);
+  
+  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();
+  
+  const GeoBox* worldBox = new GeoBox(2000*SYSTEM_OF_UNITS::cm, 2000*SYSTEM_OF_UNITS::cm, 2000*SYSTEM_OF_UNITS::cm);
+  const GeoLogVol* worldLog = new GeoLogVol("WorldLog", worldBox, air);
+  GeoPhysVol *world=new GeoPhysVol(worldLog);
+  world->ref();
+  //
+  // Loop over plugins, create the geometry and put it under the world:
+  //
+  for (const std::string & plugin : inputPlugins) {
+    GeoGeometryPluginLoader loader;
+    GeoVGeometryPlugin *factory=loader.load(plugin);
+    if (!factory) {
+      std::cerr << "Could not load plugin " << plugin << std::endl;
+      return 5;
+    }
+    factory->create(world);
+  }
+  
+  //
+  // Loop over files, create the geometry and put it under the world:
+  //
+  for (const std::string & file : inputFiles) {
+    GMDBManager* db = new GMDBManager(file.c_str());
+    if (!db->isOpen()){
+      std::cerr << "Error opening input file " << file << std::endl;
+      return 6;
+    }
+      
+    /* set the GeoModel reader */
+    GeoModelIO::ReadGeoModel readInGeo = GeoModelIO::ReadGeoModel(db);
+      
+    /* build the GeoModel geometry */
+    GeoPhysVol* dbPhys = readInGeo.buildGeoModel(); // builds the whole GeoModel tree in memory
+
+    //world->add(dbPhys);
+    GeoVolumeCursor aV(dbPhys);
+	
+    while (!aV.atEnd()) {
+      GeoNameTag *nameTag=new GeoNameTag(aV.getName());
+      GeoTransform *transform= new GeoTransform(aV.getTransform());
+      world->add(nameTag);
+      world->add(transform);
+      world->add((GeoVPhysVol *) &*aV.getVolume());
+      aV.next();
+    }
+    delete db;
+  }
+
+  //
+  // Open a new database:
+  //
+  GMDBManager db(outputFile.c_str());
+  GeoModelIO::WriteGeoModel dumpGeoModelGraph(db);
+   
+  // check the DB connection
+  if (!db.isOpen()) {
+    std::cerr << "Error opening output file " << outputFile << std::endl;
+    return 4;
+  }
+
+  world->exec(&dumpGeoModelGraph);
+  dumpGeoModelGraph.saveToDB();
+   
+  world->unref();
+  
+  
+  return 0;
+}
+