Please, help us to better know about our user community by answering the following short survey: https://forms.gle/wpyrxWi18ox9Z5ae9
Eigen  3.4.0
 
Loading...
Searching...
No Matches
SparseLU_copy_to_ucol.h
1// This file is part of Eigen, a lightweight C++ template library
2// for linear algebra.
3//
4// Copyright (C) 2012 Désiré Nuentsa-Wakam <desire.nuentsa_wakam@inria.fr>
5//
6// This Source Code Form is subject to the terms of the Mozilla
7// Public License v. 2.0. If a copy of the MPL was not distributed
8// with this file, You can obtain one at http://mozilla.org/MPL/2.0/.
9/*
10
11 * NOTE: This file is the modified version of [s,d,c,z]copy_to_ucol.c file in SuperLU
12
13 * -- SuperLU routine (version 2.0) --
14 * Univ. of California Berkeley, Xerox Palo Alto Research Center,
15 * and Lawrence Berkeley National Lab.
16 * November 15, 1997
17 *
18 * Copyright (c) 1994 by Xerox Corporation. All rights reserved.
19 *
20 * THIS MATERIAL IS PROVIDED AS IS, WITH ABSOLUTELY NO WARRANTY
21 * EXPRESSED OR IMPLIED. ANY USE IS AT YOUR OWN RISK.
22 *
23 * Permission is hereby granted to use or copy this program for any
24 * purpose, provided the above notices are retained on all copies.
25 * Permission to modify the code and to distribute modified code is
26 * granted, provided the above notices are retained, and a notice that
27 * the code was modified is included with the above copyright notice.
28 */
29#ifndef SPARSELU_COPY_TO_UCOL_H
30#define SPARSELU_COPY_TO_UCOL_H
31
32namespace Eigen {
33namespace internal {
34
49template <typename Scalar, typename StorageIndex>
50Index SparseLUImpl<Scalar,StorageIndex>::copy_to_ucol(const Index jcol, const Index nseg, IndexVector& segrep,
51 BlockIndexVector repfnz ,IndexVector& perm_r, BlockScalarVector dense, GlobalLU_t& glu)
52{
53 Index ksub, krep, ksupno;
54
55 Index jsupno = glu.supno(jcol);
56
57 // For each nonzero supernode segment of U[*,j] in topological order
58 Index k = nseg - 1, i;
59 StorageIndex nextu = glu.xusub(jcol);
60 Index kfnz, isub, segsize;
61 Index new_next,irow;
62 Index fsupc, mem;
63 for (ksub = 0; ksub < nseg; ksub++)
64 {
65 krep = segrep(k); k--;
66 ksupno = glu.supno(krep);
67 if (jsupno != ksupno ) // should go into ucol();
68 {
69 kfnz = repfnz(krep);
70 if (kfnz != emptyIdxLU)
71 { // Nonzero U-segment
72 fsupc = glu.xsup(ksupno);
73 isub = glu.xlsub(fsupc) + kfnz - fsupc;
74 segsize = krep - kfnz + 1;
75 new_next = nextu + segsize;
76 while (new_next > glu.nzumax)
77 {
78 mem = memXpand<ScalarVector>(glu.ucol, glu.nzumax, nextu, UCOL, glu.num_expansions);
79 if (mem) return mem;
80 mem = memXpand<IndexVector>(glu.usub, glu.nzumax, nextu, USUB, glu.num_expansions);
81 if (mem) return mem;
82
83 }
84
85 for (i = 0; i < segsize; i++)
86 {
87 irow = glu.lsub(isub);
88 glu.usub(nextu) = perm_r(irow); // Unlike the L part, the U part is stored in its final order
89 glu.ucol(nextu) = dense(irow);
90 dense(irow) = Scalar(0.0);
91 nextu++;
92 isub++;
93 }
94
95 } // end nonzero U-segment
96
97 } // end if jsupno
98
99 } // end for each segment
100 glu.xusub(jcol + 1) = nextu; // close U(*,jcol)
101 return 0;
102}
103
104} // namespace internal
105} // end namespace Eigen
106
107#endif // SPARSELU_COPY_TO_UCOL_H
Namespace containing all symbols from the Eigen library.
Definition: Core:141
EIGEN_DEFAULT_DENSE_INDEX_TYPE Index
The Index type as used for the API.
Definition: Meta.h:74