mirror of
https://github.com/catchorg/Catch2.git
synced 2024-11-25 14:56:10 +01:00
05cd05743a
This allows reducing the amount of friends needed for its interface and some extra tricks later. The bad part is that the pointer can become invalidated via calls to other StringRef's public methods, but c'est la vie.
128 lines
3.8 KiB
C++
128 lines
3.8 KiB
C++
/*
|
|
* 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
|
|
|
|
#include "catch_stringref.h"
|
|
|
|
#include <ostream>
|
|
#include <cstring>
|
|
#include <cstdint>
|
|
|
|
namespace {
|
|
const uint32_t byte_2_lead = 0xC0;
|
|
const uint32_t byte_3_lead = 0xE0;
|
|
const uint32_t byte_4_lead = 0xF0;
|
|
}
|
|
|
|
namespace Catch {
|
|
StringRef::StringRef( char const* rawChars ) noexcept
|
|
: StringRef( rawChars, static_cast<StringRef::size_type>(std::strlen(rawChars) ) )
|
|
{}
|
|
|
|
StringRef::operator std::string() const {
|
|
return std::string( m_start, m_size );
|
|
}
|
|
|
|
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::currentData() 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() ) {
|
|
m_data = new char[m_size+1];
|
|
memcpy( m_data, m_start, m_size );
|
|
m_data[m_size] = '\0';
|
|
m_start = m_data;
|
|
}
|
|
}
|
|
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 StringRef::operator[](size_type index) const noexcept -> char {
|
|
return m_start[index];
|
|
}
|
|
|
|
auto StringRef::numberOfCharacters() const noexcept -> size_type {
|
|
size_type noChars = m_size;
|
|
// Make adjustments for uft encodings
|
|
for( size_type i=0; i < m_size; ++i ) {
|
|
char c = m_start[i];
|
|
if( ( c & byte_2_lead ) == byte_2_lead ) {
|
|
noChars--;
|
|
if (( c & byte_3_lead ) == byte_3_lead )
|
|
noChars--;
|
|
if( ( c & byte_4_lead ) == byte_4_lead )
|
|
noChars--;
|
|
}
|
|
}
|
|
return noChars;
|
|
}
|
|
|
|
auto operator + ( StringRef const& lhs, StringRef const& rhs ) -> std::string {
|
|
std::string str;
|
|
str.reserve( lhs.size() + rhs.size() );
|
|
str += lhs;
|
|
str += rhs;
|
|
return str;
|
|
}
|
|
auto operator + ( StringRef const& lhs, const char* rhs ) -> std::string {
|
|
return std::string( lhs ) + std::string( rhs );
|
|
}
|
|
auto operator + ( char const* lhs, StringRef const& rhs ) -> std::string {
|
|
return std::string( lhs ) + std::string( rhs );
|
|
}
|
|
|
|
auto operator << ( std::ostream& os, StringRef const& str ) -> std::ostream& {
|
|
return os.write(str.currentData(), str.size());
|
|
}
|
|
|
|
auto operator+=( std::string& lhs, StringRef const& rhs ) -> std::string& {
|
|
lhs.append(rhs.currentData(), rhs.size());
|
|
return lhs;
|
|
}
|
|
|
|
} // namespace Catch
|
|
|
|
#if defined(__clang__)
|
|
# pragma clang diagnostic pop
|
|
#endif
|