diff graal/com.oracle.truffle.dsl.processor/src/com/oracle/truffle/dsl/processor/generator/PolymorphicNodeFactory.java @ 18752:1acaa69ff61b

Truffle-DSL: refactor generator classes
author Christian Humer <christian.humer@gmail.com>
date Mon, 29 Dec 2014 23:38:16 +0100
parents
children f6b8787dc113
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/graal/com.oracle.truffle.dsl.processor/src/com/oracle/truffle/dsl/processor/generator/PolymorphicNodeFactory.java	Mon Dec 29 23:38:16 2014 +0100
@@ -0,0 +1,90 @@
+/*
+ * Copyright (c) 2014, Oracle and/or its affiliates. All rights reserved.
+ * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
+ *
+ * This code is free software; you can redistribute it and/or modify it
+ * under the terms of the GNU General Public License version 2 only, as
+ * published by the Free Software Foundation.
+ *
+ * This code is distributed in the hope that it will be useful, but WITHOUT
+ * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+ * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
+ * version 2 for more details (a copy is included in the LICENSE file that
+ * accompanied this code).
+ *
+ * You should have received a copy of the GNU General Public License version
+ * 2 along with this work; if not, write to the Free Software Foundation,
+ * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
+ *
+ * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
+ * or visit www.oracle.com if you need additional information or have any
+ * questions.
+ */
+package com.oracle.truffle.dsl.processor.generator;
+
+import static com.oracle.truffle.dsl.processor.java.ElementUtils.*;
+import static javax.lang.model.element.Modifier.*;
+
+import java.util.*;
+
+import javax.lang.model.type.*;
+
+import com.oracle.truffle.api.nodes.*;
+import com.oracle.truffle.dsl.processor.java.model.*;
+import com.oracle.truffle.dsl.processor.model.*;
+
+class PolymorphicNodeFactory extends SpecializedNodeFactory {
+
+    public PolymorphicNodeFactory(CodeTypeElement nodeGen) {
+        super(nodeGen);
+    }
+
+    @Override
+    public CodeTypeElement create(SpecializationData polymorph) {
+        NodeData node = polymorph.getNode();
+        TypeMirror baseType = node.getNodeType();
+        if (nodeGen != null) {
+            baseType = nodeGen.asType();
+        }
+        CodeTypeElement clazz = createClass(node, modifiers(PRIVATE, STATIC, FINAL), nodePolymorphicClassName(node), baseType, false);
+
+        clazz.getAnnotationMirrors().add(createNodeInfo(node, NodeCost.POLYMORPHIC));
+
+        for (Parameter polymorphParameter : polymorph.getSignatureParameters()) {
+            if (!polymorphParameter.getTypeSystemType().isGeneric()) {
+                continue;
+            }
+            Set<TypeData> types = new HashSet<>();
+            for (SpecializationData specialization : node.getSpecializations()) {
+                if (!specialization.isSpecialized()) {
+                    continue;
+                }
+                Parameter parameter = specialization.findParameter(polymorphParameter.getLocalName());
+                assert parameter != null;
+                types.add(parameter.getTypeSystemType());
+            }
+
+        }
+
+        for (NodeExecutionData execution : getModel().getNode().getChildExecutions()) {
+            String fieldName = polymorphicTypeName(execution);
+            CodeVariableElement var = new CodeVariableElement(modifiers(PRIVATE), getContext().getType(Class.class), fieldName);
+            var.getAnnotationMirrors().add(new CodeAnnotationMirror(getContext().getTruffleTypes().getCompilationFinal()));
+            clazz.add(var);
+        }
+
+        return clazz;
+    }
+
+    @Override
+    protected void createChildren(SpecializationData specialization) {
+        CodeTypeElement clazz = getElement();
+
+        createConstructors(clazz);
+        createExecuteMethods(specialization);
+
+        clazz.add(createUpdateTypes0());
+        createCachedExecuteMethods(specialization);
+    }
+
+}
\ No newline at end of file