---

VisTrackerPiDigiTwig.cc

Go to the documentation of this file.

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

#include "VisReco/VisTracker/interface/VisTrackerPiDigiTwig.h"
#include "VisFramework/VisFrameworkBase/interface/debug.h"
#include "VisFramework/VisFrameworkBase/interface/VisTwigFactroyService.h"
#include "VisFramework/VisFrameworkBase/interface/VisEventSelector.h"
#include "DataFormats/SiPixelDigi/interface/PixelDigi.h"
#include "DataFormats/Common/interface/DetSetVector.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "Geometry/CommonDetUnit/interface/GeomDetType.h"
#include "Geometry/CommonDetUnit/interface/GeomDetUnit.h"
#include "Geometry/TrackerGeometryBuilder/interface/PixelGeomDetUnit.h"
#include "Geometry/TrackerTopology/interface/RectangularPixelTopology.h"
#include "Geometry/TrackerGeometryBuilder/interface/TrackerGeometry.h"
#include "Geometry/Records/interface/TrackerDigiGeometryRecord.h"
#include "Geometry/CommonTopologies/interface/StripTopology.h"
#include "Iguana/Inventor/interface/IgSoTower.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/SoSeparator.h>
#include <Inventor/nodes/SoTransform.h>
#include <Inventor/nodes/SoCube.h>
#include <Inventor/nodes/SoDrawStyle.h>
#include <Inventor/nodes/SoMaterial.h>
#include <Inventor/nodes/SoPointSet.h>
#include <Inventor/nodes/SoVertexProperty.h>
#include <qstring.h>
#include <iostream>
#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                                    >>>>>>

namespace 
{
    VisQueuedTwig *
    createThisTwig (IgState *state, IgTwig *parent,
                    const std::string &name,
                    const std::string &friendlyName,
                    const std::string &modLabel,
                    const std::string &instanceName,
                    const std::string &processName)
    {
        return new VisTrackerPiDigiTwig (state, parent, "Tracker PixelDigis (" + name + ")", 
                                         friendlyName, modLabel, instanceName, processName);
    }
}

VisTrackerPiDigiTwig::VisTrackerPiDigiTwig(IgState *state, IgTwig *parent,
                                           const std::string &name /* = "" */,
                                           const std::string &friendlyName /* = "" */,
                                           const std::string &moduleLabel /* = "" */,
                                           const std::string &instanceName /* = "" */,
                                           const std::string &processName /* = "" */)
    : VisQueuedTwig (state, parent, name),
      m_text (name),
      m_friendlyName (friendlyName),
      m_moduleLabel (moduleLabel),
      m_instanceName (instanceName),
      m_processName (processName)
{
    VisTwigFactroyService *tfService = VisTwigFactroyService::get (state);
    if (! tfService)
    {
        tfService = new VisTwigFactroyService (state);
    }   
    edm::TypeID piDigiCollID (typeid (edm::DetSetVector<PixelDigi>));
    tfService->registerTwig (piDigiCollID.friendlyClassName (), &createThisTwig);
}

void
VisTrackerPiDigiTwig::onNewEvent (const edm::Event &event,
                                  const edm::EventSetup &eventSetup)
{
    // Get debugging dump.
    VisQueuedTwig::onNewEvent (event, eventSetup);

    m_digis.clear ();
  
    m_text = (QString ("Run # %1, event # %2")
              .arg (event.id ().run ())
              .arg (event.id ().event ()).latin1 ());
 
    std::vector< edm::Handle<edm::DetSetVector<PixelDigi> > > p_digi_collections;
    try
    {
        if ((! m_friendlyName.empty ()) || (! m_moduleLabel.empty ()) || (! m_instanceName.empty ()) || (! m_processName.empty ()))
        {
            VisEventSelector visSel (m_friendlyName, m_moduleLabel, m_instanceName, m_processName);
            event.getMany (visSel, p_digi_collections);
        }
        else
        {
            event.getManyByType (p_digi_collections);
        }
        eventSetup.get<TrackerDigiGeometryRecord> ().get (m_pDD);
    }
    catch (cms::Exception& e)
    {
        e.append (" from VisTrackerPiDigiTwig: ");
        e.append (this->name ());
            
        if (this->m_onCmsException)
            this->m_onCmsException (&e);
    }
    catch (lat::Error &e) 
    {
        if (this->m_onError)
            this->m_onError (&e);
    }
    catch (std::exception &e) 
    {
        if (this->m_onException)
            this->m_onException (&e);
    }
    catch (...) 
    {
        if (this->m_onUnhandledException)
            this->m_onUnhandledException ();
    }

    if (! p_digi_collections.empty ())
    {
        std::vector< edm::Handle<edm::DetSetVector<PixelDigi> > >::iterator i;
        for (i = p_digi_collections.begin (); i != p_digi_collections.end (); i++) 
        {
            const edm::DetSetVector<PixelDigi> &c = *(*i);
 
            edm::DetSetVector<PixelDigi>::const_iterator DSViter=c.begin();
            std::cout <<  c.size() << std::endl;
            for (; DSViter!=c.end();DSViter++){
                edm::DetSet<PixelDigi> ds = *DSViter;
                if (ds.data.size())
                {
                    const uint32_t& detID = ds.id;
                    DetId detid(detID);
                    edm::DetSet<PixelDigi>::const_iterator idigi=ds.data.begin();
                    for(;idigi!=ds.data.end();idigi++){ 
                        std::pair<DetId, PixelDigi> p;
                        p.first = detid;
                        p.second = (*idigi);
                        m_digis.push_back (p);
                    } // for digis 
                }//if data.size
            }//for DSViter
        }
    }

    IgRepSet::invalidate (this, SELF_MASK);
}

void
VisTrackerPiDigiTwig::update (IgTextRep *rep)
{
    // Get debugging dump.
    VisQueuedTwig::update (rep);
  
    // Prepare property description.
    std::ostringstream  text;
    text << "Total " << m_digis.size () << " Pixel digis from";
    if(m_digis.size() > 100){
        text << " input file: printing only first 100 " << "<br>"; 
    }else text << " input file: " << m_text << "<br>";
  
    text << "<table width='100%' border=1>"
         << "<TR align = center>"
         << "<TH>Number</TH>"
         << "<TH>GeomdetId</TH>"
         << "<TH>Position</TH>"
         << "<TH>Charge</TH>"
         << "<TH>Column</TH>"
         << "<TH>Row</TH>"
         << "</TR>";
    text << setiosflags (std::ios::showpoint | std::ios::fixed);
    text.setf (std::ios::right, std::ios::adjustfield);
  
    if ((! m_digis.empty ())&& m_pDD.isValid ()) 
    {
        int nDigis = 0;
        try 
        {
            for (std::vector<std::pair<DetId, PixelDigi> >::const_iterator i = m_digis.begin (); i != m_digis.end (); ++i) 
            {
                const DetId& id = (*i).first;
                const PixelGeomDetUnit* theDet = dynamic_cast<const PixelGeomDetUnit*>(m_pDD->idToDet (id));
                const RectangularPixelTopology *theTopol = dynamic_cast<const RectangularPixelTopology *>( &(theDet->specificTopology ()));
              
                int row = (*i).second.row();
                int column =(*i).second.column();
              
                GlobalPoint pos =  (m_pDD->idToDet (id))->surface().toGlobal(theTopol->localPosition(MeasurementPoint(row,column)));
              
                text << "<TR align = right>"
                     << "<TD>" << std::setw (3) << nDigis++ << "</TD>"
                     << "<TD>" << std::setw (11) << ((*i).first).rawId()<< "</TD>"
                     << "<TD>" << pos.x () << ", " << pos.y () << ", " << pos.z () << "</TD>"
                     << "<TD>" << std::setw (6) << std::setprecision (3) << (*i).second.adc () << "</TD>"
                     << "<TD>" << std::setw (6) << std::setprecision (3) << (*i).second.column () << "</TD>"
                     << "<TD>" << std::setw (6) << std::setprecision (3) << (*i).second.row () << "</TD></TR>";
                if (nDigis>100) break;
            }
        }
        catch (cms::Exception& e)
        {
            e.append (" from VisTrackerPiDigiTwig: ");
            e.append (this->name ());
            
            if (this->m_onCmsException)
                this->m_onCmsException (&e);
        }
        catch (lat::Error &e) 
        {
            if (this->m_onError)
                this->m_onError (&e);
        }
        catch (std::exception &e) 
        {
            if (this->m_onException)
                this->m_onException (&e);
        }
        catch (...) 
        {
            if (this->m_onUnhandledException)
                this->m_onUnhandledException ();
        }
        text << "</table><p>";
    }
    // Send it over.
    IgQtLock ();
    rep->setText (text.str ());
}

void
VisTrackerPiDigiTwig::update (Ig3DRep *rep)
{
    // Get debugging dump.
    VisQueuedTwig::update (rep);
    IgQtLock ();
    rep->clear ();
  
    if (! m_digis.empty ())
    {
        SoMaterial *mat = new SoMaterial;
        mat->diffuseColor.setValue (0.0, 1.0, 0.0);//green
        mat->emissiveColor.setValue(0.0, 1.0, 0.0);
      
        SoDrawStyle *drawStyle = new SoDrawStyle;
        drawStyle->pointSize = 4.0;
      
        SoSeparator *sep = new SoSeparator; // global
        sep->addChild (mat);
        sep->addChild (drawStyle);
      
        try 
        {
            SoVertexProperty *vertices = new SoVertexProperty;
            int nVrtx = 0;
            for (std::vector<std::pair<DetId, PixelDigi> >::const_iterator 
                     i = m_digis.begin (); i != m_digis.end (); ++i)
            {
                const DetId& id = (*i).first;
                const PixelGeomDetUnit* theDet = dynamic_cast<const PixelGeomDetUnit*>(m_pDD->idToDet (id));
                const RectangularPixelTopology *theTopol = dynamic_cast<const RectangularPixelTopology *>( &(theDet->specificTopology ()));
                int row = (*i).second.row();
                int column =(*i).second.column();
                GlobalPoint pos =  (m_pDD->idToDet (id))->surface().toGlobal(theTopol->localPosition(MeasurementPoint(row,column)));
                float x = pos.x () / 100.0;  // cm -> m
                float y = pos.y () / 100.0;  // cm -> m
                float z = pos.z () / 100.0;  // cm -> m
                vertices->vertex.set1Value (nVrtx++, SbVec3f (x, y, z));
            }
          
            vertices->vertex.setNum (nVrtx);
            SoPointSet *points = new SoPointSet;
            points->vertexProperty.setValue (vertices);
            points->numPoints.setValue (nVrtx);
          
            sep->addChild (points);
        }
        catch (cms::Exception& e)
        {
            e.append (" from VisTrackerPiDigiTwig: ");
            e.append (this->name ());
            
            if (this->m_onCmsException)
                this->m_onCmsException (&e);
        }
        catch (lat::Error &e) 
        {
            if (this->m_onError)
                this->m_onError (&e);
        }
        catch (std::exception &e) 
        {
            if (this->m_onException)
                this->m_onException (&e);
        }
        catch (...) 
        {
            if (this->m_onUnhandledException)
                this->m_onUnhandledException ();
        }
        rep->node ()->addChild (sep);
    }
}

void
VisTrackerPiDigiTwig::update (IgLegoRep *rep)
{
    // Get debugging dump.
    VisQueuedTwig::update (rep);
  
    IgQtLock ();
    rep->clear ();
  
    SoSeparator *sep = new SoSeparator; // global
  
    try 
    {
        for (std::vector<std::pair<DetId, PixelDigi> >::const_iterator i = m_digis.begin (); i != m_digis.end (); ++i) 
        {
            const DetId& id = (*i).first;
            const PixelGeomDetUnit* theDet = dynamic_cast<const PixelGeomDetUnit*>(m_pDD->idToDet (id));
            const RectangularPixelTopology *theTopol = dynamic_cast<const RectangularPixelTopology *>( &(theDet->specificTopology ()));
            int row = (*i).second.row();
            int column =(*i).second.column();
            GlobalPoint pos =  (m_pDD->idToDet (id))->surface().toGlobal(theTopol->localPosition(MeasurementPoint(row,column)));
            double eta = pos.eta ();
            double phi = pos.phi ();
            if (phi < 0.0) phi = 2 * M_PI + phi; // correction in absence of one convention
            int adcCounts = (*i).second.adc ();
          
            IgSoTower *tower = new IgSoTower;
            tower->position = SbVec2f (phi, eta); // eta, phi
            tower->scaleFactor = 20.0;
            tower->energy = adcCounts;
            sep->addChild (tower);
        }
      
    }
    catch (cms::Exception& e)
    {
        e.append (" from VisTrackerPiDigiTwig: ");
        e.append (this->name ());
            
        if (this->m_onCmsException)
            this->m_onCmsException (&e);
    }
    catch (lat::Error &e) 
    {
        if (this->m_onError)
            this->m_onError (&e);
    }
    catch (std::exception &e) 
    {
        if (this->m_onException)
            this->m_onException (&e);
    }
    catch (...) 
    {
        if (this->m_onUnhandledException)
            this->m_onUnhandledException ();
    }

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

void
VisTrackerPiDigiTwig::update (IgRPhiRep *rep)
{
    // Get debugging dump.
    VisQueuedTwig::update (rep);
  
    IgQtLock ();
    rep->clear ();
  
    if (! m_digis.empty ()) 
    {
        SoSeparator *sep = new SoSeparator;    
      
        SoMaterial *mat = new SoMaterial;
        mat->diffuseColor.setValue (0.0, 1.0, 0.0);
        sep->addChild (mat);
      
        SoDrawStyle *drawStyle = new SoDrawStyle;
        drawStyle->pointSize = 4.0;
        sep->addChild (drawStyle);
      
        try 
        {
            SoVertexProperty *vertices = new SoVertexProperty;
            int nVrtx = 0;
            for (std::vector<std::pair<DetId, PixelDigi> >::const_iterator 
                     i = m_digis.begin (); i != m_digis.end (); ++i)
            {
                const DetId& id = (*i).first;
                const PixelGeomDetUnit* theDet = dynamic_cast<const PixelGeomDetUnit*>(m_pDD->idToDet (id));
                const RectangularPixelTopology *theTopol = dynamic_cast<const RectangularPixelTopology *>( &(theDet->specificTopology ()));
                int row = (*i).second.row();
                int column =(*i).second.column();
                GlobalPoint pos =  (m_pDD->idToDet (id))->surface().toGlobal(theTopol->localPosition(MeasurementPoint(row,column)));
                float x = pos.x () / 100.0;  // cm -> m
                float y = pos.y () / 100.0;  // cm -> m
                float z = 10.0;  // cm -> m
                vertices->vertex.set1Value (nVrtx++, SbVec3f (x, y, z));
            }
            vertices->vertex.setNum (nVrtx);
            SoPointSet *points = new SoPointSet;
            points->vertexProperty.setValue (vertices);
            points->numPoints.setValue (nVrtx);
          
            sep->addChild (points);
        }
        catch (cms::Exception& e)
        {
            e.append (" from VisTrackerPiDigiTwig: ");
            e.append (this->name ());
            
            if (this->m_onCmsException)
                this->m_onCmsException (&e);
        }
        catch (lat::Error &e) 
        {
            if (this->m_onError)
                this->m_onError (&e);
        }
        catch (std::exception &e) 
        {
            if (this->m_onException)
                this->m_onException (&e);
        }
        catch (...) 
        {
            if (this->m_onUnhandledException)
                this->m_onUnhandledException ();
        }

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

void
VisTrackerPiDigiTwig::update (IgRZRep *rep)
{
    // Get debugging dump.
    VisQueuedTwig::update (rep);
  
    IgQtLock ();
    rep->clear ();
  
    if (! m_digis.empty () ) 
    {
        SoSeparator *sep = new SoSeparator;    
        SoMaterial *mat = new SoMaterial;
        mat->diffuseColor.setValue (0.3, 0.8, 0.0);
        sep->addChild (mat);
      
        SoDrawStyle *drawStyle = new SoDrawStyle;
        drawStyle->pointSize = 3.0;
        sep->addChild (drawStyle);
      
        try 
        {
            SoVertexProperty *vertices = new SoVertexProperty;
            int nVrx = 0;
          
            for (std::vector<std::pair<DetId,PixelDigi> >::const_iterator i = m_digis.begin (); i != m_digis.end (); ++i) 
            {
                const DetId& id = (*i).first;
                const PixelGeomDetUnit* theDet = dynamic_cast<const PixelGeomDetUnit*>(m_pDD->idToDet (id));
                 //              Draw only if in barrel.
                int subdet = theDet->geographicalId().subdetId();
                if(subdet==1 || subdet == 3 || subdet == 5){
                const RectangularPixelTopology *theTopol = dynamic_cast<const RectangularPixelTopology *>( &(theDet->specificTopology ()));
                int row = (*i).second.row();
                int column =(*i).second.column();
                GlobalPoint pos =  (m_pDD->idToDet (id))->surface().toGlobal(theTopol->localPosition(MeasurementPoint(row,column)));
                float x = 0;
                float y = sqrt(pos.x ()*pos.x()+pos.y()*pos.y()) / 100.0; if(pos.y()< 0. )y = - y;
                float z = pos.z () / 100.0;  // cm -> m
              
                vertices->vertex.set1Value (nVrx++, SbVec3f (x, y, z));
            }
            }
            vertices->vertex.setNum (nVrx);
          
            SoPointSet *points = new SoPointSet;
            points->vertexProperty.setValue (vertices);
            points->numPoints.setValue (nVrx);
            sep->addChild (points);
        }    
        catch (cms::Exception& e)
        {
            e.append (" from VisTrackerPiDigiTwig: ");
            e.append (this->name ());
            
            if (this->m_onCmsException)
                this->m_onCmsException (&e);
        }
        catch (lat::Error &e) 
        {
            if (this->m_onError)
                this->m_onError (&e);
        }
        catch (std::exception &e) 
        {
            if (this->m_onException)
                this->m_onException (&e);
        }
        catch (...) 
        {
            if (this->m_onUnhandledException)
                this->m_onUnhandledException ();
        }

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

---