Modeling mass entrainment in a quasi-one-dimensional shock tube code