"""Original Techdokhae scenario arithmetic; NOT an Oracle/GPU benchmark.
Run with Python 3. Edit minute inputs below. No files/network/DB are accessed.
All stages are assumed sequential and non-overlapping. Costs are not modeled.
"""
import json
import math

SCENARIOS = [
    dict(name='large-build', common=60, cpu_build=80, gpu_build=8,
         transfer=12, finish=5, wait=0, deadline=120),
    dict(name='small-build', common=60, cpu_build=8, gpu_build=2,
         transfer=12, finish=5, wait=0, deadline=120),
    dict(name='large-build-with-external-queue', common=60, cpu_build=80,
         gpu_build=8, transfer=12, finish=5, wait=55, deadline=120),
]

def evaluate(s):
    keys = ('common', 'cpu_build', 'gpu_build', 'transfer', 'finish', 'wait', 'deadline')
    if any(isinstance(s[k], bool) or not isinstance(s[k], (int, float))
           or not math.isfinite(s[k]) or s[k] < 0 for k in keys):
        raise ValueError('Minute inputs must be finite nonnegative numbers')
    cpu = s['common'] + s['cpu_build']
    gpu = s['common'] + s['wait'] + s['transfer'] + s['gpu_build'] + s['finish']
    return dict(name=s['name'], cpu_total_minutes=cpu, gpu_total_minutes=gpu,
                saved_minutes=cpu-gpu, speed_ratio=cpu/gpu if gpu else None,
                cpu_meets_deadline=cpu <= s['deadline'],
                gpu_meets_deadline=gpu <= s['deadline'])

if __name__ == '__main__':
    print(json.dumps({'kind': 'hypothetical arithmetic, not measurement',
                      'inputs': SCENARIOS, 'results': [evaluate(s) for s in SCENARIOS],
                      'dense_vector_payload_bytes': {
                          '1000000_vectors_2dims_4bytes': 1000000 * 2 * 4,
                          '1000000_vectors_1536dims_4bytes': 1000000 * 1536 * 4},
                      'payload_excludes': 'graph, workspace, protocol framing; not total VRAM or wire bytes'},
                     ensure_ascii=False, indent=2))