gdml_root.C

// $Id:$
//====================================================================
//  AIDA Detector description implementation for LCD
//--------------------------------------------------------------------
//
//  Standalone ROOT script to load geometries from DD4hep:
//
//  Full GDML cycle can be tested (deplace <detector> with some meaningful identifier):
//   a) Extract gdml information
//      $ > geoConverter -compact2gdml -input file:<compact-input-xml-file> -output <detector>.gdml
//   b) Extract visualisation hints to csv file
//      $ > geoConverter -compact2vis -input file:<compact-input-xml-file> -output <detector>.vis.csv -ascii
//   c) Load GDML into ROOT (as AClick, but can also be interpreted):
//      $ > root.exe ../doc/gdml_root.C+\(\"<detector>\"\)
//      The macro expects *at least* the <detector>.gdml file. If no visualisation
//      information is found, the resulting graphics however is not very nice!
//
//
//  Author     : M.Frank
//
//====================================================================
// C/C++ include files
#include <fstream>
#include <cerrno>
#include <string>
#include <sstream>

// ROOT include files
#include <TGeoManager.h>
#include <TColor.h>
#include <TObjArray.h>
#include <TGeoVolume.h>
#include <TGDMLParse.h>

using namespace std;

TGeoVolume* gdml_root(const char* sys_name) {
  bool         debug_processing = false;
  TGeoManager* geo = new TGeoManager();
  string       system    = sys_name;
  string       gdml_file = system+".gdml";
  string       vis_file  = system+".vis.csv";
  ifstream     in(vis_file.c_str());
  TGDMLParse   parse;
  gDebug = 2;
  cout << "++ Processing gdml file:" << gdml_file << endl;

  TGeoVolume* top_vol = parse.GDMLReadFile(gdml_file.c_str());
  TObjArray*  vols    = gGeoManager->GetListOfVolumes();
  int num_volumes = vols->GetEntries();

  if ( !in.good() )
    cout << "++ Failed to open visualization file:" << vis_file 
         << " :: " << strerror(errno) << endl;
  else
    cout << "++ Processing visualization file:" << vis_file << endl;

  while(in.good() ) {
    stringstream debug;
    char         text[1024], *line;
    const char*  vol_name="";
    float        red = 0, blue = 0, green = 0;
    bool         visible = false, show_daughters = false;
    string       line_style="solid", drawing_style="solid";

    in.getline(text,sizeof(text),'\n');
    line = strtok(text,";");
    debug << "+ Process:";
    for(int count=0; line != 0; ++count, line=::strtok(0,";"))  {
      debug << line << " ";
      switch(count) {
      case 0:
        vol_name = line+::strlen("vol:");
        break;
      case 1:
        break;
      case 2:
        line += ::strlen("visible:");
        visible = *line=='1';
        break;
      case 3:
        line += ::strlen("r:");
        red = atof(line);
        break;
      case 4:
        line += ::strlen("g:");
        green = atof(line);
        break;
      case 5:
        line += ::strlen("b:");
        blue = atof(line);
        break;
      case 6:
        line += ::strlen("alpha:");
        //alpha = atof(line);
        break;
      case 7:
        line_style = line+::strlen("line_style:");
        break;
      case 8:
        drawing_style = line+::strlen("drawing_style:");
        break;
      case 9:
        line += ::strlen("show_daughters:");
        show_daughters = *line=='1';
        break;
      default:
        break;
      }
    }
    if ( debug_processing ) cout << debug.str() << endl;
    debug.str("");

    if ( vol_name && ::strlen(vol_name) )   {
      TGeoVolume* volume = 0;
      for(int i=0;i<num_volumes;++i) {
        TGeoVolume* v=(TGeoVolume*)vols->At(i);
        if ( 0 == ::strcmp(vol_name,v->GetName()) ) {
          volume = v;
          break;
        }
      }
      if ( volume ) {
        int     color  = TColor::GetColor(red,green,blue);
        Color_t bright = TColor::GetColorBright(color);
        Color_t dark   = TColor::GetColorDark(color);
        debug << "+ \tr:" << red << " g:" << green << " b:" << blue << " col:" << color 
              << " line_style:" << line_style << " drawing_style:" << drawing_style 
              << " visible:" << visible << " show_daughters:" << show_daughters;
        volume->SetLineColor(dark);
        if ( drawing_style == "solid" )  {
          volume->SetFillColor(bright);
          volume->SetFillStyle(1001); // Root: solid
        }
        else {
          //volume->SetFillColor(bright);
          volume->SetFillColor(0);
          volume->SetFillStyle(0);    // Root: hollow
        }
        if ( line_style == "unbroken" )
          volume->SetFillStyle(1);
        else
          volume->SetFillStyle(2);
      
        volume->SetLineWidth(10);
        volume->SetVisibility(visible ? kTRUE : kFALSE);
        volume->SetAttBit(TGeoAtt::kVisContainers,kTRUE);
        volume->SetVisDaughters(show_daughters ? kTRUE : kFALSE);
      }
      else {
        cout << endl << "++ Failed to find volume with name:" << vol_name;
      }
    }
    if ( debug_processing ) cout << debug.str() << endl;
  }
  cout << "++ Closing geometry and starting display.... Top volume:" << (void*)top_vol << endl;
  geo->SetTopVolume(top_vol);
  geo->CloseGeometry();
  geo->SetVisLevel(4);
  geo->SetVisOption(1);
  top_vol->Draw("ogl");
  return top_vol;
}