X-Git-Url: http://git.uio.no/git/?p=u%2Fmrichter%2FAliRoot.git;a=blobdiff_plain;f=PWG%2FDevNanoAOD%2FAliAnalysisNanoAODCuts.cxx;h=ae5b4e861918174aba853a22e14301024e4332cb;hp=fcef18cfa587861a9d9355560404354d1602b814;hb=1a468a3b24d18b41a25bc96975c834da2091b77b;hpb=778636d40b5808d8277e6a0a07690e3f632ccdca

diff --git a/PWG/DevNanoAOD/AliAnalysisNanoAODCuts.cxx b/PWG/DevNanoAOD/AliAnalysisNanoAODCuts.cxx
index fcef18cfa58..ae5b4e86191 100644
--- a/PWG/DevNanoAOD/AliAnalysisNanoAODCuts.cxx
+++ b/PWG/DevNanoAOD/AliAnalysisNanoAODCuts.cxx
@@ -3,13 +3,15 @@
 #include "AliAODEvent.h"
 #include "AliNanoAODHeader.h"
 #include "AliNanoAODTrack.h"
+#include <iomanip>
 
 ClassImp(AliAnalysisNanoAODTrackCuts)
-
+ClassImp(AliAnalysisNanoAODEventCuts)
+ClassImp(AliNanoAODSimpleSetter)
 
 
 AliAnalysisNanoAODTrackCuts::AliAnalysisNanoAODTrackCuts():
-  AliAnalysisCuts(), fBitMask(1) 
+AliAnalysisCuts(), fBitMask(1), fMinPt(0), fMaxEta(10)
 {
   // default ctor 
 }
@@ -18,9 +20,13 @@ Bool_t AliAnalysisNanoAODTrackCuts::IsSelected(TObject* obj)
 {
   // Returns true if the track is good!
   AliAODTrack* track = dynamic_cast<AliAODTrack*>(obj);
-
-  if (track->TestFilterBit(fBitMask)) return kTRUE;
-  return kFALSE;  
+  
+  
+  if (!track->TestFilterBit(fBitMask))    return kFALSE;
+  if (track->Pt() < fMinPt)               return kFALSE;
+  if (TMath::Abs(track->Eta()) > fMaxEta) return kFALSE; 
+  
+  return kTRUE;  
 
 }
 
@@ -38,6 +44,20 @@ Bool_t AliAnalysisNanoAODEventCuts::IsSelected(TObject* obj)
   AliAODVertex * vertex = evt->GetPrimaryVertex();
   if (!vertex) return kFALSE;
   if (TMath::Abs(vertex->GetZ()) > fVertexRange) return kFALSE;
+  
   return kTRUE;
   
 }
+
+
+void AliNanoAODSimpleSetter::SetNanoAODHeader(const AliAODEvent * event   , AliNanoAODHeader * head  ) {
+
+  AliAODHeader * header = dynamic_cast<AliAODHeader*>(event->GetHeader());
+  if (!header) AliFatal("Not a standard AOD");
+  // Set custom nano aod vars
+  Double_t centr    = header->GetCentralityP()->GetCentralityPercentile("V0M");
+  Double_t magfield = header->GetMagneticField();
+  head->SetVar(0, centr);
+  head->SetVar(1, magfield);
+
+}