---

VisPixelDigiTwig.cc

Go to the documentation of this file.

//<<<<<< INCLUDES                                                       >>>>>>
#include "VisReco/VisTracker/interface/VisPixelDigiTwig.h"
#include "VisFramework/VisFrameworkBase/interface/debug.h"
#include "DataFormats/SiPixelDigi/interface/PixelDigi.h"
#include "DataFormats/SiPixelDigi/interface/PixelDigiCollection.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/TrackerGeometry.h"
#include "Geometry/Records/interface/TrackerDigiGeometryRecord.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                                    >>>>>>

VisPixelDigiTwig::VisPixelDigiTwig (IgState *state, IgTwig *parent,
                                    const std::string &name /* = "" */,
                                    unsigned int detType, unsigned int subDetType)
    : VisQueuedTwig (state, parent, name),
      m_text ("no info"),
      m_detType (detType),
      m_subDetType (subDetType)
{
}

void
VisPixelDigiTwig::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 ());

    edm::Handle<PixelDigiCollection> digiCollection;
    
    try
    {
        event.getByLabel ("pixdigi", digiCollection);
        eventSetup.get<TrackerDigiGeometryRecord> ().get (m_pDD);
    }
    catch (cms::Exception& e)
    {
        e.append (" from VisPixelDigiTwig: ");
        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 (digiCollection.isValid ())
    {
        // Get vector of detIDs in map
        const std::vector<unsigned int> detIDs = digiCollection->detIDs ();

        //--- Loop over detunits.
        std::vector<unsigned int>::const_iterator detunit_it;
        for (detunit_it  = detIDs.begin (); detunit_it != detIDs.end (); 
             detunit_it++) 
        {
            unsigned int detid = *detunit_it;   // return uint, = rawid

            // Det id
            DetId detId = DetId (detid);            // Get the Detid
//          unsigned int detType = detId.det ();    // det type, pixel=1
//          unsigned int subid = detId.subdetId (); //subdetector type, barrel=1
     
//          if (detType != m_detType) continue; // look only at pixels
            
//          // Subdet it, pix barrel=1
//          if (subid != m_subDetType)
//          {
//              //if (PRINT && subid == 2) cout << " forward hit " << endl;
//              continue; // look only at barrel
//          }

            // Look at digis now
            const PixelDigiCollection::Range digiRange = digiCollection->get (detid);
            PixelDigiCollection::ContainerIterator di;
            // Loop over Digis in this det unit
            for (di = digiRange.first; di != digiRange.second; ++di) 
            { 
//              int adc = di->adc();    // charge
//              int col = di->column(); // column 
//              int row = di->row();    // row
                //cout <<numberOfDigis<< " Col: " << col << " Row: " << row << " ADC: " 
                //   << adc << endl;

                std::pair<DetId, PixelDigi> p;
                p.first = detId;
                p.second = (*di);
                //              (*digiIt).print ();

                m_digis.push_back (p);
            } // for digis 
        } // for DetId
    }
    
    IgRepSet::invalidate (this, SELF_MASK);
}

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

    // Prepare property description.
    std::ostringstream  text;
    text << "Total " << m_digis.size () << " Pixel digis from";
    text << " Cosmic run: " << m_text << "<br>";

    text << "<table width='100%' border=1>"
         << "<TR align = center>"
         << "<TH>Number</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) 
        {
            Surface::GlobalPoint pos =  (m_pDD->idToDet ((*i).first))->surface ().position ();
            text << "<TR align = right>"
                 << "<TD>" << std::setw (3) << nDigis++ << "</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>";
        }
    }
    catch (...)
    {
        text << "No Pixel digis.<br>";
    }
    text << "</table>";
    }
    
    // Send it over.
    IgQtLock ();
    rep->setText (text.str ());
}

void
VisPixelDigiTwig::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, 0.0, 1.0);
        mat->emissiveColor.setValue(0.0, 0.0, 1.0);

        SoMaterial *matcube = new SoMaterial;
        matcube->diffuseColor.setValue (0.0, 1.0, 1.0);
        matcube->emissiveColor.setValue(0.0, 1.0, 1.0);

        SoDrawStyle *drawStyle = new SoDrawStyle;
        drawStyle->pointSize = 6.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)
            {
                sepmodule = new SoSeparator;
                sepmodule->addChild (mat);

                sepcube = new SoSeparator;
                sepcube->addChild (matcube);

                cubedigi = new SoCube ();

                const DetId& id = (*i).first;
                const GeomDetUnit *det = m_pDD->idToDetUnit (id);

                theDet = dynamic_cast<const PixelGeomDetUnit*> (m_pDD->idToDet (id));

                theTopol = dynamic_cast<const RectangularPixelTopology *>( &(theDet->specificTopology ()));
                const PixelDigi pd = (*i).second;
                theNumOfRow = theTopol->nrows ();      // rows in x
                theNumOfCol = theTopol->ncolumns ();   // cols in y
                std::pair<float, float> pitchxy = theTopol->pitch ();
                thePitchX = pitchxy.first;            // pitch along x
                thePitchY = pitchxy.second;           // pitch along y

                cubedigi->width = 1./theNumOfRow;
                cubedigi->height =1./theNumOfCol;
                cubedigi->depth =(pd.adc())/10. ;

                //--- Find the offset
                MeasurementPoint  offset =
                  theTopol->measurementPosition( LocalPoint(0., 0.) );
                theOffsetX = offset.x();
                theOffsetY = offset.y();

                float xlocal = (pd.row()*thePitchX + theOffsetX)/100.0 ;
                float ylocal = (pd.column()*thePitchY + theOffsetY)/100.0 ;
                LocalPoint p = LocalPoint(xlocal, ylocal);

                //Surface::GlobalPoint pos =  (m_pDD->idToDet (id))->surface ().position ();
                //float x = pos.x () / 100.0;  // cm -> m
                //float y = pos.y () / 100.0;  // cm -> m
                //float z = pos.z () / 100.0;  // cm -> m

                float x = (m_pDD->idToDet (id))->surface().toGlobal(p).x() / 100.0;  // cm -> m
                float y = (m_pDD->idToDet (id))->surface().toGlobal(p).y() / 100.0;  // cm -> m
                float z = (m_pDD->idToDet (id))->surface().toGlobal(p).z() / 100.0;  // cm -> m

                vertices->vertex.set1Value (nVrtx++, SbVec3f (x, y, z));

                digiTransform = new SoTransform;
                digiTransform->translation.setValue (xlocal/(thePitchX*theNumOfRow), ylocal/(thePitchY*theNumOfCol),pd.adc()/20. );
                sepcube->addChild (digiTransform);
                sepcube->addChild (cubedigi);

                sepmodule->addChild (sepcube);

                rep->node ()->addChild (sepmodule);
            }
            
            vertices->vertex.setNum (nVrtx);
            SoPointSet *points = new SoPointSet;
            points->vertexProperty.setValue (vertices);
            points->numPoints.setValue (nVrtx);

            sep->addChild (points);
        }
        catch (...) 
        {
            std::cout << "No PixelDigi Digis." << std::endl;
        }
    
        rep->node ()->addChild (sep);
    }
}

void
VisPixelDigiTwig::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) 
        {
            Surface::GlobalPoint pos =  (m_pDD->idToDet ((*i).first))->surface ().position ();
            double eta = pos.eta ();
            double phi = pos.phi ();
            if (phi < 0.0) phi = 2 * M_PI + phi; // correction in absence of one convention

            // int strip = (*i).second.getStrip ();
            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 (...) 
    {
        std::cout << "No PixelDigi Digis." << std::endl;
    }
    
    rep->node ()->addChild (sep);
}

void
VisPixelDigiTwig::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, 0.0, 1.0);
        sep->addChild (mat);

        SoDrawStyle *drawStyle = new SoDrawStyle;
        drawStyle->pointSize = 6.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;

                Surface::GlobalPoint pos =  (m_pDD->idToDet (id))->surface ().position ();
                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 (...) 
        {
            std::cout << "No PixelDigi Digis." << std::endl;
        }
    
        rep->node ()->addChild (sep);
    }
}

void
VisPixelDigiTwig::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) 
            {
                Surface::GlobalPoint pos =  (m_pDD->idToDet ((*i).first))->surface ().position ();
//              int strip = (*i).second.getStrip ();
//              std::vector<int> adcCounts = (*i).second.getADCCounts ();
            
                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 (nVrx++, SbVec3f (x, y, z));
            }
            vertices->vertex.setNum (m_digis.size ());

            SoPointSet *points = new SoPointSet;
            points->vertexProperty.setValue (vertices);
            points->numPoints.setValue (m_digis.size ());
            sep->addChild (points);
        }    
        catch (...) 
        {
            std::cout << "No PixelDigi Digis." << std::endl;
        }
    
        rep->node ()->addChild (sep);
    }
}

---