1 /*
2 * Copyright (c) 2011, Michael Lehn
3 *
4 * All rights reserved.
5 *
6 * Redistribution and use in source and binary forms, with or without
7 * modification, are permitted provided that the following conditions
8 * are met:
9 *
10 * 1) Redistributions of source code must retain the above copyright
11 * notice, this list of conditions and the following disclaimer.
12 * 2) Redistributions in binary form must reproduce the above copyright
13 * notice, this list of conditions and the following disclaimer in
14 * the documentation and/or other materials provided with the
15 * distribution.
16 * 3) Neither the name of the FLENS development group nor the names of
17 * its contributors may be used to endorse or promote products derived
18 * from this software without specific prior written permission.
19 *
20 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
21 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
22 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
23 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
24 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
25 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
26 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
27 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
28 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
29 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
30 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
31 */
32
33 #ifndef FLENS_STORAGE_FULLSTORAGE_TRAPEZOIDALFILL_TCC
34 #define FLENS_STORAGE_FULLSTORAGE_TRAPEZOIDALFILL_TCC 1
35
36 #include <cxxblas/typedefs.h>
37 #include <flens/typedefs.h>
38
39 namespace flens {
40
41 template <typename IndexType, typename T>
42 void
43 trapezoidalFill(StorageOrder order, StorageUpLo upLo,
44 const T &value,
45 IndexType m, IndexType n, T *data, IndexType ld)
46 {
47 // fill the upper trapezoidal part
48 if (upLo==Upper) {
49 if (order==RowMajor) {
50 for (IndexType i=0; i<min(m,n); ++i, data+=ld+1) {
51 std::fill_n(data, n-i, value);
52 }
53 }
54 if (order==ColMajor) {
55 for (IndexType j=0; j<n; ++j, data+=ld) {
56 std::fill_n(data, min(j+1,m), value);
57 }
58 }
59 }
60 // fill the lower trapezoidal part
61 if (upLo==Lower) {
62 if (order==RowMajor) {
63 for (IndexType i=0; i<m; ++i, data+=ld) {
64 std::fill_n(data, min(i+1, n), value);
65 }
66 }
67 if (order==ColMajor) {
68 for (IndexType j=0; j<min(m, n); ++j, data+=ld+1) {
69 std::fill_n(data, m-j, value);
70 }
71 }
72 }
73 }
74
75 } // namespace flens
76
77 #endif // FLENS_STORAGE_FULLSTORAGE_TRAPEZOIDALFILL_TCC
2 * Copyright (c) 2011, Michael Lehn
3 *
4 * All rights reserved.
5 *
6 * Redistribution and use in source and binary forms, with or without
7 * modification, are permitted provided that the following conditions
8 * are met:
9 *
10 * 1) Redistributions of source code must retain the above copyright
11 * notice, this list of conditions and the following disclaimer.
12 * 2) Redistributions in binary form must reproduce the above copyright
13 * notice, this list of conditions and the following disclaimer in
14 * the documentation and/or other materials provided with the
15 * distribution.
16 * 3) Neither the name of the FLENS development group nor the names of
17 * its contributors may be used to endorse or promote products derived
18 * from this software without specific prior written permission.
19 *
20 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
21 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
22 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
23 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
24 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
25 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
26 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
27 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
28 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
29 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
30 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
31 */
32
33 #ifndef FLENS_STORAGE_FULLSTORAGE_TRAPEZOIDALFILL_TCC
34 #define FLENS_STORAGE_FULLSTORAGE_TRAPEZOIDALFILL_TCC 1
35
36 #include <cxxblas/typedefs.h>
37 #include <flens/typedefs.h>
38
39 namespace flens {
40
41 template <typename IndexType, typename T>
42 void
43 trapezoidalFill(StorageOrder order, StorageUpLo upLo,
44 const T &value,
45 IndexType m, IndexType n, T *data, IndexType ld)
46 {
47 // fill the upper trapezoidal part
48 if (upLo==Upper) {
49 if (order==RowMajor) {
50 for (IndexType i=0; i<min(m,n); ++i, data+=ld+1) {
51 std::fill_n(data, n-i, value);
52 }
53 }
54 if (order==ColMajor) {
55 for (IndexType j=0; j<n; ++j, data+=ld) {
56 std::fill_n(data, min(j+1,m), value);
57 }
58 }
59 }
60 // fill the lower trapezoidal part
61 if (upLo==Lower) {
62 if (order==RowMajor) {
63 for (IndexType i=0; i<m; ++i, data+=ld) {
64 std::fill_n(data, min(i+1, n), value);
65 }
66 }
67 if (order==ColMajor) {
68 for (IndexType j=0; j<min(m, n); ++j, data+=ld+1) {
69 std::fill_n(data, m-j, value);
70 }
71 }
72 }
73 }
74
75 } // namespace flens
76
77 #endif // FLENS_STORAGE_FULLSTORAGE_TRAPEZOIDALFILL_TCC