'
except (FileNotFoundError, IOError) as e:
self.logo_html = ''
print(f"Warning: Could not load Tachyon logo: {e}")
except (FileNotFoundError, IOError) as e:
raise RuntimeError(f"Failed to load heatmap template files: {e}") from e
class _TreeBuilder:
"""Builds hierarchical tree structure from file statistics."""
@staticmethod
def build_file_tree(file_stats: List[FileStats]) -> Dict[str, TreeNode]:
"""Build hierarchical tree grouped by module type, then by module structure.
Args:
file_stats: List of FileStats objects
Returns:
Dictionary mapping module types to their tree roots
"""
# Group by module type first
type_groups = {'stdlib': [], 'site-packages': [], 'project': [], 'other': []}
for stat in file_stats:
type_groups[stat.module_type].append(stat)
# Build tree for each type
trees = {}
for module_type, stats in type_groups.items():
if not stats:
continue
root_node = TreeNode()
for stat in stats:
module_name = stat.module_name
parts = module_name.split('.')
# Navigate/create tree structure
current_node = root_node
for i, part in enumerate(parts):
if i == len(parts) - 1:
# Last part - store the file
current_node.files.append(stat)
else:
# Intermediate part - create or navigate
if part not in current_node.children:
current_node.children[part] = TreeNode()
current_node = current_node.children[part]
# Calculate aggregate stats for this type's tree
_TreeBuilder._calculate_node_stats(root_node)
trees[module_type] = root_node
return trees
@staticmethod
def _calculate_node_stats(node: TreeNode) -> Tuple[int, int]:
"""Recursively calculate aggregate statistics for tree nodes.
Args:
node: TreeNode to calculate stats for
Returns:
Tuple of (total_samples, file_count)
"""
total_samples = 0
file_count = 0
# Count files at this level
for file_stat in node.files:
total_samples += file_stat.total_samples
file_count += 1
# Recursively process children
for child in node.children.values():
child_samples, child_count = _TreeBuilder._calculate_node_stats(child)
total_samples += child_samples
file_count += child_count
node.samples = total_samples
node.count = file_count
return total_samples, file_count
class _HtmlRenderer:
"""Renders hierarchical tree structures as HTML."""
def __init__(self, file_index: Dict[str, str]):
"""Initialize renderer with file index.
Args:
file_index: Mapping from filenames to HTML file names
"""
self.file_index = file_index
self.heatmap_bar_height = 16
def render_hierarchical_html(self, trees: Dict[str, TreeNode]) -> str:
"""Build hierarchical HTML with type sections and collapsible module folders.
Args:
trees: Dictionary mapping module types to tree roots
Returns:
Complete HTML string for all sections
"""
type_names = {
'stdlib': '📚 Standard Library',
'site-packages': '📦 Site Packages',
'project': '🏗️ Project Files',
'other': '📄 Other Files'
}
sections = []
for module_type in ['project', 'stdlib', 'site-packages', 'other']:
if module_type not in trees:
continue
tree = trees[module_type]
# Project starts expanded, others start collapsed
is_collapsed = module_type in {'stdlib', 'site-packages', 'other'}
icon = '▶' if is_collapsed else '▼'
content_style = ' style="display: none;"' if is_collapsed else ''
file_word = "file" if tree.count == 1 else "files"
sample_word = "sample" if tree.samples == 1 else "samples"
section_html = f'''