---

VisCSCSeg2HETwig.cc

Go to the documentation of this file.

//<<<<<< INCLUDES                                                       >>>>>>

#include "VisReco/VisRecoTools/interface/VisCSCSeg2HETwig.h"
#include "DataFormats/CSCRecHit/interface/CSCSegmentCollection.h"
#include "DataFormats/CSCRecHit/interface/CSCSegment.h"
#include "DataFormats/MuonDetId/interface/CSCDetId.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "Geometry/CommonDetUnit/interface/GeomDetType.h"
#include "Geometry/CommonDetUnit/interface/GeomDetUnit.h"
#include "Geometry/Records/interface/MuonGeometryRecord.h"
#include "Geometry/CSCGeometry/interface/CSCGeometry.h"
#include "Geometry/CSCGeometry/interface/CSCLayer.h"
#include "Geometry/CSCGeometry/interface/CSCLayerGeometry.h"
#include "DataFormats/GeometryVector/interface/GlobalPoint.h"
#include "Iguana/Framework/interface/IgRepSet.h"
#include "Iguana/Models/interface/IgTextRep.h"
#include "Iguana/GLModels/interface/Ig3DRep.h"
#include "Iguana/GLModels/interface/IgLegoRep.h"
#include "Iguana/GLModels/interface/IgRPhiRep.h"
#include "Iguana/GLModels/interface/IgRZRep.h"
#include "Iguana/Framework/interface/IgRepSet.h"
#include "Iguana/Studio/interface/IgQtLock.h"
#include <Inventor/nodes/SoLineSet.h>
#include <Inventor/nodes/SoMaterial.h>
#include <Inventor/nodes/SoSelection.h>
#include <Inventor/nodes/SoCube.h>
#include <Inventor/nodes/SoTranslation.h>
#include <Inventor/nodes/SoSeparator.h>
#include <Inventor/nodes/SoText2.h>
#include <qstring.h>
#include <sstream>
#include <iomanip>

//<<<<<< PRIVATE DEFINES                                                >>>>>>
//<<<<<< PRIVATE CONSTANTS                                              >>>>>>
//<<<<<< PRIVATE TYPES                                                  >>>>>>
//<<<<<< PRIVATE VARIABLE DEFINITIONS                                   >>>>>>
//<<<<<< PUBLIC VARIABLE DEFINITIONS                                    >>>>>>
//<<<<<< CLASS STRUCTURE INITIALIZATION                                 >>>>>>
//<<<<<< PRIVATE FUNCTION DEFINITIONS                                   >>>>>>
//<<<<<< PUBLIC FUNCTION DEFINITIONS                                    >>>>>>
//<<<<<< MEMBER FUNCTION DEFINITIONS                                    >>>>>>

VisCSCSeg2HETwig::VisCSCSeg2HETwig (IgState *state, IgTwig *parent,
                                    const std::string &name /* = "" */,
                                    const edm::Event &event, 
                                    const edm::EventSetup &eventSetup)
    : VisQueuedTwig (state, parent, name),
      m_event (event),
      m_eventSetup (eventSetup)
{
}

void
VisCSCSeg2HETwig::onNewEvent (const edm::Event &event,
                              const edm::EventSetup &eventSetup)
{
    VisQueuedTwig::onBaseInvalidate ();

    std::vector<edm::Handle<CSCSegmentCollection> > segCollections;
    try 
    {
        event.getManyByType (segCollections);
    }
    catch (...)
    {
        std::cout << "Failed to get the segments." << std::endl;
    }

    //
    if (! segCollections.empty ())
    {
        std::vector<edm::Handle<CSCSegmentCollection> >::iterator i;
        for (i = segCollections.begin (); i != segCollections.end (); i++)
        {
            const CSCSegmentCollection& c = *(*i);

            CSCSegmentCollection::const_iterator segIt;
            for (segIt = c.begin (); segIt != c.end (); ++segIt)
            {
                m_segs.push_back (*segIt);
            }
        }
    }

    IgRepSet::invalidate (this, SELF_MASK);
}

void
VisCSCSeg2HETwig::update (IgTextRep *rep)
{
    // Get debugging dump.
    VisQueuedTwig::update (rep);

    // Prepare property description.
    std::ostringstream  text;

    text << QString ("Run # %1, event # %2")
        .arg (m_event.id ().run ())
        .arg (m_event.id ().event ()).latin1 () << "<br>";

//     text << "<table width='100%' border=1>"
//       << "<TR align = center>"
//       << "<TH>E<SUB>em (GeV)</TH>"
//       << "<TH>E<SUB>had</TH>"
//       << "<TH>E<SUB>inv</TH>"
//       << "<TH>E<SUB>aux</TH>"
//       << "<TH>Eta</TH>"
//       << "<TH>Phi</TH>"
//       << "<TH>Particles</TH>"
//       << "</TR>";
//     text << setiosflags (std::ios::showpoint | std::ios::fixed);
//     text.setf (std::ios::right, std::ios::adjustfield);
 
//     text << "<TR align = right>"
//       << "<TD>" << std::setw (2) << std::setprecision (3) << eEm << "</TD>"
//       << "<TD>" << std::setw (6) << std::setprecision (3) << eHad << "</TD>"
//       << "<TD>" << std::setw (5) << std::setprecision (3) << eInv << "</TD>"
//       << "<TD>" << std::setw (5) << std::setprecision (3) << eAux << "</TD>"
//       << "<TD>" << std::setw (5) << std::setprecision (3) << m_jet.eta () << "</TD>"
//       << "<TD>" << std::setw (5) << std::setprecision (3) << m_jet.phi () << "</TD>"
//       << "<TD>";
//     text << "</TD></TR>";
       
//     text << "</table>";
    
    // Send it over.
    IgQtLock ();

    rep->setText (text.str ());
}

void
VisCSCSeg2HETwig::update (Ig3DRep *rep)
{
    // Get debugging dump.
    VisQueuedTwig::update (rep);

    IgQtLock ();

    SoSeparator *sep = new SoSeparator;
    SoMaterial *mat = new SoMaterial;
    mat->ambientColor.setValue (0.135, 0.2225, 0.1575);
    sep->addChild (mat);

    using namespace edm;

    ESHandle<CSCGeometry> pDD;
    try 
    {
        m_eventSetup.get<MuonGeometryRecord> ().get (pDD);
    }
    catch (...)
    {
        std::cout << "Failed to get the goemetry." << std::endl;
    }
    
//     std::vector<CSCSegment> segs;

//     std::vector<edm::Handle<CSCSegmentCollection> > segCollections;
//     try 
//     {
//      m_event.getManyByType (segCollections);
//     }
//     catch (...)
//     {
//      std::cout << "Failed to get the segments." << std::endl;
//     }

//     //
//     if (! segCollections.empty ())
//     {
//      std::vector<edm::Handle<CSCSegmentCollection> >::iterator i;
//      for (i = segCollections.begin (); i != segCollections.end (); i++)
//      {
//          const CSCSegmentCollection& c = *(*i);

//          CSCSegmentCollection::const_iterator segIt;
//          for (segIt = c.begin (); segIt != c.end (); ++segIt)
//          {
//              segs.push_back (*segIt);
//          }
//      }
//     }
//     //

    if (! m_segs.empty () && pDD.isValid ())
    {
        try 
        {
            float h ;
            std::cout << "== Run # " << m_event.id ().run () << ", event # " << m_event.id ().event () << std::endl; // latin1 *****
            for (std::vector<CSCSegment>::const_iterator iseg = m_segs.begin (); iseg != m_segs.end (); iseg++)
            {
                h = 5.7;
                const GeomDet *det = pDD->idToDet ((*iseg).cscDetId ());

                SoVertexProperty *vtx = new SoVertexProperty;
                
                SoLineSet *line = new SoLineSet;
                line->numVertices = 2;

                // Local pos & dir
                LocalPoint  pos = (*iseg).localPosition();
                LocalVector dir = (*iseg).localDirection();

                float halfThickness = det->surface ().bounds ().thickness ()/2.;

                float z1 = halfThickness;
                float x1 = pos.x() + dir.x()*z1/dir.z();
                float y1 = pos.y() + dir.y()*z1/dir.z();
                GlobalPoint g1 = det->surface().toGlobal( LocalPoint(x1,y1,z1) );

                float z2 = -halfThickness;
                float x2 = pos.x() + dir.x()*z2/dir.z();
                float y2 = pos.y() + dir.y()*z2/dir.z();
                GlobalPoint g2 = det->surface().toGlobal( LocalPoint(x2,y2,z2) );

                float x = g1.x () / 100.0;  // cm -> m
                float y = g1.y () / 100.0;  // cm -> m
                float z = g1.z () / 100.0;  // cm -> m

                //std::cout << "\n*** 1st POINT *** (" << x << ", " << y << ", " << z << ")" << std::endl;
                vtx->vertex.set1Value(0,SbVec3f(x,y,z));

                float X = g2.x () / 100.0;  // cm -> m
                float Y = g2.y () / 100.0;  // cm -> m
                float Z = g2.z () / 100.0;  // cm -> m

                //std::cout << "\n*** 2nd POINT *** (" << x << ", " << y << ", " << z << ")" << std::endl;
        
                h = (z+Z)/2 - h;
                //std::cout << " *** " << h << std::endl;

                x = (x+X)/2 - (x-X)/(z-Z)*h;
                y = (y+Y)/2 - (y-Y)/(z-Z)*h;
                z = 5.7;

                vtx->vertex.set1Value(1,SbVec3f(x,y,z));

                float phi, theta, eta;
                int iphi, ieta;

                phi = atan2(y,x);
                theta = acos(z/sqrt(x*x+y*y+z*z));

                if (phi<0) phi += 2*3.141592; // if the range of phi is defined to be from 0 to 2*PI *****1
                eta = -log(tan(theta/2));

                //std::cout << " ETA!!! = " << eta << std::endl;
                iphi = (int)((phi - 0.087/2)/0.087 + 1); // casting check *****2
                ieta = (int)((eta - 0.087/2)/0.087);

                //      if (myfile.is_open())
                //      {
                std::cout << "( iphi = " << iphi << ", ieta = " << ieta << " )" << std::endl;
                //      }
                //      else std::cout << "Unable to open file" << std::endl;

                line->vertexProperty = vtx;
                
                sep->addChild (line);
            }
        }
        catch (...)
        {
            std::cout << "Muon Endcap Segments Info: Nothing to display: an exception is caught.";
        }    
    }    

    rep->clear ();
    rep->node ()->addChild (sep);
}


void
VisCSCSeg2HETwig::update (IgLegoRep *rep)
{}

void
VisCSCSeg2HETwig::update (IgRPhiRep *rep)
{}

void
VisCSCSeg2HETwig::update (IgRZRep *rep)
{}

---