Utils Class Reference

#include <utils.h>

Public Member Functions
void	meshgrid (const vec &x, const vec &y, mat &X, mat &Y)
void	meshgrid (const vec &x, const vec &y, const vec &z, cube &X, cube &Y, cube &Z)

Static Public Member Functions
static sp_mat	spkron (const sp_mat &A, const sp_mat &B)
static sp_mat	spjoin_rows (const sp_mat &A, const sp_mat &B)
static sp_mat	spjoin_cols (const sp_mat &A, const sp_mat &B)
static vec	spsolve_eigen (const sp_mat &A, const vec &b)

Detailed Description

Definition at line 11 of file utils.h.

Member Function Documentation

meshgrid() [1/2]

void Utils::meshgrid	(	const vec &	x,
		const vec &	y,
		const vec &	z,
		cube &	X,
		cube &	Y,
		cube &	Z )

Definition at line 178 of file utils.cpp.

{
    int m = x.n_elem;
    int n = y.n_elem;
    int o = z.n_elem;
 
    assert(m > 0);
    assert(n > 0);
    assert(o > 0);
 
    // Temporary Holder of sheet of cube
    mat sheet(m, n, fill::zeros);
 
    // Build X
    vec t(n, fill::ones);
 
    X.zeros(m, n, o);
    Y.zeros(m, n, o);
    Z.zeros(m, n, o);
 
    // Sheet that repeats each slice
    for(int ii = 0; ii < m; ++ii) {
        sheet.row(ii) = x(ii) * t.t();
        t.ones();
    }
 
    for(int kk = 0; kk < o; ++kk)
        X.slice(kk) = sheet;
 
    // Y Cube, repeats same sheet as well
    for(int ii = 0; ii < m; ++ii)
        sheet.row(ii) = y.t();
 
    for(int kk = 0; kk < o; ++kk)
        Y.slice(kk) = sheet;
 
    // Z cube goes by slices each with same value
    for(int kk = 0; kk < o; ++kk)
        Z.slice(kk).fill(z(kk));
}

meshgrid() [2/2]

void Utils::meshgrid	(	const vec &	x,
		const vec &	y,
		mat &	X,
		mat &	Y )

Definition at line 155 of file utils.cpp.

{
    int m = x.n_elem;
    int n = y.n_elem;
 
    assert (m > 0);
    assert (n > 0);
 
    // Build X
    vec t(n, fill::ones);
 
    X.zeros(n, m);
    Y.zeros(n, m);
 
    for(int ii = 0; ii < m; ++ii) {
        X.col(ii) = x(ii) * t;
        t.ones();
    }
 
    for(int ii = 0; ii < m; ++ii)
        Y.col(ii) = y;
}

spjoin_cols()

sp_mat Utils::spjoin_cols ( const sp_mat & A, const sp_mat & B )

static

Definition at line 120 of file utils.cpp.

{
    sp_mat::const_iterator itA  = A.begin();
    sp_mat::const_iterator endA = A.end();
    sp_mat::const_iterator itB  = B.begin();
    sp_mat::const_iterator endB = B.end();
    u32 j = 0;
 
    vec a = nonzeros(A);
    vec b = nonzeros(B);
 
    umat locations(2, a.n_elem + b.n_elem);
    vec values(a.n_elem + b.n_elem);
 
    while(itA != endA) {
        locations(0, j) = itA.row();
        locations(1, j) = itA.col();
        values(j) = (*itA);
        ++itA;
        ++j;
    }
 
    while(itB != endB) {
        locations(0, j) = itB.row() + A.n_rows;
        locations(1, j) = itB.col();
        values(j) = (*itB);
        ++itB;
        ++j;
    }
 
    sp_mat result(locations, values, A.n_rows+B.n_rows, A.n_cols, true);
 
    return result;
}

spjoin_rows()

sp_mat Utils::spjoin_rows ( const sp_mat & A, const sp_mat & B )

static

Definition at line 85 of file utils.cpp.

{
    sp_mat::const_iterator itA  = A.begin();
    sp_mat::const_iterator endA = A.end();
    sp_mat::const_iterator itB  = B.begin();
    sp_mat::const_iterator endB = B.end();
    u32 j = 0;
 
    vec a = nonzeros(A);
    vec b = nonzeros(B);
 
    umat locations(2, a.n_elem + b.n_elem);
    vec values(a.n_elem + b.n_elem);
 
    while(itA != endA) {
        locations(0, j) = itA.row();
        locations(1, j) = itA.col();
        values(j) = (*itA);
        ++itA;
        ++j;
    }
 
    while(itB != endB) {
        locations(0, j) = itB.row();
        locations(1, j) = itB.col() + A.n_cols;
        values(j) = (*itB);
        ++itB;
        ++j;
    }
 
    sp_mat result(locations, values, A.n_rows, A.n_cols+B.n_cols, true);
 
    return result;
}

spkron()

sp_mat Utils::spkron ( const sp_mat & A, const sp_mat & B )

static

Definition at line 53 of file utils.cpp.

{
    sp_mat::const_iterator itA  = A.begin();
    sp_mat::const_iterator endA = A.end();
    sp_mat::const_iterator itB  = B.begin();
    sp_mat::const_iterator endB = B.end();
    u32 j = 0;
 
    vec a = nonzeros(A);
    vec b = nonzeros(B);
 
    umat locations(2, a.n_elem*b.n_elem);
    vec values(a.n_elem*b.n_elem);
 
    while(itA != endA) {
        while(itB != endB) {
            locations(0, j) = itA.row()*B.n_rows + itB.row();
            locations(1, j) = itA.col()*B.n_cols + itB.col();
            values(j) = (*itA)*(*itB);
            ++j;
            ++itB;
        }
 
        ++itA;
        itB = B.begin();
    }
 
    sp_mat result(locations, values, A.n_rows*B.n_rows, A.n_cols*B.n_cols, true);
 
    return result;
}

spsolve_eigen()

static vec Utils::spsolve_eigen ( const sp_mat & A, const vec & b )

static

---

The documentation for this class was generated from the following files:

• mole_C++/utils.h
• mole_C++/utils.cpp