feat(commDetection): ✨ Label as group name!

SkepticMystic · Nov 4, 2021 · 243fd05 · 243fd05
1 parent eb03e1e
commit 243fd05
Show file tree

Hide file tree

Showing 4 changed files with 26 additions and 20 deletions.
diff --git a/main.js b/main.js
diff --git a/src/Components/CommunityDetection.svelte b/src/Components/CommunityDetection.svelte
@@ -13,7 +13,6 @@
   export let settings: GraphAnalysisSettings
   export let view: AnalysisView
 
-
   let currFile = app.workspace.getActiveFile()
   $: currNode = currFile?.path.split('.md', 1)[0]
   app.workspace.on('active-leaf-change', () => {
@@ -55,7 +54,7 @@
   </div>
   {#if promiseSortedComms}
     {#await promiseSortedComms then sortedComms}
-      {#each sortedComms as comm, i}
+      {#each sortedComms as comm}
         {#if comm.comm.length > 1}
           <div class="GA-CC">
             <details>
@@ -69,7 +68,7 @@
                     : ''}"
                 >
                   <span>
-                    {i}
+                    {comm.label}
                   </span>
                   <span class={TD_MEASURE}>{comm.comm.length}</span>
                 </span>

diff --git a/src/MyGraph.ts b/src/MyGraph.ts
@@ -379,13 +379,13 @@ export default class MyGraph extends Graph {
       a: string,
       options: { iterations: number }
     ): Promise<Communities> => {
-      let labeledNodes: { [node: string]: number } = {}
+      let labeledNodes: { [node: string]: string } = {}
       this.nodes().forEach((node, i) => {
-        labeledNodes[node] = i
+        labeledNodes[node] = node
       })
 
       for (let i = 0; i < options.iterations; i++) {
-        const newLabeledNodes: {[node: string]: number} = {}
+        const newLabeledNodes: { [node: string]: string } = {}
         this.nodes().forEach((node) => {
           const neighbours = this.neighbors(node) as string[]
 
@@ -397,14 +397,14 @@ export default class MyGraph extends Graph {
             const maxKey = getMaxKey(counts)
             // console.log({neighbours,neighbourLabels, counts, maxKey})
 
-            newLabeledNodes[node] = Number.parseInt(maxKey)
+            newLabeledNodes[node] = maxKey
           }
         })
         labeledNodes = newLabeledNodes
       }
 
       const communities: Communities = {}
-      Object.entries(labeledNodes).forEach((labeledNode: [string, number]) => {
+      Object.entries(labeledNodes).forEach((labeledNode: [string, string]) => {
         const [node, label] = labeledNode
         if (communities[label] === undefined) {
           communities[label] = [node]

diff --git a/src/Utility.ts b/src/Utility.ts
@@ -237,7 +237,7 @@ export function getPromiseResults(
   return resultsPromise
 }
 
-export function getCounts(arr: number[]) {
+export function getCounts(arr: any[]) {
   const counts: { [item: string]: number } = {}
   for (const num of arr) {
     counts[num] = counts[num] ? counts[num] + 1 : 1