// SPDX-FileCopyrightText: Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
// SPDX-FileCopyrightText: Copyright (c) Kitware, Inc.
// SPDX-License-Identifier: BSD-3-Clause
/**
 * @class   vtkXMLPartitionedDataSetReader
 * @brief   Reader for partitioned datasets
 *
 * vtkXMLPartitionedDataSetReader reads the VTK XML partitioned dataset file
 * format. XML partitioned dataset files are meta-files that point to a list
 * of serial VTK XML files. When reading in parallel, it will distribute
 * sub-blocks among processors. If the number of sub-blocks is less than
 * the number of processors, some processors will not have any sub-blocks
 * for that block. If the number of sub-blocks is larger than the
 * number of processors, each processor will possibly have more than
 * 1 sub-block.
 */

#ifndef vtkXMLPartitionedDataSetReader_h
#define vtkXMLPartitionedDataSetReader_h

#include "vtkIOXMLModule.h" // For export macro
#include "vtkXMLCompositeDataReader.h"

VTK_ABI_NAMESPACE_BEGIN
class vtkMultiBlockDataSet;

class VTKIOXML_EXPORT vtkXMLPartitionedDataSetReader : public vtkXMLCompositeDataReader
{
public:
  static vtkXMLPartitionedDataSetReader* New();
  vtkTypeMacro(vtkXMLPartitionedDataSetReader, vtkXMLCompositeDataReader);
  void PrintSelf(ostream& os, vtkIndent indent) override;

protected:
  vtkXMLPartitionedDataSetReader();
  ~vtkXMLPartitionedDataSetReader() override;

  // Get the name of the data set being read.
  const char* GetDataSetName() override;

  // Read the XML element for the subtree of a composite dataset.
  // dataSetIndex is used to rank the leaf nodes in an inorder traversal.
  void ReadComposite(vtkXMLDataElement* element, vtkCompositeDataSet* composite,
    const char* filePath, unsigned int& dataSetIndex) override;

  int FillOutputPortInformation(int, vtkInformation* info) override;

  /**
   * Create the meta-data from the partitioned dataset collection from the file.
   */
  void CreateMetaData(vtkXMLDataElement* ePrimary) override;

  /**
   * Recursively synchronize the data array selection of the reader for the file specified in the
   * XML element.
   */
  void SyncCompositeDataArraySelections(vtkCompositeDataSet* composite, vtkXMLDataElement* element,
    const std::string& filePath) override;

private:
  vtkXMLPartitionedDataSetReader(const vtkXMLPartitionedDataSetReader&) = delete;
  void operator=(const vtkXMLPartitionedDataSetReader&) = delete;
};

VTK_ABI_NAMESPACE_END
#endif
