diff --git a/ql/src/printAst.ql b/ql/src/printAst.ql index adc1dd1fb48..e5b36e73fe3 100644 --- a/ql/src/printAst.ql +++ b/ql/src/printAst.ql @@ -20,7 +20,9 @@ external string selectedSourceFile(); * Hook to customize the functions printed by this query. */ class Cfg extends PrintAstConfiguration { - override predicate shouldPrintFunction(FuncDef func) { - func.getFile() = getEncodedFile(selectedSourceFile()) - } + override predicate shouldPrintFunction(FuncDef func) { shouldPrintFile(func.getFile()) } + + override predicate shouldPrintFile(File file) { file = getEncodedFile(selectedSourceFile()) } + + override predicate shouldPrintComments(File file) { none() } }