#include <iostream>
#include <vector>
#include "Figure.h"
#include "Rectangle.h"
#include "Parallelogram.h"


void printCommands() {
    printf("1: Add\n2: Print all\n3: Summ squares\n4: Sum perimeters\n5: Centre of mass\n6: Memory\n7: Sort figures\n0: Exit\n");
}

int main() {
    std::vector<CFigure *> gFigs;
    int command = 0;
    while (true) {
        printCommands();
        std::cin >> command;
        switch (command) {
            case 1:
                printf("What figure are you want to add?\n1: Rectangle 2:Parallelogram\n");
                int x;
                std::cin >> x;
                if (x == 1) {
                    CRectangle *tmp = new CRectangle();
                    tmp->initFromDialog();
                    gFigs.push_back(tmp);
                } else {
                    CParallelogram *tmp = new CParallelogram();
                    tmp->initFromDialog();
                    gFigs.push_back(tmp);
                }
                break;
            case 2: {
                printf("-----------------------\n");
                for (int i = 0; i < gFigs.size(); i++) {
                    printf("%d: ", i);
                    gFigs[i]->draw();
                }
                printf("-----------------------\n");
                break;
            }
            case 3: {
                double s = 0;
                for (int i = 0; i < gFigs.size(); i++) {
                    s += gFigs[i]->square();
                }
                std::cout << s << std::endl;
                break;
            }
            case 4: {
                double s = 0;
                for (int i = 0; i < gFigs.size(); i++) {
                    s += gFigs[i]->perimeter();
                }
                std::cout << s << std::endl;
                break;
            }
            case 5: {
                double x = 0, y = 0, n = 0;
                for (int i = 0; i < gFigs.size(); i++) {
                    x += gFigs[i]->position().x * gFigs[i]->mass();
                    y += gFigs[i]->position().y * gFigs[i]->mass();
                    n += gFigs[i]->mass();
                }
                std::cout << x / n << " " << y / n << std::endl;
                break;
            }
            case 6: {
                double s = 0;
                for (int i = 0; i < gFigs.size(); i++) {
                    s += gFigs[i]->size();
                }
                std::cout << s << std::endl;
                break;
            }
            case 7: {
                for (int i = 0; i < gFigs.size(); i++) {
                    for (int j = 0; j < gFigs.size(); j++) {
                        if ( gFigs[j] < gFigs[i]) {
                            CFigure *tmp = gFigs[i];
                            gFigs[i] = gFigs[j];
                            gFigs[j] = tmp;
                        }
                    }
                }
                break;
            }
            case 0: {
                exit(0);
            }

        }
    }
}