Modeling betatron radiation using particle-in-cell codes for plasma wakefield accelerator diagnostics