doxygen: prevent GPL license block from being parsed as doxygen comment.
[blender.git] / source / blender / editors / space_node / node_state.c
index 0fe1acd..abb3d83 100644 (file)
@@ -1,5 +1,5 @@
-/**
- * $Id:
+/*
+ * $Id$
  *
  * ***** BEGIN GPL LICENSE BLOCK *****
  *
@@ -15,7 +15,7 @@
  *
  * 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., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
+ * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
  *
  * The Original Code is Copyright (C) 2008 Blender Foundation.
  * All rights reserved.
 #include <stdio.h>
 
 #include "DNA_node_types.h"
-#include "DNA_material_types.h"
-#include "DNA_texture_types.h"
 #include "DNA_scene_types.h"
-#include "DNA_screen_types.h"
-#include "DNA_space_types.h"
+
+#include "BLI_rect.h"
+#include "BLI_utildefines.h"
 
 #include "BKE_context.h"
 #include "BKE_node.h"
-#include "BKE_global.h"
-
-#include "BLI_rect.h"
 
-#include "ED_space_api.h"
 #include "ED_screen.h"
-#include "ED_types.h"
 
 #include "RNA_access.h"
 #include "RNA_define.h"
  
 #include "node_intern.h"
 
+/* **************** Node Header Buttons ************** */
+
+/* note: call node_tree_verify_groups(snode->nodetree) after this
+ */
+void node_set_hidden_sockets(SpaceNode *snode, bNode *node, int set)
+{      
+       bNodeSocket *sock;
+
+       if(set==0) {
+               for(sock= node->inputs.first; sock; sock= sock->next)
+                       sock->flag &= ~SOCK_HIDDEN;
+               for(sock= node->outputs.first; sock; sock= sock->next)
+                       sock->flag &= ~SOCK_HIDDEN;
+       }
+       else {
+               bNode *gnode= node_tree_get_editgroup(snode->nodetree);
+
+               /* hiding inside group should not break links in other group users */
+               if(gnode) {
+                       nodeGroupSocketUseFlags((bNodeTree *)gnode->id);
+                       for(sock= node->inputs.first; sock; sock= sock->next)
+                               if(!(sock->flag & SOCK_IN_USE))
+                                       if(sock->link==NULL)
+                                               sock->flag |= SOCK_HIDDEN;
+                       for(sock= node->outputs.first; sock; sock= sock->next)
+                               if(!(sock->flag & SOCK_IN_USE))
+                                       if(nodeCountSocketLinks(snode->edittree, sock)==0)
+                                               sock->flag |= SOCK_HIDDEN;
+               }
+               else {
+                       /* hide unused sockets */
+                       for(sock= node->inputs.first; sock; sock= sock->next) {
+                               if(sock->link==NULL)
+                                       sock->flag |= SOCK_HIDDEN;
+                       }
+                       for(sock= node->outputs.first; sock; sock= sock->next) {
+                               if(nodeCountSocketLinks(snode->edittree, sock)==0)
+                                       sock->flag |= SOCK_HIDDEN;
+                       }
+               }
+       }
+}
+
+static void node_hide_unhide_sockets(SpaceNode *snode, bNode *node)
+{
+       node_set_hidden_sockets(snode, node, !node_has_hidden_sockets(node));
+       node_tree_verify_groups(snode->nodetree);
+}
 
 static int do_header_node(SpaceNode *snode, bNode *node, float mx, float my)
 {
@@ -75,39 +117,31 @@ static int do_header_node(SpaceNode *snode, bNode *node, float mx, float my)
                        node->flag ^= NODE_PREVIEW;
                        return 1;
                }
-               totr.xmin-=18.0f;
+               totr.xmin-=15.0f;
        }
        if(node->type == NODE_GROUP) {
                if(BLI_in_rctf(&totr, mx, my)) {
                        snode_make_group_editable(snode, node);
                        return 1;
                }
-               totr.xmin-=18.0f;
+               totr.xmin-=15.0f;
        }
        if(node->typeinfo->flag & NODE_OPTIONS) {
                if(BLI_in_rctf(&totr, mx, my)) {
                        node->flag ^= NODE_OPTIONS;
                        return 1;
                }
-               totr.xmin-=18.0f;
+               totr.xmin-=15.0f;
        }
        /* hide unused sockets */
        if(BLI_in_rctf(&totr, mx, my)) {
-               // XXX node_hide_unhide_sockets(snode, node);
+               node_hide_unhide_sockets(snode, node);
        }
        
-       
-       totr= node->totr;
-       totr.xmin= totr.xmax-10.0f;
-       totr.ymax= totr.ymin+10.0f;
-       if(BLI_in_rctf(&totr, mx, my)) {
-               // XXX scale_node(snode, node);
-               return 1;
-       }
        return 0;
 }
 
-static int do_header_hidden_node(SpaceNode *snode, bNode *node, float mx, float my)
+static int do_header_hidden_node(bNode *node, float mx, float my)
 {
        rctf totr= node->totr;
        
@@ -116,17 +150,10 @@ static int do_header_hidden_node(SpaceNode *snode, bNode *node, float mx, float
                node->flag &= ~NODE_HIDDEN;
                return 1;
        }       
-       
-       totr.xmax= node->totr.xmax;
-       totr.xmin= node->totr.xmax-15.0f;
-       if(BLI_in_rctf(&totr, mx, my)) {
-               scale_node(snode, node);
-               return 1;
-       }
        return 0;
 }
 
-static void node_toggle_visibility(SpaceNode *snode, ARegion *ar, short *mval)
+static int node_toggle_visibility(SpaceNode *snode, ARegion *ar, short *mval)
 {
        bNode *node;
        float mx, my;
@@ -138,31 +165,33 @@ static void node_toggle_visibility(SpaceNode *snode, ARegion *ar, short *mval)
        
        for(next_node(snode->edittree); (node=next_node(NULL));) {
                if(node->flag & NODE_HIDDEN) {
-                       if(do_header_hidden_node(snode, node, mx, my)) {
+                       if(do_header_hidden_node(node, mx, my)) {
                                ED_region_tag_redraw(ar);
-                               break;
+                               return 1;
                        }
                }
                else {
                        if(do_header_node(snode, node, mx, my)) {
                                ED_region_tag_redraw(ar);
-                               break;
+                               return 1;
                        }
                }
        }
+       return 0;
 }
 
 static int node_toggle_visibility_exec(bContext *C, wmOperator *op)
 {
-       SpaceNode *snode= (SpaceNode*)CTX_wm_space_data(C);
+       SpaceNode *snode= CTX_wm_space_node(C);
        ARegion *ar= CTX_wm_region(C);
        short mval[2];
 
        mval[0] = RNA_int_get(op->ptr, "mouse_x");
        mval[1] = RNA_int_get(op->ptr, "mouse_y");
-       node_toggle_visibility(snode, ar, mval);
-
-       return OPERATOR_FINISHED;
+       if(node_toggle_visibility(snode, ar, mval))
+               return OPERATOR_FINISHED;
+       else
+               return OPERATOR_CANCELLED|OPERATOR_PASS_THROUGH;
 }
 
 static int node_toggle_visibility_invoke(bContext *C, wmOperator *op, wmEvent *event)
@@ -184,32 +213,95 @@ void NODE_OT_visibility_toggle(wmOperatorType *ot)
        /* identifiers */
        ot->name= "Toggle Visibility";
        ot->idname= "NODE_OT_visibility_toggle";
+       ot->description= "Handle clicks on node header buttons";
        
        /* api callbacks */
        ot->invoke= node_toggle_visibility_invoke;
        ot->poll= ED_operator_node_active;
        
+       /* flags */
+       ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
+       
        RNA_def_int(ot->srna, "mouse_x", 0, INT_MIN, INT_MAX, "Mouse X", "", INT_MIN, INT_MAX);
        RNA_def_int(ot->srna, "mouse_y", 0, INT_MIN, INT_MAX, "Mouse Y", "", INT_MIN, INT_MAX);
 }
 
-static int node_fit_all_exec(bContext *C, wmOperator *op)
+/* **************** View All Operator ************** */
+
+static void snode_home(ScrArea *UNUSED(sa), ARegion *ar, SpaceNode* snode)
+{
+       bNode *node;
+       rctf *cur, *tot;
+       float oldwidth, oldheight, width, height;
+       int first= 1;
+       
+       cur= &ar->v2d.cur;
+       tot= &ar->v2d.tot;
+       
+       oldwidth= cur->xmax - cur->xmin;
+       oldheight= cur->ymax - cur->ymin;
+       
+       cur->xmin= cur->ymin= 0.0f;
+       cur->xmax=ar->winx;
+       cur->ymax=ar->winy;
+       
+       if(snode->edittree) {
+               for(node= snode->edittree->nodes.first; node; node= node->next) {
+                       if(first) {
+                               first= 0;
+                               ar->v2d.cur= node->totr;
+                       }
+                       else {
+                               BLI_union_rctf(cur, &node->totr);
+                       }
+               }
+       }
+       
+       snode->xof= 0;
+       snode->yof= 0;
+       width= cur->xmax - cur->xmin;
+       height= cur->ymax- cur->ymin;
+
+       if(width > height) {
+               float newheight;
+               newheight= oldheight * width/oldwidth;
+               cur->ymin= cur->ymin - newheight/4;
+               cur->ymax= cur->ymax + newheight/4;
+       }
+       else {
+               float newwidth;
+               newwidth= oldwidth * height/oldheight;
+               cur->xmin= cur->xmin - newwidth/4;
+               cur->xmax= cur->xmax + newwidth/4;
+       }
+
+       ar->v2d.tot= ar->v2d.cur;
+       UI_view2d_curRect_validate(&ar->v2d);
+}
+
+static int node_view_all_exec(bContext *C, wmOperator *UNUSED(op))
 {
        ScrArea *sa= CTX_wm_area(C);
        ARegion *ar= CTX_wm_region(C);
-       SpaceNode *snode= (SpaceNode *)CTX_wm_space_data(C);
+       SpaceNode *snode= CTX_wm_space_node(C);
+       
        snode_home(sa, ar, snode);
        ED_region_tag_redraw(ar);
+       
        return OPERATOR_FINISHED;
 }
 
-void NODE_OT_fit_all(wmOperatorType *ot)
+void NODE_OT_view_all(wmOperatorType *ot)
 {
        /* identifiers */
-       ot->name= "Fit All";
-       ot->idname= "NODE_OT_fit_all";
+       ot->name= "View All";
+       ot->idname= "NODE_OT_view_all";
+       ot->description= "Resize view so you can see all nodes";
        
        /* api callbacks */
-       ot->exec= node_fit_all_exec;
+       ot->exec= node_view_all_exec;
        ot->poll= ED_operator_node_active;
+       
+       /* flags */
+       ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
 }