int read_matrix_and_rhs(char *file_name, int &n,
std::map<unsigned int, MatrixEntry> &mat, std::map<unsigned int, scalar> &rhs) {
#ifndef H3D_COMPLEX
FILE *file = fopen(file_name, "r");
if (file == NULL) return ERR_FAILURE;
enum EState {
STATE_N,
STATE_MATRIX,
STATE_RHS,
} state = STATE_N;
double buffer[3];
char row[MAX_ROW_LEN];
while (fgets(row, MAX_ROW_LEN, file) != NULL) {
switch (state) {
case STATE_N:
if (read_n_nums(row, 1, buffer)) {
n = (int) buffer[0];
state = STATE_MATRIX;
}
break;
case STATE_MATRIX:
if (read_n_nums(row, 3, buffer)) {
mat[mat.size()] = (MatrixEntry((int) buffer[0], (int) buffer[1], buffer[2]));
}
else
state = STATE_RHS;
break;
case STATE_RHS:
if (read_n_nums(row, 2, buffer)) {
rhs[(int) buffer[0]] = buffer[1];
}
break;
}
}
fclose(file);
#else
n = 3;
mat[mat.size()] = MatrixEntry(0, 0, scalar(1, 2));
mat[mat.size()] = MatrixEntry(1, 1, scalar(1, 4));
mat[mat.size()] = MatrixEntry(2, 2, scalar(1, 6));
rhs[0] = scalar(2, 1);
rhs[1] = scalar(4, 1);
rhs[2] = scalar(6, 2);
#endif
return ERR_SUCCESS;
}
int read_matrix_and_rhs(char *file_name, int &n,
std::map<unsigned int, MatrixEntry> &mat, std::map<unsigned int, scalar> &rhs, bool &cplx_2_real)
{
FILE *file = fopen(file_name, "r");
if (file == NULL) return ERR_FAILURE;
enum EState {
STATE_N,
STATE_MATRIX,
STATE_RHS,
} state = STATE_N;
// Variables needed to turn complex matrix into real.
int k = 0;
int l = 0;
int param = 0;
bool fill_upper = true;
bool fill_lower = false;
int* imgn_i = NULL;
int* imgn_j = NULL;
double* imgn_value = NULL;
double* rhs_buffer = NULL;
double buffer[4];
char row[MAX_ROW_LEN];
while (fgets(row, MAX_ROW_LEN, file) != NULL) {
switch (state) {
case STATE_N:
if (read_n_nums(row, 1, buffer)) {
if (cplx_2_real) {
if (fill_upper==true) {
n = 2*((int) buffer[0]);
// Allocate and initialize temporary arrays
// we need for turning complex matrix to real.
imgn_i = new int[n];
imgn_j = new int[n];
imgn_value = new double[n];
rhs_buffer = new double[n];
for (int i = 0; i < n; i++) {
imgn_i[i] = 0;
imgn_j[i] =0;
imgn_value[i] = 0.0;
rhs_buffer[i] = 0.0;
}
}
else
printf("\n");
}
else{
n = (int) buffer[0];
}
state = STATE_MATRIX;
}
break;
#ifndef HERMES_COMMON_COMPLEX
case STATE_MATRIX:
if (cplx_2_real){
if (read_n_nums(row, 4, buffer)) {
if (fill_upper == true) { //case FILL_UPPER:
if (buffer[0] == k)
{
mat[mat.size()] = (MatrixEntry((int) buffer[0], (int) buffer[1], (scalar) buffer[2]));
imgn_value[l] = (scalar) (-1)*buffer[3];
imgn_i[l] = (int) buffer[0];
imgn_j[l] = (int) buffer[1] + int (n/2);
l=l+1;
}
else // we have read an element that belongs to next row. Now what?
{
// OK, first finish the previous row with imaginaries.
for (int i=0; i < l; i++)
{
mat[mat.size()] = (MatrixEntry(imgn_i[i], imgn_j[i], imgn_value[i]));
}
// Now begin the next row with real
mat[mat.size()] = (MatrixEntry((int) buffer[0], (int) buffer[1], buffer[2]));
// And remember imaginary part from this row
imgn_value[0] = (scalar) (-1)*buffer[3];
imgn_i[0] = (int) buffer[0];
imgn_j[0] = (int) buffer[1] + n/2;
l = 1;
while(k < buffer[0]) // Fast forward k, we need it for next read.
k++;
}
} // break; // case FILL_UPPER break.
if (fill_lower == true){ //case FILL_LOWER: We fill in lower half of the matrix, by different rules
if (buffer[0] == k)
{
// Create next matrix entry
mat[mat.size()] = (MatrixEntry((int) buffer[0] + n/2, (int) buffer[1], (scalar) buffer[3]));
// Save imaginary part for later
imgn_value[l] = (scalar) buffer[2];
imgn_i[l] = (int) buffer[0] + n/2;
imgn_j[l] = (int) buffer[1] + n/2;
l=l+1;
}
else // we have read an element that belongs to next row. Now what?
{
//OK, first finish the previous row with imaginaries.
for (int i=0; i < l; i++)
{
mat[mat.size()] = (MatrixEntry(imgn_i[i], imgn_j[i], imgn_value[i]));
}
// Now begin the next row with real
mat[mat.size()] = (MatrixEntry((int) buffer[0] + n/2, (int) buffer[1], buffer[3]));
// Save imaginary part for later
imgn_value[0] = (scalar) buffer[2];
imgn_i[0] = (int) buffer[0] + n/2;
imgn_j[0] = (int) buffer[1] + n/2;
l = 1;
// Fast forward k, we need it for next read.
while(k < buffer[0])
k++;
}
} // case FILL_LOWER break.
} // if (read_n_nums) block end.
else {
// We have reached the line in the input file,
// where the first rhs vector entries are defined.
// But before we go to STATE_RHS,
// let's finish filling the lower half of the new real matrix.
if (param == 0){
//OK, first finish the previous row with imaginaries.
for (int i=0; i < l; i++)
{
mat[mat.size()] = (MatrixEntry(imgn_i[i], imgn_j[i], imgn_value[i]));
}
k = 0;
l = 0;
//Go to the top of the file.
rewind (file);
// Anticipate the length of the first line.
state = STATE_N;
// Now we will fill lower half of the matrix.
fill_lower = true;
fill_upper = false;
// Change parameter that brought us to this place.
param = 1;
}
else {
//OK, first finish the previous row with imaginaries.
for (int i=0; i < l; i++)
{
mat[mat.size()] = (MatrixEntry(imgn_i[i], imgn_j[i], imgn_value[i]));
}
l = 0;
state = STATE_RHS;
}
}
}else{ // if cplx_2_real is false.
if (read_n_nums(row, 3, buffer))
mat[mat.size()] = (MatrixEntry((int) buffer[0], (int) buffer[1], buffer[2]));
else
state = STATE_RHS;
}
break; //case STATE_MATRIX break.
case STATE_RHS:
if (cplx_2_real) {
if (read_n_nums(row, 3, buffer)) {
if (buffer[0] != (int) n/2-1) // Then this is not the last line in the input file
{
rhs[((int) buffer[0])] = (scalar) buffer[1];
rhs_buffer[l] = (scalar) buffer[2];
l=l+1;
}
else // This is last line in the file.
{
// First read last line entry
rhs[((int) buffer[0])] = (scalar) buffer[1];
rhs_buffer[l] = (scalar) buffer[2];
l=l+1;
// Take imaginary parts you saved,
// and fill the rest of the rhs vector.
for (int i=0; i < l; i++)
{
rhs[rhs.size()] = rhs_buffer[i];
}
}
}
}
else { // if cplx_2_real is false.
if (read_n_nums(row, 2, buffer))
rhs[(int) buffer[0]] = (scalar) buffer[1];
}
break;
}
}
#else
case STATE_MATRIX:
if (read_n_nums(row, 4, buffer)) {
std::complex<double> cmplx_buffer(buffer[2], buffer[3]);
mat[mat.size()] = (MatrixEntry((int) buffer[0], (int) buffer[1], (scalar) cmplx_buffer));
}
else
state = STATE_RHS;
break;
case STATE_RHS:
if (read_n_nums(row, 3, buffer)) {
std::complex<double> cmplx_buffer(buffer[1], buffer[2]);
rhs[(int) buffer[0]] = (scalar) cmplx_buffer;
}
break;
}
