catch2/include/internal/catch_stringref.cpp

85 lines
2.4 KiB
C++
Raw Normal View History

2017-06-29 12:18:14 +02:00
/*
* Copyright 2016 Two Blue Cubes Ltd. All rights reserved.
*
* Distributed under the Boost Software License, Version 1.0. (See accompanying
* file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
*/
#if defined(__clang__)
# pragma clang diagnostic push
# pragma clang diagnostic ignored "-Wexit-time-destructors"
#endif
2017-06-29 12:18:14 +02:00
#include "catch_stringref.h"
#include <ostream>
#include <cstring>
#include <cstdint>
2017-06-29 12:18:14 +02:00
namespace Catch {
StringRef::StringRef( char const* rawChars ) noexcept
: StringRef( rawChars, static_cast<StringRef::size_type>(std::strlen(rawChars) ) )
{}
2017-06-29 12:18:14 +02:00
void StringRef::swap( StringRef& other ) noexcept {
std::swap( m_start, other.m_start );
std::swap( m_size, other.m_size );
std::swap( m_data, other.m_data );
}
2017-06-29 12:18:14 +02:00
auto StringRef::c_str() const -> char const* {
if( !isSubstring() )
return m_start;
const_cast<StringRef *>( this )->takeOwnership();
return m_data;
2017-06-29 12:18:14 +02:00
}
auto StringRef::currentData() const noexcept -> char const* {
2017-06-29 12:18:14 +02:00
return m_start;
}
auto StringRef::isOwned() const noexcept -> bool {
return m_data != nullptr;
}
auto StringRef::isSubstring() const noexcept -> bool {
return m_start[m_size] != '\0';
}
2017-06-29 12:18:14 +02:00
void StringRef::takeOwnership() {
if( !isOwned() ) {
2017-08-14 10:14:49 +02:00
m_data = new char[m_size+1];
2017-08-16 00:47:52 +02:00
memcpy( m_data, m_start, m_size );
2017-08-14 10:14:49 +02:00
m_data[m_size] = '\0';
2017-08-14 10:04:14 +02:00
}
2017-06-29 12:18:14 +02:00
}
auto StringRef::substr( size_type start, size_type size ) const noexcept -> StringRef {
if( start < m_size )
return StringRef( m_start+start, size );
else
return StringRef();
}
auto StringRef::operator == ( StringRef const& other ) const noexcept -> bool {
return
size() == other.size() &&
(std::strncmp( m_start, other.m_start, size() ) == 0);
}
auto StringRef::operator != ( StringRef const& other ) const noexcept -> bool {
return !operator==( other );
}
2017-08-14 09:50:44 +02:00
auto operator << ( std::ostream& os, StringRef const& str ) -> std::ostream& {
return os.write(str.currentData(), str.size());
2017-06-29 12:18:14 +02:00
}
auto operator+=( std::string& lhs, StringRef const& rhs ) -> std::string& {
lhs.append(rhs.currentData(), rhs.size());
return lhs;
}
2017-06-29 12:18:14 +02:00
} // namespace Catch
#if defined(__clang__)
# pragma clang diagnostic pop
#endif