File: TestProjectSphereFilter.cxx

package info (click to toggle)
vtk7 7.1.1%2Bdfsg1-12
  • links: PTS, VCS
  • area: main
  • in suites: buster
  • size: 125,776 kB
  • sloc: cpp: 1,539,582; ansic: 106,521; python: 78,038; tcl: 47,013; xml: 8,142; yacc: 5,040; java: 4,439; perl: 3,132; lex: 1,926; sh: 1,500; makefile: 122; objc: 83
file content (127 lines) | stat: -rw-r--r-- 3,959 bytes parent folder | download | duplicates (3)
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
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
/*=========================================================================

  Program:   Visualization Toolkit
  Module:    DistributedData.cxx

  Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
  All rights reserved.
  See Copyright.txt or http://www.kitware.com/Copyright.htm for details.

     This software is distributed WITHOUT ANY WARRANTY; without even
     the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
     PURPOSE.  See the above copyright notice for more information.

=========================================================================*/

// Test of vtkProjectSphereFilter. It checks the output in here
// and doesn't compare to an image.

#include "vtkArrayCalculator.h"
#include "vtkCellData.h"
#include "vtkDataArray.h"
#include "vtkNew.h"
#include "vtkPointData.h"
#include "vtkPointDataToCellData.h"
#include "vtkProjectSphereFilter.h"
#include "vtkSphereSource.h"
#include "vtkTestUtilities.h"

namespace
{
  // returns true for success. type is Point or Cell to give
  // feedback for errors in the passed in array.
  bool CheckFieldData(const char* type, vtkDataArray* array, int component,
                      double minValue, double maxValue)
  {
    double values[3];
    for(vtkIdType i=0;i<array->GetNumberOfTuples();i++)
    {
      array->GetTuple(i, values);
      for(int j=0;j<3;j++)
      {
        if(j == component)
        {
          if(values[j] < minValue || values[j] > maxValue)
          {
            vtkGenericWarningMacro("Array type " << type << " with name "
                                   << array->GetName() << " has bad value of " << values[j]
                                   << " but should be between " << minValue << " and " << maxValue);
            return false;
          }
        }
        else if(values[j] < -0.001 || values[j] > 0.001)
        {
          vtkGenericWarningMacro("Array type " << type << " with name " << array->GetName()
                                 << " should be 0 but has value of " << values[j]);
          return false;
        }
      }
    }
    return true;
  }
}

int TestProjectSphereFilter(int vtkNotUsed(argc), char* [])
{
  int numberOfErrors = 0;

  vtkNew<vtkSphereSource> sphere;
  sphere->SetRadius(1);
  sphere->SetCenter(0, 0, 0);
  sphere->SetThetaResolution(50);
  sphere->SetPhiResolution(50);

  vtkNew<vtkArrayCalculator> calculator;
  calculator->SetInputConnection(sphere->GetOutputPort());
  calculator->SetResultArrayName("result");
  calculator->SetFunction("-coordsY*iHat/sqrt(coordsY^2+coordsX^2)+coordsX*jHat/sqrt(coordsY^2+coordsX^2)");
  calculator->SetAttributeModeToUsePointData();
  calculator->AddCoordinateScalarVariable("coordsX", 0);
  calculator->AddCoordinateScalarVariable("coordsY", 1);

  vtkNew<vtkProjectSphereFilter> projectSphere;
  projectSphere->SetCenter(0, 0, 0);
  projectSphere->SetInputConnection(calculator->GetOutputPort());

  vtkNew<vtkPointDataToCellData> pointToCell;
  pointToCell->SetInputConnection(projectSphere->GetOutputPort());
  pointToCell->PassPointDataOn();

  pointToCell->Update();

  vtkDataSet* grid = pointToCell->GetOutput();
  if(grid->GetNumberOfPoints() != 2450)
  {
    vtkGenericWarningMacro(
      "Wrong number of points. There are " << grid->GetNumberOfPoints() <<
      " but should be 2450.");
    numberOfErrors++;
  }
  if(grid->GetNumberOfCells() != 4700)
  {
    vtkGenericWarningMacro(
      "Wrong number of cells. There are " << grid->GetNumberOfCells() <<
      " but should be 4700.");
    numberOfErrors++;
  }

  if(CheckFieldData("Point", grid->GetPointData()->GetArray("result"), 0, .99, 1.01)
     == false)
  {
    numberOfErrors++;
  }

  if(CheckFieldData("Point", grid->GetPointData()->GetArray("Normals"), 2, .99, 1.01)
     == false)
  {
    numberOfErrors++;
  }

  if(CheckFieldData("Cell", grid->GetCellData()->GetArray("Normals"), 2, .99, 1.01)
     == false)
  {
    numberOfErrors++;
  }

  return numberOfErrors;
}