working with collapsing nodes

Author: freeman-lab

build.js

@@ -26,6 +26,30 @@ function build (people, pairs) {
     }
   })
 
+  // get a flattened list of all people who are parents
+  var names
+  var allnames = []
+  _.forEach(groups, function (v, k) {
+    if (k != '') {
+      names = k.split(',')
+      allnames.push(names)
+    }
+  })
+  allnames = _.flatten(allnames)
+
+  // tag everyone who is a parent
+  people = _.map(people, function (p) {
+    if (_.includes(allnames, p.name)) {
+      p.is_parent = true
+    } else {
+      p.is_parent = false
+    }
+    return p
+  })
+
+  // regenerate groups
+  var groups = _.groupBy(people, 'parents')
+
   // for anyone in the empty group
   // add them to the group with their spouse
   var spouse
@@ -66,6 +90,8 @@ function build (people, pairs) {
     groups[''].splice(foundIndex, 1)
   }
 
+
+
   // go up the tree inserting groups of children
   // in between their parents
   var child
@@ -94,7 +120,7 @@ function build (people, pairs) {
   }
 
   groups[''] = sort(groups[''], max)
-  
+
   // wrap elements in arrays based on depth
   var level
   var diff
@@ -198,12 +224,23 @@ function build (people, pairs) {
   }
 
   // build the tree
+  //
+  // warning: 
+  // super hacky handling of cousin marriage
+  // manually specifying that a parent node is associated
+  // with cousin marriage if the first child is
+  // and basing the parents off the first child
+  var check
+  var cousin
   function buildTree (array, root) {
     if (!root) {
       root = {
         name: '',
         hidden: true,
         no_parent: true,
+        is_parent: true,
+        cousin_marriage: array[0].cousin_marriage,
+        parents: array[0].cousin_marriage ? array[0].parents : null,
         id: id
       }
       id += 1
@@ -213,7 +250,10 @@ function build (people, pairs) {
       if (!_.isArray(d)) {
         subtree = {
           name: d.name,
+          is_parent: d.is_parent,
           no_parent: (d.parents.length == 0) ? true : false,
+          parents: d.parents,
+          cousin_marriage: d.cousin_marriage,
           id: id
         }
         id += 1
@@ -228,6 +268,7 @@ function build (people, pairs) {
   }
 
   var id = 0
+  
   var root = buildTree(groups[''])
 
   var married = _.map(pairs, function (d) {
@@ -237,22 +278,29 @@ function build (people, pairs) {
   //display(root)
   //display(married)
 
-  return {root: root, married: married}
+  return {data: root, married: married}
 }
 
 function sort (input, max) {
 
   var level = 0
   var iter = 0
 
+  //console.log(input)
+
   // loop over levels
   // for each level keep finding nodes that need to be reordered
   // when no more are found advance to the next level
   while ((level < max) & iter < 5000) {
     var selected = _.findIndex(input, function (d, k) {
       if (getLevel(d) == level) {
         var parents = getParents(input[k])
-        if ((parents.length > 0) &&
+        if (parents.length == 1) {
+          // hacky flag to indicate the individual is the result of cousin marriage
+          display('warning! length 1 parent, probably due to cousin marriage')
+          input[k][0].cousin_marriage = true
+        }
+        if ((parents.length == 2) &&
             (!((parents[0] == (k - 1)) && (parents[1] == (k + 1)))) &&
             (!((parents[0] == (k + 1)) && (parents[1] == (k - 1))))) {
           return true

components/graph.js

@@ -16,7 +16,5 @@ module.exports = function () {
   })
   document.body.appendChild(container)
 
-  function graph () {}
-
-  return graph
+  return container
 }

examples/kk.js

@@ -1,33 +1,40 @@
 var people = [
   {name: 'sam', parents: ['brad', 'karen'], level: 0},
   {name: 'emily', parents: ['brad', 'karen'], level: 0},
+  {name: 'willem', parents: ['cherie', 'aj'], level: 0},
+  {name: 'audrey', parents: ['cherie', 'aj'], level: 0},
   {name: 'brad', parents: ['jim', 'liz'], level: 1},
   {name: 'karen', parents: [], level: 1},
   {name: 'andrew', parents: ['richard', 'marie'], level: 1},
   {name: 'kk', parents: ['richard', 'marie'], level: 1},
-  {name: 'cheri', parents: ['marie', 'dick'], level: 1},
-  {name: 'son', parents: ['cynthia', 'husband'], level: 1},
-  {name: 'daughter', parents: ['cynthia', 'husband'], level: 1},
+  {name: 'aj', parents: [], level: 1},
+  {name: 'cherie', parents: ['marie', 'dick'], level: 1},
+  {name: 'cody', parents: ['cindy', 'paul miller'], level: 1},
+  {name: 'eiann', parents: ['cindy', 'paul miller'], level: 1},
+  {name: 'rebecca', parents: ['susan', 'padre'], level: 1},
   {name: 'richard', parents: ['leonard', 'catherine'], level: 2},
   {name: 'marie',  parents: ['barton', 'jean'], level: 2},
   {name: 'liz', parents: ['barton', 'jean'], level: 2},
   {name: 'jim',  parents: [], order: -1, level: 2},
   {name: 'dick', parents: [], order: 0, level: 2},
-  {name: 'cynthia', parents: ['barton', 'housekeeper'], level: 2},
-  {name: 'husband', parents: [], level: 2},
+  {name: 'paul miller', parents: [], order: -1, level: 2},
+  {name: 'cindy', parents: ['barton', 'dede'], level: 2},
+  {name: 'susan', parents: ['barton', 'dede'], level: 2},
+  {name: 'padre', parents: [], level: 2},
+  //{name: 'mary jean', parents: ['dr jim', 'ruth'], level: 2},
+  {name: 'georganna', parents: ['lois', 'robert jr'], level: 2},
   // {name: 'michelene', parents: ['william yates', 'micky'], offset: 10, level: 2},
   // {name: 'michelene husband', parents: [], level: 2},
   // {name: 'monica erlich', parents: ['mary jean', 'ehrlich'], level: 3},
   // {name: 'ehrlich', parents: [], level: 3},
-  {name: 'mary jean', parents: ['ruth', 'dr jim'], level: 3},
   {name: 'lawrence', parents: ['mary lynn', 'hulet'], level: 1},
   {name: 'catherine', parents: ['james', 'annie'], level: 3},
   {name: 'bill', parents: ['james', 'annie'], level: 3},
   {name: 'richard pepper', parents: ['james', 'annie'], level: 3},
   {name: 'emil', parents: ['dora', 'lukas'], level: 3},
   {name: 'leonard', parents: ['dora', 'lukas'], level: 3},
   {name: 'barton', parents: ['marie swift', 'laurence'], level: 3},
-  {name: 'housekeeper', parents: [], order: 0, level: 3},
+  {name: 'dede', parents: [], order: 0, level: 3},
   {name: 'mary lynn', parents: ['marie swift', 'laurence'], level: 3},
   {name: 'beatrice', parents: ['marie swift', 'laurence'], level: 3},
   {name: 'frederick', parents: [], level: 3},
@@ -39,18 +46,18 @@ var people = [
   {name: 'micky', parents: [], level: 3},
   {name: 'thomas spens', parents: ['james andrew', 'honora'], level: 3},
   {name: 'jean', parents: ['james andrew', 'honora'], level: 3},
+  {name: 'robert jr', parents: ['lilian frances', 'robert'], level: 3},
   {name: 'jimmy', parents: ['lilian frances', 'robert'], level: 3},
   {name: 'hulet', parents: [], order: 0, level: 3},
   {name: 'gray 1', parents: ['margurite ellen', 'harold'], level: 3},
   {name: 'gray 2', parents: ['margurite ellen', 'harold'], level: 3},
+  {name: 'lois', parents: ['george a', 'leora'], level: 3},
   {name: 'boy', parents: ['clark', 'beatrice alice'], level: 4},
   {name: 'james andrew', parents: ['josephine', 'ec'], level: 4},
   {name: 'honora', parents: [], level: 4},
   {name: 'dora', parents: [], level: 4},
   {name: 'lukas', parents: [], level: 4},
   {name: 'marie swift', parents: ['samuel george', 'beatrice ann'], level: 4},
-  {name: 'george a', parents: ['samuel george', 'beatrice ann'], level: 4},
-  {name: 'leora', parents: [], level: 4},
   {name: 'cecil', parents: ['samuel george', 'beatrice ann'], level: 4},
   {name: 'edwin', parents: ['samuel george', 'beatrice ann'], level: 4},
   {name: 'samuel george II', parents: ['samuel george', 'beatrice ann'], level: 4},
@@ -66,23 +73,31 @@ var people = [
   {name: 'raymond', parents: [], order: 0, level: 4},
   {name: 'alma dean', parents: ['samuel george', 'beatrice ann'], level: 4},
   {name: 'wilford', parents: ['samuel george', 'beatrice ann'], level: 4},
+  {name: 'george a', parents: ['samuel george', 'beatrice ann'], level: 4},
+  {name: 'leora', parents: [], level: 4},
   {name: 'lilian frances', parents: ['samuel george', 'beatrice ann'], level: 4},
   {name: 'laurence', parents: ['albert joseph', 'mary ladoiska'], level: 4},
   {name: 'robert', parents: [], level: 4},
   {name: 'james', parents: [], level: 4},
   {name: 'annie', parents: [], level: 4},
-  {name: 'josephine', parents: [], level: 5},
+  {name: 'josephine', parents: ['caroline leroux', 'james yates'], level: 5},
   {name: 'ec', parents: [], level: 5},
   {name: 'samuel george', parents: ['george r', 'mary anne'], level: 5},
   {name: 'beatrice ann', parents: ['hannah marie', 'alfred swift'], level: 5},
   {name: 'albert joseph', parents: ['henry talbot', 'ruth sweetnam'], level: 5},
   {name: 'henry talbot', parents: ['jst', 'priscilla'], level: 6},
+  {name: 'caroline leroux', parents: [], level: 6},
+  {name: 'james yates', parents: ['richard yates', 'caroline nicolson'], level: 6},
   {name: 'ruth sweetnam', parents: [], level: 6},
   {name: 'mary ladoiska', parents: [], level: 5},
   {name: 'hannah marie', parents: [], level: 6},
   {name: 'alfred swift', parents: [], level: 6},
   {name: 'george r', parents: [], level: 6},
   {name: 'mary anne', parents: [], level: 6},
+  {name: 'richard yates', parents: [], level: 7},
+  {name: 'caroline nicolson', parents: [], level: 7},
+  // {name: 'benjamin leroux', parents: [], level: 7},
+  // {name: 'mary gauthier', parents: [], level: 7},
   {name: 'jst', parents: [], level: 7},
   {name: 'priscilla', parents: [], level: 7}
 ]
@@ -91,13 +106,13 @@ var pairs = [
   {married: ['marie', 'dick']},
   {married: ['frederick', 'beatrice']},
   {married: ['margurite', 'paul']},
-  {married: ['cynthia', 'husband']},
   {married: ['leora', 'george a']},
   {married: ['anona', 'arlos']},
   {married: ['anona', 'raymond']},
   {married: ['babe', 'frank']},
   {married: ['william yates', 'micky']},
-  {married: ['ruth', 'dr jim']}
+  {married: ['ruth', 'dr jim']},
+  {married: ['lois', 'robert jr']}
 ]
 
 var metadata = {