@basic-genomics/hivtrace-viz 1.1.10 → 1.1.11

package/dist/embed/index.html

@@ -1031,9 +1031,66 @@
                     user_graph.handle_attribute_categorical(user_graph.colorizer['category_id'], false);
                   }
                 }
+
+                // 更新 pill badges（唯一值数量）
+                updatePillBadges(selectedClusterId, clusters);
               }
             });
 
+            // 更新 pill badges 以反映选定簇的唯一值数量
+            function updatePillBadges(selectedClusterId, clusters) {
+              var pills = document.querySelectorAll('.hivtrace-pill-row .hivtrace-pill');
+              if (!pills.length) return;
+
+              // 确定要统计的节点集合
+              var nodes = [];
+              if (!selectedClusterId) {
+                // 全网：使用所有节点
+                nodes = user_graph.nodes || [];
+              } else {
+                // 特定簇：只使用该簇的节点
+                var cluster = clusters.find(function (c) {
+                  return String(c.cluster_id) === String(selectedClusterId);
+                });
+                if (cluster && cluster.children) {
+                  nodes = cluster.children;
+                }
+              }
+
+              // 更新每个 pill 的 badge
+              pills.forEach(function (pill) {
+                var badge = pill.querySelector('.badge');
+                if (!badge) return;
+
+                // 从 pill 的原始 HTML 中提取属性 ID（通过闭包数据不可用，需要通过文本匹配）
+                var pillText = pill.textContent.trim();
+                var attrId = null;
+
+                // 遍历 user_graph.json 中的属性来匹配
+                if (user_graph.json && user_graph.json.patient_attribute_schema) {
+                  for (var key in user_graph.json.patient_attribute_schema) {
+                    var attr = user_graph.json.patient_attribute_schema[key];
+                    if (attr.label && pillText.indexOf(attr.label) === 0) {
+                      attrId = key;
+                      break;
+                    }
+                  }
+                }
+
+                if (attrId) {
+                  // 计算选定节点中该属性的唯一值数量
+                  var uniqueValues = new Set();
+                  nodes.forEach(function (node) {
+                    var value = user_graph.attribute_node_value_by_id(node, attrId);
+                    if (value !== null && value !== undefined && value !== 'missing') {
+                      uniqueValues.add(value);
+                    }
+                  });
+                  badge.textContent = uniqueValues.size;
+                }
+              });
+            }
+
 
             // 计算并显示单个 cluster 的统计
             function updateClusterStatistics(cluster) {