convert_to_csv.py

import json
import csv
import os

def convert_to_csv(goodput_path="results/goodput.json", output_dir="results/csv", marker=None):
    with open(goodput_path, 'r') as f:
        data = json.load(f)

    output_dir = output_dir
    if not os.path.exists(output_dir):
        os.makedirs(output_dir)

    for idx, server in enumerate(data):
        samples = server['tcpstats_samples']
        prefix = f"{marker}_" if marker else ""
        filename = os.path.join(output_dir, f'{prefix}trace_{idx}.csv')
        
        with open(filename, 'w', newline='') as csvfile:
            writer = csv.writer(csvfile)
            writer.writerow(['time', 'goodput_bps', 'rtt_ms', 'loss', 'snd_cwnd'])
            
            for i, sample in enumerate(samples):
                writer.writerow([
                    sample.get('t', ''),
                    sample.get('goodput_bps', ''),
                    sample.get('rtt_ms', ''),
                    sample.get('loss', ''),
                    sample.get('cwnd', '')
                ])
        
        print(f'Created trace_{idx}.csv for server: {server["server"]}')