[Bf-blender-cvs] [9eed28fd56e] temp-image-engine-refactor: Split image engine into spaces and drawing modes.

Jeroen Bakker noreply at git.blender.org
Fri Nov 5 12:44:28 CET 2021


Commit: 9eed28fd56e0df99ab2375f9d6ceddcf84380af3
Author: Jeroen Bakker
Date:   Fri Nov 5 11:52:11 2021 +0100
Branches: temp-image-engine-refactor
https://developer.blender.org/rB9eed28fd56e0df99ab2375f9d6ceddcf84380af3

Split image engine into spaces and drawing modes.

===================================================================

A	source/blender/draw/engines/image/image_drawing_mode.hh
M	source/blender/draw/engines/image/image_engine.cc
M	source/blender/draw/engines/image/image_engine.h
M	source/blender/draw/engines/image/image_private.hh
A	source/blender/draw/engines/image/image_space_image.hh
A	source/blender/draw/engines/image/image_space_node.hh

===================================================================

diff --git a/source/blender/draw/engines/image/image_drawing_mode.hh b/source/blender/draw/engines/image/image_drawing_mode.hh
new file mode 100644
index 00000000000..5446a5603ab
--- /dev/null
+++ b/source/blender/draw/engines/image/image_drawing_mode.hh
@@ -0,0 +1,154 @@
+/*
+ * This program is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU General Public License
+ * as published by the Free Software Foundation; either version 2
+ * of the License, or (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software Foundation,
+ * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
+ *
+ * Copyright 2021, Blender Foundation.
+ */
+
+/** \file
+ * \ingroup draw_engine
+ */
+
+#pragma once
+
+#include "image_private.hh"
+
+namespace blender::draw::image_engine {
+
+class DefaultDrawingMode : public AbstractDrawingMode {
+ private:
+  DRWPass *create_image_pass() const
+  {
+    /* Write depth is needed for background overlay rendering. Near depth is used for
+     * transparency checker and Far depth is used for indicating the image size. */
+    DRWState state = static_cast<DRWState>(DRW_STATE_WRITE_COLOR | DRW_STATE_WRITE_DEPTH |
+                                           DRW_STATE_DEPTH_ALWAYS | DRW_STATE_BLEND_ALPHA_PREMUL);
+    return DRW_pass_create("Image", state);
+  }
+
+  void add_to_shgroup(AbstractSpaceAccessor *space,
+                      DRWShadingGroup *grp,
+                      const Image *image,
+                      const ImBuf *ibuf) const
+  {
+    float image_mat[4][4];
+
+    const DRWContextState *draw_ctx = DRW_context_state_get();
+    const ARegion *region = draw_ctx->region;
+    space->get_image_mat(ibuf, region, image_mat);
+
+    GPUBatch *geom = DRW_cache_quad_get();
+
+    const bool is_tiled_texture = image && image->source == IMA_SRC_TILED;
+    if (is_tiled_texture) {
+      const float translate_x = image_mat[3][0];
+      const float translate_y = image_mat[3][1];
+      LISTBASE_FOREACH (ImageTile *, tile, &image->tiles) {
+        const int tile_x = ((tile->tile_number - 1001) % 10);
+        const int tile_y = ((tile->tile_number - 1001) / 10);
+        image_mat[3][0] = (float)tile_x + translate_x;
+        image_mat[3][1] = (float)tile_y + translate_y;
+        DRW_shgroup_call_obmat(grp, geom, image_mat);
+      }
+    }
+    else {
+      DRW_shgroup_call_obmat(grp, geom, image_mat);
+    }
+  }
+
+ public:
+  void cache_init(IMAGE_Data *vedata) const override
+  {
+    IMAGE_PassList *psl = vedata->psl;
+
+    psl->image_pass = create_image_pass();
+  }
+
+  void cache_image(AbstractSpaceAccessor *space,
+                   IMAGE_Data *vedata,
+                   Image *image,
+                   ImageUser *iuser,
+                   ImBuf *image_buffer) const override
+  {
+    IMAGE_PassList *psl = vedata->psl;
+    IMAGE_StorageList *stl = vedata->stl;
+    IMAGE_PrivateData *pd = stl->pd;
+
+    GPUTexture *tex_tile_data = nullptr;
+    space->get_gpu_textures(
+        image, iuser, image_buffer, &pd->texture, &pd->owns_texture, &tex_tile_data);
+    if (pd->texture == nullptr) {
+      return;
+    }
+    const bool is_tiled_texture = tex_tile_data != nullptr;
+
+    ShaderParameters sh_params;
+    sh_params.use_premul_alpha = BKE_image_has_gpu_texture_premultiplied_alpha(image,
+                                                                               image_buffer);
+    const DRWContextState *draw_ctx = DRW_context_state_get();
+    const Scene *scene = draw_ctx->scene;
+    if (scene->camera && scene->camera->type == OB_CAMERA) {
+      Camera *camera = static_cast<Camera *>(scene->camera->data);
+      copy_v2_fl2(sh_params.far_near, camera->clip_end, camera->clip_start);
+    }
+    space->get_shader_parameters(sh_params, image_buffer, is_tiled_texture);
+
+    GPUShader *shader = IMAGE_shader_image_get(is_tiled_texture);
+    DRWShadingGroup *shgrp = DRW_shgroup_create(shader, psl->image_pass);
+    if (is_tiled_texture) {
+      DRW_shgroup_uniform_texture_ex(shgrp, "imageTileArray", pd->texture, GPU_SAMPLER_DEFAULT);
+      DRW_shgroup_uniform_texture(shgrp, "imageTileData", tex_tile_data);
+    }
+    else {
+      DRW_shgroup_uniform_texture_ex(shgrp, "imageTexture", pd->texture, GPU_SAMPLER_DEFAULT);
+    }
+    DRW_shgroup_uniform_vec2_copy(shgrp, "farNearDistances", sh_params.far_near);
+    DRW_shgroup_uniform_vec4_copy(shgrp, "color", ShaderParameters::color);
+    DRW_shgroup_uniform_vec4_copy(shgrp, "shuffle", sh_params.shuffle);
+    DRW_shgroup_uniform_int_copy(shgrp, "drawFlags", sh_params.flags);
+    DRW_shgroup_uniform_bool_copy(shgrp, "imgPremultiplied", sh_params.use_premul_alpha);
+
+    add_to_shgroup(space, shgrp, image, image_buffer);
+  }
+
+  void draw_finish(IMAGE_Data *vedata) const override
+  {
+    IMAGE_StorageList *stl = vedata->stl;
+    IMAGE_PrivateData *pd = stl->pd;
+
+    if (pd->texture && pd->owns_texture) {
+      GPU_texture_free(pd->texture);
+      pd->owns_texture = false;
+    }
+    pd->texture = nullptr;
+  }
+
+  void draw_scene(IMAGE_Data *vedata) const override
+  {
+    IMAGE_PassList *psl = vedata->psl;
+    IMAGE_PrivateData *pd = vedata->stl->pd;
+
+    DefaultFramebufferList *dfbl = DRW_viewport_framebuffer_list_get();
+    GPU_framebuffer_bind(dfbl->default_fb);
+    static float clear_col[4] = {0.0f, 0.0f, 0.0f, 0.0f};
+    GPU_framebuffer_clear_color_depth(dfbl->default_fb, clear_col, 1.0);
+
+    DRW_view_set_active(pd->view);
+    DRW_draw_pass(psl->image_pass);
+    DRW_view_set_active(nullptr);
+  }
+};
+
+}  // namespace blender::draw::image_engine
+
diff --git a/source/blender/draw/engines/image/image_engine.cc b/source/blender/draw/engines/image/image_engine.cc
index 11fe7e55d10..6ccf20ab339 100644
--- a/source/blender/draw/engines/image/image_engine.cc
+++ b/source/blender/draw/engines/image/image_engine.cc
@@ -17,7 +17,7 @@
  */
 
 /** \file
- * \ingroup draw_editors
+ * \ingroup draw_engine
  *
  * Draw engine to draw the Image/UV editor
  */
@@ -41,215 +41,14 @@
 
 #include "GPU_batch.h"
 
+#include "image_drawing_mode.hh"
 #include "image_engine.h"
 #include "image_private.hh"
+#include "image_space_image.hh"
+#include "image_space_node.hh"
 
 namespace blender::draw::image_engine {
 
-/* Forward declarations. */
-class AbstractSpaceAccessor;
-static void image_cache_image(
-    AbstractSpaceAccessor &space, IMAGE_Data *vedata, Image *image, ImageUser *iuser, ImBuf *ibuf);
-
-#define IMAGE_DRAW_FLAG_SHOW_ALPHA (1 << 0)
-#define IMAGE_DRAW_FLAG_APPLY_ALPHA (1 << 1)
-#define IMAGE_DRAW_FLAG_SHUFFLING (1 << 2)
-#define IMAGE_DRAW_FLAG_DEPTH (1 << 3)
-#define IMAGE_DRAW_FLAG_DO_REPEAT (1 << 4)
-#define IMAGE_DRAW_FLAG_USE_WORLD_POS (1 << 5)
-
-struct ShaderParameters {
-  constexpr static float color[4] = {1.0f, 1.0f, 1.0f, 1.0f};
-
-  int flags = 0;
-  float shuffle[4];
-  float far_near[2];
-  bool use_premul_alpha = false;
-
-  ShaderParameters()
-  {
-    copy_v4_fl(shuffle, 1.0f);
-    copy_v2_fl2(far_near, 100.0f, 0.0f);
-  }
-
-  void set_far_near(Camera *camera)
-  {
-    far_near[1] = camera->clip_start;
-    far_near[0] = camera->clip_end;
-  }
-};
-
-class AbstractSpaceAccessor {
- public:
-  virtual void release_buffer(Image *image, ImBuf *ibuf, void *lock) = 0;
-  virtual Image *get_image(Main *bmain) = 0;
-  virtual ImageUser *get_image_user() = 0;
-  virtual ImBuf *acquire_image_buffer(Image *image, void **lock) = 0;
-  virtual void get_shader_parameters(ShaderParameters &r_shader_parameters,
-                                     ImBuf *ibuf,
-                                     bool is_tiled) = 0;
-
-  virtual std::optional<DRWView *> create_view(const ARegion *UNUSED(region))
-  {
-    return std::nullopt;
-  }
-};
-
-class SpaceImageAccessor : public AbstractSpaceAccessor {
-  SpaceImage *sima;
-
- public:
-  SpaceImageAccessor(SpaceImage *sima) : sima(sima)
-  {
-  }
-
-  Image *get_image(Main *UNUSED(bmain)) override
-  {
-    return ED_space_image(sima);
-  }
-
-  ImageUser *get_image_user() override
-  {
-    return &sima->iuser;
-  }
-
-  ImBuf *acquire_image_buffer(Image *UNUSED(image), void **lock) override
-  {
-    return ED_space_image_acquire_buffer(sima, lock, 0);
-  }
-
-  void release_buffer(Image *UNUSED(image), ImBuf *ibuf, void *lock) override
-  {
-    ED_space_image_release_buffer(sima, ibuf, lock);
-  }
-
-  void get_shader_parameters(ShaderParameters &r_shader_parameters,
-                             ImBuf *ibuf,
-                             bool is_tiled) override
-  {
-    const int sima_flag = sima->flag & ED_space_image_get_display_channel_mask(ibuf);
-    const bool do_repeat = (!is_tiled) && ((sima->flag & SI_DRAW_TILE) != 0);
-    SET_FLAG_FROM_TEST(r_shader_parameters.flags, do_repeat, IMAGE_DRAW_FLAG_DO_REPEAT);
-    SET_FLAG_FROM_TEST(r_shader_parameters.flags, is_tiled, IMAGE_DRAW_FLAG_USE_WORLD_POS);
-    if ((sima_flag & SI_USE_ALPHA) != 0) {
-      /* Show RGBA */
-      r_shader_parameters.flags |= IMAGE_DRAW_FLAG_SHOW_ALPHA | IMAGE_DRAW_FLAG_APPLY_ALPHA;
-    }
-    else if ((sima_flag & SI_SHOW_ALPHA) != 0) {
-      r_shader_parameters.flags |= IMAGE_DRAW_FLAG_SHUFFLING;
-      copy_v4_fl4(r_shader_parameters.shuffle, 0.0f, 0.0f, 0.0f, 1.0f);
-    }
-    else if ((sima_flag & SI_SHOW_ZBUF) != 0) {
-      r_shader_parameters.flags |= IMAGE_DRAW_FLAG_DEPTH | IMAGE_DRAW_FLAG_SHUFFLING;
-      copy_v4_fl4(r_shader_parameters.shuffle, 1.0f, 0.0f, 0.0f, 0.0f);
-    }
-    else if ((sima_flag & SI_SHOW_R) != 0) {
-      r_shader_parameters.flags |= IMAGE_DRAW_FLAG_SHUFFLING;
-      if (IMB_alpha_affects_rgb(ibuf)) {
-        r_shader_parameters.flags |= IMAGE_DRAW_FLAG_APPLY_ALPHA;
-      }
-      copy_v4_fl4(r_shader_parameters.shuffle, 1.0f, 0.0f, 0.0f, 0.0f);
-    }
-    else if ((sima_flag & SI_SHOW_G) != 0) {
-      r_shader_parameters.flags |=

@@ Diff output truncated at 10240 characters. @@



More information about the Bf-blender-cvs mailing list