catch2/include/internal/catch_stringref.cpp

147 lines
4.1 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)
*/
#include "catch_stringref.h"
#include "catch_stringbuilder.h"
#include "catch_string.h"
#include "catch_stringdata.h"
#include <cstring>
#include <ostream>
#include <cassert>
2017-06-29 12:18:14 +02:00
namespace Catch {
auto getEmptyStringRef() -> StringRef {
static StringRef s_emptyStringRef("");
return s_emptyStringRef;
}
2017-06-29 12:18:14 +02:00
2017-06-29 12:47:09 +02:00
StringRef::StringRef() noexcept
: StringRef( getEmptyStringRef() )
2017-06-29 12:18:14 +02:00
{}
2017-06-29 12:47:09 +02:00
StringRef::StringRef( StringRef const& other ) noexcept
2017-06-29 12:18:14 +02:00
: m_start( other.m_start ),
m_size( other.m_size ),
m_data( other.m_data )
{
if( m_data )
m_data->addRef();
}
StringRef::StringRef( StringRef&& other ) noexcept
: m_start( other.m_start ),
m_size( other.m_size ),
m_data( other.m_data )
{
other.m_data = nullptr;
}
2017-06-29 12:47:09 +02:00
StringRef::StringRef( char const* rawChars ) noexcept
2017-06-29 12:18:14 +02:00
: m_start( rawChars ),
m_size( static_cast<size_type>( std::strlen( rawChars ) ) )
{
assert( rawChars != nullptr );
}
2017-06-29 12:18:14 +02:00
2017-06-29 12:47:09 +02:00
StringRef::StringRef( char const* rawChars, size_type size ) noexcept
2017-06-29 12:18:14 +02:00
: m_start( rawChars ),
m_size( size )
{
size_type rawSize = rawChars == nullptr ? 0 : static_cast<size_type>( std::strlen( rawChars ) );
2017-06-29 12:18:14 +02:00
if( rawSize < size )
m_size = rawSize;
2017-06-29 12:18:14 +02:00
}
2017-06-29 12:47:09 +02:00
StringRef::StringRef( String const& other ) noexcept
2017-06-29 12:18:14 +02:00
: m_start( other.c_str() ),
m_size( other.size() ),
m_data( nullptr )
{}
StringRef::StringRef( String&& str ) noexcept
: m_start( str.c_str() ),
m_size( str.size() ),
m_data( str.m_data )
{
str.m_data = StringData::getEmpty();
}
StringRef::~StringRef() noexcept {
if( isOwned() )
m_data->release();
}
2017-06-29 12:47:09 +02:00
auto StringRef::operator = ( StringRef other ) noexcept -> StringRef& {
2017-06-29 12:18:14 +02:00
swap( other );
return *this;
}
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 );
}
auto StringRef::c_str() const -> char const* {
if( isSubstring() )
const_cast<StringRef*>( this )->takeOwnership();
return m_start;
}
auto StringRef::data() const noexcept -> char const* {
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';
}
void StringRef::takeOwnership() {
if( !isOwned() ) {
StringRef temp = String( *this );
swap( temp );
}
}
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 );
}
auto operator + ( StringRef const& lhs, StringRef const& rhs ) -> String {
StringBuilder buf;
buf.reserve( lhs.size() + rhs.size() );
buf.append( lhs );
buf.append( rhs );
return String( std::move( buf ) );
}
auto operator + ( StringRef const& lhs, const char* rhs ) -> String {
return lhs + StringRef( rhs );
}
auto operator + ( char const* lhs, StringRef const& rhs ) -> String {
return StringRef( lhs ) + rhs;
}
std::ostream& operator << ( std::ostream& os, StringRef const& str ) {
return os << str.c_str();
}
} // namespace Catch