Experiments/Attic/Rook/Castle.Rook.Compiler/Visitors/BreadthFirstVisitor.cs

   1 // Copyright 2004-2007 Castle Project - http://www.castleproject.org/
   2 //
   3 // Licensed under the Apache License, Version 2.0 (the "License");
   4 // you may not use this file except in compliance with the License.
   5 // You may obtain a copy of the License at
   6 //
   7 //     http://www.apache.org/licenses/LICENSE-2.0
   8 //
   9 // Unless required by applicable law or agreed to in writing, software
  10 // distributed under the License is distributed on an "AS IS" BASIS,
  11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12 // See the License for the specific language governing permissions and
  13 // limitations under the License.
  14
  15 namespace Castle.Rook.Compiler.Visitors
  16 {
  17         using System;
  18         using System.Collections;
  19
  20         using Castle.Rook.Compiler.AST;
  21
  22
  23         public class BreadthFirstVisitor : AbstractVisitor
  24         {
  25                 private bool nowQueue = false;
  26                 private Queue nodesToBeVisited = new Queue();
  27
  28                 public override bool VisitNode(IVisitableNode node)
  29                 {
  30                         if (nowQueue)
  31                         {
  32                                 EnqueueNode(node);
  33                                 return true;
  34                         }
  35                         else
  36                         {
  37                                 return base.VisitNode(node);
  38                         }
  39                 }
  40
  41                 public override void VisitCompilationUnit(CompilationUnit compilationUnit)
  42                 {
  43                         base.VisitCompilationUnit(compilationUnit);
  44
  45                         ProcessNodesInQueue();
  46                 }
  47
  48                 public override bool VisitSourceUnit(SourceUnit unit)
  49                 {
  50                         base.VisitSourceUnit(unit);
  51
  52                         ProcessNodesInQueue();
  53
  54                         return true;
  55                 }
  56
  57                 public override bool VisitEnter(NamespaceDescriptor ns)
  58                 {
  59                         nowQueue = true;
  60                         return base.VisitEnter(ns);
  61                 }
  62
  63                 public override bool VisitLeave(NamespaceDescriptor ns)
  64                 {
  65                         nowQueue = false;
  66                         return base.VisitLeave(ns);
  67                 }
  68
  69                 public override bool VisitEnter(TypeDefinitionStatement typeDef)
  70                 {
  71                         nowQueue = true;
  72                         return base.VisitEnter(typeDef);
  73                 }
  74
  75                 public override bool VisitLeave(TypeDefinitionStatement typeDef)
  76                 {
  77                         nowQueue = false;
  78                         return base.VisitLeave(typeDef);
  79                 }
  80
  81                 public override bool VisitEnter(MethodDefinitionStatement methodDef)
  82                 {
  83                         nowQueue = true;
  84                         return base.VisitEnter(methodDef);
  85                 }
  86
  87                 public override bool VisitLeave(MethodDefinitionStatement methodDef)
  88                 {
  89                         nowQueue = false;
  90                         return base.VisitLeave(methodDef);
  91                 }
  92
  93                 private void EnqueueNode(IVisitableNode node)
  94                 {
  95                         nodesToBeVisited.Enqueue(node);
  96                 }
  97
  98                 private void ProcessNodesInQueue()
  99                 {
 100                         while(nodesToBeVisited.Count != 0)
 101                         {
 102                                 VisitNode(nodesToBeVisited.Dequeue() as IVisitableNode);
 103                         }
 104                 }
 105         }
 106 }