source: java2python/src/main/java/tudelft/utilities/j2p/t/Stub.java@ 363

package tudelft.utilities.j2p.t;

import java.util.Collections;

import com.github.javaparser.ast.expr.FieldAccessExpr;
import com.github.javaparser.ast.expr.MethodCallExpr;

import tudelft.utilities.j2p.formatting.ExpressionLine;

/**
 * a class that implements J2P translator given an arbitrary java class. The
 * translator assumes all functions in the given class can be accessed and will
 * be translated properly to the same package name in python. System classes are
 * translated using the t.java translateors.
 *
 */
public class Stub extends tudelft.utilities.j2p.t.java.lang.Object {
        private Class<?> clazz;

        /**
         * 
         * @param clazz the class to be stubbed.
         */
        public Stub(Class<?> clazz) {
                this.clazz = clazz;
        }

        @Override
        public ExpressionLine call(MethodCallExpr call) {

                // forward the translation of standard calls to Object
                switch (call.getName().asString()) {
                case "toString":
                case "hashCode":
                case "equals":
                        return super.call(call);
                }

                return Translator.callClassFunction(call);

        }

        @Override
        public ExpressionLine getName() {
                return new ExpressionLine(clazz.getSimpleName(),
                                Collections.emptyList(),
// HACK do not import the main class that is being defined.
// Hack because there are probably other Stubbed objects that DO need an import.
//                              Arrays.asList("from " + clazz.getCanonicalName() + " import "
//                                              + clazz.getSimpleName()),
                                false, Collections.emptyList());
        }

        @Override
        public ExpressionLine field(FieldAccessExpr node) {
                return getName().withAppend("." + node.getNameAsString());
        }

}