JDrawDetector2D.cc

Go to the documentation of this file.

#include <string>
#include <iostream>
#include <limits>
#include <vector>
 
#include "TROOT.h"
#include "TH2D.h"
#include "TApplication.h"
#include "TGraph.h"
#include "TEllipse.h"
#include "TCanvas.h"
#include "TMarker.h"
#include "TStyle.h"
#include "TLegend.h"
 
#include "JGizmo/JMarkerAttributes.hh"
#include "JDetector/JDetector.hh"
#include "JDetector/JDetectorToolkit.hh"
#include "JGeometry2D/JPosition2D.hh"
#include "JGeometry2D/JCircle2D.hh"
 
#include "Jeep/JParser.hh"
#include "Jeep/JMessage.hh"
 
 
/**
 * \file
 * Auxiliary program to draw the footprint of detector(s).
 * \author mdejong
 */
int main(int argc, char**argv)
{
  using namespace std;
 
  vector<string> detectorFile;
  bool           legend;
  double         markerSize;
  int            debug;
 
  try {
 
    JParser<> zap("Auxiliary program to draw the footprint of detector(s).");
 
    zap['a'] = make_field(detectorFile);
    zap['L'] = make_field(legend);
    zap['S'] = make_field(markerSize)        = 0.5;
    zap['d'] = make_field(debug)             = 1;
 
    zap(argc, argv);
  }
  catch(const exception &error) {
    FATAL(error.what() << endl);
  }
 
 
  using namespace JPP;
 
 
  JGIZMO::JMarkerAttributes::getInstance().setMarkerSize(markerSize);
 
 
  int npos = 0;
 
  vector<TGraph*> buffer;
  
  JCircle2D circle;
 
  for (vector<string>::const_iterator file_name = detectorFile.begin(); file_name != detectorFile.end(); ++file_name) {
 
    JDetector detector;
 
    try {
      load(*file_name, detector);
    }
    catch(const JException& error) {
      FATAL(error);
    }
 
    if (detector.empty()) {
      ERROR("Empty detector." << endl);
    }
 
    circle = JCircle2D(detector.begin(), detector.end());
 
    TGraph* graph = new TGraph(detector.size());
 
    graph->SetTitle(file_name->c_str());
 
    int n = 0;
 
    for (JDetector::const_iterator i = detector.begin(); i != detector.end(); ++i, ++n) {      
 
      npos += 1;
 
      graph->GetX()[n] = i->getX();
      graph->GetY()[n] = i->getY();
    }
    
    dynamic_cast<TAttMarker&>(*graph) = *JGIZMO::JMarkerAttributes::getInstance().next();
 
    buffer.push_back(graph);
  }
 
 
  if (npos == 0) {
    FATAL("No detector elements." << endl);
  }
 
 
  DEBUG("Detector (x,y,R): " << circle.getX() << ' ' << circle.getY() << ' ' << circle.getRadius() << endl);
 
  for (vector<TGraph*>::iterator graph = buffer.begin(); graph != buffer.end(); ++graph) {
    for (int i = 0; i != (*graph)->GetN(); ++i) {
      (*graph)->GetX()[i] -= circle.getX();
      (*graph)->GetY()[i] -= circle.getY();
    }
  }
 
  circle.sub(circle.getPosition());
 
  Double_t R    = circle.getRadius();
 
  if (R <= 1.0) {
    R = 1.0;
  }
 
  Double_t xmin = circle.getX() - R;
  Double_t xmax = circle.getX() + R;
  Double_t ymin = circle.getY() - R;
  Double_t ymax = circle.getY() + R;
 
 
  TApplication* tp = new TApplication("user", NULL, NULL);
 
 
  TCanvas cv("detector", "", 400, 400);
 
  cv.SetFillStyle(4000);
  cv.SetFillColor(0);
 
  cv.Divide(1, 1);
  cv.cd(1);
 
  const Double_t dx = (xmax - xmin);
  const Double_t dy = (ymax - ymin);
 
  xmin -= 0.1 * dx;
  xmax += 0.1 * dx;
  ymin -= 0.1 * dy;
  ymax += 0.1 * dy;
 
 
  TH2D h2("h2", "", 1, xmin, xmax, 1, ymin, ymax);
 
  h2.GetXaxis()->SetTitle("x [m]");
  h2.GetYaxis()->SetTitle("y [m]");
  
  h2.GetXaxis()->CenterTitle(true);
  h2.GetYaxis()->CenterTitle(true);
 
  h2.SetStats(kFALSE);
  h2.Draw();
 
 
  TEllipse ellipse(circle.getX(), circle.getY(), circle.getRadius());
 
  ellipse.Draw();
 
  for (vector<TGraph*>::iterator graph = buffer.begin(); graph != buffer.end(); ++graph) {
    (*graph)->Draw("P");
  }
 
  
  if (legend) {
 
    const Double_t x2  = gPad->GetX2() - gStyle->GetPadRightMargin();
    const Double_t y2  = gPad->GetY2() - gStyle->GetPadTopMargin();
    
    size_t   length    = 10;
    Double_t font_size = 0.03; // gStyle->GetStatFontSize();
    
    for (vector<TGraph*>::iterator graph = buffer.begin(); graph != buffer.end(); ++graph) {
      length = max(length, strlen((*graph)->GetTitle()));
    }
    
    TLegend* legend = new TLegend(x2 - length        * font_size * 0.65 - 0.02, 
                                  y2 - buffer.size() * font_size, 
                                  x2 - 0.01, 
                                  y2 - 0.01);
    
    //legend->SetFillStyle(4000);
    legend->SetFillColor(0);
    legend->SetBorderSize(0);
    legend->SetTextSize(font_size);
    
    for (vector<TGraph*>::iterator graph = buffer.begin(); graph != buffer.end(); ++graph) {
      legend->AddEntry(*graph, (*graph)->GetTitle(), "P"); 
    }
    
    legend->Draw();
  }
 
  cv.Update();
 
  tp->Run();
}