Replace tree branches with calculated true energy

Instead of reading from an ntuple branch, calculate the true energy from
true particle ID and true 3-momentum.