diff graal/com.oracle.truffle.ruby.nodes/src/com/oracle/truffle/ruby/nodes/objects/DefineOrGetClassNode.java @ 13514:0fbee3eb71f0

Ruby: import project.
author Chris Seaton <chris.seaton@oracle.com>
date Mon, 06 Jan 2014 17:12:09 +0000
parents
children
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/graal/com.oracle.truffle.ruby.nodes/src/com/oracle/truffle/ruby/nodes/objects/DefineOrGetClassNode.java	Mon Jan 06 17:12:09 2014 +0000
@@ -0,0 +1,83 @@
+/*
+ * Copyright (c) 2013 Oracle and/or its affiliates. All rights reserved. This
+ * code is released under a tri EPL/GPL/LGPL license. You can use it,
+ * redistribute it and/or modify it under the terms of the:
+ *
+ * Eclipse Public License version 1.0
+ * GNU General Public License version 2
+ * GNU Lesser General Public License version 2.1
+ */
+package com.oracle.truffle.ruby.nodes.objects;
+
+import com.oracle.truffle.api.*;
+import com.oracle.truffle.api.frame.*;
+import com.oracle.truffle.ruby.nodes.*;
+import com.oracle.truffle.ruby.runtime.*;
+import com.oracle.truffle.ruby.runtime.control.*;
+import com.oracle.truffle.ruby.runtime.core.*;
+
+/**
+ * Define a new class, or get the existing one of the same name.
+ */
+public class DefineOrGetClassNode extends RubyNode {
+
+    private final String name;
+    @Child protected RubyNode moduleDefinedIn;
+    @Child protected RubyNode superClass;
+
+    public DefineOrGetClassNode(RubyContext context, SourceSection sourceSection, String name, RubyNode moduleDefinedIn, RubyNode superClass) {
+        super(context, sourceSection);
+        this.name = name;
+        this.moduleDefinedIn = adoptChild(moduleDefinedIn);
+        this.superClass = adoptChild(superClass);
+    }
+
+    @Override
+    public Object execute(VirtualFrame frame) {
+        CompilerAsserts.neverPartOfCompilation();
+
+        final RubyContext context = getContext();
+
+        final RubyModule moduleDefinedInObject = (RubyModule) moduleDefinedIn.execute(frame);
+
+        // Look for a current definition of the class, or create a new one
+
+        final Object constantValue = moduleDefinedInObject.lookupConstant(name);
+
+        RubyClass definingClass;
+
+        if (constantValue == null) {
+            final Object self = frame.getArguments(RubyArguments.class).getSelf();
+
+            RubyModule parentModule;
+
+            if (self instanceof RubyModule) {
+                parentModule = (RubyModule) self;
+            } else {
+                // Because it's top level, and so self is the magic main object
+                parentModule = null;
+            }
+
+            final RubyClass superClassObject = (RubyClass) superClass.execute(frame);
+
+            if (superClassObject instanceof RubyException.RubyExceptionClass) {
+                definingClass = new RubyException.RubyExceptionClass(superClassObject, name);
+            } else {
+                definingClass = new RubyClass(parentModule, superClassObject, name);
+            }
+
+            moduleDefinedInObject.setConstant(name, definingClass);
+            moduleDefinedInObject.getSingletonClass().setConstant(name, definingClass);
+
+            definingClass.getRubyClass().include(moduleDefinedInObject);
+        } else {
+            if (constantValue instanceof RubyClass) {
+                definingClass = (RubyClass) constantValue;
+            } else {
+                throw new RaiseException(context.getCoreLibrary().typeErrorIsNotA(constantValue.toString(), "class"));
+            }
+        }
+
+        return definingClass;
+    }
+}