diff --git a/include/llvm/ADT/StringSwitch.h b/include/llvm/ADT/StringSwitch.h index 044cccd2948..d2a0521f73b 100644 --- a/include/llvm/ADT/StringSwitch.h +++ b/include/llvm/ADT/StringSwitch.h @@ -68,6 +68,7 @@ public: ~StringSwitch() = default; + // Case-sensitive case matchers template LLVM_ATTRIBUTE_ALWAYS_INLINE StringSwitch& Case(const char (&S)[N], const T& Value) { @@ -182,8 +183,62 @@ public: return Case(S0, Value).Cases(S1, S2, S3, S4, S5, S6, S7, S8, S9, Value); } + // Case-insensitive case matchers. + template + LLVM_ATTRIBUTE_ALWAYS_INLINE StringSwitch &CaseLower(const char (&S)[N], + const T &Value) { + if (!Result && Str.equals_lower(StringRef(S, N - 1))) + Result = &Value; + + return *this; + } + + template + LLVM_ATTRIBUTE_ALWAYS_INLINE StringSwitch &EndsWithLower(const char (&S)[N], + const T &Value) { + if (!Result && Str.endswith_lower(StringRef(S, N - 1))) + Result = &Value; + + return *this; + } + + template + LLVM_ATTRIBUTE_ALWAYS_INLINE StringSwitch &StartsWithLower(const char (&S)[N], + const T &Value) { + if (!Result && Str.startswith_lower(StringRef(S, N - 1))) + Result = &Value; + + return *this; + } + template + LLVM_ATTRIBUTE_ALWAYS_INLINE StringSwitch & + CasesLower(const char (&S0)[N0], const char (&S1)[N1], const T &Value) { + return CaseLower(S0, Value).CaseLower(S1, Value); + } + + template + LLVM_ATTRIBUTE_ALWAYS_INLINE StringSwitch & + CasesLower(const char (&S0)[N0], const char (&S1)[N1], const char (&S2)[N2], + const T &Value) { + return CaseLower(S0, Value).CasesLower(S1, S2, Value); + } + + template + LLVM_ATTRIBUTE_ALWAYS_INLINE StringSwitch & + CasesLower(const char (&S0)[N0], const char (&S1)[N1], const char (&S2)[N2], + const char (&S3)[N3], const T &Value) { + return CaseLower(S0, Value).CasesLower(S1, S2, S3, Value); + } + + template + LLVM_ATTRIBUTE_ALWAYS_INLINE StringSwitch & + CasesLower(const char (&S0)[N0], const char (&S1)[N1], const char (&S2)[N2], + const char (&S3)[N3], const char (&S4)[N4], const T &Value) { + return CaseLower(S0, Value).CasesLower(S1, S2, S3, S4, Value); + } + LLVM_ATTRIBUTE_ALWAYS_INLINE - R Default(const T& Value) const { + R Default(const T &Value) const { if (Result) return *Result; return Value; diff --git a/unittests/ADT/StringSwitchTest.cpp b/unittests/ADT/StringSwitchTest.cpp index 9f69bc3ef11..1a51629ca57 100644 --- a/unittests/ADT/StringSwitchTest.cpp +++ b/unittests/ADT/StringSwitchTest.cpp @@ -41,6 +41,36 @@ TEST(StringSwitchTest, Case) { EXPECT_EQ(-1, Translate("Test")); } +TEST(StringSwitchTest, CaseLower) { + auto Translate = [](StringRef S) { + return llvm::StringSwitch(S) + .Case("0", 0) + .Case("1", 1) + .Case("2", 2) + .Case("3", 3) + .Case("4", 4) + .Case("5", 5) + .Case("6", 6) + .Case("7", 7) + .Case("8", 8) + .Case("9", 9) + .CaseLower("A", 10) + .CaseLower("B", 11) + .CaseLower("C", 12) + .CaseLower("D", 13) + .CaseLower("E", 14) + .CaseLower("F", 15) + .Default(-1); + }; + EXPECT_EQ(1, Translate("1")); + EXPECT_EQ(2, Translate("2")); + EXPECT_EQ(11, Translate("B")); + EXPECT_EQ(11, Translate("b")); + + EXPECT_EQ(-1, Translate("")); + EXPECT_EQ(-1, Translate("Test")); +} + TEST(StringSwitchTest, StartsWith) { auto Translate = [](StringRef S) { return llvm::StringSwitch>(S) @@ -60,6 +90,29 @@ TEST(StringSwitchTest, StartsWith) { EXPECT_EQ(0, Translate("ADDER")(10, 5)); } +TEST(StringSwitchTest, StartsWithLower) { + auto Translate = [](StringRef S) { + return llvm::StringSwitch>(S) + .StartsWithLower("add", [](int X, int Y) { return X + Y; }) + .StartsWithLower("sub", [](int X, int Y) { return X - Y; }) + .StartsWithLower("mul", [](int X, int Y) { return X * Y; }) + .StartsWithLower("div", [](int X, int Y) { return X / Y; }) + .Default([](int X, int Y) { return 0; }); + }; + + EXPECT_EQ(15, Translate("adder")(10, 5)); + EXPECT_EQ(5, Translate("subtracter")(10, 5)); + EXPECT_EQ(50, Translate("multiplier")(10, 5)); + EXPECT_EQ(2, Translate("divider")(10, 5)); + + EXPECT_EQ(15, Translate("AdDeR")(10, 5)); + EXPECT_EQ(5, Translate("SuBtRaCtEr")(10, 5)); + EXPECT_EQ(50, Translate("MuLtIpLiEr")(10, 5)); + EXPECT_EQ(2, Translate("DiViDeR")(10, 5)); + + EXPECT_EQ(0, Translate("nothing")(10, 5)); +} + TEST(StringSwitchTest, EndsWith) { enum class Suffix { Possible, PastTense, Process, InProgressAction, Unknown }; @@ -80,6 +133,26 @@ TEST(StringSwitchTest, EndsWith) { EXPECT_EQ(Suffix::Unknown, Translate("OPTIMIZABLE")); } +TEST(StringSwitchTest, EndsWithLower) { + enum class Suffix { Possible, PastTense, Process, InProgressAction, Unknown }; + + auto Translate = [](StringRef S) { + return llvm::StringSwitch(S) + .EndsWithLower("able", Suffix::Possible) + .EndsWithLower("ed", Suffix::PastTense) + .EndsWithLower("ation", Suffix::Process) + .EndsWithLower("ing", Suffix::InProgressAction) + .Default(Suffix::Unknown); + }; + + EXPECT_EQ(Suffix::Possible, Translate("optimizable")); + EXPECT_EQ(Suffix::Possible, Translate("OPTIMIZABLE")); + EXPECT_EQ(Suffix::PastTense, Translate("optimized")); + EXPECT_EQ(Suffix::Process, Translate("optimization")); + EXPECT_EQ(Suffix::InProgressAction, Translate("optimizing")); + EXPECT_EQ(Suffix::Unknown, Translate("optimizer")); +} + TEST(StringSwitchTest, Cases) { enum class OSType { Windows, Linux, Unknown }; @@ -105,3 +178,29 @@ TEST(StringSwitchTest, Cases) { EXPECT_EQ(OSType::Unknown, Translate("Windows")); EXPECT_EQ(OSType::Unknown, Translate("")); } + +TEST(StringSwitchTest, CasesLower) { + enum class OSType { Windows, Linux, Unknown }; + + auto Translate = [](StringRef S) { + return llvm::StringSwitch(S) + .CasesLower("wind\0ws", "win32", "winnt", OSType::Windows) + .CasesLower("linux", "unix", "*nix", "posix", OSType::Linux) + .Default(OSType::Unknown); + }; + + EXPECT_EQ(OSType::Windows, Translate(llvm::StringRef("WIND\0WS", 7))); + EXPECT_EQ(OSType::Windows, Translate("WIN32")); + EXPECT_EQ(OSType::Windows, Translate("WINNT")); + + EXPECT_EQ(OSType::Linux, Translate("LINUX")); + EXPECT_EQ(OSType::Linux, Translate("UNIX")); + EXPECT_EQ(OSType::Linux, Translate("*NIX")); + EXPECT_EQ(OSType::Linux, Translate("POSIX")); + + EXPECT_EQ(OSType::Windows, Translate(llvm::StringRef("wind\0ws", 7))); + EXPECT_EQ(OSType::Linux, Translate("linux")); + + EXPECT_EQ(OSType::Unknown, Translate("wind")); + EXPECT_EQ(OSType::Unknown, Translate("")); +}