/* Вариант 9
Написать программу «Автоматизированная информационная система на железнодорожном вокзале».
Информационная система содержит сведения об отправлении поездов дальнего следования. Для каждого поезда указывается: 
- номер поезда; 
- станция назначения;
- время отправления.
Программа должна обеспечивать выбор с помощью меню и выполнение одной из следующих функций:
- первоначальный ввод данных в информационную систему (с клавиатуры или из файла);
- вывод сведений по всем поездам;
- вывод сведений по поезду с запрошенным номером;
- вывод сведений по тем поездам, которые следуют до запрошенной станции назначения. 
Хранение данных организовать с применением контейнерного класса vector. */
#include "stdafx.h"
#include <iostream>
#include <Windows.h>
#include <string>
#include <conio.h>
#include <vector>
using namespace std;
class Train
{
    int number;
    string destination;
    string time;
public:
    Train()
    {
        number = 0;
        destination = "";
        time = "";
    }
    Train(int NumParam, string DestParam, string TimeParam)
    {
        number = NumParam;
        destination = DestParam;
        time = TimeParam;
    }
    int GetNumber()
    {
        return number;
    }
    Train(Train const ©):number(copy.number), destination(copy.destination), time(copy.time) {}
    string GetDestination()
    {
        return destination;
    }
    string GetTime()
    {
        return time;
    }
    void operator = (Train objRight)
    {
        number = objRight.number;
        destination = objRight.destination;
        time = objRight.time;
    }
};
// Функция, для ввода данных
void AddTrain(vector <Train> &vSource)
{
    int numb;
    string station,times;
    cout << "Введите номер поезда: ";
    cin >> numb; 
    cin.ignore();
    cout << "Введите станцию назначения: ";
    getline(cin, station, '\n');
    cout << "Введите время отправления: ";
    cin >> times; 
    cin.ignore();
    vSource.push_back(Train(numb, station, times));
}
// Функция списка данных
void PrintList(vector <Train> &vSource)
{
    if (vSource.empty())
    {
        cout <<"Список пуст";
        return;
    }
    vector <Train>::iterator iterSource = vSource.begin();
    cout << "Номер поезда пункт назначения время отправления " << endl;
    while (iterSource != vSource.end())
    {
        cout << iterSource -> GetNumber() << '\t' << iterSource -> GetDestination() << '\t' << iterSource -> GetTime() << '\n';
        iterSource++;
    }
}
// Функция, для информации о поездах
void GetInfoTrain(vector <Train> &vSource)
{
    if (vSource.empty())
    {
        cout <<"Список пуст";
        return;
    }
    vector <Train>::iterator iterSource = vSource.begin();
    bool found = false;
    int train_search;
    cout << "Введите номер поезда: ";
    cin >> train_search; 
    cin.ignore();
    while (iterSource != vSource.end())
    {
        if (iterSource -> GetNumber() == train_search)
        {
            found = true;
            cout << iterSource -> GetDestination() << '\t' << iterSource -> GetTime() << '\n';
        }
        iterSource++;
    }
    if (!found) cout << "Такого поезда не найдено";
}
// Функция, для информации о станции
void GetInfoStation(vector <Train> &vSource)
{
    if (vSource.empty())
    {
        cout <<"Список пуст";
        return;
    }
    vector <Train>::iterator iterSource = vSource.begin();
    bool found = false;
    string station_search;
    cout << "Введите станцию назначения: ";
    getline(cin, station_search, '\n');
    while (iterSource != vSource.end())
    {
        if (iterSource -> GetDestination() != station_search)
        {
            found = true;
            cout << iterSource -> GetNumber() << '\t' << iterSource -> GetDestination() << '\t' << iterSource -> GetTime() << '\n';
        }
        iterSource++;
    }
    if (!found) cout << "Нет такой станции";
}
 
int main()
{
    SetConsoleCP(1251);
    SetConsoleOutputCP(1251); 
    vector <Train> vDataBase;
    char cKey;
    do{
        cout << "1 - Добавить информацию о новом поезде в базу данных\n";
        cout << "2 - Показать базу данных\n";
        cout << "3 - Показать информацию о конкретном поезде\n";
        cout << "4 - Показать информацию о прибывающих поездах на станцию\n";
        cout << "5 - Выход\n";
        cKey = _getch();
        if (cKey == '1')
        {
            AddTrain(vDataBase);
        }
        else if (cKey == '2')
            PrintList(vDataBase);
        else if (cKey == '3')
            GetInfoTrain(vDataBase);
        else if (cKey == '4')
            GetInfoStation(vDataBase);
        else if (cKey == '5') break;
        else 
        continue;
        _getch();
    }
        while (cKey != '5');
        return 0;
}