Home | History | Annotate | Download | only in compiler 
      1 // Copyright 2014 the V8 project authors. All rights reserved.
      2 // Use of this source code is governed by a BSD-style license that can be
      3 // found in the LICENSE file.
      4 
      5 #ifndef V8_CCTEST_COMPILER_GRAPH_BUILDER_TESTER_H_
      6 #define V8_CCTEST_COMPILER_GRAPH_BUILDER_TESTER_H_
      7 
      8 #include "src/v8.h"
      9 #include "test/cctest/cctest.h"
     10 
     11 #include "src/compiler/common-operator.h"
     12 #include "src/compiler/graph-builder.h"
     13 #include "src/compiler/machine-operator.h"
     14 #include "src/compiler/simplified-operator.h"
     15 #include "test/cctest/compiler/call-tester.h"
     16 #include "test/cctest/compiler/simplified-graph-builder.h"
     17 
     18 namespace v8 {
     19 namespace internal {
     20 namespace compiler {
     21 
     22 // A class that just passes node creation on to the Graph.
     23 class DirectGraphBuilder : public GraphBuilder {
     24  public:
     25   explicit DirectGraphBuilder(Graph* graph) : GraphBuilder(graph) {}
     26   virtual ~DirectGraphBuilder() {}
     27 
     28  protected:
     29   virtual Node* MakeNode(const Operator* op, int value_input_count,
     30                          Node** value_inputs) FINAL {
     31     return graph()->NewNode(op, value_input_count, value_inputs);
     32   }
     33 };
     34 
     35 
     36 class MachineCallHelper : public CallHelper {
     37  public:
     38   MachineCallHelper(Zone* zone, MachineSignature* machine_sig);
     39 
     40   Node* Parameter(size_t index);
     41 
     42   void GenerateCode() { Generate(); }
     43 
     44  protected:
     45   virtual byte* Generate();
     46   void InitParameters(GraphBuilder* builder, CommonOperatorBuilder* common);
     47 
     48  protected:
     49   size_t parameter_count() const { return machine_sig_->parameter_count(); }
     50 
     51  private:
     52   Node** parameters_;
     53   // TODO(dcarney): shouldn't need graph stored.
     54   Graph* graph_;
     55   MaybeHandle<Code> code_;
     56 };
     57 
     58 
     59 class GraphAndBuilders {
     60  public:
     61   explicit GraphAndBuilders(Zone* zone)
     62       : main_graph_(new (zone) Graph(zone)),
     63         main_common_(zone),
     64         main_simplified_(zone) {}
     65 
     66  protected:
     67   // Prefixed with main_ to avoid naiming conflicts.
     68   Graph* main_graph_;
     69   CommonOperatorBuilder main_common_;
     70   MachineOperatorBuilder main_machine_;
     71   SimplifiedOperatorBuilder main_simplified_;
     72 };
     73 
     74 
     75 template <typename ReturnType>
     76 class GraphBuilderTester
     77     : public HandleAndZoneScope,
     78       private GraphAndBuilders,
     79       public MachineCallHelper,
     80       public SimplifiedGraphBuilder,
     81       public CallHelper2<ReturnType, GraphBuilderTester<ReturnType> > {
     82  public:
     83   explicit GraphBuilderTester(MachineType p0 = kMachNone,
     84                               MachineType p1 = kMachNone,
     85                               MachineType p2 = kMachNone,
     86                               MachineType p3 = kMachNone,
     87                               MachineType p4 = kMachNone)
     88       : GraphAndBuilders(main_zone()),
     89         MachineCallHelper(
     90             main_zone(),
     91             MakeMachineSignature(
     92                 main_zone(), ReturnValueTraits<ReturnType>::Representation(),
     93                 p0, p1, p2, p3, p4)),
     94         SimplifiedGraphBuilder(main_graph_, &main_common_, &main_machine_,
     95                                &main_simplified_) {
     96     Begin(static_cast<int>(parameter_count()));
     97     InitParameters(this, &main_common_);
     98   }
     99   virtual ~GraphBuilderTester() {}
    100 
    101   Factory* factory() const { return isolate()->factory(); }
    102 };
    103 }  // namespace compiler
    104 }  // namespace internal
    105 }  // namespace v8
    106 
    107 #endif  // V8_CCTEST_COMPILER_GRAPH_BUILDER_TESTER_H_
    108