URL https://opencores.org/ocsvn/openrisc/openrisc/trunk

Subversion Repositories openrisc

[/] [openrisc/] [trunk/] [gnu-dev/] [or1k-gcc/] [libgo/] [go/] [go/] [ast/] [walk.go] - Blame information for rev 747

Details | Compare with Previous | View Log


// Copyright 2009 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.
 
package ast
 
import "fmt"
 
// A Visitor's Visit method is invoked for each node encountered by Walk.
// If the result visitor w is not nil, Walk visits each of the children
// of node with the visitor w, followed by a call of w.Visit(nil).
type Visitor interface {
        Visit(node Node) (w Visitor)
}
 
// Helper functions for common node lists. They may be empty.
 
func walkIdentList(v Visitor, list []*Ident) {
        for _, x := range list {
                Walk(v, x)
        }
}
 
func walkExprList(v Visitor, list []Expr) {
        for _, x := range list {
                Walk(v, x)
        }
}
 
func walkStmtList(v Visitor, list []Stmt) {
        for _, x := range list {
                Walk(v, x)
        }
}
 
func walkDeclList(v Visitor, list []Decl) {
        for _, x := range list {
                Walk(v, x)
        }
}
 
// TODO(gri): Investigate if providing a closure to Walk leads to
//            simpler use (and may help eliminate Inspect in turn).
 
// Walk traverses an AST in depth-first order: It starts by calling
// v.Visit(node); node must not be nil. If the visitor w returned by
// v.Visit(node) is not nil, Walk is invoked recursively with visitor
// w for each of the non-nil children of node, followed by a call of
// w.Visit(nil).
//
func Walk(v Visitor, node Node) {
        if v = v.Visit(node); v == nil {
                return
        }
 
        // walk children
        // (the order of the cases matches the order
        // of the corresponding node types in ast.go)
        switch n := node.(type) {
        // Comments and fields
        case *Comment:
                // nothing to do
 
        case *CommentGroup:
                for _, c := range n.List {
                        Walk(v, c)
                }
 
        case *Field:
                if n.Doc != nil {
                        Walk(v, n.Doc)
                }
                walkIdentList(v, n.Names)
                Walk(v, n.Type)
                if n.Tag != nil {
                        Walk(v, n.Tag)
                }
                if n.Comment != nil {
                        Walk(v, n.Comment)
                }
 
        case *FieldList:
                for _, f := range n.List {
                        Walk(v, f)
                }
 
        // Expressions
        case *BadExpr, *Ident, *BasicLit:
                // nothing to do
 
        case *Ellipsis:
                if n.Elt != nil {
                        Walk(v, n.Elt)
                }
 
        case *FuncLit:
                Walk(v, n.Type)
                Walk(v, n.Body)
 
        case *CompositeLit:
                if n.Type != nil {
                        Walk(v, n.Type)
                }
                walkExprList(v, n.Elts)
 
        case *ParenExpr:
                Walk(v, n.X)
 
        case *SelectorExpr:
                Walk(v, n.X)
                Walk(v, n.Sel)
 
        case *IndexExpr:
                Walk(v, n.X)
                Walk(v, n.Index)
 
        case *SliceExpr:
                Walk(v, n.X)
                if n.Low != nil {
                        Walk(v, n.Low)
                }
                if n.High != nil {
                        Walk(v, n.High)
                }
 
        case *TypeAssertExpr:
                Walk(v, n.X)
                if n.Type != nil {
                        Walk(v, n.Type)
                }
 
        case *CallExpr:
                Walk(v, n.Fun)
                walkExprList(v, n.Args)
 
        case *StarExpr:
                Walk(v, n.X)
 
        case *UnaryExpr:
                Walk(v, n.X)
 
        case *BinaryExpr:
                Walk(v, n.X)
                Walk(v, n.Y)
 
        case *KeyValueExpr:
                Walk(v, n.Key)
                Walk(v, n.Value)
 
        // Types
        case *ArrayType:
                if n.Len != nil {
                        Walk(v, n.Len)
                }
                Walk(v, n.Elt)
 
        case *StructType:
                Walk(v, n.Fields)
 
        case *FuncType:
                Walk(v, n.Params)
                if n.Results != nil {
                        Walk(v, n.Results)
                }
 
        case *InterfaceType:
                Walk(v, n.Methods)
 
        case *MapType:
                Walk(v, n.Key)
                Walk(v, n.Value)
 
        case *ChanType:
                Walk(v, n.Value)
 
        // Statements
        case *BadStmt:
                // nothing to do
 
        case *DeclStmt:
                Walk(v, n.Decl)
 
        case *EmptyStmt:
                // nothing to do
 
        case *LabeledStmt:
                Walk(v, n.Label)
                Walk(v, n.Stmt)
 
        case *ExprStmt:
                Walk(v, n.X)
 
        case *SendStmt:
                Walk(v, n.Chan)
                Walk(v, n.Value)
 
        case *IncDecStmt:
                Walk(v, n.X)
 
        case *AssignStmt:
                walkExprList(v, n.Lhs)
                walkExprList(v, n.Rhs)
 
        case *GoStmt:
                Walk(v, n.Call)
 
        case *DeferStmt:
                Walk(v, n.Call)
 
        case *ReturnStmt:
                walkExprList(v, n.Results)
 
        case *BranchStmt:
                if n.Label != nil {
                        Walk(v, n.Label)
                }
 
        case *BlockStmt:
                walkStmtList(v, n.List)
 
        case *IfStmt:
                if n.Init != nil {
                        Walk(v, n.Init)
                }
                Walk(v, n.Cond)
                Walk(v, n.Body)
                if n.Else != nil {
                        Walk(v, n.Else)
                }
 
        case *CaseClause:
                walkExprList(v, n.List)
                walkStmtList(v, n.Body)
 
        case *SwitchStmt:
                if n.Init != nil {
                        Walk(v, n.Init)
                }
                if n.Tag != nil {
                        Walk(v, n.Tag)
                }
                Walk(v, n.Body)
 
        case *TypeSwitchStmt:
                if n.Init != nil {
                        Walk(v, n.Init)
                }
                Walk(v, n.Assign)
                Walk(v, n.Body)
 
        case *CommClause:
                if n.Comm != nil {
                        Walk(v, n.Comm)
                }
                walkStmtList(v, n.Body)
 
        case *SelectStmt:
                Walk(v, n.Body)
 
        case *ForStmt:
                if n.Init != nil {
                        Walk(v, n.Init)
                }
                if n.Cond != nil {
                        Walk(v, n.Cond)
                }
                if n.Post != nil {
                        Walk(v, n.Post)
                }
                Walk(v, n.Body)
 
        case *RangeStmt:
                Walk(v, n.Key)
                if n.Value != nil {
                        Walk(v, n.Value)
                }
                Walk(v, n.X)
                Walk(v, n.Body)
 
        // Declarations
        case *ImportSpec:
                if n.Doc != nil {
                        Walk(v, n.Doc)
                }
                if n.Name != nil {
                        Walk(v, n.Name)
                }
                Walk(v, n.Path)
                if n.Comment != nil {
                        Walk(v, n.Comment)
                }
 
        case *ValueSpec:
                if n.Doc != nil {
                        Walk(v, n.Doc)
                }
                walkIdentList(v, n.Names)
                if n.Type != nil {
                        Walk(v, n.Type)
                }
                walkExprList(v, n.Values)
                if n.Comment != nil {
                        Walk(v, n.Comment)
                }
 
        case *TypeSpec:
                if n.Doc != nil {
                        Walk(v, n.Doc)
                }
                Walk(v, n.Name)
                Walk(v, n.Type)
                if n.Comment != nil {
                        Walk(v, n.Comment)
                }
 
        case *BadDecl:
                // nothing to do
 
        case *GenDecl:
                if n.Doc != nil {
                        Walk(v, n.Doc)
                }
                for _, s := range n.Specs {
                        Walk(v, s)
                }
 
        case *FuncDecl:
                if n.Doc != nil {
                        Walk(v, n.Doc)
                }
                if n.Recv != nil {
                        Walk(v, n.Recv)
                }
                Walk(v, n.Name)
                Walk(v, n.Type)
                if n.Body != nil {
                        Walk(v, n.Body)
                }
 
        // Files and packages
        case *File:
                if n.Doc != nil {
                        Walk(v, n.Doc)
                }
                Walk(v, n.Name)
                walkDeclList(v, n.Decls)
                for _, g := range n.Comments {
                        Walk(v, g)
                }
                // don't walk n.Comments - they have been
                // visited already through the individual
                // nodes
 
        case *Package:
                for _, f := range n.Files {
                        Walk(v, f)
                }
 
        default:
                fmt.Printf("ast.Walk: unexpected node type %T", n)
                panic("ast.Walk")
        }
 
        v.Visit(nil)
}
 
type inspector func(Node) bool
 
func (f inspector) Visit(node Node) Visitor {
        if f(node) {
                return f
        }
        return nil
}
 
// Inspect traverses an AST in depth-first order: It starts by calling
// f(node); node must not be nil. If f returns true, Inspect invokes f
// for all the non-nil children of node, recursively.
//
func Inspect(node Node, f func(Node) bool) {
        Walk(inspector(f), node)
}

Browse

Tools

Subversion Repositories openrisc

[/] [openrisc/] [trunk/] [gnu-dev/] [or1k-gcc/] [libgo/] [go/] [go/] [ast/] [walk.go] - Blame information for rev 747

Line No.	Rev	Author	Line
1	747	jeremybenn	`// Copyright 2009 The Go Authors. All rights reserved.`
2			`// Use of this source code is governed by a BSD-style`
3			`// license that can be found in the LICENSE file.`
4
5			`package ast`
6
7			`import "fmt"`
8
9			`// A Visitor's Visit method is invoked for each node encountered by Walk.`
10			`// If the result visitor w is not nil, Walk visits each of the children`
11			`// of node with the visitor w, followed by a call of w.Visit(nil).`
12			`type Visitor interface {`
13			`Visit(node Node) (w Visitor)`
14			`}`
15
16			`// Helper functions for common node lists. They may be empty.`
17
18			`func walkIdentList(v Visitor, list []*Ident) {`
19			`for _, x := range list {`
20			`Walk(v, x)`
21			`}`
22			`}`
23
24			`func walkExprList(v Visitor, list []Expr) {`
25			`for _, x := range list {`
26			`Walk(v, x)`
27			`}`
28			`}`
29
30			`func walkStmtList(v Visitor, list []Stmt) {`
31			`for _, x := range list {`
32			`Walk(v, x)`
33			`}`
34			`}`
35
36			`func walkDeclList(v Visitor, list []Decl) {`
37			`for _, x := range list {`
38			`Walk(v, x)`
39			`}`
40			`}`
41
42			`// TODO(gri): Investigate if providing a closure to Walk leads to`
43			`// simpler use (and may help eliminate Inspect in turn).`
44
45			`// Walk traverses an AST in depth-first order: It starts by calling`
46			`// v.Visit(node); node must not be nil. If the visitor w returned by`
47			`// v.Visit(node) is not nil, Walk is invoked recursively with visitor`
48			`// w for each of the non-nil children of node, followed by a call of`
49			`// w.Visit(nil).`
50			`//`
51			`func Walk(v Visitor, node Node) {`
52			`if v = v.Visit(node); v == nil {`
53			`return`
54			`}`
55
56			`// walk children`
57			`// (the order of the cases matches the order`
58			`// of the corresponding node types in ast.go)`
59			`switch n := node.(type) {`
60			`// Comments and fields`
61			`case *Comment:`
62			`// nothing to do`
63
64			`case *CommentGroup:`
65			`for _, c := range n.List {`
66			`Walk(v, c)`
67			`}`
68
69			`case *Field:`
70			`if n.Doc != nil {`
71			`Walk(v, n.Doc)`
72			`}`
73			`walkIdentList(v, n.Names)`
74			`Walk(v, n.Type)`
75			`if n.Tag != nil {`
76			`Walk(v, n.Tag)`
77			`}`
78			`if n.Comment != nil {`
79			`Walk(v, n.Comment)`
80			`}`
81
82			`case *FieldList:`
83			`for _, f := range n.List {`
84			`Walk(v, f)`
85			`}`
86
87			`// Expressions`
88			`case BadExpr, Ident, *BasicLit:`
89			`// nothing to do`
90
91			`case *Ellipsis:`
92			`if n.Elt != nil {`
93			`Walk(v, n.Elt)`
94			`}`
95
96			`case *FuncLit:`
97			`Walk(v, n.Type)`
98			`Walk(v, n.Body)`
99
100			`case *CompositeLit:`
101			`if n.Type != nil {`
102			`Walk(v, n.Type)`
103			`}`
104			`walkExprList(v, n.Elts)`
105
106			`case *ParenExpr:`
107			`Walk(v, n.X)`
108
109			`case *SelectorExpr:`
110			`Walk(v, n.X)`
111			`Walk(v, n.Sel)`
112
113			`case *IndexExpr:`
114			`Walk(v, n.X)`
115			`Walk(v, n.Index)`
116
117			`case *SliceExpr:`
118			`Walk(v, n.X)`
119			`if n.Low != nil {`
120			`Walk(v, n.Low)`
121			`}`
122			`if n.High != nil {`
123			`Walk(v, n.High)`
124			`}`
125
126			`case *TypeAssertExpr:`
127			`Walk(v, n.X)`
128			`if n.Type != nil {`
129			`Walk(v, n.Type)`
130			`}`
131
132			`case *CallExpr:`
133			`Walk(v, n.Fun)`
134			`walkExprList(v, n.Args)`
135
136			`case *StarExpr:`
137			`Walk(v, n.X)`
138
139			`case *UnaryExpr:`
140			`Walk(v, n.X)`
141
142			`case *BinaryExpr:`
143			`Walk(v, n.X)`
144			`Walk(v, n.Y)`
145
146			`case *KeyValueExpr:`
147			`Walk(v, n.Key)`
148			`Walk(v, n.Value)`
149
150			`// Types`
151			`case *ArrayType:`
152			`if n.Len != nil {`
153			`Walk(v, n.Len)`
154			`}`
155			`Walk(v, n.Elt)`
156
157			`case *StructType:`
158			`Walk(v, n.Fields)`
159
160			`case *FuncType:`
161			`Walk(v, n.Params)`
162			`if n.Results != nil {`
163			`Walk(v, n.Results)`
164			`}`
165
166			`case *InterfaceType:`
167			`Walk(v, n.Methods)`
168
169			`case *MapType:`
170			`Walk(v, n.Key)`
171			`Walk(v, n.Value)`
172
173			`case *ChanType:`
174			`Walk(v, n.Value)`
175
176			`// Statements`
177			`case *BadStmt:`
178			`// nothing to do`
179
180			`case *DeclStmt:`
181			`Walk(v, n.Decl)`
182
183			`case *EmptyStmt:`
184			`// nothing to do`
185
186			`case *LabeledStmt:`
187			`Walk(v, n.Label)`
188			`Walk(v, n.Stmt)`
189
190			`case *ExprStmt:`
191			`Walk(v, n.X)`
192
193			`case *SendStmt:`
194			`Walk(v, n.Chan)`
195			`Walk(v, n.Value)`
196
197			`case *IncDecStmt:`
198			`Walk(v, n.X)`
199
200			`case *AssignStmt:`
201			`walkExprList(v, n.Lhs)`
202			`walkExprList(v, n.Rhs)`
203
204			`case *GoStmt:`
205			`Walk(v, n.Call)`
206
207			`case *DeferStmt:`
208			`Walk(v, n.Call)`
209
210			`case *ReturnStmt:`
211			`walkExprList(v, n.Results)`
212
213			`case *BranchStmt:`
214			`if n.Label != nil {`
215			`Walk(v, n.Label)`
216			`}`
217
218			`case *BlockStmt:`
219			`walkStmtList(v, n.List)`
220
221			`case *IfStmt:`
222			`if n.Init != nil {`
223			`Walk(v, n.Init)`
224			`}`
225			`Walk(v, n.Cond)`
226			`Walk(v, n.Body)`
227			`if n.Else != nil {`
228			`Walk(v, n.Else)`
229			`}`
230
231			`case *CaseClause:`
232			`walkExprList(v, n.List)`
233			`walkStmtList(v, n.Body)`
234
235			`case *SwitchStmt:`
236			`if n.Init != nil {`
237			`Walk(v, n.Init)`
238			`}`
239			`if n.Tag != nil {`
240			`Walk(v, n.Tag)`
241			`}`
242			`Walk(v, n.Body)`
243
244			`case *TypeSwitchStmt:`
245			`if n.Init != nil {`
246			`Walk(v, n.Init)`
247			`}`
248			`Walk(v, n.Assign)`
249			`Walk(v, n.Body)`
250
251			`case *CommClause:`
252			`if n.Comm != nil {`
253			`Walk(v, n.Comm)`
254			`}`
255			`walkStmtList(v, n.Body)`
256
257			`case *SelectStmt:`
258			`Walk(v, n.Body)`
259
260			`case *ForStmt:`
261			`if n.Init != nil {`
262			`Walk(v, n.Init)`
263			`}`
264			`if n.Cond != nil {`
265			`Walk(v, n.Cond)`
266			`}`
267			`if n.Post != nil {`
268			`Walk(v, n.Post)`
269			`}`
270			`Walk(v, n.Body)`
271
272			`case *RangeStmt:`
273			`Walk(v, n.Key)`
274			`if n.Value != nil {`
275			`Walk(v, n.Value)`
276			`}`
277			`Walk(v, n.X)`
278			`Walk(v, n.Body)`
279
280			`// Declarations`
281			`case *ImportSpec:`
282			`if n.Doc != nil {`
283			`Walk(v, n.Doc)`
284			`}`
285			`if n.Name != nil {`
286			`Walk(v, n.Name)`
287			`}`
288			`Walk(v, n.Path)`
289			`if n.Comment != nil {`
290			`Walk(v, n.Comment)`
291			`}`
292
293			`case *ValueSpec:`
294			`if n.Doc != nil {`
295			`Walk(v, n.Doc)`
296			`}`
297			`walkIdentList(v, n.Names)`
298			`if n.Type != nil {`
299			`Walk(v, n.Type)`
300			`}`
301			`walkExprList(v, n.Values)`
302			`if n.Comment != nil {`
303			`Walk(v, n.Comment)`
304			`}`
305
306			`case *TypeSpec:`
307			`if n.Doc != nil {`
308			`Walk(v, n.Doc)`
309			`}`
310			`Walk(v, n.Name)`
311			`Walk(v, n.Type)`
312			`if n.Comment != nil {`
313			`Walk(v, n.Comment)`
314			`}`
315
316			`case *BadDecl:`
317			`// nothing to do`
318
319			`case *GenDecl:`
320			`if n.Doc != nil {`
321			`Walk(v, n.Doc)`
322			`}`
323			`for _, s := range n.Specs {`
324			`Walk(v, s)`
325			`}`
326
327			`case *FuncDecl:`
328			`if n.Doc != nil {`
329			`Walk(v, n.Doc)`
330			`}`
331			`if n.Recv != nil {`
332			`Walk(v, n.Recv)`
333			`}`
334			`Walk(v, n.Name)`
335			`Walk(v, n.Type)`
336			`if n.Body != nil {`
337			`Walk(v, n.Body)`
338			`}`
339
340			`// Files and packages`
341			`case *File:`
342			`if n.Doc != nil {`
343			`Walk(v, n.Doc)`
344			`}`
345			`Walk(v, n.Name)`
346			`walkDeclList(v, n.Decls)`
347			`for _, g := range n.Comments {`
348			`Walk(v, g)`
349			`}`
350			`// don't walk n.Comments - they have been`
351			`// visited already through the individual`
352			`// nodes`
353
354			`case *Package:`
355			`for _, f := range n.Files {`
356			`Walk(v, f)`
357			`}`
358
359			`default:`
360			`fmt.Printf("ast.Walk: unexpected node type %T", n)`
361			`panic("ast.Walk")`
362			`}`
363
364			`v.Visit(nil)`
365			`}`
366
367			`type inspector func(Node) bool`
368
369			`func (f inspector) Visit(node Node) Visitor {`
370			`if f(node) {`
371			`return f`
372			`}`
373			`return nil`
374			`}`
375
376			`// Inspect traverses an AST in depth-first order: It starts by calling`
377			`// f(node); node must not be nil. If f returns true, Inspect invokes f`
378			`// for all the non-nil children of node, recursively.`
379			`//`
380			`func Inspect(node Node, f func(Node) bool) {`
381			`Walk(inspector(f), node)`
382			`}`