support int.__pow__ with 3 arguments

Author: timfel

graalpython/com.oracle.graal.python/src/com/oracle/graal/python/builtins/objects/ints/IntBuiltins.java

@@ -49,6 +49,8 @@
 import com.oracle.graal.python.builtins.objects.bytes.PBytes;
 import com.oracle.graal.python.builtins.objects.type.PythonClass;
 import com.oracle.graal.python.nodes.SpecialMethodNames;
+import com.oracle.graal.python.nodes.call.special.LookupAndCallBinaryNode;
+import com.oracle.graal.python.nodes.call.special.LookupAndCallTernaryNode;
 import com.oracle.graal.python.nodes.function.PythonBuiltinBaseNode;
 import com.oracle.graal.python.nodes.function.PythonBuiltinNode;
 import com.oracle.graal.python.nodes.function.builtins.PythonBinaryBuiltinNode;
@@ -60,6 +62,7 @@
 import com.oracle.graal.python.runtime.exception.PythonErrorType;
 import com.oracle.truffle.api.CompilerDirectives;
 import com.oracle.truffle.api.CompilerDirectives.TruffleBoundary;
+import com.oracle.truffle.api.dsl.Cached;
 import com.oracle.truffle.api.dsl.Fallback;
 import com.oracle.truffle.api.dsl.GenerateNodeFactory;
 import com.oracle.truffle.api.dsl.NodeFactory;
@@ -607,6 +610,10 @@ abstract static class RMulNode extends MulNode {
     @TypeSystemReference(PythonArithmeticTypes.class)
     abstract static class PowNode extends PythonTernaryBuiltinNode {
 
+        protected static PowNode create() {
+            return null;
+        }
+
         @Specialization(guards = "right >= 0", rewriteOn = ArithmeticException.class)
         int doIntegerFast(int left, int right, @SuppressWarnings("unused") PNone none) {
             int result = 1;
@@ -688,6 +695,28 @@ PInt doPInt(PInt left, PInt right, @SuppressWarnings("unused") PNone none) {
             return factory().createInt((long) value);
         }
 
+        @Specialization
+        Object modulo(Object x, Object y, long z,
+                        @Cached("create(__POW__)") LookupAndCallTernaryNode powNode,
+                        @Cached("create(__MOD__)") LookupAndCallBinaryNode modNode) {
+            Object result = powNode.execute(x, y, PNone.NO_VALUE);
+            if (result == PNotImplemented.NOT_IMPLEMENTED) {
+                return result;
+            }
+            return modNode.executeObject(result, z);
+        }
+
+        @Specialization
+        Object modulo(Object x, Object y, PInt z,
+                        @Cached("create(__POW__)") LookupAndCallTernaryNode powNode,
+                        @Cached("create(__MOD__)") LookupAndCallBinaryNode modNode) {
+            Object result = powNode.execute(x, y, PNone.NO_VALUE);
+            if (result == PNotImplemented.NOT_IMPLEMENTED) {
+                return result;
+            }
+            return modNode.executeObject(result, z);
+        }
+
         @Fallback
         @SuppressWarnings("unused")
         Object doFallback(Object x, Object y, Object z) {