Fireworks/Electrons/src/makeSuperCluster.cc - master - githubmirror/cmssw

Fireworks/Electrons/src/makeSuperCluster.cc
116 строк4 KB
Mircho Rodozov
Add header after ROOT master clean
19 июн 2020, 14:07
19 июн 2020, 14:072893ee2
// -*- C++ -*-
//
// Package:     Electrons
// Class  :     makeSuperCluster
//
// Implementation:
//     <Notes on implementation>
//
// Original Author:  Chris Jones
//         Created:  Fri Dec  5 15:32:33 EST 2008
//

// system include files
#include "TEveGeoNode.h"
#include "TEveGeoShape.h"
#include "TGeoBBox.h"
#include "TGeoTube.h"

// user include files
#include "Fireworks/Electrons/interface/makeSuperCluster.h"

#include "Fireworks/Core/interface/BuilderUtils.h"
#include "Fireworks/Core/interface/FWEventItem.h"
#include "Fireworks/Core/interface/FWGeometry.h"
#include "Fireworks/Core/interface/Context.h"
#include "Fireworks/Core/interface/FWProxyBuilderBase.h"

namespace fireworks {
  bool makeRhoPhiSuperCluster(FWProxyBuilderBase* pb,
                              const reco::SuperClusterRef& iCluster,
                              float iPhi,
                              TEveElement& oItemHolder) {
    if (!iCluster.isAvailable())
      return false;
    TEveGeoManagerHolder gmgr(TEveGeoShape::GetGeoMangeur());

    std::vector<std::pair<DetId, float> > detids = iCluster->hitsAndFractions();
    if (detids.empty())
      return false;
    std::vector<double> phis;
    for (std::vector<std::pair<DetId, float> >::const_iterator id = detids.begin(), end = detids.end(); id != end;
         ++id) {
      const float* corners = pb->context().getGeom()->getCorners(id->first.rawId());
      if (corners != nullptr) {
        std::vector<float> centre(3, 0);

        for (unsigned int i = 0; i < 24; i += 3) {
          centre[0] += corners[i];
          centre[1] += corners[i + 1];
          centre[2] += corners[i + 2];
        }

        phis.push_back(TEveVector(centre[0], centre[1], centre[2]).Phi());
      }
    }
    std::pair<double, double> phiRange = fireworks::getPhiRange(phis, iPhi);
    const double r = pb->context().caloR1();
    TGeoBBox* sc_box = new TGeoTubeSeg(r - 2,
                                       r,
                                       1,
                                       phiRange.first * 180 / M_PI - 0.5,
                                       phiRange.second * 180 / M_PI + 0.5);  // 0.5 is roughly half size of a crystal
    TEveGeoShape* sc = fireworks::getShape("supercluster", sc_box, pb->item()->defaultDisplayProperties().color());
    sc->SetPickable(kTRUE);
    pb->setupAddElement(sc, &oItemHolder);
    return true;
  }

  bool makeRhoZSuperCluster(FWProxyBuilderBase* pb,
                            const reco::SuperClusterRef& iCluster,
                            float iPhi,
                            TEveElement& oItemHolder) {
    if (!iCluster.isAvailable())
      return false;
    TEveGeoManagerHolder gmgr(TEveGeoShape::GetGeoMangeur());
    double theta_max = 0;
    double theta_min = 10;
    std::vector<std::pair<DetId, float> > detids = iCluster->hitsAndFractions();
    if (detids.empty())
      return false;
    for (std::vector<std::pair<DetId, float> >::const_iterator id = detids.begin(), end = detids.end(); id != end;
         ++id) {
      const float* corners = pb->context().getGeom()->getCorners(id->first.rawId());
      if (corners != nullptr) {
        std::vector<float> centre(3, 0);

        for (unsigned int i = 0; i < 24; i += 3) {
          centre[0] += corners[i];
          centre[1] += corners[i + 1];
          centre[2] += corners[i + 2];
        }

        double theta = TEveVector(centre[0], centre[1], centre[2]).Theta();
        if (theta > theta_max)
          theta_max = theta;
        if (theta < theta_min)
          theta_min = theta;
      }
    }
    // expand theta range by the size of a crystal to avoid segments of zero length
    bool barrel = true;
    if ((theta_max > 0 && theta_max < pb->context().caloTransAngle()) ||
        (theta_min > (TMath::Pi() - pb->context().caloTransAngle()))) {
      barrel = false;
    }

    double z_ecal = barrel ? pb->context().caloZ1() : pb->context().caloZ2();
    double r_ecal = barrel ? pb->context().caloR1() : pb->context().caloR2();

    fireworks::addRhoZEnergyProjection(
        pb, &oItemHolder, r_ecal - 1, z_ecal - 1, theta_min - 0.003, theta_max + 0.003, iPhi);

    return true;
  }

}  // namespace fireworks