#include "pch.h"
#include <fstream>
#include <iomanip>
#include <string>
#include <sstream>
#include <iostream>
#include <conio.h>
using namespace std;

int main()
{
    setlocale(LC_ALL, "ru");
    int *C = NULL;
    fstream f1; int q = 0;
    do {
        system("cls");
        char ch; string fname;
        cout << "1. Создать файл с исходными данными." << endl <<
            "2. Загрузить исходные данные." << endl << "3. Вывести исходные данные." <<
            endl << "4. Обработать данные." << endl << "5. Завершить работу с программой." << endl;
        cout << "Для выполнения необходимого действия нажмите соответствующую цифровую клавишу: ";
        ch = _getch();
            system("cls");

            switch (ch) {
            case '1':

                do {

                    cout << "Введите количество элементов массива или \"*\" для возврата в меню: ";
                    cin >> ws;
                    if (cin.rdbuf()->in_avail() == 2 && cin.peek() == '*') {
                        cin.ignore(cin.rdbuf()->in_avail(), '\n');
                        system("cls");
                        break;
                    }
                    cin >> q;
                    if (cin.peek() != '\n')
                    {
                        cin.clear();
                        cin.ignore(cin.rdbuf()->in_avail(), '\n');
                        cout << "Введено некорректное значение! Повторите ввод!" << endl;
                        continue;
                    }
                    if (q <= 0)
                    {
                        cout << "Введено недопустимое значение! Повторите ввод!" << endl;
                        continue;
                    }
                    if (C != NULL)
                        delete[]C;

                    C = new (nothrow) int[q];
                    if (!C) {
                        cout << "Ошибка выделения памяти!" << endl;
                        system("cls");
                        _getch();
                        continue;
                    }

                    for (int j = 0; j < q;)
                    {
                        cout << "Введите " << j + 1 << " элемент массива или \"*\" для возврата в меню: ";
                        cin >> ws;
                        if (cin.rdbuf()->in_avail() == 2 && cin.peek() == '*') {
                            cin.ignore(cin.rdbuf()->in_avail(), '\n');
                            system("cls");
                            break;
                        }
                        cin >> C[j];
                        if (cin.peek() != '\n')
                        {
                            cin.clear();
                            cin.ignore(cin.rdbuf()->in_avail(), '\n');
                            cout << "Элемент содержит ошибку! Повторите ввод!" << endl;
                            continue;
                        }
                        j++;
                    }
                    break;
                } while (true);

                do {
                    cin.ignore(cin.rdbuf()->in_avail(), '\n');
                    cout << "Введите имя файла для записи или \"*\" для возврата в меню: " << endl;
                    getline(cin, fname);
                    if (fname == "*")
                    {
                        system("cls");
                        break;
                    }
                    string z;
                    f1.open(fname, ios_base::in);
                    if (f1) {
                        f1.close();
                        cout << "Файл с таким именем существует!" << endl << "Для работы с другим файлом введите \"*\", для продолжения - любую последовательность символов: ";
                        cin >> z;
                        if (z == "*")
                        {
                            continue;
                        }
                    }

                    f1.open(fname, ios_base::out | ios_base::binary);

                    if (!f1)
                    {
                        cout << "Не удалось открыть файл!" << endl;
                        continue;
                    }
                    break;
                } while (true);

                f1.write((char*)&q, sizeof(q));
                f1.write((char*)C, q*sizeof(C[0]));

                cout << "Файл успешно создан!" << endl;
                f1.close();
                _getch();
                break;

            case '2':

                do {
                    cin.ignore(cin.rdbuf()->in_avail(), '\n');
                    cout << "Введите название файла или * для возврата в меню: ";
                    getline(cin, fname);
                    if (fname == "*") {
                        system("cls");
                        break;
                    }

                    f1.open(fname, ios_base::in | ios_base::binary);
                    if (!f1)
                    {
                        cout << "Не удалось открыть файл!" << endl;
                        continue;
                    }
                    f1.read((char*)&q, sizeof(q));
                    f1.seekg(0, ios_base::end);
                    int p = f1.tellg();
                    p -= sizeof(q);
                    if (p != q * sizeof(C[0]))
                    {
                        f1.close();
                        cout << "Файл содержит некорректные данные. ";
                        cin.ignore(cin.rdbuf()->in_avail(), '\n');
                        continue;
                    }

                    if (C != NULL)
                        delete[]C;

                    C = new (nothrow) int[q];
                    if (!C)
                    {
                        f1.close();
                        cout << "Ошибка выделения памяти.";
                        continue;

                    }
                    f1.seekg(sizeof(q));
                    f1.read((char*)C, q * sizeof(C[0]));
                    f1.close();
                    break;
                } while (true);

                cout << "Данные успешно загружены!";
                _getch();
                break;

            case '3':
            {

                cout << "Ниже представлены исходные данные." << endl << endl;
                cout << setw(10) << "Номер";
                for (int i = 1; i <= q; i++)
                    cout << setw(4) << i;
                cout << endl << setw(10) << "Элемент";
                for (int i = 0; i < q; i++)
                    cout << setw(4) << C[i];
                cout << endl << endl;
                _getch();
                break;
            }

            case '4': {

                cout << "Ниже представлена обработка данных соответственно заданию." << endl << endl;
                int na = 0, nb = 0;
                for (int i = 0; i < q; i++)
                    if (C[i] < 0)
                        na++;
                    else
                        if (C[i] > 0)
                            nb++;

                int *A = new (nothrow) int[na];
                if (!A) {
                    delete[] C;
                    cout << "Ошибка выделения памяти!" << endl;
                    continue;
                }

                int *B;
                try {
                    B = new int[nb];
                }
                catch (...)
                {
                    delete[] C;
                    delete[] A;
                    cout << "Ошибка выделения памяти" << endl;
                    continue;
                }

                int pa = 1, pb = 1;
                int j1 = 0, j2 = 0;

                for (int i = 0; i < q; i++) {
                    if (C[i] < 0) {
                        A[j1] = C[i];
                        pa *= A[j1];
                        j1++;
                    }
                    else
                        if (C[i] > 0) {
                            B[j2] = C[i];
                            pb *= B[j2];
                            j2++;
                        }
                }

                cout << endl;
                if (na == 0)
                    cout << "Массив не имеет отрицательных элементов!";
                else {
                    cout << setw(10) << "Массив A" << endl;
                    cout << setw(10) << "Номер";
                    for (int i = 1; i <= na; i++)
                        cout << setw(4) << i;
                    cout << endl << setw(10) << "Элемент";
                    for (int i = 0; i < na; i++)
                        cout << setw(4) << A[i];
                }
                cout << endl;
                delete[] A;

                cout << endl;
                if (nb == 0)
                    cout << "Массив не имеет положительных элементов!";
                else {
                    cout << setw(10) << "Массив B" << endl;
                    cout << setw(10) << "Номер";
                    for (int i = 1; i <= nb; i++)
                        cout << setw(4) << i;
                    cout << endl << setw(10) << "Элемент";
                    for (int i = 0; i < nb; i++)
                        cout << setw(4) << B[i];
                }
                delete[] B;

                cout << endl << endl;

                if (na != 0 && nb != 0)
                    cout << "Отношение произведения элементов массива  В к произведению элементов массива А: " << (double)pb / pa << endl;
                else
                    cout << "Невозможно вычислить отношение произведений!";
                cout << endl;
                cout << endl;
                _getch();
                break;
            }
            case '5':
                cout << "Работа программы завершена!"<<endl;
                return 0;

            default:
                cout << "Введено неверное значение! Повторите ввод!" << endl;
                _getch();
                break;
            }
    } while (true);
}