source: java/Framclipse/com.framsticks.framclipse.ui/src/com/framsticks/framclipse/ui/outline/FramScriptTransformer.java @ 437

/*
 * generated by Xtext
 */
package com.framsticks.framclipse.ui.outline;

import java.util.ArrayList;
import java.util.List;

import org.eclipse.emf.common.util.EList;
import org.eclipse.emf.ecore.EObject;
import org.eclipse.xtext.ui.editor.outline.transformer.AbstractDeclarativeSemanticModelTransformer;

import com.framsticks.framclipse.framScript.Code;
import com.framsticks.framclipse.framScript.Function;
import com.framsticks.framclipse.framScript.Model;
import com.framsticks.framclipse.framScript.Property;
import com.framsticks.framclipse.framScript.State;
import com.framsticks.framclipse.framScript.VariableDeclarations;

/**
 * customization of the default outline structure
 */
public class FramScriptTransformer extends AbstractDeclarativeSemanticModelTransformer {

        public List<EObject> getChildren(Function e) {
                return NO_CHILDREN;
        }
        
        public List<EObject> getChildren(Model e) {
                List<EObject> children = new ArrayList<EObject>();
                for (EObject child : e.eContents()) {
                        if (child instanceof Code) {
                                addCodeContents(children, (Code) child);
                        } else if (child instanceof Property
                                        || child instanceof State) {
                                children.add(child);
                        }
                }
                return children;
        }

        private void addCodeContents(List<EObject> children, Code code) {
                if (code != null) {
                        children.addAll(code.getIncludes());
                        EList<VariableDeclarations> globals = code.getGlobals();
                        for (VariableDeclarations declarations : globals) {
                                children.addAll(declarations.getVars());
                        }
                        children.addAll(code.getFunctions());
                }
        }
}