//LABKA 2-4 v.1
#include <iostream>
#include <fstream>
#include <string>
#include "classes.h"

void putAboutStudent();

std::istream& setIStream(std::ifstream& f);
std::ostream& setOStream(std::ofstream& f);

int main() {
    putAboutStudent();
    Polynom operandA, operandB;

    // INPUT
    try {
        std::ifstream f;
        std::istream& s_inA = setIStream(f);
        s_inA >> operandA;
        std::istream& s_inB = setIStream(f);
        s_inB >> operandA;
    } catch (std::exception& ex) {
        std::cerr << "\n\t" << ex.what() << std::endl;
        system("pause");
        return -1;
    }

    Polynom resultPlus = operandA + operandB;
    Polynom resultMinus = operandA - operandB;
    operandA += operandB;

    // OUTPUT
    std::ofstream g;
    std::ostream& s_outP = setOStream(g);
    s_outP << resultPlus;
    std::ostream& s_outM = setOStream(g);
    s_outM << resultMinus;
    std::ostream& s_outPE = setOStream(g);
    s_outPE << operandA;

    // CLEANING UP MEMORY
    operandA.~Polynom();
    operandB.~Polynom();
    resultPlus.~Polynom();
    resultMinus.~Polynom();
    system("pause");
    return 0;
}

void putAboutStudent() {
    std::cout << "Laboratory work 2 - 4 Linked Lists" << std::endl;
    std::cout << "Group: K-14 Danilchenko Alexander" << std::endl;
    std::cout << "VAR: 70" << std::endl;
}

std::istream& setIStream(std::ifstream& f) {
    //int i;
    std::cout << "... Enter 1(kbd) or 2(file): ";
    //std::cin >> i;

    int z = 0;
    std::cin >> z;
    if (z == 1)
        return std::cin;
    else
        while (true) {
            std::cout << "... Enter filepath: ";
            std::string fN;
            std::cin.get(); // pass Enter from previous input
            getline(std::cin, fN);
            if (fN.length() == 0)
                fN = "in.txt";
            f.open(fN);
            if (f.fail())
                continue;
            return (std::istream&)f;
        }
}

std::ostream& setOStream(std::ofstream& f) {
    int i;
    std::cout << "... Enter for out: 1(screen) or 2(file): ";
    std::cin >> i;
    if (i == 1)
        return std::cout;
    else
        while (true) {
            std::cout << "... Enter filepath: ";
            std::string fN;
            std::cin.get(); // pass Enter from previous input
            getline(std::cin, fN);
            if (fN.length() == 0)
                fN = "out.txt";
            f.open(fN);
            if (f.fail())
                continue;
            return (std::ostream&)f;
        }
}