about summary refs log tree commit diff
path: root/sourcecodes/bnt-master/graph/children.m
diff options
context:
space:
mode:
Diffstat (limited to 'sourcecodes/bnt-master/graph/children.m')
-rw-r--r--sourcecodes/bnt-master/graph/children.m17
1 files changed, 17 insertions, 0 deletions
diff --git a/sourcecodes/bnt-master/graph/children.m b/sourcecodes/bnt-master/graph/children.m
new file mode 100644
index 00000000..d79701b7
--- /dev/null
+++ b/sourcecodes/bnt-master/graph/children.m
@@ -0,0 +1,17 @@
+function cs = children(adj_mat, i, t)
+% CHILDREN Return the indices of a node's children in sorted order
+% c = children(adj_mat, i, t)
+%
+% t is an optional argument: if present, dag is assumed to be a 2-slice DBN
+
+if nargin < 3 
+  cs = find(adj_mat(i,:));
+else
+  if t==1
+    cs = find(adj_mat(i,:));
+  else
+    ss = length(adj_mat)/2;
+    j = i+ss;
+    cs = find(adj_mat(j,:)) + (t-2)*ss;
+  end
+end