source: java/main/src/main/java/com/framsticks/params/FramsClass.java @ 84

package com.framsticks.params;

import com.framsticks.params.types.StringParam;
import com.framsticks.parsers.FileSource;
import com.framsticks.parsers.Loaders;
import com.framsticks.util.lang.Casting;

import org.apache.log4j.Logger;

import java.io.InputStream;
import java.lang.reflect.*;
import java.util.*;

/**
 * The class FramsClass represents the class / schema of connected parameters
 * (such as parameters within the class). It differs from C++ version by storing
 * information about the class that parameters belong to.
 *
 * Based loosely on c++ class Param located in cpp/gdk/param.*
 *
 * @author Jarek Szymczak <name.surname@gmail.com>, Mateusz Jarus (please
 *         replace name and surname with my personal data)
 *
 * @author Piotr Sniegowski
 */
public final class FramsClass {

        private final static Logger log = Logger.getLogger(FramsClass.class.getName());

        /**
          * The Class which represents group.
          */

        /** The offset of the parameter (applied for newly added parameter). */
        protected int fieldsNumber;

        /** The groups. */
        protected List<Group> groups = new ArrayList<Group>();

        /**
         * The param entry map <parameterId, param> (for fast accessing of parameters
         * by their name)
         */
        protected Map<String, Param> paramEntryMap = new LinkedHashMap<String, Param>();

        /** The param list (for accessing parameters by offset in O(1) time. */
        protected List<Param> paramList = new ArrayList<Param>();

        /** The param getId map (for fast lookup of offset based on name */
        protected Map<String, Integer> paramIdMap = new HashMap<String, Integer>();

        protected String id;

        protected String name;

        protected String description;

        public Collection<Param> getParamEntries() {
                return paramList;
        }

        public FramsClass() {
        }

        public FramsClass(String id, String name, String description) {
                this.setId(id);
                this.setName(name);
                this.setDescription(description);
        }

        /**
         * Adds new param entry.
         *
         * @param param
         *            the new param entry
         */
        public FramsClass append(Param param) {
                paramEntryMap.put(param.getId(), param);
                //paramEntryMap.put(param.getInternalId(), param);
                paramList.add(param);
                try {
                        Group group = groups.get(param.getGroup());
                        if (group != null) {
                                group.addProperty(param);
                        }
                } catch (IndexOutOfBoundsException ignored) {

                }

                return this;
        }

        /**
         * Adds new group.
         */
        public FramsClass appendGroup(Group group) {
                groups.add(group);
                return this;
        }

        public String getDescription() {
                return description;
        }

        public int getGroupCount() {
                return groups.size();
        }

        /**
         * Gets the group member.
         *
         * @param gi
         *            the offset of group
         * @param pi
         *            the offset of member within a group
         * @return the pi-th member of group gi
         */
        public Param getGroupMember(int gi, int pi) {
                if (gi < 0 || pi < 0 || gi >= groups.size()) {
                        return null;
                }
                Group group = groups.get(gi);
                return (group != null ? group.getProperty(pi) : null);
        }

        /**
         * Gets the group getName.
         *
         * @param gi
         *            the offset of group
         * @return the group getName
         */
        public String getGroupName(int gi) {
                if (gi < 0 || gi >= groups.size())
                        return null;
                return groups.get(gi).name;
        }

        public String getId() {
                return id;
        }

        public String getName() {
                return name;
        }

        public String getNiceName() {
                return name != null ? name : id;
        }

        /**
         * Gets the param entry.
         *
         * @param i
         *            the offset of parameter
         * @return the param entry
         */
        public <T extends Param> T getParamEntry(int i, Class<T> type) {
                if (i < 0 || i >= paramList.size()) {
                        return null;
                }
                return Casting.tryCast(type, paramList.get(i));
        }

        /**
         * Gets the param entry.
         *
         * @param id
         *            the getId of parameter
         * @return the param entry
         */
        public <T extends Param> T getParamEntry(String id, Class<T> type) {
                return Casting.tryCast(type, paramEntryMap.get(id));
        }

        public int getParamCount() {
                return paramList.size();
        }

        @Override
        public String toString() {
                return id;
        }

        public static FramsClass getFramsClass() {
                return new FramsClass("class", "class", null)
                        .append(new ParamBuilder().setId("name").setName("Name").setType(StringParam.class).build())
                        .append(new ParamBuilder().setId("id").setName("id").setType(StringParam.class).build())
                        .append(new ParamBuilder().setId("desc").setName("Description").setType(StringParam.class).build());
        }

        public void setId(String id) {
                this.id = id;
        }

        public void setName(String name) {
                this.name = name;
        }

        public void setDescription(String description) {
                this.description = description;
        }

        public static String getParamTypeForNativeType(Type type) {
                if (type instanceof ParameterizedType) {
                        ParameterizedType p = (ParameterizedType) type;
                        Type rawType = p.getRawType();
                        if (rawType.equals(Map.class)) {
                                Type containedType = p.getActualTypeArguments()[1];
                                //TODO uid should be passed along during construction
                                if (containedType instanceof Class) {
                                        return "l " + ((Class<?>) containedType).getCanonicalName() + " name";
                                }
                        }
                        if (rawType.equals(List.class)) {
                                Type containedType = p.getActualTypeArguments()[0];
                                if (containedType instanceof Class) {
                                        return "l " + ((Class<?>) containedType).getCanonicalName();
                                }
                        }
                        return null;
                }

                if (type.equals(Integer.class)) {
                        return "d";
                }
                if (type.equals(String.class)) {
                        return "s";
                }
                if (type.equals(Double.class)) {
                        return "f";
                }
                if (type instanceof Class) {
                        return "o " + ((Class<?>) type).getCanonicalName();
                }
                return null;
        }

        public static final String GENERATE_HELP_PREFIX = "automatically generated from: ";

        public static FramsClass readFromStream(InputStream stream) {
                return Loaders.loadFramsClass(new FileSource(stream));
        }

        public static class Constructor {
                protected final FramsClass result;
                protected Class<?> currentClass;

                public Constructor(Class<?> src, String name) {
                        result = new FramsClass(name, name, GENERATE_HELP_PREFIX + src.toString());
                        currentClass = src;
                        while (currentClass != null) {
                                try {
                                        currentClass.getMethod("constructFramsClass", Constructor.class).invoke(null, this);
                                } catch (Exception ignored) {
                                }
                                currentClass = currentClass.getSuperclass();
                        }
                        currentClass = src;
                }

                public final FramsClass getResult() {
                        return result;
                }

                public Constructor allFields() {
                        for (Field f : currentClass.getFields()) {
                                field(f.getName());
                        }
                        return this;
                }

                public Constructor method(String name, Class<?> ... arguments) {
                        try {
                                Method method = currentClass.getMethod(name, arguments);
                                if (!Modifier.isPublic(method.getModifiers())) {
                                        return this;
                                }
                                String returnParamClass = getParamTypeForNativeType(method.getGenericReturnType());
                                if (returnParamClass == null) {
                                        return this;
                                }
                                Class<?>[] args = method.getParameterTypes();
                                if (args.length == 0) {
                                        if (method.getName().startsWith("get")) {
                                                String fieldName = method.getName().substring(3, 4).toLowerCase() + method.getName().substring(4);
                                                Param param = new ParamBuilder().setType(returnParamClass).setName(fieldName).setId(fieldName).setHelp(GENERATE_HELP_PREFIX + method.toString()).build();
                                                assert param != null;
                                                result.append(param);
                                                return this;
                                        }
                                        return this;
                                }
                                return this;
                        } catch (NoSuchMethodException e) {
                                log.fatal("method " + name + " was not found in " + currentClass.toString());
                        }
                        return this;
                }
                public Constructor field(String name) {
                        try {
                                Field field = currentClass.getField(name);
                                if (!Modifier.isPublic(field.getModifiers())) {
                                        return this;
                                }
                                String paramClass = getParamTypeForNativeType(field.getGenericType());
                                if (paramClass == null) {
                                        return this;
                                }
                                Param param = new ParamBuilder().setType(paramClass).setName(field.getName()).setId(field.getName()).setHelp(GENERATE_HELP_PREFIX + field.toString()).build();
                                assert param != null;
                                result.append(param);
                                return this;
                        } catch (NoSuchFieldException e) {
                                log.fatal("field " + name + " was not found in " + currentClass.toString());
                        }
                        return this;
                }
        }
}