]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - START/AliSTART.cxx
Options for cuts automatically set when chosing processing option
[u/mrichter/AliRoot.git] / START / AliSTART.cxx
index 9c0091d9f272f82a222765c46e79a6bf8c5572a4..eecdddab06a962bed24c53574f9996dd32f052e6 100755 (executable)
  * provided "as is" without express or implied warranty.                  *
  **************************************************************************/
 
-/*
-$Log$
-Revision 1.19  2001/04/04 12:10:18  alla
-changes according Coding Convension
-
-Revision 1.18  2001/03/12 17:46:43  hristov
-Changes needed on Sun with CC 5.0
-
-Revision 1.17  2001/01/26 19:59:53  hristov
-Major upgrade of AliRoot code
-
-Revision 1.16  2001/01/17 10:56:08  hristov
-Corrections to destructors
-
-Revision 1.15  2001/01/01 13:10:42  hristov
-Local definition of digits removed
-
-Revision 1.14  2000/12/22 16:17:15  hristov
-Updated  START code from Alla
-
-Revision 1.13  2000/12/18 11:39:41  alibrary
-Quick fix to avoid crash in display waiting for new version
-
-Revision 1.12  2000/12/04 08:48:19  alibrary
-Fixing problems in the HEAD
-
-Revision 1.11  2000/10/13 13:14:08  hristov
-Bug fixes and code cleaning
-
-Revision 1.10  2000/10/02 21:28:13  fca
-Removal of useless dependecies via forward declarations
-
-Revision 1.9  2000/07/13 16:41:29  fca
-New START corrected for coding conventions
-
-Revision 1.8  2000/03/27 17:24:25  alla
-Modifing geometry
-
-Revision 1.6  2000/01/21 15:45:23  fca
-New Version from Alla
-
-Revision 1.5  2000/01/19 17:17:15  fca
-Introducing a list of lists of hits -- more hits allowed for detector now
-
-Revision 1.4  1999/11/12 15:04:00  fca
-Modifications from A.Maevskaya
-
-Revision 1.3  1999/09/29 09:24:29  fca
-Introduction of the Copyright and cvs Log
-
-*/
+/* $Id$ */
 
 ///////////////////////////////////////////////////////////////////////////////
 //                                                                           //
@@ -87,23 +37,27 @@ Introduction of the Copyright and cvs Log
 //                                                                           //
 ///////////////////////////////////////////////////////////////////////////////
 
-#include <iostream.h>
-#include <fstream.h>
+#include <Riostream.h>
 
-#include "TMath.h"
-#include "TTUBE.h"
-#include "TNode.h"
-#include "TRandom.h"
-#include "TGeometry.h"
-#include "TFile.h"
-#include "TParticle.h"
+#include <TFile.h>
+#include <TGeometry.h>
+#include <TMath.h>
+#include <TNode.h>
+#include <TParticle.h>
+#include <TRandom.h>
+#include <TTUBE.h>
+#include <TVirtualMC.h>
+#include <AliESD.h>
 
+#include "AliLoader.h"
 #include "AliRun.h"
 #include "AliSTART.h"
+#include "AliSTARTLoader.h"
 #include "AliSTARTdigit.h"
-#include "AliMC.h"
 #include "AliSTARThit.h"
-#include "AliSTARTvertex.h"
+#include "AliSTARThitPhoton.h"
+#include "AliMC.h"
+#include "AliSTARTDigitizer.h"
 
 ClassImp(AliSTART)
 
@@ -118,6 +72,7 @@ AliSTART::AliSTART()
   fIshunt   = 1;
   fHits     = 0;
   fDigits   = 0;
+  fPhotons  = 0;
 }
  
 //_____________________________________________________________________________
@@ -132,10 +87,14 @@ AliSTART::AliSTART(const char *name, const char *title)
   //
   // Initialise Hit array
   fHits       = new TClonesArray("AliSTARThit",  405);
-  gAlice->AddHitList(fHits);
-  
+  gAlice->GetMCApp()->AddHitList(fHits);
+
+  fPhotons  = new TClonesArray("AliSTARThitPhoton", 10000);
+  gAlice->GetMCApp()->AddHitList (fPhotons);
+  if (GetDebug()>2) cout<<" Debug "<<endl;
   fIshunt     =  1;
   fIdSens   =  0;
+  fNPhotons =  0;
   SetMarkerColor(kRed);
 }
 
@@ -145,6 +104,10 @@ AliSTART::~AliSTART() {
     fHits->Delete();
     delete fHits;
   }
+  if (fPhotons) {
+    fPhotons->Delete();
+    delete fPhotons;
+  }
 }
  
 //_____________________________________________________________________________
@@ -158,8 +121,17 @@ void AliSTART::AddHit(Int_t track, Int_t *vol, Float_t *hits)
 }
 
 //_____________________________________________________________________________
+void AliSTART::AddHitPhoton(Int_t track, Int_t *vol, Float_t *hits)
+{
+  //  Add a START hit of photons
+  
+  TClonesArray &lhits = *fPhotons;
+  new(lhits[fNPhotons++]) AliSTARThitPhoton(fIshunt,track,vol,hits);
+}
 
-void AliSTART::AddDigit(Int_t *tracks,Int_t *digits)
+//_____________________________________________________________________________
+
+void AliSTART::AddDigit(Int_t * /*tracks*/, Int_t * /*digits*/)
 {
   
   //  Add a START digit to the list. Dummy function.
@@ -178,23 +150,23 @@ void AliSTART::BuildGeometry()
   top=gAlice->GetGeometry()->GetNode("alice");
 
   // START define the different volumes
-  new TRotMatrix("rot999","rot999",  90,0,90,90,180,0);
+  new TRotMatrix("rotx999","rot999",  90,0,90,90,180,0);
 
   new TTUBE("S_0ST1","START  volume 1","void",5.,10.7,5.3);
   top->cd();
-  node = new TNode("0ST1","0ST01","S_0ST1",0,0,75.,"");
+  node = new TNode("0ST1","0ST01","S_0ST1",0,0,-69.7,"");
   node->SetLineColor(kColorSTART);
   fNodes->Add(node);
 
   new TTUBE("S_0ST2","START volume 2","void",5.,10.7,5.3);
   top->cd();
-  node = new TNode("0ST2","0ST2","S_0ST2",0,0,-75,"rot999");
+  node = new TNode("0ST2","0ST2","S_0ST2",0,0,350,"rotx999");
   node->SetLineColor(kColorSTART);
   fNodes->Add(node);
 }
  
 //_____________________________________________________________________________
-Int_t AliSTART::DistanceToPrimitive(Int_t px, Int_t py)
+Int_t AliSTART::DistanceToPrimitive(Int_t /*px*/, Int_t /*py*/)
 {
   //
   // Calculate the distance from the mouse to the START on the screen
@@ -224,160 +196,76 @@ void AliSTART::Init()
 }
 
 //---------------------------------------------------------------------------
-void AliSTART::MakeBranch(Option_t* option, const char *file)
+void AliSTART::MakeBranch(Option_t* option)
 {
   //
   // Specific START branches
   //
   // Create Tree branches for the START.
-  Int_t buffersize = 400;
-  char branchname[10];
+  Int_t buffersize = 4000;
+  char branchname[20];
   sprintf(branchname,"%s",GetName());
 
-  AliDetector::MakeBranch(option,file);
 
   const char *cD = strstr(option,"D");
+  const char *cH = strstr(option,"H");
+  
+  if (cH && fLoader->TreeH())
+  {
+     if (fPhotons == 0x0) fPhotons  = new TClonesArray("AliSTARThitPhoton", 10000);
+     sprintf (branchname, "%shitPhoton", GetName());
+     MakeBranchInTree (fLoader->TreeH(), branchname, &fPhotons, 50000, 0);
+     if (fHits == 0x0) fHits  = new TClonesArray("AliSTARThit",  405);
+  } 
   
+  AliDetector::MakeBranch(option);
+
   if (cD) {
     digits = new AliSTARTdigit();
-    MakeBranchInTree(gAlice->TreeD(), 
-                     branchname, "AliSTARTdigit", digits, buffersize, 1, file);
+    MakeBranchInTree(fLoader->TreeD(), branchname, "AliSTARTdigit", digits, buffersize, 1, 0);
   } 
-/*
-  char *cR = strstr(option,"R");
-  
-  if (cR)   {  
-    MakeBranchInTree(gAlice->TreeR(), 
-                     branchname, "Int_t", &fZposit, buffersize, 1, file);
-  }
-  */
 }    
 
 //_____________________________________________________________________________
-
-void AliSTART::Hit2digit(Int_t evnum) 
+void AliSTART::ResetHits()
 {
-  //
-  // From hits to digits
-  //
-  Float_t x,y,e;
-  Int_t nbytes = 0;
-  Int_t hit;
-  Int_t nhits;
-  Int_t volume,pmt;
-  char nameTH[8],nameTD[8];
-  Float_t timediff,timeright,timeleft,timeav;
-  Float_t besttimeright,besttimeleft,meanTime;
-  Int_t channelWidth=10;
-
-  TParticle *particle;
-  AliSTARThit  *startHit;
-
-  Int_t buffersize=256;
-  Int_t split=1;
-
-  digits= new AliSTARTdigit();
-  TBranch *bDig=0;
-
+  AliDetector::ResetHits();
   
- // Event ------------------------- LOOP  
-    sprintf(nameTD,"TreeD%d",evnum);
-    TTree *td = new TTree(nameTD,"START");
-    bDig = td->Branch("START","AliSTARTdigit",&digits,buffersize,split);
-
-    besttimeright=9999.;
-    besttimeleft=9999.;
-    Int_t timeDiff=0;
-    Int_t timeAv=0;
-
-    Int_t nparticles = gAlice->GetEvent(evnum);
-    if (nparticles <= 0) return;
-    printf("\n nparticles %d\n",nparticles);
-    
-   
-    sprintf(nameTH,"TreeH%d",evnum);
-    printf("%s\n",nameTH);
-    TTree *th = gAlice->TreeH();
-    Int_t ntracks    = (Int_t) th->GetEntries();
-    if (ntracks<=0) return;
-    // Start loop on tracks in the hits containers
-    for (Int_t track=0; track<ntracks;track++) {
-      gAlice->ResetHits();
-      nbytes += th->GetEvent(track);
-      particle=gAlice->Particle(track);
-      nhits =fHits->GetEntriesFast();
-      
-      for (hit=0;hit<nhits;hit++) {
-       startHit   = (AliSTARThit*)fHits->UncheckedAt(hit);
-       pmt=startHit->fPmt;
-       e=startHit->fEtot;
-       x=startHit->X();
-       y=startHit->Y();
-       volume = startHit->fVolume;
-       if(volume==1){
-         timeright = startHit->fTime;
-         if(timeright<besttimeright) {
-           besttimeright=timeright;
-         } //timeright
-       }//time for right shoulder
-       if(volume==2){            
-         timeleft = startHit->fTime;
-         //                printf("timeleft %f\n",timeleft);
-         if(timeleft<besttimeleft) {
-           besttimeleft=timeleft;
-         } //timeleftbest
-       }//time for left shoulder
-      } //hit loop
-    } //track loop
+  fNPhotons = 0;
+  if (fPhotons)  fPhotons->Clear();
+}
 
-    //folding with experimental time distribution
-    Float_t besttimerightGaus=gRandom->Gaus(besttimeright,0.05);
-    Float_t besttimeleftGaus=gRandom->Gaus(besttimeleft,0.05);
-    timediff=besttimerightGaus-besttimeleftGaus;
-    meanTime=(besttimerightGaus+besttimeleftGaus)/2.;
-    if ( TMath::Abs(timediff)<2. && meanTime<3.) 
-     {
-       //we assume centre of bunch is 5ns after TTS signal
-       //TOF values are relative of the end of bunch
-       Float_t ppBunch=25;
+//_____________________________________________________________________________
+void AliSTART::SetTreeAddress()
+{
+  TBranch  *branch;
+  TTree    *treeH;
+  
+  treeH = TreeH();
+  
+  if (treeH)
+    {
+      if (fPhotons == 0x0) fPhotons  = new TClonesArray("AliSTARThitPhoton", 10000);
+      branch = treeH->GetBranch("STARThitPhoton");
+      if (branch)  branch->SetAddress(&fPhotons);
+      if (fHits == 0x0) fHits  = new TClonesArray("AliSTARThit",  405);
+    }
     
-       ppBunch=ppBunch-10/2;
-       Float_t t1=1000.*besttimeleftGaus;
-       Float_t t2=1000.*besttimerightGaus;
-       t1=t1/channelWidth+ppBunch; //time in ps to channelWidth
-       t2=t2/channelWidth+ppBunch; //time in ps to channelWidth
-     
-       timeav=(t1+t2)/2.;
-     
-       // Time to TDC signal
-       // 256 channels for timediff, range 1ns
-       
-       timediff=128+1000*timediff/channelWidth; // time in ps 
-
-       timeAv = (Int_t)(timeav);   // time (ps) channel numbres
-       timeDiff = (Int_t)(timediff); // time ( ps) channel numbres
-       digits->Set(timeAv,timeDiff);
-     }
-    else
-      {timeAv=999999; timeDiff=99999;}
-      
-    td->Fill();
-    printf("digits-> %d \n",digits->GetTime());
-    td->Write();
-
+  AliDetector::SetTreeAddress();
+  
 }
 
+//______________________________________________________________________
+AliLoader* AliSTART::MakeLoader(const char* topfoldername)
+{ 
+  Info("MakeLoader", "Creating AliSTARTLoader. Top folder is %s.", topfoldername);
+  fLoader = new AliSTARTLoader(GetName(), topfoldername);
+  return fLoader;
+}
 
-
-
-
-
-
-
-
-
-
-
-
+//_____________________________________________________________________________
+AliDigitizer* AliSTART::CreateDigitizer(AliRunDigitizer* manager) const
+{
+  return new AliSTARTDigitizer(manager);
+}