octave-lyh: scripts/sparse/treelayout.m annotate

annotate scripts/sparse/treelayout.m @ 14459:a22a41ab6824

doc: Add Colin Macdonald and Mike Miller to contributors.in

author	Jordi Gutiérrez Hermoso <jordigh@octave.org>
date	Tue, 13 Mar 2012 01:30:26 -0400
parents	f3d52523cde1
children	5d3a684236b0

rev	line source
14138 72c96de7a403 maint: update copyright notices for 2012 John W. Eaton <jwe@octave.org> parents: 13060 diff changeset	1 ## Copyright (C) 2008-2012 Ivana Varekova & Radek Salac
8338 a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	2 ##
a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	3 ## This file is part of Octave.
a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	4 ##
a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	5 ## Octave is free software; you can redistribute it and/or modify it
a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	6 ## under the terms of the GNU General Public License as published by
a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	7 ## the Free Software Foundation; either version 3 of the License, or (at
a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	8 ## your option) any later version.
a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	9 ##
a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	10 ## Octave is distributed in the hope that it will be useful, but
a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	11 ## WITHOUT ANY WARRANTY; without even the implied warranty of
a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	12 ## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	13 ## General Public License for more details.
a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	14 ##
a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	15 ## You should have received a copy of the GNU General Public License
a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	16 ## along with Octave; see the file COPYING. If not, see
a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	17 ## <http://www.gnu.org/licenses/>.
a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	18
a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	19 ## -- texinfo --
11469 c776f063fefe Overhaul m-script files to use common variable name between code and documentation. Rik <octave@nomad.inbox5.com> parents: 10793 diff changeset	20 ## @deftypefn {Function File} {} treelayout (@var{tree})
c776f063fefe Overhaul m-script files to use common variable name between code and documentation. Rik <octave@nomad.inbox5.com> parents: 10793 diff changeset	21 ## @deftypefnx {Function File} {} treelayout (@var{tree}, @var{permutation})
c776f063fefe Overhaul m-script files to use common variable name between code and documentation. Rik <octave@nomad.inbox5.com> parents: 10793 diff changeset	22 ## treelayout lays out a tree or a forest. The first argument @var{tree} is a
10793 be55736a0783 Grammarcheck the documentation from m-files. Rik <octave@nomad.inbox5.com> parents: 10549 diff changeset	23 ## vector of
be55736a0783 Grammarcheck the documentation from m-files. Rik <octave@nomad.inbox5.com> parents: 10549 diff changeset	24 ## predecessors, optional parameter @var{permutation} is an optional postorder
be55736a0783 Grammarcheck the documentation from m-files. Rik <octave@nomad.inbox5.com> parents: 10549 diff changeset	25 ## permutation.
8338 a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	26 ## The complexity of the algorithm is O(n) in
a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	27 ## terms of time and memory requirements.
12575 d0b799dafede Grammarcheck files for 3.4.1 release. Rik <octave@nomad.inbox5.com> parents: 11587 diff changeset	28 ## @seealso{etreeplot, gplot, treeplot}
8338 a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	29 ## @end deftypefn
a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	30
8507 cadc73247d65 style fixes John W. Eaton <jwe@octave.org> parents: 8506 diff changeset	31 function [x_coordinate, y_coordinate, height, s] = treelayout (tree, permutation)
8338 a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	32 if (nargin < 1 \|\| nargin > 2 \|\| nargout > 4)
a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	33 print_usage ();
11587 c792872f8942 all script files: untabify and strip trailing whitespace John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	34 elseif (! isvector (tree) \|\| rows (tree) != 1 \|\| ! isnumeric (tree)
8507 cadc73247d65 style fixes John W. Eaton <jwe@octave.org> parents: 8506 diff changeset	35 \|\| any (tree > length (tree)) \|\| any (tree < 0))
8338 a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	36 error ("treelayout: the first input argument must be a vector of predecessors");
a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	37 else
8506 bc982528de11 comment style fixes John W. Eaton <jwe@octave.org> parents: 8338 diff changeset	38 ## Make it a row vector.
8507 cadc73247d65 style fixes John W. Eaton <jwe@octave.org> parents: 8506 diff changeset	39 tree = tree(:)';
8338 a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	40
8506 bc982528de11 comment style fixes John W. Eaton <jwe@octave.org> parents: 8338 diff changeset	41 ## The count of nodes of the graph.
8507 cadc73247d65 style fixes John W. Eaton <jwe@octave.org> parents: 8506 diff changeset	42 num_nodes = length (tree);
8506 bc982528de11 comment style fixes John W. Eaton <jwe@octave.org> parents: 8338 diff changeset	43 ## The number of children.
8507 cadc73247d65 style fixes John W. Eaton <jwe@octave.org> parents: 8506 diff changeset	44 num_children = zeros (1, num_nodes + 1);
8338 a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	45
8506 bc982528de11 comment style fixes John W. Eaton <jwe@octave.org> parents: 8338 diff changeset	46 ## Checking vector of predecessors.
8507 cadc73247d65 style fixes John W. Eaton <jwe@octave.org> parents: 8506 diff changeset	47 for i = 1 : num_nodes
cadc73247d65 style fixes John W. Eaton <jwe@octave.org> parents: 8506 diff changeset	48 if (tree(i) < i)
10549 95c3e38098bf Untabify .m scripts Rik <code@nomad.inbox5.com> parents: 9051 diff changeset	49 ## This part of graph was checked before.
8338 a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	50 continue;
a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	51 endif
a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	52
8506 bc982528de11 comment style fixes John W. Eaton <jwe@octave.org> parents: 8338 diff changeset	53 ## Try to find cicle in this part of graph using modified Floyd's
bc982528de11 comment style fixes John W. Eaton <jwe@octave.org> parents: 8338 diff changeset	54 ## cycle-finding algorithm.
8507 cadc73247d65 style fixes John W. Eaton <jwe@octave.org> parents: 8506 diff changeset	55 tortoise = tree(i);
cadc73247d65 style fixes John W. Eaton <jwe@octave.org> parents: 8506 diff changeset	56 hare = tree(tortoise);
8338 a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	57
a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	58 while (tortoise != hare)
10549 95c3e38098bf Untabify .m scripts Rik <code@nomad.inbox5.com> parents: 9051 diff changeset	59 ## End after finding a cicle or reaching a checked part of graph.
8338 a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	60
a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	61 if (hare < i)
8506 bc982528de11 comment style fixes John W. Eaton <jwe@octave.org> parents: 8338 diff changeset	62 ## This part of graph was checked before.
8338 a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	63 break
a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	64 endif
a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	65
8507 cadc73247d65 style fixes John W. Eaton <jwe@octave.org> parents: 8506 diff changeset	66 tortoise = tree(tortoise);
10549 95c3e38098bf Untabify .m scripts Rik <code@nomad.inbox5.com> parents: 9051 diff changeset	67 ## Hare will move faster than tortoise so in cicle hare must
95c3e38098bf Untabify .m scripts Rik <code@nomad.inbox5.com> parents: 9051 diff changeset	68 ## reach tortoise.
8507 cadc73247d65 style fixes John W. Eaton <jwe@octave.org> parents: 8506 diff changeset	69 hare = tree(tree(hare));
8338 a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	70
a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	71 endwhile
a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	72
a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	73 if (tortoise == hare)
10549 95c3e38098bf Untabify .m scripts Rik <code@nomad.inbox5.com> parents: 9051 diff changeset	74 ## If hare reach tortoise we found circle.
8338 a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	75 error ("treelayout: vector of predecessors has bad format");
a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	76 endif
a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	77
a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	78 endfor
8506 bc982528de11 comment style fixes John W. Eaton <jwe@octave.org> parents: 8338 diff changeset	79 ## Vector of predecessors has right format.
8338 a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	80
8507 cadc73247d65 style fixes John W. Eaton <jwe@octave.org> parents: 8506 diff changeset	81 for i = 1:num_nodes
cadc73247d65 style fixes John W. Eaton <jwe@octave.org> parents: 8506 diff changeset	82 ## vec_of_child is helping vector which is used to speed up the
8506 bc982528de11 comment style fixes John W. Eaton <jwe@octave.org> parents: 8338 diff changeset	83 ## choice of descendant nodes.
8338 a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	84
8507 cadc73247d65 style fixes John W. Eaton <jwe@octave.org> parents: 8506 diff changeset	85 num_children(tree(i)+1) = num_children(tree(i)+1) + 1;
8338 a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	86 endfor
a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	87
8507 cadc73247d65 style fixes John W. Eaton <jwe@octave.org> parents: 8506 diff changeset	88 pos = 1;
cadc73247d65 style fixes John W. Eaton <jwe@octave.org> parents: 8506 diff changeset	89 start = zeros (1, num_nodes+1);
cadc73247d65 style fixes John W. Eaton <jwe@octave.org> parents: 8506 diff changeset	90 xhelp = zeros (1, num_nodes+1);
cadc73247d65 style fixes John W. Eaton <jwe@octave.org> parents: 8506 diff changeset	91 stop = zeros (1, num_nodes+1);
cadc73247d65 style fixes John W. Eaton <jwe@octave.org> parents: 8506 diff changeset	92 for i = 1 : num_nodes + 1
cadc73247d65 style fixes John W. Eaton <jwe@octave.org> parents: 8506 diff changeset	93 start(i) = pos;
cadc73247d65 style fixes John W. Eaton <jwe@octave.org> parents: 8506 diff changeset	94 xhelp(i) = pos;
cadc73247d65 style fixes John W. Eaton <jwe@octave.org> parents: 8506 diff changeset	95 pos += num_children(i);
cadc73247d65 style fixes John W. Eaton <jwe@octave.org> parents: 8506 diff changeset	96 stop(i) = pos;
8338 a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	97 endfor
a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	98
a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	99 if (nargin == 1)
8507 cadc73247d65 style fixes John W. Eaton <jwe@octave.org> parents: 8506 diff changeset	100 for i = 1:num_nodes
11587 c792872f8942 all script files: untabify and strip trailing whitespace John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	101 vec_of_child(xhelp(tree(i)+1)) = i;
8507 cadc73247d65 style fixes John W. Eaton <jwe@octave.org> parents: 8506 diff changeset	102 xhelp(tree(i)+1) = xhelp(tree(i)+1) + 1;
8338 a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	103 endfor
a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	104 else
8507 cadc73247d65 style fixes John W. Eaton <jwe@octave.org> parents: 8506 diff changeset	105 vec_of_child = permutation;
8338 a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	106 endif
a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	107
8506 bc982528de11 comment style fixes John W. Eaton <jwe@octave.org> parents: 8338 diff changeset	108 ## The number of "parent" (actual) node (it's descendants will be
bc982528de11 comment style fixes John W. Eaton <jwe@octave.org> parents: 8338 diff changeset	109 ## browse in the next iteration).
8507 cadc73247d65 style fixes John W. Eaton <jwe@octave.org> parents: 8506 diff changeset	110 par_number = 0;
8338 a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	111
8506 bc982528de11 comment style fixes John W. Eaton <jwe@octave.org> parents: 8338 diff changeset	112 ## The x-coordinate of the left most descendant of "parent node"
bc982528de11 comment style fixes John W. Eaton <jwe@octave.org> parents: 8338 diff changeset	113 ## this value is increased in each leaf.
8507 cadc73247d65 style fixes John W. Eaton <jwe@octave.org> parents: 8506 diff changeset	114 left_most = 0;
8338 a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	115
8507 cadc73247d65 style fixes John W. Eaton <jwe@octave.org> parents: 8506 diff changeset	116 ## The level of "parent" node (root level is num_nodes).
cadc73247d65 style fixes John W. Eaton <jwe@octave.org> parents: 8506 diff changeset	117 level = num_nodes;
8338 a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	118
8507 cadc73247d65 style fixes John W. Eaton <jwe@octave.org> parents: 8506 diff changeset	119 ## num_nodes - max_ht is the height of this graph.
cadc73247d65 style fixes John W. Eaton <jwe@octave.org> parents: 8506 diff changeset	120 max_ht = num_nodes;
8338 a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	121
8506 bc982528de11 comment style fixes John W. Eaton <jwe@octave.org> parents: 8338 diff changeset	122 ## Main stack - each item consists of two numbers - the number of
8338 a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	123 ## node and the number it's of parent node on the top of stack
8506 bc982528de11 comment style fixes John W. Eaton <jwe@octave.org> parents: 8338 diff changeset	124 ## there is "parent node".
8507 cadc73247d65 style fixes John W. Eaton <jwe@octave.org> parents: 8506 diff changeset	125 stk = [-1, 0];
8338 a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	126
8506 bc982528de11 comment style fixes John W. Eaton <jwe@octave.org> parents: 8338 diff changeset	127 ## Number of vertices s in the top-level separator.
8338 a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	128 s = 0;
8506 bc982528de11 comment style fixes John W. Eaton <jwe@octave.org> parents: 8338 diff changeset	129 ## Flag which says if we are in top level separator.
8507 cadc73247d65 style fixes John W. Eaton <jwe@octave.org> parents: 8506 diff changeset	130 top_level = 1;
8506 bc982528de11 comment style fixes John W. Eaton <jwe@octave.org> parents: 8338 diff changeset	131 ## The top of the stack.
8507 cadc73247d65 style fixes John W. Eaton <jwe@octave.org> parents: 8506 diff changeset	132 while (par_number != -1)
cadc73247d65 style fixes John W. Eaton <jwe@octave.org> parents: 8506 diff changeset	133 if (start(par_number+1) < stop(par_number+1))
cadc73247d65 style fixes John W. Eaton <jwe@octave.org> parents: 8506 diff changeset	134 idx = vec_of_child(start(par_number+1) : stop(par_number+1) - 1);
8338 a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	135 else
a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	136 idx = zeros (1, 0);
a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	137 endif
a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	138
8506 bc982528de11 comment style fixes John W. Eaton <jwe@octave.org> parents: 8338 diff changeset	139 ## Add to idx the vector of parent descendants.
8507 cadc73247d65 style fixes John W. Eaton <jwe@octave.org> parents: 8506 diff changeset	140 stk = [stk; [idx', ones(fliplr(size(idx))) * par_number]];
8338 a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	141
8506 bc982528de11 comment style fixes John W. Eaton <jwe@octave.org> parents: 8338 diff changeset	142 ## We are in top level separator when we have one child and the
bc982528de11 comment style fixes John W. Eaton <jwe@octave.org> parents: 8338 diff changeset	143 ## flag is 1
8507 cadc73247d65 style fixes John W. Eaton <jwe@octave.org> parents: 8506 diff changeset	144 if (columns(idx) == 1 && top_level == 1)
cadc73247d65 style fixes John W. Eaton <jwe@octave.org> parents: 8506 diff changeset	145 s++;
8338 a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	146 else
8506 bc982528de11 comment style fixes John W. Eaton <jwe@octave.org> parents: 8338 diff changeset	147 # We aren't in top level separator now.
8507 cadc73247d65 style fixes John W. Eaton <jwe@octave.org> parents: 8506 diff changeset	148 top_level = 0;
8338 a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	149 endif
8506 bc982528de11 comment style fixes John W. Eaton <jwe@octave.org> parents: 8338 diff changeset	150 ## If there is not any descendant of "parent node":
8507 cadc73247d65 style fixes John W. Eaton <jwe@octave.org> parents: 8506 diff changeset	151 if (stk(end,2) != par_number)
cadc73247d65 style fixes John W. Eaton <jwe@octave.org> parents: 8506 diff changeset	152 left_most++;
11587 c792872f8942 all script files: untabify and strip trailing whitespace John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	153 x_coordinate_r(par_number) = left_most;
8507 cadc73247d65 style fixes John W. Eaton <jwe@octave.org> parents: 8506 diff changeset	154 max_ht = min (max_ht, level);
cadc73247d65 style fixes John W. Eaton <jwe@octave.org> parents: 8506 diff changeset	155 if (length(stk) > 1 && find ((shift(stk,1)-stk) == 0) > 1
10549 95c3e38098bf Untabify .m scripts Rik <code@nomad.inbox5.com> parents: 9051 diff changeset	156 && stk(end,2) != stk(end-1,2))
95c3e38098bf Untabify .m scripts Rik <code@nomad.inbox5.com> parents: 9051 diff changeset	157 ## Return to the nearest branching the position to return
95c3e38098bf Untabify .m scripts Rik <code@nomad.inbox5.com> parents: 9051 diff changeset	158 ## position is the position on the stack, where should be
8338 a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	159 ## started further search (there are two nodes which has the
8506 bc982528de11 comment style fixes John W. Eaton <jwe@octave.org> parents: 8338 diff changeset	160 ## same parent node).
8338 a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	161
8507 cadc73247d65 style fixes John W. Eaton <jwe@octave.org> parents: 8506 diff changeset	162 position = (find ((shift (stk(:,2), 1) - stk(:,2)) == 0))(end) + 1;
cadc73247d65 style fixes John W. Eaton <jwe@octave.org> parents: 8506 diff changeset	163 par_number_vec = stk(position:end,2);
8338 a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	164
8506 bc982528de11 comment style fixes John W. Eaton <jwe@octave.org> parents: 8338 diff changeset	165 ## The vector of removed nodes (the content of stack form
bc982528de11 comment style fixes John W. Eaton <jwe@octave.org> parents: 8338 diff changeset	166 ## position to end).
8338 a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	167
8507 cadc73247d65 style fixes John W. Eaton <jwe@octave.org> parents: 8506 diff changeset	168 level += length (par_number_vec);
8338 a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	169
10549 95c3e38098bf Untabify .m scripts Rik <code@nomad.inbox5.com> parents: 9051 diff changeset	170 ## The level have to be decreased.
8338 a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	171
8507 cadc73247d65 style fixes John W. Eaton <jwe@octave.org> parents: 8506 diff changeset	172 x_coordinate_r(par_number_vec) = left_most;
cadc73247d65 style fixes John W. Eaton <jwe@octave.org> parents: 8506 diff changeset	173 stk(position:end,:) = [];
11587 c792872f8942 all script files: untabify and strip trailing whitespace John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	174 endif
8338 a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	175
8506 bc982528de11 comment style fixes John W. Eaton <jwe@octave.org> parents: 8338 diff changeset	176 ## Remove the next node from "searched branch".
8338 a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	177
8507 cadc73247d65 style fixes John W. Eaton <jwe@octave.org> parents: 8506 diff changeset	178 stk(end,:) = [];
10549 95c3e38098bf Untabify .m scripts Rik <code@nomad.inbox5.com> parents: 9051 diff changeset	179 ## Choose new "parent node".
8507 cadc73247d65 style fixes John W. Eaton <jwe@octave.org> parents: 8506 diff changeset	180 par_number = stk(end,1);
10549 95c3e38098bf Untabify .m scripts Rik <code@nomad.inbox5.com> parents: 9051 diff changeset	181 ## If there is another branch start to search it.
95c3e38098bf Untabify .m scripts Rik <code@nomad.inbox5.com> parents: 9051 diff changeset	182 if (par_number != -1)
11587 c792872f8942 all script files: untabify and strip trailing whitespace John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	183 y_coordinate(par_number) = level;
8507 cadc73247d65 style fixes John W. Eaton <jwe@octave.org> parents: 8506 diff changeset	184 x_coordinate_l(par_number) = left_most + 1;
10549 95c3e38098bf Untabify .m scripts Rik <code@nomad.inbox5.com> parents: 9051 diff changeset	185 endif
8338 a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	186 else
a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	187
8506 bc982528de11 comment style fixes John W. Eaton <jwe@octave.org> parents: 8338 diff changeset	188 ## There were descendants of "parent nod" choose the last of
bc982528de11 comment style fixes John W. Eaton <jwe@octave.org> parents: 8338 diff changeset	189 ## them and go on through it.
8507 cadc73247d65 style fixes John W. Eaton <jwe@octave.org> parents: 8506 diff changeset	190 level--;
cadc73247d65 style fixes John W. Eaton <jwe@octave.org> parents: 8506 diff changeset	191 par_number = stk(end,1);
11587 c792872f8942 all script files: untabify and strip trailing whitespace John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	192 y_coordinate(par_number) = level;
8507 cadc73247d65 style fixes John W. Eaton <jwe@octave.org> parents: 8506 diff changeset	193 x_coordinate_l(par_number) = left_most + 1;
8338 a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	194 endif
a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	195 endwhile
a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	196
8506 bc982528de11 comment style fixes John W. Eaton <jwe@octave.org> parents: 8338 diff changeset	197 ## Calculate the x coordinates (the known values are the position
bc982528de11 comment style fixes John W. Eaton <jwe@octave.org> parents: 8338 diff changeset	198 ## of most left and most right descendants).
8507 cadc73247d65 style fixes John W. Eaton <jwe@octave.org> parents: 8506 diff changeset	199 x_coordinate = (x_coordinate_l + x_coordinate_r) / 2;
8338 a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	200
8507 cadc73247d65 style fixes John W. Eaton <jwe@octave.org> parents: 8506 diff changeset	201 height = num_nodes - max_ht - 1;
8338 a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	202 endif
a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	203 endfunction
a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	204
14363 f3d52523cde1 Use Octave coding conventions in all m-file %!test blocks Rik <octave@nomad.inbox5.com> parents: 14138 diff changeset	205
13060 85dd509673e7 codesprint: tests for treelayout John W. Eaton <jwe@octave.org> parents: 12575 diff changeset	206 %!test
11587 c792872f8942 all script files: untabify and strip trailing whitespace John W. Eaton <jwe@octave.org> parents: 11523 diff changeset	207 %! % Compute a simple tree layout
13060 85dd509673e7 codesprint: tests for treelayout John W. Eaton <jwe@octave.org> parents: 12575 diff changeset	208 %! [x, y, h, s] = treelayout ([0, 1, 2, 2]);
85dd509673e7 codesprint: tests for treelayout John W. Eaton <jwe@octave.org> parents: 12575 diff changeset	209 %! assert (x, [1.5, 1.5, 2, 1]);
85dd509673e7 codesprint: tests for treelayout John W. Eaton <jwe@octave.org> parents: 12575 diff changeset	210 %! assert (y, [3, 2, 1, 1]);
85dd509673e7 codesprint: tests for treelayout John W. Eaton <jwe@octave.org> parents: 12575 diff changeset	211 %! assert (h, 2);
85dd509673e7 codesprint: tests for treelayout John W. Eaton <jwe@octave.org> parents: 12575 diff changeset	212 %! assert (s, 2);
8338 a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	213
13060 85dd509673e7 codesprint: tests for treelayout John W. Eaton <jwe@octave.org> parents: 12575 diff changeset	214 %!test
8338 a35bf360b919 Add the cgs and treelayout functions Radek Salac <salac.r@gmail.com> parents: diff changeset	215 %! % Compute a simple tree layout with defined postorder permutation
13060 85dd509673e7 codesprint: tests for treelayout John W. Eaton <jwe@octave.org> parents: 12575 diff changeset	216 %! [x, y, h, s] = treelayout ([0, 1, 2, 2], [1, 2, 4, 3]);
85dd509673e7 codesprint: tests for treelayout John W. Eaton <jwe@octave.org> parents: 12575 diff changeset	217 %! assert (x, [1.5, 1.5, 1, 2]);
85dd509673e7 codesprint: tests for treelayout John W. Eaton <jwe@octave.org> parents: 12575 diff changeset	218 %! assert (y, [3, 2, 1, 1]);
85dd509673e7 codesprint: tests for treelayout John W. Eaton <jwe@octave.org> parents: 12575 diff changeset	219 %! assert (h, 2);
85dd509673e7 codesprint: tests for treelayout John W. Eaton <jwe@octave.org> parents: 12575 diff changeset	220 %! assert (s, 2);
85dd509673e7 codesprint: tests for treelayout John W. Eaton <jwe@octave.org> parents: 12575 diff changeset	221
85dd509673e7 codesprint: tests for treelayout John W. Eaton <jwe@octave.org> parents: 12575 diff changeset	222 %!test
85dd509673e7 codesprint: tests for treelayout John W. Eaton <jwe@octave.org> parents: 12575 diff changeset	223 %! % Compute a simple tree layout with defined postorder permutation
85dd509673e7 codesprint: tests for treelayout John W. Eaton <jwe@octave.org> parents: 12575 diff changeset	224 %! [x, y, h, s] = treelayout ([0, 1, 2, 2], [4, 2, 3, 1]);
85dd509673e7 codesprint: tests for treelayout John W. Eaton <jwe@octave.org> parents: 12575 diff changeset	225 %! assert (x, [0, 0, 0, 1]);
85dd509673e7 codesprint: tests for treelayout John W. Eaton <jwe@octave.org> parents: 12575 diff changeset	226 %! assert (y, [0, 0, 0, 3]);
85dd509673e7 codesprint: tests for treelayout John W. Eaton <jwe@octave.org> parents: 12575 diff changeset	227 %! assert (h, 0);
85dd509673e7 codesprint: tests for treelayout John W. Eaton <jwe@octave.org> parents: 12575 diff changeset	228 %! assert (s, 1);
14363 f3d52523cde1 Use Octave coding conventions in all m-file %!test blocks Rik <octave@nomad.inbox5.com> parents: 14138 diff changeset	229

Mercurial > hg > octave-lyh

annotate scripts/sparse/treelayout.m @ 14459:a22a41ab6824