org.stanwood.nwn2.gui.parser.BasicNWN2GUIParser.java Source code

Introduction

Here is the source code for org.stanwood.nwn2.gui.parser.BasicNWN2GUIParser.java

Source

/*
 *  Copyright (C) 2008  John-Paul.Stanford <dev@stanwood.org.uk>
 *
 *  This program is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation, either version 3 of the License, or
 *  (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
package org.stanwood.nwn2.gui.parser;

import java.lang.reflect.Constructor;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.util.HashMap;
import java.util.Map;
import java.util.Map.Entry;

import org.antlr.runtime.tree.CommonTree;
import org.antlr.runtime.tree.Tree;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.stanwood.nwn2.gui.model.NWN2GUIObject;
import org.stanwood.nwn2.gui.parser.antlr.NWN2GUIAntlrParser;

public class BasicNWN2GUIParser {

    private static final String OBJECT_PACKAGE = "org.stanwood.nwn2.gui.model.";
    private final static Log log = LogFactory.getLog(BasicNWN2GUIParser.class);

    protected void addObjectsToParent(CommonTree tree, NWN2GUIObject parent) throws GUIParseException {
        switch (tree.getType()) {
        case NWN2GUIAntlrParser.TAG_CONTENTS:
            String name = getGUIObjectName(tree);
            if (!name.equalsIgnoreCase("uiscene")) {
                NWN2GUIObject obj = createGUIObject(tree, parent);
                for (int j = 0; j < tree.getChildCount(); j++) {
                    CommonTree child2 = (CommonTree) tree.getChild(j);
                    if (child2.getType() == NWN2GUIAntlrParser.TAG_CONTENTS) {
                        addObjectsToParent(child2, obj);
                    }
                }
                parent.addChildObject(obj);
            }
            break;
        default:
            // Do nothing
        }

    }

    protected NWN2GUIObject createGUIObject(CommonTree node, NWN2GUIObject parent) throws GUIParseException {
        String name = getGUIObjectName(node);
        Map<String, String> attributes = getGUIObjectAttributes(node);

        String className = getObjectClassName(name);
        try {
            Class<?> c = Class.forName(className);
            Class<? extends NWN2GUIObject> guiObjectClass = c.asSubclass(NWN2GUIObject.class);
            Constructor<? extends NWN2GUIObject> constructor = guiObjectClass.getConstructor(NWN2GUIObject.class);
            NWN2GUIObject guiObject = constructor.newInstance(parent);
            setAttributes(guiObject, attributes);
            return guiObject;
        } catch (NoClassDefFoundError e) {
            throw new GUIParseException("Unable to find UIObject class : " + className, e);
        } catch (ClassNotFoundException e) {
            throw new GUIParseException("Unable to find UIObject class : " + className, e);
        } catch (InstantiationException e) {
            e.printStackTrace();
        } catch (IllegalAccessException e) {
            e.printStackTrace();
        } catch (IllegalArgumentException e) {
            e.printStackTrace();
        } catch (InvocationTargetException e) {
            e.printStackTrace();
        } catch (SecurityException e) {
            e.printStackTrace();
        } catch (NoSuchMethodException e) {
            e.printStackTrace();
        }

        return null;
    }

    private String getObjectClassName(String name) {
        if (name.equalsIgnoreCase("uilistbox")) {
            name = "UIListBox";
        }
        return OBJECT_PACKAGE + name;
    }

    private void setAttributes(NWN2GUIObject guiObject, Map<String, String> attributes)
            throws IllegalArgumentException, IllegalAccessException, InvocationTargetException, GUIParseException {
        for (Entry<String, String> entry : attributes.entrySet()) {
            String key = entry.getKey();
            if (key.toLowerCase().startsWith("on")) {
                // TODO set callbacks
            } else {
                String methodName = "set" + key.replaceAll("_", "");
                boolean found = false;
                for (Method m : guiObject.getClass().getMethods()) {
                    if (m.getName().equalsIgnoreCase(methodName)) {
                        found = true;
                        Class<?>[] types = m.getParameterTypes();
                        Object value = AttributeValueFactory.getAttributeValue(key, guiObject, entry.getValue(),
                                types[0]);
                        m.invoke(guiObject, value);
                        break;
                    }
                }
                if (!found) {
                    log.error("Unable to find attribute '" + key + "' in object '" + guiObject.getClass() + "'");
                }
            }
        }
    }

    protected CommonTree findGUIObjectByName(Tree parent, String name) {
        for (int i = 0; i < parent.getChildCount(); i++) {
            CommonTree child = (CommonTree) parent.getChild(i);
            if (child.getType() == NWN2GUIAntlrParser.TAG_CONTENTS) {
                if (getGUIObjectName(child).equalsIgnoreCase(name)) {
                    return child;
                } else {
                    CommonTree found = findGUIObjectByName(child, name);
                    if (found != null) {
                        return found;
                    }
                }
            }
        }
        return null;
    }

    private Map<String, String> getGUIObjectAttributes(CommonTree child) {
        Map<String, String> attributes = new HashMap<String, String>();
        Tree attributesNode = child.getFirstChildWithType(NWN2GUIAntlrParser.ATTRIBUTES);
        for (int i = 0; i < attributesNode.getChildCount(); i++) {
            Tree attributeNode = attributesNode.getChild(i);
            String value = attributeNode.getChild(1).getText();
            if (value.startsWith("'") && value.endsWith("'") || value.startsWith("\"") && value.endsWith("\"")) {
                value = value.substring(1, value.length() - 1);
            }
            attributes.put(attributeNode.getChild(0).getText(), value);
        }
        return attributes;
    }

    protected String getGUIObjectName(CommonTree child) {
        CommonTree guiElement = (CommonTree) child.getFirstChildWithType(NWN2GUIAntlrParser.GENERIC_ID);
        String guiObjectName = guiElement.getText();
        return guiObjectName;
    }
}