org.openl.codegen.tools.EnumHelper Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of org.openl.rules.gen Show documentation

OpenL Code generation project

There is a newer version: 5.19.9

package org.openl.codegen.tools;

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

import org.openl.codegen.tools.type.EnumerationDescriptor;
import org.openl.rules.enumeration.properties.EnumPropertyDefinition;
import org.openl.types.IOpenClass;
import org.openl.types.IOpenField;

public class EnumHelper {

    public static String getEnumName(String sourceName) {

        return String.format("%s%sEnum", sourceName.substring(0, 1).toUpperCase(), sourceName.substring(1));
    }

    public static List findEnumerationFields(IOpenClass openClass) {

        List enumerations = new ArrayList();

        for (IOpenField field : openClass.getFields().values()) {
            if (isEnumeration(field)) {
                enumerations.add(field);
            }
        }

        return enumerations;
    }

    public static boolean isEnumeration(IOpenField field) {

        IOpenClass type = field.getType();
        Class clazz = type.getInstanceClass();

        return clazz.equals(EnumPropertyDefinition[].class);
    }

    public static EnumerationDescriptor createDescriptor(String enumName, EnumPropertyDefinition[] values) {

        EnumerationDescriptor descriptor = new EnumerationDescriptor();
        descriptor.setEnumName(enumName);
        descriptor.setValues(values);

        return descriptor;
    }

}