diff options
Diffstat (limited to 'clang-include-fixer/find-all-symbols/HeaderMapCollector.h')
-rw-r--r-- | clang-include-fixer/find-all-symbols/HeaderMapCollector.h | 56 |
1 files changed, 56 insertions, 0 deletions
diff --git a/clang-include-fixer/find-all-symbols/HeaderMapCollector.h b/clang-include-fixer/find-all-symbols/HeaderMapCollector.h new file mode 100644 index 00000000..21358275 --- /dev/null +++ b/clang-include-fixer/find-all-symbols/HeaderMapCollector.h @@ -0,0 +1,56 @@ +//===-- HeaderMapCoolector.h - find all symbols------------------*- C++ -*-===// +// +// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. +// See https://llvm.org/LICENSE.txt for license information. +// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception +// +//===----------------------------------------------------------------------===// + +#ifndef LLVM_CLANG_TOOLS_EXTRA_FIND_ALL_SYMBOLS_HEADER_MAP_COLLECTOR_H +#define LLVM_CLANG_TOOLS_EXTRA_FIND_ALL_SYMBOLS_HEADER_MAP_COLLECTOR_H + +#include "llvm/ADT/StringMap.h" +#include "llvm/Support/Regex.h" +#include <string> +#include <vector> + +namespace clang { +namespace find_all_symbols { + +/// \brief HeaderMappCollector collects all remapping header files. This maps +/// complete header names or header name regex patterns to header names. +class HeaderMapCollector { +public: + typedef llvm::StringMap<std::string> HeaderMap; + typedef std::vector<std::pair<const char *, const char *>> RegexHeaderMap; + + HeaderMapCollector() = default; + explicit HeaderMapCollector(const RegexHeaderMap *RegexHeaderMappingTable); + + void addHeaderMapping(llvm::StringRef OrignalHeaderPath, + llvm::StringRef MappingHeaderPath) { + HeaderMappingTable[OrignalHeaderPath] = MappingHeaderPath; + }; + + /// Check if there is a mapping from \p Header or a regex pattern that matches + /// it to another header name. + /// \param Header A header name. + /// \return \p Header itself if there is no mapping for it; otherwise, return + /// a mapped header name. + llvm::StringRef getMappedHeader(llvm::StringRef Header) const; + +private: + /// A string-to-string map saving the mapping relationship. + HeaderMap HeaderMappingTable; + + // A map from header patterns to header names. + // The header names are not owned. This is only threadsafe because the regexes + // never fail. + mutable std::vector<std::pair<llvm::Regex, const char *>> + RegexHeaderMappingTable; +}; + +} // namespace find_all_symbols +} // namespace clang + +#endif // LLVM_CLANG_TOOLS_EXTRA_FIND_ALL_SYMBOLS_HEADER_MAP_COLLECTOR_H |