You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
46 lines
1.6 KiB
46 lines
1.6 KiB
6 years ago
|
(*
|
||
|
|
* Copyright (c) Facebook, Inc. and its affiliates.
|
||
|
|
*
|
||
|
|
* This source code is licensed under the MIT license found in the
|
||
|
|
* LICENSE file in the root directory of this source tree.
|
||
|
|
*)
|
||
|
|
open! IStd
|
||
|
|
module L = Logging
|
||
|
|
|
||
|
|
let process_ast ast tenv default_source_file f =
|
||
|
|
let open Clang_ast_t in
|
||
|
|
let mk_key decl = CType_decl.CProcname.from_decl ~tenv decl in
|
||
|
|
let rec extract_location decl =
|
||
|
|
match decl with
|
||
|
|
| ObjCMethodDecl (di, _, _)
|
||
|
|
| CXXConversionDecl (di, _, _, _, _)
|
||
|
|
| FunctionDecl (di, _, _, _)
|
||
|
|
| CXXMethodDecl (di, _, _, _, _)
|
||
|
|
| CXXConstructorDecl (di, _, _, _, _)
|
||
|
|
| CXXDestructorDecl (di, _, _, _, _) ->
|
||
|
|
let range = CLocation.location_of_decl_info default_source_file di in
|
||
|
|
f (mk_key decl) range
|
||
|
|
| _ -> (
|
||
|
|
match Clang_ast_proj.get_decl_context_tuple decl with
|
||
|
|
| Some (decls, _) ->
|
||
|
|
List.iter decls ~f:extract_location
|
||
|
|
| None ->
|
||
|
|
L.(debug Capture Verbose)
|
||
|
|
"@\n Found %s.., skipping"
|
||
|
|
(Clang_ast_proj.get_decl_kind_string decl) )
|
||
|
|
in
|
||
|
|
match ast with
|
||
|
|
| TranslationUnitDecl (_, decl_list, _, _) ->
|
||
|
|
CFrontend_config.global_translation_unit_decls := decl_list ;
|
||
|
|
L.(debug Capture Verbose) "@\n Processing AST...@\n" ;
|
||
|
|
List.iter decl_list ~f:(fun d ->
|
||
|
|
let info = Clang_ast_proj.get_decl_tuple d in
|
||
|
|
let source_range = info.di_source_range in
|
||
|
|
if
|
||
|
|
CLocation.should_translate_lib default_source_file source_range `DeclTraversal
|
||
|
|
~translate_when_used:true
|
||
|
|
then extract_location d ) ;
|
||
|
|
L.(debug Capture Verbose) "@\n Finished processing AST.@\n"
|
||
|
|
| _ ->
|
||
|
|
assert false
|