matrix.c

Go to the documentation of this file.

#include<stdio.h>
#include<stdlib.h>
#include"numbers.h"
#include"matrix.h"

static number addMatrix(const number x,const number y);
static number multMatrix(const number x,const number y);
static number getUnitMatrix(const number x);
static number getZeroMatrix(const number x);
static void displayMatrix(const number x);

enum matrix_mode{ ALLOC, FREE, FREEALL, INFO};

static matrix* manageMatrix(enum matrix_mode mode, int* n, number* p){
  static int count=0;
  static matrix** all=NULL;
  int i,j,t;
  number z;
  matrix *m;
  switch(mode){
  case ALLOC:
    t=*n;
    m=(matrix*)malloc(sizeof(matrix));
    m->tab =(value_type*)malloc(t * t *sizeof(value_type));
    m->prototype=*p;
    m->size=t;
    z=getZero(*p);
    for(i=0;i<t;i++)
      for(j=0;j<t;j++)
        m->tab[i+j*t]=z.val;
    if(all==NULL)
      all=(matrix**)malloc(4*sizeof(matrix*));
    else
      if(count%4==0)
        all=(matrix**)realloc(all,4*(count+1)*sizeof(matrix*));
    all[count++]=m;
    return m;
  case FREE:
    if(p->type!=MATRIX) return NULL;
    if(p->val.matval==NULL) return NULL;
    for(i=0; i<count && all[i]!=p->val.matval; i++);
    if(i==count) return NULL;
    free(p->val.matval->tab);
    free(p->val.matval);
    *p=error;
    all[i]=all[--count];
    return NULL;
  case FREEALL:
    if(p!=NULL){
      if(p->type!=MATRIX) return NULL;
      if(p->val.matval==NULL) return NULL;
    }
    j=count;
    count=0;
    for(i=0; i<j; i++){
      if(all[i]!=p->val.matval){
        free(all[i]->tab);
        free(all[i]);
      }
      else{
        all[0]=all[i];
        count=1;
      }
    }
    return NULL;
  case INFO:
    *n=count;
    return NULL;    
  }
  /* Inutile ! */
  return NULL;    
}

int howManyMatrices(){
  int n;
  manageMatrix(INFO, &n, NULL);
  return n;
}

void freeMatrix(number* p){
   manageMatrix(FREE, NULL, p);
}
 
void freeAllMatricesExcept(number p){
   manageMatrix(FREEALL, NULL, &p);
}

void freeAllMatrices(void){
   manageMatrix(FREEALL, NULL, NULL);
}
  


number getMatrix(int n, number p){
  number t ={ MATRIX, 
              {0},
              &addMatrix,
              &multMatrix,
              &getUnitMatrix,
              &getZeroMatrix,
              &displayMatrix};
  t.val.matval=manageMatrix(ALLOC, &n, &p);
  return t;
}

number getComponent(number a,int i, int j){
  number c;
  int n;
  if (a.type!=MATRIX)
    return error;
  n=a.val.matval->size;
  if(i<0 || j<0 || i>=n || j>=n)
    return error;
  c=a.val.matval->prototype;
  c.val=a.val.matval->tab[i+n*j];
  return c;
}

number setComponent(number a,int i, int j, number x){
  int n;
  if (a.type!=MATRIX)
    return error;
  n=a.val.matval->size;
  if(i<0 || j<0 || i>=n || j>=n)
    return error;
  if(x.type!=a.val.matval->prototype.type)
    return error;
  a.val.matval->tab[i+n*j]=x.val;
  return x;
}

static number addMatrix(const number a, const number b){
  number c;
  int i,j,n;
  if (a.type!=MATRIX || b.type!=MATRIX)
    return error;
  if (a.val.matval->prototype.type!=b.val.matval->prototype.type ||
      a.val.matval->size!=b.val.matval->size)
    return error;
  n=a.val.matval->size;
  c=getMatrix(n, a.val.matval->prototype);
  for(i=0; i<n; i++)
    for(j=0; j<n; j++)
      setComponent(c,i,j, add(getComponent(a,i,j),getComponent(b,i,j)));
   return c;
}

static number multMatrix(const number a, const number b){
  number c;
  int i,j,k,n;
  if (a.type!=MATRIX || b.type!=MATRIX)
    return error;
  if (a.val.matval->prototype.type!=b.val.matval->prototype.type ||
      a.val.matval->size!=b.val.matval->size)
    return error;
  n=a.val.matval->size;
  c=getMatrix(n, a.val.matval->prototype);
  for(i=0; i<n; i++)
    for(j=0; j<n; j++){
      number t=getZero(a.val.matval->prototype);
      for(k=0; k<n; k++)
        t=add(t, mult(getComponent(a,i,k),getComponent(b,k,j)));
      setComponent(c,i,j,t);
    }
   return c;
}

static number getUnitMatrix(const number a){
  number c;
  int i,n;
  if (a.type!=MATRIX)
    return error;
  n=a.val.matval->size;
  c=getMatrix(n, a.val.matval->prototype);
  number un=getUnit(a.val.matval->prototype);
  display(un);
  for(i=0; i<n; i++)
      setComponent(c,i,i,un);
  return c;
}

static number getZeroMatrix(const number a){
  number c;
  int n;
  if (a.type!=MATRIX)
    return error;
  n=a.val.matval->size;
  c=getMatrix(n, a.val.matval->prototype);
  return c;
}

void displayMatrix(const number x){
  int i,j,n;
  if (x.type!=MATRIX)
    return;
  n=x.val.matval->size;
  for(i=0;i<n;i++){
    printf("[ ");
    for(j=0;j<n;j++){
      display(getComponent(x,i,j));
      putchar(' ');
    }
    printf("]\n");
  }    
}

---