[u/mrichter/AliRoot.git] / MFT / AliMFTDigitizer.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.                  *
 **************************************************************************/

//====================================================================================================================================================
//
//      Digitizer class for the ALICE Muon Forward Tracker
//
//      Contact author: antonio.uras@cern.ch
//
//====================================================================================================================================================

#include "AliRun.h"
#include "AliRunLoader.h"
#include "AliDigitizationInput.h"
#include "AliLoader.h"
#include "AliLog.h"
#include "AliMFTDigitizer.h"
#include "AliMFTDigit.h"
#include "AliMFT.h"
#include "AliMFTSegmentation.h"
#include "TObjArray.h"
#include "TClonesArray.h"
#include "AliDigitizer.h"

ClassImp(AliMFTDigitizer)

//====================================================================================================================================================
    
AliMFTDigitizer::AliMFTDigitizer():
  AliDigitizer(),
  fNPlanes(0),
  fSegmentation(0)
{

  // default constructor 

}

//====================================================================================================================================================

AliMFTDigitizer::AliMFTDigitizer(AliDigitizationInput *digInp):
  AliDigitizer(digInp),
  fNPlanes(0),
  fSegmentation(0)
{

}

//====================================================================================================================================================

void AliMFTDigitizer::Digitize(Option_t*) {

  // This method is responsible for merging sdigits to a list of digits

  AliDebug(1, "************************************************************************");
  AliDebug(1, "************************ AliMFTDigitizer::Digitize *********************");
  AliDebug(1, "************************************************************************");
  
  if (!fSegmentation) {
    fSegmentation = new AliMFTSegmentation("AliMFTGeometry.root");
    fNPlanes = fSegmentation -> GetNPlanes();
  }

  AliDebug(1, Form("nPlanes = %d",fNPlanes));

  AliDebug(1,Form("Start with %i input(s) for event %i", fDigInput->GetNinputs(), fDigInput->GetOutputEventNr()));
    
  AliRunLoader *pInRunLoader=0;
  AliLoader    *pInMFTLoader=0;
  
  TClonesArray sDigits[fNMaxPlanes];
  for (Int_t iPlane=0; iPlane<fNMaxPlanes; iPlane++) sDigits[iPlane].SetClass("AliMFTDigit");  // tmp storage for sdigits sum up from all input files
  
  // filling the arrays of sdigits...

  for (Int_t iFile=0; iFile<fDigInput->GetNinputs(); iFile++) {
    
    pInRunLoader = AliRunLoader::GetRunLoader(fDigInput->GetInputFolderName(iFile));
    pInMFTLoader = pInRunLoader->GetLoader("MFTLoader"); 
    if (!pInMFTLoader) {
      AliDebug(1,"no MFT lodader, checking in the other input \n"); 
      continue;
    }
    
    if (!pInRunLoader->GetAliRun()) pInRunLoader->LoadgAlice();
    AliMFT *pInMFT = (AliMFT*) pInRunLoader->GetAliRun()->GetDetector("MFT"); 
    
    AliDebug(1, "pInMFTLoader->LoadSDigits()...");
    pInMFTLoader->LoadSDigits();
    AliDebug(1, "... done!");
    AliDebug(1, "    pInMFTLoader->TreeS()->GetEntry(0);");
    pInMFTLoader->TreeS()->GetEntry(0);
    AliDebug(1, "... done!");
    
    for (Int_t iPlane=0; iPlane<pInMFT->GetSDigitsList()->GetEntries(); iPlane++) {      
      for(Int_t iSDig=0; iSDig<((TClonesArray*)pInMFT->GetSDigitsList()->At(iPlane))->GetEntries(); iSDig++) {
	AliDebug(2, Form("Reading digit %03d of plane %02d (A)", iSDig, iPlane));
	AliMFTDigit *pSDig = (AliMFTDigit*) ((TClonesArray*)pInMFT->GetSDigitsList()->At(iPlane))->At(iSDig);
	AliDebug(2, Form("Reading digit %03d of plane %02d (B)", iSDig, iPlane));
	pSDig->AddOffset2TrackID(fDigInput->GetMask(iFile));   // -> To be introduced for merging (since all inputs count tracks independently from 0)
	AliDebug(2, Form("Reading digit %03d of plane %02d (C)", iSDig, iPlane));
	new ((sDigits[iPlane])[sDigits[iPlane].GetEntries()]) AliMFTDigit(*pSDig);  
	AliDebug(2, Form("Reading digit %03d of plane %02d (D)", iSDig, iPlane));
      }
    }
    
    pInMFTLoader->UnloadSDigits();   
    pInMFT->ResetSDigits();

  }
  
  AliRunLoader *pOutRunLoader = AliRunLoader::GetRunLoader(fDigInput->GetOutputFolderName());  // open output stream (only 1 possible)
  AliLoader    *pOutMFTLoader = pOutRunLoader->GetLoader("MFTLoader");                        
  AliRun       *pAliRun       = pOutRunLoader->GetAliRun();
  AliMFT       *pOutMFT       = (AliMFT*) pAliRun->GetDetector("MFT");      
  pOutMFTLoader->MakeTree("D");   
  pOutMFT->MakeBranch("D");
  pOutMFT->SetTreeAddress();
  
  SDigits2Digits(sDigits, pOutMFT->GetDigitsList());   // here the sdigits are merged into digits
  
  pOutMFTLoader->TreeD()->Fill();              // fill the output tree with the list of digits
  pOutMFTLoader->WriteDigits("OVERWRITE");     // serialize them to file
  
  for (Int_t iPlane=0; iPlane<fNPlanes; iPlane++) sDigits[iPlane].Clear();   // remove all tmp sdigits
  pOutMFTLoader->UnloadDigits();   
  pOutMFT->ResetDigits(); 

}

//====================================================================================================================================================

void AliMFTDigitizer::SDigits2Digits(TClonesArray *pSDigitList, TObjArray *pDigitList) {   

  TClonesArray *myDigitList[fNMaxPlanes] = {0};
 
  for (Int_t iPlane=0; iPlane<fNPlanes; iPlane++) { 
    myDigitList[iPlane] = (TClonesArray*)(*pDigitList)[iPlane];
    if (myDigitList[iPlane]->GetEntries()!=0) AliErrorClass("Some of digits lists is not empty");   // in principle those lists should be empty 
  }
   
  AliDebug(1,"starting loop over planes");
   
  for (Int_t iPlane=0; iPlane<fNPlanes; iPlane++) {
     
    AliMFTDigit *newDig=NULL;
    AliMFTDigit *oldDig=NULL;
    pSDigitList[iPlane].Sort();

    AliDebug(1,"starting loop over sdigits to create digits");
    AliDebug(1, Form("MFT plane #%02d has %d SDigits", iPlane, Int_t(pSDigitList[iPlane].GetEntries())));

    for (Int_t iSDig=0; iSDig<pSDigitList[iPlane].GetEntries(); iSDig++) {

      newDig = (AliMFTDigit*) (pSDigitList[iPlane].At(iSDig));
      Bool_t digitExists = kFALSE;
      Int_t nDigits = myDigitList[iPlane]->GetEntries();
      
      for (Int_t iDig=0; iDig<nDigits; iDig++) {
	oldDig = (AliMFTDigit*) (myDigitList[iPlane]->At(iDig));
	if (newDig->GetDetElemID()==oldDig->GetDetElemID() &&
	    newDig->GetPixelX()==oldDig->GetPixelX() &&
	    newDig->GetPixelY()==oldDig->GetPixelY() &&
	    newDig->GetPixelZ()==oldDig->GetPixelZ()) {
	  digitExists = kTRUE;
	  MergeDigits(oldDig, newDig);
	  break;
	}
      }

      if (!digitExists) new ((*myDigitList[iPlane])[myDigitList[iPlane]->GetEntries()]) AliMFTDigit(*newDig);

    }
    
    AliDebug(1, Form("MFT plane #%02d has %d Digits", iPlane, Int_t(myDigitList[iPlane]->GetEntries())));
    
//     for (Int_t iDigit=0; iDigit<myDigitList[iPlane]->GetEntries(); iDigit++) {
//       AliDebug(1, Form("Digit %03d of MFT plane %02d has pixel coordinates (%05d, %05d)", 
// 		       iDigit, iPlane, ((AliMFTDigit*) myDigitList[iPlane]->At(iDigit))->GetPixelX(), ((AliMFTDigit*) myDigitList[iPlane]->At(iDigit))->GetPixelY()) );
//     }

    AliDebug(1, "ending loop over sdigits to create digits");

  }

  AliDebug(1,"ending loop over layers");  

}

//====================================================================================================================================================

void AliMFTDigitizer::MergeDigits(AliMFTDigit *mainDig, AliMFTDigit *digToSum) {
  
  mainDig -> SetEloss(mainDig->GetEloss() + digToSum->GetEloss());
  
  Bool_t trackExists = kFALSE;
  for (Int_t iTrack=0; iTrack<mainDig->GetNMCTracks(); iTrack++) {
    if (digToSum->GetMCLabel(0) == mainDig->GetMCLabel(iTrack)) {
      trackExists = kTRUE;
      break;
    }
  }
  
  if (!trackExists) mainDig->AddMCLabel(digToSum->GetMCLabel(0));
  
}

//====================================================================================================================================================
Commit	Line	Data
820b4d9e	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	//
	18	// Digitizer class for the ALICE Muon Forward Tracker
	19	//
	20	// Contact author: antonio.uras@cern.ch
	21	//
	22	//====================================================================================================================================================
	23
	24	#include "AliRun.h"
	25	#include "AliRunLoader.h"
f21fc003	26	#include "AliDigitizationInput.h"
820b4d9e	27	#include "AliLoader.h"
	28	#include "AliLog.h"
	29	#include "AliMFTDigitizer.h"
	30	#include "AliMFTDigit.h"
	31	#include "AliMFT.h"
	32	#include "AliMFTSegmentation.h"
	33	#include "TObjArray.h"
	34	#include "TClonesArray.h"
	35	#include "AliDigitizer.h"
	36
	37	ClassImp(AliMFTDigitizer)
	38
	39	//====================================================================================================================================================
	40
	41	AliMFTDigitizer::AliMFTDigitizer():
	42	AliDigitizer(),
	43	fNPlanes(0),
	44	fSegmentation(0)
	45	{
	46
	47	// default constructor
	48
	49	}
	50
	51	//====================================================================================================================================================
	52
d4643a10	53	AliMFTDigitizer::AliMFTDigitizer(AliDigitizationInput *digInp):
f21fc003	54	AliDigitizer(digInp),
820b4d9e	55	fNPlanes(0),
	56	fSegmentation(0)
	57	{
	58
	59	}
	60
	61	//====================================================================================================================================================
	62
f21fc003	63	void AliMFTDigitizer::Digitize(Option_t*) {
820b4d9e	64
	65	// This method is responsible for merging sdigits to a list of digits
	66
	67	AliDebug(1, "************************************************************************");
d4643a10	68	AliDebug(1, "********************** AliMFTDigitizer::Digitize *******************");
820b4d9e	69	AliDebug(1, "************************************************************************");
	70
	71	if (!fSegmentation) {
	72	fSegmentation = new AliMFTSegmentation("AliMFTGeometry.root");
	73	fNPlanes = fSegmentation -> GetNPlanes();
	74	}
	75
	76	AliDebug(1, Form("nPlanes = %d",fNPlanes));
	77
f21fc003	78	AliDebug(1,Form("Start with %i input(s) for event %i", fDigInput->GetNinputs(), fDigInput->GetOutputEventNr()));
820b4d9e	79
	80	AliRunLoader *pInRunLoader=0;
	81	AliLoader *pInMFTLoader=0;
	82
	83	TClonesArray sDigits[fNMaxPlanes];
	84	for (Int_t iPlane=0; iPlane<fNMaxPlanes; iPlane++) sDigits[iPlane].SetClass("AliMFTDigit"); // tmp storage for sdigits sum up from all input files
	85
	86	// filling the arrays of sdigits...
	87
f21fc003	88	for (Int_t iFile=0; iFile<fDigInput->GetNinputs(); iFile++) {
820b4d9e	89
f21fc003	90	pInRunLoader = AliRunLoader::GetRunLoader(fDigInput->GetInputFolderName(iFile));
820b4d9e	91	pInMFTLoader = pInRunLoader->GetLoader("MFTLoader");
	92	if (!pInMFTLoader) {
	93	AliDebug(1,"no MFT lodader, checking in the other input \n");
	94	continue;
	95	}
	96
	97	if (!pInRunLoader->GetAliRun()) pInRunLoader->LoadgAlice();
	98	AliMFT pInMFT = (AliMFT) pInRunLoader->GetAliRun()->GetDetector("MFT");
	99
	100	AliDebug(1, "pInMFTLoader->LoadSDigits()...");
	101	pInMFTLoader->LoadSDigits();
	102	AliDebug(1, "... done!");
	103	AliDebug(1, " pInMFTLoader->TreeS()->GetEntry(0);");
	104	pInMFTLoader->TreeS()->GetEntry(0);
	105	AliDebug(1, "... done!");
	106
	107	for (Int_t iPlane=0; iPlane<pInMFT->GetSDigitsList()->GetEntries(); iPlane++) {
	108	for(Int_t iSDig=0; iSDig<((TClonesArray*)pInMFT->GetSDigitsList()->At(iPlane))->GetEntries(); iSDig++) {
	109	AliDebug(2, Form("Reading digit %03d of plane %02d (A)", iSDig, iPlane));
	110	AliMFTDigit pSDig = (AliMFTDigit) ((TClonesArray*)pInMFT->GetSDigitsList()->At(iPlane))->At(iSDig);
	111	AliDebug(2, Form("Reading digit %03d of plane %02d (B)", iSDig, iPlane));
f21fc003	112	pSDig->AddOffset2TrackID(fDigInput->GetMask(iFile)); // -> To be introduced for merging (since all inputs count tracks independently from 0)
820b4d9e	113	AliDebug(2, Form("Reading digit %03d of plane %02d (C)", iSDig, iPlane));
	114	new ((sDigits[iPlane])[sDigits[iPlane].GetEntries()]) AliMFTDigit(*pSDig);
	115	AliDebug(2, Form("Reading digit %03d of plane %02d (D)", iSDig, iPlane));
	116	}
	117	}
	118
	119	pInMFTLoader->UnloadSDigits();
	120	pInMFT->ResetSDigits();
	121
	122	}
	123
f21fc003	124	AliRunLoader *pOutRunLoader = AliRunLoader::GetRunLoader(fDigInput->GetOutputFolderName()); // open output stream (only 1 possible)
820b4d9e	125	AliLoader *pOutMFTLoader = pOutRunLoader->GetLoader("MFTLoader");
	126	AliRun *pAliRun = pOutRunLoader->GetAliRun();
	127	AliMFT pOutMFT = (AliMFT) pAliRun->GetDetector("MFT");
	128	pOutMFTLoader->MakeTree("D");
	129	pOutMFT->MakeBranch("D");
	130	pOutMFT->SetTreeAddress();
	131
	132	SDigits2Digits(sDigits, pOutMFT->GetDigitsList()); // here the sdigits are merged into digits
	133
	134	pOutMFTLoader->TreeD()->Fill(); // fill the output tree with the list of digits
	135	pOutMFTLoader->WriteDigits("OVERWRITE"); // serialize them to file
	136
	137	for (Int_t iPlane=0; iPlane<fNPlanes; iPlane++) sDigits[iPlane].Clear(); // remove all tmp sdigits
	138	pOutMFTLoader->UnloadDigits();
	139	pOutMFT->ResetDigits();
	140
	141	}
	142
	143	//====================================================================================================================================================
	144
	145	void AliMFTDigitizer::SDigits2Digits(TClonesArray pSDigitList, TObjArray pDigitList) {
	146
	147	TClonesArray *myDigitList[fNMaxPlanes] = {0};
	148
	149	for (Int_t iPlane=0; iPlane<fNPlanes; iPlane++) {
	150	myDigitList[iPlane] = (TClonesArray)(pDigitList)[iPlane];
	151	if (myDigitList[iPlane]->GetEntries()!=0) AliErrorClass("Some of digits lists is not empty"); // in principle those lists should be empty
	152	}
	153
	154	AliDebug(1,"starting loop over planes");
	155
	156	for (Int_t iPlane=0; iPlane<fNPlanes; iPlane++) {
	157
	158	AliMFTDigit *newDig=NULL;
	159	AliMFTDigit *oldDig=NULL;
	160	pSDigitList[iPlane].Sort();
	161
	162	AliDebug(1,"starting loop over sdigits to create digits");
	163	AliDebug(1, Form("MFT plane #%02d has %d SDigits", iPlane, Int_t(pSDigitList[iPlane].GetEntries())));
	164
	165	for (Int_t iSDig=0; iSDig<pSDigitList[iPlane].GetEntries(); iSDig++) {
	166
	167	newDig = (AliMFTDigit*) (pSDigitList[iPlane].At(iSDig));
	168	Bool_t digitExists = kFALSE;
	169	Int_t nDigits = myDigitList[iPlane]->GetEntries();
	170
	171	for (Int_t iDig=0; iDig<nDigits; iDig++) {
	172	oldDig = (AliMFTDigit*) (myDigitList[iPlane]->At(iDig));
	173	if (newDig->GetDetElemID()==oldDig->GetDetElemID() &&
	174	newDig->GetPixelX()==oldDig->GetPixelX() &&
	175	newDig->GetPixelY()==oldDig->GetPixelY() &&
	176	newDig->GetPixelZ()==oldDig->GetPixelZ()) {
	177	digitExists = kTRUE;
	178	MergeDigits(oldDig, newDig);
	179	break;
	180	}
	181	}
	182
	183	if (!digitExists) new ((myDigitList[iPlane])[myDigitList[iPlane]->GetEntries()]) AliMFTDigit(newDig);
	184
	185	}
	186
	187	AliDebug(1, Form("MFT plane #%02d has %d Digits", iPlane, Int_t(myDigitList[iPlane]->GetEntries())));
	188
189	// for (Int_t iDigit=0; iDigit<myDigitList[iPlane]->GetEntries(); iDigit++) {
190	// AliDebug(1, Form("Digit %03d of MFT plane %02d has pixel coordinates (%05d, %05d)",
191	// iDigit, iPlane, ((AliMFTDigit) myDigitList[iPlane]->At(iDigit))->GetPixelX(), ((AliMFTDigit) myDigitList[iPlane]->At(iDigit))->GetPixelY()) );
192	// }
193
194	AliDebug(1, "ending loop over sdigits to create digits");
195
196	}
197
198	AliDebug(1,"ending loop over layers");
199
200	}
201
202	//====================================================================================================================================================
203
204	void AliMFTDigitizer::MergeDigits(AliMFTDigit mainDig, AliMFTDigit digToSum) {
205
206	mainDig -> SetEloss(mainDig->GetEloss() + digToSum->GetEloss());
207
208	Bool_t trackExists = kFALSE;
209	for (Int_t iTrack=0; iTrack<mainDig->GetNMCTracks(); iTrack++) {
210	if (digToSum->GetMCLabel(0) == mainDig->GetMCLabel(iTrack)) {
211	trackExists = kTRUE;
212	break;
213	}
214	}
215
216	if (!trackExists) mainDig->AddMCLabel(digToSum->GetMCLabel(0));
217
218	}
219
220	//====================================================================================================================================================
221