PMD/AliPMD.cxx

   1 /**************************************************************************
   2  * Copyright(c) 1998-1999, 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
  16 /* $Id$ */
  17
  18 ///////////////////////////////////////////////////////////////////////////////
  19 //
  20 //                                                                           //
  21 //  Photon Multiplicity Detector                                             //
  22 //  This class contains the basic functions for the Photon Multiplicity      //
  23 //  Detector. Functions specific to one particular geometry are              //
  24 //  contained in the derived classes                                         //
  25 //                                                                           //
  26 //Begin_Html
  27 /*
  28 <img src="picts/AliPMDClass.gif">
  29 </pre>
  30 <br clear=left>
  31 <font size=+2 color=red>
  32 <p>The responsible person for this module is
  33 <a href="mailto:sub@vecdec.veccal.ernet.in">Subhasis Chattopadhyay</a>.
  34 </font>
  35 <pre>
  36 */
  37 //End_Html
  38 //                                                                           //
  39 ///////////////////////////////////////////////////////////////////////////////
  40
  41 #include <TBRIK.h>
  42 #include <TClonesArray.h>
  43 #include <TFile.h>
  44 #include <TGeometry.h>
  45 #include <TNode.h>
  46 #include <TTree.h>
  47 #include <TVirtualMC.h>
  48
  49 #include "AliConst.h"
  50 #include "AliLoader.h"
  51 #include "AliPMD.h"
  52 #include "AliPMDRecPoint.h"
  53 #include "AliRun.h"
  54
  55 ClassImp(AliPMD)
  56
  57 //_____________________________________________________________________________
  58 AliPMD::AliPMD()
  59 {
  60   //
  61   // Default constructor
  62   //
  63   fIshunt = 0;
  64
  65   fRecPoints  = 0;
  66
  67 }
  68
  69 //_____________________________________________________________________________
  70 AliPMD::AliPMD(const char *name, const char *title)
  71   : AliDetector(name,title)
  72 {
  73   //
  74   // Default constructor
  75   //
  76
  77   //
  78   // Allocate the array of hits
  79   fHits   = new TClonesArray("AliPMDhit",  405);
  80   gAlice->AddHitList(fHits);
  81
  82   fRecPoints  = new TClonesArray("AliPMDRecPoint",10000);
  83   fNRecPoints = 0;
  84
  85
  86   fIshunt =  0;
  87
  88   fPar[0] = 1;
  89   fPar[1] = 1;
  90   fPar[2] = 0.8;
  91   fPar[3] = 0.02;
  92   fIn[0]  = 6;
  93   fIn[1]  = 20;
  94   fIn[2]  = 600;
  95   fIn[3]  = 27;
  96   fIn[4]  = 27;
  97   fGeo[0] = 0;
  98   fGeo[1] = 0.2;
  99   fGeo[2] = 4;
 100   fPadSize[0] = 0.8;
 101   fPadSize[1] = 1.0;
 102   fPadSize[2] = 1.2;
 103   fPadSize[3] = 1.5;
 104 }
 105
 106 AliLoader* AliPMD::MakeLoader(const char* topfoldername)
 107 {
 108  cout<<"AliPMD::MakeLoader ";
 109
 110  fLoader = new AliPMDLoader(GetName(),topfoldername);
 111
 112  if (fLoader)
 113   {
 114    cout<<"Success"<<endl;
 115   }
 116  else
 117   {
 118    cout<<"Failure"<<endl;
 119   }
 120
 121  return fLoader;
 122 }
 123
 124 AliPMD::~AliPMD()
 125 {
 126   //
 127   // Default constructor
 128   //
 129     delete fRecPoints;
 130     fNRecPoints=0;
 131 }
 132
 133 //_____________________________________________________________________________
 134 void AliPMD::AddHit(Int_t track, Int_t *vol, Float_t *hits)
 135 {
 136   //
 137   // Add a PMD hit
 138   //
 139   TClonesArray &lhits = *fHits;
 140   AliPMDhit *newcell, *curcell;
 141   //  printf("PMD++ Adding energy %f, prim %d, vol %d %d %d %d %d %d %d %d\n",
 142   // hits[3],gAlice->GetPrimary(track-1),vol[0],vol[1],vol[2],vol[3],
 143   // vol[4],vol[5],vol[6],vol[7]);
 144
 145   newcell = new AliPMDhit(fIshunt, track, vol, hits);
 146   Int_t i;
 147   for (i=0; i<fNhits; i++) {
 148     //
 149     // See if this cell has already been hit
 150     curcell=(AliPMDhit*) lhits[i];
 151     if (*curcell==*newcell) {
 152 //        printf("Cell with same numbers found\n") ; curcell->Print();
 153       *curcell = *curcell+*newcell;
 154 //        printf("Cell after addition\n") ; curcell->Print();
 155       delete newcell;
 156       return;
 157     }
 158   }
 159   new(lhits[fNhits++]) AliPMDhit(newcell);
 160   delete newcell;
 161 }
 162
 163 //_____________________________________________________________________________
 164 void AliPMD::BuildGeometry()
 165 {
 166   //
 167   // Build simple ROOT TNode geometry for event display
 168   //
 169
 170   TNode *Node, *Top;
 171   const int kColorPMD  = kRed;
 172
 173   //
 174   Top=gAlice->GetGeometry()->GetNode("alice");
 175
 176   // PMD
 177   new TBRIK("S_PMD","PMD box","void",300,300,5);
 178   Top->cd();
 179   Node = new TNode("PMD","PMD","S_PMD",0,0,-600,"");
 180   Node->SetLineColor(kColorPMD);
 181   fNodes->Add(Node);
 182 }
 183
 184 //_____________________________________________________________________________
 185 Int_t AliPMD::DistancetoPrimitive(Int_t , Int_t )
 186 {
 187   //
 188   // Distance from mouse to detector on the screen
 189   // dummy routine
 190   //
 191    return 9999;
 192 }
 193
 194 //_____________________________________________________________________________
 195 void AliPMD::SetPAR(Float_t p1, Float_t p2, Float_t p3,Float_t p4)
 196 {
 197   //
 198   // Set PMD parameters
 199   //
 200   fPar[0] = p1;
 201   fPar[1] = p2;
 202   fPar[2] = p3;
 203   fPar[3] = p4;
 204 }
 205
 206 //_____________________________________________________________________________
 207 void AliPMD::SetIN(Float_t p1, Float_t p2, Float_t p3,Float_t p4,Float_t p5)
 208 {
 209   //
 210   // Set PMD parameters
 211   //
 212   fIn[0] = p1;
 213   fIn[1] = p2;
 214   fIn[2] = p3;
 215   fIn[3] = p4;
 216   fIn[4] = p5;
 217 }
 218
 219 //_____________________________________________________________________________
 220 void AliPMD::SetGEO(Float_t p1, Float_t p2, Float_t p3)
 221 {
 222   //
 223   // Set geometry parameters
 224   //
 225   fGeo[0] = p1;
 226   fGeo[1] = p2;
 227   fGeo[2] = p3;
 228 }
 229
 230 //_____________________________________________________________________________
 231 void AliPMD::SetPadSize(Float_t p1, Float_t p2, Float_t p3,Float_t p4)
 232 {
 233   //
 234   // Set pad size
 235   //
 236   fPadSize[0] = p1;
 237   fPadSize[1] = p2;
 238   fPadSize[2] = p3;
 239   fPadSize[3] = p4;
 240 }
 241
 242 //_____________________________________________________________________________
 243 void AliPMD::StepManager()
 244 {
 245   //
 246   // Called at every step in PMD
 247   //
 248 }
 249
 250 void AliPMD::AddRecPoint(const AliPMDRecPoint &p)
 251 {
 252     //
 253     // Add a PMD reconstructed hit to the list
 254     //
 255     TClonesArray &lrecpoints = *fRecPoints;
 256     new(lrecpoints[fNRecPoints++]) AliPMDRecPoint(p);
 257 }
 258
 259 void AliPMD::MakeBranch(Option_t* option)
 260 {
 261     // Create Tree branches for the PMD
 262
 263     const char *cR = strstr(option,"R");
 264     const char *cH = strstr(option,"H");
 265     if (cH && fLoader->TreeH() && (fHits == 0x0))
 266       fHits   = new TClonesArray("AliPMDhit",  405);
 267
 268     AliDetector::MakeBranch(option);
 269
 270     if (cR  && fLoader->TreeR()) {
 271       printf("Make Branch - TreeR address %p\n",fLoader->TreeR());
 272
 273       const Int_t kBufferSize = 4000;
 274       char branchname[30];
 275
 276       sprintf(branchname,"%sRecPoints",GetName());
 277       if (fRecPoints == 0x0) {
 278         fRecPoints  = new TClonesArray("AliPMDRecPoint",10000);
 279       }
 280       MakeBranchInTree(fLoader->TreeR(), branchname, &fRecPoints, kBufferSize,0);
 281    }
 282 }
 283
 284
 285 void AliPMD::SetTreeAddress()
 286 {
 287   // Set branch address for the TreeR
 288     char branchname[30];
 289
 290     if (fLoader->TreeH() && fHits==0x0)
 291       fHits   = new TClonesArray("AliPMDhit",  405);
 292
 293     AliDetector::SetTreeAddress();
 294
 295     TBranch *branch;
 296     TTree *treeR = fLoader->TreeR();
 297
 298     sprintf(branchname,"%s",GetName());
 299     if (treeR) {
 300        branch = treeR->GetBranch(branchname);
 301        if (branch)
 302        {
 303          if (fRecPoints == 0x0) {
 304            fRecPoints  = new TClonesArray("AliPMDRecPoint",10000);
 305          }
 306          branch->SetAddress(&fRecPoints);
 307        }
 308     }
 309 }
 310
 311 void AliPMD::ResetHits()
 312 {
 313   //
 314   // Reset number of hits and the hits array
 315   //
 316     AliDetector::ResetHits();
 317     fNRecPoints   = 0;
 318     if (fRecPoints)   fRecPoints->Clear();
 319 }
 320
 321 ///////////////////////////////////////////////////////////////////////////////
 322 //                                                                           //
 323 //  Photon Multiplicity Detector Version 1                                   //
 324 //                                                                           //
 325 //Begin_Html
 326 /*
 327 <img src="picts/AliPMDv1Class.gif">
 328 */
 329 //End_Html
 330 //                                                                           //
 331 ///////////////////////////////////////////////////////////////////////////////
 332
 333
 334
 335 ClassImp(AliPMDhit)
 336
 337 //_____________________________________________________________________________
 338 AliPMDhit::AliPMDhit(Int_t shunt,Int_t track, Int_t *vol, Float_t *hits):
 339   AliHit(shunt, track)
 340 {
 341   //
 342   // Add a PMD hit
 343   //
 344   Int_t i;
 345   for (i=0;i<8;i++) fVolume[i] = vol[i];
 346   fX=hits[0];
 347   fY=hits[1];
 348   fZ=hits[2];
 349   fEnergy=hits[3];
 350 }
 351