[simantics/sysdyn.git] /

/*******************************************************************************\r
 * Copyright (c) 2007, 2011 Association for Decentralized Information Management in\r
 * Industry THTH ry.\r
 * All rights reserved. This program and the accompanying materials\r
 * are made available under the terms of the Eclipse Public License v1.0\r
 * which accompanies this distribution, and is available at\r
 * http://www.eclipse.org/legal/epl-v10.html\r
 *\r
 * Contributors:\r
 *     VTT Technical Research Centre of Finland - initial API and implementation\r
 *******************************************************************************/\r
package org.simantics.sysdyn.ui.elements2.profiles;\r
\r
import java.awt.geom.AffineTransform;\r
import java.awt.geom.Rectangle2D;\r
import java.util.ArrayList;\r
\r
import org.simantics.databoard.Bindings;\r
import org.simantics.db.ReadGraph;\r
import org.simantics.db.Resource;\r
import org.simantics.db.common.request.ObjectsWithType;\r
import org.simantics.db.exception.DatabaseException;\r
import org.simantics.db.layer0.variable.Variable;\r
import org.simantics.diagram.G2DUtils;\r
import org.simantics.diagram.elements.TextNode;\r
import org.simantics.diagram.profile.StyleBase;\r
import org.simantics.diagram.stubs.DiagramResource;\r
import org.simantics.diagram.stubs.G2DResource;\r
import org.simantics.diagram.synchronization.graph.DiagramGraphUtil;\r
import org.simantics.modeling.ModelingResources;\r
import org.simantics.project.IProject;\r
import org.simantics.scenegraph.INode;\r
import org.simantics.scenegraph.g2d.nodes.ShapeNode;\r
import org.simantics.scenegraph.g2d.nodes.SingleElementNode;\r
import org.simantics.scenegraph.profile.EvaluationContext;\r
import org.simantics.scenegraph.profile.Observer;\r
import org.simantics.scenegraph.profile.common.ProfileVariables;\r
import org.simantics.scenegraph.utils.GeometryUtils;\r
import org.simantics.scenegraph.utils.NodeUtil;\r
import org.simantics.simulation.experiment.IExperiment;\r
import org.simantics.simulation.project.IExperimentManager;\r
import org.simantics.sysdyn.adapter.SysdynVariableProperties;\r
import org.simantics.sysdyn.adapter.VariableRVIUtils;\r
import org.simantics.sysdyn.manager.SysdynDataSet;\r
import org.simantics.sysdyn.manager.SysdynPlaybackExperiment;\r
import org.simantics.ui.SimanticsUI;\r
import org.simantics.utils.datastructures.Triple;\r
import org.simantics.utils.ui.color.Color;\r
import org.simantics.utils.ui.color.ColorGradient;\r
import org.simantics.utils.ui.color.ColorValue;\r
\r
/**\r
 * Profile style definition for simulation playback mode. Works only with SimulationPlaybackExperiment\r
 * \r
 * @author Teemu Lempinen\r
 *\r
 */\r
public class SimulationPlaybackStyle extends StyleBase<Triple<AffineTransform, Double, ColorGradient>> {\r
\r
    Resource gradientResource;\r
    ColorGradient cg;\r
    byte[] gradient;\r
\r
    /**\r
     * Determine if style needs to be redrawn and return objects that are needed to redraw the style.\r
     * \r
     * @return All necessary components that are needed to draw this style\r
     */\r
    @Override\r
    public Triple<AffineTransform, Double, ColorGradient> calculateStyle(ReadGraph graph, Resource runtimeDiagram, Resource entry, Resource element, Variable configuration) throws DatabaseException {\r
\r
        // Find SimulationPlaybackExperiment\r
        IProject project = SimanticsUI.getProject();\r
        IExperimentManager em = project.getHint(IExperimentManager.KEY_EXPERIMENT_MANAGER);\r
        IExperiment experiment = em.getActiveExperiment();\r
        if(!(experiment instanceof SysdynPlaybackExperiment))\r
            return null;\r
\r
        ModelingResources mr = ModelingResources.getInstance(graph);\r
        DiagramResource dr = DiagramResource.getInstance(graph);\r
\r
        ColorGradient cg = null;\r
\r
        Resource component = graph.getPossibleObject(element, mr.ElementToComponent);\r
        if (component == null)\r
            return null;\r
\r
        try {\r
\r
            // Find Variable for this component\r
            String dv = graph.getPossibleRelatedValue(runtimeDiagram, dr.RuntimeDiagram_HasVariable);\r
            Variable rootVariable = graph.getPossibleAdapter(graph.getRootLibrary(), Variable.class);\r
            if (rootVariable == null) \r
                return null;\r
            Variable diagramVariable = rootVariable.browsePossible(graph, dv.substring(6));\r
            if(diagramVariable == null) \r
                return null;\r
\r
            Variable var = diagramVariable.browsePossible(graph, component);\r
            if(var == null)\r
                return null;\r
\r
            // Get simulation result values for this component\r
            // Get values\r
            Object object =  var.getPossiblePropertyValue(graph, SysdynVariableProperties.ACTIVE_DATASETS , Bindings.VARIANT);\r
            if(object == null || !(object instanceof ArrayList<?>))\r
                return null;\r
\r
            ArrayList<SysdynDataSet> datasets = new ArrayList<SysdynDataSet>();\r
\r
            for(Object o : (ArrayList<?>)object) {\r
                if(o instanceof SysdynDataSet)\r
                    datasets.add((SysdynDataSet)o);\r
            }\r
\r
            if(datasets.size() == 0)\r
                return null;\r
\r
            if(datasets.size() > 1) {\r
                String range = datasets.get(0).name.substring(\r
                        datasets.get(0).name.indexOf('[') + 1,\r
                        datasets.get(0).name.indexOf(']'));\r
                int size = range.split(",").length;\r
                String[] filter = new String[size];\r
                for(int i = 0; i < size; i++)\r
                    filter[i] = "Sum";\r
\r
                ArrayList<SysdynDataSet> result2 = VariableRVIUtils.getDataset(datasets, filter);\r
                if(result2 != null) {\r
                    datasets = result2;\r
                }\r
\r
            }\r
\r
            // The datasets list should always be size == 1\r
            SysdynDataSet dataset = datasets.get(0);\r
            \r
            // Get values (or sum of values)\r
            double[] va = dataset.values;\r
            if(va.length < 2)\r
                return null;\r
\r
            // Get simulation timesteps for this component\r
            double[] ta = dataset.times;\r
            if(ta == null || ta.length < 2)\r
                return null;\r
\r
            if(va.length == 0 || va.length == 2)\r
                return null;\r
\r
            // Get playback time from the experiment\r
            Double time = var.getPossiblePropertyValue(graph, SysdynVariableProperties.TIME , Bindings.DOUBLE);\r
            if(time == null)\r
                return null;\r
\r
            // Find minimum and maximum values for this variable\r
            double min = va[0], max = va[0];\r
            for(double d : va) {\r
                if(d < min) \r
                    min = d;\r
                if(d > max)\r
                    max = d;\r
            }\r
\r
            // Find the index in time and value arrays for this time \r
            // (time steps may vary, we need to loop to find the correct time) \r
            int index = 0;\r
            if(ta[ta.length - 1] - time > ta[ta.length / 2] ) {\r
                index = ta.length - 1;\r
                while(ta[index] > time && index > 0)\r
                    index--;\r
            } else {\r
                while(ta[index] < time && index < ta.length - 1)\r
                    index++;\r
            }\r
\r
            double value = va[index];\r
\r
            // Calculate where the value is located between min and max values [0..1].   \r
            double multiplier = (value - min) / (max - min);\r
\r
\r
            // Get the transform of this element\r
            AffineTransform at = (AffineTransform) DiagramGraphUtil.getAffineTransform(graph, element,  G2DResource.getInstance(graph).HasTransform, true).clone();\r
\r
            // Find the gradient used to draw this style\r
            G2DResource g2d = G2DResource.getInstance(graph);\r
            Resource gradient = graph.getPossibleObject(experiment.getResource(), g2d.HasColorGradient);\r
            if(this.gradientResource == null || !this.gradientResource.equals(gradient)) {\r
                ArrayList<ColorValue> colorValues = new ArrayList<ColorValue>();\r
                for(Resource placement : graph.syncRequest(new ObjectsWithType(gradient, g2d.HasColorPlacement, g2d.ColorPlacement))) {\r
                    Double position = graph.getPossibleRelatedValue(placement, g2d.HasGradientPosition, Bindings.DOUBLE);\r
                    Resource rColor = graph.getPossibleObject(placement, g2d.HasColor);\r
                    if (rColor != null) {\r
                        colorValues.add(new ColorValue(new Color((java.awt.Color) G2DUtils.getObject(graph, rColor)), position));\r
                    }\r
                }\r
                cg = new ColorGradient(colorValues, ColorGradient.HSV);\r
            } else {\r
                cg = this.cg;\r
            }\r
\r
            return new Triple<AffineTransform, Double, ColorGradient>(at, multiplier, cg);\r
\r
        } catch(Exception ignore) {\r
            ignore.printStackTrace();\r
        }\r
        return null;\r
    }\r
\r
    @Override\r
    public void styleResultChanged(Observer observer, Resource element, Triple<AffineTransform, Double, ColorGradient> result) {\r
        if (result != null)\r
            values.put(element, result);\r
        else\r
            values.remove(element);\r
        observer.update();\r
    }\r
\r
    /**\r
     * Apply style\r
     */\r
    @Override\r
    public void applyStyleForNode(EvaluationContext observer, INode _node, Triple<AffineTransform, Double, ColorGradient> result) {\r
        Double multiplier;\r
        if (result != null && (multiplier = result.second) != null && !multiplier.isNaN()) {\r
\r
            // Create a node that will show the style effect\r
            A node = ProfileVariables.claimChild(_node, "", "playbackColour", A.class, observer);\r
            if (node == null)\r
                return;\r
\r
            AffineTransform at = result.first;\r
\r
            if(this.cg == null || !this.cg.equals(result.third)) {\r
                this.cg = result.third;\r
                this.gradient = cg.getGradientArray(101);\r
            }\r
\r
            // Get integer values for red, green and blue\r
            int i = (int)(multiplier * 100);\r
            int r = (int)(gradient[i * 3 + 0]  & 0xff);\r
            int g = (int)(gradient[i * 3 + 1]  & 0xff);\r
            int b = (int)(gradient[i * 3 + 2]  & 0xff);\r
            // Get color with r, g and b\r
            java.awt.Color c = new java.awt.Color(r, g, b, 200);\r
\r
            // Create a node for the style\r
            INode n = NodeUtil.getNearestChildByClass((SingleElementNode)_node, TextNode.class);\r
            Rectangle2D expandedElementBounds = GeometryUtils.expandRectangle( NodeUtil.getLocalElementBounds(n), 0.0);\r
            node.setFill(true);\r
            node.setColor(c);\r
            node.setStroke(null);\r
            node.setValue("shape", expandedElementBounds);\r
            node.setTransform(at);\r
\r
            // Find z-index for the TextNode associated with the element and place the style node just below that text node \r
            int zIndex = -1;\r
            if(n != null) {\r
                at = ((TextNode)n).getTransform();\r
                zIndex = ((TextNode)n).getZIndex() - 1;\r
            } else {\r
                n = _node;\r
            }\r
            node.setZIndex(zIndex);\r
\r
\r
        } else {\r
            cleanupStyleForNode(_node);\r
        }\r
    }\r
\r
    @Override\r
    protected void cleanupStyleForNode(INode node) {\r
        ProfileVariables.denyChild(node, "", "playbackColour");\r
    }\r
\r
\r
    /**\r
     * Dummy class for displaying the style\r
     * \r
     * @author Teemu Lempinen\r
     *\r
     */\r
    public static class A extends ShapeNode {\r
\r
        private static final long serialVersionUID = -5273246617906214956L;\r
\r
        @Override\r
        public Rectangle2D getBoundsInLocal() {\r
            return null;\r
        }\r
\r
        @Override\r
        public Rectangle2D getBoundsInLocal(boolean b) {\r
            return null;\r
        }\r
\r
        @Override\r
        public Rectangle2D getBounds() {\r
            return null;\r
        }\r
\r
    }\r
}\r