MetricVarMap.java

Go to the documentation of this file.

package edu.rice.cs.hpc.data.experiment.metric;

import com.graphbuilder.math.Expression;
import com.graphbuilder.math.ExpressionTree;
import com.graphbuilder.math.FuncMap;
import com.graphbuilder.math.VarMap;
import com.graphbuilder.math.func.Function;

import edu.rice.cs.hpc.data.experiment.Experiment;
import edu.rice.cs.hpc.data.experiment.scope.RootScope;
import edu.rice.cs.hpc.data.experiment.scope.Scope;

public class MetricVarMap extends VarMap {

    private Experiment experiment;
    private Scope scope;

    public MetricVarMap() {
        super(false);
    }

    
    public MetricVarMap(Experiment exp) {
        super(false);
        this.experiment = exp;
    }
    
    public MetricVarMap(Scope s, Experiment exp) {
        super(false);
        this.scope = s;
        this.experiment = exp;
    }
    
    public MetricVarMap(boolean caseSensitive) {
        super(caseSensitive);
    }

    //===========================
    

    public void setExperiment(Experiment exp) {
        this.experiment = exp;
    }
    
    public void setScope(Scope s) {
        this.scope = s;
    }
    
    public double getValue(String varName) {
        assert(experiment != null);
        
        if(varName.startsWith("$")) {
            // Metric variable
            String sIndex = varName.substring(1);
            BaseMetric metric = this.experiment.getMetric(sIndex);
            if (metric == null) 
                throw new RuntimeException("metric doesn't exist: " + sIndex);
            if (scope != null) {
                MetricValue value = metric.getValue(scope);
                if(MetricValue.isAvailable(value))
                    return MetricValue.getValue(value);
            }
            return 0.0;
            
        } else if (varName.startsWith("@")) {
            //---------------------------------------------------------
            // 2011.02.08: new interpretation of the symbol "@x" where x is the metric ID
            // @x returns the aggregate value of metric x 
            //---------------------------------------------------------
            String sIndex = varName.substring(1);

            try{
                BaseMetric metric = this.experiment.getMetric(sIndex);
                if (metric == null)
                    throw new RuntimeException("Unrecognize metric ID: " + varName);

                RootScope root = (RootScope) this.experiment.getRootScopeChildren()[0];
                return MetricValue.getValue(metric.getValue(root));

            } catch (java.lang.Exception e) {
                throw new RuntimeException("Unrecognize variable: " + varName);
            }
        } else
            return super.getValue(varName);
    }
    
    
    public static void main(String[] args) {
        String s = "@1*r^2";
        Expression x = ExpressionTree.parse(s);

        MetricVarMap vm = new MetricVarMap(false /* case sensitive */);
        vm.setValue("r", 5);

        FuncMap fm = new FuncMap(); // no functions in expression
        fm.loadDefaultFunctions();
        System.out.println(x); 
        System.out.println(x.eval(vm, fm)); 

        vm.setValue("r", 10);
        System.out.println(x.eval(vm, fm)); 
        Function []fs = fm.getFunctions();
        for( int i=0; i<fs.length ; i++) {
            System.out.println("\t<tr><td>" + " <code> " + fs[i].toString() + " </code> </td> <td></td> </tr>");
        }
    }

}