T.I. Lakoba and J. Yang, "A generalized Petviashvili method ...", J. Comp. Phys. 226, 1668-1692 (2007).
Fig. 1: 2D NLS equation with periodic potential (3.19)
the code