Separate assign_happens_before() into:

update_trace_state()
update_clock_map()