toeplitz_devtools.c

Go to the documentation of this file.

 
#include "toeplitz.h"
#include <cblas.h>
#include <math.h>
#include <stdio.h>
#include <stdlib.h>
#include <time.h>
 
//
// dev tools for cblas and print - fd@apc
 
int stmm_cblas(int n_loc, int m_loc, double *T2_loc, double *V, double *TV2) {
 
    cblas_dgemm(CblasColMajor, CblasNoTrans, CblasNoTrans, n_loc, m_loc, n_loc,
                1, T2_loc, n_loc, (V), n_loc, 1, TV2, n_loc);
 
    return 0;
}
 
 
// Build full Toeplitz matrix needed for cblas computation
int build_full_Toeplitz(int n_loc, double *T_loc, int lambda_loc,
                        double *T2_loc) {
 
    int i, j;
 
    for (j = 0; j < n_loc; j++) { // reset all the matrix to zeros
        for (i = 0; i < n_loc; i++) { T2_loc[j * n_loc + i] = 0; }
    }
 
    for (j = 0; j < n_loc;
         j++) { // Full Toeplitz matrix needed for cblas computation
        for (i = 0; i < lambda_loc; i++) {
            if (j - i >= 0) T2_loc[j * n_loc + j - i] = T_loc[i];
            if (j + i < n_loc) T2_loc[j * n_loc + j + i] = T_loc[i];
        }
    }
 
 
    return 0;
}
 
 
int print_full_Toeplitz(int n_loc, double *T2_loc) {
 
    int i, j;
 
    FILE *file;
    file = stdout;
 
    for (i = 0; i < n_loc; i++) {
        for (j = 0; j < n_loc; j++) {
            fprintf(file, "%.1f\t", T2_loc[i + j * n_loc]);
        }
        fprintf(file, "\n");
    }
 
 
    return 0;
}
 
 
int print_full_matrix(int n_loc, int m_loc, double *Mat) {
 
    int i, j;
 
    FILE *file;
    file = stdout;
 
    for (i = 0; i < n_loc; i++) {
        for (j = 0; j < m_loc; j++) {
            fprintf(file, "%.1f\t", Mat[i + j * n_loc]);
        }
        fprintf(file, "\n");
    }
 
 
    return 0;
}