From 52ed867801b3ddb5569b7f823e63555e9c322c99 Mon Sep 17 00:00:00 2001 From: "Michael J. Spencer" Date: Wed, 1 Dec 2010 00:52:55 +0000 Subject: Support/PathV2: Add replace_extension implementation. git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@120514 91177308-0d34-0410-b5e6-96231b3b80d8 --- lib/Support/PathV2.cpp | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) (limited to 'lib/Support/PathV2.cpp') diff --git a/lib/Support/PathV2.cpp b/lib/Support/PathV2.cpp index df401bd46b..7e49d8d446 100644 --- a/lib/Support/PathV2.cpp +++ b/lib/Support/PathV2.cpp @@ -511,6 +511,26 @@ error_code remove_filename(SmallVectorImpl &path) { return make_error_code(errc::success); } +error_code replace_extension(SmallVectorImpl &path, + const Twine &extension) { + StringRef p(path.begin(), path.size()); + SmallString<32> ext_storage; + StringRef ext = extension.toStringRef(ext_storage); + + // Erase existing extension. + size_t pos = p.find_last_of('.'); + if (pos != StringRef::npos && pos >= filename_pos(p)) + path.set_size(pos); + + // Append '.' if needed. + if (ext.size() > 0 && ext[0] != '.') + path.push_back('.'); + + // Append extension. + path.append(ext.begin(), ext.end()); + return make_error_code(errc::success); +} + } } } -- cgit v1.2.3-70-g09d2