ConjugateGradientSolver.h

Go to the documentation of this file.

// 
//  This file is part of the FFEA simulation package
//  
//  Copyright (c) by the Theory and Development FFEA teams,
//  as they appear in the README.md file. 
// 
//  FFEA is free software: you can redistribute it and/or modify
//  it under the terms of the GNU General Public License as published by
//  the Free Software Foundation, either version 3 of the License, or
//  (at your option) any later version.
// 
//  FFEA is distributed in the hope that it will be useful,
//  but WITHOUT ANY WARRANTY; without even the implied warranty of
//  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
//  GNU General Public License for more details.
// 
//  You should have received a copy of the GNU General Public License
//  along with FFEA.  If not, see <http://www.gnu.org/licenses/>.
// 
//  To help us fund FFEA development, we humbly ask that you cite 
//  the research papers on the package.
//

#ifndef CONJUGATEGRADIENTSOLVER_HPP_INCLUDED
#define CONJUGATEGRADIENTSOLVER_HPP_INCLUDED


#include <stdio.h>

#include "FFEA_return_codes.h"
#include "mat_vec_types.h"
#include "mesh_node.h"
#include "tetra_element_linear.h"
#include "SimulationParams.h"
#include "Solver.h"
#include "ConjugateGradientSolver.h"
#include "SparseMatrixTypes.h"

class ConjugateGradientSolver : public Solver {
public:
    ConjugateGradientSolver();

    ~ConjugateGradientSolver();

    int init(int num_nodes, int num_elements, mesh_node *node, tetra_element_linear *elem, SimulationParams *params, int num_pinned_nodes, int *pinned_nodes_list, set<int> bsite_pinned_node_list);

    int solve(vector3 *x);

    void apply_matrix(scalar *in, scalar *result);

private:

    scalar epsilon2;

    int i_max;

    int num_rows;

    sparse_entry *entry;

    int *key;

    scalar *preconditioner;

    vector3 *d, *r, *q, *s, *f;

    /* */
    scalar conjugate_gradient_residual_assume_x_zero(vector3 *b);

    /* */
    scalar parallel_sparse_matrix_apply();

    void parallel_vector_add_self(vector3 *v1, scalar a, vector3 *v2, int vec_size);

    void parallel_vector_add(vector3 *v1, scalar a, vector3 *v2, int vec_size);

    /* */
    scalar parallel_apply_preconditioner();

    /* */
    scalar residual2();
};

#endif