Modeling the wave normal distribution of chorus waves