Abstract: A optical klystron(OK) to be used for free electron laser experiments with wavelength from visible to ultra-violet is now in the stage of design. Authors provide a code for computational 2-D optical klystron magnetic field.