overlay_next_metaball.hh

Go to the documentation of this file.

/* SPDX-FileCopyrightText: 2023 Blender Authors
 *
 * SPDX-License-Identifier: GPL-2.0-or-later */
 
#pragma once
 
#include "overlay_next_private.hh"
#include "overlay_shader_shared.h"
 
#include "ED_mball.hh"
 
namespace blender::draw::overlay {
 
class Metaballs {
  using SphereOutlineInstanceBuf = ShapeInstanceBuf<BoneInstanceData>;
 
 private:
  const SelectionType selection_type_;
 
  PassSimple ps_ = {"MetaBalls"};
 
  SphereOutlineInstanceBuf circle_buf_ = {selection_type_, "metaball_data_buf"};
 
 public:
  Metaballs(const SelectionType selection_type) : selection_type_(selection_type){};
 
  void begin_sync()
  {
    circle_buf_.clear();
  }
 
  void edit_object_sync(const ObjectRef &ob_ref, Resources &res)
  {
    const Object *ob = ob_ref.object;
    const MetaBall *mb = static_cast<MetaBall *>(ob->data);
 
    const float *color;
    const float *col_radius = res.theme_settings.color_mball_radius;
    const float *col_radius_select = res.theme_settings.color_mball_radius_select;
    const float *col_stiffness = res.theme_settings.color_mball_stiffness;
    const float *col_stiffness_select = res.theme_settings.color_mball_stiffness_select;
 
    int elem_num = 0;
    LISTBASE_FOREACH (MetaElem *, ml, mb->editelems) {
      const bool is_selected = (ml->flag & SELECT) != 0;
      const bool is_scale_radius = (ml->flag & MB_SCALE_RAD) != 0;
      const float stiffness_radius = ml->rad * atanf(ml->s) * 2.0f / math::numbers::pi;
      const float3 position = float3(&ml->x);
 
      const select::ID radius_id = res.select_id(ob_ref, MBALLSEL_RADIUS | elem_num);
      color = (is_selected && is_scale_radius) ? col_radius_select : col_radius;
      circle_buf_.append({ob->object_to_world(), position, ml->rad, color}, radius_id);
 
      const select::ID stiff_id = res.select_id(ob_ref, MBALLSEL_STIFF | elem_num);
      color = (is_selected && !is_scale_radius) ? col_stiffness_select : col_stiffness;
      circle_buf_.append({ob->object_to_world(), position, stiffness_radius, color}, stiff_id);
      elem_num += 1 << 16;
    }
  }
 
  void object_sync(const ObjectRef &ob_ref, Resources &res, const State &state)
  {
    const Object *ob = ob_ref.object;
    const MetaBall *mb = static_cast<MetaBall *>(ob->data);
 
    const float4 &color = res.object_wire_color(ob_ref, state);
    const select::ID select_id = res.select_id(ob_ref);
 
    LISTBASE_FOREACH (MetaElem *, ml, &mb->elems) {
      const float3 position = float3(&ml->x);
      /* Draw radius only. */
      circle_buf_.append({ob->object_to_world(), position, ml->rad, color}, select_id);
    }
  }
 
  void end_sync(Resources &res, ShapeCache &shapes, const State &state)
  {
    ps_.init();
    ps_.state_set(DRW_STATE_WRITE_COLOR | DRW_STATE_WRITE_DEPTH | DRW_STATE_DEPTH_LESS_EQUAL,
                  state.clipping_plane_count);
    /* NOTE: Use armature sphere outline shader to have perspective correct outline instead of
     * just a circle facing the camera. */
    ps_.shader_set(res.shaders.armature_sphere_outline.get());
    ps_.bind_ubo("globalsBlock", &res.globals_buf);
    res.select_bind(ps_);
 
    circle_buf_.end_sync(ps_, shapes.metaball_wire_circle.get());
  }
 
  void draw(Framebuffer &framebuffer, Manager &manager, View &view)
  {
    GPU_framebuffer_bind(framebuffer);
    manager.submit(ps_, view);
  }
};
 
}  // namespace blender::draw::overlay