SparseSubstitutionSolver.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 SPARSESUBSTITUTIONSOLVER_H_INCLUDED
#define SPARSESUBSTITUTIONSOLVER_H_INCLUDED

#include <stdio.h>
#include <stdlib.h>
#include <math.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 "SparseSubstitutionSolver.h"

#define INDEX(I, J) ((I) * num_rows + (J))

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

    ~SparseSubstitutionSolver();

    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:

    int num_rows;


    int *L_key, *U_key;

    int total_entries_in_U;

    scalar *inverse_diag;


    scalar *L, *U;
};

#endif