@@ -363,15 +363,26 @@ def first_pass(self, path: Path, f: io.TextIOWrapper, graph:Graph) -> None:
363
363
# Process function definitions
364
364
query = C_LANGUAGE .query ("(function_definition) @function" )
365
365
captures = query .captures (tree .root_node )
366
- for capture in captures :
367
- node = capture [0 ]
366
+ # captures: {'function':
367
+ # [<Node type=function_definition, start_point=(0, 0), end_point=(7, 1)>,
368
+ # <Node type=function_definition, start_point=(15, 0), end_point=(18, 1)>
369
+ # ]
370
+ # }
371
+
372
+ functions = captures ['function' ]
373
+ for node in functions :
368
374
self .process_function_definition (file , node , path , graph , source_code )
369
375
370
376
# Process struct definitions
371
377
query = C_LANGUAGE .query ("(struct_specifier) @struct" )
372
378
captures = query .captures (tree .root_node )
373
- for capture in captures :
374
- node = capture [0 ]
379
+ structs = captures ['struct' ]
380
+ # captures: {'struct':
381
+ # [
382
+ # <Node type=struct_specifier, start_point=(9, 0), end_point=(13, 1)>
383
+ # ]
384
+ # }
385
+ for node in structs :
375
386
self .process_struct_specifier (file , node , path , graph )
376
387
377
388
def second_pass (self , path : Path , f : io .TextIOWrapper , graph : Graph ) -> None :
0 commit comments