File: vtkMultiBlockDataSetSerDesHelper.cxx

package info (click to toggle)
vtk9 9.5.2%2Bdfsg3-6
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 205,984 kB
  • sloc: cpp: 2,336,570; ansic: 327,116; python: 111,200; yacc: 4,104; java: 3,977; sh: 3,032; xml: 2,771; perl: 2,189; lex: 1,787; makefile: 181; javascript: 165; objc: 153; tcl: 59
file content (119 lines) | stat: -rw-r--r-- 3,884 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
// SPDX-FileCopyrightText: Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
// SPDX-License-Identifier: BSD-3-Clause
#include "vtkDeserializer.h"
#include "vtkInformation.h"
#include "vtkMultiBlockDataSet.h"
#include "vtkSerializer.h"

// clang-format off
#include "vtk_nlohmannjson.h"
#include VTK_NLOHMANN_JSON(json.hpp)
// clang-format on

extern "C"
{
  /**
   * Register the (de)serialization handlers of vtkMultiBlockDataSet
   * @param ser   a vtkSerializer instance
   * @param deser a vtkDeserializer instance
   */
  int RegisterHandlers_vtkMultiBlockDataSetSerDesHelper(void* ser, void* deser, void* invoker);
}

static nlohmann::json Serialize_vtkMultiBlockDataSet(
  vtkObjectBase* objectBase, vtkSerializer* serializer)
{
  using json = nlohmann::json;
  json state;
  auto object = vtkMultiBlockDataSet::SafeDownCast(objectBase);
  if (auto f = serializer->GetHandler(typeid(vtkMultiBlockDataSet::Superclass)))
  {
    state = f(object, serializer);
  }
  state["SuperClassNames"].push_back("vtkDataObjectTree");
  auto& dst = state["Blocks"] = json::array();
  for (unsigned int i = 0; i < object->GetNumberOfBlocks(); ++i)
  {
    const char* name = nullptr;
    if (object->HasMetaData(i))
    {
      auto* metadata = object->GetMetaData(i);
      if (metadata->Has(vtkCompositeDataSet::NAME()))
      {
        name = metadata->Get(vtkCompositeDataSet::NAME());
      }
    }
    dst.emplace_back(json{ { "Name", name ? name : "" },
      { "DataObject", serializer->SerializeJSON(object->GetBlock(i)) } });
  }
  return state;
}

static void Deserialize_vtkMultiBlockDataSet(
  const nlohmann::json& state, vtkObjectBase* objectBase, vtkDeserializer* deserializer)
{
  using json = nlohmann::json;
  auto object = vtkMultiBlockDataSet::SafeDownCast(objectBase);
  if (auto f = deserializer->GetHandler(typeid(vtkMultiBlockDataSet::Superclass)))
  {
    f(state, object, deserializer);
  }

  {
    const auto iter = state.find("Blocks");
    if (iter != state.end() && iter->is_array())
    {
      const auto& blocks = iter->get<json::array_t>();
      if (blocks.size() < object->GetNumberOfBlocks())
      {
        // shrink if the required number of blocks is smaller than current allocation.
        object->SetNumberOfBlocks(static_cast<unsigned int>(blocks.size()));
      }
      unsigned int i = 0;
      for (const auto& blockState : blocks)
      {
        const std::string name = blockState.at("Name").get<std::string>();
        const auto& block = blockState.at("DataObject");
        if (!block.empty())
        {
          auto* context = deserializer->GetContext();
          const auto identifier = block.at("Id").get<vtkTypeUInt32>();
          auto subObject = context->GetObjectAtId(identifier);
          deserializer->DeserializeJSON(identifier, subObject);
          object->SetBlock(i, vtkDataObject::SafeDownCast(subObject));
        }
        else
        {
          object->SetBlock(i, nullptr);
        }
        object->GetMetaData(i)->Set(vtkCompositeDataSet::NAME(), name);
        i++;
      }
    }
  }
}

int RegisterHandlers_vtkMultiBlockDataSetSerDesHelper(
  void* ser, void* deser, void* vtkNotUsed(invoker))
{
  int success = 0;
  if (auto* asObjectBase = static_cast<vtkObjectBase*>(ser))
  {
    if (auto* serializer = vtkSerializer::SafeDownCast(asObjectBase))
    {
      serializer->RegisterHandler(typeid(vtkMultiBlockDataSet), Serialize_vtkMultiBlockDataSet);
      success = 1;
    }
  }
  if (auto* asObjectBase = static_cast<vtkObjectBase*>(deser))
  {
    if (auto* deserializer = vtkDeserializer::SafeDownCast(asObjectBase))
    {
      deserializer->RegisterHandler(typeid(vtkMultiBlockDataSet), Deserialize_vtkMultiBlockDataSet);
      deserializer->RegisterConstructor(
        "vtkMultiBlockDataSet", []() { return vtkMultiBlockDataSet::New(); });
      success = 1;
    }
  }
  return success;
}