#include "Pociag.h"
#include <iostream>

using namespace std;

int Pociag::fLicznik = 0;

Pociag::Pociag()
{
  n = 0;
  fRozmiar = 10;
  nrpociagu = new int[fRozmiar];
  for(int i=0;i<fRozmiar;i++)
    nrpociagu[i] = 1;
  fLicznik++;
}

Pociag::Pociag(int N, int Rozmiar)
{
  n = N;
  fRozmiar = Rozmiar;
  nrpociagu = new int[fRozmiar];
  for(int i=0;i<fRozmiar;i++)
    {
      cout<<"Podaj wartosc nr "<<i<<" numeru pociagu"<<endl;
      cin>>nrpociagu[i];
    }
  fLicznik++;
}


Pociag::Pociag(const Pociag &p)
{
  n = p.n;
  fRozmiar = p.fRozmiar;
  nrpociagu = new int[fRozmiar];
  for(int i=0;i<fRozmiar;i++)
    nrpociagu[i] = p.nrpociagu[i];
  fLicznik++;
}

Pociag::~Pociag()
{
  delete[] nrpociagu;
  fLicznik--;
}

void Pociag::SetN(int N)
{
  n = N;
}

void Pociag::SetNrPociagu(int Rozmiar, int *tab)
{
  fRozmiar = Rozmiar;
  delete[] nrpociagu;
  nrpociagu = new int[fRozmiar];
  for(int i=0;i<fRozmiar;i++)
    nrpociagu[i] = tab[i];
}

void Pociag::Wypisz()
{
  cout<<"n = "<<n<<endl;
  cout<<"Numer pociagu: ";
  for(int i=0;i<fRozmiar;i++)
    cout<<nrpociagu[i];
  cout<<endl;
}

int Pociag::GetLicznik()
{
  return fLicznik;
}


Pociag& Pociag::operator+(const Pociag &p)
{
  Pociag *temp = new Pociag; //przez wskaznik, gdyz inaczej zwracamy lokalna zmienna (pozostaje warning)
  temp->n = n+p.n;
  temp->fRozmiar = 10;
  delete[] temp->nrpociagu;
  temp->nrpociagu = new int[fRozmiar];
  for(int i=0;i<temp->fRozmiar;i++)
    temp->nrpociagu[i] = 2;

  fLicznik--;
  return *temp;
}




Pociag& Pociag::operator=(const Pociag &p)
{
  if(this==&p)
    return *this;

  delete[] nrpociagu;

  n = p.n;
  fRozmiar = p.fRozmiar;
  nrpociagu = new int[fRozmiar];
  for(int i=0;i<fRozmiar;i++)
    nrpociagu[i] = p.nrpociagu[i];

  return *this;
  
}


ostream& operator<<(ostream &out,const Pociag &p)
{
  out<<"n = "<<p.n<<endl;
  out<<"nr pociagu: ";
  for(int i=0;i<p.fRozmiar;i++)
    out<<p.nrpociagu[i];
  out<<endl;

  return out;
}