AO2DConverter.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 "EveWorkflow/DetectorData.h"
#include "EveWorkflow/EveWorkflowHelper.h"
#include "Framework/AnalysisTask.h"
#include "DetectorsBase/Propagator.h"
#include "DataFormatsTPC/WorkflowHelper.h"
#include "CommonUtils/NameConf.h"
#include "TRDBase/Geometry.h"
#include "ReconstructionDataFormats/GlobalTrackID.h"
#include "DataFormatsMCH/TrackMCH.h"
#include "DataFormatsMCH/ROFRecord.h"
 
#include <memory>
 
using namespace o2::event_visualisation;
using namespace o2::framework;
using namespace o2::dataformats;
using namespace o2::globaltracking;
using namespace o2::tpc;
using namespace o2::trd;
 
#include "Framework/runDataProcessing.h"
 
struct AO2DConverter {
  o2::framework::Configurable<std::string> jsonPath{"jsons-folder", "./json", "name of the folder to store json files"};
 
  static constexpr float mWorkflowVersion = 1.00;
  o2::header::DataHeader::RunNumberType mRunNumber = 0;
  o2::header::DataHeader::TFCounterType mTfCounter = 0;
  o2::header::DataHeader::TForbitType mTfOrbit = 0;
  o2::framework::DataProcessingHeader::CreationTime mCreationTime;
 
  DetectorData mData;
  std::shared_ptr<EveWorkflowHelper> mHelper;
 
  void init(o2::framework::InitContext& ic)
  {
    LOGF(info, "------------------------    AO2DConverter::init version ", mWorkflowVersion, "    ------------------------------------");
 
    mData.init();
    mHelper = std::make_shared<EveWorkflowHelper>();
  }
 
  void process(o2::aod::Collision const& collision, EveWorkflowHelper::AODBarrelTracks const& barrelTracks, EveWorkflowHelper::AODForwardTracks const& fwdTracks, EveWorkflowHelper::AODMFTTracks const& mftTracks)
  {
    for (auto const& track : barrelTracks) {
      mHelper->drawAODBarrel(track, collision.collisionTime());
    }
 
    for (auto const& track : mftTracks) {
      mHelper->drawAODMFT(track, collision.collisionTime());
    }
 
    for (auto const& track : fwdTracks) {
      mHelper->drawAODFwd(track, collision.collisionTime());
    }
 
    mHelper->mEvent.setClMask(GlobalTrackID::MASK_NONE.to_ulong());
    mHelper->mEvent.setTrkMask(GlobalTrackID::MASK_ALL.to_ulong());
    mHelper->mEvent.setRunNumber(mRunNumber);
    mHelper->mEvent.setTfCounter(mTfCounter);
    mHelper->mEvent.setFirstTForbit(mTfOrbit);
    mHelper->mEvent.setCreationTime(collision.collisionTime());
 
    mHelper->save(jsonPath, ".root", -1);
    mHelper->clear();
  }
};
 
WorkflowSpec defineDataProcessing(ConfigContext const& cfgc)
{
  return WorkflowSpec{
    adaptAnalysisTask<AO2DConverter>(cfgc, TaskName{"o2-aodconverter"}),
  };
}