Coccinelle release 0.2.5-rc7.
[bpt/coccinelle.git] / parsing_c / token_helpers.ml
CommitLineData
0708f913 1(* Yoann Padioleau
ae4735db
C
2 *
3 * Copyright (C) 2010, University of Copenhagen DIKU and INRIA.
0708f913
C
4 * Copyright (C) 2007, 2008 Ecole des Mines de Nantes
5 *
6 * This program is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU General Public License (GPL)
8 * version 2 as published by the Free Software Foundation.
ae4735db 9 *
0708f913
C
10 * This program is distributed in the hope that it will be useful,
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 * file license.txt for more details.
14 *)
15
16
34e49164
C
17open Common
18
19open Parser_c
20
21(*****************************************************************************)
22(* Is_xxx, categories *)
23(*****************************************************************************)
24
ae4735db 25(* could define a type token_class = Comment | Ident | Operator | ...
0708f913 26 * update: now token_c can maybe do that.
ae4735db 27 * but still, sometimes tokens belon to multiple classes. Could maybe
0708f913 28 * return then a set of classes.
91eba41f
C
29 *)
30
34e49164
C
31let is_space = function
32 | TCommentSpace _ -> true
485bce71 33 | TCommentNewline _ -> true
34e49164
C
34 | _ -> false
35
485bce71
C
36let is_whitespace = is_space
37
0708f913 38let is_just_comment_or_space = function
34e49164
C
39 | TComment _ -> true
40 | TCommentSpace _ -> true
485bce71 41 | TCommentNewline _ -> true
34e49164 42 | _ -> false
0708f913 43let is_real_comment = is_just_comment_or_space
34e49164
C
44
45let is_just_comment = function
46 | TComment _ -> true
47 | _ -> false
48
0708f913
C
49
50
51
34e49164 52let is_comment = function
ae4735db
C
53 | TComment _
54 | TCommentSpace _ | TCommentNewline _
55 | TCommentCpp _
34e49164
C
56 | TCommentMisc _ -> true
57 | _ -> false
58
0708f913
C
59(* coupling with comment_annotater_c.ml.
60 * In fact more tokens than comments are not in the ast, but
ae4735db 61 * they were usually temporally created by ocamllex and removed
0708f913
C
62 * in parsing_hacks.
63*)
64let is_not_in_ast = is_comment
34e49164
C
65
66let is_fake_comment = function
ae4735db 67 | TCommentCpp _ | TCommentMisc _
34e49164
C
68 -> true
69 | _ -> false
70
ae4735db 71let is_not_comment x =
34e49164
C
72 not (is_comment x)
73
74
0708f913 75(* ---------------------------------------------------------------------- *)
485bce71 76
34e49164 77let is_cpp_instruction = function
ae4735db 78 | TInclude _
485bce71 79 | TDefine _
ae4735db 80 | TIfdef _ | TIfdefelse _ | TIfdefelif _ | TEndif _
34e49164 81 | TIfdefBool _ | TIfdefMisc _ | TIfdefVersion _
ae4735db 82 | TUndef _
485bce71
C
83 | TCppDirectiveOther _
84 -> true
85 | _ -> false
86
87
88let is_gcc_token = function
ae4735db
C
89 | Tasm _
90 | Tinline _
91 | Tattribute _
92 | Ttypeof _
34e49164
C
93 -> true
94 | _ -> false
95
96
97
98
0708f913 99(* ---------------------------------------------------------------------- *)
34e49164
C
100let is_opar = function
101 | TOPar _ | TOParDefine _ -> true
102 | _ -> false
103
104let is_cpar = function
105 | TCPar _ | TCParEOL _ -> true
106 | _ -> false
107
485bce71
C
108
109let is_obrace = function
110 | TOBrace _ | TOBraceDefineInit _ -> true
111 | _ -> false
112
113let is_cbrace = function
114 | TCBrace _ -> true
ae4735db 115 | _ -> false
485bce71
C
116
117
118
119
0708f913 120(* ---------------------------------------------------------------------- *)
708f4980
C
121
122(* end of file *)
34e49164
C
123let is_eof = function
124 | EOF x -> true
125 | _ -> false
126
485bce71 127
708f4980
C
128(* end of macro *)
129let is_eom = function
130 | TDefEOL _ -> true
131 | _ -> false
485bce71 132
34e49164 133let is_statement = function
ae4735db 134 | Tfor _ | Tdo _ | Tif _ | Twhile _ | Treturn _
34e49164 135 | Tbreak _ | Telse _ | Tswitch _ | Tcase _ | Tcontinue _
ae4735db 136 | Tgoto _
34e49164
C
137 | TPtVirg _
138 | TMacroIterator _
139 -> true
140 | _ -> false
141
142(* is_start_of_something is used in parse_c for error recovery, to find
143 * a synchronisation token.
ae4735db 144 *
34e49164
C
145 * Would like to put TIdent or TDefine, TIfdef but they can be in the
146 * middle of a function, for instance with label:.
ae4735db 147 *
34e49164
C
148 * Could put Typedefident but fired ? it would work in error recovery
149 * on the already_passed tokens, which has been already gone in the
150 * Parsing_hacks.lookahead machinery, but it will not work on the
151 * "next" tokens. But because the namespace for labels is different
152 * from namespace for ident/typedef, we can use the name for a typedef
ae4735db
C
153 * for a label and so dangerous to put Typedefident at true here.
154 *
34e49164
C
155 * Can look in parser_c.output to know what can be at toplevel
156 * at the very beginning.
157 *)
158
159let is_start_of_something = function
ae4735db 160 | Tchar _ | Tshort _ | Tint _ | Tdouble _ | Tfloat _ | Tlong _
1eddfd50 161 | Tunsigned _ | Tsigned _ | Tvoid _ | Tsize_t _ | Tssize_t _ | Tptrdiff_t _
34e49164
C
162 | Tauto _ | Tregister _ | Textern _ | Tstatic _
163 | Tconst _ | Tvolatile _
164 | Ttypedef _
ae4735db 165 | Tstruct _ | Tunion _ | Tenum _
34e49164
C
166 -> true
167 | _ -> false
168
169
170
171let is_binary_operator = function
ae4735db
C
172 | TOrLog _ | TAndLog _ | TOr _ | TXor _ | TAnd _
173 | TEqEq _ | TNotEq _ | TInf _ | TSup _ | TInfEq _ | TSupEq _
174 | TShl _ | TShr _
175 | TPlus _ | TMinus _ | TMul _ | TDiv _ | TMod _
34e49164 176 -> true
ae4735db 177 | _ -> false
34e49164
C
178
179let is_stuff_taking_parenthized = function
ae4735db
C
180 | Tif _
181 | Twhile _
34e49164
C
182 | Tswitch _
183 | Ttypeof _
184 | TMacroIterator _
ae4735db 185 -> true
34e49164
C
186 | _ -> false
187
91eba41f 188
0708f913 189(* used in the algorithms for "10 most problematic errors" *)
91eba41f
C
190let is_ident_like = function
191 | TIdent _
192 | TypedefIdent _
193 | TIdentDefine _
194 | TDefParamVariadic _
195
196 | TUnknown _
197
198 | TMacroAttr _
199 | TMacroAttrStorage _
200 | TMacroStmt _
201 | TMacroString _
202 | TMacroDecl _
91eba41f
C
203 | TMacroDeclConst _
204 | TMacroIterator _
205 -> true
206
ae4735db 207 | _ -> false
91eba41f
C
208
209
34e49164
C
210(*****************************************************************************)
211(* Visitors *)
212(*****************************************************************************)
213
ae4735db 214(* Because ocamlyacc force us to do it that way. The ocamlyacc token
34e49164
C
215 * cant be a pair of a sum type, it must be directly a sum type.
216 *)
217let info_of_tok = function
218 | TString ((string, isWchar), i) -> i
219 | TChar ((string, isWchar), i) -> i
220 | TFloat ((string, floatType), i) -> i
221
222 | TAssign (assignOp, i) -> i
223
224 | TIdent (s, i) -> i
f59c9fb7 225 | Tconstructorname (s, i) -> i
34e49164
C
226 | TypedefIdent (s, i) -> i
227
228 | TInt (s, i) -> i
229
ae4735db 230 | TDefine (ii) -> ii
34e49164
C
231 | TInclude (includes, filename, inifdef, i1) -> i1
232
3a314143 233 | TUndef (ii) -> ii
485bce71
C
234 | TCppDirectiveOther (ii) -> ii
235
34e49164
C
236 | TIncludeStart (i1, inifdef) -> i1
237 | TIncludeFilename (s, i1) -> i1
238
239 | TDefEOL (i1) -> i1
240 | TOParDefine (i1) -> i1
241 | TIdentDefine (s, i) -> i
242 | TCppEscapedNewline (ii) -> ii
243 | TDefParamVariadic (s, i1) -> i1
244
b1b2de81
C
245 | TCppConcatOp (ii) -> ii
246
485bce71
C
247 | TOBraceDefineInit (i1) -> i1
248
34e49164
C
249 | TUnknown (i) -> i
250
b1b2de81 251 | TMacroIdentBuilder (s, i) -> i
485bce71
C
252 | TMacroAttr (s, i) -> i
253 | TMacroAttrStorage (s, i) -> i
254 | TMacroStmt (s, i) -> i
255 | TMacroString (s, i) -> i
34e49164
C
256 | TMacroDecl (s, i) -> i
257 | TMacroDeclConst (i) -> i
258 | TMacroIterator (s,i) -> i
259(* | TMacroTop (s,i) -> i *)
260 | TCParEOL (i1) -> i1
261
262 | TAction (i) -> i
263
264 | TComment (i) -> i
265 | TCommentSpace (i) -> i
266 | TCommentNewline (i) -> i
267 | TCommentCpp (cppkind, i) -> i
268 | TCommentMisc (i) -> i
269
0708f913
C
270 | TCommentSkipTagStart (i) -> i
271 | TCommentSkipTagEnd (i) -> i
272
485bce71
C
273 | TIfdef (_, i) -> i
274 | TIfdefelse (_, i) -> i
275 | TIfdefelif (_, i) -> i
276 | TEndif (_, i) -> i
277 | TIfdefBool (b, _, i) -> i
278 | TIfdefMisc (b, _, i) -> i
279 | TIfdefVersion (b, _, i) -> i
34e49164
C
280
281 | TOPar (i) -> i
282 | TCPar (i) -> i
283 | TOBrace (i) -> i
284 | TCBrace (i) -> i
285 | TOCro (i) -> i
286 | TCCro (i) -> i
287 | TDot (i) -> i
288 | TComma (i) -> i
289 | TPtrOp (i) -> i
290 | TInc (i) -> i
291 | TDec (i) -> i
292 | TEq (i) -> i
293 | TWhy (i) -> i
294 | TTilde (i) -> i
295 | TBang (i) -> i
296 | TEllipsis (i) -> i
297 | TDotDot (i) -> i
298 | TPtVirg (i) -> i
299 | TOrLog (i) -> i
300 | TAndLog (i) -> i
301 | TOr (i) -> i
302 | TXor (i) -> i
303 | TAnd (i) -> i
304 | TEqEq (i) -> i
305 | TNotEq (i) -> i
306 | TInf (i) -> i
307 | TSup (i) -> i
308 | TInfEq (i) -> i
309 | TSupEq (i) -> i
310 | TShl (i) -> i
311 | TShr (i) -> i
312 | TPlus (i) -> i
313 | TMinus (i) -> i
314 | TMul (i) -> i
315 | TDiv (i) -> i
316 | TMod (i) -> i
317
318 | Tchar (i) -> i
319 | Tshort (i) -> i
320 | Tint (i) -> i
321 | Tdouble (i) -> i
322 | Tfloat (i) -> i
323 | Tlong (i) -> i
324 | Tunsigned (i) -> i
325 | Tsigned (i) -> i
326 | Tvoid (i) -> i
1eddfd50
C
327 | Tsize_t (i) -> i
328 | Tssize_t (i) -> i
329 | Tptrdiff_t (i) -> i
34e49164
C
330 | Tauto (i) -> i
331 | Tregister (i) -> i
332 | Textern (i) -> i
333 | Tstatic (i) -> i
334 | Tconst (i) -> i
335 | Tvolatile (i) -> i
485bce71
C
336
337 | Trestrict (i) -> i
338
34e49164
C
339 | Tstruct (i) -> i
340 | Tenum (i) -> i
341 | Ttypedef (i) -> i
342 | Tunion (i) -> i
343 | Tbreak (i) -> i
344 | Telse (i) -> i
345 | Tswitch (i) -> i
346 | Tcase (i) -> i
347 | Tcontinue (i) -> i
348 | Tfor (i) -> i
349 | Tdo (i) -> i
350 | Tif (i) -> i
351 | Twhile (i) -> i
352 | Treturn (i) -> i
353 | Tgoto (i) -> i
354 | Tdefault (i) -> i
355 | Tsizeof (i) -> i
356 | Tasm (i) -> i
357 | Tattribute (i) -> i
978fd7e5 358 | TattributeNoarg (i) -> i
34e49164
C
359 | Tinline (i) -> i
360 | Ttypeof (i) -> i
f59c9fb7 361 | Tnew (i) -> i
4dfbc1c2
C
362 | Tdelete (i) -> i
363 | TOParCplusplusInit (i) -> i
34e49164
C
364
365 | EOF (i) -> i
ae4735db 366
34e49164
C
367
368
485bce71 369
34e49164
C
370(* used by tokens to complete the parse_info with filename, line, col infos *)
371let visitor_info_of_tok f = function
ae4735db
C
372 | TString ((s, isWchar), i) -> TString ((s, isWchar), f i)
373 | TChar ((s, isWchar), i) -> TChar ((s, isWchar), f i)
374 | TFloat ((s, floatType), i) -> TFloat ((s, floatType), f i)
375 | TAssign (assignOp, i) -> TAssign (assignOp, f i)
34e49164 376
f59c9fb7
C
377 | TIdent (s, i) -> TIdent (s, f i)
378 | Tconstructorname(s, i) -> Tconstructorname (s, f i)
379 | TypedefIdent (s, i) -> TypedefIdent (s, f i)
380 | TInt (s, i) -> TInt (s, f i)
34e49164 381
ae4735db 382 | TDefine (i1) -> TDefine(f i1)
34e49164 383
3a314143 384 | TUndef (i1) -> TUndef(f i1)
ae4735db 385 | TCppDirectiveOther (i1) -> TCppDirectiveOther(f i1)
485bce71 386
ae4735db 387 | TInclude (includes, filename, inifdef, i1) ->
34e49164
C
388 TInclude (includes, filename, inifdef, f i1)
389
390 | TIncludeStart (i1, inifdef) -> TIncludeStart (f i1, inifdef)
391 | TIncludeFilename (s, i1) -> TIncludeFilename (s, f i1)
392
393 | TCppEscapedNewline (i1) -> TCppEscapedNewline (f i1)
394 | TDefEOL (i1) -> TDefEOL (f i1)
b1b2de81
C
395
396 | TCppConcatOp (ii) -> TCppConcatOp (f ii)
397
34e49164
C
398 | TOParDefine (i1) -> TOParDefine (f i1)
399 | TIdentDefine (s, i) -> TIdentDefine (s, f i)
400
401 | TDefParamVariadic (s, i1) -> TDefParamVariadic (s, f i1)
402
485bce71
C
403 | TOBraceDefineInit (i1) -> TOBraceDefineInit (f i1)
404
34e49164
C
405
406 | TUnknown (i) -> TUnknown (f i)
407
b1b2de81 408 | TMacroIdentBuilder (s, i) -> TMacroIdentBuilder (s, f i)
485bce71
C
409 | TMacroAttr (s, i) -> TMacroAttr (s, f i)
410 | TMacroAttrStorage (s, i) -> TMacroAttrStorage (s, f i)
411 | TMacroStmt (s, i) -> TMacroStmt (s, f i)
412 | TMacroString (s, i) -> TMacroString (s, f i)
413 | TMacroDecl (s, i) -> TMacroDecl (s, f i)
34e49164 414 | TMacroDeclConst (i) -> TMacroDeclConst (f i)
485bce71 415 | TMacroIterator (s, i) -> TMacroIterator (s, f i)
34e49164
C
416(* | TMacroTop (s,i) -> TMacroTop (s,f i) *)
417 | TCParEOL (i) -> TCParEOL (f i)
418
419
420 | TAction (i) -> TAction (f i)
421
ae4735db
C
422 | TComment (i) -> TComment (f i)
423 | TCommentSpace (i) -> TCommentSpace (f i)
424 | TCommentNewline (i) -> TCommentNewline (f i)
425 | TCommentCpp (cppkind, i) -> TCommentCpp (cppkind, f i)
426 | TCommentMisc (i) -> TCommentMisc (f i)
427
428 | TCommentSkipTagStart (i) -> TCommentSkipTagStart (f i)
429 | TCommentSkipTagEnd (i) -> TCommentSkipTagEnd (f i)
430
431 | TIfdef (t, i) -> TIfdef (t, f i)
432 | TIfdefelse (t, i) -> TIfdefelse (t, f i)
433 | TIfdefelif (t, i) -> TIfdefelif (t, f i)
434 | TEndif (t, i) -> TEndif (t, f i)
435 | TIfdefBool (b, t, i) -> TIfdefBool (b, t, f i)
436 | TIfdefMisc (b, t, i) -> TIfdefMisc (b, t, f i)
437 | TIfdefVersion (b, t, i) -> TIfdefVersion (b, t, f i)
438
439 | TOPar (i) -> TOPar (f i)
440 | TCPar (i) -> TCPar (f i)
441 | TOBrace (i) -> TOBrace (f i)
442 | TCBrace (i) -> TCBrace (f i)
443 | TOCro (i) -> TOCro (f i)
444 | TCCro (i) -> TCCro (f i)
445 | TDot (i) -> TDot (f i)
446 | TComma (i) -> TComma (f i)
447 | TPtrOp (i) -> TPtrOp (f i)
448 | TInc (i) -> TInc (f i)
449 | TDec (i) -> TDec (f i)
450 | TEq (i) -> TEq (f i)
451 | TWhy (i) -> TWhy (f i)
452 | TTilde (i) -> TTilde (f i)
453 | TBang (i) -> TBang (f i)
454 | TEllipsis (i) -> TEllipsis (f i)
455 | TDotDot (i) -> TDotDot (f i)
456 | TPtVirg (i) -> TPtVirg (f i)
457 | TOrLog (i) -> TOrLog (f i)
458 | TAndLog (i) -> TAndLog (f i)
459 | TOr (i) -> TOr (f i)
460 | TXor (i) -> TXor (f i)
461 | TAnd (i) -> TAnd (f i)
462 | TEqEq (i) -> TEqEq (f i)
463 | TNotEq (i) -> TNotEq (f i)
464 | TInf (i) -> TInf (f i)
465 | TSup (i) -> TSup (f i)
466 | TInfEq (i) -> TInfEq (f i)
467 | TSupEq (i) -> TSupEq (f i)
468 | TShl (i) -> TShl (f i)
469 | TShr (i) -> TShr (f i)
470 | TPlus (i) -> TPlus (f i)
471 | TMinus (i) -> TMinus (f i)
472 | TMul (i) -> TMul (f i)
473 | TDiv (i) -> TDiv (f i)
474 | TMod (i) -> TMod (f i)
475 | Tchar (i) -> Tchar (f i)
476 | Tshort (i) -> Tshort (f i)
477 | Tint (i) -> Tint (f i)
478 | Tdouble (i) -> Tdouble (f i)
479 | Tfloat (i) -> Tfloat (f i)
480 | Tlong (i) -> Tlong (f i)
481 | Tunsigned (i) -> Tunsigned (f i)
482 | Tsigned (i) -> Tsigned (f i)
483 | Tvoid (i) -> Tvoid (f i)
1eddfd50
C
484 | Tsize_t (i) -> Tsize_t (f i)
485 | Tssize_t (i) -> Tssize_t (f i)
486 | Tptrdiff_t (i) -> Tptrdiff_t (f i)
ae4735db
C
487 | Tauto (i) -> Tauto (f i)
488 | Tregister (i) -> Tregister (f i)
489 | Textern (i) -> Textern (f i)
490 | Tstatic (i) -> Tstatic (f i)
491 | Tconst (i) -> Tconst (f i)
492 | Tvolatile (i) -> Tvolatile (f i)
493
494 | Trestrict (i) -> Trestrict (f i)
495
496 | Tstruct (i) -> Tstruct (f i)
497 | Tenum (i) -> Tenum (f i)
498 | Ttypedef (i) -> Ttypedef (f i)
499 | Tunion (i) -> Tunion (f i)
500 | Tbreak (i) -> Tbreak (f i)
501 | Telse (i) -> Telse (f i)
502 | Tswitch (i) -> Tswitch (f i)
503 | Tcase (i) -> Tcase (f i)
504 | Tcontinue (i) -> Tcontinue (f i)
505 | Tfor (i) -> Tfor (f i)
506 | Tdo (i) -> Tdo (f i)
507 | Tif (i) -> Tif (f i)
508 | Twhile (i) -> Twhile (f i)
509 | Treturn (i) -> Treturn (f i)
510 | Tgoto (i) -> Tgoto (f i)
511 | Tdefault (i) -> Tdefault (f i)
512 | Tsizeof (i) -> Tsizeof (f i)
513 | Tasm (i) -> Tasm (f i)
514 | Tattribute (i) -> Tattribute (f i)
4dfbc1c2 515 | TattributeNoarg (i) -> TattributeNoarg (f i)
ae4735db
C
516 | Tinline (i) -> Tinline (f i)
517 | Ttypeof (i) -> Ttypeof (f i)
f59c9fb7 518 | Tnew (i) -> Tnew (f i)
4dfbc1c2
C
519 | Tdelete (i) -> Tdelete (f i)
520 | TOParCplusplusInit (i) -> TOParCplusplusInit (f i)
ae4735db
C
521 | EOF (i) -> EOF (f i)
522
34e49164
C
523
524(*****************************************************************************)
525(* Accessors *)
526(*****************************************************************************)
527
528let linecol_of_tok tok =
529 let info = info_of_tok tok in
530 Ast_c.line_of_info info, Ast_c.col_of_info info
531
532let col_of_tok x = snd (linecol_of_tok x)
533let line_of_tok x = fst (linecol_of_tok x)
534let pos_of_tok x = Ast_c.opos_of_info (info_of_tok x)
535let str_of_tok x = Ast_c.str_of_info (info_of_tok x)
536let file_of_tok x = Ast_c.file_of_info (info_of_tok x)
537let pinfo_of_tok x = Ast_c.pinfo_of_info (info_of_tok x)
538
539let is_origin x =
540 match pinfo_of_tok x with Ast_c.OriginTok _ -> true | _ -> false
541let is_expanded x =
542 match pinfo_of_tok x with Ast_c.ExpandedTok _ -> true | _ -> false
543let is_fake x =
544 match pinfo_of_tok x with Ast_c.FakeTok _ -> true | _ -> false
545let is_abstract x =
546 match pinfo_of_tok x with Ast_c.AbstractLineTok _ -> true | _ -> false
91eba41f
C
547
548(*****************************************************************************)
549(* Helpers *)
550(*****************************************************************************)
ae4735db
C
551let is_same_line_or_close line tok =
552 line_of_tok tok =|= line ||
b1b2de81
C
553 line_of_tok tok =|= line - 1 ||
554 line_of_tok tok =|= line - 2
0708f913 555