35#include <boost/test/unit_test.hpp>
40#include <boost/test/tools/floating_point_comparison.hpp>
41#include <boost/test/unit_test.hpp>
45namespace BlockMatrixUnitTests
50 BOOST_CHECK_EQUAL(7, m.GetNumberOfRowsInBlockRow(0));
51 BOOST_CHECK_EQUAL(7, m.GetNumberOfRowsInBlockRow(1));
52 BOOST_CHECK_EQUAL(10, m.GetNumberOfColumnsInBlockColumn(0));
53 BOOST_CHECK_EQUAL(10, m.GetNumberOfColumnsInBlockColumn(1));
54 BOOST_CHECK_EQUAL(10, m.GetNumberOfColumnsInBlockColumn(2));
62 double m1_buf[] = {1, 6, 2, 7, 3, 8};
63 double m2_buf[] = {4, 9, 5, 10};
64 double m3_buf[] = {11, 16, 12, 17, 13, 18};
65 double m4_buf[] = {14, 19, 15, 20};
67 std::shared_ptr<InnerType> m1(
new InnerType(2, 3, m1_buf));
68 std::shared_ptr<InnerType> m2(
new InnerType(2, 2, m2_buf));
69 std::shared_ptr<InnerType> m3(
new InnerType(2, 3, m3_buf));
70 std::shared_ptr<InnerType> m4(
new InnerType(2, 2, m4_buf));
72 unsigned int rowCounts[] = {2, 2};
73 unsigned int colCounts[] = {3, 2};
75 BlockType b(2, 2, rowCounts, colCounts);
81 double rhs_buf[] = {10, 20, 30, 40, 50};
86 double expected_buf[] = {550, 1300, 2050, 2800};
89 BOOST_CHECK_EQUAL(expected_result, result);
97 double m00_buf[] = {1, 2, 3, 4, 5, 6, 7, 8};
98 double m01_buf[] = {9, 10, 11, 12, 13, 14, 15, 16, 17, 18};
99 double m02_buf[] = {19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30};
100 double m10_buf[] = {31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42};
101 double m11_buf[] = {43, 44, 45, 46, 47, 48, 49, 50,
102 51, 52, 53, 54, 55, 56, 57};
103 double m12_buf[] = {58, 59, 60, 61, 62, 63, 64, 65, 66,
104 67, 68, 69, 70, 71, 72, 73, 74, 75};
106 std::shared_ptr<InnerType> m00(
new InnerType(2, 4, m00_buf));
107 std::shared_ptr<InnerType> m01(
new InnerType(2, 5, m01_buf));
108 std::shared_ptr<InnerType> m02(
new InnerType(2, 6, m02_buf));
109 std::shared_ptr<InnerType> m10(
new InnerType(3, 4, m10_buf));
110 std::shared_ptr<InnerType> m11(
new InnerType(3, 5, m11_buf));
111 std::shared_ptr<InnerType> m12(
new InnerType(3, 6, m12_buf));
113 unsigned int rowCounts[] = {2, 3};
114 unsigned int colCounts[] = {4, 5, 6};
116 BlockType b(2, 3, rowCounts, colCounts);
117 b.SetBlock(0, 0, m00);
118 b.SetBlock(0, 1, m01);
119 b.SetBlock(0, 2, m02);
120 b.SetBlock(1, 0, m10);
121 b.SetBlock(1, 1, m11);
122 b.SetBlock(1, 2, m12);
124 BOOST_CHECK_EQUAL(2, b.GetNumberOfBlockRows());
125 BOOST_CHECK_EQUAL(3, b.GetNumberOfBlockColumns());
126 BOOST_CHECK_EQUAL(2, b.GetNumberOfRowsInBlockRow(0));
127 BOOST_CHECK_EQUAL(3, b.GetNumberOfRowsInBlockRow(1));
128 BOOST_CHECK_EQUAL(4, b.GetNumberOfColumnsInBlockColumn(0));
129 BOOST_CHECK_EQUAL(5, b.GetNumberOfColumnsInBlockColumn(1));
130 BOOST_CHECK_EQUAL(6, b.GetNumberOfColumnsInBlockColumn(2));
132 BOOST_CHECK_EQUAL(0, b.CalculateBlockIndex(0, 0));
133 BOOST_CHECK_EQUAL(1, b.CalculateBlockIndex(1, 0));
134 BOOST_CHECK_EQUAL(2, b.CalculateBlockIndex(0, 1));
135 BOOST_CHECK_EQUAL(3, b.CalculateBlockIndex(1, 1));
136 BOOST_CHECK_EQUAL(4, b.CalculateBlockIndex(0, 2));
137 BOOST_CHECK_EQUAL(5, b.CalculateBlockIndex(1, 2));
139 BOOST_CHECK_EQUAL(m00->GetRawPtr(), b.GetBlock(0, 0)->GetRawPtr());
140 BOOST_CHECK_EQUAL(m01->GetRawPtr(), b.GetBlock(0, 1)->GetRawPtr());
141 BOOST_CHECK_EQUAL(m02->GetRawPtr(), b.GetBlock(0, 2)->GetRawPtr());
142 BOOST_CHECK_EQUAL(m10->GetRawPtr(), b.GetBlock(1, 0)->GetRawPtr());
143 BOOST_CHECK_EQUAL(m11->GetRawPtr(), b.GetBlock(1, 1)->GetRawPtr());
144 BOOST_CHECK_EQUAL(m12->GetRawPtr(), b.GetBlock(1, 2)->GetRawPtr());
146 BOOST_CHECK_EQUAL(
'N', b.GetBlock(0, 0)->GetTransposeFlag());
147 BOOST_CHECK_EQUAL(
'N', b.GetBlock(0, 1)->GetTransposeFlag());
148 BOOST_CHECK_EQUAL(
'N', b.GetBlock(0, 2)->GetTransposeFlag());
149 BOOST_CHECK_EQUAL(
'N', b.GetBlock(1, 0)->GetTransposeFlag());
150 BOOST_CHECK_EQUAL(
'N', b.GetBlock(1, 1)->GetTransposeFlag());
151 BOOST_CHECK_EQUAL(
'N', b.GetBlock(1, 2)->GetTransposeFlag());
153 double v_buf[] = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15};
157 BOOST_CHECK_EQUAL(5, result.
GetRows());
158 BOOST_CHECK_EQUAL(2360, result[0]);
159 BOOST_CHECK_EQUAL(2480, result[1]);
160 BOOST_CHECK_EQUAL(7080, result[2]);
161 BOOST_CHECK_EQUAL(7200, result[3]);
162 BOOST_CHECK_EQUAL(7320, result[4]);
166 BOOST_CHECK_EQUAL(3, b.GetNumberOfBlockRows());
167 BOOST_CHECK_EQUAL(2, b.GetNumberOfBlockColumns());
169 BOOST_CHECK_EQUAL(2, b.GetNumberOfColumnsInBlockColumn(0));
170 BOOST_CHECK_EQUAL(3, b.GetNumberOfColumnsInBlockColumn(1));
171 BOOST_CHECK_EQUAL(4, b.GetNumberOfRowsInBlockRow(0));
172 BOOST_CHECK_EQUAL(5, b.GetNumberOfRowsInBlockRow(1));
173 BOOST_CHECK_EQUAL(6, b.GetNumberOfRowsInBlockRow(2));
175 BOOST_CHECK_EQUAL(0, b.CalculateBlockIndex(0, 0));
176 BOOST_CHECK_EQUAL(1, b.CalculateBlockIndex(0, 1));
177 BOOST_CHECK_EQUAL(2, b.CalculateBlockIndex(1, 0));
178 BOOST_CHECK_EQUAL(3, b.CalculateBlockIndex(1, 1));
179 BOOST_CHECK_EQUAL(4, b.CalculateBlockIndex(2, 0));
180 BOOST_CHECK_EQUAL(5, b.CalculateBlockIndex(2, 1));
182 BOOST_CHECK_EQUAL(m00->GetRawPtr(), b.GetBlock(0, 0)->GetRawPtr());
183 BOOST_CHECK_EQUAL(m01->GetRawPtr(), b.GetBlock(1, 0)->GetRawPtr());
184 BOOST_CHECK_EQUAL(m02->GetRawPtr(), b.GetBlock(2, 0)->GetRawPtr());
185 BOOST_CHECK_EQUAL(m10->GetRawPtr(), b.GetBlock(0, 1)->GetRawPtr());
186 BOOST_CHECK_EQUAL(m11->GetRawPtr(), b.GetBlock(1, 1)->GetRawPtr());
187 BOOST_CHECK_EQUAL(m12->GetRawPtr(), b.GetBlock(2, 1)->GetRawPtr());
189 BOOST_CHECK_EQUAL(
'T', b.GetBlock(0, 0)->GetTransposeFlag());
190 BOOST_CHECK_EQUAL(
'T', b.GetBlock(1, 0)->GetTransposeFlag());
191 BOOST_CHECK_EQUAL(
'T', b.GetBlock(2, 0)->GetTransposeFlag());
192 BOOST_CHECK_EQUAL(
'T', b.GetBlock(0, 1)->GetTransposeFlag());
193 BOOST_CHECK_EQUAL(
'T', b.GetBlock(1, 1)->GetTransposeFlag());
194 BOOST_CHECK_EQUAL(
'T', b.GetBlock(2, 1)->GetTransposeFlag());
196 double v_transpose_buf[] = {1, 2, 3, 4, 5};
200 BOOST_CHECK_EQUAL(15, result_transpose.
GetRows());
201 BOOST_CHECK_EQUAL(391, result_transpose[0]);
202 BOOST_CHECK_EQUAL(433, result_transpose[1]);
203 BOOST_CHECK_EQUAL(475, result_transpose[2]);
204 BOOST_CHECK_EQUAL(517, result_transpose[3]);
205 BOOST_CHECK_EQUAL(559, result_transpose[4]);
206 BOOST_CHECK_EQUAL(601, result_transpose[5]);
207 BOOST_CHECK_EQUAL(643, result_transpose[6]);
208 BOOST_CHECK_EQUAL(685, result_transpose[7]);
209 BOOST_CHECK_EQUAL(727, result_transpose[8]);
210 BOOST_CHECK_EQUAL(769, result_transpose[9]);
211 BOOST_CHECK_EQUAL(811, result_transpose[10]);
212 BOOST_CHECK_EQUAL(853, result_transpose[11]);
213 BOOST_CHECK_EQUAL(895, result_transpose[12]);
214 BOOST_CHECK_EQUAL(937, result_transpose[13]);
215 BOOST_CHECK_EQUAL(979, result_transpose[14]);
217 BlockType globalTranspose =
Transpose(b);
219 BOOST_CHECK_EQUAL(5, globalResult.
GetRows());
220 BOOST_CHECK_EQUAL(2360, globalResult[0]);
221 BOOST_CHECK_EQUAL(2480, globalResult[1]);
222 BOOST_CHECK_EQUAL(7080, globalResult[2]);
223 BOOST_CHECK_EQUAL(7200, globalResult[3]);
224 BOOST_CHECK_EQUAL(7320, globalResult[4]);
233 double m00_buf[] = {1, 2, 3, 4, 5, 6, 7, 8};
234 double m01_buf[] = {9, 10, 11, 12, 13, 14, 15, 16, 17, 18};
235 double m02_buf[] = {19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30};
236 double m10_buf[] = {31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42};
237 double m11_buf[] = {43, 44, 45, 46, 47, 48, 49, 50,
238 51, 52, 53, 54, 55, 56, 57};
239 double m12_buf[] = {58, 59, 60, 61, 62, 63, 64, 65, 66,
240 67, 68, 69, 70, 71, 72, 73, 74, 75};
242 std::shared_ptr<NekMatrix<double>> m00(
244 std::shared_ptr<NekMatrix<double>> m01(
246 std::shared_ptr<NekMatrix<double>> m02(
248 std::shared_ptr<NekMatrix<double>> m10(
250 std::shared_ptr<NekMatrix<double>> m11(
252 std::shared_ptr<NekMatrix<double>> m12(
255 std::shared_ptr<InnerType> sm00(
new InnerType(2.0, m00));
256 std::shared_ptr<InnerType> sm01(
new InnerType(2.0, m01));
257 std::shared_ptr<InnerType> sm02(
new InnerType(2.0, m02));
258 std::shared_ptr<InnerType> sm10(
new InnerType(2.0, m10));
259 std::shared_ptr<InnerType> sm11(
new InnerType(2.0, m11));
260 std::shared_ptr<InnerType> sm12(
new InnerType(2.0, m12));
262 unsigned int rowCounts[] = {2, 3};
263 unsigned int colCounts[] = {4, 5, 6};
265 BlockType b(2, 3, rowCounts, colCounts);
266 b.SetBlock(0, 0, sm00);
267 b.SetBlock(0, 1, sm01);
268 b.SetBlock(0, 2, sm02);
269 b.SetBlock(1, 0, sm10);
270 b.SetBlock(1, 1, sm11);
271 b.SetBlock(1, 2, sm12);
273 BOOST_CHECK_EQUAL(2, b.GetNumberOfBlockRows());
274 BOOST_CHECK_EQUAL(3, b.GetNumberOfBlockColumns());
275 BOOST_CHECK_EQUAL(2, b.GetNumberOfRowsInBlockRow(0));
276 BOOST_CHECK_EQUAL(3, b.GetNumberOfRowsInBlockRow(1));
277 BOOST_CHECK_EQUAL(4, b.GetNumberOfColumnsInBlockColumn(0));
278 BOOST_CHECK_EQUAL(5, b.GetNumberOfColumnsInBlockColumn(1));
279 BOOST_CHECK_EQUAL(6, b.GetNumberOfColumnsInBlockColumn(2));
281 BOOST_CHECK_EQUAL(0, b.CalculateBlockIndex(0, 0));
282 BOOST_CHECK_EQUAL(1, b.CalculateBlockIndex(1, 0));
283 BOOST_CHECK_EQUAL(2, b.CalculateBlockIndex(0, 1));
284 BOOST_CHECK_EQUAL(3, b.CalculateBlockIndex(1, 1));
285 BOOST_CHECK_EQUAL(4, b.CalculateBlockIndex(0, 2));
286 BOOST_CHECK_EQUAL(5, b.CalculateBlockIndex(1, 2));
288 BOOST_CHECK_EQUAL(m00->GetRawPtr(), b.GetBlock(0, 0)->GetRawPtr());
289 BOOST_CHECK_EQUAL(m01->GetRawPtr(), b.GetBlock(0, 1)->GetRawPtr());
290 BOOST_CHECK_EQUAL(m02->GetRawPtr(), b.GetBlock(0, 2)->GetRawPtr());
291 BOOST_CHECK_EQUAL(m10->GetRawPtr(), b.GetBlock(1, 0)->GetRawPtr());
292 BOOST_CHECK_EQUAL(m11->GetRawPtr(), b.GetBlock(1, 1)->GetRawPtr());
293 BOOST_CHECK_EQUAL(m12->GetRawPtr(), b.GetBlock(1, 2)->GetRawPtr());
295 BOOST_CHECK_EQUAL(
'N', b.GetBlock(0, 0)->GetTransposeFlag());
296 BOOST_CHECK_EQUAL(
'N', b.GetBlock(0, 1)->GetTransposeFlag());
297 BOOST_CHECK_EQUAL(
'N', b.GetBlock(0, 2)->GetTransposeFlag());
298 BOOST_CHECK_EQUAL(
'N', b.GetBlock(1, 0)->GetTransposeFlag());
299 BOOST_CHECK_EQUAL(
'N', b.GetBlock(1, 1)->GetTransposeFlag());
300 BOOST_CHECK_EQUAL(
'N', b.GetBlock(1, 2)->GetTransposeFlag());
302 double v_buf[] = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15};
306 BOOST_CHECK_EQUAL(5, result.
GetRows());
307 BOOST_CHECK_EQUAL(2360 * 2, result[0]);
308 BOOST_CHECK_EQUAL(2480 * 2, result[1]);
309 BOOST_CHECK_EQUAL(7080 * 2, result[2]);
310 BOOST_CHECK_EQUAL(7200 * 2, result[3]);
311 BOOST_CHECK_EQUAL(7320 * 2, result[4]);
315 BOOST_CHECK_EQUAL(3, b.GetNumberOfBlockRows());
316 BOOST_CHECK_EQUAL(2, b.GetNumberOfBlockColumns());
318 BOOST_CHECK_EQUAL(2, b.GetNumberOfColumnsInBlockColumn(0));
319 BOOST_CHECK_EQUAL(3, b.GetNumberOfColumnsInBlockColumn(1));
320 BOOST_CHECK_EQUAL(4, b.GetNumberOfRowsInBlockRow(0));
321 BOOST_CHECK_EQUAL(5, b.GetNumberOfRowsInBlockRow(1));
322 BOOST_CHECK_EQUAL(6, b.GetNumberOfRowsInBlockRow(2));
324 BOOST_CHECK_EQUAL(0, b.CalculateBlockIndex(0, 0));
325 BOOST_CHECK_EQUAL(1, b.CalculateBlockIndex(0, 1));
326 BOOST_CHECK_EQUAL(2, b.CalculateBlockIndex(1, 0));
327 BOOST_CHECK_EQUAL(3, b.CalculateBlockIndex(1, 1));
328 BOOST_CHECK_EQUAL(4, b.CalculateBlockIndex(2, 0));
329 BOOST_CHECK_EQUAL(5, b.CalculateBlockIndex(2, 1));
331 BOOST_CHECK_EQUAL(m00->GetRawPtr(), b.GetBlock(0, 0)->GetRawPtr());
332 BOOST_CHECK_EQUAL(m01->GetRawPtr(), b.GetBlock(1, 0)->GetRawPtr());
333 BOOST_CHECK_EQUAL(m02->GetRawPtr(), b.GetBlock(2, 0)->GetRawPtr());
334 BOOST_CHECK_EQUAL(m10->GetRawPtr(), b.GetBlock(0, 1)->GetRawPtr());
335 BOOST_CHECK_EQUAL(m11->GetRawPtr(), b.GetBlock(1, 1)->GetRawPtr());
336 BOOST_CHECK_EQUAL(m12->GetRawPtr(), b.GetBlock(2, 1)->GetRawPtr());
338 BOOST_CHECK_EQUAL(
'T', b.GetBlock(0, 0)->GetTransposeFlag());
339 BOOST_CHECK_EQUAL(
'T', b.GetBlock(1, 0)->GetTransposeFlag());
340 BOOST_CHECK_EQUAL(
'T', b.GetBlock(2, 0)->GetTransposeFlag());
341 BOOST_CHECK_EQUAL(
'T', b.GetBlock(0, 1)->GetTransposeFlag());
342 BOOST_CHECK_EQUAL(
'T', b.GetBlock(1, 1)->GetTransposeFlag());
343 BOOST_CHECK_EQUAL(
'T', b.GetBlock(2, 1)->GetTransposeFlag());
345 double v_transpose_buf[] = {1, 2, 3, 4, 5};
349 BOOST_CHECK_EQUAL(15, result_transpose.
GetRows());
350 BOOST_CHECK_EQUAL(391 * 2, result_transpose[0]);
351 BOOST_CHECK_EQUAL(433 * 2, result_transpose[1]);
352 BOOST_CHECK_EQUAL(475 * 2, result_transpose[2]);
353 BOOST_CHECK_EQUAL(517 * 2, result_transpose[3]);
354 BOOST_CHECK_EQUAL(559 * 2, result_transpose[4]);
355 BOOST_CHECK_EQUAL(601 * 2, result_transpose[5]);
356 BOOST_CHECK_EQUAL(643 * 2, result_transpose[6]);
357 BOOST_CHECK_EQUAL(685 * 2, result_transpose[7]);
358 BOOST_CHECK_EQUAL(727 * 2, result_transpose[8]);
359 BOOST_CHECK_EQUAL(769 * 2, result_transpose[9]);
360 BOOST_CHECK_EQUAL(811 * 2, result_transpose[10]);
361 BOOST_CHECK_EQUAL(853 * 2, result_transpose[11]);
362 BOOST_CHECK_EQUAL(895 * 2, result_transpose[12]);
363 BOOST_CHECK_EQUAL(937 * 2, result_transpose[13]);
364 BOOST_CHECK_EQUAL(979 * 2, result_transpose[14]);
373 double m_00_buf[] = {1, 2, 3, 4, 5, 6};
374 double m_01_buf[] = {7, 8, 9, 10, 11, 12, 13, 14};
375 double m_10_buf[] = {15, 16, 17, 18, 19, 20};
376 double m_11_buf[] = {21, 22, 23, 24, 25, 26, 27, 28};
377 double m_20_buf[] = {29, 30, 31, 32, 33, 34};
378 double m_21_buf[] = {35, 36, 37, 38, 39, 40, 41, 42};
379 std::shared_ptr<Matrix> m00(
new Matrix(2, 3, m_00_buf));
380 std::shared_ptr<Matrix> m01(
new Matrix(2, 4, m_01_buf));
381 std::shared_ptr<Matrix> m10(
new Matrix(2, 3, m_10_buf));
382 std::shared_ptr<Matrix> m11(
new Matrix(2, 4, m_11_buf));
383 std::shared_ptr<Matrix> m20(
new Matrix(2, 3, m_20_buf));
384 std::shared_ptr<Matrix> m21(
new Matrix(2, 4, m_21_buf));
386 unsigned int rowCounts[] = {2, 2, 2};
387 unsigned int colCounts[] = {3, 4};
389 std::vector<std::shared_ptr<InnerMatrix>> innerMatrices;
391 for (
int i = 0; i < 6; ++i)
393 std::shared_ptr<Matrix> t00(
new Matrix(*m00));
394 std::shared_ptr<Matrix> t01(
new Matrix(*m01));
395 std::shared_ptr<Matrix> t10(
new Matrix(*m10));
396 std::shared_ptr<Matrix> t11(
new Matrix(*m11));
397 std::shared_ptr<Matrix> t20(
new Matrix(*m20));
398 std::shared_ptr<Matrix> t21(
new Matrix(*m21));
407 std::shared_ptr<InnerMatrix> matrix(
409 matrix->SetBlock(0, 0, t00);
410 matrix->SetBlock(0, 1, t01);
411 matrix->SetBlock(1, 0, t10);
412 matrix->SetBlock(1, 1, t11);
413 matrix->SetBlock(2, 0, t20);
414 matrix->SetBlock(2, 1, t21);
416 innerMatrices.push_back(matrix);
419 unsigned int topLevelRowCounts[] = {6, 6};
420 unsigned int topLevelColumnCounts[] = {7, 7, 7};
421 BlockMatrix topLevelMatrix(2, 3, topLevelRowCounts, topLevelColumnCounts);
422 topLevelMatrix.SetBlock(0, 0, innerMatrices[0]);
423 topLevelMatrix.SetBlock(0, 1, innerMatrices[1]);
424 topLevelMatrix.SetBlock(0, 2, innerMatrices[2]);
425 topLevelMatrix.SetBlock(1, 0, innerMatrices[3]);
426 topLevelMatrix.SetBlock(1, 1, innerMatrices[4]);
427 topLevelMatrix.SetBlock(1, 2, innerMatrices[5]);
429 double v_buf[] = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11,
430 12, 13, 14, 15, 16, 17, 18, 19, 20, 21};
435 double expected_result_buf[] = {4256, 4816, 12096, 12656, 19936, 20496,
436 9611, 10864, 27153, 28406, 44695, 45948};
439 BOOST_CHECK_EQUAL(expected_result, result);
441 topLevelMatrix.Transpose();
443 double transposed_v_buf[] = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12};
447 double expected_transpose_result_buf[] = {
448 4427, 4925, 5423, 5921, 6419, 6917, 7415, 5863, 6517, 7171, 7825,
449 8479, 9133, 9787, 7299, 8109, 8919, 9729, 10539, 11349, 12159};
451 expected_transpose_result_buf);
453 BOOST_CHECK_EQUAL(expected_transpose_result, transposed_result);
463 double m_00_buf[] = {1, 2, 3, 4, 5, 6};
464 double m_01_buf[] = {7, 8, 9, 10, 11, 12, 13, 14};
465 double m_10_buf[] = {15, 16, 17, 18, 19, 20};
466 double m_11_buf[] = {21, 22, 23, 24, 25, 26, 27, 28};
467 double m_20_buf[] = {29, 30, 31, 32, 33, 34};
468 double m_21_buf[] = {35, 36, 37, 38, 39, 40, 41, 42};
469 std::shared_ptr<Matrix> m00(
new Matrix(2, 3, m_00_buf));
470 std::shared_ptr<Matrix> m01(
new Matrix(2, 4, m_01_buf));
471 std::shared_ptr<Matrix> m10(
new Matrix(2, 3, m_10_buf));
472 std::shared_ptr<Matrix> m11(
new Matrix(2, 4, m_11_buf));
473 std::shared_ptr<Matrix> m20(
new Matrix(2, 3, m_20_buf));
474 std::shared_ptr<Matrix> m21(
new Matrix(2, 4, m_21_buf));
476 unsigned int rowCounts[] = {2, 2, 2};
477 unsigned int colCounts[] = {3, 4};
479 std::vector<std::shared_ptr<InnerMatrix>> innerMatrices;
481 for (
int i = 0; i < 6; ++i)
483 std::shared_ptr<Matrix> t00(
new Matrix(*m00));
484 std::shared_ptr<Matrix> t01(
new Matrix(*m01));
485 std::shared_ptr<Matrix> t10(
new Matrix(*m10));
486 std::shared_ptr<Matrix> t11(
new Matrix(*m11));
487 std::shared_ptr<Matrix> t20(
new Matrix(*m20));
488 std::shared_ptr<Matrix> t21(
new Matrix(*m21));
490 std::shared_ptr<ScaledMatrix> s00(
new ScaledMatrix(i + 1, t00));
491 std::shared_ptr<ScaledMatrix> s01(
new ScaledMatrix(i + 1, t01));
492 std::shared_ptr<ScaledMatrix> s10(
new ScaledMatrix(i + 1, t10));
493 std::shared_ptr<ScaledMatrix> s11(
new ScaledMatrix(i + 1, t11));
494 std::shared_ptr<ScaledMatrix> s20(
new ScaledMatrix(i + 1, t20));
495 std::shared_ptr<ScaledMatrix> s21(
new ScaledMatrix(i + 1, t21));
497 std::shared_ptr<InnerMatrix> matrix(
499 matrix->SetBlock(0, 0, s00);
500 matrix->SetBlock(0, 1, s01);
501 matrix->SetBlock(1, 0, s10);
502 matrix->SetBlock(1, 1, s11);
503 matrix->SetBlock(2, 0, s20);
504 matrix->SetBlock(2, 1, s21);
506 innerMatrices.push_back(matrix);
509 unsigned int topLevelRowCounts[] = {6, 6};
510 unsigned int topLevelColumnCounts[] = {7, 7, 7};
511 BlockMatrix topLevelMatrix(2, 3, topLevelRowCounts, topLevelColumnCounts);
512 topLevelMatrix.SetBlock(0, 0, innerMatrices[0]);
513 topLevelMatrix.SetBlock(0, 1, innerMatrices[1]);
514 topLevelMatrix.SetBlock(0, 2, innerMatrices[2]);
515 topLevelMatrix.SetBlock(1, 0, innerMatrices[3]);
516 topLevelMatrix.SetBlock(1, 1, innerMatrices[4]);
517 topLevelMatrix.SetBlock(1, 2, innerMatrices[5]);
519 double v_buf[] = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11,
520 12, 13, 14, 15, 16, 17, 18, 19, 20, 21};
525 double expected_result_buf[] = {4256, 4816, 12096, 12656, 19936, 20496,
526 9611, 10864, 27153, 28406, 44695, 45948};
529 BOOST_CHECK_EQUAL(expected_result, result);
531 topLevelMatrix.Transpose();
533 double transposed_v_buf[] = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12};
537 double expected_transpose_result_buf[] = {
538 4427, 4925, 5423, 5921, 6419, 6917, 7415, 5863, 6517, 7171, 7825,
539 8479, 9133, 9787, 7299, 8109, 8919, 9729, 10539, 11349, 12159};
541 expected_transpose_result_buf);
543 BOOST_CHECK_EQUAL(expected_transpose_result, transposed_result);
551 double m_00_buf[] = {3, 1, 5, 5};
552 double m_01_buf[] = {2, 4, 4, 1, 1, 3};
553 double m_02_buf[] = {3, 1, 2, 2, 4, 1, 4, 2};
555 double m_10_buf[] = {1, 3, 5, 1, 4, 2};
556 double m_11_buf[] = {4, 4, 1, 1, 1, 4, 5, 5, 3};
557 double m_12_buf[] = {5, 2, 1, 2, 3, 1, 1, 3, 1, 4, 1, 1};
559 double m_20_buf[] = {3, 1, 4, 2, 5, 1, 4, 2};
560 double m_21_buf[] = {4, 5, 2, 4, 4, 2, 3, 5, 2, 2, 5, 3};
561 double m_22_buf[] = {3, 4, 3, 1, 2, 1, 2, 4, 4, 5, 2, 3, 5, 4, 1, 1};
563 double m_30_buf[] = {2, 2, 1, 4, 5, 5};
564 double m_31_buf[] = {1, 1, 3, 2, 1, 2, 3, 3, 2};
565 double m_32_buf[] = {3, 1, 3, 1, 2, 2, 4, 2, 5, 5, 1, 1};
567 std::shared_ptr<InnerType> m00(
new InnerType(2, 2, m_00_buf));
568 std::shared_ptr<InnerType> m01(
new InnerType(2, 3, m_01_buf));
569 std::shared_ptr<InnerType> m02(
new InnerType(2, 4, m_02_buf));
571 std::shared_ptr<InnerType> m10(
new InnerType(3, 2, m_10_buf));
572 std::shared_ptr<InnerType> m11(
new InnerType(3, 3, m_11_buf));
573 std::shared_ptr<InnerType> m12(
new InnerType(3, 4, m_12_buf));
575 std::shared_ptr<InnerType> m20(
new InnerType(4, 2, m_20_buf));
576 std::shared_ptr<InnerType> m21(
new InnerType(4, 3, m_21_buf));
577 std::shared_ptr<InnerType> m22(
new InnerType(4, 4, m_22_buf));
579 std::shared_ptr<InnerType> m30(
new InnerType(3, 2, m_30_buf));
580 std::shared_ptr<InnerType> m31(
new InnerType(3, 3, m_31_buf));
581 std::shared_ptr<InnerType> m32(
new InnerType(3, 4, m_32_buf));
583 unsigned int rowCounts[] = {2, 3, 4, 3};
584 unsigned int colCounts[] = {2, 3, 4};
586 BlockType b(4, 3, rowCounts, colCounts);
587 b.SetBlock(0, 0, m00);
588 b.SetBlock(0, 1, m01);
589 b.SetBlock(0, 2, m02);
590 b.SetBlock(1, 0, m10);
591 b.SetBlock(1, 1, m11);
592 b.SetBlock(1, 2, m12);
593 b.SetBlock(2, 0, m20);
594 b.SetBlock(2, 1, m21);
595 b.SetBlock(2, 2, m22);
596 b.SetBlock(3, 0, m30);
597 b.SetBlock(3, 1, m31);
598 b.SetBlock(3, 2, m32);
602 BlockType b_initializedWithArray(arrayRowCounts, arrayColCounts);
603 b_initializedWithArray.SetBlock(0, 0, m00);
604 b_initializedWithArray.SetBlock(0, 1, m01);
605 b_initializedWithArray.SetBlock(0, 2, m02);
606 b_initializedWithArray.SetBlock(1, 0, m10);
607 b_initializedWithArray.SetBlock(1, 1, m11);
608 b_initializedWithArray.SetBlock(1, 2, m12);
609 b_initializedWithArray.SetBlock(2, 0, m20);
610 b_initializedWithArray.SetBlock(2, 1, m21);
611 b_initializedWithArray.SetBlock(2, 2, m22);
612 b_initializedWithArray.SetBlock(3, 0, m30);
613 b_initializedWithArray.SetBlock(3, 1, m31);
614 b_initializedWithArray.SetBlock(3, 2, m32);
616 double rhs_buf[] = {4, 2, 5, 5, 3, 1, 3, 2, 3};
622 double expected_buf[] = {84, 63, 71, 80, 67, 100, 76, 80, 88, 69, 51, 67};
625 BOOST_CHECK_EQUAL(expected_result, result);
626 BOOST_CHECK_EQUAL(expected_result, result1);
634 double m_00_buf[] = {1, 3, 2, 4};
635 double m_11_buf[] = {1, 4, 2, 5, 3, 6};
636 double m_22_buf[] = {1, 3, 5, 2, 4, 6};
638 std::shared_ptr<InnerType> m00(
new InnerType(2, 2, m_00_buf));
639 std::shared_ptr<InnerType> m11(
new InnerType(2, 3, m_11_buf));
640 std::shared_ptr<InnerType> m22(
new InnerType(3, 2, m_22_buf));
642 unsigned int rowCounts[] = {2, 2, 3};
643 unsigned int colCounts[] = {2, 3, 2};
645 BlockType b(3, 3, rowCounts, colCounts,
eDIAGONAL);
646 b.SetBlock(0, 0, m00);
647 b.SetBlock(1, 1, m11);
648 b.SetBlock(2, 2, m22);
650 double rhs_buf[] = {1, 2, 3, 4, 5, 6, 7};
655 double expected_buf[] = {5, 11, 26, 62, 20, 46, 72};
658 BOOST_CHECK_EQUAL(expected_result, result);
666 double m_22_buf[] = {1, 2, 3, 4, 5, 6, 7, 8};
668 std::shared_ptr<InnerType> m22(
new InnerType(1, 8, m_22_buf));
670 unsigned int rowCounts[] = {0, 0, 1};
671 unsigned int colCounts[] = {6, 6, 8};
673 BlockType b(3, 3, rowCounts, colCounts,
eDIAGONAL);
674 b.SetBlock(2, 2, m22);
676 double rhs_buf[] = {0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
677 0, 0, 1, 2, 3, 4, 5, 6, 7, 8};
682 double expected_buf[] = {1 * 1 + 2 * 2 + 3 * 3 + 4 * 4 + 5 * 5 + 6 * 6 +
686 BOOST_CHECK_EQUAL(expected_result, result);
694 double m_22_buf[] = {1, 2, 3, 4, 5, 6, 7, 8};
696 std::shared_ptr<InnerType> m22(
new InnerType(1, 8, m_22_buf));
698 unsigned int rowCounts[] = {0, 0, 1};
699 unsigned int colCounts[] = {6, 6, 8};
701 BlockType b(3, 3, rowCounts, colCounts,
eFULL);
702 b.SetBlock(2, 2, m22);
704 double rhs_buf[] = {0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
705 0, 0, 1, 2, 3, 4, 5, 6, 7, 8};
710 double expected_buf[] = {1 * 1 + 2 * 2 + 3 * 3 + 4 * 4 + 5 * 5 + 6 * 6 +
714 BOOST_CHECK_EQUAL(expected_result, result);
745 std::vector<std::shared_ptr<M2>> scaledMatrices;
746 for (
unsigned int i = 0; i < 36; ++i)
748 double values[] = {i * 4.0, i * 4.0 + 1.0, i * 4.0 + 2.0,
750 std::shared_ptr<M1> matrix(
new M1(2, 2, values));
751 std::shared_ptr<M2> scaled(
new M2(1.0, matrix));
752 scaledMatrices.push_back(scaled);
755 std::vector<std::shared_ptr<M3>> blockMatrices;
756 for (
unsigned int i = 0; i < 6; ++i)
758 std::shared_ptr<M3> blockMatrix(
new M3(2, 3, 2, 2));
759 blockMatrix->SetBlock(0, 0, scaledMatrices[i * 6]);
760 blockMatrix->SetBlock(0, 1, scaledMatrices[i * 6 + 1]);
761 blockMatrix->SetBlock(0, 2, scaledMatrices[i * 6 + 2]);
762 blockMatrix->SetBlock(1, 0, scaledMatrices[i * 6 + 3]);
763 blockMatrix->SetBlock(1, 1, scaledMatrices[i * 6 + 4]);
764 blockMatrix->SetBlock(1, 2, scaledMatrices[i * 6 + 5]);
765 blockMatrices.push_back(blockMatrix);
768 std::shared_ptr<M4> matrix(
new M4(3, 2, 4, 6));
769 matrix->SetBlock(0, 0, blockMatrices[0]);
770 matrix->SetBlock(0, 1, blockMatrices[1]);
771 matrix->SetBlock(1, 0, blockMatrices[2]);
772 matrix->SetBlock(1, 1, blockMatrices[3]);
773 matrix->SetBlock(2, 0, blockMatrices[4]);
774 matrix->SetBlock(2, 1, blockMatrices[5]);
776 double b_buf[] = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12};
782 double expected_buf[] = {1828, 1906, 2764, 2842, 5572, 5650,
783 6508, 6586, 9316, 9394, 10252, 10330};
785 BOOST_CHECK_EQUAL(result, expected_result);
791 double expected_buf[] = {-1827, -1904, -2761, -2838, -5567, -5644,
792 -6501, -6578, -9307, -9384, -10241, -10318};
795 BOOST_CHECK_EQUAL(expected_result, result);
800 result = result - (*matrix) * b;
802 double expected_buf[] = {-1827, -1904, -2761, -2838, -5567, -5644,
803 -6501, -6578, -9307, -9384, -10241, -10318};
806 BOOST_CHECK_EQUAL(expected_result, result);
unsigned int GetRows() const
BOOST_AUTO_TEST_CASE(TestEqualSizedBlockConstruction)
NekMatrix< double > InnerMatrix
The above copyright notice and this permission notice shall be included.
NekMatrix< DenseMatrix, BlockMatrixTag > BlockMatrix
NekMatrix< DenseMatrix, ScaledMatrixTag > ScaledMatrix
NekMatrix< InnerMatrixType, BlockMatrixTag > Transpose(NekMatrix< InnerMatrixType, BlockMatrixTag > &rhs)