src/org/mozilla/javascript/PreorderNodeIterator.java

   1 /* -*- Mode: java; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 4 -*-\r
   2  *\r
   3  * The contents of this file are subject to the Netscape Public\r
   4  * License Version 1.1 (the "License"); you may not use this file\r
   5  * except in compliance with the License. You may obtain a copy of\r
   6  * the License at http://www.mozilla.org/NPL/\r
   7  *\r
   8  * Software distributed under the License is distributed on an "AS\r
   9  * IS" basis, WITHOUT WARRANTY OF ANY KIND, either express oqr\r
  10  * implied. See the License for the specific language governing\r
  11  * rights and limitations under the License.\r
  12  *\r
  13  * The Original Code is Rhino code, released\r
  14  * May 6, 1999.\r
  15  *\r
  16  * The Initial Developer of the Original Code is Netscape\r
  17  * Communications Corporation.  Portions created by Netscape are\r
  18  * Copyright (C) 1997-1999 Netscape Communications Corporation. All\r
  19  * Rights Reserved.\r
  20  *\r
  21  * Contributor(s): \r
  22  * Norris Boyd\r
  23  * Roger Lawrence\r
  24  *\r
  25  * Alternatively, the contents of this file may be used under the\r
  26  * terms of the GNU Public License (the "GPL"), in which case the\r
  27  * provisions of the GPL are applicable instead of those above.\r
  28  * If you wish to allow use of your version of this file only\r
  29  * under the terms of the GPL and not to allow others to use your\r
  30  * version of this file under the NPL, indicate your decision by\r
  31  * deleting the provisions above and replace them with the notice\r
  32  * and other provisions required by the GPL.  If you do not delete\r
  33  * the provisions above, a recipient may use your version of this\r
  34  * file under either the NPL or the GPL.\r
  35  */\r
  36 \r
  37 package org.mozilla.javascript;\r
  38 \r
  39 import java.util.Stack;\r
  40 \r
  41 /**\r
  42  * This class implements a preorder tree iterator for the Node class.\r
  43  *\r
  44  * @see Node\r
  45  * @author Norris Boyd\r
  46  */\r
  47 public class PreorderNodeIterator {\r
  48     public PreorderNodeIterator(Node n) {\r
  49         start = n;\r
  50         stack = new Stack();\r
  51     }\r
  52 \r
  53     public Node currentNode() {\r
  54         return current;\r
  55     }\r
  56 \r
  57     public Node getCurrentParent() {\r
  58         return currentParent;\r
  59     }\r
  60 \r
  61     public Node nextNode() {\r
  62         if (current == null)\r
  63             return current = start;\r
  64         if (current.first != null) {\r
  65             stack.push(current);\r
  66             currentParent = current;\r
  67             current = current.first;\r
  68         } else {\r
  69             current = current.next;\r
  70             boolean isEmpty;\r
  71             for (;;) {\r
  72                 isEmpty = stack.isEmpty();\r
  73                 if (isEmpty || current != null)\r
  74                         break;\r
  75                 current = (Node) stack.pop();\r
  76                 current = current.next;\r
  77             }\r
  78             currentParent = isEmpty ? null : (Node) stack.peek();\r
  79         }\r
  80         return current;\r
  81     }\r
  82 \r
  83     public void replaceCurrent(Node newNode) {\r
  84         currentParent.replaceChild(current, newNode);\r
  85         current = newNode;\r
  86     }\r
  87 \r
  88     private Node start;\r
  89     private Node current;\r
  90     private Node currentParent;\r
  91     private Stack stack;\r
  92 }\r