Support/PathV2: Add replace_extension implementation.

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@120514 91177308-0d34-0410-b5e6-96231b3b80d8
This commit is contained in:
Michael J. Spencer 2010-12-01 00:52:55 +00:00
parent dbfb56bebd
commit 52ed867801
2 changed files with 24 additions and 0 deletions

View File

@ -511,6 +511,26 @@ error_code remove_filename(SmallVectorImpl<char> &path) {
return make_error_code(errc::success); return make_error_code(errc::success);
} }
error_code replace_extension(SmallVectorImpl<char> &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);
}
} }
} }
} }

View File

@ -103,6 +103,10 @@ TEST(Support, Path) {
if (error_code ec = sys::path::remove_filename(temp_store)) if (error_code ec = sys::path::remove_filename(temp_store))
ASSERT_FALSE(ec.message().c_str()); ASSERT_FALSE(ec.message().c_str());
outs() << " remove_filename: " << temp_store << '\n'; outs() << " remove_filename: " << temp_store << '\n';
temp_store = *i;
if (error_code ec = sys::path::replace_extension(temp_store, "ext"))
ASSERT_FALSE(ec.message().c_str());
outs() << " replace_extension: " << temp_store << '\n';
outs().flush(); outs().flush();
} }