Expand the instruction visitor to visit computations and values.

runtime/vm/intermediate_language.h

 // Copyright (c) 2012, the Dart project authors.  Please see the AUTHORS file
 // for details. All rights reserved. Use of this source code is governed by a
 // BSD-style license that can be found in the LICENSE file.
 
 #ifndef VM_INTERMEDIATE_LANGUAGE_H_
 #define VM_INTERMEDIATE_LANGUAGE_H_
 
 #include "vm/allocation.h"
 #include "vm/growable_array.h"
 #include "vm/handles_impl.h"
 #include "vm/object.h"
 
 namespace dart {
 
+class FlowGraphVisitor;
 class LocalVariable;
-class ConstantValue;
-class TempValue;
 
 // Computations and values.
 //
 // <Computation> ::= <Value>
 //                 | AssertAssignable <Value> <AbstractType>
 //                 | InstanceCall <cstring> <Value> ...
 //                 | StaticCall <Function> <Value> ...
 //                 | LoadLocal <LocalVariable>
 //                 | StoreLocal <LocalVariable> <Value>
 //                 | StrictCompare <Token::kind> <Value> <Value>
 //
 // <Value> ::= Temp <int>
 //           | Constant <Instance>
 
+// M is a two argument macro.  It is applied to each concrete value or
+// computation (include value) type name and class name.
+#define FOR_EACH_VALUE(M)                                                      \
+  M(Temp, TempVal)                                                             \
+  M(Constant, ConstantVal)
+
+#define FOR_EACH_COMPUTATION(M)                                                \
+  FOR_EACH_VALUE(M)                                                            \
+  M(AssertAssignable, AssertAssignableComp)                                    \
+  M(InstanceCall, InstanceCallComp)                                            \
+  M(StrictCompare, StrictCompareComp)                                          \
+  M(StaticCall, StaticCallComp)                                                \
+  M(LoadLocal, LoadLocalComp)                                                  \
+  M(StoreLocal, StoreLocalComp)
+
+
+#define FORWARD_DECLARATION(ShortName, ClassName) class ClassName;
+FOR_EACH_COMPUTATION(FORWARD_DECLARATION)
+#undef FORWARD_DECLARATION
+
 class Computation : public ZoneAllocated {
  public:
   Computation() { }
 
-  // Prints a computation without indentation or trailing newlines.
-  virtual void Print() const = 0;
+  // Visiting support.
+  virtual void Accept(FlowGraphVisitor* visitor) = 0;
 
  private:
   DISALLOW_COPY_AND_ASSIGN(Computation);
 };
 
 
 class Value : public Computation {
  public:
   Value() { }
 
-  virtual TempValue* AsTemp() { return NULL; }
-  virtual ConstantValue* AsConstant() { return NULL; }
+#define DEFINE_TESTERS(ShortName, ClassName)                                   \
+  virtual ClassName* As##ShortName() { return NULL; }                          \
+  bool Is##ShortName() { return As##ShortName() != NULL; }
 
-  bool IsTemp() { return AsTemp() != NULL; }
-  bool IsConstant() { return AsConstant() != NULL; }
+  FOR_EACH_VALUE(DEFINE_TESTERS)
+#undef DEFINE_TESTERS
 
  private:
   DISALLOW_COPY_AND_ASSIGN(Value);
 };
 
 
+// Functions defined in all concrete computation classes.
+#define DECLARE_COMPUTATION(ShortName)                                         \
+  virtual void Accept(FlowGraphVisitor* visitor);
+
+// Functions defined in all concrete value classes.
+#define DECLARE_VALUE(ShortName)                                               \
+  DECLARE_COMPUTATION(ShortName)                                               \
+  virtual ShortName##Val* As##ShortName() { return this; }
+
+
+class TempVal : public Value {
+ public:
+  explicit TempVal(intptr_t index) : index_(index) { }
+
+  DECLARE_VALUE(Temp)
+
+  intptr_t index() const { return index_; }
+
+ private:
+  intptr_t index_;
+
+  DISALLOW_COPY_AND_ASSIGN(TempVal);
+};
+
+
+class ConstantVal: public Value {
+ public:
+  explicit ConstantVal(const Instance& instance) : instance_(instance) {
+    ASSERT(instance.IsZoneHandle());
+  }
+
+  DECLARE_VALUE(Constant)
+
+  const Instance& instance() const { return instance_; }
+
+ private:
+  const Instance& instance_;
+
+  DISALLOW_COPY_AND_ASSIGN(ConstantVal);
+};
+
+#undef DECLARE_VALUE
+
+
 class AssertAssignableComp : public Computation {
  public:
   AssertAssignableComp(Value* value, const AbstractType& type)
       : value_(value), type_(type) { }
 
-  virtual void Print() const;
+  DECLARE_COMPUTATION(AssertAssignable)
+
+  Value* value() const { return value_; }
+  const AbstractType& type() const { return type_; }
 
  private:
   Value* value_;
   const AbstractType& type_;
 
   DISALLOW_COPY_AND_ASSIGN(AssertAssignableComp);
 };
 
 
 class InstanceCallComp : public Computation {
  public:
   InstanceCallComp(const char* name, ZoneGrowableArray<Value*>* arguments)
       : name_(name), arguments_(arguments) { }
 
-  virtual void Print() const;
+  DECLARE_COMPUTATION(InstanceCall)
+
+  const char* name() const { return name_; }
+  int ArgumentCount() const { return arguments_->length(); }
+  Value* ArgumentAt(int index) const { return (*arguments_)[index]; }
 
  private:
   const char* name_;
   ZoneGrowableArray<Value*>* arguments_;
 
   DISALLOW_COPY_AND_ASSIGN(InstanceCallComp);
 };
 
 
 class StrictCompareComp : public Computation {
  public:
   StrictCompareComp(Token::Kind kind, Value* left, Value* right)
       : kind_(kind), left_(left), right_(right) {
     ASSERT((kind_ == Token::kEQ_STRICT) || (kind_ == Token::kNE_STRICT));
   }
 
-  virtual void Print() const;
+  DECLARE_COMPUTATION(StrictCompare)
+
+  Token::Kind kind() const { return kind_; }
+  Value* left() const { return left_; }
+  Value* right() const { return right_; }
 
  private:
   const Token::Kind kind_;
   Value* left_;
   Value* right_;
 
   DISALLOW_COPY_AND_ASSIGN(StrictCompareComp);
 };
 
 
 class StaticCallComp : public Computation {
  public:
   StaticCallComp(const Function& function, ZoneGrowableArray<Value*>* arguments)
       : function_(function), arguments_(arguments) {
     ASSERT(function.IsZoneHandle());
   }
 
-  virtual void Print() const;
+  DECLARE_COMPUTATION(StaticCall)
+
+  const Function& function() const { return function_; }
+  int ArgumentCount() const { return arguments_->length(); }
+  Value* ArgumentAt(int index) const { return (*arguments_)[index]; }
 
  private:
   const Function& function_;
   ZoneGrowableArray<Value*>* arguments_;
 
   DISALLOW_COPY_AND_ASSIGN(StaticCallComp);
 };
 
 
 class LoadLocalComp : public Computation {
  public:
   explicit LoadLocalComp(const LocalVariable& local) : local_(local) { }
 
-  virtual void Print() const;
+  DECLARE_COMPUTATION(LoadLocal)
+
+  const LocalVariable& local() const { return local_; }
 
  private:
   const LocalVariable& local_;
 
   DISALLOW_COPY_AND_ASSIGN(LoadLocalComp);
 };
 
 
 class StoreLocalComp : public Computation {
  public:
   StoreLocalComp(const LocalVariable& local, Value* value)
       : local_(local), value_(value) { }
 
-  virtual void Print() const;
+  DECLARE_COMPUTATION(StoreLocal)
+
+  const LocalVariable& local() const { return local_; }
+  Value* value() const { return value_; }
 
  private:
   const LocalVariable& local_;
   Value* value_;
 
   DISALLOW_COPY_AND_ASSIGN(StoreLocalComp);
 };
 
-
-class TempValue : public Value {
- public:
-  explicit TempValue(intptr_t index) : index_(index) { }
-
-  virtual TempValue* AsTemp() { return this; }
-  virtual void Print() const;
-
- private:
-  intptr_t index_;
-
-  DISALLOW_COPY_AND_ASSIGN(TempValue);
-};
-
-
-class ConstantValue: public Value {
- public:
-  explicit ConstantValue(const Instance& instance) : instance_(instance) {
-    ASSERT(instance.IsZoneHandle());
-  }
-
-  virtual ConstantValue* AsConstant() { return this; }
-  virtual void Print() const;
-
-  const Instance& instance() const { return instance_; }
-
- private:
-  const Instance& instance_;
-
-  DISALLOW_COPY_AND_ASSIGN(ConstantValue);
-};
+#undef DECLARE_COMPUTATION
 
 
 // Instructions.
 //
 // <Instruction> ::= Do <Computation> <Instruction>
 //                 | Bind <int> <Computation> <Instruction>
 //                 | Return <Value>
 //                 | Branch <Value> <Instruction> <Instruction>
 //                 | Empty <Instruction>
 
 // M is a single argument macro.  It is applied to each concrete instruction
 // type name.  The concrete instruction classes are the name with Instr
 // concatenated.
 #define FOR_EACH_INSTRUCTION(M)                                                \
   M(JoinEntry)                                                                 \
   M(TargetEntry)                                                               \
   M(Do)                                                                        \
   M(Bind)                                                                      \
   M(Return)                                                                    \
   M(Branch)
 
 
 // Forward declarations for Instruction classes.
 class BlockEntryInstr;
-class InstructionVisitor;
 #define FORWARD_DECLARATION(type) class type##Instr;
 FOR_EACH_INSTRUCTION(FORWARD_DECLARATION)
 #undef FORWARD_DECLARATION
 
 
 // Functions required in all concrete instruction classes.
 #define DECLARE_INSTRUCTION(type)                                              \
-  virtual Instruction* Accept(InstructionVisitor* visitor);                    \
+  virtual Instruction* Accept(FlowGraphVisitor* visitor);                      \
   virtual bool Is##type() const { return true; }                               \
   virtual type##Instr* As##type() { return this; }                             \
 
 
 class Instruction : public ZoneAllocated {
  public:
   Instruction() : mark_(false) { }
 
   virtual bool IsBlockEntry() const { return false; }
 
   // Visiting support.
-  virtual Instruction* Accept(InstructionVisitor* visitor) = 0;
+  virtual Instruction* Accept(FlowGraphVisitor* visitor) = 0;
 
   virtual void SetSuccessor(Instruction* instr) = 0;
   // Perform a postorder traversal of the instruction graph reachable from
   // this instruction.  Accumulate basic block entries in the order visited
   // in the in/out parameter 'block_entries'.
   virtual void Postorder(GrowableArray<BlockEntryInstr*>* block_entries) = 0;
 
   // Mark bit to support non-reentrant recursive traversal (i.e.,
   // identification of cycles).  Before and after a traversal, all the nodes
   // must have the same mark.
@@ skipping 177 matching lines @@
   Value* value_;
   TargetEntryInstr* true_successor_;
   TargetEntryInstr* false_successor_;
 
   DISALLOW_COPY_AND_ASSIGN(BranchInstr);
 };
 
 #undef DECLARE_INSTRUCTION
 
 
-class InstructionVisitor {
+class FlowGraphVisitor {
  public:
-  InstructionVisitor() { }
-  virtual ~InstructionVisitor() { }
+  FlowGraphVisitor() { }
+  virtual ~FlowGraphVisitor() { }
 
   // Visit each block in the array list in reverse, and for each block its
   // instructions in order from the block entry to exit.
   virtual void VisitBlocks(const GrowableArray<BlockEntryInstr*>& block_order);
 
-#define DECLARE_VISIT(type)                             \
-  virtual void Visit##type(type##Instr* instr) { }
-  FOR_EACH_INSTRUCTION(DECLARE_VISIT)
-#undef DECLARE_VISIT
+#define DECLARE_VISIT_COMPUTATION(ShortName, ClassName)                        \
+  virtual void Visit##ShortName(ClassName* comp) { }
+
+#define DECLARE_VISIT_INSTRUCTION(ShortName)                                   \
+  virtual void Visit##ShortName(ShortName##Instr* instr) { }
+
+  FOR_EACH_COMPUTATION(DECLARE_VISIT_COMPUTATION)
+  FOR_EACH_INSTRUCTION(DECLARE_VISIT_INSTRUCTION)
+
+#undef DECLARE_VISIT_COMPUTATION
+#undef DECLARE_VISIT_INSTRUCTION
 
  private:
-  DISALLOW_COPY_AND_ASSIGN(InstructionVisitor);
+  DISALLOW_COPY_AND_ASSIGN(FlowGraphVisitor);
 };
 
 
 }  // namespace dart
 
 #endif  // VM_INTERMEDIATE_LANGUAGE_H_