ExamplePrinterSpec.h

Go to the documentation of this file.

// 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.


#ifndef QUALITYCONTROL_EXAMPLEPRINTERSPEC_H
#define QUALITYCONTROL_EXAMPLEPRINTERSPEC_H

#include <string>
#include <memory>

#include <TH1F.h>
#include <TObjArray.h>

#include <Framework/Task.h>
#include <Framework/DataRefUtils.h>

#include "QualityControl/MonitorObject.h"
#include "QualityControl/QualityObject.h"

namespace o2::quality_control::example
{

class ExamplePrinterSpec : public framework::Task
{
 public:
  void run(ProcessingContext& processingContext) final
  {
    LOG(INFO) << "Received data";
    std::shared_ptr<TObjArray> moArray{ DataRefUtils::as<TObjArray>(*processingContext.inputs().begin()) };

    if (moArray->IsEmpty()) {
      LOG(INFO) << "Array is empty";
      return;
    }

    // get the object
    auto* mo = dynamic_cast<MonitorObject*>(moArray->At(0));
    if (mo == nullptr) {
      LOG(INFO) << "First element is not a MonitorObject";
      return;
    }
    auto* histo = dynamic_cast<TH1F*>(mo->getObject());
    if (histo == nullptr) {
      LOG(INFO) << "MonitorObject does not contain a TH1";
      return;
    }

    std::string bins = "BINS:";
    for (int i = 0; i < histo->GetNbinsX(); i++) {
      bins += " " + std::to_string((int)histo->GetBinContent(i));
    }
    LOG(INFO) << bins;
  }
};

class ExampleQualityPrinterSpec : public framework::Task
{
 public:
  void run(ProcessingContext& processingContext) final
  {
    auto qo = processingContext.inputs().get<QualityObject*>("checked-mo");

    LOG(INFO) << "Received Quality: " << qo->getQuality();
  }
};

} // namespace o2::quality_control::example

#endif //QUALITYCONTROL_EXAMPLEPRINTERSPEC_H