FWSiStripDigiProxyBuilder

Line Code
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
// -*- C++ -*-
//
// Package:     Tracks
// Class  :     FWSiStripDigiProxyBuilder
//
//
// Original Author:
//         Created:  Thu Dec  6 18:01:21 PST 2007
//

#include "TEveStraightLineSet.h"

#include "Fireworks/Core/interface/FWProxyBuilderBase.h"
#include "Fireworks/Core/interface/FWEventItem.h"
#include "Fireworks/Core/interface/FWGeometry.h"
#include "Fireworks/Core/interface/fwLog.h"
#include "Fireworks/Tracks/interface/TrackUtils.h"

#include "DataFormats/SiStripDigi/interface/SiStripDigi.h"
#include "DataFormats/Common/interface/DetSetVector.h"
#include "DataFormats/DetId/interface/DetId.h"

class FWSiStripDigiProxyBuilder : public FWProxyBuilderBase {
public:
  FWSiStripDigiProxyBuilder(void) {}
  ~FWSiStripDigiProxyBuilder(void) override {}

  REGISTER_PROXYBUILDER_METHODS();

private:
  using FWProxyBuilderBase::build;
  void build(const FWEventItem* iItem, TEveElementList* product, const FWViewContext*) override;

public:
  FWSiStripDigiProxyBuilder(const FWSiStripDigiProxyBuilder&) = delete;
  const FWSiStripDigiProxyBuilder& operator=(const FWSiStripDigiProxyBuilder&) = delete;
};

void FWSiStripDigiProxyBuilder::build(const FWEventItem* iItem, TEveElementList* product, const FWViewContext*) {
  const edm::DetSetVector<SiStripDigi>* digis = nullptr;

  iItem->get(digis);

  if (!digis) {
    return;
  }
  const FWGeometry* geom = iItem->getGeom();

  for (edm::DetSetVector<SiStripDigi>::const_iterator it = digis->begin(), end = digis->end(); it != end; ++it) {
    edm::DetSet<SiStripDigi> ds = *it;
    const uint32_t& id = ds.id;

    const float* pars = geom->getParameters(id);

    for (edm::DetSet<SiStripDigi>::const_iterator idigi = ds.data.begin(), idigiEnd = ds.data.end(); idigi != idigiEnd;
         ++idigi) {
      TEveStraightLineSet* lineSet = new TEveStraightLineSet;
      setupAddElement(lineSet, product);

      if (!geom->contains(id)) {
        fwLog(fwlog::kWarning) << "failed get geometry and topology of SiStripDigi with detid: " << id << std::endl;
        continue;
      }
      float localTop[3] = {0.0, 0.0, 0.0};
      float localBottom[3] = {0.0, 0.0, 0.0};

      fireworks::localSiStrip((*idigi).strip(), localTop, localBottom, pars, id);

      float globalTop[3];
      float globalBottom[3];
      geom->localToGlobal(id, localTop, globalTop, localBottom, globalBottom);

      lineSet->AddLine(globalTop[0], globalTop[1], globalTop[2], globalBottom[0], globalBottom[1], globalBottom[2]);
    }  // end of iteration over digis
  }  // end of iteration over the DetSetVector
}

REGISTER_FWPROXYBUILDER(FWSiStripDigiProxyBuilder,
                        edm::DetSetVector<SiStripDigi>,
                        "SiStripDigi",
                        FWViewType::kAll3DBits | FWViewType::kAllRPZBits);