The Schwartz space and homogeneous distributions on $H$-type groups