EveDisplayPlugin.cc

Go to the documentation of this file.

// -*- C++ -*-
//
// Package:    EveDisplayPlugin
// Class:      EveDisplayPlugin
// 
//
// Original Author:  Chris D Jones
//         Created:  Wed Sep 26 08:27:23 EDT 2007
//
//
#define private public // workaround for bug in 5.34.18
#include "TROOT.h"
#include "TSystem.h"
#include "TColor.h"
#include "TStyle.h"
#include "TEnv.h"

// system include files
#include <memory>
#include <iostream>
#include <sstream>

// user include files
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "Fireworks/Geometry/interface/DisplayPlugin.h"
#include "Fireworks/Geometry/interface/DisplayPluginFactory.h"

#include "Fireworks/Geometry/interface/DisplayGeomRecord.h"

#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/EventSetup.h"

#include "TGeoManager.h"
#include "TEveManager.h"
#include "TEveGeoNode.h"

//
// class decleration
//

class EveDisplayPlugin : public fireworks::geometry::DisplayPlugin {

   public:
      explicit EveDisplayPlugin();
      ~EveDisplayPlugin();


   private:
      virtual void run(const edm::EventSetup&) override;

};

//
// constants, enums and typedefs
//

//
// static data member definitions
//

//
// constructors and destructor
//
EveDisplayPlugin::EveDisplayPlugin()
{
   //now do what ever initialization is needed

}


EveDisplayPlugin::~EveDisplayPlugin()
{
 
   // do anything here that needs to be done at desctruction time
   // (e.g. close files, deallocate resources etc.)

}


//
// member functions
//
// ------------ method called to for each event  ------------
void
EveDisplayPlugin::run(const edm::EventSetup& iSetup)
{
  std::cout << "In the EveDisplayPlugin::analyze method..." << std::endl;
   using namespace edm;

   ESHandle<TGeoManager> geom;
   iSetup.get<DisplayGeomRecord>().get(geom);



// AMT workaround for an agressive clenup in 5.43.18
#if ROOT_VERSION_CODE >= ROOT_VERSION(5,34,18)
   if (!gStyle) {
      TColor::fgInitDone=false;
      TColor::InitializeColors();
      TStyle::BuildStyles();
      gROOT->SetStyle(gEnv->GetValue("Canvas.Style", "Modern"));
      gStyle = gROOT->GetStyle("Classic");
   }
#endif

   TEveManager::Create();

   TEveGeoTopNode* trk = new TEveGeoTopNode(const_cast<TGeoManager*>(geom.product()),
                        geom->GetTopNode());
   trk->SetVisLevel(2);
   gEve->AddGlobalElement(trk);

}


//define this as a plug-in
DEFINE_FIREWORKS_GEOM_DISPLAY(EveDisplayPlugin);