src/func-name-inferrer.h

   1 // Copyright 2006-2009 the V8 project authors. All rights reserved.
   2 // Redistribution and use in source and binary forms, with or without
   3 // modification, are permitted provided that the following conditions are
   4 // met:
   5 //
   6 //     * Redistributions of source code must retain the above copyright
   7 //       notice, this list of conditions and the following disclaimer.
   8 //     * Redistributions in binary form must reproduce the above
   9 //       copyright notice, this list of conditions and the following
  10 //       disclaimer in the documentation and/or other materials provided
  11 //       with the distribution.
  12 //     * Neither the name of Google Inc. nor the names of its
  13 //       contributors may be used to endorse or promote products derived
  14 //       from this software without specific prior written permission.
  15 //
  16 // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  17 // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  18 // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  19 // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  20 // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  21 // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  22 // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  23 // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  24 // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  25 // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  26 // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  27
  28 #ifndef V8_FUNC_NAME_INFERRER_H_
  29 #define V8_FUNC_NAME_INFERRER_H_
  30
  31 namespace v8 { namespace internal {
  32
  33 // FuncNameInferrer is a stateful class that is used to perform name
  34 // inference for anonymous functions during static analysis of source code.
  35 // Inference is performed in cases when an anonymous function is assigned
  36 // to a variable or a property (see test-func-name-inference.cc for examples.)
  37
  38 // The basic idea is that during AST traversal LHSs of expressions are
  39 // always visited before RHSs. Thus, during visiting the LHS, a name can be
  40 // collected, and during visiting the RHS, a function literal can be collected.
  41 // Inference is performed while leaving the assignment node.
  42
  43 class FuncNameInferrer BASE_EMBEDDED {
  44  public:
  45   FuncNameInferrer() :
  46       entries_stack_(10),
  47       names_stack_(5),
  48       func_to_infer_(NULL),
  49       dot_(Factory::NewStringFromAscii(CStrVector("."))) {
  50   }
  51
  52   bool IsOpen() const { return !entries_stack_.is_empty(); }
  53
  54   void PushEnclosingName(Handle<String> name);
  55
  56   void Enter() {
  57     entries_stack_.Add(names_stack_.length());
  58   }
  59
  60   void Leave() {
  61     ASSERT(IsOpen());
  62     names_stack_.Rewind(entries_stack_.RemoveLast());
  63   }
  64
  65   void PushName(Handle<String> name) {
  66     if (IsOpen()) {
  67       names_stack_.Add(name);
  68     }
  69   }
  70
  71   void SetFuncToInfer(FunctionLiteral* func_to_infer) {
  72     if (IsOpen()) {
  73       ASSERT(func_to_infer_ == NULL);
  74       func_to_infer_ = func_to_infer;
  75     }
  76   }
  77
  78   void InferAndLeave() {
  79     ASSERT(IsOpen());
  80     MaybeInferFunctionName();
  81     Leave();
  82   }
  83
  84  private:
  85   Handle<String> MakeNameFromStack();
  86   Handle<String> MakeNameFromStackHelper(int pos, Handle<String> prev);
  87   void MaybeInferFunctionName();
  88
  89   List<int> entries_stack_;
  90   List<Handle<String> > names_stack_;
  91   FunctionLiteral* func_to_infer_;
  92   Handle<String> dot_;
  93
  94   DISALLOW_COPY_AND_ASSIGN(FuncNameInferrer);
  95 };
  96
  97
  98 // A wrapper class that automatically calls InferAndLeave when
  99 // leaving scope.
 100 class ScopedFuncNameInferrer BASE_EMBEDDED {
 101  public:
 102   ScopedFuncNameInferrer(FuncNameInferrer* inferrer) : inferrer_(inferrer),
 103                                                        is_entered_(false) {};
 104   ~ScopedFuncNameInferrer() {
 105     if (is_entered_) {
 106       inferrer_->InferAndLeave();
 107     }
 108   }
 109
 110   void Enter() {
 111     inferrer_->Enter();
 112     is_entered_ = true;
 113   }
 114
 115  private:
 116   FuncNameInferrer* inferrer_;
 117   bool is_entered_;
 118
 119   DISALLOW_COPY_AND_ASSIGN(ScopedFuncNameInferrer);
 120 };
 121
 122
 123 } }  // namespace v8::internal
 124
 125 #endif  // V8_FUNC_NAME_INFERRER_H_