JSONDataSetWriter: support array selection

IO/Export/vtkJSONDataSetWriter.cxx

@@ -39,6 +39,8 @@ vtkJSONDataSetWriter::vtkJSONDataSetWriter()
 {
   this->Archiver = vtkArchiver::New();
   this->ValidStringCount = 1;
+  this->GetPointArraySelection()->SetUnknownArraySetting(1);
+  this->GetCellArraySelection()->SetUnknownArraySetting(1);
 }
 
 //------------------------------------------------------------------------------
@@ -94,6 +96,23 @@ std::string vtkJSONDataSetWriter::WriteDataSetAttributes(
       continue;
     }
 
+    if (std::string(className) == "pointData")
+    {
+      if (!this->GetPointArraySelection()->ArrayIsEnabled(field->GetName()))
+      {
+        vtkDebugMacro("Skipping writing point array " << field->GetName());
+        continue;
+      }
+    }
+    else if (std::string(className) == "cellData")
+    {
+      if (!this->GetCellArraySelection()->ArrayIsEnabled(field->GetName()))
+      {
+        vtkDebugMacro("Skipping cell array " << field->GetName());
+        continue;
+      }
+    }
+
     if (nbArrayWritten)
     {
       jsonSnippet << ",\n";

IO/Export/vtkJSONDataSetWriter.h

@@ -32,6 +32,9 @@
 
 #include <string> // std::string used as parameters in a few methods
 
+#include "vtkDataArraySelection.h" // Instantiated
+#include "vtkNew.h"                // Used for DataArray selection
+
 VTK_ABI_NAMESPACE_BEGIN
 class vtkDataSet;
 class vtkDataArray;
@@ -118,6 +121,16 @@ public:
   vtkGetObjectMacro(Archiver, vtkArchiver);
   ///@}
 
+  /**
+   * Specify which point arrays should be written.
+   */
+  vtkGetObjectMacro(PointArraySelection, vtkDataArraySelection);
+
+  /**
+   * Specify which cell arrays should be written.
+   */
+  vtkGetObjectMacro(CellArraySelection, vtkDataArraySelection);
+
   void Write(vtkDataSet*);
 
   bool IsDataSetValid() { return this->ValidDataSet; }
@@ -139,6 +152,9 @@ protected:
 private:
   vtkJSONDataSetWriter(const vtkJSONDataSetWriter&) = delete;
   void operator=(const vtkJSONDataSetWriter&) = delete;
+
+  vtkNew<vtkDataArraySelection> PointArraySelection;
+  vtkNew<vtkDataArraySelection> CellArraySelection;
 };
 
 VTK_ABI_NAMESPACE_END