fix proton mask constant error, extend getlorentzVector

Please check explicitly which pTypes you want to have (e.g. "pType == 2 or pType ==3"). It is less confusing to read and more robust should one add further pTypes.