gpu_select_next.cc

Go to the documentation of this file.

/* SPDX-FileCopyrightText: 2017 Blender Authors. All rights reserved.
 *
 * SPDX-License-Identifier: GPL-2.0-or-later */
 
#include <cfloat>
 
#include "BLI_rect.h"
#include "BLI_span.hh"
 
#include "GPU_select.hh"
 
#include "gpu_select_private.hh"
 
struct GPUSelectNextState {
  GPUSelectBuffer *buffer;
  rcti rect;
  uint hits;
  eGPUSelectMode mode;
};
 
static GPUSelectNextState g_state = {};
 
void gpu_select_next_begin(GPUSelectBuffer *buffer, const rcti *input, eGPUSelectMode mode)
 
{
  g_state.buffer = buffer;
  g_state.rect = *input;
  g_state.mode = mode;
}
 
int gpu_select_next_get_pick_area_center()
{
  BLI_assert(BLI_rcti_size_x(&g_state.rect) == BLI_rcti_size_y(&g_state.rect));
  return BLI_rcti_size_x(&g_state.rect) / 2;
}
 
eGPUSelectMode gpu_select_next_get_mode()
{
  return g_state.mode;
}
 
void gpu_select_next_set_result(GPUSelectResult *hit_buf, uint hit_len)
{
  const int old_size = g_state.buffer->storage.size();
  g_state.buffer->storage.resize(old_size + hit_len);
  blender::MutableSpan<GPUSelectResult> hit_results =
      g_state.buffer->storage.as_mutable_span().slice(old_size, hit_len);
  const blender::Span<GPUSelectResult> hits(hit_buf, hit_len);
 
  /* TODO(fclem): There might be some conversion to do to align to the other APIs output. */
  switch (g_state.mode) {
    case eGPUSelectMode::GPU_SELECT_ALL:
      hit_results.copy_from(hits);
      break;
    case eGPUSelectMode::GPU_SELECT_NEAREST_FIRST_PASS:
      hit_results.copy_from(hits);
      break;
    case eGPUSelectMode::GPU_SELECT_NEAREST_SECOND_PASS:
      hit_results.copy_from(hits);
      break;
    case eGPUSelectMode::GPU_SELECT_PICK_ALL:
      hit_results.copy_from(hits);
      break;
    case eGPUSelectMode::GPU_SELECT_PICK_NEAREST:
      hit_results.copy_from(hits);
      break;
  }
 
  g_state.hits = hit_len;
}
 
uint gpu_select_next_end()
{
  return g_state.hits;
}