1 //============================================================================
2 //  Copyright (c) Kitware, Inc.
3 //  All rights reserved.
4 //  See LICENSE.txt for details.
5 //  This software is distributed WITHOUT ANY WARRANTY; without even
6 //  the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
7 //  PURPOSE.  See the above copyright notice for more information.
8 //
9 //  Copyright 2014 National Technology & Engineering Solutions of Sandia, LLC (NTESS).
10 //  Copyright 2014 UT-Battelle, LLC.
11 //  Copyright 2014 Los Alamos National Security.
12 //
13 //  Under the terms of Contract DE-NA0003525 with NTESS,
14 //  the U.S. Government retains certain rights in this software.
15 //
16 //  Under the terms of Contract DE-AC52-06NA25396 with Los Alamos National
17 //  Laboratory (LANL), the U.S. Government retains certain rights in
18 //  this software.
19 //============================================================================
20 
21 #include <vtkm/CellShape.h>
22 #include <vtkm/VectorAnalysis.h>
23 #include <vtkm/cont/ArrayHandle.h>
24 #include <vtkm/cont/CellSetStructured.h>
25 #include <vtkm/cont/DataSet.h>
26 #include <vtkm/cont/DataSetFieldAdd.h>
27 #include <vtkm/cont/DynamicArrayHandle.h>
28 
29 #include <vtkm/cont/MultiBlock.h>
30 #include <vtkm/cont/serial/DeviceAdapterSerial.h>
31 #include <vtkm/exec/ConnectivityStructured.h>
32 
33 #include <vtkm/cont/testing/MakeTestDataSet.h>
34 #include <vtkm/cont/testing/Testing.h>
35 #include <vtkm/filter/CellAverage.h>
36 
37 
38 template <typename T>
MultiBlockBuilder(std::size_t BlockNum,std::string FieldName)39 vtkm::cont::MultiBlock MultiBlockBuilder(std::size_t BlockNum, std::string FieldName)
40 {
41   vtkm::cont::DataSetBuilderUniform dataSetBuilder;
42   vtkm::cont::DataSet dataSet;
43   vtkm::cont::DataSetFieldAdd dsf;
44 
45   vtkm::Vec<T, 2> origin(0);
46   vtkm::Vec<T, 2> spacing(1);
47   vtkm::cont::MultiBlock Blocks;
48   for (vtkm::Id BlockId = 0; BlockId < static_cast<vtkm::Id>(BlockNum); BlockId++)
49   {
50     vtkm::Id2 dimensions((BlockId + 2) * (BlockId + 2), (BlockId + 2) * (BlockId + 2));
51 
52     if (FieldName == "cellvar")
53     {
54       vtkm::Id numCells = (dimensions[0] - 1) * (dimensions[1] - 1);
55 
56       std::vector<T> varC2D(static_cast<std::size_t>(numCells));
57       for (vtkm::Id i = 0; i < numCells; i++)
58       {
59         varC2D[static_cast<std::size_t>(i)] = static_cast<T>(BlockId * i);
60       }
61       dataSet = dataSetBuilder.Create(vtkm::Id2(dimensions[0], dimensions[1]),
62                                       vtkm::Vec<T, 2>(origin[0], origin[1]),
63                                       vtkm::Vec<T, 2>(spacing[0], spacing[1]));
64       dsf.AddCellField(dataSet, "cellvar", varC2D);
65     }
66 
67     if (FieldName == "pointvar")
68     {
69       vtkm::Id numPoints = dimensions[0] * dimensions[1];
70       std::vector<T> varP2D(static_cast<std::size_t>(numPoints));
71       for (vtkm::Id i = 0; i < numPoints; i++)
72       {
73         varP2D[static_cast<std::size_t>(i)] = static_cast<T>(BlockId);
74       }
75       dataSet = dataSetBuilder.Create(vtkm::Id2(dimensions[0], dimensions[1]),
76                                       vtkm::Vec<T, 2>(origin[0], origin[1]),
77                                       vtkm::Vec<T, 2>(spacing[0], spacing[1]));
78       dsf.AddPointField(dataSet, "pointvar", varP2D);
79     }
80 
81     Blocks.AddBlock(dataSet);
82   }
83   return Blocks;
84 }
85 template <typename D>
Result_Verify(const vtkm::cont::MultiBlock & Result,D & Filter,const vtkm::cont::MultiBlock & Blocks,std::string FieldName)86 void Result_Verify(const vtkm::cont::MultiBlock& Result,
87                    D& Filter,
88                    const vtkm::cont::MultiBlock& Blocks,
89                    std::string FieldName)
90 {
91   VTKM_TEST_ASSERT(Result.GetNumberOfBlocks() == Blocks.GetNumberOfBlocks(),
92                    "result block number incorrect");
93   const std::string outputFieldName = Filter.GetOutputFieldName();
94   for (vtkm::Id j = 0; j < Result.GetNumberOfBlocks(); j++)
95   {
96     Filter.SetActiveField(FieldName);
97     vtkm::cont::DataSet BlockResult = Filter.Execute(Blocks.GetBlock(j));
98 
99     VTKM_TEST_ASSERT(Result.GetBlock(j).GetField(outputFieldName).GetData().GetNumberOfValues() ==
100                        BlockResult.GetField(outputFieldName).GetData().GetNumberOfValues(),
101                      "result vectors' size incorrect");
102 
103     vtkm::cont::ArrayHandle<vtkm::Id> MBlockArray;
104     Result.GetBlock(j).GetField(outputFieldName).GetData().CopyTo(MBlockArray);
105     vtkm::cont::ArrayHandle<vtkm::Id> SDataSetArray;
106     BlockResult.GetField(outputFieldName).GetData().CopyTo(SDataSetArray);
107 
108     for (vtkm::Id i = 0;
109          i < Result.GetBlock(j).GetField(outputFieldName).GetData().GetNumberOfValues();
110          i++)
111     {
112       VTKM_TEST_ASSERT(MBlockArray.GetPortalConstControl().Get(i) ==
113                          SDataSetArray.GetPortalConstControl().Get(i),
114                        "result values incorrect");
115     }
116   }
117   return;
118 }
119 
TestMultiBlockFilters()120 void TestMultiBlockFilters()
121 {
122   std::size_t BlockNum = 7;
123   vtkm::cont::MultiBlock result;
124   vtkm::cont::MultiBlock Blocks;
125 
126   Blocks = MultiBlockBuilder<vtkm::Id>(BlockNum, "pointvar");
127   vtkm::filter::CellAverage cellAverage;
128   cellAverage.SetOutputFieldName("average");
129   cellAverage.SetActiveField("pointvar");
130   result = cellAverage.Execute(Blocks);
131   Result_Verify(result, cellAverage, Blocks, std::string("pointvar"));
132 }
133 
UnitTestMultiBlockFilters(int,char * [])134 int UnitTestMultiBlockFilters(int, char* [])
135 {
136   return vtkm::cont::testing::Testing::Run(TestMultiBlockFilters);
137 }
138