-
Notifications
You must be signed in to change notification settings - Fork 444
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Adding workflow to read in integrated clusters from root files
- writing TFIDInfo to TTree - moving writer to workflowIO
- Loading branch information
1 parent
858a822
commit 773b62b
Showing
10 changed files
with
246 additions
and
16 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,25 @@ | ||
| // 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 "TOFWorkflowIO/TOFIntegrateClusterReaderSpec.h" | ||
| #include "CommonUtils/ConfigurableParam.h" | ||
| #include "Framework/ConfigParamSpec.h" | ||
|
|
||
| using namespace o2::framework; | ||
|
|
||
| #include "Framework/runDataProcessing.h" | ||
|
|
||
| WorkflowSpec defineDataProcessing(ConfigContext const& cfgc) | ||
| { | ||
| WorkflowSpec wf; | ||
| wf.emplace_back(o2::tof::getTOFIntegrateClusterReaderSpec()); | ||
| return wf; | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
27 changes: 27 additions & 0 deletions
27
Detectors/TOF/workflowIO/include/TOFWorkflowIO/TOFIntegrateClusterReaderSpec.h
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,27 @@ | ||
| // 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. | ||
|
|
||
| #ifndef O2_TOF_TOFINTEGRATECLUSTERREADERSPEC_SPEC | ||
| #define O2_TOF_TOFINTEGRATECLUSTERREADERSPEC_SPEC | ||
|
|
||
| #include "Framework/DataProcessorSpec.h" | ||
|
|
||
| namespace o2 | ||
| { | ||
| namespace tof | ||
| { | ||
|
|
||
| o2::framework::DataProcessorSpec getTOFIntegrateClusterReaderSpec(); | ||
|
|
||
| } // end namespace tof | ||
| } // end namespace o2 | ||
|
|
||
| #endif |
File renamed without changes.
158 changes: 158 additions & 0 deletions
158
Detectors/TOF/workflowIO/src/TOFIntegrateClusterReaderSpec.cxx
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,158 @@ | ||
| // 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 TOFIntegrateClusterReaderSpec.cxx | ||
|
|
||
| #include <vector> | ||
| #include <boost/algorithm/string/predicate.hpp> | ||
|
|
||
| #include "TOFWorkflowIO/TOFIntegrateClusterReaderSpec.h" | ||
| #include "Framework/Task.h" | ||
| #include "Framework/ControlService.h" | ||
| #include "Framework/ConfigParamRegistry.h" | ||
| #include "CommonUtils/NameConf.h" | ||
| #include "CommonDataFormat/TFIDInfo.h" | ||
| #include "Algorithm/RangeTokenizer.h" | ||
| #include "TChain.h" | ||
| #include "TGrid.h" | ||
|
|
||
| using namespace o2::framework; | ||
|
|
||
| namespace o2 | ||
| { | ||
| namespace tof | ||
| { | ||
|
|
||
| class IntegratedClusterReader : public Task | ||
| { | ||
| public: | ||
| IntegratedClusterReader() = default; | ||
| ~IntegratedClusterReader() override = default; | ||
| void init(InitContext& ic) final; | ||
| void run(ProcessingContext& pc) final; | ||
|
|
||
| private: | ||
| void connectTrees(); | ||
|
|
||
| int mChainEntry = 0; ///< processed entries in the chain | ||
| std::unique_ptr<TChain> mChain; ///< input TChain | ||
| std::vector<std::string> mFileNames; ///< input files | ||
| std::vector<float> mTOFCNCl, *mTOFCNClPtr = &mTOFCNCl; ///< branch integrated number of cluster TOF currents | ||
| std::vector<float> mTOFCqTot, *mTOFCqTotPtr = &mTOFCqTot; ///< branch integrated q TOF currents | ||
| o2::dataformats::TFIDInfo mTFinfo, *mTFinfoPtr = &mTFinfo; ///< branch TFIDInfo for injecting correct time | ||
| std::vector<std::pair<unsigned long, int>> mIndices; ///< firstTfOrbit, file, index | ||
| }; | ||
|
|
||
| void IntegratedClusterReader::init(InitContext& ic) | ||
| { | ||
| const auto dontCheckFileAccess = ic.options().get<bool>("dont-check-file-access"); | ||
| auto fileList = o2::RangeTokenizer::tokenize<std::string>(ic.options().get<std::string>("tofcurrents-infiles")); | ||
|
|
||
| // check if only one input file (a txt file contaning a list of files is provided) | ||
| if (fileList.size() == 1) { | ||
| if (boost::algorithm::ends_with(fileList.front(), "txt")) { | ||
| LOGP(info, "Reading files from input file {}", fileList.front()); | ||
| std::ifstream is(fileList.front()); | ||
| std::istream_iterator<std::string> start(is); | ||
| std::istream_iterator<std::string> end; | ||
| std::vector<std::string> fileNamesTmp(start, end); | ||
| fileList = fileNamesTmp; | ||
| } | ||
| } | ||
|
|
||
| const std::string inpDir = o2::utils::Str::rectifyDirectory(ic.options().get<std::string>("input-dir")); | ||
| for (const auto& file : fileList) { | ||
| if ((file.find("alien://") == 0) && !gGrid && !TGrid::Connect("alien://")) { | ||
| LOG(fatal) << "Failed to open alien connection"; | ||
| } | ||
| const auto fileDir = o2::utils::Str::concat_string(inpDir, file); | ||
| if (!dontCheckFileAccess) { | ||
| std::unique_ptr<TFile> filePtr(TFile::Open(fileDir.data())); | ||
| if (!filePtr || !filePtr->IsOpen() || filePtr->IsZombie()) { | ||
| LOGP(warning, "Could not open file {}", fileDir); | ||
| continue; | ||
| } | ||
| } | ||
| mFileNames.emplace_back(fileDir); | ||
| } | ||
|
|
||
| if (mFileNames.size() == 0) { | ||
| LOGP(error, "No input files to process"); | ||
| } | ||
| connectTrees(); | ||
| } | ||
|
|
||
| void IntegratedClusterReader::run(ProcessingContext& pc) | ||
| { | ||
| // check time order inside the TChain | ||
| if (mChainEntry == 0) { | ||
| mIndices.clear(); | ||
| mIndices.reserve(mChain->GetEntries()); | ||
| for (unsigned long i = 0; i < mChain->GetEntries(); i++) { | ||
| mChain->GetEntry(i); | ||
| mIndices.emplace_back(std::make_pair(mTFinfo.firstTForbit, i)); | ||
| } | ||
| std::sort(mIndices.begin(), mIndices.end()); | ||
| } | ||
|
|
||
| LOGP(debug, "Processing entry {}", mIndices[mChainEntry].second); | ||
| mChain->GetEntry(mIndices[mChainEntry++].second); | ||
|
|
||
| // inject correct timing informations | ||
| auto& timingInfo = pc.services().get<o2::framework::TimingInfo>(); | ||
| timingInfo.firstTForbit = mTFinfo.firstTForbit; | ||
| timingInfo.tfCounter = mTFinfo.tfCounter; | ||
| timingInfo.runNumber = mTFinfo.runNumber; | ||
| timingInfo.creation = mTFinfo.creation; | ||
|
|
||
| pc.outputs().snapshot(Output{header::gDataOriginTOF, "ITOFCN"}, mTOFCNCl); | ||
| pc.outputs().snapshot(Output{header::gDataOriginTOF, "ITOFCQ"}, mTOFCqTot); | ||
| usleep(100); | ||
|
|
||
| if (mChainEntry >= mChain->GetEntries()) { | ||
| pc.services().get<ControlService>().endOfStream(); | ||
| pc.services().get<ControlService>().readyToQuit(QuitRequest::Me); | ||
| } | ||
| } | ||
|
|
||
| void IntegratedClusterReader::connectTrees() | ||
| { | ||
| mChain.reset(new TChain("itofc")); | ||
| for (const auto& file : mFileNames) { | ||
| LOGP(info, "Adding file to chain: {}", file); | ||
| mChain->AddFile(file.data()); | ||
| } | ||
| assert(mChain->GetEntries()); | ||
| mChain->SetBranchAddress("ITOFCN", &mTOFCNClPtr); | ||
| mChain->SetBranchAddress("ITOFCQ", &mTOFCqTotPtr); | ||
| mChain->SetBranchAddress("tfID", &mTFinfoPtr); | ||
| } | ||
|
|
||
| DataProcessorSpec getTOFIntegrateClusterReaderSpec() | ||
| { | ||
| std::vector<OutputSpec> outputs; | ||
| outputs.emplace_back(o2::header::gDataOriginTOF, "ITOFCN", 0, Lifetime::Sporadic); | ||
| outputs.emplace_back(o2::header::gDataOriginTOF, "ITOFCQ", 0, Lifetime::Sporadic); | ||
|
|
||
| return DataProcessorSpec{ | ||
| "tof-integrated-cluster-reader", | ||
| Inputs{}, | ||
| outputs, | ||
| AlgorithmSpec{adaptFromTask<IntegratedClusterReader>()}, | ||
| Options{ | ||
| {"tofcurrents-infiles", VariantType::String, "o2currents_tof.root", {"comma-separated list of input files or .txt file containing list of input files"}}, | ||
| {"input-dir", VariantType::String, "none", {"Input directory"}}, | ||
| {"dont-check-file-access", VariantType::Bool, false, {"Deactivate check if all files are accessible before adding them to the list of files"}}, | ||
| }}; | ||
| } | ||
|
|
||
| } // namespace tof | ||
| } // namespace o2 |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters