gizmo_library_presets.c

Go to the documentation of this file.

/*
 * 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.
 */
 
#include "BLI_math.h"
 
#include "DNA_object_types.h"
 
#include "BKE_context.h"
 
#include "GPU_matrix.h"
#include "GPU_select.h"
 
#include "DEG_depsgraph.h"
 
#include "WM_types.h"
 
#include "ED_view3d.h"
 
/* own includes */
#include "ED_gizmo_library.h"     /* own include */
#include "gizmo_library_intern.h" /* own include */
 
/* TODO, this is to be used by RNA. might move to ED_gizmo_library */
 
static void single_axis_convert(int src_axis,
                                const float src_mat[4][4],
                                int dst_axis,
                                float dst_mat[4][4])
{
  copy_m4_m4(dst_mat, src_mat);
  if (src_axis == dst_axis) {
    return;
  }
 
  float rotmat[3][3];
  mat3_from_axis_conversion_single(src_axis, dst_axis, rotmat);
  transpose_m3(rotmat);
  mul_m4_m4m3(dst_mat, src_mat, rotmat);
}
 
static void ed_gizmo_draw_preset_geometry(const struct wmGizmo *gz,
                                          const float mat[4][4],
                                          int select_id,
                                          const GizmoGeomInfo *info)
{
  const bool is_select = (select_id != -1);
  const bool is_highlight = is_select && (gz->state & WM_GIZMO_STATE_HIGHLIGHT) != 0;
 
  float color[4];
  gizmo_color_get(gz, is_highlight, color);
 
  if (is_select) {
    GPU_select_load_id(select_id);
  }
 
  GPU_matrix_push();
  GPU_matrix_mul(mat);
  wm_gizmo_geometryinfo_draw(info, is_select, color);
  GPU_matrix_pop();
 
  if (is_select) {
    GPU_select_load_id(-1);
  }
}
 
void ED_gizmo_draw_preset_box(const struct wmGizmo *gz, float mat[4][4], int select_id)
{
  ed_gizmo_draw_preset_geometry(gz, mat, select_id, &wm_gizmo_geom_data_cube);
}
 
void ED_gizmo_draw_preset_arrow(const struct wmGizmo *gz, float mat[4][4], int axis, int select_id)
{
  float mat_rotate[4][4];
  single_axis_convert(OB_POSZ, mat, axis, mat_rotate);
  ed_gizmo_draw_preset_geometry(gz, mat_rotate, select_id, &wm_gizmo_geom_data_arrow);
}
 
void ED_gizmo_draw_preset_circle(const struct wmGizmo *gz,
                                 float mat[4][4],
                                 int axis,
                                 int select_id)
{
  float mat_rotate[4][4];
  single_axis_convert(OB_POSZ, mat, axis, mat_rotate);
  ed_gizmo_draw_preset_geometry(gz, mat_rotate, select_id, &wm_gizmo_geom_data_dial);
}
 
void ED_gizmo_draw_preset_facemap(
    const bContext *C, const struct wmGizmo *gz, Object *ob, const int facemap, int select_id)
{
  /* Dependency graph is supposed to be evaluated prior to draw. */
  Depsgraph *depsgraph = CTX_data_expect_evaluated_depsgraph(C);
  const bool is_select = (select_id != -1);
  const bool is_highlight = is_select && (gz->state & WM_GIZMO_STATE_HIGHLIGHT) != 0;
 
  float color[4];
  gizmo_color_get(gz, is_highlight, color);
 
  if (is_select) {
    GPU_select_load_id(select_id);
  }
 
  GPU_matrix_push();
  GPU_matrix_mul(ob->obmat);
  ED_draw_object_facemap(depsgraph, ob, color, facemap);
  GPU_matrix_pop();
 
  if (is_select) {
    GPU_select_load_id(-1);
  }
}