FWECALDetailViewBuilder.cc

Go to the documentation of this file.

// FIXME - needed to set fixed eta-phi limits. Without the
//         visible area may change widely depending on energy
//         deposition availability

#include "TEveCaloData.h"
#include "TEveViewer.h"
#include "TEvePointSet.h"
#include "TEveCalo.h"
#include "TEveCompound.h"
#include "TAxis.h"
#include "TMath.h"
#include "THLimitsFinder.h"
#include "TLatex.h"

#include "Fireworks/Core/interface/FWDetailViewBase.h"
#include "Fireworks/Calo/interface/FWECALDetailViewBuilder.h"
#include "Fireworks/Calo/interface/FWBoxRecHit.h"
#include "Fireworks/Core/interface/FWGeometry.h"
#include "Fireworks/Core/interface/fw3dlego_xbins.h"
#include "Fireworks/Core/interface/fwLog.h"
#include "Fireworks/Core/interface/Context.h"


#include "DataFormats/FWLite/interface/Handle.h"
#include "DataFormats/EgammaReco/interface/SuperCluster.h"
#include "DataFormats/EcalDetId/interface/EcalSubdetector.h"

#include "TGeoMatrix.h"
#include "TEveTrans.h"

#include <utility>


FWECALDetailViewBuilder::FWECALDetailViewBuilder(const edm::EventBase *event, const FWGeometry* geom, 
                                                 float eta, float phi, int size , Color_t defaultColor)
                                                
: m_event(event), m_geom(geom),
  m_eta(eta), m_phi(phi), m_size(size),
  m_defaultColor(defaultColor), m_towerList(nullptr)
{
}


TEveCaloData* FWECALDetailViewBuilder::buildCaloData(bool)
{
   // get the hits from the event

   // data
   TEveCaloDataVec* data = new TEveCaloDataVec( 1);
   data->SetWrapTwoPi(false);
   data->RefSliceInfo(0).Setup("hits (not clustered)", 0.0, m_defaultColor );
   
   fillData(data);

   // axis
   float etaMin = m_eta - sizeRad();
   float etaMax = m_eta + sizeRad();
   float phiMin = m_phi - sizeRad();
   float phiMax = m_phi + sizeRad();

   data->AddTower(m_eta - sizeRad(), m_eta + sizeRad(), m_phi - sizeRad(), m_phi + sizeRad());

   data->FillSlice(0, 0.1);      
   

 
   TAxis* eta_axis = nullptr;
   TAxis* phi_axis = nullptr; 

   //  printf("data rng %f %f %f %f\n",etaMin, etaMax, phiMin, phiMax );  
   std::vector<double> etaBinsWithinLimits;
   etaBinsWithinLimits.push_back(etaMin);
   for (unsigned int i=0; i<83; ++i)
       if ( fw3dlego::xbins[i] > etaMin && fw3dlego::xbins[i] < etaMax )
           etaBinsWithinLimits.push_back(fw3dlego::xbins[i]);
   etaBinsWithinLimits.push_back(etaMax);

   std::vector<double> phiBinsWithinLimits;
   phiBinsWithinLimits.push_back(phiMin);
   for ( double phi = -M_PI; phi < M_PI; phi += M_PI/36 )
       if ( phi > phiMin && phi < phiMax )
           phiBinsWithinLimits.push_back(phi);
   phiBinsWithinLimits.push_back(phiMax);

   eta_axis = new TAxis((int)etaBinsWithinLimits.size() -1, &etaBinsWithinLimits[0]);
   phi_axis = new TAxis((int)phiBinsWithinLimits.size() -1, &phiBinsWithinLimits[0]);

   eta_axis->SetTitleFont(122);
   eta_axis->SetTitle("h");
   eta_axis->SetTitleSize(0.07);
   phi_axis->SetTitleFont(122);
   phi_axis->SetTitle("f");
   phi_axis->SetTitleSize(0.07);

   eta_axis->SetNdivisions(510);
   phi_axis->SetNdivisions(510);
   data->SetEtaBins(eta_axis);
   data->SetPhiBins(phi_axis);
   return data;
}

//_______________________________________________________________   
TEveCaloLego* FWECALDetailViewBuilder::build()
{

   // axis
   float etaMin = m_eta - sizeRad();
   float etaMax = m_eta + sizeRad();
   float phiMin = m_phi - sizeRad();
   float phiMax = m_phi + sizeRad();
   
   m_towerList = new TEveElementList("TowerHolder");
   TEveCaloData* data = buildCaloData(true);
      
   // lego
   TEveCaloLego *lego = new TEveCaloLego();
   lego->SetData(data);
   lego->AddElement(m_towerList);
   lego->SetAutoRange(false);
   lego->SetDrawNumberCellPixels(100);
   // scale and translate to real world coordinates
   lego->SetEta(etaMin, etaMax);
   lego->SetPhiWithRng((phiMin+phiMax)*0.5, (phiMax-phiMin)*0.5); // phi range = 2* phiOffset
   Double_t legoScale = sizeRad() *2;
   lego->InitMainTrans();
   lego->RefMainTrans().SetScale(legoScale, legoScale, legoScale*0.5);
   lego->RefMainTrans().SetPos(m_eta, m_phi, -0.01);
   lego->SetAutoRebin(kFALSE);
   lego->SetName("ECALDetail Lego");

   // cut & paste from FWLegoViewBase
   lego->SetScaleAbs(true);
   lego->SetHasFixedHeightIn2DMode(true);
   lego->SetFixedHeightValIn2DMode(0.001);

   
   TEvePointSet* ps = new TEvePointSet("origin");
   ps->SetNextPoint(m_eta, m_phi, 0.01);
   ps->SetMarkerSize(0.05);
   ps->SetMarkerStyle(2);
   ps->SetMainColor(kGreen);
   ps->SetMarkerColor(kGreen);
   lego->AddElement(ps);

   return lego;

}

void FWECALDetailViewBuilder::setColor(Color_t color, const std::vector<DetId> &detIds)
{
   for (size_t i = 0; i < detIds.size(); ++i)
      m_detIdsToColor[detIds[i]] = color;
}

void
FWECALDetailViewBuilder::showSuperCluster( const reco::SuperCluster &cluster, Color_t color )
{
   std::vector<DetId> clusterDetIds;
   const std::vector<std::pair<DetId, float> > &hitsAndFractions = cluster.hitsAndFractions();
   for (size_t j = 0; j < hitsAndFractions.size(); ++j)
   {
      clusterDetIds.push_back(hitsAndFractions[j].first);
   }

   setColor( color, clusterDetIds );
}

void
FWECALDetailViewBuilder::showSuperClusters( Color_t color1, Color_t color2 )
{
   // get the superclusters from the event
   edm::Handle<reco::SuperClusterCollection> collection;

   if( fabs( m_eta ) < 1.5 ) {
      try {
         m_event->getByLabel(edm::InputTag("correctedHybridSuperClusters"), collection);
      }
      catch (...)
      {
         fwLog(fwlog::kWarning) <<"no barrel superclusters are available" << std::endl;
      }
   } else {
      try {
         m_event->getByLabel(edm::InputTag("correctedMulti5x5SuperClustersWithPreshower"), collection);
      }
      catch (...)
      {
         fwLog(fwlog::kWarning) <<"no endcap superclusters are available" << std::endl;
      }
   }
   if( collection.isValid() )
   {
      unsigned int colorIndex = 0;
      // sort clusters in eta so neighboring clusters have distinct colors
      reco::SuperClusterCollection sorted = *collection.product();
      std::sort( sorted.begin(), sorted.end(), superClusterEtaLess );
      for( size_t i = 0; i < sorted.size(); ++i )
      {
     if( !(fabs(sorted[i].eta() - m_eta) < sizeRad()
           && fabs(sorted[i].phi() - m_phi) < sizeRad()) )
       continue;

     if( colorIndex %2 == 0 )
        showSuperCluster( sorted[i], color1 );
     else
        showSuperCluster( sorted[i], color2 );
     ++colorIndex;
      }
   }
}


namespace {
float
calculateEt( const TEveVector &centre, float e )
{
   TEveVector vec = centre;
   float et;

   vec.Normalize();
   vec *= e;
   et = vec.Perp();

   return et;
}

}
//------------------------------------------------------------------
void
FWECALDetailViewBuilder::fillEtaPhi( const EcalRecHitCollection *hits,TEveCaloDataVec *data)
{
    // printf("filletaphi \n");  
   const float area = sizeRad(); // barrel cell range, AMT this is available in context

   double eta1 = m_eta - area;
   double eta2 = m_eta + area;
   double phi1 = m_phi - area;
   double phi2 = m_phi + area;


   std::vector<FWBoxRecHit*>  boxes;
   for( EcalRecHitCollection::const_iterator hitIt = hits->begin(); hitIt != hits->end(); ++hitIt)
   {    
       const float *corners = m_geom->getCorners( hitIt->detid() );
       float energy, et;
       std::vector<TEveVector> etaphiCorners(8);

       if( corners == nullptr )
           continue;


       for( int i = 0; i < 4; ++i )
       {
           TEveVector cv = TEveVector( corners[i*3], corners[i*3+1], corners[i*3+2] );
           etaphiCorners[i].fX = cv.Eta();                                     // Conversion of rechit X/Y values for plotting in Eta/Phi
           etaphiCorners[i].fY = cv.Phi();
           etaphiCorners[i].fZ = 0.0;

           etaphiCorners[i+4].fX = etaphiCorners[i].fX;                        // Top can simply be plotted exactly over the top of the bottom face
           etaphiCorners[i+4].fY = etaphiCorners[i].fY;
           etaphiCorners[i+4].fZ = 0.001;
           //  printf("%f %f %d \n",  etaphiCorners[i].fX, etaphiCorners[i].fY, i);
       }

       TEveVector center;
       for( int i = 0; i < 4; ++i )
           center += etaphiCorners[i];
       center *= 1.f / 4.f;

       
       if ( center.fX < eta1 || center.fX > eta2) continue;
       if ( center.fY < phi1 || center.fY > phi2) continue;



       // Stop phi wrap
       float dPhi1 = etaphiCorners[2].fY - etaphiCorners[1].fY;
       float dPhi2 = etaphiCorners[3].fY - etaphiCorners[0].fY;
       float dPhi3 = etaphiCorners[1].fY - etaphiCorners[2].fY;
       float dPhi4 = etaphiCorners[0].fY - etaphiCorners[3].fY;

       if( dPhi1 > 1 )
           etaphiCorners[2].fY = etaphiCorners[2].fY - ( 2 * TMath::Pi() );
       if( dPhi2 > 1 )
           etaphiCorners[3].fY = etaphiCorners[3].fY - ( 2 * TMath::Pi() );
       if( dPhi3 > 1 )
           etaphiCorners[2].fY = etaphiCorners[2].fY + ( 2 * TMath::Pi() );
       if( dPhi4 > 1 )
           etaphiCorners[3].fY = etaphiCorners[3].fY + ( 2 * TMath::Pi() );



       energy = hitIt->energy();
       et = calculateEt( center, energy );
       Color_t bcolor = m_defaultColor;
       std::map<DetId, int>::const_iterator itr = m_detIdsToColor.find(hitIt->id());
       if (itr != m_detIdsToColor.end()) bcolor = itr->second;

       m_boxes.push_back(new FWBoxRecHit( etaphiCorners, m_towerList, energy, et ));
       TEveElement::List_i pIt = m_boxes.back()->getTower()->BeginParents();
       TEveCompound* comp = dynamic_cast<TEveCompound*>(*pIt);
       comp->SetMainColor(bcolor);
       m_boxes.back()->getTower()->SetPickable(true);
       m_boxes.back()->getTower()->SetElementTitle(Form("rawId = %d, et = %f", hitIt->id().rawId(), et));
   } // loop hits

}


//---------------------------------------------------------------------------------------


void
FWECALDetailViewBuilder::fillData(  TEveCaloDataVec *data)
{
   { // barrel
      const EcalRecHitCollection *hitsEB = nullptr;
      edm::Handle<EcalRecHitCollection> handle_hitsEB;

      // RECO
      try
      {
         edm::InputTag tag("ecalRecHit", "EcalRecHitsEB");
         m_event->getByLabel(tag, handle_hitsEB);
     if (handle_hitsEB.isValid())
         {
        hitsEB = &*handle_hitsEB;
         }
      }
      catch (...)
      {
         fwLog(fwlog::kWarning) <<"FWECALDetailViewBuilder::fillData():: Failed to access EcalRecHitsEB collection." << std::endl;
      }


      // AOD
      if ( ! handle_hitsEB.isValid()) {
         try{
            edm::InputTag tag("reducedEcalRecHitsEB");
            m_event->getByLabel(tag, handle_hitsEB);
            if (handle_hitsEB.isValid())
            {
               hitsEB = &*handle_hitsEB;
            }

         }
         catch (...)
         {
            fwLog(fwlog::kWarning) <<"FWECALDetailViewBuilder::filData():: Failed to access reducedEcalRecHitsEB collection." << std::endl;
         }
      }

      // MINIAOD
      if ( ! handle_hitsEB.isValid()) {
         try{
            edm::InputTag tag("reducedEgamma", "reducedEBRecHits");
            m_event->getByLabel(tag, handle_hitsEB);
            if (handle_hitsEB.isValid())
            {
               hitsEB = &*handle_hitsEB;
            }




         }
         catch (...)
         {
            fwLog(fwlog::kWarning) <<"FWECALDetailViewBuilder::filData():: Failed to access reducedEgamma collection." << std::endl;
         }
      }

      if( handle_hitsEB.isValid() ) 
      {
         fillEtaPhi( hitsEB, data);
      }
   }

   {// endcap

      const EcalRecHitCollection *hitsEE = nullptr;
      edm::Handle<EcalRecHitCollection> handle_hitsEE;

      // RECO
      try
      {
         edm::InputTag tag("ecalRecHit", "EcalRecHitsEE");
         m_event->getByLabel(tag, handle_hitsEE);
     if (handle_hitsEE.isValid())
        hitsEE = &*handle_hitsEE;
      }
      catch (...)
      {
         fwLog(fwlog::kWarning) <<"FWECALDetailViewBuilder::fillData():: Failed to access ecalRecHitsEE collection." << std::endl;
      }

      // AOD
      if ( ! handle_hitsEE.isValid()) {
         try {
            edm::InputTag tag("reducedEcalRecHitsEE");
            m_event->getByLabel(tag, handle_hitsEE);
            if (handle_hitsEE.isValid())
            {
               hitsEE = &*handle_hitsEE;
            }

         }
         catch (...)
         {     
            fwLog(fwlog::kWarning) <<"FWECALDetailViewBuilder::fillData():: Failed to access reducedEcalRecHitsEE collection." << std::endl;
         }

      // MINIAOD
      if ( ! handle_hitsEE.isValid()) {
         try {
            edm::InputTag tag("reducedEgamma", "reducedEERecHits");
            m_event->getByLabel(tag, handle_hitsEE);
            if (handle_hitsEE.isValid())
            {
               hitsEE = &*handle_hitsEE;
            }

         }
         catch (...)
         {     
            fwLog(fwlog::kWarning) <<"FWECALDetailViewBuilder::fillData():: Failed to access reducedEcalRecHitsEE collection." << std::endl;
         }
      }
   
      }

      if( handle_hitsEE.isValid() ) 
      {
          fillEtaPhi( hitsEE, data);
      }
   }

   if ( m_boxes.empty()) return;

   bool plotEt = true;
   float maxEnergy = 0;
   int maxEnergyIdx = 0;
   // get max energy in EE and EB

   int cnt = 0;
   for (auto & i : m_boxes)  {
       if (i->getEnergy(plotEt) > maxEnergy) {
           maxEnergy = i->getEnergy(plotEt);
           maxEnergyIdx = cnt;
       }
       cnt++;
   }

   m_boxes[maxEnergyIdx]->setIsTallest();

   // AMT ... max size can be an external parameter
   float scale = 0.3/maxEnergy;
   for (auto & i : m_boxes) {
        i->updateScale(scale, log(maxEnergy + 1), plotEt);
        i->getTower()->SetDrawFrame(true);
   }
   data->DataChanged();
}

double
FWECALDetailViewBuilder::makeLegend( double x0, double y0,
                                    Color_t clustered1, Color_t clustered2,
                                    Color_t supercluster
                                    )
{
   Double_t fontsize = 0.07;
   TLatex* latex = new TLatex();
   Double_t x = x0;
   Double_t y = y0;
   Double_t boxH = 0.25*fontsize;
   Double_t yStep = 0.04;
   
   y -= yStep;
   latex->DrawLatex(x, y, "Energy types:");
   y -= yStep;
   
   Double_t pos[4];
   pos[0] = x+0.05;
   pos[2] = x+0.20;
   
   pos[1] = y; pos[3] = pos[1] + boxH;
   FWDetailViewBase::drawCanvasBox(pos, m_defaultColor);
   latex->DrawLatex(x+0.25, y, "unclustered");
   y -= yStep;
   if (clustered1<0) return y;
   
   pos[1] = y; pos[3] = pos[1] + boxH;
   FWDetailViewBase::drawCanvasBox(pos, clustered1);
   latex->DrawLatex(x+0.25, y, "clustered");
   y -= yStep;
   if (clustered2<0) return y;
   
   pos[1] = y; pos[3] = pos[1] + boxH;
   FWDetailViewBase::drawCanvasBox(pos, clustered2);
   latex->DrawLatex(x+0.25, y, "clustered");
   y -= yStep;
   if (supercluster<0) return y;
   
   pos[1] = y; pos[3] = pos[1] + boxH;
   FWDetailViewBase::drawCanvasBox(pos, supercluster);
   latex->DrawLatex(x+0.25, y, "super-cluster");
   y -= yStep;
   
   return y;
}
//______________________________________________________________________________

float FWECALDetailViewBuilder::sizeRad() const
{
   float rs = m_size * TMath::DegToRad();
   return rs;
}