EcalTBMCInfoProducer.cc

Go to the documentation of this file.

/*
 * \file EcalTBMCInfoProducer.cc
 *
 *
 */

#include "DataFormats/Common/interface/Handle.h"
#include "DataFormats/Math/interface/Point3D.h"

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/Framework/interface/stream/EDProducer.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/PluginManager/interface/ModuleDef.h"
#include "FWCore/ServiceRegistry/interface/Service.h"
#include "FWCore/Utilities/interface/Exception.h"
#include "FWCore/Utilities/interface/RandomNumberGenerator.h"

#include "Geometry/EcalTestBeam/interface/EcalTBCrystalMap.h"
#include "SimDataFormats/EcalTestBeam/interface/PEcalTBInfo.h"
#include "SimDataFormats/GeneratorProducts/interface/HepMCProduct.h"

#include "Math/GenVector/Rotation3D.h"
#include <CLHEP/Random/RandFlat.h>

#include <fstream>
#include <iostream>
#include <vector>

class EcalTBMCInfoProducer : public edm::stream::EDProducer<> {
public:
  explicit EcalTBMCInfoProducer(const edm::ParameterSet &ps);

  ~EcalTBMCInfoProducer() override;

  void produce(edm::Event &event, const edm::EventSetup &eventSetup) override;

private:
  double beamEta;
  double beamPhi;
  double beamTheta;

  int crysNumber;

  double beamXoff;
  double beamYoff;

  double partXhodo;
  double partYhodo;

  EcalTBCrystalMap *theTestMap;

  ROOT::Math::Rotation3D *fromCMStoTB;

  edm::EDGetTokenT<edm::HepMCProduct> GenVtxToken;
};

EcalTBMCInfoProducer::EcalTBMCInfoProducer(const edm::ParameterSet &ps) {
  produces<PEcalTBInfo>();

  edm::FileInPath CrystalMapFile = ps.getParameter<edm::FileInPath>("CrystalMapFile");
  GenVtxToken = consumes<edm::HepMCProduct>(edm::InputTag("moduleLabelVtx", "source"));
  double fMinEta = ps.getParameter<double>("MinEta");
  double fMaxEta = ps.getParameter<double>("MaxEta");
  double fMinPhi = ps.getParameter<double>("MinPhi");
  double fMaxPhi = ps.getParameter<double>("MaxPhi");
  beamEta = (fMaxEta + fMinEta) / 2.;
  beamPhi = (fMaxPhi + fMinPhi) / 2.;
  beamTheta = 2.0 * atan(exp(-beamEta));
  beamXoff = ps.getParameter<double>("BeamMeanX");
  beamYoff = ps.getParameter<double>("BeamMeanX");

  std::string fullMapName = CrystalMapFile.fullPath();
  theTestMap = new EcalTBCrystalMap(fullMapName);
  crysNumber = 0;

  double deltaEta = 999.;
  double deltaPhi = 999.;
  for (int cryIndex = 1; cryIndex <= EcalTBCrystalMap::NCRYSTAL; ++cryIndex) {
    double eta = 0;
    double phi = 0.;
    theTestMap->findCrystalAngles(cryIndex, eta, phi);
    if (fabs(beamEta - eta) < deltaEta && fabs(beamPhi - phi) < deltaPhi) {
      deltaEta = fabs(beamEta - eta);
      deltaPhi = fabs(beamPhi - phi);
      crysNumber = cryIndex;
    } else if (fabs(beamEta - eta) < deltaEta && fabs(beamPhi - phi) > deltaPhi) {
      if (fabs(beamPhi - phi) < 0.017) {
        deltaEta = fabs(beamEta - eta);
        deltaPhi = fabs(beamPhi - phi);
        crysNumber = cryIndex;
      }
    } else if (fabs(beamEta - eta) > deltaEta && fabs(beamPhi - phi) < deltaPhi) {
      if (fabs(beamEta - eta) < 0.017) {
        deltaEta = fabs(beamEta - eta);
        deltaPhi = fabs(beamPhi - phi);
        crysNumber = cryIndex;
      }
    }
  }

  edm::LogVerbatim("EcalTBInfo") << "Initialize TB MC ECAL info producer with parameters: \n"
                                 << "Crystal map file:  " << CrystalMapFile << "\n"
                                 << "Beam average eta = " << beamEta << "\n"
                                 << "Beam average phi = " << beamPhi << "\n"
                                 << "Corresponding to crystal number = " << crysNumber << "\n"
                                 << "Beam X offset =    " << beamXoff << "\n"
                                 << "Beam Y offset =    " << beamYoff;

  // rotation matrix to move from the CMS reference frame to the test beam one

  double xx = -cos(beamTheta) * cos(beamPhi);
  double xy = -cos(beamTheta) * sin(beamPhi);
  double xz = sin(beamTheta);

  double yx = sin(beamPhi);
  double yy = -cos(beamPhi);
  double yz = 0.;

  double zx = sin(beamTheta) * cos(beamPhi);
  double zy = sin(beamTheta) * sin(beamPhi);
  double zz = cos(beamTheta);

  fromCMStoTB = new ROOT::Math::Rotation3D(xx, xy, xz, yx, yy, yz, zx, zy, zz);

  // random number
  edm::Service<edm::RandomNumberGenerator> rng;
  if (!rng.isAvailable()) {
    throw cms::Exception("Configuration") << "EcalTBMCInfoProducer requires the RandomNumberGeneratorService\n"
                                             "which is not present in the configuration file.  You must add the "
                                             "service\n"
                                             "in the configuration file or remove the modules that require it.";
  }
}

EcalTBMCInfoProducer::~EcalTBMCInfoProducer() { delete theTestMap; }

void EcalTBMCInfoProducer::produce(edm::Event &event, const edm::EventSetup &eventSetup) {
  edm::Service<edm::RandomNumberGenerator> rng;
  CLHEP::HepRandomEngine *engine = &rng->getEngine(event.streamID());

  std::unique_ptr<PEcalTBInfo> product(new PEcalTBInfo());

  // Fill the run information

  product->setCrystal(crysNumber);

  product->setBeamDirection(beamEta, beamPhi);
  product->setBeamOffset(beamXoff, beamYoff);

  // Compute the event x,y vertex coordinates in the beam reference system
  // e.g. in the place orthogonal to the beam average direction

  partXhodo = partYhodo = 0.;

  edm::Handle<edm::HepMCProduct> GenEvt;
  event.getByToken(GenVtxToken, GenEvt);

  const HepMC::GenEvent *Evt = GenEvt->GetEvent();
  HepMC::GenEvent::vertex_const_iterator Vtx = Evt->vertices_begin();

  math::XYZPoint eventCMSVertex((*Vtx)->position().x(), (*Vtx)->position().y(), (*Vtx)->position().z());

  LogDebug("EcalTBInfo") << "Generated vertex position = " << eventCMSVertex.x() << " " << eventCMSVertex.y() << " "
                         << eventCMSVertex.z();

  math::XYZPoint eventTBVertex = (*fromCMStoTB) * eventCMSVertex;

  LogDebug("EcalTBInfo") << "Rotated vertex position   = " << eventTBVertex.x() << " " << eventTBVertex.y() << " "
                         << eventTBVertex.z();

  partXhodo = eventTBVertex.x();
  partYhodo = eventTBVertex.y();

  product->setBeamPosition(partXhodo, partYhodo);

  // Asynchronous phase shift
  double thisPhaseShift = CLHEP::RandFlat::shoot(engine);

  product->setPhaseShift(thisPhaseShift);
  LogDebug("EcalTBInfo") << "Asynchronous Phaseshift = " << thisPhaseShift;

  // store the object in the framework event

  event.put(std::move(product));
}

DEFINE_FWK_MODULE(EcalTBMCInfoProducer);