EveService.cc

Go to the documentation of this file.

// -*- C++ -*-
//
// Package:     Fireworks/Eve
// Class  :     EveService
//
// Implementation:
//     [Notes on implementation]
//
// Original Author:  Matevz Tadel
//         Created:  Fri Jun 25 18:57:39 CEST 2010
//
 
// system include files
#include <iostream>
 
// user include files
#include "Fireworks/Eve/interface/EveService.h"
 
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/ServiceRegistry/interface/ActivityRegistry.h"
 
#include "TROOT.h"
#include "TSystem.h"
#include "TColor.h"
#include "TStyle.h"
#include "TEnv.h"
 
// To extract coil current from ConditionsDB
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "CondFormats/RunInfo/interface/RunInfo.h"
#include "CondFormats/DataRecord/interface/RunSummaryRcd.h"
 
// To extract coil current from ConditionsInEdm
#include <FWCore/Framework/interface/Run.h>
#include <DataFormats/Common/interface/Handle.h>
#include "DataFormats/Common/interface/ConditionsInEdm.h"
 
#include "TROOT.h"
#include "TSystem.h"
#include "TRint.h"
#include "TEveManager.h"
#include "TEveEventManager.h"
#include "TEveTrackPropagator.h"
 
// GUI widgets
#include "TEveBrowser.h"
#include "TGFrame.h"
#include "TGButton.h"
#include "TGLabel.h"
 
namespace {
  class CmsEveMagField : public TEveMagField {
  private:
    Float_t fField;
    Float_t fFieldMag;
 
  public:
    CmsEveMagField() : TEveMagField(), fField(-3.8), fFieldMag(3.8) {}
    ~CmsEveMagField() override {}
 
    // set current
    void SetFieldByCurrent(Float_t avg_current) {
      fField = -3.8 * avg_current / 18160.0;
      fFieldMag = TMath::Abs(fField);
    }
 
    // get field values
    Float_t GetMaxFieldMag() const override { return fFieldMag; }
 
    TEveVector GetField(Float_t x, Float_t y, Float_t z) const override {
      static const Float_t barrelFac = 1.2 / 3.8;
      static const Float_t endcapFac = 2.0 / 3.8;
 
      const Float_t R = sqrt(x * x + y * y);
      const Float_t absZ = TMath::Abs(z);
 
      //barrel
      if (absZ < 724.0f) {
        //inside solenoid
        if (R < 300.0f)
          return TEveVector(0, 0, fField);
 
        // outside solinoid
        if ((R > 461.0f && R < 490.5f) || (R > 534.5f && R < 597.5f) || (R > 637.0f && R < 700.0f)) {
          return TEveVector(0, 0, -fField * barrelFac);
        }
      } else {
        if ((absZ > 724.0f && absZ < 786.0f) || (absZ > 850.0f && absZ < 910.0f) || (absZ > 975.0f && absZ < 1003.0f)) {
          const Float_t fac = (z >= 0 ? fField : -fField) * endcapFac / R;
          return TEveVector(x * fac, y * fac, 0);
        }
      }
      return TEveVector(0, 0, 0);
    }
  };
}  // namespace
 
//
// constants, enums and typedefs
//
 
//
// static data member definitions
//
 
//==============================================================================
// constructors and destructor
//==============================================================================
 
EveService::EveService(const edm::ParameterSet&, edm::ActivityRegistry& ar)
    : m_EveManager(nullptr),
      m_Rint(nullptr),
      m_MagField(nullptr),
      m_AllowStep(true),
      m_ShowEvent(true),
      m_ContinueButton(nullptr),
      m_StepButton(nullptr),
      m_StepLabel(nullptr) {
  printf("EveService::EveService CTOR\n");
 
  std::cout << " gApplication " << gApplication << std::endl;
  std::cout << " is batch " << gROOT->IsBatch() << std::endl;
  std::cout << " display " << gSystem->Getenv("DISPLAY") << std::endl;
 
  const char* dummyArgvArray[] = {"cmsRun"};
  char** dummyArgv = const_cast<char**>(dummyArgvArray);
  int dummyArgc = 1;
 
  m_Rint = new TRint("App", &dummyArgc, dummyArgv);
  assert(TApplication::GetApplications()->GetSize());
 
  gROOT->SetBatch(kFALSE);
  std::cout << "calling NeedGraphicsLibs()" << std::endl;
  TApplication::NeedGraphicsLibs();
 
  m_EveManager = TEveManager::Create();
 
  m_EveManager->AddEvent(new TEveEventManager("Event", "Event Data"));
 
  m_MagField = new CmsEveMagField();
 
  createEventNavigationGUI();
 
  // ----------------------------------------------------------------
 
  ar.watchPostBeginJob(this, &EveService::postBeginJob);
  ar.watchPostEndJob(this, &EveService::postEndJob);
 
  ar.watchPostGlobalBeginRun(this, &EveService::postGlobalBeginRun);
 
  ar.watchPostEvent(this, &EveService::postEvent);
}
 
EveService::~EveService() {
  printf("EveService::~EveService DTOR\n");
 
  delete m_MagField;
}
 
//==============================================================================
// Service watchers
//==============================================================================
 
void EveService::postBeginJob() {
  printf("EveService::postBeginJob\n");
 
  // Show the GUI ...
  gSystem->ProcessEvents();
}
 
void EveService::postEndJob() {
  printf("EveService::postEndJob\n");
 
  TEveManager::Terminate();
}
 
//------------------------------------------------------------------------------
 
void EveService::postGlobalBeginRun(edm::GlobalContext const&) {
  float current = 18160.0f;
  /*
   try 
   {
      edm::Handle<edm::ConditionsInRunBlock> runCond;
      bool res = iRun.getByLabel("conditionsInEdm", runCond);
      if (res && runCond.isValid())
      {
         printf("Got current from conds in edm %f\n", runCond->BAvgCurrent);
         current = runCond->BAvgCurrent;
      }
      else
      {
         printf("Could not extract run-conditions get-result=%d, is-valid=%d\n", res, runCond.isValid());
 
         edm::ESHandle<RunInfo> sum;
         iSetup.get<RunInfoRcd>().get(sum);
 
         current = sum->m_avg_current;
         printf("Got current from RunInfoRcd %f\n", sum->m_avg_current);
      }
   }
   catch (...) 
   {
   }
   */
  printf("RunInfo not available \n");
  static_cast<CmsEveMagField*>(m_MagField)->SetFieldByCurrent(current);
}
 
//------------------------------------------------------------------------------
 
void EveService::postEvent(edm::StreamContext const&) {
  printf("EveService::postProcessEvent: Starting GUI loop.\n");
 
  m_StepButton->SetEnabled(kFALSE);
  m_ContinueButton->SetEnabled(kFALSE);
  m_StepLabel->SetText("");
 
  if (m_ShowEvent) {
    gEve->Redraw3D();
    m_Rint->Run(kTRUE);
  }
  m_ShowEvent = true;
  m_AllowStep = true;
 
  gEve->GetCurrentEvent()->DestroyElements();
}
 
//------------------------------------------------------------------------------
 
void EveService::display(const std::string& info) {
  // Display whatever was registered so far, wait until user presses
  // the "Step" button.
 
  if (m_AllowStep) {
    m_ContinueButton->SetEnabled(kTRUE);
    m_StepButton->SetEnabled(kTRUE);
    m_StepLabel->SetText(info.c_str());
    gEve->Redraw3D();
    m_Rint->Run(kTRUE);
  }
}
 
//==============================================================================
// Getters for cleints
//==============================================================================
 
TEveManager* EveService::getManager() {
  gEve = m_EveManager;
  return m_EveManager;
}
 
TEveMagField* EveService::getMagField() { return m_MagField; }
void EveService::setupFieldForPropagator(TEveTrackPropagator* prop) { prop->SetMagFieldObj(m_MagField, kFALSE); }
 
//==============================================================================
// Redirectors to gEve
//==============================================================================
 
void EveService::AddElement(TEveElement* el) { m_EveManager->AddElement(el); }
 
void EveService::AddGlobalElement(TEveElement* el) { m_EveManager->AddGlobalElement(el); }
 
//==============================================================================
// GUI Builders and callback slots
//==============================================================================
 
namespace {
  TGTextButton* MkTxtButton(
      TGCompositeFrame* p, const char* txt, Int_t width = 0, Int_t lo = 0, Int_t ro = 0, Int_t to = 0, Int_t bo = 0) {
    // Create a standard button.
    // If width is not zero, the fixed-width flag is set.
 
    TGTextButton* b = new TGTextButton(p, txt);
    if (width > 0) {
      b->SetWidth(width);
      b->ChangeOptions(b->GetOptions() | kFixedWidth);
    }
    p->AddFrame(b, new TGLayoutHints(kLHintsNormal, lo, ro, to, bo));
    return b;
  }
}  // namespace
 
void EveService::createEventNavigationGUI() {
  const TString cls("EveService");
 
  TEveBrowser* browser = gEve->GetBrowser();
  browser->StartEmbedding(TRootBrowser::kBottom);
 
  TGMainFrame* mf = new TGMainFrame(gClient->GetRoot(), 400, 100, kVerticalFrame);
 
  TGHorizontalFrame* f = new TGHorizontalFrame(mf);
  mf->AddFrame(f, new TGLayoutHints(kLHintsExpandX, 0, 0, 2, 2));
 
  MkTxtButton(f, "Exit", 100, 2, 2)->Connect("Clicked()", cls, this, "slotExit()");
 
  MkTxtButton(f, "Next Event", 100, 2, 2)->Connect("Clicked()", cls, this, "slotNextEvent()");
 
  m_ContinueButton = MkTxtButton(f, "Continue", 100, 2, 2);
  m_ContinueButton->Connect("Clicked()", cls, this, "slotContinue()");
 
  m_StepButton = MkTxtButton(f, "Step", 100, 2, 2);
  m_StepButton->Connect("Clicked()", cls, this, "slotStep()");
 
  m_StepLabel = new TGLabel(mf, "");
  m_StepLabel->SetTextJustify(kTextTop | kTextLeft);
  mf->AddFrame(m_StepLabel, new TGLayoutHints(kLHintsNormal | kLHintsExpandX | kLHintsExpandY, 2, 2, 2, 2));
 
  mf->SetCleanup(kDeepCleanup);
  mf->Layout();
  mf->MapSubwindows();
  mf->MapWindow();
 
  browser->StopEmbedding("EventCtrl");
}
 
void EveService::slotExit() {
  gSystem->ExitLoop();
  printf("EveService exiting on user request.\n");
 
  // Throwing exception here is bad because:
  //   a) it does not work when in a "debug step";
  //   b) does not restore terminal state.
  // So we do exit instead for now.
  // throw cms::Exception("UserTerminationRequest");
 
  gSystem->Exit(0);
}
 
void EveService::slotNextEvent() {
  gSystem->ExitLoop();
  m_ShowEvent = false;
  m_AllowStep = false;
}
 
void EveService::slotContinue() {
  gSystem->ExitLoop();
  m_AllowStep = false;
}
 
void EveService::slotStep() { gSystem->ExitLoop(); }