Method RequestData

Filters/Statistics/vtkExtractStatisticalModelTables.cxx:136–177

}

int vtkExtractStatisticalModelTables::RequestData(vtkInformation* vtkNotUsed(request),
  vtkInformationVector** inInfoVec, vtkInformationVector* outInfoVec)
{
  auto* model = vtkStatisticalModel::GetData(inInfoVec[0]);
  auto* pdc = vtkPartitionedDataSetCollection::GetData(inInfoVec[0]);
  auto* out = vtkPartitionedDataSetCollection::GetData(outInfoVec);

  if (pdc && (pdc->GetNumberOfPartitionedDataSets() == 0 || !pdc->GetDataAssembly()))
  {
    // Empty input ⇒ empty output
    out->Initialize();
    return 1;
  }

  vtkNew<vtkDataAssembly> resultAssy;

  if (model)
  {
    AddModelToAssembly(out, model, resultAssy, /*root node of model*/ 0);
  }
  else if (pdc)
  {
    // Copy the input's structure so all node IDs match, but remove dataset references:
    auto* assemblyIn = pdc->GetDataAssembly();
    resultAssy->DeepCopy(assemblyIn);
    resultAssy->RemoveAllDataSetIndices(0, /*traverse_subtree*/ true);
    // Create a visitor for the input collection.
    vtkNew<ModelExtractor> extractor;
    extractor->Initialize(pdc, out, resultAssy);
    // Add new node IDs for statistical models by traversing the input:
    assemblyIn->Visit(extractor);
  }
  else
  {
    auto* data = vtkDataObject::GetData(inInfoVec[0]);
    vtkErrorMacro("Unhandled input type \"" << (data ? data->GetClassName() : "null") << "\".");
    return 0;
  }
  out->SetDataAssembly(resultAssy);

  return 1;
}

VTK_ABI_NAMESPACE_END