Add threeMomentum for State

This MR implements the use of the functors P and PT on a State. This is achieved implementing the function threeMomentum when the input is a State.