[u/mrichter/AliRoot.git] / HLT / src / AliL3FileHandler.cxx

//$Id$

// Author: Uli Frankenfeld <mailto:franken@fi.uib.no>, Anders Vestbo <mailto:vestbo$fi.uib.no>
//*-- Copyright &copy Uli 

#include <math.h>
#include <iostream.h>

#include "AliL3Transform.h"
#include "AliL3Logging.h"
#include "AliL3MemHandler.h"
#include "AliL3FileHandler.h"

#include "AliTPCDigitsArray.h"
#include "AliTPCClustersArray.h"
#include "AliTPCcluster.h"
#include "AliTPCClustersRow.h"

#include "AliL3DigitData.h"
#include "AliL3TrackSegmentData.h"
#include "AliL3SpacePointData.h"
#include "AliL3TrackArray.h"

//_____________________________________________________________
// AliL3FileHandler
//
// The HLT ROOT <-> binary files handling class
//
// This class provides the interface between AliROOT files,
// and HLT binary files. It should be used for converting 
// TPC data stored in AliROOT format (outputfile from a simulation),
// into the data format currently used by in the HLT framework. 
// This enables the possibility to always use the same data format, 
// whether you are using a binary file as an input, or a AliROOT file.
//
// For example on how to create binary files from a AliROOT simulation,
// see example macro exa/Binary.C.
//
// For reading a AliROOT file into HLT format in memory, do the following:
//
// AliL3FileHandler file;
// file.SetAliInput("galice.root");
// AliL3DigitRowData *dataPt = (AliL3DigitRowData*)file.AliDigits2Memory(nrows,eventnr);
// 
// All the data are then stored in memory and accessible via the pointer dataPt.
// Accesing the data is then identical to the example 1) showed in AliL3MemHandler class.
//
// For converting the data back, and writing it to a new AliROOT file do:
//
// AliL3FileHandler file;
// file.SetAliInput("galice.root");
// file.Init(slice,patch,NumberOfRowsInPatch);
// file.AliDigits2RootFile(dataPt,"new_galice.root");
// file.CloseAliInput();

ClassImp(AliL3FileHandler)

AliL3FileHandler::AliL3FileHandler()
{
  //Default constructor
  fInAli = 0;
  fParam = 0;
  fMC =0;
  fLastIndex=0;
  fDigits=0;
  fDigitsTree=0;
}

AliL3FileHandler::~AliL3FileHandler()
{
  //Destructor
  if(fMC) CloseMCOutput();
  if(fDigitsTree) delete fDigitsTree;
  if(fInAli) CloseAliInput();
  
}

Bool_t AliL3FileHandler::SetMCOutput(char *name)
{
  fMC = fopen(name,"w");
  if(!fMC){
    LOG(AliL3Log::kWarning,"AliL3FileHandler::SetMCOutput","File Open")
      <<"Pointer to File = 0x0 "<<ENDLOG;
    return kFALSE;
  }
  return kTRUE;
}

Bool_t AliL3FileHandler::SetMCOutput(FILE *file)
{
  fMC = file;
  if(!fMC){
    LOG(AliL3Log::kWarning,"AliL3FileHandler::SetMCOutput","File Open")
      <<"Pointer to File = 0x0 "<<ENDLOG;
    return kFALSE;
  }
  return kTRUE;
}

void AliL3FileHandler::CloseMCOutput()
{
  if(!fMC){
    LOG(AliL3Log::kWarning,"AliL3FileHandler::CloseMCOutPut","File Close")
      <<"Nothing to Close"<<ENDLOG;
    return;
  }
  fclose(fMC);
  fMC =0;
}

Bool_t AliL3FileHandler::SetAliInput()
{
  if(!fInAli->IsOpen()){
    LOG(AliL3Log::kError,"AliL3FileHandler::SetAliInput","File Open")
      <<"Ali File "<<fInAli->GetName()<<" does not exist"<<ENDLOG;
    return kFALSE;
  }
  fParam = (AliTPCParam*)fInAli->Get("75x40_100x60");
  if(!fParam){ 
    LOG(AliL3Log::kError,"AliL3FileHandler::SetAliInput","File Open")
      <<"No AliTPCParam 75x40_100x60 in File "<<fInAli->GetName()<<ENDLOG;
    return kFALSE;
  }
  return kTRUE;
}

Bool_t AliL3FileHandler::SetAliInput(char *name)
{
  //Open the AliROOT file with name.
  
  fInAli= new TFile(name,"READ");
  if(!fInAli){
    LOG(AliL3Log::kWarning,"AliL3FileHandler::SetAliInput","File Open")
    <<"Pointer to TFile = 0x0 "<<ENDLOG;
    return kFALSE;
  }
  return SetAliInput();
}

Bool_t AliL3FileHandler::SetAliInput(TFile *file)
{
  //Specify already opened AliROOT file to use as an input.
  
  fInAli=file;
  if(!fInAli){
    LOG(AliL3Log::kWarning,"AliL3FileHandler::SetAliInput","File Open")
    <<"Pointer to TFile = 0x0 "<<ENDLOG;
    return kFALSE;
  }
  return SetAliInput();
}

void AliL3FileHandler::CloseAliInput()
{
  if(!fInAli){
    LOG(AliL3Log::kWarning,"AliL3FileHandler::CloseAliInput","File Close")
      <<"Nothing to Close"<<ENDLOG;
    return;
  }
  if(fInAli->IsOpen()) fInAli->Close();
  delete fInAli;
  fInAli = 0;
  
}

Bool_t AliL3FileHandler::IsDigit()
{
  //Check if there is a TPC digit tree in the current file.
  //Return kTRUE if tree was found, and kFALSE if not found.
  
  if(!fInAli){
    LOG(AliL3Log::kWarning,"AliL3FileHandler::IsDigit","File")
    <<"Pointer to TFile = 0x0 "<<ENDLOG;
    return kTRUE;  //may you are use binary input which is Digits!!
  }
  TTree *t=(TTree*)fInAli->Get("TreeD_75x40_100x60_0");
  if(t){
    LOG(AliL3Log::kInformational,"AliL3FileHandler::IsDigit","File Type")
    <<"Found Digit Tree -> Use Fast Cluster Finder"<<ENDLOG;
    return kTRUE;
  }
  else{
    LOG(AliL3Log::kInformational,"AliL3FileHandler::IsDigit","File Type")
    <<"No Digit Tree -> Use Cluster Tree"<<ENDLOG;
    return kFALSE;
  }
}

///////////////////////////////////////// Digit IO  
Bool_t AliL3FileHandler::AliDigits2Binary(Int_t event)
{
  
  Bool_t out = kTRUE;
  UInt_t nrow;
  AliL3DigitRowData* data = AliDigits2Memory(nrow,event);
  out = Memory2Binary(nrow,data);
  Free();
  return out;
}


Bool_t AliL3FileHandler::AliDigits2CompBinary(Int_t event)
{
  //Convert AliROOT TPC data, into HLT data format.
  //event specifies the event you want in the aliroot file.
  
  Bool_t out = kTRUE;
  UInt_t ndigits=0;
  AliL3DigitRowData* digits = AliDigits2Memory(ndigits,event);
  out = Memory2CompBinary(ndigits,digits);
  Free();
  return out;
}


AliL3DigitRowData * AliL3FileHandler::AliDigits2Memory(UInt_t & nrow,Int_t event)
{
  //Read data from AliROOT file into memory, and store it in the HLT data format.
  //Returns a pointer to the data.

  AliL3DigitRowData *data = 0;
  nrow=0;
  if(!fInAli){
    LOG(AliL3Log::kWarning,"AliL3FileHandler::AliDigits2Memory","File")
    <<"No Input avalible: no object TFile"<<ENDLOG;
    return 0; 
  }
  if(!fInAli->IsOpen()){
    LOG(AliL3Log::kWarning,"AliL3FileHandler::AliDigits2Memory","File")
    <<"No Input avalible: TFile not opend"<<ENDLOG;
    return 0;
  }
  
  if(!fDigitsTree)
    GetDigitsTree(event);
    
  UShort_t dig;
  Int_t time,pad,sector,row;
  Int_t nrows=0;
  Int_t ndigitcount=0;
  Int_t entries = (Int_t)fDigitsTree->GetEntries();
  Int_t ndigits[entries];
  Int_t lslice,lrow;
  
  for(Int_t n=fLastIndex; n<fDigitsTree->GetEntries(); n++)
    {
      fDigitsTree->GetEvent(n);
      fParam->AdjustSectorRow(fDigits->GetID(),sector,row);
      AliL3Transform::Sector2Slice(lslice,lrow,sector,row);
      //if(fSlice != lslice || lrow<fRowMin || lrow>fRowMax) continue;
      if(lslice < fSlice) continue;
      if(lslice != fSlice) break;
      if(lrow < fRowMin) continue;
      if(lrow > fRowMax) break;

      Float_t xyz[3];
      ndigits[lrow] = 0;
      fDigits->First();
      do {
        time=fDigits->CurrentRow();
        pad=fDigits->CurrentColumn();
        dig = fDigits->GetDigit(time,pad);
        if(dig<=fParam->GetZeroSup()) continue;
        if(time < fParam->GetMaxTBin()-1 && time > 0)
          if(fDigits->GetDigit(time+1,pad) <= fParam->GetZeroSup()
             && fDigits->GetDigit(time-1,pad) <= fParam->GetZeroSup())
            continue;

        AliL3Transform::Raw2Local(xyz,sector,row,pad,time);
        if(fParam->GetPadRowRadii(sector,row)<230./250.*fabs(xyz[2]))
          continue;

        ndigits[lrow]++; //for this row only
        ndigitcount++;  //total number of digits to be published

      } while (fDigits->Next());

      nrows++;
    }
  Int_t size = sizeof(AliL3DigitData)*ndigitcount
    + nrows*sizeof(AliL3DigitRowData);

  LOG(AliL3Log::kDebug,"AliL3FileHandler::AliDigits2Memory","Digits")
    <<AliL3Log::kDec<<"Found "<<ndigitcount<<" Digits"<<ENDLOG;
  
  data=(AliL3DigitRowData*) Allocate(size);
  nrow = (UInt_t)nrows;
  AliL3DigitRowData *tempPt = data;
  for(Int_t n=fLastIndex; n<fDigitsTree->GetEntries(); n++)
    {
      fDigitsTree->GetEvent(n);
      Float_t xyz[3];
      fParam->AdjustSectorRow(fDigits->GetID(),sector,row);
      AliL3Transform::Sector2Slice(lslice,lrow,sector,row);
      //if(fSlice != lslice || lrow<fRowMin || lrow>fRowMax) continue;
      if(lslice < fSlice) continue;
      if(lslice != fSlice) break;
      if(lrow < fRowMin) continue;
      if(lrow > fRowMax) break;

      tempPt->fRow = lrow;
      tempPt->fNDigit = ndigits[lrow];

      Int_t localcount=0;
      fDigits->First();
      do {
        //dig=fDigits->CurrentDigit();
	//if (dig<=fParam->GetZeroSup()) continue;
        time=fDigits->CurrentRow();
        pad=fDigits->CurrentColumn();
        dig = fDigits->GetDigit(time,pad);
	if (dig <= fParam->GetZeroSup()) continue;
	if(time < fParam->GetMaxTBin()-1 && time > 0)
          if(fDigits->GetDigit(time-1,pad) <= fParam->GetZeroSup() &&
             fDigits->GetDigit(time+1,pad) <= fParam->GetZeroSup()) continue;

        //Exclude data outside cone:
        AliL3Transform::Raw2Local(xyz,sector,row,pad,time);
        if(fParam->GetPadRowRadii(sector,row)<230./250.*fabs(xyz[2]))
          continue;

        if(localcount >= ndigits[lrow])
          LOG(AliL3Log::kFatal,"AliL3FileHandler::AliDigits2Binary","Memory")
	    <<AliL3Log::kDec<<"Mismatch: localcount "<<localcount<<" ndigits "
	    <<ndigits[lrow]<<ENDLOG;
	
        tempPt->fDigitData[localcount].fCharge=dig;
        tempPt->fDigitData[localcount].fPad=pad;
        tempPt->fDigitData[localcount].fTime=time;
#ifdef do_mc
	tempPt->fDigitData[localcount].fTrackID[0] = fDigits->GetTrackID(time,pad,0);
	tempPt->fDigitData[localcount].fTrackID[1] = fDigits->GetTrackID(time,pad,1);
	tempPt->fDigitData[localcount].fTrackID[2] = fDigits->GetTrackID(time,pad,2);
#endif
        localcount++;
      } while (fDigits->Next());

      Byte_t *tmp = (Byte_t*)tempPt;
      Int_t size = sizeof(AliL3DigitRowData)
                                      + ndigits[lrow]*sizeof(AliL3DigitData);
      tmp += size;
      tempPt = (AliL3DigitRowData*)tmp;
      //fLastIndex=n;
    }
  //fLastIndex++;
  return data;
}

Bool_t AliL3FileHandler::GetDigitsTree(Int_t event)
{
  //Connects to the TPC digit tree in the AliROOT file.
  
  fInAli->cd();
  Char_t dname[100];
  sprintf(dname,"TreeD_75x40_100x60_%d",event);
  fDigitsTree = (TTree*)fInAli->Get("TreeD_75x40_100x60_0");
  if(!fDigitsTree) 
    {
      LOG(AliL3Log::kError,"AliL3FileHandler::GetDigitsTree","Digits Tree")
	<<AliL3Log::kHex<<"Error getting digitstree "<<(Int_t)fDigitsTree<<ENDLOG;
      return kFALSE;
    }
  fDigitsTree->GetBranch("Segment")->SetAddress(&fDigits);
  return kTRUE;
}

void AliL3FileHandler::AliDigits2RootFile(AliL3DigitRowData *rowPt,Char_t *new_digitsfile)
{
  //Write the data stored in rowPt, into a new AliROOT file.
  //The data is stored in the AliROOT format 
  //This is specially a nice thing if you have modified data, and wants to run it  
  //through the offline reconstruction chain.
  //The arguments is a pointer to the data, and the name of the new AliROOT file.
  //Remember to pass the original AliROOT file (the one that contains the original
  //simulated data) to this object, in order to retrieve the MC id's of the digits.
  
  if(!fInAli)
    {
      printf("AliL3FileHandler::AliDigits2RootFile : No rootfile\n");
      return;
    }
  if(!fParam)
    {
      printf("AliL3FileHandler::AliDigits2RootFile : No parameter object. Run on rootfile\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_0");
  if(!ok)
    {
      printf("AliL3FileHandler::AliDigits2RootFile : No digits tree object\n");
      return;
    }
  
  Bool_t create=kFALSE;
  TFile *digFile;
  
  digFile = TFile::Open(new_digitsfile,"NEW");
  if(digFile->IsOpen())
    {    
      create = kTRUE;
      fParam->Write(fParam->GetTitle());
    }
  else
    {
      LOG(AliL3Log::kDebug,"AliL3FileHandler::AliDigits2RootFile","Rootfile")
	<<"Rootfile did already exist, so I will just open it for updates"<<ENDLOG;
      digFile = TFile::Open(new_digitsfile,"UPDATE");
      create=kFALSE;
    }
  if(!digFile->IsOpen())
    {
      LOG(AliL3Log::kError,"AliL3FileHandler::AliDigits2RootFile","Rootfile")
	<<"Error opening rootfile "<<new_digitsfile<<ENDLOG;
      return;
    }
  
  digFile->cd();
    
  //setup a new one, or connect it to the existing one:
  AliTPCDigitsArray *arr = new AliTPCDigitsArray; 
  arr->SetClass("AliSimDigits");
  arr->Setup(fParam);
  if(create)
    arr->MakeTree();
  else
    {
      Bool_t ok = arr->ConnectTree("TreeD_75x40_100x60_0");
      if(!ok)
	{
	  printf("AliL3FileHandler::AliDigits2RootFile : No digits tree object in existing file\n");
	  return;
	}
    }
  Int_t digcounter=0;

  for(Int_t i=fRowMin; i<=fRowMax; i++)
    {
      
      if((Int_t)rowPt->fRow != i) printf("AliL3FileHandler::AliDigits2RootFile : Mismatching row numbering!!!\n");
            
      Int_t sector,row;
      AliL3Transform::Slice2Sector(fSlice,i,sector,row);
      AliSimDigits * dig = (AliSimDigits*)arr->CreateRow(sector,row);
      AliSimDigits *old_dig = (AliSimDigits*)old_array->LoadRow(sector,row);
      if(!old_dig)
	printf("AliL3FileHandler::AliDigits2RootFile : No padrow %d %d\n",sector,row);
      
      AliL3DigitData *digPt = rowPt->fDigitData;
      digcounter=0;
      for(UInt_t j=0; j<rowPt->fNDigit; j++)
	{
	  Int_t charge = (Int_t)digPt[j].fCharge;
	  Int_t pad = (Int_t)digPt[j].fPad;
	  Int_t time = (Int_t)digPt[j].fTime;
	  
	  if(charge == 0) //Only write the digits that has not been removed
	    continue;
	  digcounter++;
	  dig->SetDigitFast(charge,time,pad);
	  
	  Int_t trackID[3] = {old_dig->GetTrackID(time,pad,0),old_dig->GetTrackID(time,pad,1),old_dig->GetTrackID(time,pad,2)};
	  Int_t s_pad = pad;
	  Int_t s_time = time - 1;
	  while(trackID[0] < 0)
	    {
	      if(s_time >= 0 && s_time < AliL3Transform::GetNTimeBins() && s_pad >= 0 && s_pad < AliL3Transform::GetNPads(i))
		{
		  if(old_dig->GetTrackID(s_time,s_pad,0) > 0)
		    {
		      trackID[0]=old_dig->GetTrackID(s_time,s_pad,0); 
		      trackID[1]=old_dig->GetTrackID(s_time,s_pad,1); 
		      trackID[2]=old_dig->GetTrackID(s_time,s_pad,2); 
		    }
		}
	      if(s_pad == pad && s_time == time - 1)
		s_time = time + 1;
	      else if(s_pad == pad && s_time == time + 1)
		{s_pad = pad - 1; s_time = time;}
	      else if(s_pad == pad - 1 && s_time == time)
		s_time = time - 1;
	      else if(s_pad == pad - 1 && s_time == time - 1)
		s_time = time + 1;
	      else if(s_pad == pad - 1 && s_time == time + 1)
		{s_pad = pad + 1; s_time = time;}
	      else if(s_pad == pad + 1 && s_time == time)
		s_time = time - 1;
	      else if(s_pad == pad + 1 && s_time == time - 1)
		s_time = time + 1;
	      else 
		break;
	    }
	  
	  dig->SetTrackIDFast(trackID[0],time,pad,0);
	  dig->SetTrackIDFast(trackID[1],time,pad,1);
	  dig->SetTrackIDFast(trackID[2],time,pad,2);
	  
	}
      //cout<<"Wrote "<<digcounter<<" on row "<<i<<endl;
      UpdateRowPointer(rowPt);
      arr->StoreRow(sector,row);
      arr->ClearRow(sector,row);  
      old_array->ClearRow(sector,row);
    }
  digFile->cd();
  char treeName[100];
  sprintf(treeName,"TreeD_%s_0",fParam->GetTitle());
  printf("Writing tree to file.....");
  arr->GetTree()->Write(treeName,TObject::kOverwrite);
  printf("done\n");
  digFile->Close();
  //arr->GetTree()->Delete();
  //delete arr;
}

///////////////////////////////////////// Point IO  
Bool_t AliL3FileHandler::AliPoints2Binary(){
  Bool_t out = kTRUE;
  UInt_t npoint;
  AliL3SpacePointData *data = AliPoints2Memory(npoint);
  out = Memory2Binary(npoint,data);
  Free();
  return out;
}

AliL3SpacePointData * AliL3FileHandler::AliPoints2Memory(UInt_t & npoint){
  AliL3SpacePointData *data = 0;
  npoint=0;
  if(!fInAli){
    LOG(AliL3Log::kWarning,"AliL3FileHandler::AliPoints2Memory","File")
    <<"No Input avalible: no object TFile"<<ENDLOG;
    return 0;
  }
  if(!fInAli->IsOpen()){
    LOG(AliL3Log::kWarning,"AliL3FileHandler::AliPoints2Memory","File")
    <<"No Input avalible: TFile not opend"<<ENDLOG;
    return 0;
  }

  TDirectory *savedir = gDirectory;
  fInAli->cd();
  
  Char_t cname[100];
  Int_t eventn = 0;
  sprintf(cname,"TreeC_TPC_%d",eventn);
  AliTPCClustersArray carray;
  carray.Setup(fParam);
  carray.SetClusterType("AliTPCcluster");
  Bool_t clusterok = carray.ConnectTree(cname);
  if(!clusterok) return 0;

  AliTPCClustersRow ** clusterrow = 
               new AliTPCClustersRow*[ (int)carray.GetTree()->GetEntries()];
  Int_t *rows = new int[ (int)carray.GetTree()->GetEntries()];
  Int_t *sects = new int[  (int)carray.GetTree()->GetEntries()];
  Int_t sum=0;

  Int_t lslice,lrow;
  for(Int_t i=0; i<carray.GetTree()->GetEntries(); i++){
    AliSegmentID *s = carray.LoadEntry(i);
    Int_t sector,row;
    fParam->AdjustSectorRow(s->GetID(),sector,row);
    rows[i] = row;
    sects[i] = sector;
    clusterrow[i] = 0;
    AliL3Transform::Sector2Slice(lslice,lrow,sector,row);
    if(fSlice != lslice || lrow<fRowMin || lrow>fRowMax) continue;
    clusterrow[i] = carray.GetRow(sector,row);
    if(clusterrow[i])
      sum+=clusterrow[i]->GetArray()->GetEntriesFast();
  }
  UInt_t size = sum*sizeof(AliL3SpacePointData);

  LOG(AliL3Log::kDebug,"AliL3FileHandler::AliPoints2Memory","File")
  <<AliL3Log::kDec<<"Found "<<sum<<" SpacePoints"<<ENDLOG;

  data = (AliL3SpacePointData *) Allocate(size);
  npoint = sum;
  UInt_t n=0; 
  for(Int_t i=0; i<carray.GetTree()->GetEntries(); i++){
    if(!clusterrow[i]) continue;
    Int_t row = rows[i];
    Int_t sector = sects[i];
    AliL3Transform::Sector2Slice(lslice,lrow,sector,row);
    Int_t entries_in_row = clusterrow[i]->GetArray()->GetEntriesFast();
    for(Int_t j = 0;j<entries_in_row;j++){
      AliTPCcluster *c = (AliTPCcluster*)(*clusterrow[i])[j];
      data[n].fZ = c->GetZ();
      data[n].fY = c->GetY();
      data[n].fX = fParam->GetPadRowRadii(sector,row);
      data[n].fID = n+((fSlice&0x7f)<<25)+((fPatch&0x7)<<22);//uli
      data[n].fPadRow = lrow;
      data[n].fXYErr = c->GetSigmaY2();
      data[n].fZErr = c->GetSigmaZ2();
      if(fMC) fprintf(fMC,"%d %d\n",data[n].fID,c->GetLabel(0));
      n++;
    }
  }
  for(Int_t i=0;i<carray.GetTree()->GetEntries();i++){
    Int_t row = rows[i];
    Int_t sector = sects[i];
    if(carray.GetRow(sector,row))
      carray.ClearRow(sector,row);
  }

  delete [] clusterrow;
  delete [] rows;
  delete [] sects;
  savedir->cd();   

  return data;
}
Commit	Line	Data
31d9405a	1	//$Id$
31d9405a	2
5d2d32e3	3	// Author: Uli Frankenfeld <mailto:franken@fi.uib.no>, Anders Vestbo <mailto:vestbo$fi.uib.no>
b661165c	4	//*-- Copyright &copy Uli
108615fc	5
	6	#include <math.h>
	7	#include <iostream.h>
108615fc	8
	9	#include "AliL3Transform.h"
	10	#include "AliL3Logging.h"
	11	#include "AliL3MemHandler.h"
	12	#include "AliL3FileHandler.h"
	13
0d319e67	14	#include "AliTPCDigitsArray.h"
108615fc	15	#include "AliTPCClustersArray.h"
	16	#include "AliTPCcluster.h"
	17	#include "AliTPCClustersRow.h"
108615fc	18
	19	#include "AliL3DigitData.h"
	20	#include "AliL3TrackSegmentData.h"
	21	#include "AliL3SpacePointData.h"
	22	#include "AliL3TrackArray.h"
b661165c	23
108615fc	24	//_____________________________________________________________
b661165c	25	// AliL3FileHandler
108615fc	26	//
9183aa27	27	// The HLT ROOT <-> binary files handling class
108615fc	28	//
9183aa27	29	// This class provides the interface between AliROOT files,
	30	// and HLT binary files. It should be used for converting
	31	// TPC data stored in AliROOT format (outputfile from a simulation),
	32	// into the data format currently used by in the HLT framework.
	33	// This enables the possibility to always use the same data format,
	34	// whether you are using a binary file as an input, or a AliROOT file.
	35	//
	36	// For example on how to create binary files from a AliROOT simulation,
	37	// see example macro exa/Binary.C.
	38	//
	39	// For reading a AliROOT file into HLT format in memory, do the following:
	40	//
	41	// AliL3FileHandler file;
	42	// file.SetAliInput("galice.root");
	43	// AliL3DigitRowData dataPt = (AliL3DigitRowData)file.AliDigits2Memory(nrows,eventnr);
	44	//
	45	// All the data are then stored in memory and accessible via the pointer dataPt.
	46	// Accesing the data is then identical to the example 1) showed in AliL3MemHandler class.
	47	//
	48	// For converting the data back, and writing it to a new AliROOT file do:
	49	//
	50	// AliL3FileHandler file;
	51	// file.SetAliInput("galice.root");
	52	// file.Init(slice,patch,NumberOfRowsInPatch);
	53	// file.AliDigits2RootFile(dataPt,"new_galice.root");
	54	// file.CloseAliInput();
108615fc	55
	56	ClassImp(AliL3FileHandler)
	57
9183aa27	58	AliL3FileHandler::AliL3FileHandler()
9183aa27	59	{
108615fc	60	//Default constructor
	61	fInAli = 0;
	62	fParam = 0;
108615fc	63	fMC =0;
a6e4f9d6	64	fLastIndex=0;
	65	fDigits=0;
	66	fDigitsTree=0;
108615fc	67	}
108615fc	68
9183aa27	69	AliL3FileHandler::~AliL3FileHandler()
9183aa27	70	{
108615fc	71	//Destructor
108615fc	72	if(fMC) CloseMCOutput();
a6e4f9d6	73	if(fDigitsTree) delete fDigitsTree;
8f1a9904	74	if(fInAli) CloseAliInput();
a6e4f9d6	75
108615fc	76	}
108615fc	77
9183aa27	78	Bool_t AliL3FileHandler::SetMCOutput(char *name)
9183aa27	79	{
108615fc	80	fMC = fopen(name,"w");
	81	if(!fMC){
	82	LOG(AliL3Log::kWarning,"AliL3FileHandler::SetMCOutput","File Open")
9183aa27	83	<<"Pointer to File = 0x0 "<<ENDLOG;
108615fc	84	return kFALSE;
	85	}
	86	return kTRUE;
	87	}
	88
9183aa27	89	Bool_t AliL3FileHandler::SetMCOutput(FILE *file)
9183aa27	90	{
108615fc	91	fMC = file;
	92	if(!fMC){
	93	LOG(AliL3Log::kWarning,"AliL3FileHandler::SetMCOutput","File Open")
9183aa27	94	<<"Pointer to File = 0x0 "<<ENDLOG;
108615fc	95	return kFALSE;
	96	}
	97	return kTRUE;
	98	}
	99
9183aa27	100	void AliL3FileHandler::CloseMCOutput()
9183aa27	101	{
108615fc	102	if(!fMC){
108615fc	103	LOG(AliL3Log::kWarning,"AliL3FileHandler::CloseMCOutPut","File Close")
9183aa27	104	<<"Nothing to Close"<<ENDLOG;
108615fc	105	return;
	106	}
	107	fclose(fMC);
	108	fMC =0;
	109	}
	110
9183aa27	111	Bool_t AliL3FileHandler::SetAliInput()
9183aa27	112	{
108615fc	113	if(!fInAli->IsOpen()){
108615fc	114	LOG(AliL3Log::kError,"AliL3FileHandler::SetAliInput","File Open")
9183aa27	115	<<"Ali File "<<fInAli->GetName()<<" does not exist"<<ENDLOG;
108615fc	116	return kFALSE;
	117	}
	118	fParam = (AliTPCParam*)fInAli->Get("75x40_100x60");
	119	if(!fParam){
	120	LOG(AliL3Log::kError,"AliL3FileHandler::SetAliInput","File Open")
9183aa27	121	<<"No AliTPCParam 75x40_100x60 in File "<<fInAli->GetName()<<ENDLOG;
9183aa27	122	return kFALSE;
108615fc	123	}
108615fc	124	return kTRUE;
	125	}
	126
9183aa27	127	Bool_t AliL3FileHandler::SetAliInput(char *name)
	128	{
	129	//Open the AliROOT file with name.
	130
108615fc	131	fInAli= new TFile(name,"READ");
	132	if(!fInAli){
	133	LOG(AliL3Log::kWarning,"AliL3FileHandler::SetAliInput","File Open")
	134	<<"Pointer to TFile = 0x0 "<<ENDLOG;
	135	return kFALSE;
	136	}
	137	return SetAliInput();
	138	}
	139
9183aa27	140	Bool_t AliL3FileHandler::SetAliInput(TFile *file)
	141	{
	142	//Specify already opened AliROOT file to use as an input.
	143
108615fc	144	fInAli=file;
	145	if(!fInAli){
	146	LOG(AliL3Log::kWarning,"AliL3FileHandler::SetAliInput","File Open")
	147	<<"Pointer to TFile = 0x0 "<<ENDLOG;
	148	return kFALSE;
	149	}
	150	return SetAliInput();
	151	}
	152
9183aa27	153	void AliL3FileHandler::CloseAliInput()
9183aa27	154	{
108615fc	155	if(!fInAli){
108615fc	156	LOG(AliL3Log::kWarning,"AliL3FileHandler::CloseAliInput","File Close")
9183aa27	157	<<"Nothing to Close"<<ENDLOG;
9183aa27	158	return;
108615fc	159	}
	160	if(fInAli->IsOpen()) fInAli->Close();
	161	delete fInAli;
	162	fInAli = 0;
9183aa27	163
108615fc	164	}
108615fc	165
9183aa27	166	Bool_t AliL3FileHandler::IsDigit()
	167	{
	168	//Check if there is a TPC digit tree in the current file.
	169	//Return kTRUE if tree was found, and kFALSE if not found.
	170
108615fc	171	if(!fInAli){
	172	LOG(AliL3Log::kWarning,"AliL3FileHandler::IsDigit","File")
	173	<<"Pointer to TFile = 0x0 "<<ENDLOG;
	174	return kTRUE; //may you are use binary input which is Digits!!
	175	}
03b6adf7	176	TTree t=(TTree)fInAli->Get("TreeD_75x40_100x60_0");
108615fc	177	if(t){
	178	LOG(AliL3Log::kInformational,"AliL3FileHandler::IsDigit","File Type")
	179	<<"Found Digit Tree -> Use Fast Cluster Finder"<<ENDLOG;
	180	return kTRUE;
	181	}
	182	else{
	183	LOG(AliL3Log::kInformational,"AliL3FileHandler::IsDigit","File Type")
	184	<<"No Digit Tree -> Use Cluster Tree"<<ENDLOG;
	185	return kFALSE;
	186	}
	187	}
	188
	189	///////////////////////////////////////// Digit IO
9183aa27	190	Bool_t AliL3FileHandler::AliDigits2Binary(Int_t event)
	191	{
	192
108615fc	193	Bool_t out = kTRUE;
108615fc	194	UInt_t nrow;
31d9405a	195	AliL3DigitRowData* data = AliDigits2Memory(nrow,event);
108615fc	196	out = Memory2Binary(nrow,data);
	197	Free();
	198	return out;
	199	}
	200
	201
9183aa27	202	Bool_t AliL3FileHandler::AliDigits2CompBinary(Int_t event)
	203	{
	204	//Convert AliROOT TPC data, into HLT data format.
	205	//event specifies the event you want in the aliroot file.
	206
108615fc	207	Bool_t out = kTRUE;
108615fc	208	UInt_t ndigits=0;
31d9405a	209	AliL3DigitRowData* digits = AliDigits2Memory(ndigits,event);
108615fc	210	out = Memory2CompBinary(ndigits,digits);
	211	Free();
	212	return out;
	213	}
	214
	215
9183aa27	216	AliL3DigitRowData * AliL3FileHandler::AliDigits2Memory(UInt_t & nrow,Int_t event)
	217	{
	218	//Read data from AliROOT file into memory, and store it in the HLT data format.
	219	//Returns a pointer to the data.
	220
108615fc	221	AliL3DigitRowData *data = 0;
	222	nrow=0;
	223	if(!fInAli){
	224	LOG(AliL3Log::kWarning,"AliL3FileHandler::AliDigits2Memory","File")
	225	<<"No Input avalible: no object TFile"<<ENDLOG;
	226	return 0;
	227	}
	228	if(!fInAli->IsOpen()){
	229	LOG(AliL3Log::kWarning,"AliL3FileHandler::AliDigits2Memory","File")
	230	<<"No Input avalible: TFile not opend"<<ENDLOG;
	231	return 0;
	232	}
b25c64e5	233
a6e4f9d6	234	if(!fDigitsTree)
	235	GetDigitsTree(event);
	236
108615fc	237	UShort_t dig;
	238	Int_t time,pad,sector,row;
	239	Int_t nrows=0;
	240	Int_t ndigitcount=0;
a6e4f9d6	241	Int_t entries = (Int_t)fDigitsTree->GetEntries();
108615fc	242	Int_t ndigits[entries];
108615fc	243	Int_t lslice,lrow;
a6e4f9d6	244
a6e4f9d6	245	for(Int_t n=fLastIndex; n<fDigitsTree->GetEntries(); n++)
108615fc	246	{
a6e4f9d6	247	fDigitsTree->GetEvent(n);
a6e4f9d6	248	fParam->AdjustSectorRow(fDigits->GetID(),sector,row);
494fad94	249	AliL3Transform::Sector2Slice(lslice,lrow,sector,row);
a6e4f9d6	250	//if(fSlice != lslice \|\| lrow<fRowMin \|\| lrow>fRowMax) continue;
	251	if(lslice < fSlice) continue;
	252	if(lslice != fSlice) break;
	253	if(lrow < fRowMin) continue;
	254	if(lrow > fRowMax) break;
108615fc	255
	256	Float_t xyz[3];
	257	ndigits[lrow] = 0;
a6e4f9d6	258	fDigits->First();
108615fc	259	do {
a6e4f9d6	260	time=fDigits->CurrentRow();
	261	pad=fDigits->CurrentColumn();
	262	dig = fDigits->GetDigit(time,pad);
108615fc	263	if(dig<=fParam->GetZeroSup()) continue;
108615fc	264	if(time < fParam->GetMaxTBin()-1 && time > 0)
a6e4f9d6	265	if(fDigits->GetDigit(time+1,pad) <= fParam->GetZeroSup()
a6e4f9d6	266	&& fDigits->GetDigit(time-1,pad) <= fParam->GetZeroSup())
108615fc	267	continue;
108615fc	268
494fad94	269	AliL3Transform::Raw2Local(xyz,sector,row,pad,time);
108615fc	270	if(fParam->GetPadRowRadii(sector,row)<230./250.*fabs(xyz[2]))
	271	continue;
	272
	273	ndigits[lrow]++; //for this row only
	274	ndigitcount++; //total number of digits to be published
	275
a6e4f9d6	276	} while (fDigits->Next());
108615fc	277
	278	nrows++;
	279	}
	280	Int_t size = sizeof(AliL3DigitData)*ndigitcount
8f1a9904	281	+ nrows*sizeof(AliL3DigitRowData);
108615fc	282
108615fc	283	LOG(AliL3Log::kDebug,"AliL3FileHandler::AliDigits2Memory","Digits")
a6e4f9d6	284	<<AliL3Log::kDec<<"Found "<<ndigitcount<<" Digits"<<ENDLOG;
a6e4f9d6	285
108615fc	286	data=(AliL3DigitRowData*) Allocate(size);
	287	nrow = (UInt_t)nrows;
	288	AliL3DigitRowData *tempPt = data;
a6e4f9d6	289	for(Int_t n=fLastIndex; n<fDigitsTree->GetEntries(); n++)
108615fc	290	{
a6e4f9d6	291	fDigitsTree->GetEvent(n);
108615fc	292	Float_t xyz[3];
a6e4f9d6	293	fParam->AdjustSectorRow(fDigits->GetID(),sector,row);
494fad94	294	AliL3Transform::Sector2Slice(lslice,lrow,sector,row);
a6e4f9d6	295	//if(fSlice != lslice \|\| lrow<fRowMin \|\| lrow>fRowMax) continue;
	296	if(lslice < fSlice) continue;
	297	if(lslice != fSlice) break;
	298	if(lrow < fRowMin) continue;
	299	if(lrow > fRowMax) break;
	300
108615fc	301	tempPt->fRow = lrow;
	302	tempPt->fNDigit = ndigits[lrow];
	303
	304	Int_t localcount=0;
a6e4f9d6	305	fDigits->First();
108615fc	306	do {
a6e4f9d6	307	//dig=fDigits->CurrentDigit();
03b6adf7	308	//if (dig<=fParam->GetZeroSup()) continue;
a6e4f9d6	309	time=fDigits->CurrentRow();
	310	pad=fDigits->CurrentColumn();
	311	dig = fDigits->GetDigit(time,pad);
03b6adf7	312	if (dig <= fParam->GetZeroSup()) continue;
03b6adf7	313	if(time < fParam->GetMaxTBin()-1 && time > 0)
a6e4f9d6	314	if(fDigits->GetDigit(time-1,pad) <= fParam->GetZeroSup() &&
a6e4f9d6	315	fDigits->GetDigit(time+1,pad) <= fParam->GetZeroSup()) continue;
108615fc	316
108615fc	317	//Exclude data outside cone:
494fad94	318	AliL3Transform::Raw2Local(xyz,sector,row,pad,time);
108615fc	319	if(fParam->GetPadRowRadii(sector,row)<230./250.*fabs(xyz[2]))
	320	continue;
	321
	322	if(localcount >= ndigits[lrow])
	323	LOG(AliL3Log::kFatal,"AliL3FileHandler::AliDigits2Binary","Memory")
03b6adf7	324	<<AliL3Log::kDec<<"Mismatch: localcount "<<localcount<<" ndigits "
	325	<<ndigits[lrow]<<ENDLOG;
	326
108615fc	327	tempPt->fDigitData[localcount].fCharge=dig;
	328	tempPt->fDigitData[localcount].fPad=pad;
	329	tempPt->fDigitData[localcount].fTime=time;
10f815d9	330	#ifdef do_mc
	331	tempPt->fDigitData[localcount].fTrackID[0] = fDigits->GetTrackID(time,pad,0);
	332	tempPt->fDigitData[localcount].fTrackID[1] = fDigits->GetTrackID(time,pad,1);
	333	tempPt->fDigitData[localcount].fTrackID[2] = fDigits->GetTrackID(time,pad,2);
	334	#endif
108615fc	335	localcount++;
a6e4f9d6	336	} while (fDigits->Next());
108615fc	337
	338	Byte_t tmp = (Byte_t)tempPt;
	339	Int_t size = sizeof(AliL3DigitRowData)
	340	+ ndigits[lrow]*sizeof(AliL3DigitData);
	341	tmp += size;
	342	tempPt = (AliL3DigitRowData*)tmp;
a6e4f9d6	343	//fLastIndex=n;
108615fc	344	}
a6e4f9d6	345	//fLastIndex++;
108615fc	346	return data;
	347	}
	348
a6e4f9d6	349	Bool_t AliL3FileHandler::GetDigitsTree(Int_t event)
a6e4f9d6	350	{
9183aa27	351	//Connects to the TPC digit tree in the AliROOT file.
a6e4f9d6	352
	353	fInAli->cd();
	354	Char_t dname[100];
	355	sprintf(dname,"TreeD_75x40_100x60_%d",event);
	356	fDigitsTree = (TTree*)fInAli->Get("TreeD_75x40_100x60_0");
	357	if(!fDigitsTree)
	358	{
	359	LOG(AliL3Log::kError,"AliL3FileHandler::GetDigitsTree","Digits Tree")
	360	<<AliL3Log::kHex<<"Error getting digitstree "<<(Int_t)fDigitsTree<<ENDLOG;
	361	return kFALSE;
	362	}
	363	fDigitsTree->GetBranch("Segment")->SetAddress(&fDigits);
	364	return kTRUE;
	365	}
	366
0d319e67	367	void AliL3FileHandler::AliDigits2RootFile(AliL3DigitRowData rowPt,Char_t new_digitsfile)
0d319e67	368	{
9183aa27	369	//Write the data stored in rowPt, into a new AliROOT file.
	370	//The data is stored in the AliROOT format
	371	//This is specially a nice thing if you have modified data, and wants to run it
	372	//through the offline reconstruction chain.
	373	//The arguments is a pointer to the data, and the name of the new AliROOT file.
	374	//Remember to pass the original AliROOT file (the one that contains the original
	375	//simulated data) to this object, in order to retrieve the MC id's of the digits.
	376
0d319e67	377	if(!fInAli)
0d319e67	378	{
8f1a9904	379	printf("AliL3FileHandler::AliDigits2RootFile : No rootfile\n");
0d319e67	380	return;
	381	}
	382	if(!fParam)
	383	{
	384	printf("AliL3FileHandler::AliDigits2RootFile : No parameter object. Run on rootfile\n");
	385	return;
	386	}
0d319e67	387
	388	//Get the original digitstree:
	389	fInAli->cd();
	390	AliTPCDigitsArray *old_array = new AliTPCDigitsArray();
	391	old_array->Setup(fParam);
	392	old_array->SetClass("AliSimDigits");
03b6adf7	393	Bool_t ok = old_array->ConnectTree("TreeD_75x40_100x60_0");
0d319e67	394	if(!ok)
	395	{
	396	printf("AliL3FileHandler::AliDigits2RootFile : No digits tree object\n");
	397	return;
	398	}
0d319e67	399
03b6adf7	400	Bool_t create=kFALSE;
	401	TFile *digFile;
	402
8f1a9904	403	digFile = TFile::Open(new_digitsfile,"NEW");
8f1a9904	404	if(digFile->IsOpen())
03b6adf7	405	{
03b6adf7	406	create = kTRUE;
	407	fParam->Write(fParam->GetTitle());
	408	}
	409	else
	410	{
8f1a9904	411	LOG(AliL3Log::kDebug,"AliL3FileHandler::AliDigits2RootFile","Rootfile")
8f1a9904	412	<<"Rootfile did already exist, so I will just open it for updates"<<ENDLOG;
03b6adf7	413	digFile = TFile::Open(new_digitsfile,"UPDATE");
	414	create=kFALSE;
	415	}
	416	if(!digFile->IsOpen())
	417	{
8f1a9904	418	LOG(AliL3Log::kError,"AliL3FileHandler::AliDigits2RootFile","Rootfile")
8f1a9904	419	<<"Error opening rootfile "<<new_digitsfile<<ENDLOG;
03b6adf7	420	return;
	421	}
	422
	423	digFile->cd();
	424
	425	//setup a new one, or connect it to the existing one:
0d319e67	426	AliTPCDigitsArray *arr = new AliTPCDigitsArray;
	427	arr->SetClass("AliSimDigits");
	428	arr->Setup(fParam);
03b6adf7	429	if(create)
	430	arr->MakeTree();
	431	else
	432	{
	433	Bool_t ok = arr->ConnectTree("TreeD_75x40_100x60_0");
	434	if(!ok)
	435	{
	436	printf("AliL3FileHandler::AliDigits2RootFile : No digits tree object in existing file\n");
	437	return;
	438	}
	439	}
54efe57a	440	Int_t digcounter=0;
49898c33	441
0d319e67	442	for(Int_t i=fRowMin; i<=fRowMax; i++)
0d319e67	443	{
03b6adf7	444
49898c33	445	if((Int_t)rowPt->fRow != i) printf("AliL3FileHandler::AliDigits2RootFile : Mismatching row numbering!!!\n");
03b6adf7	446
0d319e67	447	Int_t sector,row;
494fad94	448	AliL3Transform::Slice2Sector(fSlice,i,sector,row);
49898c33	449	AliSimDigits * dig = (AliSimDigits*)arr->CreateRow(sector,row);
49898c33	450	AliSimDigits old_dig = (AliSimDigits)old_array->LoadRow(sector,row);
0d319e67	451	if(!old_dig)
	452	printf("AliL3FileHandler::AliDigits2RootFile : No padrow %d %d\n",sector,row);
	453
	454	AliL3DigitData *digPt = rowPt->fDigitData;
54efe57a	455	digcounter=0;
0d319e67	456	for(UInt_t j=0; j<rowPt->fNDigit; j++)
0d319e67	457	{
49898c33	458	Int_t charge = (Int_t)digPt[j].fCharge;
	459	Int_t pad = (Int_t)digPt[j].fPad;
	460	Int_t time = (Int_t)digPt[j].fTime;
03b6adf7	461
	462	if(charge == 0) //Only write the digits that has not been removed
	463	continue;
54efe57a	464	digcounter++;
54efe57a	465	dig->SetDigitFast(charge,time,pad);
49898c33	466
	467	Int_t trackID[3] = {old_dig->GetTrackID(time,pad,0),old_dig->GetTrackID(time,pad,1),old_dig->GetTrackID(time,pad,2)};
	468	Int_t s_pad = pad;
	469	Int_t s_time = time - 1;
	470	while(trackID[0] < 0)
	471	{
494fad94	472	if(s_time >= 0 && s_time < AliL3Transform::GetNTimeBins() && s_pad >= 0 && s_pad < AliL3Transform::GetNPads(i))
49898c33	473	{
	474	if(old_dig->GetTrackID(s_time,s_pad,0) > 0)
	475	{
	476	trackID[0]=old_dig->GetTrackID(s_time,s_pad,0);
	477	trackID[1]=old_dig->GetTrackID(s_time,s_pad,1);
	478	trackID[2]=old_dig->GetTrackID(s_time,s_pad,2);
	479	}
	480	}
	481	if(s_pad == pad && s_time == time - 1)
	482	s_time = time + 1;
	483	else if(s_pad == pad && s_time == time + 1)
	484	{s_pad = pad - 1; s_time = time;}
	485	else if(s_pad == pad - 1 && s_time == time)
	486	s_time = time - 1;
	487	else if(s_pad == pad - 1 && s_time == time - 1)
	488	s_time = time + 1;
	489	else if(s_pad == pad - 1 && s_time == time + 1)
	490	{s_pad = pad + 1; s_time = time;}
	491	else if(s_pad == pad + 1 && s_time == time)
	492	s_time = time - 1;
	493	else if(s_pad == pad + 1 && s_time == time - 1)
	494	s_time = time + 1;
	495	else
	496	break;
	497	}
	498
	499	dig->SetTrackIDFast(trackID[0],time,pad,0);
	500	dig->SetTrackIDFast(trackID[1],time,pad,1);
	501	dig->SetTrackIDFast(trackID[2],time,pad,2);
0d319e67	502
0d319e67	503	}
54efe57a	504	//cout<<"Wrote "<<digcounter<<" on row "<<i<<endl;
0d319e67	505	UpdateRowPointer(rowPt);
	506	arr->StoreRow(sector,row);
	507	arr->ClearRow(sector,row);
	508	old_array->ClearRow(sector,row);
	509	}
	510	digFile->cd();
	511	char treeName[100];
03b6adf7	512	sprintf(treeName,"TreeD_%s_0",fParam->GetTitle());
8f1a9904	513	printf("Writing tree to file.....");
0d319e67	514	arr->GetTree()->Write(treeName,TObject::kOverwrite);
8f1a9904	515	printf("done\n");
0d319e67	516	digFile->Close();
03b6adf7	517	//arr->GetTree()->Delete();
03b6adf7	518	//delete arr;
0d319e67	519	}
0d319e67	520
108615fc	521	///////////////////////////////////////// Point IO
	522	Bool_t AliL3FileHandler::AliPoints2Binary(){
	523	Bool_t out = kTRUE;
	524	UInt_t npoint;
	525	AliL3SpacePointData *data = AliPoints2Memory(npoint);
	526	out = Memory2Binary(npoint,data);
	527	Free();
	528	return out;
	529	}
	530
	531	AliL3SpacePointData * AliL3FileHandler::AliPoints2Memory(UInt_t & npoint){
	532	AliL3SpacePointData *data = 0;
	533	npoint=0;
	534	if(!fInAli){
	535	LOG(AliL3Log::kWarning,"AliL3FileHandler::AliPoints2Memory","File")
	536	<<"No Input avalible: no object TFile"<<ENDLOG;
	537	return 0;
	538	}
	539	if(!fInAli->IsOpen()){
	540	LOG(AliL3Log::kWarning,"AliL3FileHandler::AliPoints2Memory","File")
	541	<<"No Input avalible: TFile not opend"<<ENDLOG;
	542	return 0;
	543	}
494fad94	544
108615fc	545	TDirectory *savedir = gDirectory;
108615fc	546	fInAli->cd();
8f1a9904	547
	548	Char_t cname[100];
	549	Int_t eventn = 0;
	550	sprintf(cname,"TreeC_TPC_%d",eventn);
108615fc	551	AliTPCClustersArray carray;
	552	carray.Setup(fParam);
	553	carray.SetClusterType("AliTPCcluster");
8f1a9904	554	Bool_t clusterok = carray.ConnectTree(cname);
108615fc	555	if(!clusterok) return 0;
	556
	557	AliTPCClustersRow ** clusterrow =
	558	new AliTPCClustersRow*[ (int)carray.GetTree()->GetEntries()];
	559	Int_t *rows = new int[ (int)carray.GetTree()->GetEntries()];
	560	Int_t *sects = new int[ (int)carray.GetTree()->GetEntries()];
	561	Int_t sum=0;
	562
	563	Int_t lslice,lrow;
	564	for(Int_t i=0; i<carray.GetTree()->GetEntries(); i++){
	565	AliSegmentID *s = carray.LoadEntry(i);
	566	Int_t sector,row;
	567	fParam->AdjustSectorRow(s->GetID(),sector,row);
	568	rows[i] = row;
	569	sects[i] = sector;
	570	clusterrow[i] = 0;
494fad94	571	AliL3Transform::Sector2Slice(lslice,lrow,sector,row);
108615fc	572	if(fSlice != lslice \|\| lrow<fRowMin \|\| lrow>fRowMax) continue;
	573	clusterrow[i] = carray.GetRow(sector,row);
	574	if(clusterrow[i])
	575	sum+=clusterrow[i]->GetArray()->GetEntriesFast();
	576	}
	577	UInt_t size = sum*sizeof(AliL3SpacePointData);
	578
	579	LOG(AliL3Log::kDebug,"AliL3FileHandler::AliPoints2Memory","File")
	580	<<AliL3Log::kDec<<"Found "<<sum<<" SpacePoints"<<ENDLOG;
	581
	582	data = (AliL3SpacePointData *) Allocate(size);
	583	npoint = sum;
	584	UInt_t n=0;
	585	for(Int_t i=0; i<carray.GetTree()->GetEntries(); i++){
	586	if(!clusterrow[i]) continue;
	587	Int_t row = rows[i];
	588	Int_t sector = sects[i];
494fad94	589	AliL3Transform::Sector2Slice(lslice,lrow,sector,row);
108615fc	590	Int_t entries_in_row = clusterrow[i]->GetArray()->GetEntriesFast();
	591	for(Int_t j = 0;j<entries_in_row;j++){
	592	AliTPCcluster c = (AliTPCcluster)(*clusterrow[i])[j];
	593	data[n].fZ = c->GetZ();
	594	data[n].fY = c->GetY();
	595	data[n].fX = fParam->GetPadRowRadii(sector,row);
	596	data[n].fID = n+((fSlice&0x7f)<<25)+((fPatch&0x7)<<22);//uli
	597	data[n].fPadRow = lrow;
	598	data[n].fXYErr = c->GetSigmaY2();
	599	data[n].fZErr = c->GetSigmaZ2();
	600	if(fMC) fprintf(fMC,"%d %d\n",data[n].fID,c->GetLabel(0));
	601	n++;
	602	}
	603	}
	604	for(Int_t i=0;i<carray.GetTree()->GetEntries();i++){
	605	Int_t row = rows[i];
	606	Int_t sector = sects[i];
	607	if(carray.GetRow(sector,row))
	608	carray.ClearRow(sector,row);
	609	}
	610
	611	delete [] clusterrow;
	612	delete [] rows;
	613	delete [] sects;
	614	savedir->cd();
	615
	616	return data;
	617	}
	618