updates in cluster error parametrization + documentation

[u/mrichter/AliRoot.git] / TRD / qaRec / AliTRDalignmentTask.cxx

/**************************************************************************
* Copyright(c) 1998-1999, 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-commercialf 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 "TTreeStream.h"
#include "TROOT.h"
#include "TH1.h"
#include "TH2.h"

#include "AliTrackPointArray.h"
#include "AliLog.h"

#include "AliTRDgeometry.h"
#include "AliTRDcluster.h"
#include "AliTRDseedV1.h"
#include "AliTRDtrackV1.h"

#include "AliTRDalignmentTask.h"

ClassImp(AliTRDalignmentTask)

//________________________________________________________
AliTRDalignmentTask::AliTRDalignmentTask()
  :AliTRDrecoTask("Alignment", "TRD alignment")
  ,fTree(0x0)
  ,fArray(0x0)
{
  InitFunctorList();
  DefineOutput(1, TTree::Class());

}

//________________________________________________________
AliTRDalignmentTask::~AliTRDalignmentTask()
{
  if (fArray) delete fArray;
}


//________________________________________________________
void AliTRDalignmentTask::CreateOutputObjects()
{
  // spatial resolution
  OpenFile(1, "RECREATE");

  fTree = new TTree("spTree", "Tree with track space point arrays");
  fTree->Branch("SP","AliTrackPointArray", &fArray);
}


//________________________________________________________
void AliTRDalignmentTask::Exec(Option_t *opt)
{
  AliTRDrecoTask::Exec(opt);
  PostData(1, fTree);
}


//________________________________________________________
TH1* AliTRDalignmentTask::PlotTrackPoints(const AliTRDtrackV1 *track)
{
  if(track) fTrack = track;
  if(!fTrack){
    AliWarning("No Track defined.");
    return 0x0;
  }

  if (fArray) delete fArray;
  fArray = new AliTrackPointArray(fTrack->GetNumberOfTracklets());

  // Filling the track points array
  Float_t x, y, z;
  AliTrackPoint p; Int_t ip = 0;
  AliTRDseedV1 *tracklet = 0x0;
  for(Int_t il=0; il<AliTRDgeometry::kNlayer; il++){
    if(!(tracklet = fTrack->GetTracklet(il))) continue;
    if(!tracklet->IsOK()) continue;
    tracklet->Fit(kTRUE);

    x = tracklet->GetX0();
    y = tracklet->GetYfit(0)-tracklet->GetYfit(1)*(tracklet->GetX0()-x);
    z = tracklet->GetZfit(0);
    p.SetXYZ(x,y,z);
    fArray->AddPoint(ip++, &p);
  }
  fTree->Fill();

/*  if(fDebugLevel>=1){
    Float_t yt = fRim.GetYat(x[il]);
    (*fDebugStream) << "TrkltResiduals"
      << "layer="  << il
      << "x="      <<x[il]
      << "y="      <<y[il]
      << "yt="     <<yt
      << "dydx="   << dydx[il]
      << "dy="     << dy
      << "\n";
  }*/
  return 0x0;
}


//________________________________________________________
void AliTRDalignmentTask::Terminate(Option_t *)
{
  if(fDebugStream){ 
    delete fDebugStream;
    fDebugStream = 0x0;
    fDebugLevel = 0;
  }
  if(HasPostProcess()) PostProcess();
}


//________________________________________________________
TObjArray* AliTRDalignmentTask::Histos()
{
  return 0x0;
/*  if(fContainer) return fContainer;

  fContainer  = new TObjArray(3);

  TH1 *h = 0x0;
  
  if(!(h = (TH1D*)gROOT->FindObject("cuttra"))){
    h = new TH1D("cuttra","cuttra",20,0.5,20.5); 
  } else h->Reset();

  if(!(h = (TH1D*)gROOT->FindObject("cutpoi"))){
    h = new TH1D("cutpoi","cutpoi",20,0.5,20.5);
  } else h->Reset();

  if(!(h = (TH2D*)gROOT->FindObject("modpop"))){
    h = new TH2D("modpop","modpop",90,-0.5,89.5,30,-0.5,29.5);
    h->SetXTitle("module nr");
    h->SetYTitle("layer nr");
  } else h->Reset();*/
}


//________________________________________________________
Bool_t AliTRDalignmentTask::IsIdenticalWithOneOf(AliTrackPoint *p, AliTrackPointArray *parray, int nmax) {

  // Is the point p identical with one of the points on the list parray?
  // This is a fix for aliroot 4-16-Rev-01 (and before) writing some 
  // spurious unitialized points. 
 
  for (int i=0; i<parray->GetNPoints() && i<nmax; i++) {
    AliTrackPoint pa;
    parray->GetPoint(pa,i);
    //printf("comparing %7.3f with %7.3f\n",p->GetY(),pa.GetY());
    if (p->GetResidual(pa,0)<1e-8) return kTRUE;
    //printf("different\n");
  }
  return kFALSE;
}