It's clearly much better than that. The problem is that, in a medium like water, light of different wavelengths absorb differently. So the longer distance the l

It's clearly much better than that. The problem is that, in a medium like water, light of different wavelengths absorb differently. So the longer distance the light travels through water, the more the effect is, and this effect is different depending on the wavelength. So you can't just open Photoshop and change the color levels of everything in the picture simultaneously. You actually have to know/estimate distances to different objects and make adjustments locally based on that.