Allow creation of mask nodes in compositor with drag&drop of mask datablocks
authorSv. Lockal <lockalsash@gmail.com>
Mon, 22 Jul 2013 19:24:39 +0000 (19:24 +0000)
committerSv. Lockal <lockalsash@gmail.com>
Mon, 22 Jul 2013 19:24:39 +0000 (19:24 +0000)
Also remove superfluous "deselect all" operation in drag&drop for images

source/blender/editors/space_node/node_add.c
source/blender/editors/space_node/node_intern.h
source/blender/editors/space_node/node_ops.c
source/blender/editors/space_node/space_node.c

index d5224a37358f478acd951e17336d4fe0864d007a..0dc1415ea69f1c3d7a5f03f085925a73f914db60 100644 (file)
@@ -339,8 +339,6 @@ static int node_add_file_exec(bContext *C, wmOperator *op)
                }
        }
 
-       node_deselect_all(snode);
-
        switch (snode->nodetree->type) {
                case NTREE_SHADER:
                        type = SH_NODE_TEX_IMAGE;
@@ -410,6 +408,69 @@ void NODE_OT_add_file(wmOperatorType *ot)
        RNA_def_string(ot->srna, "name", "Image", MAX_ID_NAME - 2, "Name", "Datablock name to assign");
 }
 
+/* ****************** Add Mask Node Operator  ******************* */
+
+static int node_add_mask_poll(bContext *C)
+{
+       SpaceNode *snode = CTX_wm_space_node(C);
+
+       return ED_operator_node_editable(C) && snode->nodetree->type == NTREE_COMPOSIT;
+}
+
+static int node_add_mask_invoke(bContext *C, wmOperator *op, const wmEvent *event)
+{
+       ARegion *ar = CTX_wm_region(C);
+       SpaceNode *snode = CTX_wm_space_node(C);
+       bNode *node;
+       ID *mask = NULL;
+
+       /* check input variables */
+       char name[MAX_ID_NAME - 2];
+       RNA_string_get(op->ptr, "name", name);
+       mask = BKE_libblock_find_name(ID_MSK, name);
+       if (!mask) {
+               BKE_reportf(op->reports, RPT_ERROR, "Mask '%s' not found", name);
+               return OPERATOR_CANCELLED;
+       }
+
+       ED_preview_kill_jobs(C);
+
+       /* convert mouse coordinates to v2d space */
+       UI_view2d_region_to_view(&ar->v2d, event->mval[0], event->mval[1],
+                                                        &snode->cursor[0], &snode->cursor[1]);
+       node = node_add_node(C, NULL, CMP_NODE_MASK, snode->cursor[0], snode->cursor[1]);
+
+       if (!node) {
+               BKE_report(op->reports, RPT_WARNING, "Could not add an mask node");
+               return OPERATOR_CANCELLED;
+       }
+
+       node->id = mask;
+       id_us_plus(mask);
+
+       snode_notify(C, snode);
+       snode_dag_update(C, snode);
+
+       return OPERATOR_FINISHED;
+}
+
+void NODE_OT_add_mask(wmOperatorType *ot)
+{
+       /* identifiers */
+       ot->name = "Add Mask Node";
+       ot->description = "Add a mask node to the current node editor";
+       ot->idname = "NODE_OT_add_mask";
+
+       /* callbacks */
+       ot->invoke = node_add_mask_invoke;
+       ot->poll = node_add_mask_poll;
+
+       /* flags */
+       ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
+
+       RNA_def_string(ot->srna, "name", "Mask", MAX_ID_NAME - 2, "Name", "Datablock name to assign");
+}
+
 /********************** New node tree operator *********************/
 
 static int new_node_tree_exec(bContext *C, wmOperator *op)
index a9a17b182a4a87291247c60dd0396fb3914fa17c..1a2e90e55225dfc9d6803768d94dfa6a6a70bbf3 100644 (file)
@@ -143,6 +143,7 @@ void draw_nodespace_back_pix(const struct bContext *C, struct ARegion *ar, struc
 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_file(struct wmOperatorType *ot);
+void NODE_OT_add_mask(struct wmOperatorType *ot);
 void NODE_OT_new_node_tree(struct wmOperatorType *ot);
 
 
index 2476a9b8b99a2fd19d7103716a6ca8c9858b50af..566bb1600cc2b307a5f2a31d6e4f103f2b1aaed1 100644 (file)
@@ -101,6 +101,7 @@ void node_operatortypes(void)
        WM_operatortype_append(NODE_OT_backimage_sample);
        
        WM_operatortype_append(NODE_OT_add_file);
+       WM_operatortype_append(NODE_OT_add_mask);
        
        WM_operatortype_append(NODE_OT_new_node_tree);
        
index f12f77434294e46e38645b9974ed53ec89113f4b..922912fa5407e61b32b85765bcb2778381fc98ec 100644 (file)
@@ -635,7 +635,7 @@ static void node_main_area_draw(const bContext *C, ARegion *ar)
 
 /* ************* dropboxes ************* */
 
-static int node_drop_poll(bContext *UNUSED(C), wmDrag *drag, const wmEvent *UNUSED(event))
+static int node_ima_drop_poll(bContext *UNUSED(C), wmDrag *drag, const wmEvent *UNUSED(event))
 {
        if (drag->type == WM_DRAG_ID) {
                ID *id = (ID *)drag->poin;
@@ -649,6 +649,23 @@ static int node_drop_poll(bContext *UNUSED(C), wmDrag *drag, const wmEvent *UNUS
        return 0;
 }
 
+static int node_mask_drop_poll(bContext *UNUSED(C), wmDrag *drag, const wmEvent *UNUSED(event))
+{
+       if (drag->type == WM_DRAG_ID) {
+               ID *id = (ID *)drag->poin;
+               if (GS(id->name) == ID_MSK)
+                       return 1;
+       }
+       return 0;
+}
+
+static void node_id_drop_copy(wmDrag *drag, wmDropBox *drop)
+{
+       ID *id = (ID *)drag->poin;
+
+       RNA_string_set(drop->ptr, "name", id->name + 2);
+}
+
 static void node_id_path_drop_copy(wmDrag *drag, wmDropBox *drop)
 {
        ID *id = (ID *)drag->poin;
@@ -666,7 +683,8 @@ static void node_dropboxes(void)
 {
        ListBase *lb = WM_dropboxmap_find("Node Editor", SPACE_NODE, RGN_TYPE_WINDOW);
 
-       WM_dropbox_add(lb, "NODE_OT_add_file", node_drop_poll, node_id_path_drop_copy);
+       WM_dropbox_add(lb, "NODE_OT_add_file", node_ima_drop_poll, node_id_path_drop_copy);
+       WM_dropbox_add(lb, "NODE_OT_add_mask", node_mask_drop_poll, node_id_drop_copy);
 
 }