Use abstract inteface for the calibration components using tracks

[u/mrichter/AliRoot.git] / TPC / AliAnaTPCTrackCalib.cxx

#include "AliAnaTPCTrackCalib.h"\r
//\r
// This class is ment as an example of how to get access to the \r
// clusters associated with reconstructed ESD tracks \r
//\r
//\r
\r
// ROOT includes\r
#include <TChain.h>\r
#include <TMath.h>\r
\r
// ALIROOT includes\r
#include <AliTPCclusterMI.h>\r
#include <AliTPCcalibTracksCuts.h>\r
#include <AliTPCClusterParam.h>\r
\r
// STL includes\r
#include <iostream>\r
\r
using namespace std;\r
\r
ClassImp(AliAnaTPCTrackCalib)\r
  \r
//________________________________________________________________________\r
AliAnaTPCTrackCalib::AliAnaTPCTrackCalib() : \r
  AliAnaTPCTrackBase(), fNtracks(0), fNClusters(0), fCalibTracks(0)\r
{\r
  //\r
  // Default constructor (should not be used)\r
  //\r
}\r
\r
//________________________________________________________________________\r
AliAnaTPCTrackCalib::AliAnaTPCTrackCalib(const char *name) : \r
  AliAnaTPCTrackBase(name), fNtracks(0), fNClusters(0), fCalibTracks(0)\r
{\r
  //\r
  // Normal constructor\r
  //\r
  printf("Normal constructor called with name %s \n", name);  \r
  // Input slot #1 works with a AliTPCCalibTracksCuts\r
  DefineInput(1, AliTPCcalibTracksCuts::Class());\r
  // Input slot #2 works with a AliTPCClusterParam\r
  DefineInput(2, AliTPCClusterParam::Class());\r
  \r
  // Output slot #0 writes into a TList\r
  // DefineOutput(0, TList::Class());\r
 }\r
\r
//________________________________________________________________________\r
void AliAnaTPCTrackCalib::CreateOutputObjects() \r
{\r
   //\r
   // Connect the output objects\r
   //\r
   if(fDebug>0)\r
    cout << "AliAnaTPCTrackCalib::CreateOutputObjects()" << endl;\r
\r
   AliAnaTPCTrackBase::CreateOutputObjects();\r
  \r
   // user code to go here\r
   if (GetInputData(0)) printf("Input slo 0, Class_Name: %s\n", GetInputData(0)->Class_Name());\r
   \r
   fNtracks       = new TH1I("ntracks","Number of tracks", 100, 0, 400);\r
   fListOfHists->Add(fNtracks);\r
   fNClusters     = new TH1I("ncluster","Number of clusters",100, 0, 200);\r
   fListOfHists->Add(fNClusters);\r
   \r
   AliTPCcalibTracksCuts *cuts = (AliTPCcalibTracksCuts*)GetInputData(1);\r
   AliTPCClusterParam *clusterParam = (AliTPCClusterParam*)GetInputData(2);\r
   if (!cuts) Error("CreateOutputObjects", "No CUTS found in input slot 1");\r
   else {\r
      printf("\nCuts found :-) \n");\r
      cuts->Print();\r
   }\r
   if (!clusterParam) Error("CreateOutputObjects", "No CLUSTERPARAM found in input slot 2");\r
   \r
/*   if ( !fCalibTracks ) {\r
      OpenFile(0, "RECREATE");*/\r
      \r
   fCalibTracks = new AliTPCcalibTracks("calibTracks", "Resolution calibration object for tracks", clusterParam, cuts);\r
   fListOfHists->Add(fCalibTracks);\r
}\r
\r
\r
//________________________________________________________________________\r
Int_t AliAnaTPCTrackCalib::FillTrackHistograms(Int_t nTracks, AliESDtrack* track, AliESDfriendTrack* friendTrack, AliTPCseed* seed) {\r
  //\r
  // This is the main method which rejects noise tracks and fills \r
  // the histograms\r
  //\r
\r
  if(!nTracks || !track || !friendTrack) {\r
    if (fDebug > 1) AliWarning("WARNING: missing track information in AliAnaTPCTrackCalib");\r
    return 0;\r
  }\r
  if(seed==0) {\r
    if (fDebug > 1) AliWarning("WARNING: Missing seed in AliAnaTPCTrackCalib");\r
    return 0;\r
  }\r
   \r
   // calibration components to go here:  \r
  \r
   if (seed) {\r
      fNClusters->Fill(seed->GetNumberOfClusters());\r
      fCalibTracks->Process(seed);   // analysis is done in fCalibTracks\r
   }\r
 \r
  return 1;\r
}      \r
\r