为什么我的vtkContourFilter没有输出?

时间:2017-11-18 12:01:59

标签: c++ vtk

我正在用C ++编写一个VTK程序来处理有限元分析结果。为了显示isosurfaces,我正在尝试vtkContourFilter,但是在这个过滤器后我什么都没得到。

所以我写了一个简单版本:自定义阅读器生成一个2单元非结构化网格,后面的vtkContourFilter提取等值面。我的标量数据是点数据。

我打算得到等值面,但我仍然得到零输出。我想知道我错过了什么。

我的节目:

#include <vtkSmartPointer.h>
#include <vtkUnstructuredGrid.h>
#include <vtkPoints.h>
#include <vtkPointData.h>
#include <vtkCellArray.h>
#include <vtkIdList.h>
#include <vtkDoubleArray.h>
#include <vtkPointData.h>
#include <vtkUnstructuredGridReader.h>
#include <vtkInformation.h>
#include <vtkInformationVector.h>
#include <vtkClipDataSet.h>
#include <vtkPlane.h>
#include <vtkContourFilter.h>
#include <iostream>
using namespace std;

// This is a fake reader that outputs an unstructured mesh
class VTKIOLEGACY_EXPORT FakeUnstructuredGridReader : public vtkUnstructuredGridReader {
public:
    static FakeUnstructuredGridReader* New() { return new FakeUnstructuredGridReader(); }
    vtkTypeMacro(FakeUnstructuredGridReader, vtkUnstructuredGridReader);
    void PrintSelf(ostream& os, vtkIndent indent) VTK_OVERRIDE {}

protected:
    FakeUnstructuredGridReader() { this->SetNumberOfInputPorts(0); }// No input ports
    // This method passes a 2-cell unstructured grid with 1 set of scalar point data
    int RequestData(vtkInformation*, vtkInformationVector**, vtkInformationVector* outputVector) VTK_OVERRIDE {
        static const double rawPoints[15] = { 0, 0, 0, 1, 0, 0, 0, 1, 0, 0, 0, 1, 1, 1, 1 };
        static const vtkIdType rawIndices[8] = { 0, 1, 2, 3, 1, 2, 3, 4};
        static const double rawScalar[5] = { 5, 10, 20, 15, 50 };
        static const char* scalarName = "Stress";
        // Build Points
        auto points = vtkSmartPointer<vtkPoints>::New();
        for (int i = 0; i < 15; i += 3)
            points->InsertNextPoint(rawPoints + i);
        // Build Cells
        auto cells = vtkSmartPointer<vtkCellArray>::New();
        // Build Cell 1
        auto indices1 = vtkSmartPointer<vtkIdList>::New();
        for (int i = 0; i < 4; ++i)
            indices1->InsertNextId(rawIndices[i]);      
        cells->InsertNextCell(indices1);
        // Build Cell 2
        auto indices2 = vtkSmartPointer<vtkIdList>::New();
        for (int i = 0; i < 4; ++i)
            indices2->InsertNextId(rawIndices[i + 4]);
        cells->InsertNextCell(indices2);
        // Scalar Data
        auto dataArray = vtkSmartPointer<vtkDoubleArray>::New();
        dataArray->SetName(scalarName);
        dataArray->SetNumberOfComponents(1);
        for (int i = 0; i < 5; ++i)
            dataArray->InsertNextValue(rawScalar[i]);
        // Point Data
        auto pointData = vtkSmartPointer<vtkPointData>::New();
        pointData->AddArray(dataArray);
        // Pass Data
        vtkInformation* outInfo = outputVector->GetInformationObject(0);
        vtkUnstructuredGrid *output = vtkUnstructuredGrid::SafeDownCast(outInfo->Get(vtkDataObject::DATA_OBJECT()));
        output->SetPoints(points);
        output->SetCells(VTK_TETRA, cells);
        output->SetFieldData(pointData);
        return 1;
    }
};


int main() {
    // Reader
    auto fakeReader = vtkSmartPointer<FakeUnstructuredGridReader>::New();

    // Isosurface Filter
    auto isoFilter = vtkSmartPointer<vtkContourFilter>::New();
    isoFilter->SetInputConnection(fakeReader->GetOutputPort());
    isoFilter->GenerateValues(8, 0.0, 50.0);
    isoFilter->Update();

    cout << fakeReader->GetOutput()->GetNumberOfPoints() << endl;
    cout << fakeReader->GetOutput()->GetNumberOfCells() << endl;
    cout << fakeReader->GetOutput()->GetFieldData()->GetNumberOfArrays() << endl;
    cout << isoFilter->GetOutput()->GetNumberOfPoints() << endl;
    cout << isoFilter->GetOutput()->GetNumberOfCells() << endl;
    cout << isoFilter->GetOutput()->GetFieldData()->GetNumberOfArrays() << endl;

    system("pause");
    return 0;
}

其输出:

5
2
1
0
0
1

1 个答案:

答案 0 :(得分:1)

在函数RequestData()中,首先需要将标量值与output->GetPointData()->AddArray(dataArray)的数据集的点相关联,然后使用output->GetPointData()->SetActiveScalars(scalarName)激活标量字段,以便vtkContourFilter将是能够选择它来计算轮廓。您可以通过执行output->GetPointData()->SetScalars(dataArray)来执行这两项操作。请注意,使用此解决方案,您不再需要分配pointData变量。

简而言之,你可以改变:

int RequestData(vtkInformation*, vtkInformationVector**, vtkInformationVector* outputVector) VTK_OVERRIDE {
  /* body of the function here */

  // old version
  // output->SetFieldData(pointData);
  // new version
  output->GetPointData()->SetScalars(dataArray);
  return 1;
}

您可能需要相应地修改打印。

// old version:
// cout << fakeReader->GetOutput()->GetFieldData()->GetNumberOfArrays() << endl;
// new version:
cout << fakeReader->GetOutput()->GetPointData()->GetNumberOfArrays() << endl;

// old version
// cout << isoFilter->GetOutput()->GetFieldData()->GetNumberOfArrays() << endl;
// new version
cout << isoFilter->GetOutput()->GetPointData()->GetNumberOfArrays() << endl;