mirror of
https://github.com/OpenFOAM/ThirdParty-6.git
synced 2025-12-08 06:57:43 +00:00
68 lines
2.3 KiB
C++
68 lines
2.3 KiB
C++
// 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);
|
|
}
|
|
}
|