add_child_ids.cpp

Go to the documentation of this file.

#include "add_child_ids.h"
#include "../utils/map_to_index.h"
#include "../utils/runtime_error.h"

namespace ut = utils;

/* ------------------------------------------------------------------ */

AddChildIDs::AddChildIDs(
  const ElPtr &elem,
  const Str &child_name_,
  const Str &key_for_parent_id_)
{
  ext_generator = elem;
  child_name = child_name_;
  key_for_parent_id = key_for_parent_id_;
}

/* ------------------------------------------------------------------ */

void AddChildIDs::compute_impl(
  Json &data,
  JsonToVoidFunc check_required_keys,
  JsonToBoolFunc check_optional_keys)
{
  check_required_keys("id");

  auto el_children = ext_generator->get_element();

  el_children->check_required_keys({"id", key_for_parent_id});

  auto key_for_child_ids = child_name + "-ids";

  if (check_optional_keys(key_for_child_ids))
  {
    ut::runtime_error("Key '" + key_for_child_ids + "' already exists");
  }

  auto id2index = ut::map_to_index(data, "id");

  for (const auto &child : el_children->get_data())
  {
    auto mol_index = id2index[child[key_for_parent_id]];
    data[mol_index][key_for_child_ids].push_back(child["id"]);
  }
}