Added function AliDigits2RootFile

diff --git a/HLT/src/AliL3FileHandler.cxx b/HLT/src/AliL3FileHandler.cxx
index 2bac33fb39f403c82cfe7f8e83c4d0dc4a49f019..86464eeff042fd3c5f54d52136aebc320a568215 100644 (file)
--- a/HLT/src/AliL3FileHandler.cxx
+++ b/HLT/src/AliL3FileHandler.cxx
@@ -14,6 +14,7 @@
 #include "AliL3MemHandler.h"
 #include "AliL3FileHandler.h"
 
+#include "AliTPCDigitsArray.h"
 #include "AliTPCClustersArray.h"
 #include "AliTPCcluster.h"
 #include "AliTPCClustersRow.h"
@@ -284,6 +285,83 @@ AliL3DigitRowData * AliL3FileHandler::AliDigits2Memory(UInt_t & nrow){
   return data;
 }
 
+void AliL3FileHandler::AliDigits2RootFile(AliL3DigitRowData *rowPt,Char_t *new_digitsfile)
+{
+  //Write digits to a new alirootfile.
+
+  if(!fInAli)
+    {
+      printf("AliL3FileHandler::AliDigits2RootFile : No input alirootfile\n");
+      return;
+    }
+  if(!fParam)
+    {
+      printf("AliL3FileHandler::AliDigits2RootFile : No parameter object. Run on rootfile\n");
+      return;
+    }
+  if(!fTransformer)
+    {
+      printf("AliL3FileHandler::AliDigits2RootFile : No transform object\n");
+      return;
+    }
+  
+  //Get the original digitstree:
+  fInAli->cd();
+  AliTPCDigitsArray *old_array = new AliTPCDigitsArray();
+  old_array->Setup(fParam);
+  old_array->SetClass("AliSimDigits");
+  Bool_t ok = old_array->ConnectTree("TreeD_75x40_100x60");
+  if(!ok)
+    {
+      printf("AliL3FileHandler::AliDigits2RootFile : No digits tree object\n");
+      return;
+    }
+    
+  TFile *digFile = new TFile(new_digitsfile,"RECREATE");
+  digFile->cd();
+  
+  //setup a new one:
+  AliTPCDigitsArray *arr = new AliTPCDigitsArray; 
+  arr->SetClass("AliSimDigits");
+  arr->Setup(fParam);
+  arr->MakeTree();
+  if(fRowMin !=0 || fRowMax != 175)
+    printf("\n AliL3FileHandler::AliDigits2RootFile : Rather stupid row numbers...%d %d\n\n",fRowMin,fRowMax);
+  for(Int_t i=fRowMin; i<=fRowMax; i++)
+    {
+      Int_t sector,row;
+      fTransformer->Slice2Sector(fSlice,i,sector,row);
+      AliDigits * dig = arr->CreateRow(sector,row);
+      AliDigits *old_dig = old_array->LoadRow(sector,row);
+      if(!old_dig)
+       printf("AliL3FileHandler::AliDigits2RootFile : No padrow %d %d\n",sector,row);
+      
+      AliL3DigitData *digPt = rowPt->fDigitData;
+      for(UInt_t j=0; j<rowPt->fNDigit; j++)
+       {
+         UShort_t charge = digPt[j].fCharge;
+         UChar_t pad = digPt[j].fPad;
+         UShort_t time = digPt[j].fTime;
+         dig->SetDigitFast(charge,time,pad);
+         ((AliSimDigits*)dig)->SetTrackIDFast(((AliSimDigits*)old_dig)->GetTrackID((Int_t)time,(Int_t)pad,0),time,pad,0);
+         ((AliSimDigits*)dig)->SetTrackIDFast(((AliSimDigits*)old_dig)->GetTrackID((Int_t)time,(Int_t)pad,1),time,pad,1);
+         ((AliSimDigits*)dig)->SetTrackIDFast(((AliSimDigits*)old_dig)->GetTrackID((Int_t)time,(Int_t)pad,2),time,pad,2);
+         
+       }
+      UpdateRowPointer(rowPt);
+      arr->StoreRow(sector,row);
+      arr->ClearRow(sector,row);  
+      old_array->ClearRow(sector,row);
+    }
+  digFile->cd();
+  char treeName[100];
+  sprintf(treeName,"TreeD_%s",fParam->GetTitle());
+  arr->GetTree()->Write(treeName,TObject::kOverwrite);
+  fParam->Write(fParam->GetTitle());
+  digFile->Close();
+  delete digFile;
+}
+
 ///////////////////////////////////////// Point IO  
 Bool_t AliL3FileHandler::AliPoints2Binary(){
   Bool_t out = kTRUE;

diff --git a/HLT/src/AliL3FileHandler.h b/HLT/src/AliL3FileHandler.h
index 6038b2f29593b61391a403fabefe930d1f5af141..4014292ac73af9fee2642bbc46e6e66fd325df79 100644 (file)
--- a/HLT/src/AliL3FileHandler.h
+++ b/HLT/src/AliL3FileHandler.h
@@ -43,6 +43,7 @@ class AliL3FileHandler:public AliL3MemHandler{
   Bool_t AliDigits2Binary();
   AliL3DigitRowData *AliDigits2Memory(UInt_t & nrow); //Allocates Memory
   Bool_t AliDigits2CompBinary();  
+  void AliDigits2RootFile(AliL3DigitRowData *rowPt,Char_t *new_digitsfile);
 
   //Point IO
   Bool_t AliPoints2Binary();