sched: Refactor the power savings balance code