Make context available in TreeTransform.prepareForXX methods