[UpdateTestChecks][NFC] Share the code to get CHECK prefix between all scripts

[MaskRay]

Looks great!

llvm/utils/UpdateTestChecks/common.py

 def check_prefix(prefix):
   if not PREFIX_RE.match(prefix):
     hint = ""
     if ',' in prefix:
       hint = " Did you mean '--check-prefixes=" + prefix + "'?"
     warn(("Supplied prefix '%s' is invalid. Prefix must contain only alphanumeric characters, hyphens and underscores." + hint) %
          (prefix))
 
+def get_check_prefixes(filecheck_cmd):
+  check_prefixes = [item for m in CHECK_PREFIX_RE.finditer(filecheck_cmd)
+                           for item in m.group(1).split(',')]
+  if not check_prefixes:
+    check_prefixes = ['CHECK']
+  return check_prefixes
 
 def verify_filecheck_prefixes(fc_cmd):
   fc_cmd_parts = fc_cmd.split()
   for part in fc_cmd_parts:
     if "check-prefix=" in part:
       prefix = part.split('=', 1)[1]
       check_prefix(prefix)
     elif "check-prefixes=" in part:

llvm/utils/update_analyze_test_checks.py

for l in ti.run_lines:
         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 = [item for m in common.CHECK_PREFIX_RE.finditer(filecheck_cmd)
-                               for item in m.group(1).split(',')]
-      if not check_prefixes:
-        check_prefixes = ['CHECK']
+      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))
 
     builder = common.FunctionTestBuilder(
       run_list = prefix_list,
       flags = type('', (object,), {

llvm/utils/update_cc_test_checks.py

for l in ti.run_lines:
       filecheck_cmd = commands[-1]
       common.verify_filecheck_prefixes(filecheck_cmd)
       if not filecheck_cmd.startswith('FileCheck '):
         # Execute non-filechecked clang runline.
         exe = [ti.args.clang] + clang_args
         run_list.append((None, exe, None, None))
         continue
 
-      check_prefixes = [item for m in common.CHECK_PREFIX_RE.finditer(filecheck_cmd)
-                               for item in m.group(1).split(',')]
-      if not check_prefixes:
-        check_prefixes = ['CHECK']
+      check_prefixes = common.get_check_prefixes(filecheck_cmd)
       run_list.append((check_prefixes, clang_args, commands[1:-1], triple_in_cmd))
 
     # Execute clang, generate LLVM IR, and extract functions.
 
     # Store only filechecked runlines.
     filecheck_run_list = [i for i in run_list if i[0]]
     builder = common.FunctionTestBuilder(
       run_list=filecheck_run_list,

llvm/utils/update_llc_test_checks.py

for l in ti.run_lines:
       if not filecheck_cmd.startswith('FileCheck '):
         common.warn('Skipping non-FileChecked RUN line: ' + l)
         continue
 
       llc_cmd_args = llc_cmd[len(llc_tool):].strip()
       llc_cmd_args = llc_cmd_args.replace('< %s', '').replace('%s', '').strip()
       if ti.path.endswith('.mir'):
         llc_cmd_args += ' -x mir'
-      check_prefixes = [item for m in common.CHECK_PREFIX_RE.finditer(filecheck_cmd)
-                               for item in m.group(1).split(',')]
-      if not check_prefixes:
-        check_prefixes = ['CHECK']
+      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.
       run_list.append((check_prefixes, llc_tool, llc_cmd_args, preprocess_cmd,
                        triple_in_cmd, march_in_cmd))
 
     if ti.path.endswith('.mir'):
       check_indent = '  '

llvm/utils/update_mca_test_checks.py

for run_line in run_lines:
 
     if not filecheck_cmd.startswith('FileCheck '):
       _warn('skipping non-FileCheck RUN line: {}'.format(run_line))
       continue
 
     tool_cmd_args = tool_cmd[len(tool_basename):].strip()
     tool_cmd_args = tool_cmd_args.replace('< %s', '').replace('%s', '').strip()
 
-    check_prefixes = [item
-                      for m in common.CHECK_PREFIX_RE.finditer(filecheck_cmd)
-                      for item in m.group(1).split(',')]
-    if not check_prefixes:
-      check_prefixes = ['CHECK']
+    check_prefixes = common.get_check_prefixes(filecheck_cmd)
 
     run_infos.append((check_prefixes, tool_cmd_args))
 
   return run_infos
 
 
 def _break_down_block(block_info, common_prefix):
   """ Given a block_info, see if we can analyze it further to let us break it

llvm/utils/update_mir_test_checks.py

for l in run_lines:
             triple = m.group(1)
         # If we find -march but not -mtriple, use that.
         m = common.MARCH_ARG_RE.search(llc_cmd)
         if m and not triple:
             triple = '{}--'.format(m.group(1))
 
         cmd_args = llc_cmd[len('llc'):].strip()
         cmd_args = cmd_args.replace('< %s', '').replace('%s', '').strip()
-
-        check_prefixes = [
-            item
-            for m in common.CHECK_PREFIX_RE.finditer(filecheck_cmd)
-            for item in m.group(1).split(',')]
-        if not check_prefixes:
-            check_prefixes = ['CHECK']
+        check_prefixes = common.get_check_prefixes(filecheck_cmd)
         all_prefixes += check_prefixes
 
         run_list.append(Run(check_prefixes, cmd_args, triple))
 
     # Sort prefixes that are shared between run lines before unshared prefixes.
     # This causes us to prefer printing shared prefixes.
     for run in run_list:
         run.prefixes.sort(key=lambda prefix: -all_prefixes.count(prefix))

llvm/utils/update_test_checks.py

for l in ti.run_lines:
         continue
 
       if not filecheck_cmd.startswith('FileCheck '):
         common.warn('Skipping non-FileChecked RUN line: ' + l)
         continue
 
       tool_cmd_args = tool_cmd[len(tool_basename):].strip()
       tool_cmd_args = tool_cmd_args.replace('< %s', '').replace('%s', '').strip()
-
-      check_prefixes = [item for m in
-                        common.CHECK_PREFIX_RE.finditer(filecheck_cmd)
-                        for item in m.group(1).split(',')]
-      if not check_prefixes:
-        check_prefixes = ['CHECK']
+      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, preprocess_cmd))
 
     global_vars_seen_dict = {}
     builder = common.FunctionTestBuilder(
       run_list=prefix_list,