Compiler Warning removed

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

// @(#) $Id$

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

#include "AliL3StandardIncludes.h"

#ifndef no_root
#include <TFile.h>
#include <TDirectory.h>
#include <TClonesArray.h>
#include <TStopwatch.h>
#endif

#include "AliL3Logging.h"
#include "AliLevel3.h"
#include "AliL3ConfMapper.h"
#include "AliL3Vertex.h"
#include "AliL3VertexFinder.h"
#include "AliL3TrackMerger.h"
#include "AliL3GlobalMerger.h"
#include "AliL3InterMerger.h"
#include "AliL3ConfMapPoint.h"
#include "AliL3ConfMapTrack.h"
#include "AliL3Transform.h"
#include "AliL3ClustFinderNew.h"
#include "AliL3DigitData.h"
#include "AliL3TrackArray.h"
#include "AliL3MemHandler.h"
#include "AliL3Fitter.h"
#ifdef use_aliroot
#include "AliL3FileHandler.h"
#endif
#include "AliL3Benchmark.h"
#include "AliL3DigitData.h"
#include "AliL3TrackSegmentData.h"
#include "AliL3SpacePointData.h"
#include "AliL3VertexData.h"
#include "AliL3DDLDataFileHandler.h"

/** \class AliLevel3
<pre>
//_____________________________________________________________
//
//  AliLevel3
//
//  Interface class for Level3 tracker.
//  For example how to use, see exa/runtracker.C (root)
//  or programs/runtracker.cxx (standalone program).
//Begin_Html 
//<img src="tpcsectorsnb.gif">
//End_Html
</pre>
*/

ClassImp(AliLevel3)

Bool_t AliLevel3::fSetTracks2FirstPoint = kTRUE;//Define track parameters at first point

AliLevel3::AliLevel3()
{
  //Default constructor. Should also be used when input is from binary files.
  //In that case the path to where the binary files are located has to be
  //passed to the AliLevel::Init function.
  
  fInputFile=0;
}

AliLevel3::AliLevel3(Char_t *infile)
{
  //Constructor to use for when input is anything else but binary files,
  //meaning rootfiles or raw files.
  
  fInputFile = infile;

}

void AliLevel3::Init(Char_t *path,EFileType filetype,Int_t npatches)
{
  if((filetype!=kBinary) && (filetype!=kDate) && !fInputFile)
    {
      LOG(AliL3Log::kError,"AliLevel3::Init","Files")
	<<"You have not supplied the input rootfile; use the appropriate ctor!"<<ENDLOG;
      return;
    }
  
  fWriteOut = kFALSE;
  fPileUp = kFALSE;
  fNoCF=kFALSE;
  fUseBinary = (filetype==kBinary);
  SetPath(path);
  
  fDoRoi = kFALSE;
  fDoNonVertex = kFALSE;
  fFindVertex = kFALSE;
  SetClusterFinderParam();

  fEta[0] = 0.;
  fEta[1] = 1.1;

  fEvent=0;

  switch(npatches){
  case 0:
    fNPatch = 1;
    fRow[0][0] = AliL3Transform::GetFirstRow(3);
    fRow[0][1] = AliL3Transform::GetLastRow(5);
    break;
  case 1:
    fNPatch = 1;        //number of patches change row in process
    fRow[0][0] = 0;
    fRow[0][1] = AliL3Transform::GetLastRow(-1);
    break;
  case 2:
    fNPatch = 2;        //number of patches change row in process
    fRow[0][0] = 0;     // first row
    fRow[0][1] = AliL3Transform::GetLastRow(1);
    fRow[1][0] = AliL3Transform::GetFirstRow(2);
    fRow[1][1] = AliL3Transform::GetLastRow(5);
    break;
  default: 
    fNPatch = 6;        
    fRow[0][0] = AliL3Transform::GetFirstRow(0);
    fRow[0][1] = AliL3Transform::GetLastRow(0);
    fRow[1][0] = AliL3Transform::GetFirstRow(1);
    fRow[1][1] = AliL3Transform::GetLastRow(1);
    fRow[2][0] = AliL3Transform::GetFirstRow(2);
    fRow[2][1] = AliL3Transform::GetLastRow(2);
    fRow[3][0] = AliL3Transform::GetFirstRow(3);
    fRow[3][1] = AliL3Transform::GetLastRow(3);
    fRow[4][0] = AliL3Transform::GetFirstRow(4);
    fRow[4][1] = AliL3Transform::GetLastRow(4);
    fRow[5][0] = AliL3Transform::GetFirstRow(5);
    fRow[5][1] = AliL3Transform::GetLastRow(5);
  }

  fVertexFinder = new AliL3VertexFinder();
  fVertex = new AliL3Vertex();
  fTracker = new AliL3ConfMapper();
  fTrackMerger = new AliL3TrackMerger(fNPatch);
  fInterMerger = new AliL3InterMerger();
  fGlobalMerger = new AliL3GlobalMerger();
  SetMergerParameters();//Set default merger parameters
#ifdef use_aliroot
  if(filetype==kRoot){
    fFileHandler = new AliL3FileHandler();
    fFileHandler->SetAliInput(fInputFile);
  }else if(filetype==kRaw){
    fFileHandler = new AliL3DDLDataFileHandler();
    fFileHandler->SetReaderInput(fInputFile);
  }else if(filetype==kDate){
    fFileHandler = new AliL3DDLDataFileHandler();
    fFileHandler->SetReaderInput(fInputFile,-1);
  }else{
    fFileHandler = new AliL3MemHandler();
  }
#else
  if(filetype==kRaw){
    fFileHandler = new AliL3DDLDataFileHandler();
    fFileHandler->SetReaderInput(fInputFile);
  }else{
    fFileHandler = new AliL3MemHandler();
  }
#endif
  fBenchmark = new AliL3Benchmark();
}

void AliLevel3::DoBench(char* name){
  fBenchmark->Analyze(name);
  delete fBenchmark;
  fBenchmark = new AliL3Benchmark();
}

void AliLevel3::DoMc(char* file){
#ifdef use_aliroot
  if(!fFileHandler->IsDigit(fEvent))
    fFileHandler->SetMCOutput(file);
#endif
}

AliLevel3::~AliLevel3(){
  //Destructor
  if(fVertexFinder) delete fVertexFinder;
  if(fVertex) delete fVertex;
  if(fTracker) delete fTracker;
  if(fTrackMerger) delete fTrackMerger;
  if(fInterMerger) delete fInterMerger;
  if(fFileHandler) delete fFileHandler;
  if(fGlobalMerger) delete fGlobalMerger;
}

void AliLevel3::SetClusterFinderParam(Float_t fXYError, Float_t fZError, Bool_t deconv)
{
  fXYClusterError=fXYError;
  fZClusterError=fZError;
  fClusterDeconv=deconv;
}

void AliLevel3::SetTrackerParam(Int_t phi_segments, Int_t eta_segments,
				Int_t trackletlength, Int_t tracklength,
				Int_t rowscopetracklet, Int_t rowscopetrack,
				Double_t min_pt_fit, Double_t maxangle,
				Double_t goodDist, Double_t hitChi2Cut,
				Double_t goodHitChi2, Double_t trackChi2Cut,
				Int_t maxdist,Double_t maxphi,Double_t maxeta,Bool_t vertexconstraint)
{
  //Set parameters input to the tracker
  //If no arguments are given, default parameters will be used
  
  fTracker->SetNSegments(phi_segments,eta_segments);
  fTracker->SetMaxDca(min_pt_fit);
  fTracker->SetTrackCuts(hitChi2Cut,goodHitChi2,trackChi2Cut,maxdist,vertexconstraint);
  fTracker->SetTrackletCuts(maxangle,goodDist,vertexconstraint);
  if(vertexconstraint)
    fTracker->MainVertexSettings(trackletlength,tracklength,rowscopetracklet,rowscopetrack,maxphi,maxeta);
  else
    fTracker->NonVertexSettings(trackletlength,tracklength,rowscopetracklet,rowscopetrack);
  fTracker->InitVolumes();
}

void AliLevel3::SetMergerParameters(Double_t maxy,Double_t maxz,Double_t maxkappa,Double_t maxpsi,Double_t maxtgl)
{
  fGlobalMerger->SetParameter(maxy,maxz,maxkappa,maxpsi,maxtgl);
}

void AliLevel3::ProcessEvent(Int_t first,Int_t last,Int_t event){
  //Do tracking on all slices in region [first,last]
  //Slices numbering in TPC goes from 0-35, which means that one slice
  //corresponds to inner+outer sector.E.g. slice 2 corresponds to
  //inner=2 + outer=38.

  fGlobalMerger->Setup(first,last);
  fEvent=event;
  for(Int_t i=first; i<=last; i++){
    ProcessSlice(i);
    fGlobalMerger->SetVertex(fVertex);
    fGlobalMerger->InitSlice(i);
    fGlobalMerger->FillTracks(fNTrackData,fTrackData);
    fFileHandler->Free();   //free the memory
    fNTrackData=0;
    fTrackData=0;
  }
  fBenchmark->Start("Global track merger");
  //fGlobalMerger->AddAllTracks();
  fGlobalMerger->Merge();
  //fGlobalMerger->SlowMerge(fWriteOutPath);
  fBenchmark->Stop("Global track merger");
  
  FitGlobalTracks();
  
  if(fWriteOut) WriteResults(); 
  fFileHandler->FreeDigitsTree();
}

void AliLevel3::ProcessSlice(Int_t slice){
  char name[256];
  Bool_t UseCF = kFALSE;
#ifdef use_aliroot
  UseCF = fFileHandler->IsDigit(fEvent);
#endif
  if(fUseBinary)
    UseCF = kTRUE;   //In case you are not using aliroot
  if(fNoCF == kTRUE) //In case you don't want to run with cluster finder
    UseCF = kFALSE;

  const Int_t maxpoints=120000;
  const Int_t pointsize = maxpoints * sizeof(AliL3SpacePointData);
  AliL3MemHandler *memory = new AliL3MemHandler();

  fTrackMerger->Reset();
  fTrackMerger->SetRows(fRow[0]);
  
  for(Int_t patch=fNPatch-1;patch>=0;patch--){
    fFileHandler->Init(slice,patch,&fRow[patch][0]);
    UInt_t npoints=0;
    AliL3SpacePointData *points =0;
    UInt_t ndigits=0;
    AliL3DigitRowData *digits =0;
    if(UseCF){
      if(fUseBinary){
        if(!fDoRoi){ 
          if(1){     //Binary to Memory
	    fFileHandler->Free();
            if(fNPatch == 1)
	      sprintf(name,"%sdigits_%d_%d_%d.raw",fPath,fEvent,slice,-1);
	    else
	      sprintf(name,"%sdigits_%d_%d_%d.raw",fPath,fEvent,slice,patch);
	    if(!fFileHandler->SetBinaryInput(name)) return;
	    if(fPileUp)
	      { //Read binary files which are not RLE
		digits = (AliL3DigitRowData*)fFileHandler->Allocate();
		fFileHandler->Binary2Memory(ndigits,digits); 
	      }
	    else //Read RLE binary files
	      digits= (AliL3DigitRowData *)fFileHandler->CompBinary2Memory(ndigits);

	    fFileHandler->CloseBinaryInput(); 
          }

          if(0){     //Binary to Memory with Benchmark 
	    fFileHandler->Free();
            if(fNPatch == 1)
	      sprintf(name,"%sdigits_%d_%d_%d.raw",fPath,fEvent,slice,-1);
	    else
	      sprintf(name,"%sdigits_%d_%d_%d.raw",fPath,fEvent,slice,patch);
            if(!memory->SetBinaryInput(name)) return;
            UInt_t compsize=memory->GetFileSize();
            UInt_t *comp=(UInt_t *)memory->Allocate(compsize);
            memory->CompBinary2CompMemory(ndigits,comp);
            memory->CloseBinaryInput();
            UInt_t datasize=memory->GetMemorySize(ndigits,comp);
            digits=(AliL3DigitRowData *)fFileHandler->Allocate(datasize);
            fBenchmark->Start("Unpacker"); 
            fFileHandler->CompMemory2Memory(ndigits,digits,comp); 
            fBenchmark->Stop("Unpacker");
            memory->Free();
          }
  
          if(0){     //Binary to Memory with Random
            fFileHandler->Free();
            fFileHandler->ResetRandom();
            fFileHandler->SetRandomCluster(100);
            fFileHandler->SetNGenerate(100);
            if(fNPatch == 1)
	      sprintf(name,"%sdigits_%d_%d_%d.raw",fPath,fEvent,slice,-1);
	    else
	      sprintf(name,"%sdigits_%d_%d_%d.raw",fPath,fEvent,slice,patch);
            if(!memory->SetBinaryInput(name)) return;
            UInt_t compsize=memory->GetFileSize();
            UInt_t *comp=(UInt_t *)memory->Allocate(compsize);
            memory->CompBinary2CompMemory(ndigits,comp);
            memory->CloseBinaryInput();
            UInt_t dsize=memory->GetMemorySize(ndigits,comp);
            UInt_t rsize=fFileHandler->GetRandomSize();       
            digits=(AliL3DigitRowData*)fFileHandler->Allocate(dsize+rsize);
            fBenchmark->Start("Unpacker");
            fFileHandler->CompMemory2Memory(ndigits,digits,comp); 
            fBenchmark->Stop("Unpacker");
            memory->Free();
          }
        }

        else{     //Binary to Memory with Roi
          fFileHandler->Free();
          Int_t sli[2]={0,0};
          fFileHandler->SetROI(fEta,sli);
          if(fNPatch==1)
	    sprintf(name,"%sdigits_%d_%d_%d.raw",fPath,fEvent,slice,-1);
	  else
	    sprintf(name,"%sdigits_%d_%d_%d.raw",fPath,fEvent,slice,patch);
          if(!memory->SetBinaryInput(name)) return;
          UInt_t compsize=memory->GetFileSize();
          UInt_t *comp=(UInt_t *)memory->Allocate(compsize);
          memory->CompBinary2CompMemory(ndigits,comp);
          memory->CloseBinaryInput();
          UInt_t datasize=memory->GetMemorySize(ndigits,comp);
          digits=(AliL3DigitRowData *)fFileHandler->Allocate(datasize);
          fBenchmark->Start("Unpacker"); 
          datasize = fFileHandler->CompMemory2Memory(ndigits,digits,comp); 
          fBenchmark->Stop("Unpacker"); 
          memory->Free();
        }
      }//end UseBinary
      else{
#ifdef use_aliroot
        fBenchmark->Start("Dummy Unpacker");
        if(fNPatch==1)
	  sprintf(name,"digits_%d_%d_%d.raw",fEvent,slice,-1);
	else
	  sprintf(name,"digits_%d_%d_%d.raw",fEvent,slice,patch);
        fBenchmark->Stop("Dummy Unpacker");

        if(0){   //Ali to Binary
          fFileHandler->SetBinaryOutput(name);
          fFileHandler->AliDigits2CompBinary();
          fFileHandler->CloseBinaryOutput();
        }
  
        if(1){   //Ali to Memory
          digits=(AliL3DigitRowData *)fFileHandler->AliAltroDigits2Memory(ndigits,fEvent);
          if(0){ //Memory to Binary
            fFileHandler->SetBinaryOutput(name);
            fFileHandler->Memory2CompBinary(ndigits,digits);
            fFileHandler->CloseBinaryOutput();
          }
        }
#endif
      }//end else UseBinary

      points = (AliL3SpacePointData *) memory->Allocate(pointsize);
      fClusterFinder = new AliL3ClustFinderNew();
      fClusterFinder->InitSlice(slice,patch,fRow[patch][0],fRow[patch][1],maxpoints);
      fClusterFinder->SetDeconv(fClusterDeconv);
      fClusterFinder->SetXYError(fXYClusterError);
      fClusterFinder->SetZError(fZClusterError);
      if((fXYClusterError>0)&&(fZClusterError>0))
	fClusterFinder->SetCalcErr(kFALSE);
      fClusterFinder->SetOutputArray(points);
      fBenchmark->Start("Cluster finder");
      fClusterFinder->Read(ndigits,digits);
      fClusterFinder->ProcessDigits();
      fBenchmark->Stop("Cluster finder");
      npoints = fClusterFinder->GetNumberOfClusters();
      delete fClusterFinder;
      fClusterFinder = 0;
      fFileHandler->Free();
      LOG(AliL3Log::kInformational,"AliLevel3::ProcessSlice","Cluster Finder")
        <<AliL3Log::kDec<<"Found "<<npoints<<" Points"<<ENDLOG;
    }//end UseCF
    else{// if not use Clusterfinder
      if(fUseBinary){//Binary to Memory
        memory->Free();
        if(fNPatch==1)
	  sprintf(name,"%s/points_%d_%d_%d.raw",fPath,fEvent,slice,-1);
	else
	  sprintf(name,"%s/points_%d_%d_%d.raw",fPath,fEvent,slice,patch);
        if(!memory->SetBinaryInput(name)) return;
        points = (AliL3SpacePointData *) memory->Allocate();
        memory->Binary2Memory(npoints,points);
        memory->CloseBinaryInput();
        LOG(AliL3Log::kInformational,"AliLevel3::ProcessSlice","Read Cluster")
        <<AliL3Log::kDec<<"Found "<<npoints<<" Points in File"<<ENDLOG;
      }
#ifdef use_aliroot
      else{
        points = fFileHandler->AliPoints2Memory(npoints);
      }
#endif
      fBenchmark->Start("Dummy Unpacker");
      fBenchmark->Stop("Dummy Unpacker");
      fBenchmark->Start("Dummy CF");
      fBenchmark->Stop("Dummy CF");
    }
   
    if(patch == fNPatch-1){
      // Vertex
      if(fFindVertex){
      // Vertex Finder
      
        fBenchmark->Start("Vertex Finder Read");
        fVertexFinder->Reset();
        fVertexFinder->Read(npoints,points);
        fBenchmark->Stop("Vertex Finder Read"); 
        fBenchmark->Start("Vertex Finder");
        fVertexFinder->Analyze();
        AliL3VertexData vertex[1];
        fVertexFinder->Write(vertex);
        fVertex->Read(vertex);
        fBenchmark->Stop("Vertex Finder"); 
      }
      else{
        //use 0,0,0 for vertex
        fVertex->SetZero();
      }
      fTrackMerger->SetVertex(fVertex);
    }

    fTracker->InitSector(slice,fRow[patch],fEta);
    fTracker->SetVertex(fVertex);
    fBenchmark->Start("Tracker setup"); 
    fTracker->ReadHits(npoints,points);
    fTracker->MainVertexTracking_a();
    fBenchmark->Stop("Tracker setup");
    fBenchmark->Start("Track follower");
    fTracker->MainVertexTracking_b();
    fBenchmark->Stop("Track follower");
    if(fDoNonVertex)
      fTracker->NonVertexTracking();//Do a second pass for nonvertex tracks
    fBenchmark->Start("Sector track fitting");
    fTracker->FillTracks();
    fBenchmark->Stop("Sector track fitting");

    if(fWriteOut) 
       WriteSpacePoints(npoints, points, slice, patch); //do after Tracking

    //free memory
    if(UseCF)
      memory->Free();
    else
      fFileHandler->Free();

    UInt_t ntracks0 =0;
    AliL3TrackSegmentData *trackdata0  = 
         (AliL3TrackSegmentData *) memory->Allocate(fTracker->GetTracks());
    memory->TrackArray2Memory(ntracks0,trackdata0,fTracker->GetTracks());

    //write tracks
    if(fWriteOut){
      if(fNPatch==1)
	sprintf(name,"%s/tracks_tr_%d_%d_%d.raw",fWriteOutPath,fEvent,slice,-1);
      else
	sprintf(name,"%s/tracks_tr_%d_%d_%d.raw",fWriteOutPath,fEvent,slice,patch);
      memory->SetBinaryOutput(name);
      memory->Memory2Binary(ntracks0,trackdata0);
      memory->CloseBinaryOutput();
    }

    fInterMerger->Reset();
    fInterMerger->Init(fRow[patch],patch);

    fInterMerger->FillTracks(ntracks0,trackdata0);
    
    //fBenchmark->Start("Inter Merger");
    // fInterMerger->Merge();
    //    fInterMerger->SlowMerge();
    
    //fBenchmark->Stop("Inter Merger");
    /*
    //write inter merged tracks
    if(fWriteOut){
      sprintf(name,"%stracks_im_%d_%d.raw",fWriteOutPath,slice,patch);
      WriteTracks(name,fInterMerger,'i'); //write output of intermerger
      }
    */
    memory->Free();
    
    UInt_t ntracks1 =0;
    AliL3TrackSegmentData *trackdata1 =
      (AliL3TrackSegmentData *) memory->Allocate(fInterMerger->GetInTracks(0));
    memory->TrackArray2Memory(ntracks1,trackdata1,fInterMerger->GetInTracks(0));

    fTrackMerger->InitSector(slice,patch);
    fTrackMerger->FillTracks(ntracks1,trackdata1);

    memory->Free();
  }
  //fBenchmark->Start("Patch Merger");
  //fTrackMerger->SlowMerge();
  fTrackMerger->AddAllTracks();
  //fTrackMerger->Merge();
  //fBenchmark->Stop("Patch Merger");
  /*
  //write merged tracks
  if(fWriteOut){
    sprintf(name,"%stracks_tm_%d.raw",fWriteOutPath,slice);
    WriteTracks(name,fTrackMerger,'o'); //write output of trackmerger
  }
  */
  fTrackData = (AliL3TrackSegmentData *) 
                         fFileHandler->Allocate(fTrackMerger->GetOutTracks());

  fFileHandler->TrackArray2Memory(fNTrackData,fTrackData,
                                                fTrackMerger->GetOutTracks());

  delete memory;
}

void AliLevel3::FitGlobalTracks()
{
  AliL3Fitter *fitter = new AliL3Fitter(fVertex);
  if(fNPatch==1)
    fitter->LoadClusters(fWriteOutPath,fEvent,kTRUE);
  else
    fitter->LoadClusters(fWriteOutPath,fEvent,kFALSE);
  
  fBenchmark->Start("Global track fitter");
  AliL3TrackArray *tracks = fGlobalMerger->GetOutTracks();
  for(Int_t i=0; i<tracks->GetNTracks(); i++)
    {
      AliL3Track *tr = tracks->GetCheckedTrack(i);
      if(!tr) continue;
      fitter->FitHelix(tr);
      fitter->UpdateTrack(tr);
    }
  fBenchmark->Stop("Global track fitter");
  delete fitter;
}

void AliLevel3::WriteSpacePoints(UInt_t npoints,AliL3SpacePointData *points,
				 Int_t slice,Int_t patch)
{
  char name[256];
  if(fNPatch==1)
    sprintf(name,"%s/points_%d_%d_%d.raw",fWriteOutPath,fEvent,slice,-1);
  else
    sprintf(name,"%s/points_%d_%d_%d.raw",fWriteOutPath,fEvent,slice,patch);
  AliL3MemHandler * memory = new AliL3MemHandler();
  memory->SetBinaryOutput(name);
  memory->Transform(npoints,points,slice);
  memory->Memory2Binary(npoints,points);
  memory->CloseBinaryOutput();
  delete  memory;
}

Int_t AliLevel3::WriteTracks(char *filename,AliL3Merger *merger,char opt)
{
  AliL3MemHandler *memory = new AliL3MemHandler();
  memory->SetBinaryOutput(filename);
  if(opt=='a'||opt=='i'){  //add intracks
    for(Int_t i=0;i<merger->GetNIn();i++){
      AliL3TrackArray *tr=merger->GetInTracks(i);
      memory->TrackArray2Binary(tr);
    }
  }

  if(opt=='o'||opt=='a'){
    AliL3TrackArray *tr=merger->GetOutTracks();
    memory->TrackArray2Binary(tr);
  }

  memory->CloseBinaryOutput();
 
  return 1;
}

void AliLevel3::WriteResults()
{
  //Write the resulting tracks to outputfile
  Char_t fname[256];
  sprintf(fname,"%s/tracks_%d.raw",fWriteOutPath,fEvent);
  WriteTracks(fname,fGlobalMerger,'a');
  //WriteTracks("tracks.raw",fGlobalMerger,'a');
  sprintf(fname,"%s/tracks_gl_%d.raw",fWriteOutPath,fEvent);
  WriteTracks(fname,fGlobalMerger,'o');
  //WriteTracks("tracks_gl.raw",fGlobalMerger,'o');
}