[u/mrichter/AliRoot.git] / RICH / AliRICHMerger.cxx

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

/*
$Log$
Revision 1.3  2001/05/16 14:57:20  alibrary
New files for folders and Stack

Revision 1.2  2001/03/14 18:16:08  jbarbosa
Corrected bug (more to correct).
File "points.dat" is no longer created.

Revision 1.1  2001/02/27 22:13:34  jbarbosa
Implementing merger class.

*/

#include <TTree.h> 
#include <TVector.h>
#include <TObjArray.h>
#include <TFile.h>
#include <TDirectory.h>
#include <TParticle.h>


// #include "AliMerger.h"
// #include "AliMergable.h"
#include "AliRICHMerger.h"
#include "AliRICHChamber.h"
#include "AliHitMap.h"
#include "AliRICHHitMapA1.h"
#include "AliRICH.h"
#include "AliRICHHit.h"
#include "AliRICHSDigit.h"
#include "AliRICHDigit.h"
#include "AliRICHTransientDigit.h"
#include "AliRun.h"
#include "AliPDG.h"

ClassImp(AliRICHMerger)

//___________________________________________
AliRICHMerger::AliRICHMerger()
{
// Default constructor    
    fEvNrSig = 0;
    fEvNrBgr = 0;
    fMerge   = kDigitize;
    fFnBgr   = 0;
    fHitMap  = 0;
    fList    = 0;
    fTrH1    = 0;
    fHitsBgr = 0;
    fSDigitsBgr = 0;
    fHitMap     = 0;
    fList       = 0;
    fTrList     = 0;
    fAddress    = 0;
    fBgrFile    = 0;
}

//------------------------------------------------------------------------
AliRICHMerger::~AliRICHMerger()
{
// Destructor
    if (fTrH1)       delete fTrH1;
    if (fHitsBgr)    delete fHitsBgr;
    if (fSDigitsBgr) delete fSDigitsBgr;
    if (fHitMap)     delete fHitMap;
    if (fList)       delete fList;
    if (fTrList)     delete fTrList;
    if (fAddress)    delete fAddress; 
}

//------------------------------------------------------------------------
Bool_t AliRICHMerger::Exists(const AliRICHSDigit *mergable)
{
    AliRICHSDigit *padhit = (AliRICHSDigit*) mergable;
    return (fHitMap[fNch]->TestHit(padhit->PadX(),padhit->PadY()));
}

//------------------------------------------------------------------------
void AliRICHMerger::Update(AliRICHSDigit *mergable)
{
    AliRICHSDigit *padhit = (AliRICHSDigit*) mergable;    
    AliRICHTransientDigit* pdigit;
    Int_t ipx      = padhit->PadX();        // pad number on X
    Int_t ipy      = padhit->PadY();        // pad number on Y
    Int_t iqpad    = Int_t(padhit->QPad()); // charge per pad

    pdigit = (AliRICHTransientDigit*) fHitMap[fNch]->GetHit(ipx, ipy);
    // update charge
    //
    (*pdigit).AddSignal(iqpad);
    (*pdigit).AddPhysicsSignal(iqpad);		
    // update list of tracks
    //
    TObjArray* fTrList = (TObjArray*)pdigit->TrackList();
    Int_t lastEntry = fTrList->GetLast();
    TVector *pTrack = (TVector*)fTrList->At(lastEntry);
    TVector &ptrk   = *pTrack;
    TVector &trinfo = *((TVector*) (*fAddress)[fCountadr-1]);
    Int_t lastTrack = Int_t(ptrk(0));

    if (trinfo(0) != kBgTag) {
	if (lastTrack == fTrack) {
	    Int_t lastCharge = Int_t(ptrk(1));
	    lastCharge += iqpad;
	    fTrList->RemoveAt(lastEntry);
	    trinfo(1) = lastCharge;
	    fTrList->AddAt(&trinfo,lastEntry);
	} else {
	    fTrList->Add(&trinfo);
	}
    } else {
	if (lastTrack != -1) fTrList->Add(&trinfo);
    }
}

//------------------------------------------------------------------------
void AliRICHMerger::CreateNew(AliRICHSDigit *mergable)
{
    AliRICHSDigit *padhit = (AliRICHSDigit*) mergable;    
    AliRICHTransientDigit* pdigit;

    Int_t ipx      = padhit->PadX();       // pad number on X
    Int_t ipy      = padhit->PadY();       // pad number on Y
    fList->AddAtAndExpand(
	new AliRICHTransientDigit(fNch,fDigits),fCounter);
    fHitMap[fNch]->SetHit(ipx, ipy, fCounter);
    fCounter++;
    pdigit = (AliRICHTransientDigit*)fList->At(fList->GetLast());
    // list of tracks
    TObjArray *fTrList = (TObjArray*)pdigit->TrackList();
    TVector &trinfo    = *((TVector*) (*fAddress)[fCountadr-1]);
    fTrList->Add(&trinfo);
}


//------------------------------------------------------------------------
void AliRICHMerger::Init()
{
// Initialisation
    
    if (fMerge) fBgrFile = InitBgr();
}


//------------------------------------------------------------------------
TFile* AliRICHMerger::InitBgr()
{
// Initialise background event
    TFile *file = new TFile(fFnBgr);
// add error checking later
    printf("\n AliRICHMerger has opened %s file with background event \n", fFnBgr);
    fHitsBgr     = new TClonesArray("AliRICHHit",1000);
    fSDigitsBgr  = new TClonesArray("AliRICHSDigit",1000);
    return file;
}

//------------------------------------------------------------------------
void AliRICHMerger::Digitise(Int_t nev, Int_t flag)
{

 // keep galice.root for signal and name differently the file for 
    // background when add! otherwise the track info for signal will be lost !

  Int_t particle;

  //FILE* points; //these will be the digits...
  
  //points=fopen("points.dat","w");
  
  AliRICHChamber*       iChamber;
  AliSegmentation*  segmentation;
  
    Int_t digitise=0;
    Int_t trk[50];
    Int_t chtrk[50];  
    TObjArray *list=new TObjArray;
    static TClonesArray *pAddress=0;
    if(!pAddress) pAddress=new TClonesArray("TVector",1000);
    Int_t digits[5]; 
    
    AliRICH *pRICH = (AliRICH *) gAlice->GetDetector("RICH");
    AliHitMap* pHitMap[10];
    Int_t i;
    for (i=0; i<10; i++) {pHitMap[i]=0;}
    
    if (fMerge ) {
      fBgrFile->cd();
      //fBgrFile->ls();
      //
      // Get Hits Tree header from file
      if(fHitsBgr) fHitsBgr->Clear();
      if(fSDigitsBgr) fSDigitsBgr->Clear();
      if(fTrH1) delete fTrH1;
      fTrH1 = 0;
      
      char treeName[20];
      sprintf(treeName,"TreeH%d",fEvNrBgr);
      fTrH1 = (TTree*)gDirectory->Get(treeName);
      if (!fTrH1) {
	printf("ERROR: cannot find Hits Tree for event:%d\n",fEvNrBgr);
      }
      //
      // Set branch addresses
      TBranch *branch;
      char branchname[20];
      sprintf(branchname,"%s",pRICH->GetName());
      if (fTrH1 && fHitsBgr) {
	branch = fTrH1->GetBranch(branchname);
	if (branch) branch->SetAddress(&fHitsBgr);
      }
      if (fTrH1 && fSDigitsBgr) {
	branch = fTrH1->GetBranch("RICHSDigits");
	if (branch) branch->SetAddress(&fSDigitsBgr);
      }
    }
    
    AliHitMap* hm;
    Int_t countadr=0;
    Int_t counter=0;
    for (i =0; i<kNCH; i++) {
      iChamber= &(pRICH->Chamber(i));
      segmentation=iChamber->GetSegmentationModel(1);
      pHitMap[i] = new AliRICHHitMapA1(segmentation, list);
    }
    //
    //   Loop over tracks
    //
    
    TTree *treeH = gAlice->TreeH();
    Int_t ntracks =(Int_t) treeH->GetEntries();
    for (Int_t track=0; track<ntracks; track++) {
      gAlice->ResetHits();
      treeH->GetEvent(track);
      //
      //   Loop over hits
      for(AliRICHHit* mHit=(AliRICHHit*)pRICH->FirstHit(-1); 
	  mHit;
	  mHit=(AliRICHHit*)pRICH->NextHit()) 
	{
	  
	  Int_t   nch   = mHit->fChamber-1;  // chamber number
	  Int_t   index = mHit->Track();
	  if (nch >kNCH) continue;
	  iChamber = &(pRICH->Chamber(nch));
	  
	  TParticle *current = (TParticle*)gAlice->Particle(index);
	  
	  if (current->GetPdgCode() >= 50000050)
	    {
	      TParticle *motherofcurrent = (TParticle*)gAlice->Particle(current->GetFirstMother());
	      particle = motherofcurrent->GetPdgCode();
	    }
	  else
	    {
	      particle = current->GetPdgCode();
	    }

	  
	  //printf("Flag:%d\n",flag);
	  //printf("Track:%d\n",track);
	  //printf("Particle:%d\n",particle);
	  
	  digitise=1;
	  
	  if (flag == 1) 
	    if(TMath::Abs(particle)==211 || TMath::Abs(particle)==111)
	      digitise=0;
	  
	  if (flag == 2)
	    if(TMath::Abs(particle)==321 || TMath::Abs(particle)==130 || TMath::Abs(particle)==310 
	       || TMath::Abs(particle)==311)
	      digitise=0;
	  
	  if (flag == 3 && TMath::Abs(particle)==2212)
	    digitise=0;
	  
	  if (flag == 4 && TMath::Abs(particle)==13)
	    digitise=0;
	  
	  if (flag == 5 && TMath::Abs(particle)==11)
	    digitise=0;
	  
	  if (flag == 6 && TMath::Abs(particle)==2112)
	    digitise=0;
	  
	  
	  //printf ("Particle: %d, Flag: %d, Digitise: %d\n",particle,flag,digitise); 
	  
	  
	  if (digitise)
	    {
	      
	      //
	      // Loop over pad hits
	      for (AliRICHSDigit* mPad=
		     (AliRICHSDigit*)pRICH->FirstPad(mHit,pRICH->SDigits());
		   mPad;
		   mPad=(AliRICHSDigit*)pRICH->NextPad(pRICH->SDigits()))
		{
		  Int_t ipx      = mPad->fPadX;       // pad number on X
		  Int_t ipy      = mPad->fPadY;       // pad number on Y
		  Int_t iqpad    = mPad->fQpad;       // charge per pad
		  //
		  //
		  //printf("X:%d, Y:%d, Q:%d\n",ipx,ipy,iqpad);
		  
		  Float_t thex, they, thez;
		  segmentation=iChamber->GetSegmentationModel(0);
		  segmentation->GetPadC(ipx,ipy,thex,they,thez);
		  new((*pAddress)[countadr++]) TVector(2);
		  TVector &trinfo=*((TVector*) (*pAddress)[countadr-1]);
		  trinfo(0)=(Float_t)track;
		  trinfo(1)=(Float_t)iqpad;
		  
		  digits[0]=ipx;
		  digits[1]=ipy;
		  digits[2]=iqpad;
		  
		  AliRICHTransientDigit* pdigit;
		  // build the list of fired pads and update the info
		  if (!pHitMap[nch]->TestHit(ipx, ipy)) {
		    list->AddAtAndExpand(new AliRICHTransientDigit(nch,digits),counter);
		    pHitMap[nch]->SetHit(ipx, ipy, counter);
		    counter++;
		    pdigit=(AliRICHTransientDigit*)list->At(list->GetLast());
		    // list of tracks
		    TObjArray *trlist=(TObjArray*)pdigit->TrackList();
		    trlist->Add(&trinfo);
		  } else {
		    pdigit=(AliRICHTransientDigit*) pHitMap[nch]->GetHit(ipx, ipy);
		    // update charge
		    (*pdigit).fSignal+=iqpad;
		    // update list of tracks
		    TObjArray* trlist=(TObjArray*)pdigit->TrackList();
		    Int_t lastEntry=trlist->GetLast();
		    TVector *ptrkP=(TVector*)trlist->At(lastEntry);
		    TVector &ptrk=*ptrkP;
		    Int_t lastTrack=Int_t(ptrk(0));
		    Int_t lastCharge=Int_t(ptrk(1));
		    if (lastTrack==track) {
		      lastCharge+=iqpad;
		      trlist->RemoveAt(lastEntry);
		      trinfo(0)=lastTrack;
		      trinfo(1)=lastCharge;
		      trlist->AddAt(&trinfo,lastEntry);
		    } else {
		      trlist->Add(&trinfo);
		    }
		    // check the track list
		    Int_t nptracks=trlist->GetEntriesFast();
		    if (nptracks > 2) {
		      printf("Attention - tracks:  %d (>2)\n",nptracks);
		      //printf("cat,nch,ix,iy %d %d %d %d  \n",icat+1,nch,ipx,ipy);
		      for (Int_t tr=0;tr<nptracks;tr++) {
			TVector *pptrkP=(TVector*)trlist->At(tr);
			TVector &pptrk=*pptrkP;
			trk[tr]=Int_t(pptrk(0));
			chtrk[tr]=Int_t(pptrk(1));
		      }
		    } // end if nptracks
		  } //  end if pdigit
		} //end loop over clusters
	    }// track type condition
	} // hit loop
    } // track loop
    
    // open the file with background
    
    if (fMerge) {
      ntracks = (Int_t)fTrH1->GetEntries();
      //
      //   Loop over tracks
      //
      for (fTrack = 0; fTrack < ntracks; fTrack++) {
	
	if (fHitsBgr)       fHitsBgr->Clear();
	if (fSDigitsBgr)    fSDigitsBgr->Clear();
	
	fTrH1->GetEvent(fTrack);
	//
	//   Loop over hits
	AliRICHHit* mHit;
	for(int i = 0; i < fHitsBgr->GetEntriesFast(); ++i) 
	  {	
	    mHit   = (AliRICHHit*) (*fHitsBgr)[i];
	    fNch   = mHit->Chamber()-1;  // chamber number
	    iChamber = &(pRICH->Chamber(fNch));
	    //Float_t xbgr = mHit->X();
	    //Float_t ybgr = mHit->Y();
	    Bool_t cond  = kFALSE;
	    cond  = kTRUE;
	    //
	    // Loop over pad hits
	    for (AliRICHSDigit* mPad =
		   (AliRICHSDigit*)pRICH->FirstPad(mHit,fSDigitsBgr);
		 mPad;
		 mPad = (AliRICHSDigit*)pRICH->NextPad(fSDigitsBgr))
	      {
		Int_t ipx      = mPad->PadX();        // pad number on X
		Int_t ipy      = mPad->PadY();        // pad number on Y
		Int_t iqpad    = Int_t(mPad->QPad()); // charge per pad
		
		new((*fAddress)[fCountadr++]) TVector(2);
		TVector &trinfo = *((TVector*) (*fAddress)[fCountadr-1]);
		trinfo(0) = kBgTag;  // tag background
		trinfo(1) = kBgTag;
		
		fDigits[0] = ipx;
		fDigits[1] = ipy;
		fDigits[3] = iqpad;
		
		// build the list of fired pads and update the info
		if (!Exists(mPad)) {
		  CreateNew(mPad);
		} else {
		  Update(mPad);
		} //  end if !Exists
	      } //end loop over clusters
	  } // hit loop
      } // track loop
      
      TTree *fAli = gAlice->TreeK();
      TFile *file = NULL;
      
      if (fAli) file = fAli->GetCurrentFile();
      file->cd();
    } // if fMerge
    
    Int_t tracks[10];
    Int_t charges[10];
    //cout<<"Start filling digits \n "<<endl;
    Int_t nentries=list->GetEntriesFast();
    //printf(" \n \n nentries %d \n",nentries);
    
    // start filling the digits
    
    for (Int_t nent=0;nent<nentries;nent++) {
      AliRICHTransientDigit *address=(AliRICHTransientDigit*)list->At(nent);
      if (address==0) continue; 
      
      Int_t ich=address->fChamber;
      Int_t q=address->fSignal; 
      iChamber=&(pRICH->Chamber(ich));
      AliRICHResponse * response=iChamber->GetResponseModel();
      Int_t adcmax= (Int_t) response->MaxAdc();
      
      
      // add white noise and do zero-suppression and signal truncation (new electronics,old electronics gaus 1.2,0.2)
      //printf("Treshold: %d\n",iChamber->fTresh->GetHitIndex(address->fPadX,address->fPadY));
      Int_t pedestal = iChamber->fTresh->GetHitIndex(address->fPadX,address->fPadY);

      //printf("Pedestal:%d\n",pedestal);
      //Int_t pedestal=0;
      Float_t treshold = (pedestal + 4*2.2);
      
      Float_t meanNoise = gRandom->Gaus(2.2, 0.3);
      Float_t noise     = gRandom->Gaus(0, meanNoise);
      q+=(Int_t)(noise + pedestal);
      //q+=(Int_t)(noise);
      //          magic number to be parametrised !!! 
      if ( q <= treshold) 
	{
	  q = q - pedestal;
	  continue;
	}
      q = q - pedestal;
      if ( q >= adcmax) q=adcmax;
      digits[0]=address->fPadX;
      digits[1]=address->fPadY;
      digits[2]=q;
      
      TObjArray* trlist=(TObjArray*)address->TrackList();
      Int_t nptracks=trlist->GetEntriesFast();
      
      // this was changed to accomodate the real number of tracks
      if (nptracks > 10) {
	printf("Attention - tracks > 10 %d\n",nptracks);
	nptracks=10;
      }
      if (nptracks > 2) {
	printf("Attention - tracks > 2  %d \n",nptracks);
	//printf("cat,ich,ix,iy,q %d %d %d %d %d \n",
	//icat,ich,digits[0],digits[1],q);
      }
      for (Int_t tr=0;tr<nptracks;tr++) {
	TVector *ppP=(TVector*)trlist->At(tr);
	TVector &pp  =*ppP;
	tracks[tr]=Int_t(pp(0));
	charges[tr]=Int_t(pp(1));
      }      //end loop over list of tracks for one pad
      if (nptracks < 10 ) {
	for (Int_t t=nptracks; t<10; t++) {
	  tracks[t]=0;
	  charges[t]=0;
	}
      }
      //write file
      //if (ich==2)
	//fprintf(points,"%4d,      %4d,      %4d\n",digits[0],digits[1],digits[2]);
      
      // fill digits
      pRICH->AddDigits(ich,tracks,charges,digits);
    }	
    gAlice->TreeD()->Fill();
    
    list->Delete();
    for(Int_t ii=0;ii<kNCH;++ii) {
      if (pHitMap[ii]) {
	hm=pHitMap[ii];
	delete hm;
	pHitMap[ii]=0;
      }
    }
    
    //TTree *TD=gAlice->TreeD();
    //Stat_t ndig=TD->GetEntries();
    //cout<<"number of digits  "<<ndig<<endl;
    TClonesArray *fDch;
    for (int k=0;k<kNCH;k++) {
      fDch= pRICH->DigitsAddress(k);
      int ndigit=fDch->GetEntriesFast();
      printf ("Chamber %d digits %d \n",k,ndigit);
    }
    pRICH->ResetDigits();
    // char hname[30];
    // sprintf(hname,"TreeD%d",nev);
    // gAlice->TreeD()->Write(hname);
    gAlice->TreeD()->Write(0,TObject::kOverwrite);
    // reset tree
    //    gAlice->TreeD()->Reset();
    delete list;
    pAddress->Clear();
    // gObjectTable->Print();

}
Commit	Line	Data
d8a72780	1	/**************************************************************************
	2	* Copyright(c) 1998-2000, 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	/*
	17	$Log$
2685bf00	18	Revision 1.3 2001/05/16 14:57:20 alibrary
	19	New files for folders and Stack
	20
9e1a0ddb	21	Revision 1.2 2001/03/14 18:16:08 jbarbosa
	22	Corrected bug (more to correct).
	23	File "points.dat" is no longer created.
	24
14886cb7	25	Revision 1.1 2001/02/27 22:13:34 jbarbosa
	26	Implementing merger class.
	27
d8a72780	28	*/
	29
	30	#include <TTree.h>
	31	#include <TVector.h>
	32	#include <TObjArray.h>
	33	#include <TFile.h>
	34	#include <TDirectory.h>
	35	#include <TParticle.h>
	36
	37
	38	// #include "AliMerger.h"
	39	// #include "AliMergable.h"
	40	#include "AliRICHMerger.h"
	41	#include "AliRICHChamber.h"
	42	#include "AliHitMap.h"
	43	#include "AliRICHHitMapA1.h"
	44	#include "AliRICH.h"
	45	#include "AliRICHHit.h"
	46	#include "AliRICHSDigit.h"
	47	#include "AliRICHDigit.h"
	48	#include "AliRICHTransientDigit.h"
	49	#include "AliRun.h"
	50	#include "AliPDG.h"
	51
	52	ClassImp(AliRICHMerger)
	53
	54	//___________________________________________
	55	AliRICHMerger::AliRICHMerger()
	56	{
	57	// Default constructor
	58	fEvNrSig = 0;
	59	fEvNrBgr = 0;
	60	fMerge = kDigitize;
	61	fFnBgr = 0;
	62	fHitMap = 0;
	63	fList = 0;
	64	fTrH1 = 0;
	65	fHitsBgr = 0;
	66	fSDigitsBgr = 0;
	67	fHitMap = 0;
	68	fList = 0;
	69	fTrList = 0;
2685bf00	70	fAddress = 0;
2685bf00	71	fBgrFile = 0;
d8a72780	72	}
	73
	74	//------------------------------------------------------------------------
	75	AliRICHMerger::~AliRICHMerger()
	76	{
	77	// Destructor
	78	if (fTrH1) delete fTrH1;
	79	if (fHitsBgr) delete fHitsBgr;
	80	if (fSDigitsBgr) delete fSDigitsBgr;
	81	if (fHitMap) delete fHitMap;
	82	if (fList) delete fList;
	83	if (fTrList) delete fTrList;
	84	if (fAddress) delete fAddress;
	85	}
	86
	87	//------------------------------------------------------------------------
	88	Bool_t AliRICHMerger::Exists(const AliRICHSDigit *mergable)
	89	{
	90	AliRICHSDigit padhit = (AliRICHSDigit) mergable;
	91	return (fHitMap[fNch]->TestHit(padhit->PadX(),padhit->PadY()));
	92	}
	93
	94	//------------------------------------------------------------------------
	95	void AliRICHMerger::Update(AliRICHSDigit *mergable)
	96	{
	97	AliRICHSDigit padhit = (AliRICHSDigit) mergable;
	98	AliRICHTransientDigit* pdigit;
	99	Int_t ipx = padhit->PadX(); // pad number on X
	100	Int_t ipy = padhit->PadY(); // pad number on Y
	101	Int_t iqpad = Int_t(padhit->QPad()); // charge per pad
	102
	103	pdigit = (AliRICHTransientDigit*) fHitMap[fNch]->GetHit(ipx, ipy);
	104	// update charge
	105	//
	106	(*pdigit).AddSignal(iqpad);
	107	(*pdigit).AddPhysicsSignal(iqpad);
	108	// update list of tracks
	109	//
	110	TObjArray* fTrList = (TObjArray*)pdigit->TrackList();
	111	Int_t lastEntry = fTrList->GetLast();
	112	TVector pTrack = (TVector)fTrList->At(lastEntry);
	113	TVector &ptrk = *pTrack;
	114	TVector &trinfo = ((TVector) (*fAddress)[fCountadr-1]);
	115	Int_t lastTrack = Int_t(ptrk(0));
	116
	117	if (trinfo(0) != kBgTag) {
	118	if (lastTrack == fTrack) {
	119	Int_t lastCharge = Int_t(ptrk(1));
	120	lastCharge += iqpad;
	121	fTrList->RemoveAt(lastEntry);
	122	trinfo(1) = lastCharge;
	123	fTrList->AddAt(&trinfo,lastEntry);
	124	} else {
	125	fTrList->Add(&trinfo);
	126	}
	127	} else {
	128	if (lastTrack != -1) fTrList->Add(&trinfo);
	129	}
	130	}
	131
	132	//------------------------------------------------------------------------
	133	void AliRICHMerger::CreateNew(AliRICHSDigit *mergable)
	134	{
	135	AliRICHSDigit padhit = (AliRICHSDigit) mergable;
136	AliRICHTransientDigit* pdigit;
137
138	Int_t ipx = padhit->PadX(); // pad number on X
139	Int_t ipy = padhit->PadY(); // pad number on Y
140	fList->AddAtAndExpand(
141	new AliRICHTransientDigit(fNch,fDigits),fCounter);
142	fHitMap[fNch]->SetHit(ipx, ipy, fCounter);
143	fCounter++;
144	pdigit = (AliRICHTransientDigit*)fList->At(fList->GetLast());
145	// list of tracks
146	TObjArray fTrList = (TObjArray)pdigit->TrackList();
147	TVector &trinfo = ((TVector) (*fAddress)[fCountadr-1]);
148	fTrList->Add(&trinfo);
149	}
150
151
152	//------------------------------------------------------------------------
153	void AliRICHMerger::Init()
154	{
155	// Initialisation
156
157	if (fMerge) fBgrFile = InitBgr();
158	}
159
160
161
162	//------------------------------------------------------------------------
163	TFile* AliRICHMerger::InitBgr()
164	{
165	// Initialise background event
166	TFile *file = new TFile(fFnBgr);
167	// add error checking later
168	printf("\n AliRICHMerger has opened %s file with background event \n", fFnBgr);
169	fHitsBgr = new TClonesArray("AliRICHHit",1000);
170	fSDigitsBgr = new TClonesArray("AliRICHSDigit",1000);
171	return file;
172	}
173
174	//------------------------------------------------------------------------
175	void AliRICHMerger::Digitise(Int_t nev, Int_t flag)
176	{
177
178	// keep galice.root for signal and name differently the file for
179	// background when add! otherwise the track info for signal will be lost !
180
181	Int_t particle;
182
14886cb7	183	//FILE* points; //these will be the digits...
d8a72780	184
14886cb7	185	//points=fopen("points.dat","w");
d8a72780	186
	187	AliRICHChamber* iChamber;
	188	AliSegmentation* segmentation;
	189
	190	Int_t digitise=0;
	191	Int_t trk[50];
	192	Int_t chtrk[50];
	193	TObjArray *list=new TObjArray;
	194	static TClonesArray *pAddress=0;
	195	if(!pAddress) pAddress=new TClonesArray("TVector",1000);
	196	Int_t digits[5];
	197
	198	AliRICH pRICH = (AliRICH ) gAlice->GetDetector("RICH");
	199	AliHitMap* pHitMap[10];
	200	Int_t i;
	201	for (i=0; i<10; i++) {pHitMap[i]=0;}
	202
	203	if (fMerge ) {
	204	fBgrFile->cd();
14886cb7	205	//fBgrFile->ls();
d8a72780	206	//
	207	// Get Hits Tree header from file
	208	if(fHitsBgr) fHitsBgr->Clear();
	209	if(fSDigitsBgr) fSDigitsBgr->Clear();
	210	if(fTrH1) delete fTrH1;
	211	fTrH1 = 0;
	212
	213	char treeName[20];
	214	sprintf(treeName,"TreeH%d",fEvNrBgr);
	215	fTrH1 = (TTree*)gDirectory->Get(treeName);
	216	if (!fTrH1) {
	217	printf("ERROR: cannot find Hits Tree for event:%d\n",fEvNrBgr);
	218	}
	219	//
	220	// Set branch addresses
	221	TBranch *branch;
	222	char branchname[20];
	223	sprintf(branchname,"%s",pRICH->GetName());
	224	if (fTrH1 && fHitsBgr) {
	225	branch = fTrH1->GetBranch(branchname);
	226	if (branch) branch->SetAddress(&fHitsBgr);
	227	}
	228	if (fTrH1 && fSDigitsBgr) {
14886cb7	229	branch = fTrH1->GetBranch("RICHSDigits");
d8a72780	230	if (branch) branch->SetAddress(&fSDigitsBgr);
	231	}
	232	}
	233
	234	AliHitMap* hm;
	235	Int_t countadr=0;
	236	Int_t counter=0;
	237	for (i =0; i<kNCH; i++) {
	238	iChamber= &(pRICH->Chamber(i));
	239	segmentation=iChamber->GetSegmentationModel(1);
	240	pHitMap[i] = new AliRICHHitMapA1(segmentation, list);
	241	}
	242	//
	243	// Loop over tracks
	244	//
	245
	246	TTree *treeH = gAlice->TreeH();
	247	Int_t ntracks =(Int_t) treeH->GetEntries();
	248	for (Int_t track=0; track<ntracks; track++) {
	249	gAlice->ResetHits();
	250	treeH->GetEvent(track);
	251	//
	252	// Loop over hits
	253	for(AliRICHHit* mHit=(AliRICHHit*)pRICH->FirstHit(-1);
	254	mHit;
	255	mHit=(AliRICHHit*)pRICH->NextHit())
	256	{
	257
	258	Int_t nch = mHit->fChamber-1; // chamber number
	259	Int_t index = mHit->Track();
	260	if (nch >kNCH) continue;
	261	iChamber = &(pRICH->Chamber(nch));
	262
	263	TParticle current = (TParticle)gAlice->Particle(index);
	264
	265	if (current->GetPdgCode() >= 50000050)
	266	{
	267	TParticle motherofcurrent = (TParticle)gAlice->Particle(current->GetFirstMother());
	268	particle = motherofcurrent->GetPdgCode();
	269	}
	270	else
	271	{
	272	particle = current->GetPdgCode();
	273	}
	274
	275
	276	//printf("Flag:%d\n",flag);
	277	//printf("Track:%d\n",track);
	278	//printf("Particle:%d\n",particle);
	279
	280	digitise=1;
	281
	282	if (flag == 1)
	283	if(TMath::Abs(particle)==211 \|\| TMath::Abs(particle)==111)
	284	digitise=0;
	285
	286	if (flag == 2)
	287	if(TMath::Abs(particle)==321 \|\| TMath::Abs(particle)==130 \|\| TMath::Abs(particle)==310
	288	\|\| TMath::Abs(particle)==311)
	289	digitise=0;
	290
	291	if (flag == 3 && TMath::Abs(particle)==2212)
	292	digitise=0;
	293
294	if (flag == 4 && TMath::Abs(particle)==13)
295	digitise=0;
296
297	if (flag == 5 && TMath::Abs(particle)==11)
298	digitise=0;
299
300	if (flag == 6 && TMath::Abs(particle)==2112)
301	digitise=0;
302
303
304	//printf ("Particle: %d, Flag: %d, Digitise: %d\n",particle,flag,digitise);
305
306
307	if (digitise)
308	{
309
310	//
311	// Loop over pad hits
312	for (AliRICHSDigit* mPad=
313	(AliRICHSDigit*)pRICH->FirstPad(mHit,pRICH->SDigits());
314	mPad;
315	mPad=(AliRICHSDigit*)pRICH->NextPad(pRICH->SDigits()))
316	{
317	Int_t ipx = mPad->fPadX; // pad number on X
318	Int_t ipy = mPad->fPadY; // pad number on Y
319	Int_t iqpad = mPad->fQpad; // charge per pad
320	//
321	//
322	//printf("X:%d, Y:%d, Q:%d\n",ipx,ipy,iqpad);
323
324	Float_t thex, they, thez;
325	segmentation=iChamber->GetSegmentationModel(0);
326	segmentation->GetPadC(ipx,ipy,thex,they,thez);
327	new((*pAddress)[countadr++]) TVector(2);
328	TVector &trinfo=((TVector) (*pAddress)[countadr-1]);
329	trinfo(0)=(Float_t)track;
330	trinfo(1)=(Float_t)iqpad;
331
332	digits[0]=ipx;
333	digits[1]=ipy;
334	digits[2]=iqpad;
335
336	AliRICHTransientDigit* pdigit;
337	// build the list of fired pads and update the info
338	if (!pHitMap[nch]->TestHit(ipx, ipy)) {
339	list->AddAtAndExpand(new AliRICHTransientDigit(nch,digits),counter);
340	pHitMap[nch]->SetHit(ipx, ipy, counter);
341	counter++;
342	pdigit=(AliRICHTransientDigit*)list->At(list->GetLast());
343	// list of tracks
344	TObjArray trlist=(TObjArray)pdigit->TrackList();
345	trlist->Add(&trinfo);
346	} else {
347	pdigit=(AliRICHTransientDigit*) pHitMap[nch]->GetHit(ipx, ipy);
348	// update charge
349	(*pdigit).fSignal+=iqpad;
350	// update list of tracks
351	TObjArray* trlist=(TObjArray*)pdigit->TrackList();
352	Int_t lastEntry=trlist->GetLast();
353	TVector ptrkP=(TVector)trlist->At(lastEntry);
354	TVector &ptrk=*ptrkP;
355	Int_t lastTrack=Int_t(ptrk(0));
356	Int_t lastCharge=Int_t(ptrk(1));
357	if (lastTrack==track) {
358	lastCharge+=iqpad;
359	trlist->RemoveAt(lastEntry);
360	trinfo(0)=lastTrack;
361	trinfo(1)=lastCharge;
362	trlist->AddAt(&trinfo,lastEntry);
363	} else {
364	trlist->Add(&trinfo);
365	}
366	// check the track list
367	Int_t nptracks=trlist->GetEntriesFast();
368	if (nptracks > 2) {
369	printf("Attention - tracks: %d (>2)\n",nptracks);
370	//printf("cat,nch,ix,iy %d %d %d %d \n",icat+1,nch,ipx,ipy);
371	for (Int_t tr=0;tr<nptracks;tr++) {
372	TVector pptrkP=(TVector)trlist->At(tr);
373	TVector &pptrk=*pptrkP;
374	trk[tr]=Int_t(pptrk(0));
375	chtrk[tr]=Int_t(pptrk(1));
376	}
377	} // end if nptracks
378	} // end if pdigit
379	} //end loop over clusters
380	}// track type condition
381	} // hit loop
382	} // track loop
383
384	// open the file with background
385
386	if (fMerge) {
387	ntracks = (Int_t)fTrH1->GetEntries();
388	//
389	// Loop over tracks
390	//
391	for (fTrack = 0; fTrack < ntracks; fTrack++) {
392
393	if (fHitsBgr) fHitsBgr->Clear();
394	if (fSDigitsBgr) fSDigitsBgr->Clear();
395
396	fTrH1->GetEvent(fTrack);
397	//
398	// Loop over hits
399	AliRICHHit* mHit;
400	for(int i = 0; i < fHitsBgr->GetEntriesFast(); ++i)
401	{
402	mHit = (AliRICHHit) (fHitsBgr)[i];
403	fNch = mHit->Chamber()-1; // chamber number
404	iChamber = &(pRICH->Chamber(fNch));
405	//Float_t xbgr = mHit->X();
406	//Float_t ybgr = mHit->Y();
407	Bool_t cond = kFALSE;
408	cond = kTRUE;
409	//
410	// Loop over pad hits
411	for (AliRICHSDigit* mPad =
412	(AliRICHSDigit*)pRICH->FirstPad(mHit,fSDigitsBgr);
413	mPad;
414	mPad = (AliRICHSDigit*)pRICH->NextPad(fSDigitsBgr))
415	{
416	Int_t ipx = mPad->PadX(); // pad number on X
417	Int_t ipy = mPad->PadY(); // pad number on Y
418	Int_t iqpad = Int_t(mPad->QPad()); // charge per pad
419
420	new((*fAddress)[fCountadr++]) TVector(2);
421	TVector &trinfo = ((TVector) (*fAddress)[fCountadr-1]);
422	trinfo(0) = kBgTag; // tag background
423	trinfo(1) = kBgTag;
424
425	fDigits[0] = ipx;
426	fDigits[1] = ipy;
427	fDigits[3] = iqpad;
428
429	// build the list of fired pads and update the info
430	if (!Exists(mPad)) {
431	CreateNew(mPad);
432	} else {
433	Update(mPad);
434	} // end if !Exists
435	} //end loop over clusters
436	} // hit loop
437	} // track loop
438
439	TTree *fAli = gAlice->TreeK();
440	TFile *file = NULL;
441
442	if (fAli) file = fAli->GetCurrentFile();
443	file->cd();
444	} // if fMerge
445
446	Int_t tracks[10];
447	Int_t charges[10];
448	//cout<<"Start filling digits \n "<<endl;
449	Int_t nentries=list->GetEntriesFast();
450	//printf(" \n \n nentries %d \n",nentries);
451
452	// start filling the digits
453
454	for (Int_t nent=0;nent<nentries;nent++) {
455	AliRICHTransientDigit address=(AliRICHTransientDigit)list->At(nent);
456	if (address==0) continue;
457
458	Int_t ich=address->fChamber;
459	Int_t q=address->fSignal;
460	iChamber=&(pRICH->Chamber(ich));
461	AliRICHResponse * response=iChamber->GetResponseModel();
462	Int_t adcmax= (Int_t) response->MaxAdc();
463
464
465	// add white noise and do zero-suppression and signal truncation (new electronics,old electronics gaus 1.2,0.2)
466	//printf("Treshold: %d\n",iChamber->fTresh->GetHitIndex(address->fPadX,address->fPadY));
467	Int_t pedestal = iChamber->fTresh->GetHitIndex(address->fPadX,address->fPadY);
468
469	//printf("Pedestal:%d\n",pedestal);
470	//Int_t pedestal=0;
471	Float_t treshold = (pedestal + 4*2.2);
472
473	Float_t meanNoise = gRandom->Gaus(2.2, 0.3);
474	Float_t noise = gRandom->Gaus(0, meanNoise);
475	q+=(Int_t)(noise + pedestal);
476	//q+=(Int_t)(noise);
477	// magic number to be parametrised !!!
478	if ( q <= treshold)
479	{
480	q = q - pedestal;
481	continue;
482	}
483	q = q - pedestal;
484	if ( q >= adcmax) q=adcmax;
485	digits[0]=address->fPadX;
486	digits[1]=address->fPadY;
487	digits[2]=q;
488
489	TObjArray* trlist=(TObjArray*)address->TrackList();
490	Int_t nptracks=trlist->GetEntriesFast();
491
492	// this was changed to accomodate the real number of tracks
493	if (nptracks > 10) {
494	printf("Attention - tracks > 10 %d\n",nptracks);
495	nptracks=10;
496	}
497	if (nptracks > 2) {
498	printf("Attention - tracks > 2 %d \n",nptracks);
499	//printf("cat,ich,ix,iy,q %d %d %d %d %d \n",
500	//icat,ich,digits[0],digits[1],q);
501	}
502	for (Int_t tr=0;tr<nptracks;tr++) {
503	TVector ppP=(TVector)trlist->At(tr);
504	TVector &pp =*ppP;
505	tracks[tr]=Int_t(pp(0));
506	charges[tr]=Int_t(pp(1));
507	} //end loop over list of tracks for one pad
508	if (nptracks < 10 ) {
509	for (Int_t t=nptracks; t<10; t++) {
510	tracks[t]=0;
511	charges[t]=0;
512	}
513	}
514	//write file
14886cb7	515	//if (ich==2)
14886cb7	516	//fprintf(points,"%4d, %4d, %4d\n",digits[0],digits[1],digits[2]);
d8a72780	517
	518	// fill digits
	519	pRICH->AddDigits(ich,tracks,charges,digits);
	520	}
	521	gAlice->TreeD()->Fill();
	522
	523	list->Delete();
	524	for(Int_t ii=0;ii<kNCH;++ii) {
	525	if (pHitMap[ii]) {
	526	hm=pHitMap[ii];
	527	delete hm;
	528	pHitMap[ii]=0;
	529	}
	530	}
	531
	532	//TTree *TD=gAlice->TreeD();
	533	//Stat_t ndig=TD->GetEntries();
	534	//cout<<"number of digits "<<ndig<<endl;
	535	TClonesArray *fDch;
	536	for (int k=0;k<kNCH;k++) {
	537	fDch= pRICH->DigitsAddress(k);
	538	int ndigit=fDch->GetEntriesFast();
	539	printf ("Chamber %d digits %d \n",k,ndigit);
	540	}
	541	pRICH->ResetDigits();
9e1a0ddb	542	// char hname[30];
	543	// sprintf(hname,"TreeD%d",nev);
	544	// gAlice->TreeD()->Write(hname);
	545	gAlice->TreeD()->Write(0,TObject::kOverwrite);
d8a72780	546	// reset tree
	547	// gAlice->TreeD()->Reset();
	548	delete list;
	549	pAddress->Clear();
	550	// gObjectTable->Print();
	551
	552	}
	553
	554
	555
	556
	557