[Feature] plot particle energy