4momentum calculation