tsv.cpp

/*************************************************************************************
 * MechSys - A C++ library to simulate (Continuum) Mechanical Systems                *
 * Copyright (C) 2005 Dorival de Moraes Pedroso <dorival.pedroso at gmail.com>       *
 * Copyright (C) 2005 Raul Dario Durand Farfan  <raul.durand at gmail.com>           *
 *                                                                                   *
 * This file is part of MechSys.                                                     *
 *                                                                                   *
 * MechSys 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 2 of the License, or (at your option) any later        *
 * version.                                                                          *
 *                                                                                   *
 * MechSys 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      *
 * MechSys; if not, write to the Free Software Foundation, Inc., 51 Franklin Street, *
 * Fifth Floor, Boston, MA 02110-1301, USA                                           *
 *************************************************************************************/

#define _NMBFMT std::setw(12) <<

#include <iostream>
#include <cmath>
#include <iomanip>

#include "linalg/lawrap.h"
#include "linalg/matrix.h"
#include "linalg/vector.h"
#include "util/exception.h"

using namespace std;
using namespace LinAlg;

int main() try
{
    // ================================================================= Linear Solver

    cout << "\n============================================== Linear Solver\n";

    Matrix<REAL> A(5,5,"3 1 3 1 8   1 5 9 -2 4   2 6 5 0 -7   1 2 2 3 1   1 2 3 1 -1");
    Vector<REAL> Y(5,"-1 -1  1 0 2");

    cout << "A = \n" << A << endl;
    cout << "Y = \n" << Y << endl;
    
    if (Gesv(A,Y)==0)
    {
        cout << "Y := inv(A)*Y\n" << Y << endl;
        cout << "matrix A after solver = \n" << A << endl;
    }
    else
        cout << "ERROR: Impossible to solve [A]{X} = {Y} !\n";

    // =================================================== Linear Solver (Tridiagonal)
    
    cout << "\n================================== Linear Solver (Tridiagonal)\n";

    /*         [M]            {W}     {N}
     *  / 2 -1  0  0  0 \   / w0 \   / 1 \
     * | -1  2 -1  0  0 |   | w1 |   | 2 |
     * |  0 -2  3 -1  0 | * | w2 | = | 3 |.
     * |  0  0 -1  3 -2 |   | w3 |   | 2 |
     * \  0  0  0 -1  1 /   \ w4 /   \ 1 /
     *              \  \  \  
     *               L  D  U
    */
    
    Vector<REAL> L(4,"-1 -2 -1 -1");
    Vector<REAL> D(5,"2 2 3 3 1");
    Vector<REAL> U(4,"-1 -1 -1 -2");
    Vector<REAL> N(5,"1 2 3 2 1");
    
    cout << "L = \n" << L << endl;
    cout << "D = \n" << D << endl;
    cout << "U = \n" << U << endl;
    cout << "N = \n" << N << endl;
    
    Gtsv(L,D,U,N);
    cout << "N := W = inv(M)*N = \n" << N << endl;
    

    return 0;
}
catch (Exception * e) //{{{ 
{
    e->Cout();
    if (e->IsFatal()) {delete e; exit(1);}
    delete e;
}
catch (char const * m)
{
    std::cout << "Fatal: " << m << std::endl;
    exit (1);
}
catch (...)
{
    std::cout << "Some exception (...) ocurred\n";
} //}}} 

// vim:fdm=marker

---