Method RequestData

Infovis/Core/vtkAddMembershipArray.cxx:78–233

------------------------------------------------------------------------------

//------------------------------------------------------------------------------
int vtkAddMembershipArray::RequestData(
  vtkInformation*, vtkInformationVector** inputVector, vtkInformationVector* outputVector)
{
  vtkInformation* inInfo = inputVector[0]->GetInformationObject(0);
  vtkDataObject* input = inInfo->Get(vtkDataObject::DATA_OBJECT());
  vtkSelection* inputSelection = vtkSelection::GetData(inputVector[1]);
  vtkAnnotationLayers* inputAnnotations = vtkAnnotationLayers::GetData(inputVector[2]);
  vtkInformation* outputInfo = outputVector->GetInformationObject(0);
  vtkDataObject* output = outputInfo->Get(vtkDataObject::DATA_OBJECT());
  vtkGraph* graph = vtkGraph::SafeDownCast(output);
  vtkTable* table = vtkTable::SafeDownCast(output);

  output->ShallowCopy(input);

  if (!inputSelection)
  {
    if (!this->InputArrayName || !this->InputValues)
      return 1;

    vtkDataSetAttributes* ds = nullptr;
    switch (this->FieldType)
    {
      case vtkAddMembershipArray::VERTEX_DATA:
        if (graph)
        {
          ds = graph->GetVertexData();
        }
        break;
      case vtkAddMembershipArray::EDGE_DATA:
        if (graph)
        {
          ds = graph->GetEdgeData();
        }
        break;
      case vtkAddMembershipArray::ROW_DATA:
        if (table)
        {
          ds = table->GetRowData();
        }
        break;
    }

    if (!ds)
    {
      vtkErrorMacro("Unsupported input field type.");
      return 0;
    }

    vtkIntArray* vals = vtkIntArray::New();
    vals->SetNumberOfTuples(ds->GetNumberOfTuples());
    vals->SetNumberOfComponents(1);
    vals->SetName(this->OutputArrayName);
    vals->FillComponent(0, 0);

    vtkAbstractArray* inputArray = ds->GetAbstractArray(this->InputArrayName);
    if (inputArray && this->InputValues)
    {
      for (vtkIdType i = 0; i < inputArray->GetNumberOfTuples(); ++i)