#include "stdafx.h"
#include <iostream>
#include <conio.h>
#include <vector>


using namespace std;

template <typename ChooseType>
class Matrix {

public:
    static int n;

    Matrix(){
        Arr = new ChooseType*[50];
        for(int i=0;i<50;i++){
            Arr[i] = new ChooseType[50];
        }

        for(int i=0;i<50;i++){
            for(int j=0;j<50;j++){
                Arr[i][j] = 0;
            }
        }

    }

    void setArrElem(int a,int b){
        scanf_s("%f",&(Arr[a][b]));
        fflush(stdin);
    };

    float* getArr(){
        return *Arr;
    };

    void setN(){
        printf("N: ");
        scanf_s("%d",&n);
        fflush(stdin);
    }

    void Matrix<float>::operator -(float x){
        for(int i=0;i<n;i++){
            for(int j=0;j<n;j++){
                Arr[i][j] -= x;
            }
        }
    }

    void Matrix<int>::operator -(int x){
        for(int i=0;i<n;i++){
            for(int j=0;j<n;j++){
                Arr[i][j] -= x;
            }
        }
    }

    void operator /(float x){
        for(int i=0;i<n;i++){
            for(int j=0;j<n;j++){
                Arr[i][j] /= x;
            }
        }
    }

    void operator -(Matrix second){
        for(int i=0;i<n;i++){
            for(int j=0;j<n;j++){
                Arr[i][j] -= second.Arr[i][j];
            }
        }
    }

    void operator *(Matrix second){
        float temp[50][50];

        for(int i=0;i<n;i++){
            for(int j=0;j<n;j++){
                temp[i][j] = Arr[i][j];
            }
        }

        for(int p=0;p<n;p++){
            for(int w=0;w<n;w++){
                Arr[p][w] = 0;
                for(int j=0;j<n;j++){
                    Arr[p][w] += temp[p][j]*(second.Arr[j][w]);
                }
            }
        }
    }

    friend Matrix operator <<(Matrix Matr, float x){
        int p = (int)x;

        float temp[50];

        for(int m=0;m<p;m++){
            for(int i=0;i<n;i++){
                temp[i] = Matr.Arr[i][0];

                for(int j=0;j<n;j++){
                    Matr.Arr[i][j] = Matr.Arr[i][j+1];
                }

                Matr.Arr[i][n-1] = temp[i];
            }
        }
        return Matr;
    }

    void operator --(){
        for(int i=0;i<n;i++){
            for(int j=0;j<n;j++){
                Arr[i][j]--;
            }
        }
    }

    void Add(){
        for(int i=0;i<n;i++){
            for(int j=0;j<n;j++){
                printf("[%d][%d] = ",i,j);
                this->setArrElem(i,j);
            }
            printf("\n");
        }
    }

    void Show(){
        for(int i=0;i<n;i++){
            for(int j=0;j<n;j++){
                printf("%8.2f ",Arr[i][j]);
            }
            printf("\n");
        }
    }

private:
    ChooseType **Arr;
};
template <typename ChooseType> int Matrix<ChooseType>::n = 0;

template <typename ChooseType>
float setX<float>(){
    float x;
    printf("x = ");
    scanf_s("%f",&x);
    fflush(stdin);
    return x;
}

template <typename ChooseType>
int setX<int>(){
    int x;
    printf("x = ");
    scanf_s("%d",&x);
    fflush(stdin);
    return x;
}

void showX(float x){
    printf("x = %f \n",x);
}


int main ()
{
    int count = 0;
    int type;


    char s = 0;

    printf("Matrix type\n");
    printf("1) Int\n");
    printf("2) Float\n");
    scanf("%d",type);

    switch(type){
    case 1:
        Matrix<float> m1,m2;
        float x;

        while(s!=48){

            printf("\n Menu: \n");
            printf("1) A-x \n");
            printf("2) A/x \n");
            printf("3) A-B \n");
            printf("4) A*B \n");
            printf("5) A << x\n");
            printf("6) B-- \n");
            printf("0) Exit \n\n");

            s=_getch();


            switch (s)
            {
            case 48:
                break;
            case 49:
                m1.setN();
                m1.Add();
                x = setX();

                m1 - x;
                m1.Show();
                break;
            case 50:
                m1.Add();
                x = setX();

                m1 / x;
                m1.Show();
                break;
            case 51:
                m1.setN();
                cout << "A" << endl;
                m1.Add();
                cout << "B" << endl;
                m2.Add();

                m1 - m2;
                m1.Show();

                break;
            case 52:
                m1.setN();
                cout << "A" << endl;
                m1.Add();
                cout << "B" << endl;
                m2.Add();

                m1 * m2;
                m1.Show();
                break;
            case 53:
                m1.setN();
                m1.Add();
                x = setX();


                m1 = m1 << x;
                m1.Show();
                break;
            case 54:
                m1.setN();
                m1.Add();
                m1.Show();
                cout << endl;

                --m1;
                m1.Show();
                break;
            default:
                printf("Error: Wrong key. Try 1-6 \n");
            }
        }
        return 0;
    case 2:
        Matrix<int> m3,m4;
        int y;

        while(s!=48){

            printf("\n Menu: \n");
            printf("1) A-x \n");
            printf("2) A/x \n");
            printf("3) A-B \n");
            printf("4) A*B \n");
            printf("5) A << x\n");
            printf("6) B-- \n");
            printf("0) Exit \n\n");

            s=_getch();


            switch (s)
            {
            case 48:
                break;
            case 49:
                m3.setN();
                m3.Add();
                x = setX();

                m3 - x;
                m3.Show();
                break;
            case 50:
                m3.Add();
                x = setX();

                m3 / x;
                m3.Show();
                break;
            case 51:
                m3.setN();
                cout << "A" << endl;
                m3.Add();
                cout << "B" << endl;
                m3.Add();

                m3 - m4;
                m3.Show();

                break;
            case 52:
                m3.setN();
                cout << "A" << endl;
                m3.Add();
                cout << "B" << endl;
                m3.Add();

                m3 * m4;
                m3.Show();
                break;
            case 53:
                m3.setN();
                m3.Add();
                x = setX();


                m3 = m3 << x;
                m3.Show();
                break;
            case 54:
                m3.setN();
                m3.Add();
                m3.Show();
                cout << endl;

                --m3;
                m3.Show();
                break;
            default:
                printf("Error: Wrong key. Try 1-6 \n");
            }
        }
        return 0;
    default:
        printf("Wrong type\n");
        _getch();
        return 0;
    }


}