FWHGCalMultiClusterProxyBuilder.cc

Go to the documentation of this file.

#include "TEveBoxSet.h"
#include "Fireworks/Core/interface/FWSimpleProxyBuilderTemplate.h"
#include "Fireworks/Core/interface/FWEventItem.h"
#include "Fireworks/Core/interface/FWGeometry.h"
#include "Fireworks/Core/interface/BuilderUtils.h"
#include "DataFormats/ParticleFlowReco/interface/HGCalMultiCluster.h"

class FWHGCalMultiClusterProxyBuilder : public FWSimpleProxyBuilderTemplate<reco::HGCalMultiCluster>
{
public:
   FWHGCalMultiClusterProxyBuilder( void ) {}
   ~FWHGCalMultiClusterProxyBuilder( void ) override {}

   REGISTER_PROXYBUILDER_METHODS();

private:
   FWHGCalMultiClusterProxyBuilder( const FWHGCalMultiClusterProxyBuilder& ) = delete;          // stop default
   const FWHGCalMultiClusterProxyBuilder& operator=( const FWHGCalMultiClusterProxyBuilder& ) = delete;     // stop default

   using FWSimpleProxyBuilderTemplate<reco::HGCalMultiCluster>::build;
   void build( const reco::HGCalMultiCluster& iData, unsigned int iIndex, TEveElement& oItemHolder, const FWViewContext* ) override;
};

void
FWHGCalMultiClusterProxyBuilder::build( const reco::HGCalMultiCluster& iData, unsigned int iIndex, TEveElement& oItemHolder, const FWViewContext* )
{
  const auto & clusters = iData.clusters();

  TEveBoxSet* boxset = new TEveBoxSet();
  boxset->Reset(TEveBoxSet::kBT_FreeBox, true, 64);
  boxset->UseSingleColor();
  boxset->SetPickable(true);
  for (const auto & c : clusters)
    {
      std::vector<std::pair<DetId, float> > clusterDetIds = c->hitsAndFractions();


      for( std::vector<std::pair<DetId, float> >::iterator it = clusterDetIds.begin(), itEnd = clusterDetIds.end();
       it != itEnd; ++it )
    {
      const float* corners = item()->getGeom()->getCorners( it->first );
      const float* parameters = item()->getGeom()->getParameters( it->first );
      const float* shapes = item()->getGeom()->getShapePars(it->first);

      if( corners == nullptr || parameters == nullptr || shapes == nullptr ) {
         continue;
      }

#if 0
      const int total_points = parameters[0];
      const int total_vertices = 3*total_points;
#else // using broken boxes(half hexagon) until there's support for hexagons in TEveBoxSet
      const int total_points = 4;
      const int total_vertices = 3*total_points;

      const float thickness = shapes[3];

      std::vector<float> pnts(24);
      for(int i = 0; i < total_points; ++i){
         pnts[i*3+0] = corners[i*3];
         pnts[i*3+1] = corners[i*3+1];
         pnts[i*3+2] = corners[i*3+2];

         pnts[(i*3+0)+total_vertices] = corners[i*3];
         pnts[(i*3+1)+total_vertices] = corners[i*3+1];
         pnts[(i*3+2)+total_vertices] = corners[i*3+2]+thickness;
      }
      boxset->AddBox( &pnts[0]);
#endif
    }
    }
   boxset->RefitPlex();
   setupAddElement(boxset, &oItemHolder);
}

REGISTER_FWPROXYBUILDER( FWHGCalMultiClusterProxyBuilder, reco::HGCalMultiCluster, "HGCal MultiCluster", FWViewType::kISpyBit );