plugins/elements/compose.py: Only calculate snapshot when needed

This is an optimisation for the case where the compose element doesn't
do any splitting, and also brings 2 related code fragments closer to
each other.