Wrong calculation of raw softdrop mass for AK8 jets?

[nurfikri89]

From the snippet below, it looks like the raw softdrop mass is wrongly calculated. The subjets retrieved here are with jet energy corrections (JECs) applied on them [1]. From what I understand of what we mean by the raw softdrop mass is the invariant mass of the subjet pair with the JECs removed from each subjet four-vector.

nanoAOD-tools/python/postprocessing/modules/jme/fatJetUncertainties.py

Lines 549 to 555 in 4396652

	if jet.subJetIdx1 >= 0 and jet.subJetIdx2 >= 0:
	groomedP4 = subJets[jet.subJetIdx1].p4() + subJets[
	jet.subJetIdx2].p4() # check subjet jecs
	else:
	groomedP4 = None
	jet_msdcorr_raw = groomedP4.M() if groomedP4 is not None else 0.0

[1] at the nanoAOD production level, or probably even earlier at the miniAOD production level

[nurfikri89]

I think line 550-551 can be replaced by:

groomedP4 = (subJets[jet.subJetIdx1].p4() * (1. - subJets[jet.subJetIdx1].rawFactor)) \
     + (subJets[jet.subJetIdx2].p4() * (1. - subJets[jet.subJetIdx2].rawFactor))