[u/mrichter/AliRoot.git] / HMPID / AliHMPIDDigitizer.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.                  *
 **************************************************************************/
//.
//.
//.
//.
//.
#include <AliRun.h>
#include <AliRunLoader.h>
#include "AliRunDigitizer.h"
#include <AliLoader.h>
#include <AliLog.h>
#include "AliHMPIDDigitizer.h"
#include "AliHMPIDDigit.h"
#include "AliHMPID.h"
#include "AliHMPIDParam.h"
#include <TRandom.h>
#include <TTree.h>

ClassImp(AliHMPIDDigitizer)

Bool_t AliHMPIDDigitizer::fgDoNoise=kFALSE;
//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
void AliHMPIDDigitizer::Exec(Option_t*)
{
// This methode is responsible for merging sdigits to a list of digits
//Disintegration leeds to the fact that one hit affects several neighbouring pads, which means that the same pad might be affected by few hits.     
  AliDebug(1,Form("Start with %i input(s) for event %i",fManager->GetNinputs(),fManager->GetOutputEventNr()));
//First we read all sdigits from all inputs  
  AliRunLoader *pInRunLoader=0;//in and out Run loaders
  AliLoader    *pInRichLoader=0;//in and out HMPID loaders  
  static TClonesArray sdigs("AliHMPIDDigit");//tmp storage for sdigits sum up from all input files
  Int_t total=0;
  for(Int_t inFileN=0;inFileN<fManager->GetNinputs();inFileN++){//files loop
    pInRunLoader  = AliRunLoader::GetRunLoader(fManager->GetInputFolderName(inFileN));          //get run loader from current input 
    pInRichLoader = pInRunLoader->GetLoader("HMPIDLoader"); if(pInRichLoader==0) continue;       //no HMPID in this input, check the next input
    if (!pInRunLoader->GetAliRun()) pInRunLoader->LoadgAlice();
    AliHMPID* pInRich=(AliHMPID*)pInRunLoader->GetAliRun()->GetDetector("HMPID");                  //take HMPID from current input
    pInRichLoader->LoadSDigits(); pInRichLoader->TreeS()->GetEntry(0);                          //take list of HMPID sdigits from current input 
    AliDebug(1,Form("input %i has %i sdigits",inFileN,pInRich->SdiLst()->GetEntries()));
    for(Int_t i=0;i<pInRich->SdiLst()->GetEntries();i++){                                        //collect sdigits from current input
      AliHMPIDDigit *pSDig=(AliHMPIDDigit*)pInRich->SdiLst()->At(i);
      pSDig->AddTidOffset(fManager->GetMask(inFileN));                                          //apply TID shift since all inputs count tracks independently starting from 0
      new(sdigs[total++]) AliHMPIDDigit(*pSDig);       
    }
    pInRichLoader->UnloadSDigits();   pInRich->SdiReset(); //close current input and reset 
  }//files loop

  //PH  if(sdigs.GetEntries()==0) return;                                                              //no sdigits collected, nothing to convert  
  
  AliRunLoader *pOutRunLoader  = AliRunLoader::GetRunLoader(fManager->GetOutputFolderName());    //open output stream (only 1 possible)
  AliLoader    *pOutRichLoader = pOutRunLoader->GetLoader("HMPIDLoader");                         //take output HMPID loader
  AliRun *pArun = pOutRunLoader->GetAliRun();
  AliHMPID      *pOutRich       = (AliHMPID*)pArun->GetDetector("HMPID");      //take output HMPID
  pOutRichLoader->MakeTree("D");   pOutRich->MakeBranch("D");                                    //create TreeD in output stream

  Sdi2Dig(&sdigs,pOutRich->DigLst());
  
  pOutRichLoader->TreeD()->Fill();              //fill the output tree with the list of digits
  pOutRichLoader->WriteDigits("OVERWRITE");     //serialize them to file
  
  sdigs.Clear();                      //remove all tmp sdigits
  pOutRichLoader->UnloadDigits();   pOutRich->DigReset();
}//Exec()
//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
void AliHMPIDDigitizer::Sdi2Dig(TClonesArray *pSdiLst,TObjArray *pDigLst)
{
// Converts list of sdigits to 7 lists of digits, one per each chamber
// Arguments: pSDigLst - list of all sdigits
//            pDigLst  - list of 7 lists of digits        
//   Returns: none  
  
  TClonesArray *pLst[7]; Int_t iCnt[7];

  for(Int_t i=0;i<7;i++){
    pLst[i]=(TClonesArray*)(*pDigLst)[i];
    iCnt[i]=0; if(pLst[i]->GetEntries()!=0) AliErrorClass("Some of digits lists is not empty");         //in principle those lists should be empty                                                                       
  }

  // make noise array
  Float_t arrNoise[7][6][80][48];
  if(fgDoNoise) {
    for (Int_t iCh=AliHMPIDParam::kMinCh;iCh<=AliHMPIDParam::kMaxCh;iCh++)
      for (Int_t iPc=AliHMPIDParam::kMinPc;iPc<=AliHMPIDParam::kMaxPc;iPc++)
        for(Int_t iPx=AliHMPIDParam::kMinPx;iPx<=AliHMPIDParam::kMaxPx;iPx++)
          for(Int_t iPy=AliHMPIDParam::kMinPy;iPy<=AliHMPIDParam::kMaxPy;iPy++) arrNoise[iCh][iPc][iPx][iPy] = gRandom->Gaus(0,1.1);
  }  
  
  pSdiLst->Sort();  
                     
  Int_t iPad=-1,iCh=-1,iNdigPad=-1,aTids[3]={-1,-1,-1}; Float_t q=-1;
  for(Int_t i=0;i<pSdiLst->GetEntries();i++){                                                                  //sdigits loop (sorted)
    AliHMPIDDigit *pSdig=(AliHMPIDDigit*)pSdiLst->At(i);                                                         //take current sdigit
    if(pSdig->Pad()==iPad){                                                                                    //if the same pad 
      q+=pSdig->Q();                                                                                           //sum up charge
      iNdigPad++; if(iNdigPad<=3) aTids[iNdigPad-1]=pSdig->GetTrack(0);                                        //collect TID 
      continue;
    }
    if(i!=0 && AliHMPIDParam::IsOverTh(q) 
            && iCh>=AliHMPIDParam::kMinCh
            && iCh<=AliHMPIDParam::kMaxCh)  new((*pLst[iCh])[iCnt[iCh]++]) AliHMPIDDigit(iPad,(Int_t)q,aTids);  //do not create digit for the very first sdigit 
    
    iPad=pSdig->Pad(); iCh=AliHMPIDParam::A2C(iPad);                                                            //new sdigit comes, reset collectors
    iNdigPad=1;
    aTids[0]=pSdig->GetTrack(0);aTids[1]=aTids[2]=-1; 
    q=pSdig->Q();
    if(fgDoNoise) q+=arrNoise[iCh][pSdig->Pc()][pSdig->PadPcX()][pSdig->PadPcY()];
    arrNoise[iCh][pSdig->Pc()][pSdig->PadPcX()][pSdig->PadPcY()]=0;
  }//sdigits loop (sorted)
  
  if(AliHMPIDParam::IsOverTh(q)
      && iCh>=AliHMPIDParam::kMinCh
      && iCh<=AliHMPIDParam::kMaxCh)  new((*pLst[iCh])[iCnt[iCh]++]) AliHMPIDDigit(iPad,(Int_t)q,aTids);        //add the last one, in case of empty sdigits list q=-1
  
// add noise pad above threshold with no signal merged...if any
  if(!fgDoNoise) return;
  aTids[0]=aTids[1]=aTids[2]=-1;
  for (Int_t iChCurr=AliHMPIDParam::kMinCh;iChCurr<=AliHMPIDParam::kMaxCh;iChCurr++)
    for (Int_t iPc=AliHMPIDParam::kMinPc;iPc<=AliHMPIDParam::kMaxPc;iPc++)
      for(Int_t iPx=AliHMPIDParam::kMinPx;iPx<=AliHMPIDParam::kMaxPx;iPx++)
        for(Int_t iPy=AliHMPIDParam::kMinPy;iPy<=AliHMPIDParam::kMaxPy;iPy++) {
          Float_t qNoise = arrNoise[iChCurr][iPc][iPx][iPy];
          if(AliHMPIDParam::IsOverTh(qNoise)) new((*pLst[iChCurr])[iCnt[iChCurr]++]) AliHMPIDDigit(AliHMPIDParam::Abs(iChCurr,iPc,iPx,iPy),(Int_t)qNoise,aTids);
        }
        
}//Sdi2Dig()
//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
Commit	Line	Data
d3da6dc4	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	**************************************************************************/
1b10372c	15	//.
	16	//.
	17	//.
	18	//.
	19	//.
	20	#include <AliRun.h>
d3da6dc4	21	#include <AliRunLoader.h>
	22	#include "AliRunDigitizer.h"
	23	#include <AliLoader.h>
	24	#include <AliLog.h>
1b10372c	25	#include "AliHMPIDDigitizer.h"
	26	#include "AliHMPIDDigit.h"
	27	#include "AliHMPID.h"
ae5a42aa	28	#include "AliHMPIDParam.h"
1b10372c	29	#include <TRandom.h>
c93255fe	30	#include <TTree.h>
d3da6dc4	31
	32	ClassImp(AliHMPIDDigitizer)
	33
94b1fbfa	34	Bool_t AliHMPIDDigitizer::fgDoNoise=kFALSE;
d3da6dc4	35	//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
	36	void AliHMPIDDigitizer::Exec(Option_t*)
	37	{
	38	// This methode is responsible for merging sdigits to a list of digits
	39	//Disintegration leeds to the fact that one hit affects several neighbouring pads, which means that the same pad might be affected by few hits.
	40	AliDebug(1,Form("Start with %i input(s) for event %i",fManager->GetNinputs(),fManager->GetOutputEventNr()));
	41	//First we read all sdigits from all inputs
	42	AliRunLoader *pInRunLoader=0;//in and out Run loaders
	43	AliLoader *pInRichLoader=0;//in and out HMPID loaders
d6183365	44	static TClonesArray sdigs("AliHMPIDDigit");//tmp storage for sdigits sum up from all input files
d3da6dc4	45	Int_t total=0;
	46	for(Int_t inFileN=0;inFileN<fManager->GetNinputs();inFileN++){//files loop
	47	pInRunLoader = AliRunLoader::GetRunLoader(fManager->GetInputFolderName(inFileN)); //get run loader from current input
	48	pInRichLoader = pInRunLoader->GetLoader("HMPIDLoader"); if(pInRichLoader==0) continue; //no HMPID in this input, check the next input
	49	if (!pInRunLoader->GetAliRun()) pInRunLoader->LoadgAlice();
	50	AliHMPID* pInRich=(AliHMPID*)pInRunLoader->GetAliRun()->GetDetector("HMPID"); //take HMPID from current input
	51	pInRichLoader->LoadSDigits(); pInRichLoader->TreeS()->GetEntry(0); //take list of HMPID sdigits from current input
	52	AliDebug(1,Form("input %i has %i sdigits",inFileN,pInRich->SdiLst()->GetEntries()));
	53	for(Int_t i=0;i<pInRich->SdiLst()->GetEntries();i++){ //collect sdigits from current input
	54	AliHMPIDDigit pSDig=(AliHMPIDDigit)pInRich->SdiLst()->At(i);
	55	pSDig->AddTidOffset(fManager->GetMask(inFileN)); //apply TID shift since all inputs count tracks independently starting from 0
	56	new(sdigs[total++]) AliHMPIDDigit(*pSDig);
	57	}
	58	pInRichLoader->UnloadSDigits(); pInRich->SdiReset(); //close current input and reset
	59	}//files loop
	60
	61	//PH if(sdigs.GetEntries()==0) return; //no sdigits collected, nothing to convert
	62
	63	AliRunLoader *pOutRunLoader = AliRunLoader::GetRunLoader(fManager->GetOutputFolderName()); //open output stream (only 1 possible)
	64	AliLoader *pOutRichLoader = pOutRunLoader->GetLoader("HMPIDLoader"); //take output HMPID loader
1b10372c	65	AliRun *pArun = pOutRunLoader->GetAliRun();
1b10372c	66	AliHMPID pOutRich = (AliHMPID)pArun->GetDetector("HMPID"); //take output HMPID
d3da6dc4	67	pOutRichLoader->MakeTree("D"); pOutRich->MakeBranch("D"); //create TreeD in output stream
	68
	69	Sdi2Dig(&sdigs,pOutRich->DigLst());
	70
	71	pOutRichLoader->TreeD()->Fill(); //fill the output tree with the list of digits
	72	pOutRichLoader->WriteDigits("OVERWRITE"); //serialize them to file
	73
	74	sdigs.Clear(); //remove all tmp sdigits
	75	pOutRichLoader->UnloadDigits(); pOutRich->DigReset();
	76	}//Exec()
	77	//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
	78	void AliHMPIDDigitizer::Sdi2Dig(TClonesArray pSdiLst,TObjArray pDigLst)
	79	{
	80	// Converts list of sdigits to 7 lists of digits, one per each chamber
	81	// Arguments: pSDigLst - list of all sdigits
	82	// pDigLst - list of 7 lists of digits
	83	// Returns: none
	84
	85	TClonesArray *pLst[7]; Int_t iCnt[7];
1b10372c	86
d3da6dc4	87	for(Int_t i=0;i<7;i++){
d3da6dc4	88	pLst[i]=(TClonesArray)(pDigLst)[i];
3c6274c1	89	iCnt[i]=0; if(pLst[i]->GetEntries()!=0) AliErrorClass("Some of digits lists is not empty"); //in principle those lists should be empty
d3da6dc4	90	}
c5c19d6a	91
	92	// make noise array
	93	Float_t arrNoise[7][6][80][48];
94b1fbfa	94	if(fgDoNoise) {
ae5a42aa	95	for (Int_t iCh=AliHMPIDParam::kMinCh;iCh<=AliHMPIDParam::kMaxCh;iCh++)
	96	for (Int_t iPc=AliHMPIDParam::kMinPc;iPc<=AliHMPIDParam::kMaxPc;iPc++)
	97	for(Int_t iPx=AliHMPIDParam::kMinPx;iPx<=AliHMPIDParam::kMaxPx;iPx++)
73badab2	98	for(Int_t iPy=AliHMPIDParam::kMinPy;iPy<=AliHMPIDParam::kMaxPy;iPy++) arrNoise[iCh][iPc][iPx][iPy] = gRandom->Gaus(0,1.1);
c5c19d6a	99	}
d3da6dc4	100
	101	pSdiLst->Sort();
	102
	103	Int_t iPad=-1,iCh=-1,iNdigPad=-1,aTids[3]={-1,-1,-1}; Float_t q=-1;
	104	for(Int_t i=0;i<pSdiLst->GetEntries();i++){ //sdigits loop (sorted)
	105	AliHMPIDDigit pSdig=(AliHMPIDDigit)pSdiLst->At(i); //take current sdigit
	106	if(pSdig->Pad()==iPad){ //if the same pad
	107	q+=pSdig->Q(); //sum up charge
	108	iNdigPad++; if(iNdigPad<=3) aTids[iNdigPad-1]=pSdig->GetTrack(0); //collect TID
	109	continue;
	110	}
aa8db9d6	111	if(i!=0 && AliHMPIDParam::IsOverTh(q)
	112	&& iCh>=AliHMPIDParam::kMinCh
	113	&& iCh<=AliHMPIDParam::kMaxCh) new((*pLst[iCh])[iCnt[iCh]++]) AliHMPIDDigit(iPad,(Int_t)q,aTids); //do not create digit for the very first sdigit
	114
ae5a42aa	115	iPad=pSdig->Pad(); iCh=AliHMPIDParam::A2C(iPad); //new sdigit comes, reset collectors
d3da6dc4	116	iNdigPad=1;
d3da6dc4	117	aTids[0]=pSdig->GetTrack(0);aTids[1]=aTids[2]=-1;
c5c19d6a	118	q=pSdig->Q();
94b1fbfa	119	if(fgDoNoise) q+=arrNoise[iCh][pSdig->Pc()][pSdig->PadPcX()][pSdig->PadPcY()];
c5c19d6a	120	arrNoise[iCh][pSdig->Pc()][pSdig->PadPcX()][pSdig->PadPcY()]=0;
d3da6dc4	121	}//sdigits loop (sorted)
d3da6dc4	122
aa8db9d6	123	if(AliHMPIDParam::IsOverTh(q)
	124	&& iCh>=AliHMPIDParam::kMinCh
	125	&& iCh<=AliHMPIDParam::kMaxCh) new((*pLst[iCh])[iCnt[iCh]++]) AliHMPIDDigit(iPad,(Int_t)q,aTids); //add the last one, in case of empty sdigits list q=-1
	126
c5c19d6a	127	// add noise pad above threshold with no signal merged...if any
94b1fbfa	128	if(!fgDoNoise) return;
c5c19d6a	129	aTids[0]=aTids[1]=aTids[2]=-1;
ec575c20	130	for (Int_t iChCurr=AliHMPIDParam::kMinCh;iChCurr<=AliHMPIDParam::kMaxCh;iChCurr++)
ae5a42aa	131	for (Int_t iPc=AliHMPIDParam::kMinPc;iPc<=AliHMPIDParam::kMaxPc;iPc++)
	132	for(Int_t iPx=AliHMPIDParam::kMinPx;iPx<=AliHMPIDParam::kMaxPx;iPx++)
	133	for(Int_t iPy=AliHMPIDParam::kMinPy;iPy<=AliHMPIDParam::kMaxPy;iPy++) {
ec575c20	134	Float_t qNoise = arrNoise[iChCurr][iPc][iPx][iPy];
ec575c20	135	if(AliHMPIDParam::IsOverTh(qNoise)) new((*pLst[iChCurr])[iCnt[iChCurr]++]) AliHMPIDDigit(AliHMPIDParam::Abs(iChCurr,iPc,iPx,iPy),(Int_t)qNoise,aTids);
c5c19d6a	136	}
c5c19d6a	137
d3da6dc4	138	}//Sdi2Dig()
d3da6dc4	139	//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++