add_wrapped_position.cpp

Go to the documentation of this file.

#include "add_wrapped_position.h"

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

AddWrappedPosition::AddWrappedPosition(
  const ElPtr &elem)
{
  ext_generator = elem;
}

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

void AddWrappedPosition::compute_impl(
  Json &data,
  JsonToVoidFunc check_required_keys,
  JsonToBoolFunc check_optional_keys)
{
  check_required_keys({"xu", "yu", "zu"});

  auto el_box = ext_generator->get_element();

  el_box->check_required_keys({"lo_x", "lo_y", "lo_z", "hi_x", "hi_y", "hi_z"});

  auto &box = el_box->get_data();

  ArrayXd offset(3);
  offset << box["lo_x"], box["lo_y"], box["lo_z"];

  ArrayXd length(3);
  length << box["hi_x"], box["hi_y"], box["hi_z"];
  length -= offset;

  for (auto &d : data)
  {
    ArrayXd unwrapped(3);
    unwrapped << d["xu"], d["yu"], d["zu"];

    ArrayXd tmp_wrapped = unwrapped;

    tmp_wrapped -= offset;
    tmp_wrapped /= length;

    tmp_wrapped = -tmp_wrapped.floor();
    tmp_wrapped *= length;
    tmp_wrapped += unwrapped;

    d["x"] = tmp_wrapped(0);
    d["y"] = tmp_wrapped(1);
    d["z"] = tmp_wrapped(2);
  }
}