aboutsummaryrefslogtreecommitdiff
blob: efa4d7e592605ed943c404b30e66d81a1316310d (plain)
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
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
import glob
from ply import lex
from ply import yacc

#lex stuff begins here

def scanincludes(string,inclst,curdir):
    tokens = (
            "GINCLUDE",
            "LINCLUDE",
            #"BUNDLEINC",
            "IFDEF",
            "ENDIF",
            )

    states = (
            ("com","exclusive"), #comment
            ("ifdef","inclusive"),
            )

    t_ANY_ignore = " \t"

    def t_begin_com(t):
        r"/\*"
        t.lexer.push_state("com")

    def t_com_end(t):
        r"\*/"
        t.lexer.pop_state()
        pass

    def t_line_com(t):
        r"//.*"
        pass

    def t_ANY_begin_if0(t):
        r"\#if[ \t]+0"
        t.lexer.push_state("com")

    def t_com_endif(t):
        r"\#endif"
        t.lexer.pop_state()
        pass

    def t_com_ifdef(t):
        r"\#ifdef"
        t.lexer.push_state("com")

    def t_IFDEF(t):
        r"\#ifdef[ \t]+[a-zA-Z_][a-zA-Z0-9_]*"
        t.value = t.value[6:].strip() #return the ifdef name
        t.lexer.push_state("ifdef")
        return t

    def t_ifdef_ENDIF(t):
        r"\#endif"
        t.lexer.pop_state()
        return t

    def t_GINCLUDE(t):
        r"\#[Ii][Nn][Cc][Ll][Uu][Dd][Ee][ \t]+<.*\.h>"
        t.value = t.value[8:].strip().strip("<>")
        return t

    def t_LINCLUDE(t):
        r"\#[Ii][Nn][Cc][Ll][Uu][Dd][Ee][ \t]+\".*\.h\""
        t.value = t.value[8:].strip().strip('""')
        return t

    def t_BUNDLEINC(t):
        r"\#[Ii][Nn][Cc][Ll][Uu][Dd][Ee][ \t]+<.*>"
        pass

    def t_ANY_error(t):
        #print("Illegal character '%s'" % t.value[0])
        t.lexer.skip(1)

    lexer = lex.lex()

    #lexer.input(string)
    #
    #for tok in lexer:
    #    print(tok)
    #
    #YACC stuff here

    def p_includes2(p):
        """
        includes : includes ginc
        """
        p[1][0].add(p[2])
        p[0] = p[1]

    def p_lincludes(p):
        """
        includes : includes linc
        """
        if islocalinc(p[2],curdir):
            p[1][1].add(p[2])
        else:
            p[1][0].add(p[2])
        p[0] = p[1]

    def p_ifdef(p):
        """
        includes : includes IFDEF includes ENDIF
                 | IFDEF includes ENDIF
        """
        if len(p) == 5:
            p[1][2] = addnewifdefs(p[1][2],{p[2] : p[3]})
            p[0] = p[1]
        else:
            ifdef = {}
            ifdef[p[1]] = p[2]
            p[0] = [set(),set(),ifdef]

    def p_ginc(p):
        "includes : ginc"
        globinc = set()
        globinc.add(p[1])
        p[0] = [globinc,set(),{}]

    def p_linc(p):
        "includes : linc"
        locinc = set()
        locinc.add(p[1])
        if islocalinc(p[1], curdir):
            p[0] = [set(),locinc,{}]
        else:
            p[0] = [locinc,set(),{}]

    def p_ginclude(p):
        "ginc : GINCLUDE"
        p[0] = p[1]

    def p_linclude(p):
        "linc : LINCLUDE"
        p[0] = p[1]

    def p_error(p):
        #print("syntax error at '%s'" % p.type)
        pass

    yacc.yacc()

    newinclst = yacc.parse(string)
    if newinclst == None:
        #Check if the file didn't have any includes
        return(inclst)
    newinclst = addnewincludes(newinclst,inclst)
    return(newinclst)

def islocalinc(inc, curdir):
    if glob.glob(curdir + "/" + inc) == []:
        return False
    else:
        return True


def addnewincludes(inclist1,inclist2):
    #come up with better names!!
    inclist1[0] = inclist1[0] | inclist2[0]
    inclist1[1] = inclist1[1] | inclist2[1]
    inclist1[2] = addnewifdefs(inclist1[2],inclist2[2])
    return(inclist1)

def addnewifdefs(dict1,dict2):
    if dict1 == {} and dict2 == {}:
        #we are done here
        return(dict())
    dups = dict1.keys() & dict2.keys()
    if dups == set():
        #no duplicates, empty set()
        for name in dict2:
            dict1[name] = dict2[name]
        return(dict1)

    for name in dups:
        dict1[name][0] = dict1[name][0] | dict2[name][0]
        dict1[name][1] = dict1[name][1] | dict2[name][1]
        dict1[name][2] = addnewifdefs(dict1[name][2],dict2[name][2])
    return(dict1)