The node you are looking for is at set-display-callback.