about summary refs log tree commit diff
path: root/src/libnix/aterm.hh
blob: 16d8d6bb6dfd2458a771d6ece85429d48f0a4e9d (plain) (blame)
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
#ifndef __ATERM_H
#define __ATERM_H

extern "C" {
#include <aterm2.h>
}

#include "util.hh"


/* Print an ATerm. */
string atPrint(ATerm t);

/* Write an ATerm to an output stream. */
ostream & operator << (ostream & stream, ATerm e);

class ATermIterator
{
    ATermList t;

public:
    ATermIterator(ATermList _t) : t(_t) { }
    ATermIterator & operator ++ ()
    {
        t = ATgetNext(t);
        return *this;
    }
    ATerm operator * ()
    {
        return ATgetFirst(t);
    }
    operator bool ()
    {
        return t != ATempty;
    }
};


/* Type-safe matching. */

struct ATMatcher 
{
    ATerm t;
    int pos;
    const static int failPos = -2;
    const static int funPos = -1;

    ATMatcher() : t(0), pos(failPos)
    {
    }

    operator bool() const
    {
        return pos != failPos;
    }
};

/* Initiate matching of a term. */
ATMatcher & atMatch(ATMatcher & pos, ATerm t);

/* Get the next argument of an application. */
ATMatcher & operator >> (ATMatcher & pos, ATerm & out);

/* Get the name of the function symbol of an applicatin, or the next
   argument of an application as a string. */
ATMatcher & operator >> (ATMatcher & pos, string & out);

/* Like the previous, but check that the string is equal to the given
   string. */
ATMatcher & operator >> (ATMatcher & pos, const string & s);

/* Get the next argument of an application, and verify that it is a
   list. */
ATMatcher & operator >> (ATMatcher & pos, ATermList & out);


#endif /* !__ATERM_H */