Mediation analysis with principal stratification