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 
17 using namespace llvm;
18 using namespace llvm::remarks;
19 
20 Expected<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 
35 Expected<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