particleTransform.h

Go to the documentation of this file.

/*
 * -------------------------------------------
 * Copyright (c) 2021 - 2026 Prashant K. Jha
 * -------------------------------------------
 * PeriDEM https://github.com/prashjha/PeriDEM
 *
 * Distributed under the Boost Software License, Version 1.0. (See accompanying
 * file LICENSE)
 */
 
#pragma once
 
#include "util/point.h"  // definition of Point
#include "util/transformationFunctions.h"
#include "util/io.h"
 
#include <cstring> // string and size_t type
#include <vector>
 
namespace geom {
 
struct ParticleTransform {
 
  util::Point d_translation;
 
  util::Point d_axis;
 
  double d_theta;
 
  double d_scale;
 
  util::Point d_rotationPoint;
 
  ParticleTransform()
      : d_translation(util::Point()), d_axis(util::Point()),
        d_theta(0.), d_scale(1.), d_rotationPoint(util::Point()){};
 
  ParticleTransform(util::Point translate, util::Point axis, double theta,
                    double scale = 1.)
      : d_translation(translate), d_axis(axis / axis.length()), d_theta(theta),
        d_scale(scale), d_rotationPoint(util::Point()){};
 
  ParticleTransform(util::Point translate, util::Point axis, double theta, double scale,
                    util::Point rotation_point)
      : d_translation(translate), d_axis(axis / axis.length()), d_theta(theta),
        d_scale(scale), d_rotationPoint(rotation_point){};
 
  ParticleTransform(const ParticleTransform &t)
      : d_translation(t.d_translation), d_axis(t.d_axis),
        d_theta(t.d_theta), d_scale(t.d_scale), d_rotationPoint(t.d_rotationPoint){};
 
  util::Point apply(const util::Point &v) const {
    return d_rotationPoint +
           d_scale * util::rotate(v - d_rotationPoint, d_theta, d_axis) +
           d_translation;
  };
 
  std::string printStr(int nt = 0, int lvl = 0) const {
 
    auto tabS = util::io::getTabS(nt);
    std::ostringstream oss;
    oss << tabS << "------- ParticleTransform --------" << std::endl << std::endl;
    oss << tabS << "Scale = " << d_scale << std::endl;
    oss << tabS << "Angle = " << d_theta << std::endl;
    oss << tabS << "Translation = " << d_translation.printStr() << std::endl;
    oss << tabS << "Rotation pivot = " << d_rotationPoint.printStr() << std::endl;
    oss << tabS << "Axis = " << d_axis.printStr() << std::endl;
    oss << tabS << std::endl;
 
    return oss.str();
  }
 
  void print(int nt = 0, int lvl = 0) const { std::cout << printStr(nt, lvl); }
};
 
} // namespace geom