source: java/FramclipsePlugin/src/main/java/com/framsticks/framclipse/editors/outline/FramclipseEditorOutlineContentProvider.java @ 193

package com.framsticks.framclipse.editors.outline;

import java.util.List;

import org.eclipse.jface.text.BadPositionCategoryException;
import org.eclipse.jface.text.DefaultPositionUpdater;
import org.eclipse.jface.text.IDocument;
import org.eclipse.jface.text.IPositionUpdater;
import org.eclipse.jface.viewers.ITreeContentProvider;
import org.eclipse.jface.viewers.Viewer;

import com.framsticks.framclipse.editors.FramclipseEditor;
import com.framsticks.framclipse.internal.parser.ASTFunction;
import com.framsticks.framclipse.internal.parser.ASTGlobalDecl;
import com.framsticks.framclipse.internal.parser.ASTIdentList;
import com.framsticks.framclipse.internal.parser.Node;


public class FramclipseEditorOutlineContentProvider implements ITreeContentProvider
{
        private Node root;
        private FramclipseEditor editor;

        protected final static String FRAMCLIPSE_ELEM_POSITIONS = "__framclipse_elem_positions";
        protected IPositionUpdater positionUpdater = new DefaultPositionUpdater(FRAMCLIPSE_ELEM_POSITIONS);
        
        public FramclipseEditorOutlineContentProvider(FramclipseEditor framsEditor)
        {
                this.editor = framsEditor;
        }
        
        public Object[] getChildren(Object parentElement) {
                
                if(parentElement instanceof Node)
                {
                        Node node = (Node)parentElement;
                        
                        if(node instanceof ASTFunction)
                        {
                                if(node.jjtGetNumChildren() > 1)
                                {
                                        ASTIdentList params = (ASTIdentList) node.jjtGetChild(1);
                                        List<String> names = params.getIdents();
                                        
                                        Node[] children = new Node[names.size()];
                                        
                                        for(int l = 0; l < names.size(); ++l)
                                                children[l] = new IdentifierNode(names.get(l), "Parameter", node);
                                        
                                        return children;
                                }
                                return new Object[0];
                        }
                        if(node instanceof ASTGlobalDecl)
                        {
                                if(node.jjtGetNumChildren() > 0)
                                {
                                        ASTIdentList params = (ASTIdentList) node.jjtGetChild(0);
                                        List<String> names = params.getIdents();
                                        
                                        Node[] children = new Node[names.size()];
                                        
                                        for(int l = 0; l < names.size(); ++l)
                                                children[l] = new IdentifierNode(names.get(l), "Variable", node);
                                        
                                        return children;
                                }
                                return new Object[0];
                        }
                        
                        if(node.jjtGetNumChildren() > 0)
                        {
                                Node[] children = new Node[node.jjtGetNumChildren()];
                                
                                for(int k = 0; k < node.jjtGetNumChildren(); ++k)
                                        children[k] = node.jjtGetChild(k);
                                
                                return children;
                        }
                }
                

                return new Object[0];
        }

        public Object getParent(Object element) {

                if(element instanceof Node)
                {
                        return ((Node)element).jjtGetParent();
                }
                return null;
        }

        public boolean hasChildren(Object element) {
                
                if(element instanceof Node)
                {
                        if(element instanceof ASTFunction)
                        {
                                return ((Node)element).jjtGetNumChildren() > 1;
                        }
                        
                        return ((Node)element).jjtGetNumChildren() > 0;
                }
                
                return false;
        }

        public Object[] getElements(Object inputElement) {
                if (root == null)
                        return new Object[0];
                if(root.jjtGetNumChildren() > 0)
                {
                        Node[] children = new Node[root.jjtGetNumChildren()];
                        
                        for(int k = 0; k < root.jjtGetNumChildren(); ++k)
                                children[k] = root.jjtGetChild(k);
                        
                        return children;
                }
                
                return new Object[0];
        }

        public void dispose() {
                // TODO Auto-generated method stub
                
        }

        public void inputChanged(Viewer viewer, Object oldInput, Object newInput) {
                if (oldInput != null)
                {
                        IDocument document = editor.getDocumentProvider().getDocument(oldInput);
                        if (document != null)
                        {
                                try
                                {
                                        document.removePositionCategory(FRAMCLIPSE_ELEM_POSITIONS);
                                }
                                catch (BadPositionCategoryException x)
                                {
                                }
                                document.removePositionUpdater(positionUpdater);
                        }
                }
                
                root =  (Node) newInput;

                if (newInput != null)
                {
                        IDocument document = editor.getDocumentProvider().getDocument(newInput);
                        if (document != null)
                        {
                                document.addPositionCategory(FRAMCLIPSE_ELEM_POSITIONS);
                                document.addPositionUpdater(positionUpdater);
                                
                        }
                }
        }

}