com.epam.deltix.util.jcg.scg.MethodImpl 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.util.jcg.*;
import java.io.IOException;
import java.lang.reflect.Modifier;
import java.util.*;

import static com.epam.deltix.util.jcg.scg.JavaSrcGenContext.*;

/**
 *
 */
 class MethodImpl
    extends CallableImpl
    implements JMethod
{
    private final String                  type;
    private final String                  name;
    private final Collection  annotations = new ArrayList<> ();

    MethodImpl(
        ClassImpl                   container,
        int                         modifiers,
        String                      type,
        String                      name
    )
    {
        super (container, modifiers, name);
        this.type = type;
        this.name = name;
    }


    @Override
    public void                     printDeclaration (SourceCodePrinter out)
        throws IOException
    {
        if ( annotations.size () > 0) {
            out.newLine ();
            for (JAnnotation annotation : annotations) {
                out.print (annotation);
                out.newLine ();
            }
        }
        super.printDeclaration (out);
    }

    @Override
    void                            printHead (SourceCodePrinter out) throws IOException {
        context.printType (type, out);
        out.print (" ", name);
    }

    @Override
    public JExpr                    call (final JExpr obj, final JExpr... args) {
        return (
            new JExprImplBase (context) {
                @Override
                public void     print (int outerPriority, SourceCodePrinter out) throws IOException {
                    out.print (obj, " (");
                    px (out, args);
                    out.print (")");
                }
            }
        );
    }

    @Override
    public JExpr                    callThis (final JExpr ... args) {
        return call(accessThis(), args);
    }

    @Override
    public JExpr                    staticCall (final JExpr... args) {
        return (
            new JExprImplBase (context) {
                @Override
                public void     print (int outerPriority, SourceCodePrinter out) throws IOException {
                    out.print (containerClass ().fullName (), ".", name (), " (");
                    px (out, args);
                    out.print (")");
                }
            }
        );
    }

    private JExpr            accessThis () {
        return (
            new JExprImplBase (context) {
                @Override
                public void print (int outerPriority, SourceCodePrinter out) throws IOException {
                    final JClass containerClass = containerClass();
                    if (containerClass == null)
                        out.print (name ());
                    else if (containerClass == out.currentClass)
                        out.print ("this.", name ());
                    else if (Modifier.isStatic (modifiers ()))
                        out.print (containerClass.fullName (), ".", name ());
                    else
                        out.print (containerClass.name (), ".this.", name ());
                }
            }
        );
    }

    @Override
    public void             addAnnotation (JAnnotation annotation) {
        annotations.add (annotation);
    }
}