com.epam.deltix.util.jcg.scg.JEnumImpl Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2021 EPAM Systems, Inc
 *
 * See the NOTICE file distributed with this work for additional information
 * regarding copyright ownership. Licensed under the Apache License,
 * Version 2.0 (the "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.  See the
 * License for the specific language governing permissions and limitations under
 * the License.
 */
package com.epam.deltix.util.jcg.scg;

import com.epam.deltix.qsrv.hf.pub.md.EnumClassDescriptor;
import com.epam.deltix.qsrv.hf.pub.md.EnumValue;
import com.epam.deltix.util.jcg.*;

import java.io.IOException;

public class JEnumImpl extends ClassImpl {

    private EnumClassDescriptor ecd;

    JEnumImpl (JContextImpl context, int modifiers, String packageName, String simpleName, String parentName) {
        super(context, modifiers, packageName, simpleName, parentName);
    }

    public JEnumImpl (JContextImpl context, int modifiers, String packageName, String simpleName, String parentName, EnumClassDescriptor ecd) {
        this (context, modifiers, packageName, simpleName, parentName);
        this.ecd = ecd;
    }

    @Override
    ClassImpl       innerClassImpl (int modifiers, String simpleName, String parentName) {
        throw new UnsupportedOperationException (); 
    }

    @Override
    ConstructorImpl newConstructor (int modifiers) {
        throw new UnsupportedOperationException (); 
    }

    @Override
    MethodImpl      createMethod (int modifiers, String typeName, String name) {
        throw new UnsupportedOperationException ();
    }    
    
    @Override
    public JExpr callSuperMethod(String name, JExpr... args) {
        return null;
    }

    @Override
    public void printDeclaration(SourceCodePrinter out) throws IOException {
        out.newLine();

        JClass save = out.currentClass;
        out.currentClass = this;

        printAnnotations (out);
        
        out.printModifiers(modifiers());
        out.print("enum ", name());

        out.print(" {");
        out.indent(1);

        final EnumValue[] values = ecd.getValues();
        if (values != null) {
            for (int i = 0; i < values.length; i++) {
                out.print(values[i].getNormalizedSymbol());
                if (i != values.length - 1) {
                    out.println(",");
                }
            }

            if (ecd.needsNormalization()) {
                out.println(";");
                // use HashMap for the lookup
                out.println("private final static java.util.HashMap map = new java.util.HashMap();");
                out.print("static {");
                out.indent(1);
                for (EnumValue value : values) {
                    out.println("map.put(\"", value.symbol, "\",", value.getNormalizedSymbol(), ");");
                }
                out.indent(-1);
                out.println("}");
                out.print(String.format("public static %1$s lookup(String value) {\n" +
                        "     final %1$s v = map.get(value);\n" +
                        "     if (v == null) throw new java.lang.IllegalArgumentException(\"No enum const \\\"\" + value + \"\\\" in %1$s\");\n" +
                        "     return v;\n" +
                        "    }", name()));
            }
        }

        out.indent(-1);
        out.println("}");

        out.currentClass = save;

    }
}