Reservoir energy storage calculation