JDrawDetector3D.cc

Go to the documentation of this file.

#include <string>
#include <iostream>
#include <limits>
#include <memory>
 
#include "TROOT.h"
#include "TApplication.h"
#include "TCanvas.h"
#include "TRootCanvas.h"
#include "TView.h"
#include "TGeometry.h"
#include "TGeoManager.h"
#include "TGeoMatrix.h"
#include "TGeoMaterial.h"
#include "TGeoMedium.h"
#include "TGeoVolume.h"
 
#include "JDetector/JDetector.hh"
#include "JDetector/JDetectorToolkit.hh"
 
#include "JGeometry3D/JCylinder3D.hh"
 
#include "Jeep/JParser.hh"
#include "Jeep/JMessage.hh"
 
 
/**
 * \file
 * Auxiliary program to draw the detector in 3D.
 * \author mdejong
 */
int main(int argc, char**argv)
{
  using namespace std;
  using namespace JPP;
 
  string         detectorFile;
  int            debug;
 
  try {
 
    JParser<> zap("Auxiliary program to draw the detector in 3D.");
 
    zap['a'] = make_field(detectorFile);
    zap['d'] = make_field(debug)             = 1;
 
    zap(argc, argv);
  }
  catch(const exception &error) {
    FATAL(error.what() << endl);
  }
 
 
  JDetector detector;
 
  try {
    load(detectorFile, detector);
  }
  catch(const JException& error) {
    FATAL(error);
  }
 
  if (detector.empty()) {
    FATAL("Empty detector " << detectorFile << endl);
  }
 
  JCylinder3D cylinder(detector.begin(), detector.end());
 
  NOTICE("Detector volume: " << cylinder << endl);
 
  TApplication* tp = new TApplication("user", NULL, NULL);
  TCanvas*      cv = new TCanvas("a", detectorFile.c_str(), 1200, 1200);
 
  ((TRootCanvas *) cv->GetCanvasImp())->Connect("CloseWindow()", "TApplication", tp, "Terminate()");
 
  cv->SetFillColor(0);
 
  TGeoManager*  geom     = new TGeoManager ("geometry", "");
  TGeoMaterial* material = new TGeoMaterial("vacuum", 0, 0, 0);
  TGeoMedium*   medium   = new TGeoMedium  ("vacuum", 1, material);
  TGeoVolume*   top      = geom->MakeBox   ("Top",    medium, cylinder.getRadius(), cylinder.getRadius(), cylinder.getZmax() - cylinder.getZmin());
  TGeoVolume*   shape    = geom->MakeSphere("Module", medium, 0.4, 0.5); 
 
  shape->SetLineColor(kBlue);
 
  int N = 0;
 
  for (JDetector::const_iterator i = detector.begin(); i != detector.end(); ++i) {
    top->AddNode(shape, N++, new TGeoTranslation(i->getX() - cylinder.getX(), i->getY() - cylinder.getY(), i->getZ() - cylinder.getZmin()));
  }
 
  geom->SetTopVolume(top);
  geom->CloseGeometry();
  //geom->SetVisLevel(4);
 
  top->Draw();
 
  cv->GetView()->ShowAxis();
  cv->Update();
 
  tp->Run();
}