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
     86
     87
     88
     89
     90
     91
     92
     93
     94
#include <flens/flens.cxx>
#include <algorithm>
#include <iostream>

using namespace std;
using namespace flens;


// Initialize a triangular/trapezoidal matrix column wise with 1, 2, 3, ...
template <typename MA>
void
myInit(TrMatrix<MA> &A)
{
    int counter = 1;

    int m = A.numRows();
    int n = A.numCols();

    if (A.upLo()==Upper) {
        if (A.diag()==NonUnit) {
            for (int j=1; j<=n; ++j) {
                for (int i=1; i<=std::min(j,m); ++i) {
                    A(i,j) = counter++;
                }
            }
        } else {
            for (int j=1; j<=n; ++j) {
                for (int i=1; i<std::min(j,m); ++i) {
                    A(i,j) = counter++;
                }
            }
        }
    } else {
        if (A.diag()==NonUnit) {
            for (int j=1; j<=n; ++j) {
                for (int i=j; i<=m; ++i) {
                    A(i,j) = counter++;
                }
            }
        } else {
            for (int j=1; j<=n; ++j) {
                for (int i=j+1; i<=m; ++i) {
                    A(i,j) = counter++;
                }
            }
        }
    }
}

int
main()
{
    TrMatrix<FullStorage<double> >  A(3, Upper);
    TrMatrix<FullStorage<double> >  B(3, Upper, Unit);
    TrMatrix<FullStorage<double> >  C(34, Upper);
    TrMatrix<FullStorage<double> >  D(34, Upper, Unit);
    TrMatrix<FullStorage<double> >  E(43, Upper);
    TrMatrix<FullStorage<double> >  F(43, Upper, Unit);

    TrMatrix<FullStorage<double> >  G(3, Lower);
    TrMatrix<FullStorage<double> >  H(3, Lower, Unit);
    TrMatrix<FullStorage<double> >  I(34, Lower);
    TrMatrix<FullStorage<double> >  J(34, Lower, Unit);
    TrMatrix<FullStorage<double> >  K(43, Lower);
    TrMatrix<FullStorage<double> >  L(43, Lower, Unit);

    myInit(A);
    myInit(B);
    myInit(C);
    myInit(D);
    myInit(E);
    myInit(F);

    myInit(G);
    myInit(H);
    myInit(I);
    myInit(J);
    myInit(K);
    myInit(L);

    cout << "A = " << A << endl;
    cout << "B = " << B << endl;
    cout << "C = " << C << endl;
    cout << "D = " << D << endl;
    cout << "E = " << E << endl;
    cout << "F = " << F << endl;

    cout << "G = " << G << endl;
    cout << "H = " << H << endl;
    cout << "I = " << I << endl;
    cout << "J = " << J << endl;
    cout << "K = " << K << endl;
    cout << "L = " << L << endl;
}