JDrawModule3D.cc File Reference

Auxiliary program to draw a given module in 3D. More...

#include <string>
#include <iostream>
#include <sstream>
#include <iomanip>
#include "TROOT.h"
#include "TApplication.h"
#include "TCanvas.h"
#include "TView.h"
#include "TGeometry.h"
#include "TGeoManager.h"
#include "TGeoMatrix.h"
#include "TGeoMaterial.h"
#include "TGeoMedium.h"
#include "TGeoVolume.h"
#include "TGeoTube.h"
#include "TGeoCone.h"
#include "TPaveText.h"
#include "JMath/JConstants.hh"
#include "JDetector/JDetector.hh"
#include "JDetector/JDetectorToolkit.hh"
#include "JDetector/JModuleRouter.hh"
#include "JDetector/JDetectorAddressMapToolkit.hh"
#include "Jeep/JPrint.hh"
#include "Jeep/JParser.hh"
#include "Jeep/JMessage.hh"

Go to the source code of this file.

Classes
class	JGeoVolume

Functions
int	main (int argc, char **argv)

Variables
TCanvas *	c1 = NULL
	Global variables to handle mouse events. More...
TPaveText *	p1 = NULL

Detailed Description

Auxiliary program to draw a given module in 3D.

Author

mdejong

Definition in file JDrawModule3D.cc.

Function Documentation

int main	(	int	argc,
		char **	argv
	)

Definition at line 108 of file JDrawModule3D.cc.

{
  using namespace std;

  string         detectorFile;
  int            moduleID;
  int            debug;

  try {

    JParser<> zap("Auxiliary program to draw a given module in 3D.");

    zap['a'] = make_field(detectorFile);
    zap['M'] = make_field(moduleID)        = -1;
    zap['d'] = make_field(debug)           =  1;

    zap(argc, argv);
  }
  catch(const exception &error) {
    FATAL(error.what() << endl);
  }


  using namespace JPP;

  
  JDetector detector;

  try {
    load(detectorFile, detector);
  }
  catch(const JException& error) {
    FATAL(error);
  }

  if (detector.empty()) {
    FATAL("Empty detector.");
  }


  JModule module;

  if (moduleID == -1) {

    module = detector.front();

  } else {

    JModuleRouter router(detector);

    if (!router.hasModule(moduleID)) {
      FATAL("Missing module " << moduleID << endl);
    } else {
      module = router.getModule(moduleID);
    }
  }
  
  module -= module.getPosition();        // ~center module


  const JDetectorAddressMap& demo = getDetectorAddressMap(detector.getID());
  const JModuleAddressMap&   memo = demo.get(module.getID());


  TApplication* app = new TApplication("user", NULL, NULL);
  
  c1 = new TCanvas("module", detectorFile.c_str(), 600, 600);
  p1 = new TPaveText(0.7, 0.9, 1.0, 1.0, "NB");

  c1->SetFillColor(0);

  p1->SetFillColor(0);
  p1->SetBorderSize(0);
  p1->SetTextSize(0.03);

  TGeoManager*  gGeoManager = new TGeoManager("geometry", "");

  TGeoMaterial* mat = new TGeoMaterial("vacuum", 0, 0, 0);
  TGeoMedium*   med = new TGeoMedium  ("vacuum", 1, mat);
  TGeoVolume*   top = gGeoManager->MakeBox("top", med, 30.0, 30.0, 30.0);

  // PMT
  //
  //              ->dz<-
  //
  //               _/|  
  //          R1  |_ |  R2
  //                \|
  //
  //            ->dz<-

  const double dz   =  3.0;  // [cm]
  const double R1   =  2.0;  // [cm]
  const double R2   =  4.0;  // [cm]


  const Int_t color[] = { kRed, kOrange, kYellow, kGreen, kBlue, kMagenta };

  for (unsigned int tdc = 0; tdc != module.size(); ++tdc) {

    const JPMTAddressTranslator& address = memo.getAddressTranslator(tdc);
    const int                    index   = address.ring     - 'A';

    ostringstream os[2];

    os[0] << "PMT " << setw(1) << address.ring << setw(1) << address.position;
    os[1] << "TDC " << setw(2) << tdc;

    TGeoVolumeAssembly* pPMT = new TGeoVolumeAssembly("PMT");

    JGeoVolume* pTube = new JGeoVolume("dynode",  new TGeoTube("",           0.0, R1, 0.5 * dz), med);
    JGeoVolume* pCone = new JGeoVolume("cathode", new TGeoCone("", 0.5 * dz, 0.0, R1, 0.0,  R2), med);
    
    for (int i = 0; i != sizeof(os)/sizeof(os[0]); ++i) {
      pTube->AddText(os[i].str());
      pCone->AddText(os[i].str());
    }

    pTube->SetLineColor(color[index]);
    pCone->SetLineColor(color[index]);

    pPMT->AddNode(pTube, 0, new TGeoTranslation(0.0, 0.0, -dz));
    pPMT->AddNode(pCone, 1, new TGeoTranslation(0.0, 0.0, 0.0));

    const JPMT& pmt = module.getPMT(tdc);
    
    const Double_t x     = pmt.getX() * 100;             // [cm]
    const Double_t y     = pmt.getY() * 100;             // [cm]
    const Double_t z     = pmt.getZ() * 100;             // [cm]
    
    const Double_t theta = pmt.getTheta() * 180.0 / PI;  // [deg]
    const Double_t phi   = pmt.getPhi()   * 180.0 / PI;  // [deg]
    
    DEBUG("PMT"                                           << " "
          << setw(1) << address.ring     
          << setw(1) << address.position                  << " "
          << " TDC "
          << setw(2) << tdc                               << " "
          << FIXED(3,0) << x     << " [cm] " 
          << FIXED(3,0) << y     << " [cm] " 
          << FIXED(3,0) << z     << " [cm] " 
          << FIXED(4,0) << theta << " [deg] " 
          << FIXED(4,0) << phi   << " [deg] " 
          << endl);

    TGeoRotation* rot = new TGeoRotation();
    
    rot->RotateY(theta);
    rot->RotateZ(phi);

    top->AddNode(pPMT, tdc, new TGeoCombiTrans(x,y,z,rot));
  }

  gGeoManager->SetTopVolume(top);
  gGeoManager->CloseGeometry();

  top->Draw();

  c1->GetView()->ShowAxis();
  c1->Update();

  app->Run();
}

Variable Documentation

TCanvas* c1 = NULL

Global variables to handle mouse events.

Definition at line 34 of file JDrawModule3D.cc.

TPaveText* p1 = NULL

Definition at line 35 of file JDrawModule3D.cc.