FWPCaloHitProxyBuilder

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
/*
 *  FWPCaloHitProxyBuilder.cc
 *  FWorks
 *
 *  Created by Ianna Osborne on 9/9/10.
 *
 */

#include "Fireworks/Core/interface/FWDigitSetProxyBuilder.h"
#include "Fireworks/Core/interface/FWGeometry.h"
#include "Fireworks/Core/interface/FWEventItem.h"
#include "Fireworks/Core/interface/BuilderUtils.h"
#include "DataFormats/EcalDetId/interface/EcalSubdetector.h"
#include "SimDataFormats/CaloHit/interface/PCaloHit.h"
#include "SimDataFormats/CaloHit/interface/PCaloHitContainer.h"

class FWPCaloHitProxyBuilder : public FWDigitSetProxyBuilder {
public:
  FWPCaloHitProxyBuilder(void) {}
  ~FWPCaloHitProxyBuilder(void) override {}

  REGISTER_PROXYBUILDER_METHODS();

  FWPCaloHitProxyBuilder(const FWPCaloHitProxyBuilder&) = delete;
  const FWPCaloHitProxyBuilder& operator=(const FWPCaloHitProxyBuilder&) = delete;

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

void FWPCaloHitProxyBuilder::build(const FWEventItem* iItem, TEveElementList* product, const FWViewContext*) {
  const edm::PCaloHitContainer* collection = nullptr;
  iItem->get(collection);
  if (!collection)
    return;

  TEveBoxSet* boxSet = addBoxSetToProduct(product);
  int index = 0;
  for (std::vector<PCaloHit>::const_iterator it = collection->begin(); it != collection->end(); ++it) {
    const float* corners = item()->getGeom()->getCorners((*it).id());

    std::vector<float> scaledCorners(24);
    if (corners)
      fireworks::energyTower3DCorners(corners, (*it).energy() * 10, scaledCorners);

    addBox(boxSet, &scaledCorners[0], iItem->modelInfo(index++).displayProperties());
  }
}

// !AMT TEveBoxSet is not projectable. Can't be added to RPZ views empty.
REGISTER_FWPROXYBUILDER(FWPCaloHitProxyBuilder,
                        edm::PCaloHitContainer,
                        "PCaloHits",
                        FWViewType::kAll3DBits | FWViewType::kAllRPZBits);