1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
|
#ifndef __MY_AST_H__
# define __MY_AST_H__
#include <antlr/CommonAST.hpp>
class MyAST;
typedef ANTLR_USE_NAMESPACE(antlr)ASTRefCount<MyAST> RefMyAST;
/** Custom AST class that adds line numbers to the AST nodes.
* easily extended with columns. Filenames will take more work since
* you'll need a custom token class as well (one that contains the
* filename)
*/
class MyAST : public ANTLR_USE_NAMESPACE(antlr)CommonAST {
public:
// copy constructor
MyAST( const MyAST& other )
: CommonAST(other)
, line(other.line)
{
}
// Default constructor
MyAST( void ) : CommonAST(), line(0) {}
virtual ~MyAST( void ) {}
// get the line number of the node (or try to derive it from the child node
virtual int getLine( void ) const
{
// most of the time the line number is not set if the node is a
// imaginary one. Usually this means it has a child. Refer to the
// child line number. Of course this could be extended a bit.
// based on an example by Peter Morling.
if ( line != 0 )
return line;
if( getFirstChild() )
return ( RefMyAST(getFirstChild())->getLine() );
return 0;
}
virtual void setLine( int l )
{
line = l;
}
/** the initialize methods are called by the tree building constructs
* depending on which version is called the line number is filled in.
* e.g. a bit depending on how the node is constructed it will have the
* line number filled in or not (imaginary nodes!).
*/
virtual void initialize(int t, const ANTLR_USE_NAMESPACE(std)string& txt)
{
CommonAST::initialize(t,txt);
line = 0;
}
virtual void initialize( ANTLR_USE_NAMESPACE(antlr)RefToken t )
{
CommonAST::initialize(t);
line = t->getLine();
}
virtual void initialize( RefMyAST ast )
{
CommonAST::initialize(ANTLR_USE_NAMESPACE(antlr)RefAST(ast));
line = ast->getLine();
}
// for convenience will also work without
void addChild( RefMyAST c )
{
BaseAST::addChild( ANTLR_USE_NAMESPACE(antlr)RefAST(c) );
}
// for convenience will also work without
void setNextSibling( RefMyAST c )
{
BaseAST::setNextSibling( ANTLR_USE_NAMESPACE(antlr)RefAST(c) );
}
// provide a clone of the node (no sibling/child pointers are copied)
virtual ANTLR_USE_NAMESPACE(antlr)RefAST clone( void )
{
return ANTLR_USE_NAMESPACE(antlr)RefAST(new MyAST(*this));
}
static ANTLR_USE_NAMESPACE(antlr)RefAST factory( void )
{
return ANTLR_USE_NAMESPACE(antlr)RefAST(RefMyAST(new MyAST()));
}
private:
int line;
};
#endif
|
