particleTransform.h

Go to the documentation of this file.

/*
 * -------------------------------------------
 * Copyright (c) 2021 - 2024 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/transformation.h"
 
#include <cstring> // string and size_t type
#include <vector>
 
namespace particle {
 
struct ParticleTransform {
 
  util::Point d_translation;
 
  util::Point d_axis;
 
  double d_theta;
 
  double d_scale;
 
  ParticleTransform()
      : d_translation(util::Point()), d_axis(util::Point()),
        d_theta(0.), d_scale(1.){};
 
  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){};
 
  ParticleTransform(const ParticleTransform &t)
      : d_translation(t.d_translation), d_axis(t.d_axis),
        d_theta(t.d_theta), d_scale(t.d_scale){};
 
  util::Point apply(const util::Point &v) const {
 
    return d_translation +
           d_scale * util::rotate(v, d_theta, d_axis);
 
    // return d_translation + d_scale * util::Point(v.d_x, -v.d_y, 0.);
    // return d_translation + v;
  };
 
  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 << "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 particle