mipt_cpp/seminar02_encapsulation/classroom_tasks/code/2string/12full_string.cpp

/*
    Собираем все методы вместе. Получилась строка, которой можно удобно пользоваться и не задумываться о выделении памяти.
*/

#include <iostream>
#include <cstdlib>
using std::cout, std::cin, std::endl, std::size_t;


class String 
{
private:

    size_t mSize;
    size_t mCapacity;
    char* mpData;

public:

    String(const char* str) 
    {
        size_t i = 0;
        while (str[i] != '\0')
            i++;
        mSize = i;
        mCapacity = i + 1;

        mpData = (char*)std::malloc(sizeof(char) * mCapacity);

        for (size_t i = 0; str[i]; i++)
            mpData[i] = str[i];
        mpData[mSize] = '\0';
    }

    String() : String("") {}
    String(const String& s) : String(s.cStr()) {}

    String(size_t n, char a)
    {
        mSize = n;
        mCapacity = n + 1;
        mpData = (char*)std::malloc(sizeof(char) * mCapacity);

        for (size_t i = 0; i < mSize; ++i)
            mpData[i] = a;
        mpData[mSize] = '\0';
    }

    ~String()
    {
        std::free(mpData);
    }

    void reserve(size_t capacity)
    {
        if (capacity <= mCapacity)
            return;

        mCapacity = capacity;
        char* newData = (char*)std::malloc(sizeof(char) * mCapacity);

        for (size_t i = 0; i < mSize; ++i)
            newData[i] = mpData[i];
        newData[mSize] = '\0';

        std::free(mpData);
        mpData = newData;
    }


    void resize(size_t size)
    {
        reserve(size + 1);
        mSize = size;
        mpData[mSize] = '\0';
    }


    String& operator=(const String& right)
    {
        if (this == &right)
            return *this;

        mSize = right.mSize;
        resize(mSize);

        for (size_t i = 0; i <= mSize; ++i)
            mpData[i] = right.mpData[i];

        return *this;
    }


    String operator+(const String& b)
    {
        String result;
        result.resize(mSize + b.mSize);

        for (size_t i = 0; i < mSize; ++i)
            result.mpData[i] = mpData[i];

        for (size_t i = 0; i < b.mSize; ++i)
            result.mpData[mSize + i] = b.mpData[i];

        result.mpData[result.mSize] = '\0';

        return result;
    }

    String& operator+=(const String& right)
    {
        *this = *this + right;
        return *this;
    }

    bool operator==(const String& right) const
    {
        if (mSize != right.mSize)
            return false;

        size_t i = 0;
        while (i < mSize && mpData[i] == right.mpData[i])
            i++;

        return i == mSize;
    }

    bool operator<(const String& right) const
    {
        size_t i = 0;
        while (i < mSize && i < right.mSize && mpData[i] == right.mpData[i])
            i++;

        return mpData[i] < right.mpData[i];
    }

    bool operator<=(const String& right) const
    {
        size_t i = 0;
        while (i < mSize && i < right.mSize && mpData[i] == right.mpData[i])
            i++;

        return mpData[i] <= right.mpData[i];
    }

    bool operator!=(const String& right) const
    {
        return !(*this == right);
    }

    bool operator>(const String& right) const
    {
        return !(*this <= right);
    }

    bool operator>=(const String& right) const
    {
        return !(*this < right);
    }

    char& operator[](size_t i)
    {
        return mpData[i];
    }

    char& at(size_t i)
    {
        if (i >= mSize)
        {
            cout << "Error! Index is out of bounds." << endl;
            std::exit(1);
        }
        return mpData[i];
    }

    void clear()
    {
        std::free(mpData);

        mSize = 0;
        mCapacity = 1;
        mpData = (char*)std::malloc(sizeof(char) * mCapacity);
        mpData[0] = '\0';
    }

    void addCharacter(char c)
    {
        if (mSize + 1 == mCapacity)
            reserve(2 * mCapacity);

        mpData[mSize] = c;
        resize(mSize + 1);
    }


    size_t getSize()        const   {return mSize;}
    size_t getCapacity()    const   {return mCapacity;}
    const char* cStr()      const   {return mpData;}
};


std::ostream& operator<<(std::ostream& out, const String& s) 
{
    out << s.cStr();
    return out;
}

std::istream& operator>>(std::istream& in, String& s) 
{
    s.clear();
    while (true)
    {
        char x = in.get();
        if (x == ' ' || x == '\n' || x == '\t')
            break;
        s.addCharacter(x);
    }
    return in;
}


int main() 
{
    String a = "Mouse";
    String b;
    cin >> b;
    String c = b;

    if (a + c == "MouseLion")
        cout << "Yes" << endl;
    else
        cout << "No" << endl;


    if (a > "Mice")
        cout << "Yes" << endl;
    else
        cout << "No" << endl;


    c[0] = 'P';
    cout << c << endl;

    c += a;
    cout << c << endl;

    c = c + String(10, 'q');
    cout << c << endl;
}



/*
    Задача:


        1)  Напищите программу, которая будет считывать слова (используйте cin) в бесконечном цикле и каждый
            раз печатать сумму всех слов. Например, если пользователь ввёл Hello, то программа должна напечатать
            Hello и запросить следующее слово. Если затем пользователь введёт World, то программа должна будет
            напечатать HelloWorld и запросить следуещее слово и так далее.
            Программа должна останавливаться если пользователь ввёл слово "quit".
*/