A wavefront of a light beam that exits an acousto-optic material is estimated; a control signal for an acousto-optic system that includes the acousto-optic material is generated, the control signal being based on the estimated wavefront of the light beam; andthe control signal is applied to the acousto-optic system to generate a frequency-chirped acoustic wave that propagates in the acousto-optic material, the frequency-chirped acoustic wave forming a transient diffractive element in the acousto-optic material, an interaction between the transient diffractive element and the light beam adjusting the wavefront of the light beam to compensate for a distortion of the wavefront of the light beam, the distortion of the wavefront being at least partially caused by a physical effect in the acousto-optic material.