Add streamer for std::vector<o2::tpc::PadFlags>

Author: ktf

DataFormats/Detectors/TPC/include/DataFormatsTPC/Defs.h

@@ -182,4 +182,23 @@ typename Enum<T>::Iterator end(Enum<T>)
 }
 } // namespace o2::tpc
 
+#ifdef __CLING__
+#include "DataFormatsTPC/VectorPadFlagsStreamer.h"
+
+struct VectorPadFlagsStreamerInit {
+  VectorPadFlagsStreamerInit()
+  {
+    auto cls = TClass::GetClass("std::vector<o2::tpc::PadFlags>");
+    if (cls) {
+      cls->AdoptStreamer(new VectorPadFlagsStreamer());
+    } else {
+      throw std::runtime_error("VectorPadFlagsStreamerInit: Class not found!");
+    }
+  }
+};
+
+static VectorPadFlagsStreamerInit _vectorPadFlagsStreamerInit;
+
+#endif // __CLING__
+
 #endif

DataFormats/Detectors/TPC/include/DataFormatsTPC/VectorPadFlagsStreamer.h

@@ -0,0 +1,51 @@
+// 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_DATAFORMATSTPC_VECTORPAD_FLAGSSTREAMER_H_
+#define O2_DATAFORMATSTPC_VECTORPAD_FLAGSSTREAMER_H_
+
+#include "DataFormatsTPC/Defs.h"
+#include <TClassStreamer.h>
+#include <TBuffer.h>
+#include <vector>
+
+class VectorPadFlagsStreamer : public TClassStreamer
+{
+ public:
+  void operator()(TBuffer& R__b, void* objp) override
+  {
+    std::vector<o2::tpc::PadFlags>* obj = static_cast<std::vector<o2::tpc::PadFlags>*>(objp);
+    if (R__b.IsReading()) {
+      std::vector<int> R__stl;
+      R__stl.clear();
+      int R__n;
+      R__b >> R__n;
+      R__stl.reserve(R__n);
+      for (int R__i = 0; R__i < R__n; R__i++) {
+        Int_t readtemp;
+        R__b >> readtemp;
+        R__stl.push_back(readtemp);
+      }
+      auto data = reinterpret_cast<unsigned short*>(R__stl.data());
+      for (int i = 0; i < R__n; ++i) {
+        obj->push_back(static_cast<o2::tpc::PadFlags>(data[i]));
+      }
+    } else {
+      // We always save things with the old format.
+      R__b << (int)obj->size() / 2;
+      for (size_t i = 0; i < obj->size(); i++) {
+        R__b << (short)obj->at(i);
+      }
+    }
+  }
+};
+
+#endif // O2_DATAFORMATSTPC_VECTORPAD_FLAGSSTREAMER_H_

GPU/Workflow/src/GPUWorkflowTPC.cxx

@@ -80,6 +80,8 @@
 #include <chrono>
 #include "GPUReconstructionConvert.h"
 #include "DetectorsRaw/RDHUtils.h"
+#include "DataFormatsTPC/VectorPadFlagsStreamer.h"
+
 #include <TStopwatch.h>
 #include <TObjArray.h>
 #include <TH1F.h>
@@ -98,6 +100,8 @@ namespace o2::gpu
 
 void GPURecoWorkflowSpec::initFunctionTPCCalib(InitContext& ic)
 {
+  TClass* cls = TClass::GetClass("std::vector<o2::tpc::PadFlags>");
+  cls->AdoptStreamer(new VectorPadFlagsStreamer());
   mTPCDeadChannelMapCreator.reset(new o2::tpc::DeadChannelMapCreator());
   const auto deadMapSource = (mSpecConfig.tpcDeadMapSources > -1) ? static_cast<tpc::SourcesDeadMap>(mSpecConfig.tpcDeadMapSources) : tpc::SourcesDeadMap::All;
   mTPCDeadChannelMapCreator->init();