Cast traslated only for primitives

iuliux · iuliux · commit 2922fbd5a83c · 2012-04-23T23:09:14.000+03:00
diff --git a/java2python/compiler/visitor.py b/java2python/compiler/visitor.py
@@ -678,7 +678,6 @@ def acceptPreformatted(self, node, memo):
         return acceptPreformatted
 
     acceptArrayElementAccess = makeNodePreformattedExpr(FS.l + '[' + FS.r + ']')
-    acceptCastExpr  = makeNodePreformattedExpr(FS.l + '(' + FS.r + ')' ) # problem?
     acceptDiv = makeNodePreformattedExpr(FS.l + ' / ' + FS.r)
     acceptLogicalAnd = makeNodePreformattedExpr(FS.l + ' and ' + FS.r)
     acceptLogicalNot = makeNodePreformattedExpr('not ' + FS.l)
@@ -690,6 +689,25 @@ def acceptPreformatted(self, node, memo):
     acceptUnaryMinus = makeNodePreformattedExpr('-' + FS.l)
     acceptUnaryPlus = makeNodePreformattedExpr('+' + FS.l)
 
+    def acceptCastExpr(self, node, memo): # problem?
+        """ Accept and process a cast expression. """
+        # If the type of casting is a primitive type,
+        # then do the cast, else drop it.
+        factory = self.factory.expr
+        typeTok = node.firstChildOfType(tokens.TYPE)
+        typeIdent = typeTok.firstChild()
+        typeName = typeIdent.text
+        if typeIdent.type == tokens.QUALIFIED_TYPE_IDENT:
+            typeName = typeIdent.firstChild().text
+
+        if typeName in tokens.primitiveTypeNames:
+            # Cast using the primitive type constructor
+            self.fs = typeName + '(' + FS.r + ')'
+        else:
+            self.fs = FS.r
+        self.left, self.right = visitors = factory(parent=self), factory(parent=self)
+        self.zipWalk(node.children, visitors, memo)
+
     def makeAcceptPrePost(suffix, pre):
         """ Make an accept method for pre- and post- assignment expressions. """
         def acceptPrePost(self, node, memo):
diff --git a/java2python/lang/base.py b/java2python/lang/base.py
@@ -88,6 +88,11 @@ def methodTypes(self):
         mod = self.module
         return (mod.VOID_METHOD_DECL, mod.FUNCTION_METHOD_DECL, )
 
+    @property
+    def primitiveTypeNames(self):
+        """ Type name of well-known primitive types """
+        return ('bool', 'str', 'int', 'long', 'float', )
+
     @property
     def module(self):
         """ Provides lazy import to the parser module. """
