The pore morphology method (PMM) has been widely applied to simulate the quasi–static drainage process of porous media, which requires less computational time and computer memory compared to other two–phase flow simulation methods. However, most of the existing pore morphology simulations were restricted to totally wetting systems (i.e., contact angle is equal to zero). Here, we propose a new pore morphology method (Lvca–PMM) enhanced by considering locally variable contact angles, which now enables a reproduction of the primary drainage process for porous media under different wettability conditions featuring the full range of contact angle for wetting phase for various spatial distribution. A series of simulations are performed by using the Lvca–PMM to a porous medium under different uniform and mixed wettability conditions. The proposed Lvca–PMM presents a better result compared to the method by Schulz et al. (2014) for different wettability conditions. The capillary pressure–saturation relations as well as the pore–scale fluid distributions can be well captured by the proposed Lvca-PMM.