Test solution

This commit is contained in:
2025-09-08 20:10:23 +03:00
parent 5784ca2b83
commit 5ab3cf7986
4 changed files with 76 additions and 14 deletions

3
Header.h Normal file
View File

@@ -0,0 +1,3 @@
#include <iostream>
#include <fstream>
#include <Eigen/Dense>

View File

@@ -1,25 +1,76 @@
#include <iostream>
#include <fstream>
#include <Eigen/Dense>
#include "Header.h"
using namespace Eigen;
int main() {
std::ifstream file("matrix.txt");
int N = 20;
int upper = 6, lower = 1;
double A = 1, B = 1, C = 1, L = upper - lower, du_dx = 1.;
// Если размеры известны заранее (3x3 в вашем случае)
Matrix <double, 3, 3> matrix;
double dx = L / N;
for (int i = 0; i < 3; ++i) {
for (int j = 0; j < 3; ++j) {
if (!(file >> matrix(i, j))) {
std::cerr << "Error in matrix initialization";
return -1;
}
MatrixXd local = MatrixXd::Zero(2, 2);
local(0, 0) = A / dx - B / 2;
local(0, 1) = -A / dx + B / 2;
local(1, 0) = -A / dx - B / 2;
local(1, 1) = A / dx + B / 2;
}
VectorXd local_load(2);
local_load(0) = C * dx / 2 - A * du_dx;
local_load(1) = C * dx / 2 + A * du_dx;
// Ансамблированные матрица и вектор
MatrixXd ansamb = MatrixXd::Zero(N + 1, N + 1);
VectorXd global_load = VectorXd::Zero(N + 1);
// Ансамблирование для каждого конечного элемента
for (int elem = 0; elem < N; ++elem) {
int node_i = elem;
int node_j = elem + 1;
// Матрица жесткости
ansamb(node_i, node_i) += local(0, 0);
ansamb(node_i, node_j) += local(0, 1);
ansamb(node_j, node_i) += local(1, 0);
ansamb(node_j, node_j) += local(1, 1);
// Вектор нагрузки
global_load(node_i) += local_load(0);
global_load(node_j) += local_load(1);
}
std::cout << "Matrix:\n" << matrix << std::endl;
std::cout << "Before:" << std::endl;
std::cout << "Ansamb matrix:\n" << ansamb << std::endl;
std::cout << "Ansamb load vector:\n" << global_load << std::endl;
// ГУ 1-го рода
double u_left = 5.0; // u(1) = 5
double u_right = 15.0; // u(6) = 15
// 1. Учёт условия u(1)=5
for (int i = 1; i < N + 1; ++i) {
global_load(i) -= ansamb(i, 0) * u_left;
}
// Обнуляем первую строку и столбец матрицы, устанавливаем диагональ = 1
ansamb.row(0).setZero();
ansamb.col(0).setZero();
ansamb(0, 0) = 1.0;
global_load(0) = u_left;
// 2. Учёт условия u(6)=15
for (int i = 0; i < N; ++i) {
global_load(i) -= ansamb(i, N) * u_right;
}
// Обнуляем последнюю строку и столбец матрицы, устанавливаем диагональ = 1
ansamb.row(N).setZero();
ansamb.col(N).setZero();
ansamb(N, N) = 1.0;
global_load(N) = u_right;
std::cout << "\nAfter:" << std::endl;
std::cout << "Modified matrix:\n" << ansamb << std::endl;
std::cout << "Modified load vector:\n" << global_load << std::endl;
return 0;
}

View File

@@ -128,6 +128,9 @@
<ItemGroup>
<None Include="packages.config" />
</ItemGroup>
<ItemGroup>
<ClInclude Include="Header.h" />
</ItemGroup>
<Import Project="$(VCTargetsPath)\Microsoft.Cpp.targets" />
<ImportGroup Label="ExtensionTargets">
<Import Project="packages\Eigen.3.3.3\build\native\Eigen.targets" Condition="Exists('packages\Eigen.3.3.3\build\native\Eigen.targets')" />

View File

@@ -22,4 +22,9 @@
<ItemGroup>
<None Include="packages.config" />
</ItemGroup>
<ItemGroup>
<ClInclude Include="Header.h">
<Filter>Исходные файлы</Filter>
</ClInclude>
</ItemGroup>
</Project>