---

VisG4TrackPtTwig Class Reference

#include <VisSimulation/VisSimBase/interface/VisG4TrackPtTwig.h>

Inheritance diagram for VisG4TrackPtTwig:

List of all members.

Public Member Functions
virtual void	cutChanged (void)
virtual void	update (Ig3DRep *rep)
virtual void	update (IgTextRep *rep)
	VisG4TrackPtTwig (IgState *state, IgTwig *parent, std::string name, unsigned colour, bool highpt, VisG4TracksTwig *tracks, const int *ids)
Private Member Functions
bool	rejectTrajectory (const G4Trajectory *traj) const
Private Attributes
const int *	m_ids
bool	m_inclusive
bool	m_isHighPt
unsigned	m_rgba
VisG4TracksTwig *	m_tracks

---

Detailed Description

Definition at line 20 of file VisG4TrackPtTwig.h.

---

Constructor & Destructor Documentation

VisG4TrackPtTwig::VisG4TrackPtTwig	(	IgState *	state,
		IgTwig *	parent,
		std::string	name,
		unsigned	colour,
		bool	highpt,
		VisG4TracksTwig *	tracks,
		const int *	ids
	)

Definition at line 41 of file VisG4TrackPtTwig.cc.

    : VisQueuedTwig (state, parent, name),
      m_tracks (tracks),
      m_isHighPt (highpt),
      m_ids (ids),
      m_rgba (colour)
{}

---

Member Function Documentation

void VisG4TrackPtTwig::cutChanged

(

void

)

[virtual]

Definition at line 52 of file VisG4TrackPtTwig.cc.

References IgRepSet::invalidate(), and IgTwig::SELF_MASK.

{ IgRepSet::invalidate (this, SELF_MASK); }

bool VisG4TrackPtTwig::rejectTrajectory

(

const G4Trajectory *

traj

)

const [private]

Definition at line 56 of file VisG4TrackPtTwig.cc.

References ASSERT, VisG4TracksTwig::CHARGE, VisG4TracksTwig::CHARGED, i, int, m_ids, m_isHighPt, VisG4TracksTwig::NEUTRAL, p, VisG4TracksTwig::PARTICLE, and funct::sqrt().

Referenced by update().

{
    // Check if it passes pt cut
    Hep3Vector  p = traj->GetInitialMomentum ();
    double      px = p.x () / GeV;
    double      py = p.y () / GeV;
    double      pt = sqrt (px * px + py * py);

    if ((m_isHighPt)
        || (! m_isHighPt))
        return true;

    // Check if it passes eta cut?
    // if ( ... > m_etaCut.value ())
    //     return true;

    // Check if the particle should be accepted.  We accept three params
    // in the "m_ids" array.  Firstly, there can be a "CHARGE" setting,
    // accepting or passing by particle charge.  Then there can be a
    // section of "PARTICLE" type; the list can indicate an inclusive
    // list (only particles mentioned will be accepted) or exclusive
    // (particles other than those mentioned will be accepted).
    const int *ids = m_ids;
    
    // First check for general charge setting; if it fails, quit early
    if (ids [0] == VisG4TracksTwig::CHARGE)
    {
        int charge = (int) traj->GetCharge ();

        if (ids [1] == VisG4TracksTwig::CHARGED)
        {
            if (charge == 0)
                return true;
        }
        else if (ids [1] == VisG4TracksTwig::NEUTRAL)
        {
            if (charge != 0)
                return true;
        }
        else
            ASSERT (false);
        
        ids += 2;
    }

    // Now check for particle choice.  Negative PARTICLE means exclusion.
    ASSERT (ids [0] == VisG4TracksTwig::PARTICLE
            || ids [0] == -VisG4TracksTwig::PARTICLE);

    int         partId = traj->GetPDGEncoding ();
    bool        inclusive = (ids [0] == VisG4TracksTwig::PARTICLE);
    int         nparts = ids [1];
    bool        isOfType = false;

    for (int i = 0; !isOfType && i < nparts; ++i)
         isOfType = isOfType || (partId == ids [2 + i]);

    return inclusive != isOfType;
}

void VisG4TrackPtTwig::update

(

Ig3DRep *

rep

)

[virtual]

Reimplemented from VisQueuedTwig.

Definition at line 199 of file VisG4TrackPtTwig.cc.

References ASSERT, darken(), group, i, j, parsecf::pyparsing::line(), funct::log(), m, m_rgba, m_tracks, Ig3DBaseRep::node(), p, rejectTrajectory(), funct::sqrt(), cmsRelvalreportInput::step, VisG4TracksTwig::trajectories(), and VisQueuedTwig::update().

{
    // Get debugging dump
    VisQueuedTwig::update (rep);

    // Prepare parent container
    qApp->lock ();
    SoGroup *contents = rep->node ();
    contents->removeAllChildren ();

    // Process all trajectories
    G4TrajectoryContainer *trajs = m_tracks->trajectories ();
    for (int i = 0; i < trajs->entries (); ++i)
    {
        // FIXME: separate ideal track from momentum and charge?

        ASSERT (dynamic_cast<G4Trajectory *> ((*trajs) [i]));
        G4Trajectory *traj = static_cast<G4Trajectory *> ((*trajs) [i]);
        if (rejectTrajectory (traj))
            continue;

        static SoDrawStyle *thick = 0;
        static SoDrawStyle *thin = 0;
        if (! thick)
        {
            thick = new SoDrawStyle;
            thick->lineWidth = 2.0;
            thick->ref ();

            thin = new SoDrawStyle;
            thin->lineWidth = 1.0;
            thin->ref ();
        }

        SoSeparator     *group = new SoSeparator;
        SoLineSet       *line = new SoLineSet;
        SoVertexProperty *vtx = new SoVertexProperty;
        int             npoints = traj->GetPointEntries ();
        SbVec3f         *points;

        // Colour the track by pT.  Ramp logarithmically from 2 GeV (50%)
        // to 20 GeV (100%) as a function of pT.
        Hep3Vector      p = traj->GetInitialMomentum ();
        double          px = p.x () / GeV;
        double          py = p.y () / GeV;
        double          pt = sqrt (px * px + py * py);
        double          ptscale = (pt <= 2 ? .5
                                   : pt <= 20 ? .5 + log(pt) / log(20.) * .5
                                   : 1);
        SoDrawStyle     *style = pt < 20 ? thin : thick;
        vtx->materialBinding = SoVertexProperty::OVERALL;
        vtx->orderedRGBA = darken (m_rgba, ptscale);

        // Get which step to draw
        int step = 1; // FIXME: m_step.value ();
        
        // Collect trajectory points
        int nVertices = 0;
        
        vtx->vertex.setNum (npoints);
        points = vtx->vertex.startEditing ();
        for (int j = 0; j < npoints; ++j)
        {
            G4VTrajectoryPoint *vp = traj->GetPoint (j);
            ASSERT (dynamic_cast<G4TrajectoryPoint *> (vp));
            G4ThreeVector pt = static_cast<G4TrajectoryPoint *> (vp)->GetPosition ();
            if (j % step == 0 || j == npoints - 1) 
            {
                points [nVertices] = SbVec3f (pt.x () / m, pt.y () / m, pt.z () / m);
                nVertices++;
            }
        }
        
        vtx->vertex.finishEditing ();
        line->numVertices = nVertices;
        line->vertexProperty = vtx;
        group->addChild (style);
        group->addChild (line);
        contents->addChild (group);
    }

    qApp->unlock (false);
}

void VisG4TrackPtTwig::update

(

IgTextRep *

rep

)

[virtual]

Reimplemented from VisQueuedTwig.

Definition at line 117 of file VisG4TrackPtTwig.cc.

References ASSERT, i, int, m_isHighPt, m_tracks, IgSimpleTwig::name(), p, rejectTrajectory(), IgTextRep::setText(), funct::sqrt(), VisG4TracksTwig::trajectories(), and VisQueuedTwig::update().

{
    // Get debugging dump
    VisQueuedTwig::update (rep);

    // Prepare property description.
    G4TrajectoryContainer       *trajs = m_tracks->trajectories ();
    std::ostringstream          text;
    std::ostringstream          preamble;
    int                         nSelected = 0;

    text << std::setiosflags (std::ios::showpoint | std::ios::fixed);
    text.setf (std::ios::right, std::ios::adjustfield);

    for (int i = 0; i < trajs->entries (); ++i)
    {
        ASSERT (dynamic_cast<G4Trajectory *> ((*trajs) [i]));
        G4Trajectory *traj = static_cast<G4Trajectory *> ((*trajs) [i]);
        if (rejectTrajectory (traj))
            continue;

        if (++nSelected == 1)
        {
            text << "<table width='100%' border='1'>\n"
                 << "<tr align='center'>"
                 << "<th>ID</th>"
                 << "<th>Parent ID</th>"
                 << "<th>Name</th>"
                 << "<th>Charge</th>"
                 << "<th>p<sub>T</sub></th>"
                 << "<th>p<sub>x</sub></th>"
                 << "<th>p<sub>y</sub></th>"
                 << "<th>p<sub>z</sub></th>"
                 << "<th>Points</th>"
                 // << "<th>E [GeV]</th>"
                 // << "<th>p<sub>T</sub> [GeV/c]</th>"
                 // << "<th>charge</th>"
                 // << "<th>eta</th>"
                 // << "<th>phi [rad]</th>"
                 // << "<th>vertex position [cm]</th>"
                 << "</tr>";
        }

        Hep3Vector      p = traj->GetInitialMomentum ();
        double          px = p.x () / GeV;
        double          py = p.y () / GeV;
        double          pz = p.z () / GeV;
        double          pt = sqrt (px * px + py * py);

        text << "<tr align='right'>"
             << "<td>" << traj->GetTrackID () << "</td>"
             << "<td>" << traj->GetParentID () << "</td>"
             << "<td>" << traj->GetParticleName () << "</td>"
             << "<td>" << (int) traj->GetCharge () << "</td>"
             << "<td>" << std::setprecision (2) << pt << "</td>"
             << "<td>" << std::setprecision (2) << px << "</td>"
             << "<td>" << std::setprecision (2) << py << "</td>"
             << "<td>" << std::setprecision (2) << pz << "</td>"
             << "<td>" << traj->GetPointEntries () << "</td>"
             << "</tr>";
    }

    if (nSelected)
    {
        text << "</table>\n";
        preamble << "<p>" << name () << " with p<sub>T</sub> "
                 << (m_isHighPt ? "over " : "under ")
                 << std::setprecision (2) << "FIXME: NO CUT"
                 << ": " << nSelected << " trajectories</p>";
    }
    else
        text << "<p>No " << name () << " with p<sub>T</sub> "
             << (m_isHighPt ? "over " : "under ")
             << std::setprecision (2) << "FIXME: NO CUT" << "</p>";

    // Send it over.
    qApp->lock ();
    rep->setText (preamble.str () + text.str ());
    qApp->unlock (false);
}

---

Member Data Documentation

const int* VisG4TrackPtTwig::m_ids [private]

Definition at line 38 of file VisG4TrackPtTwig.h.

Referenced by rejectTrajectory().

bool VisG4TrackPtTwig::m_inclusive [private]

Definition at line 37 of file VisG4TrackPtTwig.h.

bool VisG4TrackPtTwig::m_isHighPt [private]

Definition at line 36 of file VisG4TrackPtTwig.h.

Referenced by rejectTrajectory(), and update().

unsigned VisG4TrackPtTwig::m_rgba [private]

Definition at line 39 of file VisG4TrackPtTwig.h.

Referenced by update().

VisG4TracksTwig* VisG4TrackPtTwig::m_tracks [private]

Definition at line 35 of file VisG4TrackPtTwig.h.

Referenced by update().

---

The documentation for this class was generated from the following files:

• VisSimulation/VisSimBase/interface/VisG4TrackPtTwig.h
• VisSimulation/VisSimBase/src/VisG4TrackPtTwig.cc

---