src/rc_string.cpp


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61

/*
 * MRustC - Mutabah's Rust Compiler
 * - By John Hodge (Mutabah/thePowersGang)
 *
 * rc_string.cpp
 * - Reference-counted string
 */
#include <rc_string.hpp>
#include <cstring>
#include <iostream>

RcString::RcString(const char* s, unsigned int len):
    m_ptr(nullptr),
    m_len(len)
{
    if( len > 0 )
    {
        m_ptr = new unsigned int[1 + (len+1 + sizeof(unsigned int)-1) / sizeof(unsigned int)];
        *m_ptr = 1;
        char* data_mut = reinterpret_cast<char*>(m_ptr + 1);
        for(unsigned int j = 0; j < len; j ++ )
            data_mut[j] = s[j];
        data_mut[len] = '\0';

        //::std::cout << "RcString(" << m_ptr << " \"" << *this << "\") - " << *m_ptr << " (creation)" << ::std::endl;
    }
}
RcString::RcString(const RcString& x):
    m_ptr(x.m_ptr),
    m_len(x.m_len)
{
    if( m_ptr )
    {
        *m_ptr += 1;
        //::std::cout << "RcString(" << m_ptr << " \"" << *this << "\") - " << *m_ptr << " refs present (copy)" << ::std::endl;
    }
}
RcString::~RcString()
{
    if(m_ptr)
    {
        *m_ptr -= 1;
        //::std::cout << "RcString(" << m_ptr << " \"" << *this << "\") - " << *m_ptr << " refs left (drop)" << ::std::endl;
        if( *m_ptr == 0 )
        {
            delete[] m_ptr;
            m_ptr = nullptr;
        }
    }
}
bool RcString::operator==(const char* s) const
{
    if( m_len == 0 )
        return *s == '\0';
    auto m = this->c_str();
    do {
        if( *m != *s )
            return false;
    } while( *m++ != '\0' && *s++ != '\0' );
    return true;
}