Welcome to mirror list, hosted at ThFree Co, Russian Federation.

git.blender.org/blender.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
Diffstat
-rw-r--r--source/blender/editors/space_node/node_add.c301
-rw-r--r--source/blender/editors/space_node/node_intern.h3
-rw-r--r--source/blender/editors/space_node/node_ops.c3
-rw-r--r--source/blender/editors/space_node/space_node.c39
4 files changed, 346 insertions, 0 deletions
diff --git a/source/blender/editors/space_node/node_add.c b/source/blender/editors/space_node/node_add.c
index e0de2393917..a646804e0fd 100644
--- a/source/blender/editors/space_node/node_add.c
+++ b/source/blender/editors/space_node/node_add.c
@@ -23,7 +23,9 @@
#include "MEM_guardedalloc.h"
+#include "DNA_collection_types.h"
#include "DNA_node_types.h"
+#include "DNA_texture_types.h"
#include "BLI_listbase.h"
#include "BLI_math.h"
@@ -418,6 +420,305 @@ void NODE_OT_add_group(wmOperatorType *ot)
/** \} */
/* -------------------------------------------------------------------- */
+/** \name Add Node Object Operator
+ * \{ */
+
+static Object *node_add_object_get_and_poll_object_node_tree(Main *bmain, wmOperator *op)
+{
+ char name[MAX_ID_NAME - 2];
+ RNA_string_get(op->ptr, "name", name);
+
+ Object *object = (Object *)BKE_libblock_find_name(bmain, ID_OB, name);
+ if (!object) {
+ return NULL;
+ }
+
+ return object;
+}
+
+static int node_add_object_exec(bContext *C, wmOperator *op)
+{
+ Main *bmain = CTX_data_main(C);
+ SpaceNode *snode = CTX_wm_space_node(C);
+ bNodeTree *ntree = snode->edittree;
+ Object *object;
+
+ if (!(object = node_add_object_get_and_poll_object_node_tree(bmain, op))) {
+ return OPERATOR_CANCELLED;
+ }
+
+ ED_preview_kill_jobs(CTX_wm_manager(C), CTX_data_main(C));
+
+ bNode *object_node = node_add_node(
+ C, NULL, GEO_NODE_OBJECT_INFO, snode->runtime->cursor[0], snode->runtime->cursor[1]);
+ if (!object_node) {
+ BKE_report(op->reports, RPT_WARNING, "Could not add node object");
+ return OPERATOR_CANCELLED;
+ }
+
+ bNodeSocket *sock = nodeFindSocket(object_node, SOCK_IN, "Object");
+ if (!sock) {
+ BKE_report(op->reports, RPT_WARNING, "Could not find node object socket");
+ return OPERATOR_CANCELLED;
+ }
+
+ bNodeSocketValueObject *socket_data = sock->default_value;
+ socket_data->value = object;
+ id_us_plus(&object->id);
+
+ nodeSetActive(ntree, object_node);
+ ntreeUpdateTree(bmain, ntree);
+
+ snode_notify(C, snode);
+ snode_dag_update(C, snode);
+
+ return OPERATOR_FINISHED;
+}
+
+static int node_add_object_invoke(bContext *C, wmOperator *op, const wmEvent *event)
+{
+ ARegion *region = CTX_wm_region(C);
+ SpaceNode *snode = CTX_wm_space_node(C);
+
+ /* Convert mouse coordinates to v2d space. */
+ UI_view2d_region_to_view(&region->v2d,
+ event->mval[0],
+ event->mval[1],
+ &snode->runtime->cursor[0],
+ &snode->runtime->cursor[1]);
+
+ snode->runtime->cursor[0] /= UI_DPI_FAC;
+ snode->runtime->cursor[1] /= UI_DPI_FAC;
+
+ return node_add_object_exec(C, op);
+}
+
+static bool node_add_object_poll(bContext *C)
+{
+ const SpaceNode *snode = CTX_wm_space_node(C);
+ return ED_operator_node_editable(C) && ELEM(snode->nodetree->type, NTREE_GEOMETRY);
+}
+
+void NODE_OT_add_object(wmOperatorType *ot)
+{
+ /* identifiers */
+ ot->name = "Add Node Object";
+ ot->description = "Add an object info node to the current node editor";
+ ot->idname = "NODE_OT_add_object";
+
+ /* callbacks */
+ ot->exec = node_add_object_exec;
+ ot->invoke = node_add_object_invoke;
+ ot->poll = node_add_object_poll;
+
+ /* flags */
+ ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO | OPTYPE_INTERNAL;
+
+ RNA_def_string(ot->srna, "name", "Object", MAX_ID_NAME - 2, "Name", "Data-block name to assign");
+}
+
+/** \} */
+
+/* -------------------------------------------------------------------- */
+/** \name Add Node Texture Operator
+ * \{ */
+
+static Tex *node_add_texture_get_and_poll_texture_node_tree(Main *bmain, wmOperator *op)
+{
+ char name[MAX_ID_NAME - 2];
+ RNA_string_get(op->ptr, "name", name);
+
+ Tex *texture = (Tex *)BKE_libblock_find_name(bmain, ID_TE, name);
+ if (!texture) {
+ return NULL;
+ }
+
+ return texture;
+}
+
+static int node_add_texture_exec(bContext *C, wmOperator *op)
+{
+ Main *bmain = CTX_data_main(C);
+ SpaceNode *snode = CTX_wm_space_node(C);
+ bNodeTree *ntree = snode->edittree;
+ Tex *texture;
+
+ if (!(texture = node_add_texture_get_and_poll_texture_node_tree(bmain, op))) {
+ return OPERATOR_CANCELLED;
+ }
+
+ ED_preview_kill_jobs(CTX_wm_manager(C), CTX_data_main(C));
+
+ bNode *texture_node = node_add_node(C,
+ NULL,
+ GEO_NODE_ATTRIBUTE_SAMPLE_TEXTURE,
+ snode->runtime->cursor[0],
+ snode->runtime->cursor[1]);
+ if (!texture_node) {
+ BKE_report(op->reports, RPT_WARNING, "Could not add texture node");
+ return OPERATOR_CANCELLED;
+ }
+
+ texture_node->id = &texture->id;
+ id_us_plus(&texture->id);
+
+ nodeSetActive(ntree, texture_node);
+ ntreeUpdateTree(bmain, ntree);
+
+ snode_notify(C, snode);
+ snode_dag_update(C, snode);
+
+ return OPERATOR_FINISHED;
+}
+
+static int node_add_texture_invoke(bContext *C, wmOperator *op, const wmEvent *event)
+{
+ ARegion *region = CTX_wm_region(C);
+ SpaceNode *snode = CTX_wm_space_node(C);
+
+ /* Convert mouse coordinates to v2d space. */
+ UI_view2d_region_to_view(&region->v2d,
+ event->mval[0],
+ event->mval[1],
+ &snode->runtime->cursor[0],
+ &snode->runtime->cursor[1]);
+
+ snode->runtime->cursor[0] /= UI_DPI_FAC;
+ snode->runtime->cursor[1] /= UI_DPI_FAC;
+
+ return node_add_texture_exec(C, op);
+}
+
+static bool node_add_texture_poll(bContext *C)
+{
+ const SpaceNode *snode = CTX_wm_space_node(C);
+ return ED_operator_node_editable(C) && ELEM(snode->nodetree->type, NTREE_GEOMETRY);
+}
+
+void NODE_OT_add_texture(wmOperatorType *ot)
+{
+ /* identifiers */
+ ot->name = "Add Node Texture";
+ ot->description = "Add a texture to the current node editor";
+ ot->idname = "NODE_OT_add_texture";
+
+ /* callbacks */
+ ot->exec = node_add_texture_exec;
+ ot->invoke = node_add_texture_invoke;
+ ot->poll = node_add_texture_poll;
+
+ /* flags */
+ ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO | OPTYPE_INTERNAL;
+
+ RNA_def_string(
+ ot->srna, "name", "Texture", MAX_ID_NAME - 2, "Name", "Data-block name to assign");
+}
+
+/** \} */
+
+/* -------------------------------------------------------------------- */
+/** \name Add Node Collection Operator
+ * \{ */
+
+static Collection *node_add_collection_get_and_poll_collection_node_tree(Main *bmain,
+ wmOperator *op)
+{
+ char name[MAX_ID_NAME - 2];
+ RNA_string_get(op->ptr, "name", name);
+
+ Collection *collection = (Collection *)BKE_libblock_find_name(bmain, ID_GR, name);
+ if (!collection) {
+ return NULL;
+ }
+
+ return collection;
+}
+
+static int node_add_collection_exec(bContext *C, wmOperator *op)
+{
+ Main *bmain = CTX_data_main(C);
+ SpaceNode *snode = CTX_wm_space_node(C);
+ bNodeTree *ntree = snode->edittree;
+ Collection *collection;
+
+ if (!(collection = node_add_collection_get_and_poll_collection_node_tree(bmain, op))) {
+ return OPERATOR_CANCELLED;
+ }
+
+ ED_preview_kill_jobs(CTX_wm_manager(C), CTX_data_main(C));
+
+ bNode *collection_node = node_add_node(
+ C, NULL, GEO_NODE_COLLECTION_INFO, snode->runtime->cursor[0], snode->runtime->cursor[1]);
+ if (!collection_node) {
+ BKE_report(op->reports, RPT_WARNING, "Could not add node collection");
+ return OPERATOR_CANCELLED;
+ }
+
+ bNodeSocket *sock = nodeFindSocket(collection_node, SOCK_IN, "Collection");
+ if (!sock) {
+ BKE_report(op->reports, RPT_WARNING, "Could not find node collection socket");
+ return OPERATOR_CANCELLED;
+ }
+
+ bNodeSocketValueCollection *socket_data = sock->default_value;
+ socket_data->value = collection;
+ id_us_plus(&collection->id);
+
+ nodeSetActive(ntree, collection_node);
+ ntreeUpdateTree(bmain, ntree);
+
+ snode_notify(C, snode);
+ snode_dag_update(C, snode);
+
+ return OPERATOR_FINISHED;
+}
+
+static int node_add_collection_invoke(bContext *C, wmOperator *op, const wmEvent *event)
+{
+ ARegion *region = CTX_wm_region(C);
+ SpaceNode *snode = CTX_wm_space_node(C);
+
+ /* Convert mouse coordinates to v2d space. */
+ UI_view2d_region_to_view(&region->v2d,
+ event->mval[0],
+ event->mval[1],
+ &snode->runtime->cursor[0],
+ &snode->runtime->cursor[1]);
+
+ snode->runtime->cursor[0] /= UI_DPI_FAC;
+ snode->runtime->cursor[1] /= UI_DPI_FAC;
+
+ return node_add_collection_exec(C, op);
+}
+
+static bool node_add_collection_poll(bContext *C)
+{
+ const SpaceNode *snode = CTX_wm_space_node(C);
+ return ED_operator_node_editable(C) && ELEM(snode->nodetree->type, NTREE_GEOMETRY);
+}
+
+void NODE_OT_add_collection(wmOperatorType *ot)
+{
+ /* identifiers */
+ ot->name = "Add Node Collection";
+ ot->description = "Add an collection info node to the current node editor";
+ ot->idname = "NODE_OT_add_collection";
+
+ /* callbacks */
+ ot->exec = node_add_collection_exec;
+ ot->invoke = node_add_collection_invoke;
+ ot->poll = node_add_collection_poll;
+
+ /* flags */
+ ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO | OPTYPE_INTERNAL;
+
+ RNA_def_string(
+ ot->srna, "name", "Collection", MAX_ID_NAME - 2, "Name", "Data-block name to assign");
+}
+
+/** \} */
+
+/* -------------------------------------------------------------------- */
/** \name Add File Node Operator
* \{ */
diff --git a/source/blender/editors/space_node/node_intern.h b/source/blender/editors/space_node/node_intern.h
index bf27ab18bd4..1840ec93f6f 100644
--- a/source/blender/editors/space_node/node_intern.h
+++ b/source/blender/editors/space_node/node_intern.h
@@ -208,6 +208,9 @@ bNode *node_add_node(
const struct bContext *C, const char *idname, int type, float locx, float locy);
void NODE_OT_add_reroute(struct wmOperatorType *ot);
void NODE_OT_add_group(struct wmOperatorType *ot);
+void NODE_OT_add_object(struct wmOperatorType *ot);
+void NODE_OT_add_collection(struct wmOperatorType *ot);
+void NODE_OT_add_texture(struct wmOperatorType *ot);
void NODE_OT_add_file(struct wmOperatorType *ot);
void NODE_OT_add_mask(struct wmOperatorType *ot);
void NODE_OT_new_node_tree(struct wmOperatorType *ot);
diff --git a/source/blender/editors/space_node/node_ops.c b/source/blender/editors/space_node/node_ops.c
index 7671547363b..ef28cfe8a8b 100644
--- a/source/blender/editors/space_node/node_ops.c
+++ b/source/blender/editors/space_node/node_ops.c
@@ -89,6 +89,9 @@ void node_operatortypes(void)
WM_operatortype_append(NODE_OT_backimage_sample);
WM_operatortype_append(NODE_OT_add_group);
+ WM_operatortype_append(NODE_OT_add_object);
+ WM_operatortype_append(NODE_OT_add_collection);
+ WM_operatortype_append(NODE_OT_add_texture);
WM_operatortype_append(NODE_OT_add_file);
WM_operatortype_append(NODE_OT_add_mask);
diff --git a/source/blender/editors/space_node/space_node.c b/source/blender/editors/space_node/space_node.c
index 289b7d9efa1..5d14919502e 100644
--- a/source/blender/editors/space_node/space_node.c
+++ b/source/blender/editors/space_node/space_node.c
@@ -667,6 +667,30 @@ static bool node_group_drop_poll(bContext *UNUSED(C),
return WM_drag_is_ID_type(drag, ID_NT);
}
+static bool node_object_drop_poll(bContext *UNUSED(C),
+ wmDrag *drag,
+ const wmEvent *UNUSED(event),
+ const char **UNUSED(r_tooltip))
+{
+ return WM_drag_is_ID_type(drag, ID_OB);
+}
+
+static bool node_collection_drop_poll(bContext *UNUSED(C),
+ wmDrag *drag,
+ const wmEvent *UNUSED(event),
+ const char **UNUSED(r_tooltip))
+{
+ return WM_drag_is_ID_type(drag, ID_GR);
+}
+
+static bool node_texture_drop_poll(bContext *UNUSED(C),
+ wmDrag *drag,
+ const wmEvent *UNUSED(event),
+ const char **UNUSED(r_tooltip))
+{
+ return WM_drag_is_ID_type(drag, ID_TE);
+}
+
static bool node_ima_drop_poll(bContext *UNUSED(C),
wmDrag *drag,
const wmEvent *UNUSED(event),
@@ -721,6 +745,21 @@ static void node_dropboxes(void)
ListBase *lb = WM_dropboxmap_find("Node Editor", SPACE_NODE, RGN_TYPE_WINDOW);
WM_dropbox_add(lb,
+ "NODE_OT_add_object",
+ node_object_drop_poll,
+ node_id_drop_copy,
+ WM_drag_free_imported_drag_ID);
+ WM_dropbox_add(lb,
+ "NODE_OT_add_collection",
+ node_collection_drop_poll,
+ node_id_drop_copy,
+ WM_drag_free_imported_drag_ID);
+ WM_dropbox_add(lb,
+ "NODE_OT_add_texture",
+ node_texture_drop_poll,
+ node_id_drop_copy,
+ WM_drag_free_imported_drag_ID);
+ WM_dropbox_add(lb,
"NODE_OT_add_group",
node_group_drop_poll,
node_group_drop_copy,
generated by cgit v1.2.3 (git 2.25.1) at 2024-07-23 04:59:53 +0300