SiStripApproximateCluster reduced data format for HI

DataFormats/SiStripCluster/interface/SiStripApproximateClusterv1.h

@@ -0,0 +1,31 @@
+#ifndef DATAFORMATS_SISTRIPAPPROXIMATECLUSTERv1_H
+#define DATAFORMATS_SISTRIPAPPROXIMATECLUSTERv1_H
+
+#include <vector>
+#include <numeric>
+#include "FWCore/MessageLogger/interface/MessageLogger.h"
+
+class SiStripApproximateClusterv1  {
+public:
+
+  SiStripApproximateClusterv1() {}
+
+  explicit SiStripApproximateClusterv1(uint8_t avgCharge, uint16_t barycenter, uint8_t width):avgCharge_(avgCharge) {
+    baryWidth_ = barycenter;
+    baryWidth_ = baryWidth_ << 6;
+    if(width>0x3F) width=0x3F;
+    baryWidth_ += width;
+  }
+
+  uint16_t barycenter() const {return (uint16_t)((baryWidth_ & 0xFFC0) >> 6);}
+  uint8_t width() const {return (uint8_t) (baryWidth_ & 0x3F);}
+  uint16_t baryWidth() const {return baryWidth_;}
+  uint8_t  avgCharge() const{return avgCharge_;} 
+
+private:
+
+  uint16_t                baryWidth_ = 0;
+
+  uint8_t                 avgCharge_ = 0;
+};
+#endif // DATAFORMATS_SISTRIPAPPROXIMATECLUSTERv1_H

DataFormats/SiStripCluster/src/classes.h

@@ -5,6 +5,8 @@
 #include "DataFormats/Common/interface/DetSetVectorNew.h"
 #include "DataFormats/Common/interface/Ref.h"
 #include "DataFormats/SiStripCluster/interface/SiStripCluster.h"
+#include "DataFormats/SiStripCluster/interface/SiStripApproximateClusterv1.h"
+
 #include "DataFormats/Common/interface/ContainerMask.h"
 
 #endif  // SISTRIPCLUSTER_CLASSES_H

DataFormats/SiStripCluster/src/classes_def.xml

@@ -23,4 +23,21 @@
  <class name="edm::Wrapper<edmNew::DetSetVector<edm::Ref<edmNew::DetSetVector<SiStripCluster>,SiStripCluster,edmNew::DetSetVector<SiStripCluster>::FindForDetSetVector> > >" />
 
 
+<class name="SiStripApproximateClusterv1"/>
+ <class name="edmNew::DetSetVector<SiStripApproximateClusterv1>"/>
+ <class name="edm::Wrapper<edmNew::DetSetVector<SiStripApproximateClusterv1>>"/>
+
+ <class name="std::vector<SiStripApproximateClusterv1>"/>
+
+ <class name="edm::Ref<edmNew::DetSetVector<SiStripApproximateClusterv1>, SiStripApproximateClusterv1, edmNew::DetSetVector<SiStripApproximateClusterv1>::FindForDetSetVector>"/>
+
+
+ <class name="edm::ContainerMask<edmNew::DetSetVector<SiStripApproximateClusterv1> >"/>
+ <class name="edm::Wrapper<edm::ContainerMask<edmNew::DetSetVector<SiStripApproximateClusterv1> > >"/>
+
+
+ <class name="std::vector<edm::Ref<edmNew::DetSetVector<SiStripApproximateClusterv1>,SiStripApproximateClusterv1,edmNew::DetSetVector<SiStripApproximateClusterv1>::FindForDetSetVector> >" />
+ <class name="edmNew::DetSetVector<edm::Ref<edmNew::DetSetVector<SiStripApproximateClusterv1>,SiStripApproximateClusterv1,edmNew::DetSetVector<SiStripApproximateClusterv1>::FindForDetSetVector> >" />
+ <class name="edm::Wrapper<edmNew::DetSetVector<edm::Ref<edmNew::DetSetVector<SiStripApproximateClusterv1>,SiStripApproximateClusterv1,edmNew::DetSetVector<SiStripApproximateClusterv1>::FindForDetSetVector> > >" />
+
 </lcgdict>

RecoLocalTracker/SiStripClusterizer/BuildFile.xml

@@ -8,6 +8,9 @@
 <use name="CalibFormats/SiStripObjects"/>
 <use name="CalibTracker/Records"/>
 <use name="RecoLocalTracker/Records"/>
+<use name="CommonTools/UtilAlgos"/>
+<use name="CommonTools/Utils"/>
+<use name="root"/>
 <export>
   <lib name="1"/>
 </export>

RecoLocalTracker/SiStripClusterizer/interface/SiStripClusters2ApproxClustersv1.h

@@ -0,0 +1,32 @@
+#ifndef RecoLocalTracker_SiStripClusters2ApproxClustersv1_h
+#define RecoLocalTracker_SiStripClusters2ApproxClustersv1_h
+
+#include "FWCore/Framework/interface/MakerMacros.h"
+#include "FWCore/Framework/interface/Frameworkfwd.h"
+#include "FWCore/Framework/interface/stream/EDProducer.h"
+#include "FWCore/ParameterSet/interface/ParameterSet.h"
+#include "FWCore/Utilities/interface/InputTag.h"
+#include "DataFormats/SiStripCluster/interface/SiStripApproximateClusterv1.h"
+#include "DataFormats/SiStripCluster/interface/SiStripCluster.h"
+#include "DataFormats/Common/interface/DetSetVectorNew.h"
+#include "DataFormats/Common/interface/DetSetVector.h"
+
+#include <vector>
+#include <memory>
+
+class SiStripClusters2ApproxClustersv1: public edm::stream::EDProducer<>  {
+
+public:
+
+  explicit SiStripClusters2ApproxClustersv1(const edm::ParameterSet& conf);
+  void produce(edm::Event&, const edm::EventSetup&) override;
+
+private:
+
+  edm::InputTag inputClusters;
+  edm::EDGetTokenT< edmNew::DetSetVector<SiStripCluster> > clusterToken;  
+};
+
+DEFINE_FWK_MODULE(SiStripClusters2ApproxClustersv1);
+#endif
+

RecoLocalTracker/SiStripClusterizer/plugins/SiStripClusters2ApproxClustersv1.cc

@@ -0,0 +1,51 @@
+#include "RecoLocalTracker/SiStripClusterizer/interface/SiStripClusters2ApproxClustersv1.h"
+#include <iostream>
+
+
+SiStripClusters2ApproxClustersv1::SiStripClusters2ApproxClustersv1(const edm::ParameterSet& conf){
+   inputClusters = conf.getParameter< edm::InputTag >("inputClusters");
+   clusterToken = consumes< edmNew::DetSetVector< SiStripCluster > >(inputClusters);
+
+   produces< edmNew::DetSetVector< SiStripApproximateClusterv1 > >(); 
+
+}
+
+void SiStripClusters2ApproxClustersv1::produce(edm::Event& e, edm::EventSetup const&){
+  std::unique_ptr<edmNew::DetSetVector< SiStripApproximateClusterv1 > > result(new edmNew::DetSetVector< SiStripApproximateClusterv1 > );
+
+  edm::Handle<edmNew::DetSetVector< SiStripCluster >> clusterCollection;
+  e.getByToken(clusterToken, clusterCollection);
+
+  uint32_t minID = 470444276;
+  int maxFirst = -1;
+
+  for( edmNew::DetSetVector<SiStripCluster>::const_iterator i = clusterCollection->begin(); i!=clusterCollection->end(); i++){
+
+    std::vector< SiStripApproximateClusterv1 > tempVec;    
+
+    edmNew::DetSetVector<SiStripApproximateClusterv1>::FastFiller ff = edmNew::DetSetVector<SiStripApproximateClusterv1>::FastFiller(*result, i->id());
+
+    for( edmNew::DetSet<SiStripCluster>::const_iterator j = i->begin(); j!=i->end(); j++){
+
+      if(maxFirst<j->firstStrip()) maxFirst = j->firstStrip();
+      if(minID>i->id()) minID = i->id();
+
+      uint8_t width = j->amplitudes().size();
+      uint16_t barycenter = (uint16_t)j->barycenter();
+
+      int charge = 0;
+      for (unsigned k = 0; k < j->amplitudes().size(); k++) {
+        charge += (int)j->amplitudes()[k];
+      }
+
+      SiStripApproximateClusterv1 approxCluster = SiStripApproximateClusterv1( (uint8_t)(charge/width), barycenter, width );
+      ff.push_back(approxCluster);
+    }
+
+
+  }
+
+  e.put(std::move(result));
+}
+
+

RecoLocalTracker/SiStripClusterizer/python/SiStripApproximatedClustersDump_cfi.py

@@ -0,0 +1,5 @@
+import FWCore.ParameterSet.Config as cms
+
+SiStripApproximatedClustersDump = cms.EDAnalyzer("SiStripApproximatedClustersDump",
+    approximatedClustersTag = cms.InputTag("SiStripClusters2ApproxClustersv1")
+)

RecoLocalTracker/SiStripClusterizer/python/SiStripClusters2ApproxClustersv1_cfi.py

@@ -0,0 +1,6 @@
+import FWCore.ParameterSet.Config as cms
+
+SiStripClusters2ApproxClustersv1 = cms.EDProducer("SiStripClusters2ApproxClustersv1",
+	inputClusters = cms.InputTag("siStripClusters")
+)
+

RecoLocalTracker/SiStripClusterizer/test/SealModules.cc

@@ -9,9 +9,12 @@
 #include "RecoLocalTracker/SiStripClusterizer/test/StripByStripTestDriver.h"
 #include "RecoLocalTracker/SiStripClusterizer/test/ClusterizerUnitTesterESProducer.h"
 #include "RecoLocalTracker/SiStripClusterizer/test/ClusterRefinerTagMCmerged.h"
+#include "RecoLocalTracker/SiStripClusterizer/test/SiStripApproximatedClustersDump.h"
 
 DEFINE_FWK_MODULE(CompareClusters);
 DEFINE_FWK_MODULE(ClusterizerUnitTester);
 DEFINE_FWK_MODULE(StripByStripTestDriver);
 DEFINE_FWK_EVENTSETUP_MODULE(ClusterizerUnitTesterESProducer);
 DEFINE_FWK_MODULE(ClusterRefinerTagMCmerged);
+DEFINE_FWK_MODULE(SiStripApproximatedClustersDump);
+

RecoLocalTracker/SiStripClusterizer/test/SiStripApproximatedClustersDump.cc

@@ -0,0 +1,41 @@
+#include "RecoLocalTracker/SiStripClusterizer/test/SiStripApproximatedClustersDump.h"
+
+
+SiStripApproximatedClustersDump::SiStripApproximatedClustersDump(const edm::ParameterSet& conf) {
+  inputTagClusters = conf.getParameter< edm::InputTag >("approximatedClustersTag");
+  clusterToken = consumes< edmNew::DetSetVector<SiStripApproximateClusterv1>>(inputTagClusters);
+
+  usesResource("TFileService");
+
+  outNtuple = fs->make<TTree>("ApproxClusters", "ApproxClusters");
+  outNtuple->Branch("event", &eventN, "event/i");
+  outNtuple->Branch("detId", &detId, "detId/i");
+  outNtuple->Branch("barycenter", &barycenter, "barycenter/F");
+  outNtuple->Branch("width", &width, "width/b");
+  outNtuple->Branch("charge", &avCharge, "charge/b");
+}
+
+SiStripApproximatedClustersDump::~SiStripApproximatedClustersDump() {}
+
+void SiStripApproximatedClustersDump::analyze(const edm::Event& event, const edm::EventSetup& es) {
+
+  edm::Handle<edmNew::DetSetVector<SiStripApproximateClusterv1>> inClusters;
+  event.getByToken(clusterToken, inClusters);
+
+
+  for (edmNew::DetSetVector<SiStripApproximateClusterv1>::const_iterator itApprox = inClusters->begin(); itApprox!= inClusters->end(); itApprox++) {
+    detId = itApprox->detId();
+    eventN = event.id().event();
+
+    for (edmNew::DetSet<SiStripApproximateClusterv1>::const_iterator itClusters = itApprox->begin(); itClusters!= itApprox->end(); itClusters++){
+
+      barycenter = itClusters->barycenter();
+      width = itClusters->width();
+      avCharge=itClusters->avgCharge();
+      outNtuple->Fill();
+
+    }
+  }
+}
+
+

RecoLocalTracker/SiStripClusterizer/test/SiStripApproximatedClustersDump.h

@@ -0,0 +1,59 @@
+#ifndef RecoLocalTracker_SiStripApproximatedClustersDump_h
+#define RecoLocalTracker_SiStripApproximatedClustersDump_h
+
+
+// user include files
+#include "FWCore/Framework/interface/Frameworkfwd.h"
+#include "FWCore/Framework/interface/one/EDAnalyzer.h"
+#include "FWCore/Framework/interface/MakerMacros.h"
+
+#include "FWCore/Framework/interface/Event.h"
+#include "FWCore/ParameterSet/interface/ParameterSet.h"
+#include "DataFormats/Common/interface/DetSet.h"
+#include "DataFormats/Common/interface/DetSetVector.h"
+#include "DataFormats/Common/interface/DetSetVectorNew.h"
+#include "DataFormats/SiStripCluster/interface/SiStripApproximateClusterv1.h"
+
+#include "FWCore/ServiceRegistry/interface/Service.h"
+#include "CommonTools/UtilAlgos/interface/TFileService.h"
+#include "CommonTools/Utils/interface/TFileDirectory.h"
+
+#include <memory>
+#include <iostream>
+
+//ROOT inclusion
+#include "TROOT.h"
+#include "TFile.h"
+#include "TNtuple.h"
+#include "TTree.h"
+#include "TMath.h"
+#include "TList.h"
+#include "TString.h"
+
+//
+// class decleration
+//
+
+
+class SiStripApproximatedClustersDump : public edm::one::EDAnalyzer<edm::one::SharedResources> {
+public:
+  explicit SiStripApproximatedClustersDump(const edm::ParameterSet&);
+  ~SiStripApproximatedClustersDump() override;
+
+
+private:
+  void analyze(const edm::Event&, const edm::EventSetup&) override;
+
+  edm::InputTag inputTagClusters;
+  edm::EDGetTokenT< edmNew::DetSetVector<SiStripApproximateClusterv1> > clusterToken;  
+
+  TTree* outNtuple;
+  edm::Service<TFileService> fs;
+
+  uint32_t detId;
+  uint16_t barycenter; 
+  uint16_t width;
+  uint8_t avCharge;
+  edm::EventNumber_t eventN;
+};
+ #endif