#include <iostream.h>
#include<stdlib.h>

typedef int * matrix_member;

void ask_matrix_size(int *size);
matrix_member allocate_memory(int size);
matrix_member read_matrix(int size);
void print_matrix(matrix_member matrix,int size);
matrix_member create_matrix(int *size);
void add_matrix(matrix_member m1,matrix_member m2,matrix_member result,int size);
matrix_member return_address(int i,int j,matrix_member m,int size);



int main () 
{

	int size=0;
	int choice=0;
	matrix_member matrix1;
	matrix_member matrix2 ;
	matrix_member result_matrix;

	matrix1=create_matrix( &size);
	matrix2=create_matrix (&size);

	print_matrix (matrix1,size);
	print_matrix (matrix2,size);

	while (choice!=3)
	{
		cout<<"==========MENU==========\n";
		cout<<"1)Add matrices\n";
		cout<<"2)Multiply matrices\n";
		cout<<"3)Exit\n";

		cout<<"Enter your choice:";
		cin>>choice;

		switch (choice) {

			case 1:
				add_matrix (matrix1,matrix2,result_matrix,size);
				break;
			case 2://multiplication will come here but first addition. . .
				
				break;
			case 3:
				cout<<"Thanks!... See you soon";
				cout<<"copyright by birben";
				return -1;
			default:
				cout<<"Choose the right number for your operation!";
		}
	}

	return 0;
}

void ask_matrix_size (int * size) //ask the user matrix size 
{
	cout<<"Enter the size of the matrix that you want to make operations";
	cin>> *size;
}

matrix_member allocate_memory (int size) //allocate memory for the array
{
	matrix_member start; //we define a integer pointer for the start of the array
	if(!(start=(matrix_member)calloc( (size*size) ,sizeof(matrix_member) ) )){
		cout<<"not enough memory";
		
	} 

	return start;
}

matrix_member read_matrix ( int size ) //this will take the matrix elements and place them inside it
{
	int i;
	matrix_member m;
	for (i=0; i<size*size ;i++) 
	{
		cout<<"Enter the number for the matrix then press enter";
		cin>>*(m+i);
		
	}
	
	return m;
} 

void print_matrix (matrix_member matrix, int size)
{
	int i;
	for (i=0 ;i<size ;i++)
	{
		cout<<*(matrix+i);
	}
}

matrix_member create_matrix (int * size)
{
	
	matrix_member matrix;
	ask_matrix_size (size);
	matrix=allocate_memory(*size);
	read_matrix(*size);

	return matrix;
}

matrix_member return_adress (int i,int j,matrix_member m, int size)//return the adress of the desired component of the matrix
{
	   return (m + ((i*size)+j));
}

void add_matrix(matrix_member m1,matrix_member m2,matrix_member result,int size)
{
	int i,j;
	for(i=0  ; i<size ;i++) {
		for( j=0 ;j<size ;j++) {
			
			*(return_adress(i,j,result,size) )= (* (return_adress(i,j,m1,size) ) ) + ( * (return_adress(i,j,m2,size) ) );
			
			cout<<*(return_adress(i,j,result,size) );

		}
	}
}