RemarkFormat.cpp revision 360784 
1//===- RemarkFormat.cpp --------------------------------------------------===//
2//
3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4// See https://llvm.org/LICENSE.txt for license information.
5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6//
7//===----------------------------------------------------------------------===//
8//
9// Implementation of utilities to handle the different remark formats.
10//
11//===----------------------------------------------------------------------===//
12
13#include "llvm/Remarks/RemarkFormat.h"
14#include "llvm/ADT/StringSwitch.h"
15#include "llvm/Remarks/BitstreamRemarkContainer.h"
16
17using namespace llvm;
18using namespace llvm::remarks;
19
20Expected<Format> llvm::remarks::parseFormat(StringRef FormatStr) {
21  auto Result = StringSwitch<Format>(FormatStr)
22                    .Cases("", "yaml", Format::YAML)
23                    .Case("yaml-strtab", Format::YAMLStrTab)
24                    .Case("bitstream", Format::Bitstream)
25                    .Default(Format::Unknown);
26
27  if (Result == Format::Unknown)
28    return createStringError(std::make_error_code(std::errc::invalid_argument),
29                             "Unknown remark format: '%s'",
30                             FormatStr.data());
31
32  return Result;
33}
34
35Expected<Format> llvm::remarks::magicToFormat(StringRef MagicStr) {
36  auto Result =
37      StringSwitch<Format>(MagicStr)
38          .StartsWith("--- ", Format::YAML) // This is only an assumption.
39          .StartsWith(remarks::Magic, Format::YAMLStrTab)
40          .StartsWith(remarks::ContainerMagic, Format::Bitstream)
41          .Default(Format::Unknown);
42
43  if (Result == Format::Unknown)
44    return createStringError(std::make_error_code(std::errc::invalid_argument),
45                             "Unknown remark magic: '%s'", MagicStr.data());
46  return Result;
47}
48