mirror of
https://github.com/OpenFOAM/ThirdParty-6.git
synced 2025-12-08 06:57:43 +00:00
ParaView-5.0.1: Added the source-tree to ThirdParty-dev and patched as described in the README file
Resolves bug-report http://bugs.openfoam.org/view.php?id=2098
This commit is contained in:
@ -0,0 +1,67 @@
|
||||
// Adaptor for getting Fortran simulation code into ParaView CoProcessor.
|
||||
|
||||
// CoProcessor specific headers
|
||||
#include "vtkCPDataDescription.h"
|
||||
#include "vtkCPInputDataDescription.h"
|
||||
#include "vtkCPProcessor.h"
|
||||
#include "vtkCPPythonScriptPipeline.h"
|
||||
#include "vtkSmartPointer.h"
|
||||
#include "vtkDoubleArray.h"
|
||||
#include "vtkPointData.h"
|
||||
#include "vtkImageData.h"
|
||||
|
||||
// Fortran specific header
|
||||
#include "vtkCPPythonAdaptorAPI.h"
|
||||
|
||||
|
||||
// These will be called from the Fortran "glue" code"
|
||||
// Completely dependent on data layout, structured vs. unstructured, etc.
|
||||
// since VTK/ParaView uses different internal layouts for each.
|
||||
|
||||
// Creates the data container for the CoProcessor.
|
||||
extern "C" void createcpimagedata_(int* nxstart, int* nxend, int* nx,
|
||||
int* ny, int* nz)
|
||||
{
|
||||
if (!vtkCPPythonAdaptorAPI::GetCoProcessorData())
|
||||
{
|
||||
vtkGenericWarningMacro("Unable to access CoProcessorData.");
|
||||
return;
|
||||
}
|
||||
|
||||
// The simulation grid is a 3-dimensional topologically and geometrically
|
||||
// regular grid. In VTK/ParaView, this is considered an image data set.
|
||||
vtkSmartPointer<vtkImageData> grid = vtkSmartPointer<vtkImageData>::New();
|
||||
|
||||
grid->SetExtent(*nxstart-1, *nxend-1, 0, *ny-1, 0, *nz-1);
|
||||
|
||||
// Name should be consistent between here, Fortran and Python client script.
|
||||
vtkCPPythonAdaptorAPI::GetCoProcessorData()->GetInputDescriptionByName("input")->SetGrid(grid);
|
||||
vtkCPPythonAdaptorAPI::GetCoProcessorData()->GetInputDescriptionByName("input")->SetWholeExtent(0, *nx-1, 0, *ny-1, 0, *nz-1);
|
||||
}
|
||||
|
||||
// Add field(s) to the data container.
|
||||
// Separate from above because this will be dynamic, grid is static.
|
||||
// By hand name mangling for fortran.
|
||||
extern "C" void addfield_(double* scalars, char* name)
|
||||
{
|
||||
vtkCPInputDataDescription* idd =
|
||||
vtkCPPythonAdaptorAPI::GetCoProcessorData()->GetInputDescriptionByName("input");
|
||||
|
||||
vtkImageData* Image = vtkImageData::SafeDownCast(idd->GetGrid());
|
||||
|
||||
if (!Image)
|
||||
{
|
||||
vtkGenericWarningMacro("No adaptor grid to attach field data to.");
|
||||
return;
|
||||
}
|
||||
|
||||
// field name must match that in the fortran code.
|
||||
if (idd->IsFieldNeeded(name))
|
||||
{
|
||||
vtkSmartPointer<vtkDoubleArray> field = vtkSmartPointer<vtkDoubleArray>::New();
|
||||
field->SetNumberOfComponents(2);
|
||||
field->SetName(name);
|
||||
field->SetArray(scalars, 2*Image->GetNumberOfPoints(), 1);
|
||||
Image->GetPointData()->AddArray(field);
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user