o2AnalysisWorkflow.cxx

Go to the documentation of this file.

// Copyright 2019-2020 CERN and copyright holders of ALICE O2.
// See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
// All rights not expressly granted are reserved.
//
// This software is distributed under the terms of the GNU General Public
// License v3 (GPL Version 3), copied verbatim in the file "COPYING".
//
// 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/runDataProcessing.h"
#include "Framework/AnalysisTask.h"
#include "Framework/AnalysisDataModel.h"
#include <TH2F.h>
 
using namespace o2;
using namespace o2::framework;
using namespace o2::framework::expressions;
 
namespace o2::aod
{
namespace test
{
DECLARE_SOA_COLUMN(X, x, float);
DECLARE_SOA_COLUMN(Y, y, float);
DECLARE_SOA_COLUMN(Z, z, float);
} // namespace test
DECLARE_SOA_TABLE(Points, "AOD", "POINTS",
                  test::X, test::Y, test::Z);
} // namespace o2::aod
 
struct EtaAndClsHistograms {
  OutputObj<TH3F> etaClsH{TH3F("eta_vs_cls_vs_sigmapT", "#eta vs N_{cls} vs sigma_{1/pT}", 102, -2.01, 2.01, 160, -0.5, 159.5, 100, 0, 10)};
  Produces<aod::Points> points;
 
  void process(soa::Join<aod::FullTracks, aod::TracksCov> const& tracks)
  {
    for (auto& track : tracks) {
      etaClsH->Fill(track.eta(), track.tpcNClsFindable(), track.sigma1Pt());
      points(1, 2, 3);
    }
  }
};
 
WorkflowSpec defineDataProcessing(ConfigContext const& cfgc)
{
  // For the sake of running without an option, we do not throw an exception
  // in case the option is not present.
  if (cfgc.options().hasOption("aod-metadata-Run") == false ||
      cfgc.options().get<std::string>("aod-metadata-Run") == "2") {
    return WorkflowSpec{
      adaptAnalysisTask<EtaAndClsHistograms>(cfgc),
    };
  } else {
    throw std::runtime_error("Unsupported run type");
  }
}