Add support for initial wave states (i.e. letting the wave function know about sky/ground/walls in the input image).