started migration of TPC hough tracking code to TPC library

[u/mrichter/AliRoot.git] / HLT / TPCLib / tracking / AliHLTTPCHoughTracker.cxx

diff --git a/HLT/TPCLib/tracking/AliHLTTPCHoughTracker.cxx b/HLT/TPCLib/tracking/AliHLTTPCHoughTracker.cxx
new file mode 100644 (file)
index 0000000..71024d5
--- /dev/null
+++ b/HLT/TPCLib/tracking/AliHLTTPCHoughTracker.cxx
@@ -0,0 +1,94 @@
+// $Id$
+// origin: src/AliL3TPCtracker.cxx 1.2 Fri Jun 10 04:25:00 2005 UTC by cvetan
+
+/**************************************************************************
+ * 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-commercial 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.                  *
+ **************************************************************************/
+
+//-------------------------------------------------------------------------
+//       Implementation of the HLT TPC hough transform tracker class
+//
+//    It reads directly TPC digits using runloader and runs the HT
+//    algorithm over them.
+//    It stores the reconstructed hough tracks in the HLT ESD using the
+//    the off-line AliESDtrack format.
+//
+//       Origin: Cvetan Cheshkov, CERN, Cvetan.Cheshkov@cern.ch
+//-------------------------------------------------------------------------
+
+#include "AliESD.h"
+#include "AliRunLoader.h"
+#include "AliHLTTPCHoughTracker.h"
+#include "AliHLTHough.h"
+
+ClassImp(AliHLTTPCHoughTracker)
+
+AliHLTTPCHoughTracker::AliHLTTPCHoughTracker(AliRunLoader *runLoader):AliTracker()
+{
+  //--------------------------------------------------------------
+  // Constructor
+  //--------------------------------------------------------------
+
+  if(AliHLTTransform::GetVersion() == AliHLTTransform::kVdefault) {
+    Bool_t isinit=AliHLTTransform::Init(runLoader);
+    if(!isinit) AliWarning("Could not init AliHLTTransform settings, using defaults!");
+  }
+
+  fRunLoader = runLoader;
+}
+
+Int_t AliHLTTPCHoughTracker::Clusters2Tracks(AliESD *event)
+{
+  //--------------------------------------------------------------------
+  // This method reconstructs HLT TPC Hough tracks
+  //--------------------------------------------------------------------
+  
+  if (!fRunLoader) {
+    AliError("Missing runloader!");
+    return kTRUE;
+  }
+  Int_t iEvent = fRunLoader->GetEventNumber();
+  
+  Float_t ptmin = 0.1*AliHLTTransform::GetSolenoidField();
+
+  Float_t zvertex = GetZ();
+
+  AliInfo(Form("Hough Transform will run with ptmin=%f and zvertex=%f",ptmin,zvertex));
+
+  AliHLTHough *hough = new AliHLTHough();
+    
+  hough->SetThreshold(4);
+  hough->CalcTransformerParams(ptmin);
+  hough->SetPeakThreshold(70,-1);
+  hough->SetRunLoader(fRunLoader);
+  hough->Init("./", kFALSE, 100, kFALSE,4,0,0,zvertex);
+  hough->SetAddHistograms();
+
+  for(Int_t slice=0; slice<=35; slice++)
+    {
+      hough->ReadData(slice,iEvent);
+      hough->Transform();
+      hough->AddAllHistogramsRows();
+      hough->FindTrackCandidatesRow();
+      hough->AddTracks();
+    }
+
+  Int_t ntrk = hough->FillESD(event);
+
+  Info("Clusters2Tracks","Number of found tracks: %d\n",ntrk);
+  
+  delete hough;
+
+  return 0;
+}