[u/mrichter/AliRoot.git] / FMD / AliFMDFancy.cxx

/**************************************************************************
 * Copyright(c) 2004, ALICE Experiment at CERN, All rights reserved. *
 *                                                                        *
 * Author: The ALICE Off-line Project.                                    *
 * Contributors are mentioned in the code where appropriate.              *
 *                                                                        *
 * Permission to use, copy, modify and distribute this software and its   *
 * documentation strictly for non-commercial purposes is hereby granted   *
 * without fee, provided that the above copyright notice appears in all   *
 * copies and that both the copyright notice and this permission notice   *
 * appear in the supporting documentation. The authors make no claims     *
 * about the suitability of this software for any purpose. It is          *
 * provided "as is" without express or implied warranty.                  *
 **************************************************************************/
/* $Id$ */
/** @file    AliFMDFancy.cxx
    @author  Christian Holm Christensen <cholm@nbi.dk>
    @date    Mon Mar 27 12:39:09 2006
    @brief   FMD 2D Event display 
*/
//___________________________________________________________________
//
// The classes defined here, are utility classes for reading in data
// for the FMD.  They are  put in a seperate library to not polute the
// normal libraries.  The classes are intended to be used as base
// classes for customized class that do some sort of analysis on the
// various types of data produced by the FMD. 
//
// Latest changes by Christian Holm Christensen
//
#include "AliFMDFancy.h"	// ALIFMDDISPLAY_H
#include "AliFMDGeometry.h"	// ALIFMDGEOMETRY_H
#include "AliFMDParameters.h"	// ALIFMDPARAMETERS_H
#include <AliLog.h>
#include <TStyle.h>
#include <TApplication.h>
#include <TButton.h>
#include <TCanvas.h>
#include <TView.h>
#include <TH2F.h>
#include <TH3F.h>
#include <TSystem.h>
#include <TVector2.h>
#include "AliFMDRing.h"
#include "AliFMDDetector.h"
#include "AliFMDHit.h"
#include <TRandom.h>
#include <TLatex.h>
#include <TLine.h>
#include <iostream>
#include <iomanip>

//____________________________________________________________________
ClassImp(AliFMDFancy)
#if 0
  ; // This is here to keep Emacs for indenting the next line
#endif

//____________________________________________________________________
AliFMDFancy::AliFMDFancy(const char* gAliceFile)
  : AliFMDDisplay(kTRUE, gAliceFile),
    fFMD1Pad(0),
    fFMD1(1),
    fFMD2Pad(0),
    fFMD2(2),
    fFMD3Pad(0),
    fFMD3(3),
    fEvent(.1, .8, "Event #"),
    fFMD1IHits(.2, .7, "# in FMD1I: "),
    fFMD2IHits(.2, .6, "# in FMD2I: "),
    fFMD2OHits(.2, .5, "# in FMD2O: "),
    fFMD3IHits(.2, .4, "# in FMD3I: "),
    fFMD3OHits(.2, .3, "# in FMD3O: "),
    fLine(.15, .27, .85, .27),
    fTotal(.2, .15, "Total:   ")
{
  fEvent.SetBit(TLatex::kTextNDC);
  fFMD1IHits.SetBit(TLatex::kTextNDC);
  fFMD2IHits.SetBit(TLatex::kTextNDC);
  fFMD2OHits.SetBit(TLatex::kTextNDC);
  fFMD3IHits.SetBit(TLatex::kTextNDC);
  fFMD3OHits.SetBit(TLatex::kTextNDC);
  fLine.SetBit(TLine::kLineNDC);
  fTotal.SetBit(TLatex::kTextNDC);
}

//____________________________________________________________________
AliFMDFancy::Detector::Detector(UShort_t id)
  : fId(id)
{
  fInnerHits.SetName(Form("FMD%dI", id));
  fInnerHits.SetMarkerStyle(1); // 20);
  fInnerHits.SetMarkerSize(.2);
  fInnerHits.SetMarkerColor(50); // 12);
  fOuterHits.SetName(Form("FMD%dO", id));
  fOuterHits.SetMarkerStyle(1); // 20);
  fOuterHits.SetMarkerSize(.2);
  fOuterHits.SetMarkerColor(50); // 12);
}

//____________________________________________________________________
AliFMDFancy::Detector::~Detector()
{
  fShapes.Delete();
  if (fFrame) delete fFrame;
}

//____________________________________________________________________
AliFMDFancy::~AliFMDFancy()
{}

//____________________________________________________________________
void
AliFMDFancy::Detector::AddHistogram(TGraph2D& g, const char* opt)
{
  TH2* h = g.GetHistogram(opt);
  if (!h) return;
  h->SetBins(1, -fMaxR, fMaxR, 1, -fMaxR, fMaxR);
  h->GetZaxis()->SetRangeUser(fMinZ, fMaxZ);
}

  

//____________________________________________________________________
void
AliFMDFancy::Detector::Init()
{
  AliFMDGeometry* geom = AliFMDGeometry::Instance();
  AliFMDDetector* det  = geom->GetDetector(fId);
  if (!det) return;
  Char_t   rs[]   = { 'I' , 'O', '\0' };
  Char_t*  rp     = rs;
  Char_t   r;
  Double_t maxR = 0;
  Double_t minZ = 10000;
  Double_t maxZ = -10000;
  Int_t    ns   = 0;
  while ((r = *(rp++))) {
    AliFMDRing* ring = det->GetRing(r);
    if (!ring) continue;
    // if (r == 'O') continue;
    const TObjArray& vs = ring->GetVerticies();
    Int_t            nm = ring->GetNModules();
    Double_t         zd = (r == 'I' ? det->GetInnerZ() : det->GetOuterZ());
    for (Int_t m = 0; m < nm; m++) {
      Int_t          nv = vs.GetEntries();
      Double_t       a  = TMath::Pi() / 180 * (m * 2 + 1) * ring->GetTheta();
      TGraph2D*      g  = new TGraph2D(nv);
      Double_t       x0 = 0, y0 = 0, z0 = 0;
      Double_t       z  = zd + (m % 2==0 ? 0 : 
				TMath::Sign(ring->GetModuleSpacing(), zd));
      minZ              = TMath::Min(minZ, z);
      maxZ              = TMath::Max(maxZ, z);
      g->SetName(Form("FMD%d%cM%02d", fId, r, m));
      fShapes.AddAtAndExpand(g, ns++);
      for (Int_t c = 0; c < nv; c++) {
	TVector2* v = static_cast<TVector2*>(vs.At(nv - 1 - c));
	TVector2  w(v->Rotate(a));
	if (c == 0) { x0 = w.X(); y0 = w.Y(); z0 = z; }
	g->SetPoint(c, w.X(), w.Y(), z);
	maxR        = TMath::Max(maxR, v->Mod());
      }
      //g->SetPoint(nv, x0, y0, z0);
      g->SetFillColor(2);
      g->SetFillStyle(3002);
      g->SetLineColor(2);
      g->SetLineWidth(1);
    }
  }
  fMaxR = 1.05 * maxR;
  fMinZ = (minZ  > 0 ? 0.95 * minZ : 1.05 * minZ);
  fMaxZ = (maxZ  > 0 ? 1.05 * maxZ : 0.95 * maxZ);

  TIter next(&fShapes);
  TGraph2D* g = 0;
  while ((g = static_cast<TGraph2D*>(next()))) AddHistogram(*g);
  if (det->GetInner()) AddHistogram(fInnerHits);
  if (det->GetOuter()) AddHistogram(fOuterHits);

  fFrame = new TH2F(Form("FMD%d", fId), Form("FMD%d", fId), 
		    1, -fMaxR, fMaxR, 1, -fMaxR, fMaxR);
  fFrame->SetStats(kFALSE);
  fFrame->GetXaxis()->SetTitle("x [cm]");
  fFrame->GetYaxis()->SetTitle("y [cm]");
  fFrame->GetZaxis()->SetTitle("z [cm]");
  fFrame->SetDirectory(0);
}

  
//____________________________________________________________________
Bool_t 
AliFMDFancy::Init()
{
  // Initialize.  GEt transforms and such, 
  if (!AliFMDInput::Init()) return kFALSE;
  AliFMDGeometry* geom = AliFMDGeometry::Instance();
  geom->Init();
  geom->InitTransformations();
  
  fFMD1.Init();
  fFMD2.Init();
  fFMD3.Init();
  return kTRUE;
}

//____________________________________________________________________
void
AliFMDFancy::Detector::Begin(Int_t /* event */)
{
  TIter next(&fShapes);
  TGraph2D* g = 0;
  fFrame->Draw("surf fb");
  fFrame->GetZaxis()->SetRangeUser(fMinZ, fMaxZ);
  while ((g = static_cast<TGraph2D*>(next()))) g->Draw("tri2 FB same");
}

//____________________________________________________________________
void
AliFMDFancy::Detector::Clear(Int_t /* event */)
{
  fNInnerHits = 0;
  fNOuterHits = 0;
}

  
//____________________________________________________________________
Bool_t 
AliFMDFancy::Begin(Int_t event) 
{
  if (!fCanvas) {
    const char* which[] = { "Continue", "Redisplay", 0 };
    MakeCanvas(which);

    AliFMDGeometry* geom = AliFMDGeometry::Instance();
    AliFMDDetector* det;
    if ((det = geom->GetDetector(1))) {
      fPad->cd();
      fFMD1Pad = new TPad("FMD1", "FMD1", 0.0, 0.50, 0.5, 1.0, 0, 0);
      fFMD1Pad->Draw();
      fFMD1Pad->cd();
      fFMD1.Begin(event);
    }
    if ((det = geom->GetDetector(2))) {
      fPad->cd();
      fFMD2Pad = new TPad("FMD2", "FMD2", 0.5, 0.50, 1.0, 1.0, 0, 0);
      fFMD2Pad->Draw();
      fFMD2Pad->cd();
      fFMD2.Begin(event);
    }
    if ((det = geom->GetDetector(3))) {
      fPad->cd();
      fFMD3Pad = new TPad("FMD3", "FMD3", 0.0, 0.05, .5, .5, 0, 0);
      fFMD3Pad->Draw();
      fFMD3Pad->cd();
      fFMD3.Begin(event);
    }
    fPad->cd();
    fSummary = new TPad("display", "Display", 0.5, 0.05, 1.0, 0.5, 0, 0);
    fSummary->Draw();
    fSummary->cd();
    fEvent.Draw();
    fFMD1IHits.Draw();
    fFMD2IHits.Draw();
    fFMD2OHits.Draw();
    fFMD3IHits.Draw();
    fFMD3OHits.Draw();
    fLine.Draw();
    fTotal.Draw();
  }
  fEvent.SetTitle(Form("Event # %6d", event));
  // fEvent.Modify();
  fCanvas->Modified();
  fCanvas->Update();
  fCanvas->cd();
  fFMD1.Clear(event);
  fFMD2.Clear(event);
  fFMD3.Clear(event);
  return AliFMDInput::Begin(event);
}

//____________________________________________________________________
void
AliFMDFancy::Detector::End()
{
  Char_t  rs[] = { 'I', 'O', '\0' };
  Char_t* rp   = rs;
  Char_t  r;
  while ((r = *(rp++))) {
    TGraph2D& g = (r == 'I' ? fInnerHits : fOuterHits);
    Int_t&    n = (r == 'I' ? fNInnerHits : fNOuterHits);
    Int_t     m = (r == 'I' ? 512 * 10 * 2 : 256 * 20 * 2);
    if (n == 0) continue;
    for (Int_t i = n; i < g.GetN(); i++)  g.RemovePoint(i);
    std::cout << g.GetName() << " has " << std::setw(4) << n << "/" 
	      << std::setw(5) << m << " points" << std::endl;
    g.Draw("same fb p");
    AddHistogram(g, "empty");
  }
}

//____________________________________________________________________
Bool_t 
AliFMDFancy::End() 
{
  AliFMDGeometry* geom = AliFMDGeometry::Instance();
  AliFMDDetector* det;
  Int_t total = 0;
  if ((det = geom->GetDetector(1))) {
    fFMD1Pad->cd();
    fFMD1.End();
    fFMD1Pad->Modified();
    fFMD1IHits.SetTitle(Form("# hits in FMD1I:  %5d", fFMD1.fNInnerHits));
    total += fFMD1.fNInnerHits;
  }
  if ((det = geom->GetDetector(2))) {
    fFMD2Pad->cd();
    fFMD2.End();
    fFMD2Pad->Modified();
    fFMD2IHits.SetTitle(Form("# hits in FMD2I:  %5d", fFMD2.fNInnerHits));
    fFMD2OHits.SetTitle(Form("# hits in FMD2O: %5d", fFMD2.fNOuterHits));
    total += fFMD2.fNInnerHits;
    total += fFMD2.fNOuterHits;    
  }
  if ((det = geom->GetDetector(3))) {
    fFMD3Pad->cd();
    fFMD3.End();
    fFMD3Pad->Modified();
    fFMD3IHits.SetTitle(Form("# hits in FMD3I:  %5d", fFMD3.fNInnerHits));
    fFMD3OHits.SetTitle(Form("# hits in FMD3O: %5d", fFMD3.fNOuterHits));
    total += fFMD3.fNInnerHits;
    total += fFMD3.fNOuterHits;    
  }
  fTotal.SetTitle(Form("Total:    %5d/51200 (%3d%%)", 
		      total, Int_t(100. / 51200 * total)));
  fSummary->Modified();
  fCanvas->Modified();
  fCanvas->Update();
  fCanvas->cd();
  fWait = kTRUE;
  while (fWait) {
    gApplication->StartIdleing();
    gSystem->InnerLoop();
    gApplication->StopIdleing();
  }
  return AliFMDInput::End();
}

//____________________________________________________________________
Bool_t 
AliFMDFancy::ProcessHit(AliFMDHit* hit, TParticle*) 
{
  AddMarker(hit->Detector(), hit->Ring(), hit->Sector(), hit->Strip(), 
	    hit, hit->Edep(), 0);
  return kTRUE;
}
//____________________________________________________________________
void
AliFMDFancy::Detector::AddMarker(Char_t rng, UShort_t sec, UShort_t str,
				 Float_t, Float_t)
{
  AliFMDGeometry*   geom = AliFMDGeometry::Instance();
  Double_t x, y, z;
  geom->Detector2XYZ(fId, rng, sec, str, x, y, z);
  if (true) {
    AliFMDRing* r  = geom->GetRing(rng);
    Double_t    t  = .9 * r->GetTheta() / 2;
    Double_t    a  = gRandom->Uniform(-t,t) * TMath::Pi() / 180;
    Double_t    x1 = x * TMath::Cos(a) - y * TMath::Sin(a);
    Double_t    y1 = x * TMath::Sin(a) + y * TMath::Cos(a);
    x = x1;
    y = y1;
  }
  switch (rng) {
  case 'I':
  case 'i': fInnerHits.SetPoint(fNInnerHits++, x, y, z); break;
  case 'O': 
  case 'o': fOuterHits.SetPoint(fNOuterHits++, x, y, z); break;
  default:  return;
  }
}

    

//____________________________________________________________________
void
AliFMDFancy::AddMarker(UShort_t det, Char_t rng, UShort_t sec, UShort_t str,
		       TObject*, Float_t, Float_t)
{
  // Add a marker to the display
  //
  //    det 	Detector
  //    rng 	Ring
  //    sec 	Sector 
  //    str 	Strip
  //    o   	Object to refer to
  //    s   	Signal 
  //    max 	Maximum of signal 
  //
  switch (det) {
  case 1: fFMD1.AddMarker(rng,sec,str,0,0); break;
  case 2: fFMD2.AddMarker(rng,sec,str,0,0); break;
  case 3: fFMD3.AddMarker(rng,sec,str,0,0); break;
  }
}

//____________________________________________________________________
//
// EOF
//
Commit	Line	Data
a9579262	1	/**************************************************************************
	2	* Copyright(c) 2004, ALICE Experiment at CERN, All rights reserved. *
	3	* *
	4	* Author: The ALICE Off-line Project. *
	5	* Contributors are mentioned in the code where appropriate. *
	6	* *
	7	* Permission to use, copy, modify and distribute this software and its *
	8	* documentation strictly for non-commercial purposes is hereby granted *
	9	* without fee, provided that the above copyright notice appears in all *
	10	* copies and that both the copyright notice and this permission notice *
	11	* appear in the supporting documentation. The authors make no claims *
	12	* about the suitability of this software for any purpose. It is *
	13	* provided "as is" without express or implied warranty. *
	14	**************************************************************************/
	15	/* $Id$ */
	16	/** @file AliFMDFancy.cxx
	17	@author Christian Holm Christensen <cholm@nbi.dk>
	18	@date Mon Mar 27 12:39:09 2006
	19	@brief FMD 2D Event display
	20	*/
	21	//___________________________________________________________________
	22	//
	23	// The classes defined here, are utility classes for reading in data
	24	// for the FMD. They are put in a seperate library to not polute the
	25	// normal libraries. The classes are intended to be used as base
	26	// classes for customized class that do some sort of analysis on the
	27	// various types of data produced by the FMD.
	28	//
	29	// Latest changes by Christian Holm Christensen
	30	//
	31	#include "AliFMDFancy.h" // ALIFMDDISPLAY_H
	32	#include "AliFMDGeometry.h" // ALIFMDGEOMETRY_H
	33	#include "AliFMDParameters.h" // ALIFMDPARAMETERS_H
	34	#include <AliLog.h>
	35	#include <TStyle.h>
	36	#include <TApplication.h>
	37	#include <TButton.h>
	38	#include <TCanvas.h>
	39	#include <TView.h>
	40	#include <TH2F.h>
	41	#include <TH3F.h>
	42	#include <TSystem.h>
	43	#include <TVector2.h>
	44	#include "AliFMDRing.h"
	45	#include "AliFMDDetector.h"
	46	#include "AliFMDHit.h"
	47	#include <TRandom.h>
	48	#include <TLatex.h>
	49	#include <TLine.h>
	50	#include <iostream>
	51	#include <iomanip>
	52
	53	//____________________________________________________________________
	54	ClassImp(AliFMDFancy)
	55	#if 0
	56	; // This is here to keep Emacs for indenting the next line
	57	#endif
	58
	59	//____________________________________________________________________
	60	AliFMDFancy::AliFMDFancy(const char* gAliceFile)
	61	: AliFMDDisplay(kTRUE, gAliceFile),
	62	fFMD1Pad(0),
	63	fFMD1(1),
	64	fFMD2Pad(0),
65	fFMD2(2),
66	fFMD3Pad(0),
67	fFMD3(3),
68	fEvent(.1, .8, "Event #"),
69	fFMD1IHits(.2, .7, "# in FMD1I: "),
70	fFMD2IHits(.2, .6, "# in FMD2I: "),
71	fFMD2OHits(.2, .5, "# in FMD2O: "),
72	fFMD3IHits(.2, .4, "# in FMD3I: "),
73	fFMD3OHits(.2, .3, "# in FMD3O: "),
74	fLine(.15, .27, .85, .27),
75	fTotal(.2, .15, "Total: ")
76	{
77	fEvent.SetBit(TLatex::kTextNDC);
78	fFMD1IHits.SetBit(TLatex::kTextNDC);
79	fFMD2IHits.SetBit(TLatex::kTextNDC);
80	fFMD2OHits.SetBit(TLatex::kTextNDC);
81	fFMD3IHits.SetBit(TLatex::kTextNDC);
82	fFMD3OHits.SetBit(TLatex::kTextNDC);
83	fLine.SetBit(TLine::kLineNDC);
84	fTotal.SetBit(TLatex::kTextNDC);
85	}
86
87	//____________________________________________________________________
88	AliFMDFancy::Detector::Detector(UShort_t id)
89	: fId(id)
90	{
91	fInnerHits.SetName(Form("FMD%dI", id));
92	fInnerHits.SetMarkerStyle(1); // 20);
93	fInnerHits.SetMarkerSize(.2);
94	fInnerHits.SetMarkerColor(50); // 12);
95	fOuterHits.SetName(Form("FMD%dO", id));
96	fOuterHits.SetMarkerStyle(1); // 20);
97	fOuterHits.SetMarkerSize(.2);
98	fOuterHits.SetMarkerColor(50); // 12);
99	}
100
101	//____________________________________________________________________
102	AliFMDFancy::Detector::~Detector()
103	{
104	fShapes.Delete();
105	if (fFrame) delete fFrame;
106	}
107
108	//____________________________________________________________________
109	AliFMDFancy::~AliFMDFancy()
110	{}
111
112	//____________________________________________________________________
113	void
114	AliFMDFancy::Detector::AddHistogram(TGraph2D& g, const char* opt)
115	{
116	TH2* h = g.GetHistogram(opt);
117	if (!h) return;
118	h->SetBins(1, -fMaxR, fMaxR, 1, -fMaxR, fMaxR);
119	h->GetZaxis()->SetRangeUser(fMinZ, fMaxZ);
120	}
121
122
123
124	//____________________________________________________________________
125	void
126	AliFMDFancy::Detector::Init()
127	{
128	AliFMDGeometry* geom = AliFMDGeometry::Instance();
129	AliFMDDetector* det = geom->GetDetector(fId);
130	if (!det) return;
131	Char_t rs[] = { 'I' , 'O', '\0' };
132	Char_t* rp = rs;
133	Char_t r;
134	Double_t maxR = 0;
135	Double_t minZ = 10000;
136	Double_t maxZ = -10000;
137	Int_t ns = 0;
138	while ((r = *(rp++))) {
139	AliFMDRing* ring = det->GetRing(r);
140	if (!ring) continue;
141	// if (r == 'O') continue;
142	const TObjArray& vs = ring->GetVerticies();
143	Int_t nm = ring->GetNModules();
144	Double_t zd = (r == 'I' ? det->GetInnerZ() : det->GetOuterZ());
145	for (Int_t m = 0; m < nm; m++) {
146	Int_t nv = vs.GetEntries();
147	Double_t a = TMath::Pi() / 180 * (m * 2 + 1) * ring->GetTheta();
148	TGraph2D* g = new TGraph2D(nv);
149	Double_t x0 = 0, y0 = 0, z0 = 0;
150	Double_t z = zd + (m % 2==0 ? 0 :
151	TMath::Sign(ring->GetModuleSpacing(), zd));
152	minZ = TMath::Min(minZ, z);
153	maxZ = TMath::Max(maxZ, z);
154	g->SetName(Form("FMD%d%cM%02d", fId, r, m));
155	fShapes.AddAtAndExpand(g, ns++);
156	for (Int_t c = 0; c < nv; c++) {
157	TVector2* v = static_cast<TVector2*>(vs.At(nv - 1 - c));
158	TVector2 w(v->Rotate(a));
159	if (c == 0) { x0 = w.X(); y0 = w.Y(); z0 = z; }
160	g->SetPoint(c, w.X(), w.Y(), z);
161	maxR = TMath::Max(maxR, v->Mod());
162	}
163	//g->SetPoint(nv, x0, y0, z0);
164	g->SetFillColor(2);
165	g->SetFillStyle(3002);
166	g->SetLineColor(2);
167	g->SetLineWidth(1);
168	}
169	}
170	fMaxR = 1.05 * maxR;
171	fMinZ = (minZ > 0 ? 0.95 * minZ : 1.05 * minZ);
172	fMaxZ = (maxZ > 0 ? 1.05 * maxZ : 0.95 * maxZ);
173
174	TIter next(&fShapes);
175	TGraph2D* g = 0;
176	while ((g = static_cast<TGraph2D>(next()))) AddHistogram(g);
177	if (det->GetInner()) AddHistogram(fInnerHits);
178	if (det->GetOuter()) AddHistogram(fOuterHits);
179
180	fFrame = new TH2F(Form("FMD%d", fId), Form("FMD%d", fId),
181	1, -fMaxR, fMaxR, 1, -fMaxR, fMaxR);
182	fFrame->SetStats(kFALSE);
183	fFrame->GetXaxis()->SetTitle("x [cm]");
184	fFrame->GetYaxis()->SetTitle("y [cm]");
185	fFrame->GetZaxis()->SetTitle("z [cm]");
186	fFrame->SetDirectory(0);
187	}
188
189
190	//____________________________________________________________________
191	Bool_t
192	AliFMDFancy::Init()
193	{
194	// Initialize. GEt transforms and such,
195	if (!AliFMDInput::Init()) return kFALSE;
196	AliFMDGeometry* geom = AliFMDGeometry::Instance();
197	geom->Init();
198	geom->InitTransformations();
199
200	fFMD1.Init();
201	fFMD2.Init();
202	fFMD3.Init();
203	return kTRUE;
204	}
205
206	//____________________________________________________________________
207	void
208	AliFMDFancy::Detector::Begin(Int_t /* event */)
209	{
210	TIter next(&fShapes);
211	TGraph2D* g = 0;
212	fFrame->Draw("surf fb");
213	fFrame->GetZaxis()->SetRangeUser(fMinZ, fMaxZ);
214	while ((g = static_cast<TGraph2D*>(next()))) g->Draw("tri2 FB same");
215	}
216
217	//____________________________________________________________________
218	void
219	AliFMDFancy::Detector::Clear(Int_t /* event */)
220	{
221	fNInnerHits = 0;
222	fNOuterHits = 0;
223	}
224
225
226	//____________________________________________________________________
227	Bool_t
228	AliFMDFancy::Begin(Int_t event)
229	{
230	if (!fCanvas) {
231	const char* which[] = { "Continue", "Redisplay", 0 };
232	MakeCanvas(which);
233
234	AliFMDGeometry* geom = AliFMDGeometry::Instance();
235	AliFMDDetector* det;
236	if ((det = geom->GetDetector(1))) {
237	fPad->cd();
238	fFMD1Pad = new TPad("FMD1", "FMD1", 0.0, 0.50, 0.5, 1.0, 0, 0);
239	fFMD1Pad->Draw();
240	fFMD1Pad->cd();
241	fFMD1.Begin(event);
242	}
243	if ((det = geom->GetDetector(2))) {
244	fPad->cd();
245	fFMD2Pad = new TPad("FMD2", "FMD2", 0.5, 0.50, 1.0, 1.0, 0, 0);
246	fFMD2Pad->Draw();
247	fFMD2Pad->cd();
248	fFMD2.Begin(event);
249	}
250	if ((det = geom->GetDetector(3))) {
251	fPad->cd();
252	fFMD3Pad = new TPad("FMD3", "FMD3", 0.0, 0.05, .5, .5, 0, 0);
253	fFMD3Pad->Draw();
254	fFMD3Pad->cd();
255	fFMD3.Begin(event);
256	}
257	fPad->cd();
258	fSummary = new TPad("display", "Display", 0.5, 0.05, 1.0, 0.5, 0, 0);
259	fSummary->Draw();
260	fSummary->cd();
261	fEvent.Draw();
262	fFMD1IHits.Draw();
263	fFMD2IHits.Draw();
264	fFMD2OHits.Draw();
265	fFMD3IHits.Draw();
266	fFMD3OHits.Draw();
267	fLine.Draw();
268	fTotal.Draw();
269	}
270	fEvent.SetTitle(Form("Event # %6d", event));
271	// fEvent.Modify();
272	fCanvas->Modified();
273	fCanvas->Update();
274	fCanvas->cd();
275	fFMD1.Clear(event);
276	fFMD2.Clear(event);
277	fFMD3.Clear(event);
278	return AliFMDInput::Begin(event);
279	}
280
281	//____________________________________________________________________
282	void
283	AliFMDFancy::Detector::End()
284	{
285	Char_t rs[] = { 'I', 'O', '\0' };
286	Char_t* rp = rs;
287	Char_t r;
288	while ((r = *(rp++))) {
289	TGraph2D& g = (r == 'I' ? fInnerHits : fOuterHits);
290	Int_t& n = (r == 'I' ? fNInnerHits : fNOuterHits);
291	Int_t m = (r == 'I' ? 512 * 10 * 2 : 256 * 20 * 2);
292	if (n == 0) continue;
293	for (Int_t i = n; i < g.GetN(); i++) g.RemovePoint(i);
294	std::cout << g.GetName() << " has " << std::setw(4) << n << "/"
295	<< std::setw(5) << m << " points" << std::endl;
296	g.Draw("same fb p");
297	AddHistogram(g, "empty");
298	}
299	}
300
301	//____________________________________________________________________
302	Bool_t
303	AliFMDFancy::End()
304	{
305	AliFMDGeometry* geom = AliFMDGeometry::Instance();
306	AliFMDDetector* det;
307	Int_t total = 0;
308	if ((det = geom->GetDetector(1))) {
309	fFMD1Pad->cd();
310	fFMD1.End();
311	fFMD1Pad->Modified();
312	fFMD1IHits.SetTitle(Form("# hits in FMD1I: %5d", fFMD1.fNInnerHits));
313	total += fFMD1.fNInnerHits;
314	}
315	if ((det = geom->GetDetector(2))) {
316	fFMD2Pad->cd();
317	fFMD2.End();
318	fFMD2Pad->Modified();
319	fFMD2IHits.SetTitle(Form("# hits in FMD2I: %5d", fFMD2.fNInnerHits));
320	fFMD2OHits.SetTitle(Form("# hits in FMD2O: %5d", fFMD2.fNOuterHits));
321	total += fFMD2.fNInnerHits;
322	total += fFMD2.fNOuterHits;
323	}
324	if ((det = geom->GetDetector(3))) {
325	fFMD3Pad->cd();
326	fFMD3.End();
327	fFMD3Pad->Modified();
328	fFMD3IHits.SetTitle(Form("# hits in FMD3I: %5d", fFMD3.fNInnerHits));
329	fFMD3OHits.SetTitle(Form("# hits in FMD3O: %5d", fFMD3.fNOuterHits));
330	total += fFMD3.fNInnerHits;
331	total += fFMD3.fNOuterHits;
332	}
333	fTotal.SetTitle(Form("Total: %5d/51200 (%3d%%)",
334	total, Int_t(100. / 51200 * total)));
335	fSummary->Modified();
336	fCanvas->Modified();
337	fCanvas->Update();
338	fCanvas->cd();
339	fWait = kTRUE;
340	while (fWait) {
341	gApplication->StartIdleing();
342	gSystem->InnerLoop();
343	gApplication->StopIdleing();
344	}
345	return AliFMDInput::End();
346	}
347
348	//____________________________________________________________________
349	Bool_t
350	AliFMDFancy::ProcessHit(AliFMDHit* hit, TParticle*)
351	{
352	AddMarker(hit->Detector(), hit->Ring(), hit->Sector(), hit->Strip(),
353	hit, hit->Edep(), 0);
354	return kTRUE;
355	}
356	//____________________________________________________________________
357	void
358	AliFMDFancy::Detector::AddMarker(Char_t rng, UShort_t sec, UShort_t str,
359	Float_t, Float_t)
360	{
361	AliFMDGeometry* geom = AliFMDGeometry::Instance();
362	Double_t x, y, z;
363	geom->Detector2XYZ(fId, rng, sec, str, x, y, z);
364	if (true) {
365	AliFMDRing* r = geom->GetRing(rng);
366	Double_t t = .9 * r->GetTheta() / 2;
367	Double_t a = gRandom->Uniform(-t,t) * TMath::Pi() / 180;
368	Double_t x1 = x * TMath::Cos(a) - y * TMath::Sin(a);
369	Double_t y1 = x * TMath::Sin(a) + y * TMath::Cos(a);
370	x = x1;
371	y = y1;
372	}
373	switch (rng) {
374	case 'I':
375	case 'i': fInnerHits.SetPoint(fNInnerHits++, x, y, z); break;
376	case 'O':
377	case 'o': fOuterHits.SetPoint(fNOuterHits++, x, y, z); break;
378	default: return;
379	}
380	}
381
382
383
384	//____________________________________________________________________
385	void
386	AliFMDFancy::AddMarker(UShort_t det, Char_t rng, UShort_t sec, UShort_t str,
387	TObject*, Float_t, Float_t)
388	{
389	// Add a marker to the display
390	//
391	// det Detector
392	// rng Ring
393	// sec Sector
394	// str Strip
395	// o Object to refer to
396	// s Signal
397	// max Maximum of signal
398	//
399	switch (det) {
400	case 1: fFMD1.AddMarker(rng,sec,str,0,0); break;
401	case 2: fFMD2.AddMarker(rng,sec,str,0,0); break;
402	case 3: fFMD3.AddMarker(rng,sec,str,0,0); break;
403	}
404	}
405
406	//____________________________________________________________________
407	//
408	// EOF
409	//