forked from AliceO2Group/AliceO2
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathAO2DConverter.cxx
More file actions
88 lines (72 loc) · 3.06 KB
/
AO2DConverter.cxx
File metadata and controls
88 lines (72 loc) · 3.06 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
// 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.
/// \file AO2DConverter.cxx
/// \author Piotr Nowakowski
#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"}),
};
}