1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206
|
/**************************************************************************
**
** Copyright (C) 1993 David E. Steward & Zbigniew Leyk, all rights reserved.
**
** Meschach Library
**
** This Meschach Library is provided "as is" without any express
** or implied warranty of any kind with respect to this software.
** In particular the authors shall not be liable for any direct,
** indirect, special, incidental or consequential damages arising
** in any way from use of the software.
**
** Everyone is granted permission to copy, modify and redistribute this
** Meschach Library, provided:
** 1. All copies contain this copyright notice.
** 2. All modified copies shall carry a notice stating who
** made the last modification and the date of such modification.
** 3. No charge is made for this software or works derived from it.
** This clause shall not be construed as constraining other software
** distributed on the same medium as this software, nor is a
** distribution fee considered a charge.
**
***************************************************************************/
/* 1.2 submat.c 11/25/87 */
#include <stdio.h>
#include "matrix.h"
static char rcsid[] = "$Id: submat.c,v 1.2 1994/01/13 05:28:12 des Exp $";
/* get_col -- gets a specified column of a matrix and retruns it as a vector */
#ifndef ANSI_C
VEC *get_col(mat,col,vec)
unsigned int col;
MAT *mat;
VEC *vec;
#else
VEC *get_col(const MAT *mat, unsigned int col, VEC *vec)
#endif
{
unsigned int i;
if ( mat==(MAT *)NULL )
error(E_NULL,"get_col");
if ( col >= mat->n )
error(E_RANGE,"get_col");
if ( vec==(VEC *)NULL || vec->dim<mat->m )
vec = v_resize(vec,mat->m);
for ( i=0; i<mat->m; i++ )
vec->ve[i] = mat->me[i][col];
return (vec);
}
/* get_row -- gets a specified row of a matrix and retruns it as a vector */
#ifndef ANSI_C
VEC *get_row(mat,row,vec)
unsigned int row;
MAT *mat;
VEC *vec;
#else
VEC *get_row(const MAT *mat, unsigned int row, VEC *vec)
#endif
{
unsigned int i;
if ( mat==(MAT *)NULL )
error(E_NULL,"get_row");
if ( row >= mat->m )
error(E_RANGE,"get_row");
if ( vec==(VEC *)NULL || vec->dim<mat->n )
vec = v_resize(vec,mat->n);
for ( i=0; i<mat->n; i++ )
vec->ve[i] = mat->me[row][i];
return (vec);
}
/* _set_col -- sets column of matrix to values given in vec (in situ)
-- that is, mat(i0:lim,col) <- vec(i0:lim) */
#ifndef ANSI_C
MAT *_set_col(mat,col,vec,i0)
MAT *mat;
VEC *vec;
unsigned int col,i0;
#else
MAT *_set_col(MAT *mat, unsigned int col, const VEC *vec, unsigned int i0)
#endif
{
unsigned int i,lim;
if ( mat==(MAT *)NULL || vec==(VEC *)NULL )
error(E_NULL,"_set_col");
if ( col >= mat->n )
error(E_RANGE,"_set_col");
lim = min(mat->m,vec->dim);
for ( i=i0; i<lim; i++ )
mat->me[i][col] = vec->ve[i];
return (mat);
}
/* _set_row -- sets row of matrix to values given in vec (in situ) */
#ifndef ANSI_C
MAT *_set_row(mat,row,vec,j0)
MAT *mat;
VEC *vec;
unsigned int row,j0;
#else
MAT *_set_row(MAT *mat, unsigned int row, const VEC *vec, unsigned int j0)
#endif
{
unsigned int j,lim;
if ( mat==(MAT *)NULL || vec==(VEC *)NULL )
error(E_NULL,"_set_row");
if ( row >= mat->m )
error(E_RANGE,"_set_row");
lim = min(mat->n,vec->dim);
for ( j=j0; j<lim; j++ )
mat->me[row][j] = vec->ve[j];
return (mat);
}
/* sub_mat -- returns sub-matrix of old which is formed by the rectangle
from (row1,col1) to (row2,col2)
-- Note: storage is shared so that altering the "new"
matrix will alter the "old" matrix */
#ifndef ANSI_C
MAT *sub_mat(old,row1,col1,row2,col2,new)
MAT *old,*new;
unsigned int row1,col1,row2,col2;
#else
MAT *sub_mat(const MAT *old,
unsigned int row1, unsigned int col1,
unsigned int row2, unsigned int col2,
MAT *new)
#endif
{
unsigned int i;
if ( old==(MAT *)NULL )
error(E_NULL,"sub_mat");
if ( row1 > row2 || col1 > col2 || row2 >= old->m || col2 >= old->n )
error(E_RANGE,"sub_mat");
if ( new==(MAT *)NULL || new->m < row2-row1+1 )
{
new = NEW(MAT);
new->me = NEW_A(row2-row1+1,Real *);
if ( new==(MAT *)NULL || new->me==(Real **)NULL )
error(E_MEM,"sub_mat");
else if (mem_info_is_on()) {
mem_bytes(TYPE_MAT,0,sizeof(MAT)+
(row2-row1+1)*sizeof(Real *));
}
}
new->m = row2-row1+1;
new->n = col2-col1+1;
new->base = (Real *)NULL;
for ( i=0; i < new->m; i++ )
new->me[i] = (old->me[i+row1]) + col1;
return (new);
}
/* sub_vec -- returns sub-vector which is formed by the elements i1 to i2
-- as for sub_mat, storage is shared */
#ifndef ANSI_C
VEC *sub_vec(old,i1,i2,new)
VEC *old, *new;
int i1, i2;
#else
VEC *sub_vec(const VEC *old, int i1, int i2, VEC *new)
#endif
{
if ( old == (VEC *)NULL )
error(E_NULL,"sub_vec");
if ( i1 > i2 || old->dim < i2 )
error(E_RANGE,"sub_vec");
if ( new == (VEC *)NULL )
new = NEW(VEC);
if ( new == (VEC *)NULL )
error(E_MEM,"sub_vec");
else if (mem_info_is_on()) {
mem_bytes(TYPE_VEC,0,sizeof(VEC));
}
new->dim = i2 - i1 + 1;
new->ve = &(old->ve[i1]);
return new;
}
|