Modeling chromatic emittance growth in staged plasma wakefield acceleration to 1 TeV using nonlinear transfer matrices