main.cpp

#include <iostream>
#include <string>
#include <algorithm>

#include "Vector.h"
#include "Square.h"

void menu() {
	std::cout << "exit : EXIT\n";
	std::cout << "add : INSERT OBJECT BY INDEX\n";
	std::cout << "erase : ERASE OBJECT BY INDEX\n";
    std::cout << "count : PRINT COUNT OF FIGURES, WHICH SQUARE IS GREATER THAN 'N'\n";
    std::cout << "print : PRINT ALL FIGURES\n";
    std::cout << "size : PRINT COUNT OF FIGURES\n";
	std::cout << "> ";
}


int main() {
    menu();

    std::string command;
    Containers::Vector<Square<double>> figures;
    while (std::cin >> command) {
        if (command == "add") {
            size_t position;
            std::cin >> position;
            auto it = figures.begin();
            try {
                it = std::next(it, position);
            } catch(std::exception& e) {
                std::cout << "Position is too big\n";
                continue;
            }
            Square<double> new_figure;
            try {
                std::cin >> new_figure;
                figures.Insert(it, new_figure);
                std::cout << new_figure << "\n";
            } catch (std::exception& ex) {
                std::cout << ex.what() << "\n";
            }
        } else if (command == "erase") {
            size_t index;
            std::cin >> index;
            try {
                if(figures.Size() == 0) {
                std::cout << "Array is empty, cant erase." << std::endl;
            }
            else{
                auto it = std::next(figures.begin(), index);
                figures.Erase(it);
            }
            } catch (...) {
                std::cout << "Index is too big\n";
                continue;
            }
        } else if (command == "size") {
          std::cout << figures.Size() << "\n";
        } else if (command == "print") {
            std::for_each(figures.begin(), figures.end(), [] (const Square<double>& fig) {
                std::cout << fig << " "<< std::endl;
            });
            std::cout << "\n";
        } else if (command == "count") {
            size_t required_area;
            std::cin >> required_area;
            std::cout << std::count_if(figures.begin(), figures.end(), [&required_area] (const Square<double >& fig) {
                return fig.Area() < required_area;
            });
            std::cout << "\n";
        } else if (command == "exit") {
            break;
        } else {
                std::cout << "Incorrect command" << "\n";
                std::cin.ignore(32767, '\n');
            }
        }
    }