SDigitizer and Merger added

[u/mrichter/AliRoot.git] / TOF / AliTOFSDigitizer.cxx

/**************************************************************************
 * Copyright(c) 1998-1999, 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.                  *
 **************************************************************************/

//_________________________________________________________________________
// This is a TTask that constructs SDigits out of Hits
// A Summable Digits is the sum of all hits in a pad
// 
//
//-- Author: F. Pierella
//////////////////////////////////////////////////////////////////////////////


#include "TTask.h"
#include "TTree.h"
#include "TSystem.h"
#include "TFile.h"

#include "AliTOFdigit.h"
#include "AliTOFhit.h"
#include "AliTOF.h"
#include "AliTOFv1.h"
#include "AliTOFv2.h"
#include "AliTOFv3.h"
#include "AliTOFv4.h"
#include "AliTOFSDigitizer.h"
#include "AliRun.h"
#include "AliDetector.h"
#include "AliMC.h"

#include "TFile.h"
#include "TTask.h"
#include "TTree.h"
#include "TSystem.h"
#include "TROOT.h"
#include "TFolder.h"
#include <stdlib.h>
#include <iostream.h>
#include <fstream.h>

ClassImp(AliTOFSDigitizer)

//____________________________________________________________________________ 
  AliTOFSDigitizer::AliTOFSDigitizer():TTask("AliTOFSDigitizer","") 
{
  // ctor
  fNevents = 0 ;     
  fSDigits = 0 ;
  fHits = 0 ;

}
           
//____________________________________________________________________________ 
  AliTOFSDigitizer::AliTOFSDigitizer(char* HeaderFile,char *SdigitsFile ):TTask("AliTOFSDigitizer","") 
{
  fNevents = 0 ;     // Number of events to digitize, 0 means all evens in current file
  // add Task to //root/Tasks folder
  TTask * roottasks = (TTask*)gROOT->GetRootFolder()->FindObject("Tasks") ; 
  roottasks->Add(this) ; 
}

//____________________________________________________________________________ 
  AliTOFSDigitizer::~AliTOFSDigitizer()
{
  // dtor
}

//____________________________________________________________________________
void AliTOFSDigitizer::Exec(Option_t *option) { 


  // Initialise Hit array
  fHits = new TClonesArray ("AliTOFhit", 1000);
  fSDigits = new TClonesArray ("AliTOFdigit", 1000);

  AliTOF *TOF = (AliTOF *) gAlice->GetDetector ("TOF");

  if (fNevents == 0)
    fNevents = (Int_t) gAlice->TreeE ()->GetEntries ();

  for (Int_t ievent = 0; ievent < fNevents; ievent++)
    {
      gAlice->GetEvent (ievent);
      if (gAlice->TreeH () == 0)
	return;
      if (gAlice->TreeS () == 0)
	gAlice->MakeTree ("S");

      
            //Make branches
      char branchname[20];
       sprintf (branchname, "%s", TOF->GetName ());
      //Make branch for digits
        TOF->MakeBranch ("S");
    
       //Now made SDigits from hits

      Int_t    tracks[3];    // track info
      Int_t    vol[5];       // location for a digit
      Float_t  digit[2];     // TOF digit variables
      Int_t hit, nbytes;
      TParticle *particle;
      AliTOFhit *tofHit;
      TClonesArray *TOFhits = TOF->Hits();


      // Event ------------------------- LOOP  


      if (TOF)
	{
	  TOFhits = TOF->Hits ();
	  TTree *TH = gAlice->TreeH ();
	  Stat_t ntracks = TH->GetEntries ();
	  for (Int_t track = 0; track < ntracks; track++)
	    {
	      gAlice->ResetHits ();
	      nbytes += TH->GetEvent (track);
	      particle = gAlice->Particle (track);
	      Int_t nhits = TOFhits->GetEntriesFast ();

	      for (hit = 0; hit < nhits; hit++)
		{
		  tofHit = (AliTOFhit *) TOFhits->UncheckedAt(hit);
	        vol[0] = tofHit->GetSector();
	        vol[1] = tofHit->GetPlate();
	        vol[2] = tofHit->GetPadx();
	        vol[3] = tofHit->GetPadz();
	        vol[4] = tofHit->GetStrip();

	        // 95% of efficiency to be inserted here
	        // edge effect to be inserted here
	        // cross talk  to be inserted here

	        Float_t idealtime = tofHit->GetTof(); // unit s
	        idealtime *= 1.E+12;  // conversion from s to ps
                              // fTimeRes is given usually in ps
//	        Float_t tdctime   = gRandom->Gaus(idealtime, fTimeRes);	
	        Float_t tdctime   = gRandom->Gaus(idealtime, TOF->GetTimeRes());	
	        digit[0] = tdctime;

	        // typical Landau Distribution to be inserted here
	        // instead of Gaussian Distribution
	        Float_t idealcharge = tofHit->GetEdep();
//	        Float_t adccharge = gRandom->Gaus(idealcharge, fChrgRes);
	        Float_t adccharge = gRandom->Gaus(idealcharge, TOF->GetChrgRes());
	        digit[1] = adccharge;
	        Int_t tracknum = tofHit->GetTrack();
	        tracks[0] = tracknum;
		tracks[1] = 0;
		tracks[2] = 0;

		// check if two digit are on the same pad; in that case we sum
		// the two or more digits
//	        Bool_t overlap = AliTOF::CheckOverlap(vol, digit, tracknum);
	        Bool_t overlap = TOF->CheckOverlap(vol, digit, tracknum);
	        if(!overlap) 
		  //			new ((*fSDigits)[nSdigits++]) AliTOFdigit(tracks, vol, digit);
		  TOF->AddSDigit(tracks, vol, digit);
     		cout << "nSdigits" << endl; 
	        } // end loop on hits for the current track

	     } // end loop on ntracks

	  } // close if TOF switched ON
      
      gAlice->TreeS()->Reset();
      gAlice->TreeS()->Fill();
      gAlice->TreeS()->Write(0,TObject::kOverwrite) ;
    }				//event loop


}
 
//__________________________________________________________________
void AliTOFSDigitizer::SetSDigitsFile(char * file ){
  if(!fSDigitsFile.IsNull())
    cout << "Changing SDigits file from " <<(char *)fSDigitsFile.Data() << " to " << file << endl ;
  fSDigitsFile=file ;
}
//__________________________________________________________________
void AliTOFSDigitizer::Print(Option_t* option)const
{
  cout << "------------------- "<< GetName() << " -------------" << endl ;
  if(fSDigitsFile.IsNull())
    cout << " Writing SDigitis to file galice.root "<< endl ;
  else
    cout << "    Writing SDigitis to file  " << (char*) fSDigitsFile.Data() << endl ;

}