Experimental: use the Result class to manage the lifetime of registers...

src/register-allocator-ia32.h

 // Copyright 2008 the V8 project authors. All rights reserved.
 // Redistribution and use in source and binary forms, with or without
 // modification, are permitted provided that the following conditions are
 // met:
 //
 //     * Redistributions of source code must retain the above copyright
 //       notice, this list of conditions and the following disclaimer.
 //     * Redistributions in binary form must reproduce the above
 //       copyright notice, this list of conditions and the following
 //       disclaimer in the documentation and/or other materials provided
@@ skipping 14 matching lines @@
 // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 
 #ifndef V8_REGISTER_ALLOCATOR_IA32_H_
 #define V8_REGISTER_ALLOCATOR_IA32_H_
 
 #include "macro-assembler.h"
 
 namespace v8 { namespace internal {
 
+
+// -------------------------------------------------------------------------
+// Results
+//
+// Results encapsulate the compile-time values manipulated by the code
+// generator.  They can represent registers or constants.
+
+class Result BASE_EMBEDDED {
+ public:
+  // Construct an invalid result.
+  explicit Result(CodeGenerator* cgen) : type_(INVALID), cgen_(cgen) {}
+
+  // Construct a register Result.
+  Result(Register reg, CodeGenerator* cgen);
+
+  // Construct a Result whose value is a compile-time constant.
+  Result(Handle<Object> value, CodeGenerator * cgen)
+      : type_(CONSTANT),
+        cgen_(cgen) {
+    data_.handle_ = value.location();
+  }
+
+  // The copy constructor and assignment operators could each create a new
+  // register reference.
+  Result(const Result& other) {
+    other.CopyTo(this);
+  }
+
+  Result& operator=(Result& other) {
+    if (this != &other) {
+      Unuse();
+      other.CopyTo(this);
+      other.Unuse();
+    }
+    return *this;
+  }
+
+  ~Result() { Unuse(); }
+
+  void Unuse();
+
+  bool is_valid() const { return type() != INVALID; }
+  bool is_register() const { return type() == REGISTER; }
+  bool is_constant() const { return type() == CONSTANT; }
+
+  Register reg() const {
+    ASSERT(type() == REGISTER);
+    return data_.reg_;
+  }
+
+  Handle<Object> handle() const {
+    ASSERT(type() == CONSTANT);
+    return Handle<Object>(data_.handle_);
+  }
+
+  // Change a result to a register result.  If the result is not already
+  // in a register, allocate a register from the code generator, and emit
+  // code to move the value into that register.
+  void ToRegister();
+

[William Hesse, 2008/12/19 08:41:03]

Do we also want ToRegister(Register target)?

[Kevin Millikin (Chromium), 2008/12/19 08:48:11]

Ultimately, yes.  It was in a withdrawn change I sent out a few days ago, and
it'll pop up again soon.

+ private:
+  enum Type {
+    INVALID,
+    REGISTER,
+    CONSTANT
+  };
+
+  Type type_;
+
+  union {
+    Register reg_;
+    Object** handle_;
+  } data_;
+
+  CodeGenerator* cgen_;
+
+  Type type() const { return type_; }
+
+  void CopyTo(Result* destination) const;
+};
+
+
 // -------------------------------------------------------------------------
 // Register file
 //
 // The register file tracks reference counts for the processor registers.
 // It is used by both the register allocator and the virtual frame.
 
 class RegisterFile BASE_EMBEDDED {
  public:
   RegisterFile() { Reset(); }
 
@@ skipping 29 matching lines @@
   int ref_counts_[kNumRegisters];
 };
 
 
 // -------------------------------------------------------------------------
 // Register allocator
 //
 
 class RegisterAllocator BASE_EMBEDDED {
  public:
-  explicit RegisterAllocator(CodeGenerator* cgen) : code_generator_(cgen) {}
+  explicit RegisterAllocator(CodeGenerator* cgen) : cgen_(cgen) {}
 
   int num_registers() const { return RegisterFile::kNumRegisters; }
 
+  // Predicates and accessors for the registers' reference counts.
+  bool is_used(int reg_code) const { return registers_.is_used(reg_code); }
   int count(int reg_code) { return registers_.count(reg_code); }
 
   // Explicitly record a reference to a register.
   void Use(Register reg) { registers_.Use(reg); }
 
   // Explicitly record that a register will no longer be used.
   void Unuse(Register reg) { registers_.Unuse(reg); }
 
   // Initialize the register allocator for entry to a JS function.  On
   // entry, esp, ebp, esi, and edi are externally referenced (ie, outside
   // the virtual frame); and the other registers are free.
   void Initialize();
 
-  // Allocate a free register if possible or fail by returning no_reg.
-  Register Allocate();
+  // Allocate a free register and return a register result if possible or
+  // fail and return an invalid result.
+  Result Allocate();
 
-  // Allocate a free register without spilling any or fail and return
-  // no_reg.
-  Register AllocateWithoutSpilling();
+  // Allocate a free register without spilling any from the current frame or
+  // fail and return an invalid result.
+  Result AllocateWithoutSpilling();
 
  private:
-  CodeGenerator* code_generator_;
+  CodeGenerator* cgen_;
   RegisterFile registers_;
 };
 
 } }  // namespace v8::internal
 
 #endif  // V8_REGISTER_ALLOCATOR_IA32_H_