Resumen
Phase mask optimization is one of the critical steps in designing a wavefront coding system to extend the depth of field (DoF). As a classical phase mask, a cubic phase mask was taken as an example. An improved particle swarm optimization (PSO) algorithm was applied to calculate the parameters of the cubic phase mask by introducing the modulation transfer function as the optimization criterion and a threshold as a constraint. The quality of the subsequent image restoration is guaranteed on the premise of the extended DoF. Finally, the improved PSO was proved to be faster, more efficient, and more accurate compared to the simulated annealing algorithm and the traditional PSO. The experimental results verify that the cubic phase mask optimized by the improved PSO can achieve DoF extension in the wavefront coding system. The improved PSO can also be applied to other phase masks of wavefront coding systems.