test/source/blender/editors/util/ed_util_ops.cc

/* SPDX-FileCopyrightText: 2023 Blender Authors
 *
 * SPDX-License-Identifier: GPL-2.0-or-later */

/** \file
 * \ingroup edutil
 *
 * Utility operators for UI data or for the UI to use.
 */

#include <cstring>

#include "DNA_space_types.h"
#include "DNA_windowmanager_types.h"

#include "BLI_fileops.h"
#include "BLI_utildefines.h"
#include "BLI_vector.hh"

#include "BKE_context.hh"
#include "BKE_lib_id.hh"
#include "BKE_lib_override.hh"
#include "BKE_library.hh"
#include "BKE_preview_image.hh"
#include "BKE_report.hh"

#include "ED_asset.hh"
#include "ED_render.hh"
#include "ED_undo.hh"
#include "ED_util.hh"

#include "RNA_access.hh"
#include "RNA_prototypes.hh"

#include "UI_interface.hh"

#include "WM_api.hh"
#include "WM_types.hh"

/* -------------------------------------------------------------------- */
/** \name Context Query Helpers
 * \{ */

blender::Vector<PointerRNA> ED_operator_single_id_from_context_as_vec(const bContext *C)
{
  blender::Vector<PointerRNA> ids;
  PointerRNA idptr = CTX_data_pointer_get_type(C, "id", &RNA_ID);
  if (idptr.data) {
    ids.append(idptr);
  }
  return ids;
}

blender::Vector<PointerRNA> ED_operator_get_ids_from_context_as_vec(const bContext *C)
{
  blender::Vector<PointerRNA> ids;

  /* "selected_ids" context member. */
  CTX_data_selected_ids(C, &ids);
  if (!ids.is_empty()) {
    return ids;
  }

  /* "id" context member. */
  return ED_operator_single_id_from_context_as_vec(C);
}

/** \} */

/* -------------------------------------------------------------------- */
/** \name ID Previews
 * \{ */

static bool lib_id_preview_editing_poll_ex(const ID *id, const char **r_disabled_hint)
{
  if (!id) {
    return false;
  }
  if (!ID_IS_EDITABLE(id)) {
    if (r_disabled_hint) {
      *r_disabled_hint = "Can't edit external library data";
    }
    return false;
  }
  if (ID_IS_OVERRIDE_LIBRARY(id)) {
    if (r_disabled_hint) {
      *r_disabled_hint = "Can't edit previews of overridden library data";
    }
    return false;
  }
  if (!BKE_previewimg_id_get_p(id)) {
    if (r_disabled_hint) {
      *r_disabled_hint = "Data-block does not support previews";
    }
    return false;
  }

  return true;
}

static bool lib_id_preview_editing_poll(bContext *C)
{
  const PointerRNA idptr = CTX_data_pointer_get(C, "id");
  BLI_assert(!idptr.data || RNA_struct_is_ID(idptr.type));

  const ID *id = (ID *)idptr.data;
  const char *disabled_hint = nullptr;
  if (!lib_id_preview_editing_poll_ex(id, &disabled_hint)) {
    CTX_wm_operator_poll_msg_set(C, disabled_hint);
    return false;
  }

  return true;
}

static ID *lib_id_load_custom_preview_id_get(bContext *C, const wmOperator *op)
{
  /* #invoke() gets the ID from context and saves it in the custom data. */
  if (op->customdata) {
    return static_cast<ID *>(op->customdata);
  }

  PointerRNA idptr = CTX_data_pointer_get(C, "id");
  return static_cast<ID *>(idptr.data);
}

static wmOperatorStatus lib_id_load_custom_preview_exec(bContext *C, wmOperator *op)
{
  char filepath[FILE_MAX];

  RNA_string_get(op->ptr, "filepath", filepath);

  if (!BLI_is_file(filepath)) {
    BKE_reportf(op->reports, RPT_ERROR, "File not found '%s'", filepath);
    return OPERATOR_CANCELLED;
  }

  ID *id = lib_id_load_custom_preview_id_get(C, op);
  if (!id) {
    BKE_report(
        op->reports, RPT_ERROR, "Failed to set preview: no ID in context (incorrect context?)");
    return OPERATOR_CANCELLED;
  }

  BKE_previewimg_id_custom_set(id, filepath);

  WM_event_add_notifier(C, NC_ASSET | NA_EDITED, nullptr);

  return OPERATOR_FINISHED;
}

/**
 * Obtain the ID from context, and spawn a File Browser to select the preview image. The
 * File Browser may re-use the Asset Browser under the cursor, and clear the file-list on
 * confirmation, leading to failure to obtain the ID at that point. So get it before spawning the
 * File Browser (store it in the operator custom data).
 */
static wmOperatorStatus lib_id_load_custom_preview_invoke(bContext *C,
                                                          wmOperator *op,
                                                          const wmEvent *event)
{
  op->customdata = lib_id_load_custom_preview_id_get(C, op);
  return WM_operator_filesel(C, op, event);
}

static void ED_OT_lib_id_load_custom_preview(wmOperatorType *ot)
{
  ot->name = "Load Custom Preview";
  ot->description = "Choose an image to help identify the data-block visually";
  ot->idname = "ED_OT_lib_id_load_custom_preview";

  /* API callbacks. */
  ot->poll = lib_id_preview_editing_poll;
  ot->exec = lib_id_load_custom_preview_exec;
  ot->invoke = lib_id_load_custom_preview_invoke;

  /* flags */
  ot->flag = OPTYPE_UNDO | OPTYPE_INTERNAL;

  WM_operator_properties_filesel(ot,
                                 FILE_TYPE_FOLDER | FILE_TYPE_IMAGE,
                                 FILE_SPECIAL,
                                 FILE_OPENFILE,
                                 WM_FILESEL_FILEPATH,
                                 FILE_DEFAULTDISPLAY,
                                 FILE_SORT_DEFAULT);
}

/**
 * Helper for batch editing previews. Gets selected or active IDs from context and calls \a
 * foreach_id for each ID that supports previews.
 */
static void lib_id_batch_edit_previews(bContext *C, blender::FunctionRef<void(ID *)> foreach_id)
{
  blender::Vector<PointerRNA> id_pointers = ED_operator_get_ids_from_context_as_vec(C);
  for (PointerRNA &idptr : id_pointers) {
    ID *id = static_cast<ID *>(idptr.data);

    if (lib_id_preview_editing_poll_ex(id, nullptr)) {
      foreach_id(id);
    }
  }
}

/**
 * Helper for batch editing previews. Check if at least one of the selected or active IDs supports
 * previews, setting a disabled hint if not. Note that only one disabled hint can be set, this
 * simply uses the first one set while polling individual IDs. That's more useful than a generic
 * message still.
 *
 * \param additional_condition: When set, IDs need to additionally pass this check (return true) to
 * be considered as supporting this operation.
 */
static bool lib_id_batch_editing_preview_poll(
    bContext *C,
    blender::FunctionRef<bool(const ID *, const char **r_disabled_hint)> additional_condition =
        nullptr)
{
  blender::Vector<PointerRNA> id_pointers = ED_operator_get_ids_from_context_as_vec(C);
  if (id_pointers.is_empty()) {
    CTX_wm_operator_poll_msg_set(C, "No data-block selected or active");
    return false;
  }

  const char *disabled_hint = nullptr;

  for (const PointerRNA &idptr : id_pointers) {
    const ID *id = static_cast<const ID *>(idptr.data);

    const char *iter_disabled_hint = nullptr;
    if (lib_id_preview_editing_poll_ex(id, &iter_disabled_hint) &&
        (!additional_condition || additional_condition(id, &iter_disabled_hint)))
    {
      /* Operator can run if there's at least one ID supporting previews. */
      return true;
    }

    if (iter_disabled_hint && !disabled_hint) {
      disabled_hint = iter_disabled_hint;
    }
  }

  /* Will only hold the first disabled hint set. That often gives some more specific information,
   * so it's more useful than a generic message. */
  if (disabled_hint) {
    CTX_wm_operator_poll_msg_set(C, disabled_hint);
  }
  else {
    CTX_wm_operator_poll_msg_set(C, "None of the selected data-blocks supports previews");
  }
  return false;
}

static bool lib_id_generate_preview_poll(bContext *C)
{
  return lib_id_batch_editing_preview_poll(C, [](const ID *id, const char **r_disabled_hint) {
    return ED_preview_id_is_supported(id, r_disabled_hint);
  });
}

static wmOperatorStatus lib_id_generate_preview_exec(bContext *C, wmOperator * /*op*/)
{
  using namespace blender::ed;

  ED_preview_kill_jobs(CTX_wm_manager(C), CTX_data_main(C));

  lib_id_batch_edit_previews(C, [&](ID *id) {
    if (ED_preview_id_is_supported(id, nullptr)) {
      PreviewImage *preview = BKE_previewimg_id_get(id);

      if (preview) {
        BKE_previewimg_clear(preview);
      }

      UI_icon_render_id(C, nullptr, id, ICON_SIZE_PREVIEW, true);
    }
  });

  WM_event_add_notifier(C, NC_ASSET | NA_EDITED, nullptr);
  asset::list::storage_tag_main_data_dirty();

  return OPERATOR_FINISHED;
}

static void ED_OT_lib_id_generate_preview(wmOperatorType *ot)
{
  ot->name = "Generate Preview";
  ot->description = "Create an automatic preview for the selected data-block";
  ot->idname = "ED_OT_lib_id_generate_preview";

  /* API callbacks. */
  ot->poll = lib_id_generate_preview_poll;
  ot->exec = lib_id_generate_preview_exec;

  /* flags */
  ot->flag = OPTYPE_INTERNAL | OPTYPE_REGISTER | OPTYPE_UNDO;
}

static bool lib_id_generate_preview_from_object_poll(bContext *C)
{
  /* This already checks if the IDs in context (e.g. selected in the Asset browser) can generate
   * previews... */
  if (!lib_id_batch_editing_preview_poll(C)) {
    return false;
  }

  /* ... but we also need to check this for the active object (since this is what is being
   * rendered). */
  Object *object_to_render = CTX_data_active_object(C);
  if (object_to_render == nullptr) {
    return false;
  }
  const char *disabled_hint = nullptr;
  if (!ED_preview_id_is_supported(&object_to_render->id, &disabled_hint)) {
    CTX_wm_operator_poll_msg_set(C, disabled_hint);
    return false;
  }

  return true;
}

static wmOperatorStatus lib_id_generate_preview_from_object_exec(bContext *C, wmOperator * /*op*/)
{
  using namespace blender::ed;

  ED_preview_kill_jobs(CTX_wm_manager(C), CTX_data_main(C));

  Object *object_to_render = CTX_data_active_object(C);

  lib_id_batch_edit_previews(C, [&](ID *id) {
    BKE_previewimg_id_free(id);

    PreviewImage *preview_image = BKE_previewimg_id_ensure(id);
    UI_icon_render_id_ex(
        C, nullptr, &object_to_render->id, ICON_SIZE_PREVIEW, true, preview_image);
  });

  WM_event_add_notifier(C, NC_ASSET | NA_EDITED, nullptr);
  asset::list::storage_tag_main_data_dirty();

  return OPERATOR_FINISHED;
}

static void ED_OT_lib_id_generate_preview_from_object(wmOperatorType *ot)
{
  ot->name = "Generate Preview from Object";
  ot->description = "Create a preview for this asset by rendering the active object";
  ot->idname = "ED_OT_lib_id_generate_preview_from_object";

  /* API callbacks. */
  ot->poll = lib_id_generate_preview_from_object_poll;
  ot->exec = lib_id_generate_preview_from_object_exec;

  /* flags */
  ot->flag = OPTYPE_INTERNAL | OPTYPE_REGISTER | OPTYPE_UNDO;
}

static bool lib_id_remove_preview_poll(bContext *C)
{
  if (!lib_id_batch_editing_preview_poll(C)) {
    return false;
  }

  bool has_any_removable = false;
  lib_id_batch_edit_previews(C, [&](ID *id) {
    if (BKE_previewimg_id_get(id)) {
      has_any_removable = true;
    }
  });

  if (!has_any_removable) {
    CTX_wm_operator_poll_msg_set(C, "No preview available to remove");
    return false;
  }

  return true;
}

static wmOperatorStatus lib_id_remove_preview_exec(bContext *C, wmOperator * /*op*/)
{
  lib_id_batch_edit_previews(C, [&](ID *id) { BKE_previewimg_id_free(id); });

  WM_event_add_notifier(C, NC_ASSET | NA_EDITED, nullptr);

  return OPERATOR_FINISHED;
}

static void ED_OT_lib_id_remove_preview(wmOperatorType *ot)
{
  /* identifiers */
  ot->name = "Remove Preview";
  ot->description = "Remove the preview of this data-block";
  ot->idname = "ED_OT_lib_id_remove_preview";

  /* API callbacks. */
  ot->poll = lib_id_remove_preview_poll;
  ot->exec = lib_id_remove_preview_exec;

  /* flags */
  ot->flag = OPTYPE_UNDO | OPTYPE_INTERNAL;
}

/** \} */

/* -------------------------------------------------------------------- */
/** \name Generic ID Operators
 * \{ */

static wmOperatorStatus lib_id_fake_user_toggle_exec(bContext *C, wmOperator *op)
{
  PropertyPointerRNA pprop;
  PointerRNA idptr = PointerRNA_NULL;

  UI_context_active_but_prop_get_templateID(C, &pprop.ptr, &pprop.prop);

  if (pprop.prop) {
    idptr = RNA_property_pointer_get(&pprop.ptr, pprop.prop);
  }

  if ((pprop.prop == nullptr) || RNA_pointer_is_null(&idptr) || !RNA_struct_is_ID(idptr.type)) {
    BKE_report(
        op->reports, RPT_ERROR, "Incorrect context for running data-block fake user toggling");
    return OPERATOR_CANCELLED;
  }

  ID *id = (ID *)idptr.data;

  if (!BKE_id_is_editable(CTX_data_main(C), id) ||
      ELEM(GS(id->name), ID_GR, ID_SCE, ID_SCR, ID_TXT, ID_OB, ID_WS))
  {
    BKE_report(op->reports, RPT_ERROR, "Data-block type does not support fake user");
    return OPERATOR_CANCELLED;
  }

  if (ID_FAKE_USERS(id)) {
    id_fake_user_clear(id);
  }
  else {
    id_fake_user_set(id);
  }

  return OPERATOR_FINISHED;
}

static void ED_OT_lib_id_fake_user_toggle(wmOperatorType *ot)
{
  /* identifiers */
  ot->name = "Toggle Fake User";
  ot->description = "Save this data-block even if it has no users";
  ot->idname = "ED_OT_lib_id_fake_user_toggle";

  /* API callbacks. */
  ot->exec = lib_id_fake_user_toggle_exec;

  /* flags */
  ot->flag = OPTYPE_UNDO | OPTYPE_INTERNAL;
}

static wmOperatorStatus lib_id_unlink_exec(bContext *C, wmOperator *op)
{
  PropertyPointerRNA pprop;
  PointerRNA idptr;

  UI_context_active_but_prop_get_templateID(C, &pprop.ptr, &pprop.prop);

  if (pprop.prop) {
    idptr = RNA_property_pointer_get(&pprop.ptr, pprop.prop);
  }

  if ((pprop.prop == nullptr) || RNA_pointer_is_null(&idptr) || !RNA_struct_is_ID(idptr.type)) {
    BKE_report(
        op->reports, RPT_ERROR, "Incorrect context for running data-block fake user toggling");
    return OPERATOR_CANCELLED;
  }

  idptr = {};
  RNA_property_pointer_set(&pprop.ptr, pprop.prop, idptr, nullptr);
  RNA_property_update(C, &pprop.ptr, pprop.prop);

  return OPERATOR_FINISHED;
}

static void ED_OT_lib_id_unlink(wmOperatorType *ot)
{
  /* identifiers */
  ot->name = "Unlink Data-Block";
  ot->description = "Remove a usage of a data-block, clearing the assignment";
  ot->idname = "ED_OT_lib_id_unlink";

  /* API callbacks. */
  ot->exec = lib_id_unlink_exec;

  /* flags */
  ot->flag = OPTYPE_UNDO | OPTYPE_INTERNAL;
}

static bool lib_id_override_editable_toggle_poll(bContext *C)
{
  const PointerRNA id_ptr = CTX_data_pointer_get_type(C, "id", &RNA_ID);
  const ID *id = static_cast<ID *>(id_ptr.data);

  return id && ID_IS_OVERRIDE_LIBRARY_REAL(id) && !ID_IS_LINKED(id);
}

static wmOperatorStatus lib_id_override_editable_toggle_exec(bContext *C, wmOperator * /*op*/)
{
  Main *bmain = CTX_data_main(C);
  const PointerRNA id_ptr = CTX_data_pointer_get_type(C, "id", &RNA_ID);
  ID *id = static_cast<ID *>(id_ptr.data);

  const bool is_system_override = BKE_lib_override_library_is_system_defined(bmain, id);
  if (is_system_override) {
    /* A system override is not editable. Make it an editable (non-system-defined) one. */
    id->override_library->flag &= ~LIBOVERRIDE_FLAG_SYSTEM_DEFINED;
  }
  else {
    /* Reset override, which makes it non-editable (i.e. a system define override). */
    BKE_lib_override_library_id_reset(bmain, id, true);

    WM_event_add_notifier(C, NC_SPACE | ND_SPACE_VIEW3D, nullptr);
    WM_event_add_notifier(C, NC_WINDOW, nullptr);
  }

  WM_main_add_notifier(NC_WM | ND_LIB_OVERRIDE_CHANGED, nullptr);

  return OPERATOR_FINISHED;
}

static void ED_OT_lib_id_override_editable_toggle(wmOperatorType *ot)
{
  /* identifiers */
  ot->name = "Toggle Library Override Editable";
  ot->description = "Set if this library override data-block can be edited";
  ot->idname = "ED_OT_lib_id_override_editable_toggle";

  /* API callbacks. */
  ot->poll = lib_id_override_editable_toggle_poll;
  ot->exec = lib_id_override_editable_toggle_exec;

  /* flags */
  ot->flag = OPTYPE_UNDO | OPTYPE_INTERNAL;
}

/** \} */

/* -------------------------------------------------------------------- */
/** \name General editor utils.
 * \{ */

static wmOperatorStatus ed_flush_edits_exec(bContext *C, wmOperator * /*op*/)
{
  Main *bmain = CTX_data_main(C);
  ED_editors_flush_edits(bmain);
  return OPERATOR_FINISHED;
}

static void ED_OT_flush_edits(wmOperatorType *ot)
{
  /* identifiers */
  ot->name = "Flush Edits";
  ot->description = "Flush edit data from active editing modes";
  ot->idname = "ED_OT_flush_edits";

  /* API callbacks. */
  ot->exec = ed_flush_edits_exec;

  /* flags */
  ot->flag = OPTYPE_INTERNAL;
}

/** \} */

void ED_operatortypes_edutils()
{
  WM_operatortype_append(ED_OT_lib_id_load_custom_preview);
  WM_operatortype_append(ED_OT_lib_id_generate_preview);
  WM_operatortype_append(ED_OT_lib_id_generate_preview_from_object);
  WM_operatortype_append(ED_OT_lib_id_remove_preview);

  WM_operatortype_append(ED_OT_lib_id_fake_user_toggle);
  WM_operatortype_append(ED_OT_lib_id_unlink);
  WM_operatortype_append(ED_OT_lib_id_override_editable_toggle);

  WM_operatortype_append(ED_OT_flush_edits);

  WM_operatortype_append(ED_OT_undo);
  WM_operatortype_append(ED_OT_undo_push);
  WM_operatortype_append(ED_OT_redo);
  WM_operatortype_append(ED_OT_undo_redo);
  WM_operatortype_append(ED_OT_undo_history);
}