// Copyright CERN and copyright holders of ALICE O2. This software is

// distributed under the terms of the GNU General Public License v3 (GPL

// Version 3), copied verbatim in the file "COPYING".

//

// See http://alice-o2.web.cern.ch/license for full licensing information.

//

// In applying this license CERN does not waive the privileges and immunities

// granted to it by virtue of its status as an Intergovernmental Organization

// or submit itself to any jurisdiction.

#include "Framework/AnalysisDataModel.h"

#include "Framework/AnalysisTask.h"

#include "Framework/runDataProcessing.h"

#include <TH1F.h>

#include <cmath>

#include "AnalysisCore/TrackSelection.h"

#include "AnalysisDataModel/TrackSelectionTables.h"

using namespace o2;

using namespace o2::framework;

using namespace o2::framework::expressions;

//--------------------------------------------------------------------

// Task generating pT spectrum of charged particles

//--------------------------------------------------------------------

struct HistogramTrackSelection {

Configurable<int> selectedTracks{"select", 1, "Choice of track selection. 0 = no selection, 1 = globalTracks, 2 = globalTracksSDD"};

OutputObj<TH1F> pt{TH1F("pt", "pt", 100, 0., 50.)};

void init(o2::framework::InitContext&) {}

void process(aod::Collision const& collision,

soa::Join<aod::Tracks, aod::TrackSelection> const& tracks)

{

for (auto& track : tracks) {

if (selectedTracks == 1 && !track.isGlobalTrack()) {

continue;

} else if (selectedTracks == 2 && !track.isGlobalTrackSDD()) {

continue;

}

pt->Fill(track.pt());

}

}

};

//--------------------------------------------------------------------

// Workflow definition

//--------------------------------------------------------------------

WorkflowSpec defineDataProcessing(ConfigContext const&)

{

return WorkflowSpec{

adaptAnalysisTask<HistogramTrackSelection>("histogram-track-selection")};

}