[simantics/sysdyn.git] /

package org.simantics.sysdyn.ui.properties.widgets.expressions;\r
\r
import java.io.StringReader;\r
import java.util.ArrayList;\r
import java.util.HashMap;\r
import java.util.HashSet;\r
import java.util.List;\r
import java.util.Set;\r
\r
import org.eclipse.jface.text.Position;\r
import org.eclipse.swt.custom.StyledText;\r
import org.eclipse.swt.graphics.Color;\r
import org.eclipse.swt.widgets.TableItem;\r
import org.simantics.db.Resource;\r
import org.simantics.db.exception.DatabaseException;\r
import org.simantics.sysdyn.expressionParser.ExpressionParser;\r
import org.simantics.sysdyn.expressionParser.ParseException;\r
import org.simantics.sysdyn.expressionParser.Token;\r
import org.simantics.sysdyn.expressionParser.TokenMgrError;\r
import org.simantics.sysdyn.manager.SysdynModel;\r
import org.simantics.sysdyn.manager.SysdynModelManager;\r
import org.simantics.sysdyn.representation.Configuration;\r
import org.simantics.sysdyn.representation.IElement;\r
import org.simantics.sysdyn.representation.Variable;\r
import org.simantics.sysdyn.ui.properties.widgets.expressions.IExpression;\r
import org.simantics.ui.SimanticsUI;\r
\r
public class ExpressionUtils {\r
        \r
        /**\r
         * Determines if the given expression is a parameter expression. Parameters are numbers.\r
         * If the expression contains anything other than numbers, it is not a parameter.\r
         * \r
         * @param expression The expression to be checked\r
         * @return is the expression a parameter\r
         */\r
        static public boolean isParameter(String expression) {\r
                try {\r
                        Double.parseDouble(expression);\r
                        return true;\r
                } catch (NumberFormatException e) {\r
                        return false;\r
                }\r
        }\r
\r
    static public void validateExpressionFields(IExpression expression, TableItem[] connectedVariables, Resource configuration) {\r
        ExpressionParser parser = new ExpressionParser(new StringReader(""));\r
        Set<String> variables = new HashSet<String>();\r
        HashMap<ExpressionField, HashMap<String, List<Token>>> references = new HashMap<ExpressionField, HashMap<String, List<Token>>>();\r
\r
\r
        // Build references and variable array\r
        for(ExpressionField ef : expression.getExpressionFields()) {\r
            ef.resetAnnotations();\r
            String textString = ef.getExpression();\r
            parser.ReInit(new StringReader(textString));\r
            try {\r
                parser.expr();\r
                HashMap<String, List<Token>> cr = parser.getReferences();\r
                references.put(ef, cr);\r
                for(String t : cr.keySet())\r
                    variables.add(t);\r
            } catch (ParseException e1) {\r
                ef.setSyntaxError(e1.currentToken);\r
            } catch (TokenMgrError err) {\r
                ef.setSyntaxError(0, textString.length(), "MissingLink", "Expression contains unsupported characters");\r
            }\r
        }\r
\r
        // Remove variables from variable array that don't exist in the model. Create annotations\r
        if(!variables.isEmpty()) {\r
            ArrayList<String> modelVariables = new ArrayList<String>();\r
            Set<String> noSuchVariables = new HashSet<String>();\r
            SysdynModelManager sdm = SysdynModelManager.getInstance(SimanticsUI.getSession());\r
            SysdynModel model = sdm.getModel(configuration);\r
            try {\r
                model.update();\r
            } catch (DatabaseException e1) {\r
                e1.printStackTrace();\r
            }\r
            Configuration conf = model.getConfiguration();\r
            ArrayList<IElement> elements = conf.getElements();\r
            for(IElement e : elements) {\r
                if(e instanceof Variable) {\r
                    Variable v = (Variable) e;\r
                    modelVariables.add(v.getName());\r
                }\r
            }\r
\r
\r
            for(String v : variables) {\r
                if(!modelVariables.contains(v)) {\r
                    noSuchVariables.add(v);\r
                }\r
            }\r
\r
            if(!noSuchVariables.isEmpty()) {\r
                // remove no such variables from variable list\r
                for(String s : noSuchVariables)\r
                    variables.remove(s);\r
                // create annotations\r
                HashMap<ExpressionField ,ArrayList<Position>> positions = getPositionsForVariables(references, noSuchVariables);\r
                for(ExpressionField ef : positions.keySet()) {\r
                    ef.setNoSuchVariableAnnotations(positions.get(ef));\r
                }\r
            }      \r
        }\r
\r
        // Check that the variables that exist have connections and the connected variables have references in the expressions\r
        if(!(expression instanceof StockExpression)) { \r
            for(TableItem ti : connectedVariables) {\r
                if(!variables.contains(ti.getText())) {\r
                    ti.setForeground(new Color(ti.getDisplay(), 255, 0, 0));\r
                } else {\r
                    ti.setForeground(new Color(ti.getDisplay(), 0, 0, 0));\r
                    variables.remove(ti.getText());\r
                }\r
            }\r
\r
            if(!variables.isEmpty()) {\r
                HashMap<ExpressionField ,ArrayList<Position>> positions = getPositionsForVariables(references, variables);\r
                for(ExpressionField ef : positions.keySet()) {\r
                    ef.setMissingLinkAnnotations(positions.get(ef));\r
                }\r
\r
            }\r
        } \r
\r
    }\r
\r
    @SuppressWarnings("unchecked")\r
    static private HashMap<ExpressionField ,ArrayList<Position>> getPositionsForVariables(HashMap<ExpressionField, HashMap<String, List<Token>>> references, Set<String> variables) {\r
        HashMap<ExpressionField ,ArrayList<Position>> result = new HashMap<ExpressionField ,ArrayList<Position>>();\r
        for(String s : variables) {\r
            List<Token> tlist = new ArrayList<Token>();\r
            for(ExpressionField ef : references.keySet()) {\r
                ArrayList<Position> positions = new ArrayList<Position>();\r
                tlist = references.get(ef).get(s);\r
                if(tlist != null)\r
                    for(Token t : tlist) {\r
                        StyledText st = ef.getSourceViewer().getTextWidget();\r
                        int start =  st.getOffsetAtLine(t.beginLine - 1) + t.beginColumn - 1;\r
                        int offset = st.getOffsetAtLine(t.endLine - 1) + t.endColumn - start;\r
                        positions.add(new Position(\r
                                start,\r
                                offset));\r
                    }\r
                if(result.keySet().contains(ef)) {\r
                    result.get(ef).addAll((ArrayList<Position>)positions.clone());\r
                } else {\r
                    result.put(ef, (ArrayList<Position>)positions.clone());\r
                }\r
            }\r
        }\r
        return result;\r
    }\r
}\r