VisClient/org/hfbk/vis/visnode/VisProfiler.java

Go to the documentation of this file.

package org.hfbk.vis.visnode;

import java.util.Map;

import org.dronus.gl.GLFont;
import org.dronus.gl.GLSphereRenderer;
import org.dronus.gl.GLTextPanel;
import org.dronus.gl.GLUtil;
import org.dronus.graph.Edge;
import org.dronus.graph.Graph;
import org.dronus.graph.Node;
import org.hfbk.util.Counter;
import org.hfbk.util.Sleeper;
import org.lwjgl.opengl.GL11;
import org.lwjgl.util.vector.Vector3f;

public class VisProfiler extends VisSet {
        
//      final int DELAY=10;  
        final int DELAY=1000;  
        
        Graph profile=new Graph();

        Counter<Integer> callCounter=new Counter<Integer>();
        
        //if something was sampled but not yet visualised.
        boolean dirty=true;
        
        class VisCallBall extends VisNodeDraggable {
                int dl;
                GLTextPanel helpText;           
                Node node;
                boolean isTriggered=true; 
                
                VisCallBall(Node n, Vector3f position) {
                        super(n, position);
                        this.node=n;
                        w=h=radius=1.6f;
                        helpText=new GLTextPanel(n.text,0,0);
                }
                
                void renderSelf() {
                        Integer count=callCounter.get(node.id);
                        if (count!=null){
                                radius=.3f+10*(float)Math.sqrt(count/(float)callCounter.total);
                                w=h=2*radius;
                        }
                        if (isHoovered || isTriggered)                          
                                GL11.glColor3f(1,1,1);
                        else if (node.type.equals("Thread"))
                                GL11.glColor3f(1,1,.4f);                        
                        else if (node.type.equals("Call")){
                                boolean isOur=node.text.startsWith("org.dronus")||node.text.startsWith("org.hfbk");
                                GL11.glColor4f(1,0,0,(isOur ? 1 : .5f ));
                        }
                                
                        GLSphereRenderer.renderSphere(radius);
                        
                        //if (isHoovered || radius>=1.6f) { //describe "fat" calls unhoovered also              
                        if (isHoovered) {
                                GL11.glTranslatef(radius,radius,0);
                                GLUtil.billboardCylinder();
                                GL11.glScalef(1,2,1);
                                GLFont.getDefault().render();                           
                                helpText.render();
                                GL11.glDisable(GL11.GL_TEXTURE_2D);
                        }
                        
                        isTriggered=false;
                }
        }
        
        VisNode create(Node node, Vector3f pos) {
                return new VisCallBall(node, pos);
        }
        
        public VisProfiler(Node dummy, Vector3f position) {     
                super(null, position);
                                
                profile.addNode(new Node("0","Vis/Client","Set"));
                
                Thread profiler=new Thread(){                   
                        public void run() {
                                setName("VisProfiler");
                                while(true){
                                        synchronized(VisProfiler.this){ profile();      };
                                        Sleeper.sleep(DELAY);
                                }
                        }
                };                              
                profiler.start();
        }
        
        void profile(){
                Node rootset=profile.getRoot();
                
                //for all existing threads...
                for(Map.Entry<Thread,StackTraceElement[]> trace : Thread.getAllStackTraces().entrySet()){
                        
                        String threadName=trace.getKey().getName();
                        int threadId=threadName.hashCode();
                        Node threadNode=profile.findNode(threadId);
                        if(threadNode==null) {
                                threadNode=new Node(threadId, threadName, "Thread");
                                profile.addNode(threadNode);
                                profile.addEdge(new Edge(threadNode,rootset, null, "in"));                                      
                        }
                        StackTraceElement[] stes=trace.getValue();
                        
                        //reverse stack trace
                        for (int i=0; i<stes.length/2; i++){
                                StackTraceElement tmp=stes[i];
                                stes[i]=stes[stes.length-i-1];
                                stes[stes.length-i-1]=tmp;
                        }
                        
                        //for all elements of current thread's call stack...
                        Node last=null;
                        for (StackTraceElement ste: stes){
                                //id by thread too to show threads independently
                                int id=threadId^ste.getClassName().hashCode()^ste.getMethodName().hashCode();
                                
                                callCounter.add(id); //count method sample
                                
                                Node n=profile.findNode(id); //know this method already?
                                
                                if (n==null){
                                        n=new Node(id, ste.getClassName()+"."+ste.getMethodName(),"Call");
                                        profile.addNode(n);                                     
                                        dirty=true; //mark for update
                                }       
                                profile.addEdge(new Edge(n,rootset, null, "in"));
                                
                                if (last!=null) //link to caller method 
                                        profile.addEdge(new Edge(last,n, null, ""));
                                else
                                        profile.addEdge(new Edge(n,threadNode, null, "in"));
                                last=n;
                        }       
                        
                        // flash active node
                        VisNode vn=nodes.get(last);
                        if (vn instanceof VisCallBall)
                                ((VisCallBall)vn).isTriggered=true;
                }
        }
        
        void renderSelf() {
                super.renderSelf();
                //everytime the profiler added something to
                //profile graph, we update the visualisation.
                if (dirty) synchronized(this){
                        Node root=profile.getRoot();
                        update(root);
                        
                        dirty=false;
                }
        }
}

---