overlay_next_lattice.hh

Go to the documentation of this file.

/* SPDX-FileCopyrightText: 2024 Blender Authors
 *
 * SPDX-License-Identifier: GPL-2.0-or-later */
 
#pragma once
 
#include "draw_cache_impl.hh"
#include "draw_common_c.hh"
#include "overlay_next_private.hh"
 
#include "ED_lattice.hh"
 
namespace blender::draw::overlay {
 
class Lattices {
 private:
  PassMain ps_ = {"Lattice"};
 
  PassMain::Sub *lattice_ps_;
  PassMain::Sub *edit_lattice_wire_ps_;
  PassMain::Sub *edit_lattice_point_ps_;
 
 public:
  void begin_sync(Resources &res, const State &state)
  {
    const DRWState pass_state = DRW_STATE_WRITE_COLOR | DRW_STATE_WRITE_DEPTH |
                                DRW_STATE_DEPTH_LESS_EQUAL;
 
    auto create_sub_pass = [&](const char *name, GPUShader *shader, bool add_weight_tex) {
      PassMain::Sub &sub_pass = ps_.sub(name);
      sub_pass.state_set(pass_state, state.clipping_plane_count);
      sub_pass.shader_set(shader);
      sub_pass.bind_ubo("globalsBlock", &res.globals_buf);
      if (add_weight_tex) {
        sub_pass.bind_texture("weightTex", &res.weight_ramp_tx);
      }
      return &sub_pass;
    };
 
    ps_.init();
    edit_lattice_wire_ps_ = create_sub_pass(
        "edit_lattice_wire", res.shaders.lattice_wire.get(), true);
    edit_lattice_point_ps_ = create_sub_pass(
        "edit_lattice_points", res.shaders.lattice_points.get(), false);
    lattice_ps_ = create_sub_pass("lattice", res.shaders.extra_wire_object.get(), false);
    res.select_bind(ps_);
  }
 
  void edit_object_sync(Manager &manager, const ObjectRef &ob_ref, Resources &res)
  {
    ResourceHandle res_handle = manager.unique_handle(ob_ref);
    {
      gpu::Batch *geom = DRW_cache_lattice_wire_get(ob_ref.object, true);
      edit_lattice_wire_ps_->draw(geom, res_handle, res.select_id(ob_ref).get());
    }
    {
      gpu::Batch *geom = DRW_cache_lattice_vert_overlay_get(ob_ref.object);
      edit_lattice_point_ps_->draw(geom, res_handle, res.select_id(ob_ref).get());
    }
  }
 
  void object_sync(Manager &manager, const ObjectRef &ob_ref, Resources &res, const State &state)
  {
    gpu::Batch *geom = DRW_cache_lattice_wire_get(ob_ref.object, false);
    if (geom) {
      const float4 &color = res.object_wire_color(ob_ref, state);
      float4x4 draw_mat(ob_ref.object->object_to_world().ptr());
      for (int i : IndexRange(3)) {
        draw_mat[i][3] = color[i];
      }
      draw_mat[3][3] = 0.0f /* No stipples. */;
      ResourceHandle res_handle = manager.resource_handle(ob_ref, &draw_mat, nullptr, nullptr);
      lattice_ps_->draw(geom, res_handle, res.select_id(ob_ref).get());
    }
  }
 
  void draw(Framebuffer &framebuffer, Manager &manager, View &view)
  {
    GPU_framebuffer_bind(framebuffer);
    manager.submit(ps_, view);
  }
};
}  // namespace blender::draw::overlay