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();
}