gen_wall_box_plane.hpp

Go to the documentation of this file.

#pragma once
 
#include "scene.hpp"
#include "shape_plane.hpp"
#include "wall.hpp"
 
namespace netdem {
 
class WallBoxPlane {
private:
  Plane p_mx, p_px, p_my, p_py, p_mz, p_pz;
  Wall w_mx, w_px, w_my, w_py, w_mz, w_pz;
 
public:
  WallBoxPlane(double len_x, double len_y, double len_z, double center_x,
               double center_y, double center_z) {
    p_mx = Plane(center_x - 0.5 * len_x, center_y, center_z, 1, 0, 0);
    p_px = Plane(center_x + 0.5 * len_x, center_y, center_z, -1, 0, 0);
    p_my = Plane(center_x, center_y - 0.5 * len_y, center_z, 0, 1, 0);
    p_py = Plane(center_x, center_y + 0.5 * len_y, center_z, 0, -1, 0);
    p_mz = Plane(center_x, center_y, center_z - 0.5 * len_z, 0, 0, 1);
    p_pz = Plane(center_x, center_y, center_z + 0.5 * len_z, 0, 0, -1);
 
    double sqrt_2_by_2 = 2.0 * std::sqrt(2.0);
    p_mx.SetExtent(std::max(len_y, len_z) * sqrt_2_by_2);
    p_px.SetExtent(std::max(len_y, len_z) * sqrt_2_by_2);
    p_my.SetExtent(std::max(len_x, len_z) * sqrt_2_by_2);
    p_py.SetExtent(std::max(len_x, len_z) * sqrt_2_by_2);
    p_mz.SetExtent(std::max(len_x, len_y) * sqrt_2_by_2);
    p_pz.SetExtent(std::max(len_x, len_y) * sqrt_2_by_2);
 
    w_mx = Wall(&p_mx);
    w_px = Wall(&p_px);
    w_my = Wall(&p_my);
    w_py = Wall(&p_py);
    w_mz = Wall(&p_mz);
    w_pz = Wall(&p_pz);
  }
 
  VecXT<Shape *> GetShapes() {
    VecXT<Shape *> plane_list;
    plane_list.emplace_back(&p_mx);
    plane_list.emplace_back(&p_px);
    plane_list.emplace_back(&p_my);
    plane_list.emplace_back(&p_py);
    plane_list.emplace_back(&p_mz);
    plane_list.emplace_back(&p_pz);
    return plane_list;
  }
 
  VecXT<Wall *> GetWall() {
    VecXT<Wall *> wall_list;
    wall_list.emplace_back(&w_mx);
    wall_list.emplace_back(&w_px);
    wall_list.emplace_back(&w_my);
    wall_list.emplace_back(&w_py);
    wall_list.emplace_back(&w_mz);
    wall_list.emplace_back(&w_pz);
    return wall_list;
  }
 
  void ImportToScene(Scene *scene) {
    auto shape_ptr_list_local = GetShapes();
    auto wall_ptr_list_local = GetWall();
 
    auto shape_ptr_list = scene->InsertShape(shape_ptr_list_local);
 
    for (int i = 0; i < 6; i++) {
      wall_ptr_list_local[i]->SetShape(shape_ptr_list[i]);
      scene->InsertWall(wall_ptr_list_local[i]);
      wall_ptr_list_local[i]->SetShape(shape_ptr_list_local[i]);
    }
  }
};
 
} // namespace netdem