Перегрузка операций += и -= - C++ - Ответ 10513330

#ifndef BOTTLE_H_
#define BOTTLE_H_
#include <iostream>
class Port {
private:
    char * brand;
    char style[20];     // tawny, ruby, vintage
    int bottles;
public:
    Port(const char * br = "NONE", const char * st = "NONE", int b = 0);
    Port(const Port & p);
    virtual ~Port() {   delete [] brand;    }
    Port & operator=(const Port & p);
    Port & operator+(int b);    // add b to bottles
    Port & operator-=(int b);   // del b from bottles, if it's possible
    int BottleCount() { return bottles; }
    virtual void Show() const;
    friend std::ostream & operator<<(std::ostream & os, const Port & p);
};
class VintagePort : public Port {
private:
    char * nickname;    //"The Noble" or "Old Velvet"
    int year;           
public:
    VintagePort();
    VintagePort(const char * br, int b, const char * nn, int y);
    VintagePort(const VintagePort & vp);
    ~VintagePort() {    delete [] nickname; }
    VintagePort & operator=(const VintagePort & vp);
    void Show() const;
    friend std::ostream & operator<<(std::ostream & os, const VintagePort & vp);
};
#endif

#include <iostream>
#include "bottle.h"
 
////////////////////////////////    Port        //////////////////////////
 
Port::Port(const char *br, const char *st, int b)
{
    brand = new char[std::strlen(br) + 1];
    std::strcpy(brand, br);
    std::strncpy(style, st, std::strlen(st) + 1);
    style[std::strlen(st) + 1] = '\0';
    bottles = b;
}
 
Port::Port(const Port &p)
{
    brand = new char[std::strlen(p.brand) + 1];
    std::strcpy(brand, p.brand);
    std::strncpy(style, p.style, std::strlen(p.style) + 1);
    style[std::strlen(p.style) + 1] = '\0';
    bottles = p.bottles;
}
 
Port & Port::operator =(const Port &p)
{
    if(this == &p)
        return *this;
    delete [] brand;
    brand = new char[std::strlen(p.brand) + 1];
    std::strcpy(brand, p.brand);
    std::strncpy(style, p.style, std::strlen(p.style) + 1);
    style[std::strlen(p.style) + 1] = '\0';
    bottles = p.bottles;
    return * this;
}
 
Port & Port::operator+=(int b)
{
    bottles += b;
    return *this;
}
 
Port & Port::operator -=(int b)
{
    if (b > BottleCount())
        std::cout << "Count of botteles on storage less than you want to take. The try is not successful.\n";
    else
        bottles -= b;
    return * this;
}
 
void Port::Show() const
{   std::cout << "Brand: " << brand << "\nKind: " << style << "\nBotteles: " << bottles << std::endl;   }
 
std::ostream & operator<<(std::ostream & os, const Port & p)
{   
    os << p.brand << ", " << p.style << ", " << p.bottles;
    return os;
}
 
////////////////////////////////    VintagePort     //////////////////////////
 
VintagePort::VintagePort() : Port()
{
    nickname = new char[1];
    nickname[0] = '\0';
    year = 0;
}
 
VintagePort::VintagePort(const char *br, int b, const char *nn, int y) : Port(br, "Vintage", b)
{
    nickname = new char[std::strlen(nn) + 1];
    std::strcpy(nickname, nn);
    year = y;
}
 
VintagePort::VintagePort(const VintagePort &vp) : Port(vp)
{
    nickname = new char[std::strlen(vp.nickname) + 1];
    std::strcpy(nickname, vp.nickname);
    year = vp.year;
}
 
VintagePort & VintagePort::operator =(const VintagePort &vp)
{
    if (this == &vp)
        return *this;
    delete [] nickname;
    nickname = new char[std::strlen(vp.nickname) + 1];
    std::strcpy(nickname, vp.nickname);
    year = vp.year;
    return *this;
}
 
void VintagePort::Show() const
{
    Port::Show();
    std::cout << "NickName: " << nickname << "\nYear: " << year << std::endl;
}
 
std::ostream & operator <<(std::ostream & os, const VintagePort & vp)
{
    os << (const Port &) vp;
    os << ", " << vp.nickname << ", " << vp.year << std::endl;
    return os;
}

#include <iostream>
#include "bottle.h"
 
using namespace std;
 
char * setNickName(int tnn);
char * setStyle(int s);
 
int main()
{
    int numVine;
    cout << "Enter count of brands: ";
    cin >> numVine;
    Port ** p_vines = new Port*[numVine];   
    char tempBrand[20];
    int tempStyle;                              // 1 - VintagePort, 2 - Tawny, 3 - Ruby
    int tempBottles;
    int tempNickName;                       // 1 - The Noble, 2 - Old Velvet
    int tempYear;
 
    for (int i = 0; i < numVine; i++)
    {
        cout << "\nEnter Brand " << i + 1 << ": ";
        cin >> tempBrand;
        cout << "Set style of bottles: 1 - VintagePort, 2 - Tawny, 3 - Ruby: ";
        cin >> tempStyle;
        cout << "Enter count of botteles: ";
        while (cin >> tempBottles && tempBottles < 0)
            cout << "Count of bottles must be positive. Enter count of botteles AGAIN: ";
        if (tempStyle == 1)
        {
            cout << "Enter 1 for nickName "The Noble" and 2 for nickName "Old Velvet": ";
            cin >> tempNickName;
            cout << "Enter year: ";
            while (cin >> tempYear && ( tempYear < 1000 || tempYear > 2017))
                cout << "Enter year: ";
            p_vines[i] = new VintagePort(tempBrand, tempBottles, setNickName(tempNickName), tempYear);
        }
        else
        {
            p_vines[i] = new Port(tempBrand, setStyle(tempStyle), tempBottles);
        }
    }
 
    cout << endl;
 
    for (int i = 0; i < numVine; i++)
    {
        p_vines[i] ->Show();
        cout << endl;
    }
 
    int b;
    cout << "Enter count of bottles you want to take: ";
    cin >> b;
    for (int i = 0; i < numVine; i++)
        p_vines[i] = p_vines[i] + b;
 
    for (int i = 0; i < numVine; i++)
        delete p_vines[i];
 
    return 0;
}
 
char * setStyle(int s)
{
    if (s == 1)         return "VintagePort";
    else if (s == 2)    return "Tawny";
    else if (s == 3)    return "Ruby";
    else                return "none";
}
 
char * setNickName(int tnn)
{
    if (tnn == 1)       return "The Noble";
    else if (tnn == 2)  return "Old Velvet";
    else                return "none";
}