ASTFactory.cpp
上传用户:afrynkmhm
上传日期:2007-01-06
资源大小:1262k
文件大小:6k
源码类别:

编译器/解释器

开发平台:

Others

  1. /**
  2.  * <b>SOFTWARE RIGHTS</b>
  3.  * <p>
  4.  * ANTLR 2.6.0 MageLang Insitute, 1998
  5.  * <p>
  6.  * We reserve no legal rights to the ANTLR--it is fully in the
  7.  * public domain. An individual or company may do whatever
  8.  * they wish with source code distributed with ANTLR or the
  9.  * code generated by ANTLR, including the incorporation of
  10.  * ANTLR, or its output, into commerical software.
  11.  * <p>
  12.  * We encourage users to develop software with ANTLR. However,
  13.  * we do ask that credit is given to us for developing
  14.  * ANTLR. By "credit", we mean that if you use ANTLR or
  15.  * incorporate any source code into one of your programs
  16.  * (commercial product, research project, or otherwise) that
  17.  * you acknowledge this fact somewhere in the documentation,
  18.  * research report, etc... If you like ANTLR and have
  19.  * developed a nice tool with the output, please mention that
  20.  * you developed it using ANTLR. In addition, we ask that the
  21.  * headers remain intact in our source code. As long as these
  22.  * guidelines are kept, we expect to continue enhancing this
  23.  * system and expect to make other tools available as they are
  24.  * completed.
  25.  * <p>
  26.  * The ANTLR gang:
  27.  * @version ANTLR 2.6.0 MageLang Insitute, 1998
  28.  * @author Terence Parr, <a href=http://www.MageLang.com>MageLang Institute</a>
  29.  * @author <br>John Lilley, <a href=http://www.Empathy.com>Empathy Software</a>
  30.  * @author <br><a href="mailto:pete@yamuna.demon.co.uk">Pete Wells</a>
  31.  */
  32. #include "antlr/ASTFactory.hpp"
  33. #include "antlr/CommonAST.hpp"
  34. ANTLR_BEGIN_NAMESPACE(antlr)
  35. /** AST Support code shared by TreeParser and Parser.
  36.  *  We use delegation to share code (and have only one
  37.  *  bit of code to maintain) rather than subclassing
  38.  *  or superclassing (forces AST support code to be
  39.  *  loaded even when you don't want to do AST stuff).
  40.  *
  41.  *  Typically, setASTNodeType is used to specify the
  42.  *  type of node to create, but you can override
  43.  *  create to make heterogeneous nodes etc...
  44.  */
  45. ASTFactory::ASTFactory()
  46. {
  47. nodeFactory = &CommonAST::factory;
  48. }
  49. /** Add a child to the current AST */
  50. void ASTFactory::addASTChild(ASTPair& currentAST, RefAST child)
  51. {
  52. if (child) {
  53. if (!currentAST.root) {
  54. // Make new child the current root
  55. currentAST.root = child;
  56. }
  57. else {
  58. if (!currentAST.child) {
  59. // Add new child to current root
  60. currentAST.root->setFirstChild(child);
  61. }
  62. else {
  63. currentAST.child->setNextSibling(child);
  64. }
  65. }
  66. // Make new child the current child
  67. currentAST.child = child;
  68. currentAST.advanceChildToEnd();
  69. }
  70. }
  71. /** Create a new empty AST node; if the user did not specify
  72.  *  an AST node type, then create a default one: CommonAST.
  73.  */
  74. RefAST ASTFactory::create()
  75. {
  76. RefAST node = nodeFactory(); //new CommonASTNode();
  77. node->setType(Token::INVALID_TYPE);
  78. return node;
  79. }
  80. RefAST ASTFactory::create(int type)
  81. {
  82. RefAST t = create();
  83. t->initialize(type,"");
  84. return t;
  85. }
  86. RefAST ASTFactory::create(int type, const ANTLR_USE_NAMESPACE(std)string& txt)
  87. {
  88. RefAST t = create();
  89. t->initialize(type,txt);
  90. return t;
  91. }
  92. /** Create a new empty AST node; if the user did not specify
  93.  *  an AST node type, then create a default one: CommonAST.
  94.  */
  95. RefAST ASTFactory::create(RefAST tr)
  96. {
  97. if (!tr) return nullAST; // create(null) == null
  98. RefAST t = create();
  99. t->initialize(tr);
  100. return t;
  101. }
  102. RefAST ASTFactory::create(RefToken tok)
  103. {
  104. RefAST t = create();
  105. t->initialize(tok);
  106. return t;
  107. }
  108. /** Copy a single node.  clone() is not used because
  109.  *  we want to return an AST not a plain object...a type
  110.  *  safety issue.  Further, we want to have all AST node
  111.  *  creation go through the factory so creation can be
  112.  *  tracked.  Returns null if t is null.
  113.  */
  114. RefAST ASTFactory::dup(RefAST t)
  115. {
  116. return create(t); // if t==null, create returns null
  117. }
  118. /** Duplicate tree including siblings of root. */
  119. RefAST ASTFactory::dupList(RefAST t)
  120. {
  121. RefAST result = dupTree(t);         // if t == null, then result==null
  122. RefAST nt = result;
  123. while (t) { // for each sibling of the root
  124. t = t->getNextSibling();
  125. nt->setNextSibling(dupTree(t)); // dup each subtree, building new tree
  126. nt = nt->getNextSibling();
  127. }
  128. return result;
  129. }
  130. /**Duplicate a tree, assuming this is a root node of a tree--
  131.  * duplicate that node and what's below; ignore siblings of root node.
  132.  */
  133. RefAST ASTFactory::dupTree(RefAST t)
  134. {
  135. RefAST result = dup(t); // make copy of root
  136. // copy all children of root.
  137. if (t) {
  138. result->setFirstChild( dupList(t->getFirstChild()) );
  139. }
  140. return result;
  141. }
  142. /** Make a tree from a list of nodes.  The first element in the
  143.  *  array is the root.  If the root is null, then the tree is
  144.  *  a simple list not a tree.  Handles null children nodes correctly.
  145.  *  For example, build(a, b, null, c) yields tree (a b c).  build(null,a,b)
  146.  *  yields tree (nil a b).
  147.  */
  148. RefAST ASTFactory::make(ANTLR_USE_NAMESPACE(std)vector<RefAST> nodes)
  149. {
  150. if ( nodes.size()==0 )
  151. return RefAST(nullASTptr);
  152. RefAST root = nodes[0];
  153. RefAST tail = RefAST(nullASTptr);
  154. if (root) {
  155. root->setFirstChild(RefAST(nullASTptr)); // don't leave any old pointers set
  156. }
  157. // link in children;
  158. for (unsigned int i=1; i<nodes.size(); i++) {
  159. if ( !nodes[i] ) continue; // ignore null nodes
  160. if ( !root ) {
  161. // Set the root and set it up for a flat list
  162. root = tail = nodes[i];
  163. }
  164. else if ( !tail ) {
  165. root->setFirstChild(nodes[i]);
  166. tail = root->getFirstChild();
  167. }
  168. else {
  169. tail->setNextSibling(nodes[i]);
  170. tail = tail->getNextSibling();
  171. }
  172. // Chase tail to last sibling
  173. while (tail->getNextSibling()) {
  174. tail = tail->getNextSibling();
  175. }
  176. }
  177. return root;
  178. }
  179. /** Make a tree from a list of nodes, where the nodes are contained
  180.   * in an ASTArray object
  181.   */
  182. RefAST ASTFactory::make(ASTArray* nodes)
  183. {
  184. RefAST ret = make(nodes->array);
  185. delete nodes;
  186. return ret;
  187. }
  188. /** Make an AST the root of current AST */
  189. void ASTFactory::makeASTRoot(ASTPair& currentAST, RefAST root)
  190. {
  191. if (root) {
  192. // Add the current root as a child of new root
  193. root->addChild(currentAST.root);
  194. // The new current child is the last sibling of the old root
  195. currentAST.child = currentAST.root;
  196. currentAST.advanceChildToEnd();
  197. // Set the new root
  198. currentAST.root = root;
  199. }
  200. }
  201. void ASTFactory::setASTNodeFactory(factory_type factory)
  202. {
  203. nodeFactory = factory;
  204. }
  205. ANTLR_END_NAMESPACE