src/sparse/matrix.c File Reference

#include <stdio.h>
#include "sparse_int.h"

Include dependency graph for matrix.c:

Go to the source code of this file.

Functions
sm_matrix *	sm_allocate (void)
sm_matrix *	sm_alloc_size (int row, int col)
void	sm_free_space (sm_matrix *A)
sm_matrix *	sm_dup (sm_matrix *A)
void	sm_resize (sm_matrix *A, int row, int col)
sm_element *	sm_insert (sm_matrix *A, int row, int col)
sm_element *	sm_find (sm_matrix *A, int rownum, int colnum)
void	sm_remove (sm_matrix *A, int rownum, int colnum)
void	sm_remove_element (sm_matrix *A, sm_element *p)
void	sm_delrow (sm_matrix *A, int i)
void	sm_delcol (sm_matrix *A, int i)
void	sm_copy_row (sm_matrix *dest, int dest_row, sm_row *prow)
void	sm_copy_col (sm_matrix *dest, int dest_col, sm_col *pcol)
sm_row *	sm_longest_row (sm_matrix *A)
sm_col *	sm_longest_col (sm_matrix *A)
int	sm_num_elements (sm_matrix *A)
int	sm_read (FILE *fp, sm_matrix **A)
int	sm_read_compressed (FILE *fp, sm_matrix **A)
void	sm_write (FILE *fp, sm_matrix *A)
void	sm_print (FILE *fp, sm_matrix *A)
void	sm_dump (sm_matrix *A, char *s, int max)
void	sm_cleanup (void)

---

Function Documentation

sm_matrix* sm_alloc_size	(	int	row,
		int	col
	)

Definition at line 41 of file matrix.c.

{
    register sm_matrix *A;

    A = sm_alloc();
    sm_resize(A, row, col);
    return A;
}

sm_matrix* sm_allocate

(

void

)

Definition at line 24 of file matrix.c.

{
    register sm_matrix *A;

    A = ALLOC(sm_matrix, 1);
    A->rows = NIL(sm_row *);
    A->cols = NIL(sm_col *);
    A->nrows = A->ncols = 0;
    A->rows_size = A->cols_size = 0;
    A->first_row = A->last_row = NIL(sm_row);
    A->first_col = A->last_col = NIL(sm_col);
    A->user_word = NIL(char);           /* for our user ... */
    return A;
}

void sm_cleanup

(

void

)

Definition at line 508 of file matrix.c.

{
#ifdef FAST_AND_LOOSE
    register sm_element *p, *pnext;
    register sm_row *prow, *pnextrow;
    register sm_col *pcol, *pnextcol;

    for(p = sm_element_freelist; p != 0; p = pnext) {
        pnext = p->next_col;
        FREE(p);
    }
    sm_element_freelist = 0;

    for(prow = sm_row_freelist; prow != 0; prow = pnextrow) {
        pnextrow = prow->next_row;
        FREE(prow);
    }
    sm_row_freelist = 0;

    for(pcol = sm_col_freelist; pcol != 0; pcol = pnextcol) {
        pnextcol = pcol->next_col;
        FREE(pcol);
    }
    sm_col_freelist = 0;
#endif
}

void sm_copy_col	(	sm_matrix *	dest,
		int	dest_col,
		sm_col *	pcol
	)

Definition at line 328 of file matrix.c.

{
    register sm_element *p;

    for(p = pcol->first_row; p != 0; p = p->next_row) {
        (void) sm_insert(dest, dest_col, p->row_num);
    }
}

void sm_copy_row	(	sm_matrix *	dest,
		int	dest_row,
		sm_row *	prow
	)

Definition at line 317 of file matrix.c.

{
    register sm_element *p;

    for(p = prow->first_col; p != 0; p = p->next_col) {
        (void) sm_insert(dest, dest_row, p->col_num);
    }
}

void sm_delcol	(	sm_matrix *	A,
		int	i
	)

Definition at line 285 of file matrix.c.

{
    register sm_element *p, *pnext;
    sm_row *prow;
    sm_col *pcol;

    pcol = sm_get_col(A, i);
    if (pcol != NIL(sm_col)) {
        /* walk down the column */
        for(p = pcol->first_row; p != 0; p = pnext) {
            pnext = p->next_row;

            /* unlink the element from the row (and delete it) */
            prow = sm_get_row(A, p->row_num);
            sm_row_remove_element(prow, p);

            /* discard the row if it is now empty */
            if (prow->first_col == NIL(sm_element)) {
                sm_delrow(A, prow->row_num);
            }
        }

        /* discard the column -- we already threw away the elements */ 
        A->cols[i] = NIL(sm_col);
        dll_unlink(pcol, A->first_col, A->last_col, 
                            next_col, prev_col, A->ncols);
        pcol->first_row = pcol->last_row = NIL(sm_element);
        sm_col_free(pcol);
    }
}

void sm_delrow	(	sm_matrix *	A,
		int	i
	)

Definition at line 253 of file matrix.c.

{
    register sm_element *p, *pnext;
    sm_col *pcol;
    sm_row *prow;

    prow = sm_get_row(A, i);
    if (prow != NIL(sm_row)) {
        /* walk across the row */
        for(p = prow->first_col; p != 0; p = pnext) {
            pnext = p->next_col;

            /* unlink the item from the column (and delete it) */
            pcol = sm_get_col(A, p->col_num);
            sm_col_remove_element(pcol, p);

            /* discard the column if it is now empty */
            if (pcol->first_row == NIL(sm_element)) {
                sm_delcol(A, pcol->col_num);
            }
        }

        /* discard the row -- we already threw away the elements */ 
        A->rows[i] = NIL(sm_row);
        dll_unlink(prow, A->first_row, A->last_row, 
                                next_row, prev_row, A->nrows);
        prow->first_col = prow->last_col = NIL(sm_element);
        sm_row_free(prow);
    }
}

void sm_dump	(	sm_matrix *	A,
		char *	s,
		int	max
	)

Definition at line 497 of file matrix.c.

{
    FILE *fp = stdout;

    (void) fprintf(fp, "%s %d rows by %d cols\n", s, A->nrows, A->ncols);
    if (A->nrows < max) {
        sm_print(fp, A);
    }
}

sm_matrix* sm_dup

(

sm_matrix *

A

)

Definition at line 95 of file matrix.c.

{
    register sm_row *prow;
    register sm_element *p;
    register sm_matrix *B;

    B = sm_alloc();
    if (A->last_row != 0) {
        sm_resize(B, A->last_row->row_num, A->last_col->col_num);
        for(prow = A->first_row; prow != 0; prow = prow->next_row) {
            for(p = prow->first_col; p != 0; p = p->next_col) {
                (void) sm_insert(B, p->row_num, p->col_num);
            }
        }
    }
    return B;
}

sm_element* sm_find	(	sm_matrix *	A,
		int	rownum,
		int	colnum
	)

Definition at line 191 of file matrix.c.

{
    sm_row *prow;
    sm_col *pcol;

    prow = sm_get_row(A, rownum);
    if (prow == NIL(sm_row)) {
        return NIL(sm_element);
    } else {
        pcol = sm_get_col(A, colnum);
        if (pcol == NIL(sm_col)) {
            return NIL(sm_element);
        }
        if (prow->length < pcol->length) {
            return sm_row_find(prow, colnum);
        } else {
            return sm_col_find(pcol, rownum);
        }
    }
}

void sm_free_space

(

sm_matrix *

A

)

Definition at line 52 of file matrix.c.

{
#ifdef FAST_AND_LOOSE
    register sm_row *prow;

    if (A->first_row != 0) {
        for(prow = A->first_row; prow != 0; prow = prow->next_row) {
            /* add the elements to the free list of elements */
            prow->last_col->next_col = sm_element_freelist;
            sm_element_freelist = prow->first_col;
        }

        /* Add the linked list of rows to the row-free-list */
        A->last_row->next_row = sm_row_freelist;
        sm_row_freelist = A->first_row;

        /* Add the linked list of cols to the col-free-list */
        A->last_col->next_col = sm_col_freelist;
        sm_col_freelist = A->first_col;
    }
#else
    register sm_row *prow, *pnext_row;
    register sm_col *pcol, *pnext_col;

    for(prow = A->first_row; prow != 0; prow = pnext_row) {
        pnext_row = prow->next_row;
        sm_row_free(prow);
    }
    for(pcol = A->first_col; pcol != 0; pcol = pnext_col) {
        pnext_col = pcol->next_col;
        pcol->first_row = pcol->last_row = NIL(sm_element);
        sm_col_free(pcol);
    }
#endif

    /* Free the arrays to map row/col numbers into pointers */
    FREE(A->rows);
    FREE(A->cols);
    FREE(A);
}

sm_element* sm_insert	(	sm_matrix *	A,
		int	row,
		int	col
	)

Definition at line 143 of file matrix.c.

{
    register sm_row *prow;
    register sm_col *pcol;
    register sm_element *element;
    sm_element *save_element;

    if (row >= A->rows_size || col >= A->cols_size) {
        sm_resize(A, row, col);
    }

    prow = A->rows[row];
    if (prow == NIL(sm_row)) {
        prow = A->rows[row] = sm_row_alloc();
        prow->row_num = row;
        sorted_insert(sm_row, A->first_row, A->last_row, A->nrows, 
                        next_row, prev_row, row_num, row, prow);
    }

    pcol = A->cols[col];
    if (pcol == NIL(sm_col)) {
        pcol = A->cols[col] = sm_col_alloc();
        pcol->col_num = col;
        sorted_insert(sm_col, A->first_col, A->last_col, A->ncols, 
                        next_col, prev_col, col_num, col, pcol);
    }

    /* get a new item, save its address */
    sm_element_alloc(element);
    save_element = element;

    /* insert it into the row list */
    sorted_insert(sm_element, prow->first_col, prow->last_col, 
                prow->length, next_col, prev_col, col_num, col, element);

    /* if it was used, also insert it into the column list */
    if (element == save_element) {
        sorted_insert(sm_element, pcol->first_row, pcol->last_row, 
                pcol->length, next_row, prev_row, row_num, row, element);
    } else {
        /* otherwise, it was already in matrix -- free element we allocated */
        sm_element_free(save_element);
    }
    return element;
}

sm_col* sm_longest_col

(

sm_matrix *

A

)

Definition at line 357 of file matrix.c.

{
    register sm_col *large_col, *pcol;
    register int max_length;

    max_length = 0;
    large_col = NIL(sm_col);
    for(pcol = A->first_col; pcol != 0; pcol = pcol->next_col) {
        if (pcol->length > max_length) {
            max_length = pcol->length;
            large_col = pcol;
        }
    }
    return large_col;
}

sm_row* sm_longest_row

(

sm_matrix *

A

)

Definition at line 339 of file matrix.c.

{
    register sm_row *large_row, *prow;
    register int max_length;

    max_length = 0;
    large_row = NIL(sm_row);
    for(prow = A->first_row; prow != 0; prow = prow->next_row) {
        if (prow->length > max_length) {
            max_length = prow->length;
            large_row = prow;
        }
    }
    return large_row;
}

int sm_num_elements

(

sm_matrix *

A

)

Definition at line 374 of file matrix.c.

{
    register sm_row *prow;
    register int num;

    num = 0;
    sm_foreach_row(A, prow) {
        num += prow->length;
    }
    return num;
}

void sm_print	(	FILE *	fp,
		sm_matrix *	A
	)

Definition at line 450 of file matrix.c.

{
    register sm_row *prow;
    register sm_col *pcol;
    int c;

    if (A->last_col->col_num >= 100) {
        (void) fprintf(fp, "    ");
        for(pcol = A->first_col; pcol != 0; pcol = pcol->next_col) {
            (void) fprintf(fp, "%d", (pcol->col_num / 100)%10);
        }
        putc('\n', fp);
    }

    if (A->last_col->col_num >= 10) {
        (void) fprintf(fp, "    ");
        for(pcol = A->first_col; pcol != 0; pcol = pcol->next_col) {
            (void) fprintf(fp, "%d", (pcol->col_num / 10)%10);
        }
        putc('\n', fp);
    }

    (void) fprintf(fp, "    ");
    for(pcol = A->first_col; pcol != 0; pcol = pcol->next_col) {
        (void) fprintf(fp, "%d", pcol->col_num % 10);
    }
    putc('\n', fp);

    (void) fprintf(fp, "    ");
    for(pcol = A->first_col; pcol != 0; pcol = pcol->next_col) {
        (void) fprintf(fp, "-");
    }
    putc('\n', fp);

    for(prow = A->first_row; prow != 0; prow = prow->next_row) {
        (void) fprintf(fp, "%3d:", prow->row_num);

        for(pcol = A->first_col; pcol != 0; pcol = pcol->next_col) {
            c = sm_row_find(prow, pcol->col_num) ? '1' : '.';
            putc(c, fp);
        }
        putc('\n', fp);
    }
}

int sm_read	(	FILE *	fp,
		sm_matrix **	A
	)

Definition at line 387 of file matrix.c.

{
    int i, j, err;

    *A = sm_alloc();
    while (! feof(fp)) {
        err = fscanf(fp, "%d %d", &i, &j);
        if (err == EOF) {
            return 1;
        } else if (err != 2) {
            return 0;
        }
        (void) sm_insert(*A, i, j);
    }
    return 1;
}

int sm_read_compressed	(	FILE *	fp,
		sm_matrix **	A
	)

Definition at line 406 of file matrix.c.

{
    int i, j, k, nrows, ncols;
    unsigned long x;

    *A = sm_alloc();
    if (fscanf(fp, "%d %d", &nrows, &ncols) != 2) {
        return 0;
    }
    sm_resize(*A, nrows, ncols);

    for(i = 0; i < nrows; i++) {
        if (fscanf(fp, "%lx", &x) != 1) {
            return 0;
        }
        for(j = 0; j < ncols; j += 32) {
            if (fscanf(fp, "%lx", &x) != 1) {
                return 0;
            }
            for(k = j; x != 0; x >>= 1, k++) {
                if (x & 1) {
                    (void) sm_insert(*A, i, k);
                }
            }
        }
    }
    return 1;
}

void sm_remove	(	sm_matrix *	A,
		int	rownum,
		int	colnum
	)

Definition at line 214 of file matrix.c.

{
    sm_remove_element(A, sm_find(A, rownum, colnum));
}

void sm_remove_element	(	sm_matrix *	A,
		sm_element *	p
	)

Definition at line 222 of file matrix.c.

{
    register sm_row *prow;
    register sm_col *pcol;

    if (p == 0) return;

    /* Unlink the element from its row */
    prow = sm_get_row(A, p->row_num);
    dll_unlink(p, prow->first_col, prow->last_col, 
                        next_col, prev_col, prow->length);

    /* if no more elements in the row, discard the row header */
    if (prow->first_col == NIL(sm_element)) {
        sm_delrow(A, p->row_num);
    }

    /* Unlink the element from its column */
    pcol = sm_get_col(A, p->col_num);
    dll_unlink(p, pcol->first_row, pcol->last_row, 
                        next_row, prev_row, pcol->length);

    /* if no more elements in the column, discard the column header */
    if (pcol->first_row == NIL(sm_element)) {
        sm_delcol(A, p->col_num);
    }

    sm_element_free(p);
}

void sm_resize	(	sm_matrix *	A,
		int	row,
		int	col
	)

Definition at line 115 of file matrix.c.

{
    register int i, new_size;

    if (row >= A->rows_size) {
        new_size = MAX(A->rows_size*2, row+1);
        A->rows = REALLOC(sm_row *, A->rows, new_size);
        for(i = A->rows_size; i < new_size; i++) {
            A->rows[i] = NIL(sm_row);
        }
        A->rows_size = new_size;
    }

    if (col >= A->cols_size) {
        new_size = MAX(A->cols_size*2, col+1);
        A->cols = REALLOC(sm_col *, A->cols, new_size);
        for(i = A->cols_size; i < new_size; i++) {
            A->cols[i] = NIL(sm_col);
        }
        A->cols_size = new_size;
    }
}

void sm_write	(	FILE *	fp,
		sm_matrix *	A
	)

Definition at line 437 of file matrix.c.

{
    register sm_row *prow;
    register sm_element *p;

    for(prow = A->first_row; prow != 0; prow = prow->next_row) {
        for(p = prow->first_col; p != 0; p = p->next_col) {
            (void) fprintf(fp, "%d %d\n", p->row_num, p->col_num);
        }
    }
}