-
Notifications
You must be signed in to change notification settings - Fork 0
/
JacobiP.hpp
85 lines (82 loc) · 1.82 KB
/
JacobiP.hpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
#ifndef JACOBI_P_HPP
#define JACOBI_P_HPP
//----------------------------------
// Librerias Necesarias
#include <iostream>
#include <string>
//----------------------------------
template<class P>
class JacobiP
{
private:
double *D; // Contiene la diagonal de la matriz
int n; // Numero de elementos de la diagonal
void reserveMemory(); // Reserva memoria para la diagonal
std::string namep;
// Libera memoria en la diagonal
void freeMemory()
{
if(D == nullptr)
{
delete [] D;
}
}
public:
// Constructor vacio
JacobiP():namep("Jacobi"){};
// Libera la memoria utilizada
~JacobiP(){freeMemory();};
// Devuelve el nombre del precondicionador
inline std::string name(){return namep;};
// Calcula el precondicionador
void calculate(P &Mtx);
// Resuelve el sistema de ecuaciones
void solve(Vector &z, Vector const &r);
// Muestra en pantalla el contenido de la diagonal
void impDiag();
};
// -------------------------------------------------
// Reservamos memoria
template <class P>
void JacobiP<P> :: reserveMemory()
{
// Seleccionamos espacio para la diagonal
D = new double[n];
// Verificamos que la memoria fuera asignada
if (D == nullptr)
{
std::cout << "ERROR: MEMORIA NO ASIGNADA: reserveMemory: JacobiP" << std::endl;
exit(0);
}
}
// Calculamos el precondicionador
template<class P>
void JacobiP<P> :: calculate(P &Mtx)
{
n = Mtx.n;
reserveMemory();
// Hacemos una busqueda sobre el arreglo
for (int i = 0; i < n; ++i)
{
D[i] = Mtx.search(i,i);
}
}
// Calculamos la solucion del sistema
template<class P>
void JacobiP<P> ::solve(Vector &z, Vector const &r)
{
// Calculamos la nueva z
for (int i = 0; i < n; ++i)
{
z[i] = r[i]/D[i];
}
}
template <class P>
void JacobiP<P> :: impDiag()
{
for (int i = 0; i < n; ++i)
{
std::cout<<D[i]<<std::endl;
}
}
#endif // JACOBI_P_HPP