proof_of_concept_04_charts.py

import json
import matplotlib.pyplot as plt
from collections import defaultdict, Counter
import os

# =======================
# INITIAL CONFIGURATION
# =======================

TRACEABILITY_FILE = 'proof_of_concept/traceability_map.json'
OUTPUT_DIR = 'analysis_charts'

# Create output directory if it doesn't exist
os.makedirs(OUTPUT_DIR, exist_ok=True)

# Relevant tags per research question
TAG_GROUPS = {
    'testing_needed': ['testing', 'fix', 'error'],
    'mock_usage': ['mock'],
    'refactor_needed': ['mock', 'refactor'],
    'high_activity': ['testing', 'model', 'config'],
    'ci_cd_issues': ['config', 'model', 'fix'],
}

# Nice titles for the charts
TITLES = {
    'testing_needed': '🔬 Features that need more testing',
    'mock_usage': '🧪 Features with heavy mock usage',
    'refactor_needed': '🔧 Features requiring refactor for testability',
    'high_activity': '📈 Most active features (testing, model, config)',
    'ci_cd_issues': '⚙️ Features with CI/CD issues',
}

# =======================
# LOAD DATA
# =======================

with open(TRACEABILITY_FILE) as f:
    traceability_map = json.load(f)

# Group by feature and count tags
feature_tag_count = defaultdict(Counter)
for feature, source_type, tag in traceability_map:
    feature_tag_count[feature][tag] += 1

# Helper function to sum tag counts per group
def sum_tags(counter, tags):
    return sum(counter[tag] for tag in tags)

# =======================
# GENERATE CHARTS
# =======================

def generate_chart(question_key, tags):
    data = [
        (feature, sum_tags(counter, tags))
        for feature, counter in feature_tag_count.items()
        if sum_tags(counter, tags) > 0
    ]
    if not data:
        return

    data.sort(key=lambda x: x[1], reverse=True)
    features, counts = zip(*data)

    plt.figure(figsize=(10, 5))
    plt.bar(features, counts)
    plt.title(TITLES[question_key])
    plt.ylabel('Occurrences')
    plt.xlabel('Feature')
    plt.xticks(rotation=45, ha='right')
    plt.tight_layout()

    output_path = os.path.join(OUTPUT_DIR, f'{question_key}.png')
    plt.savefig(output_path)
    plt.close()

    print(f'[✓] Chart generated: {output_path}')

# Generate all charts
for key, tags in TAG_GROUPS.items():
    generate_chart(key, tags)