[UpdateTestChecks] Add preprocess_cmd in update_analyze_test_checks.py

[luke]

Summary

Similar to how update_llc_test_checks.py and update_test_checks.py, this
allows a sed command or similar to be used to preprocess an analysis
test.
This code was copied straight from update_llc_test_checks, and could
probably be shared.

[luke]

Rebase

[nikic]

Could you please add a test?

llvm/utils/update_analyze_test_checks.py

):
                 break
 
         prefix_list = []
         for l in ti.run_lines:
             if "|" not in l:
                 common.warn("Skipping unparsable RUN line: " + l)
                 continue
 
-            (tool_cmd, filecheck_cmd) = tuple([cmd.strip() for cmd in l.split("|", 1)])
+            commands = [cmd.strip() for cmd in l.split("|")]
+            assert len(commands) >= 2
+            preprocess_cmd = None
+            if len(commands) > 2:
+                preprocess_cmd = " | ".join(commands[:-2])
+            tool_cmd = commands[-2]
+            filecheck_cmd = commands[-1]
+
             common.verify_filecheck_prefixes(filecheck_cmd)
 
             if not tool_cmd.startswith(opt_basename + " "):
                 common.warn("WSkipping non-%s RUN line: %s" % (opt_basename, l))
                 continue
 
             if not filecheck_cmd.startswith("FileCheck "):
                 common.warn("Skipping non-FileChecked RUN line: " + l)
                 continue
 
             tool_cmd_args = tool_cmd[len(opt_basename) :].strip()
             tool_cmd_args = tool_cmd_args.replace("< %s", "").replace("%s", "").strip()
             check_prefixes = common.get_check_prefixes(filecheck_cmd)
 
             # FIXME: We should use multiple check prefixes to common check lines. For
             # now, we just ignore all but the last.
-            prefix_list.append((check_prefixes, tool_cmd_args))
+            prefix_list.append((check_prefixes, tool_cmd_args, preprocess_cmd))
 
         builder = common.FunctionTestBuilder(
             run_list=prefix_list,
             flags=type(
                 "",
                 (object,),
                 {
                     "verbose": ti.args.verbose,
                     "filters": ti.args.filters,
                     "function_signature": False,
                     "check_attributes": False,
                     "replace_value_regex": [],
                 },
             ),
             scrubber_args=[],
             path=ti.path,
         )
 
-        for prefixes, opt_args in prefix_list:
+        for prefixes, opt_args, preprocess_cmd in prefix_list:
             common.debug("Extracted opt cmd:", opt_basename, opt_args, file=sys.stderr)
             common.debug(
                 "Extracted FileCheck prefixes:", str(prefixes), file=sys.stderr
             )
+            common.debug("Extracted preprocess cmd:", preprocess_cmd)
 
-            raw_tool_outputs = common.invoke_tool(ti.args.opt_binary, opt_args, ti.path)
+            raw_tool_outputs = common.invoke_tool(
+                ti.args.opt_binary, opt_args, ti.path, preprocess_cmd
+            )
 
             if re.search(r"Printing analysis ", raw_tool_outputs) is not None:
                 # Split analysis outputs by "Printing analysis " declarations.
                 for raw_tool_output in re.split(
                     r"Printing analysis ", raw_tool_outputs
                 ):
                     builder.process_run_line(
                         common.ANALYZE_FUNCTION_RE,
):
                 common.warn("Don't know how to deal with this output")
                 continue
 
             builder.processed_prefixes(prefixes)
 
         func_dict = builder.finish_and_get_func_dict()
         is_in_function = False
         is_in_function_start = False
-        prefix_set = set([prefix for prefixes, _ in prefix_list for prefix in prefixes])
+        prefix_set = set(
+            [prefix for prefixes, _, _ in prefix_list for prefix in prefixes]
+        )
         common.debug("Rewriting FileCheck prefixes:", str(prefix_set), file=sys.stderr)
         output_lines = []
 
         generated_prefixes = []
         for input_info in ti.iterlines(output_lines):
             input_line = input_info.line
             args = input_info.args
             if is_in_function_start: