src/linalg.h

Go to the documentation of this file.

/***************************************************************************
 *   Copyright (C) 2010 by Matthias Ihrke                                  *
 *   ihrke@nld.ds.mpg.de
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU General Public License as published by  *
 *   the Free Software Foundation; either version 2 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 *   This program is distributed in the hope that it will be useful,       *
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
 *   GNU General Public License for more details.                          *
 *                                                                         *
 *   You should have received a copy of the GNU General Public License     *
 *   aint with this program; if not, write to the                         *
 *   Free Software Foundation, Inc.,                                       *
 *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
 ***************************************************************************/

#ifndef LINALG_H
# define LINALG_H
#include "definitions.h"
#include "array.h"
#include <gsl/gsl_matrix.h>

#define matrix_CHECK( flag, m )                                                 \
  if( (m)==NULL ){                                                                  \
     errprintf("got NULL as matrix\n");                                         \
     flag=FALSE;                                                                        \
  } else if(!( (m)->ndim==2 && (m)->dtype==DOUBLE )){                       \
     char *dts="";                                                                      \
     array_DTYPESTRING( dts, m->dtype );                                        \
     errprintf("not a matrix, ndim=%i, dtype=%s\n", m->ndim, dts ); \
     flag=FALSE;                                                                        \
  } else { flag=TRUE; } 

#define matrix_CHECKSQR( flag, m )                                                  \
  if(!( (m)->ndim==2 && (m)->dtype==DOUBLE )){                                  \
     char *dts="";                                                                          \
     array_DTYPESTRING( dts, (m)->dtype );                                          \
     errprintf("not a matrix, ndim=%i, dtype=%s\n", (m)->ndim, dts );       \
     flag=FALSE;                                                                            \
  } else if( (m)->size[0]!=(m)->size[1] ) {                                     \
     errprintf("Matrix is not square, got (%i,%i)\n",(m)->size[0],(m)->size[1]); \
     flag=FALSE;                                                                            \
  } else { flag=TRUE; }                                                             

#define mat_IDX( m, i, j )                              \
  array_INDEX2( m, double, i, j )

#define vec_IDX( v, i )                                 \
  array_INDEX1( v, double, i )

#define vector_CHECK( flag, v )                                                     \
  if( (v)==NULL ){                                                                      \
     errprintf("got NULL as vector\n");                                             \
     flag=FALSE;                                                                            \
  } else if(!( (v)->ndim==1 && (v)->dtype==DOUBLE ) ){                      \
     char *dts="";                                                                          \
     array_DTYPESTRING( dts, v->dtype );                                            \
     errprintf("not a vector, ndim=%i, dtype=%s\n", v->ndim, dts );     \
     flag=FALSE;                                                                            \
  } else { flag=TRUE; }                                                             


#ifdef __cplusplus
extern "C" {
#endif

  /* -------------- FUNCTIONS ---------------- */
  gsl_matrix* matrix_to_gsl( Array *in, bool alloc );

  Array* matrix_get_col( Array *m, int col );
  Array* matrix_get_row( Array *m, int row, bool alloc );
  
  Array* matrix_mult( const Array *m1, const Array *m2 );
  Array* matrix_transpose( Array *m, bool alloc );

  Array* matrix_mean( Array *a, int dim );
  Array* matrix_pca( Array *X, Array **var, bool alloc );


#ifdef __cplusplus
}
#endif

#endif /* LINALG_H */