blob: 09d2092d43b07ca9a34a6e1f9b77d44d97fbd5ed [file] [log] [blame]
Andrew Scull5e1ddfa2018-08-14 10:06:54 +01001//===- StringMatcher.h - Generate a matcher for input strings ---*- C++ -*-===//
2//
3// The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9//
10// This file implements the StringMatcher class.
11//
12//===----------------------------------------------------------------------===//
13
14#ifndef LLVM_TABLEGEN_STRINGMATCHER_H
15#define LLVM_TABLEGEN_STRINGMATCHER_H
16
17#include "llvm/ADT/StringRef.h"
18#include <string>
19#include <utility>
20#include <vector>
21
22namespace llvm {
23
24class raw_ostream;
25
26/// StringMatcher - Given a list of strings and code to execute when they match,
27/// output a simple switch tree to classify the input string.
28///
29/// If a match is found, the code in Vals[i].second is executed; control must
30/// not exit this code fragment. If nothing matches, execution falls through.
31///
32class StringMatcher {
33public:
34 using StringPair = std::pair<std::string, std::string>;
35
36private:
37 StringRef StrVariableName;
38 const std::vector<StringPair> &Matches;
39 raw_ostream &OS;
40
41public:
42 StringMatcher(StringRef strVariableName,
43 const std::vector<StringPair> &matches, raw_ostream &os)
44 : StrVariableName(strVariableName), Matches(matches), OS(os) {}
45
46 void Emit(unsigned Indent = 0, bool IgnoreDuplicates = false) const;
47
48private:
49 bool EmitStringMatcherForChar(const std::vector<const StringPair *> &Matches,
50 unsigned CharNo, unsigned IndentCount,
51 bool IgnoreDuplicates) const;
52};
53
54} // end namespace llvm
55
56#endif // LLVM_TABLEGEN_STRINGMATCHER_H