Puer, concordo apenas parcialmente com o que disseste. Se imaginar-mos uma foca a dormir no mar alto(a meio de uma rota migratória por exemplo), esta corre tanto risco como um peixe a dormir no mesmo ambiente.
Não acho que os diferentes tipos de sono estejam condicionados apenas pelo habitat, a evolução deu-se também em termos do tipo de sono. Os animais mais evoluídos (logo mais complexos) tem um sono mais evoluído, apesar de entre estes teres diferentes padrões dentro do mesmo tipo de sono.
Logo o sono não é um "luxo"(no sentido em que é o habitat que o permite) mas sim uma necessidade biológica, que tem uma derribada função(que não é o sono, mas sim o que ocorre durante este).
Se tomar-mos como objecto de estudo os mamíferos e analisar-mos uma característica do sono que é conhecida como Rapid Eye Movement(REM) mas que está também associada ao ritmo respiratório e a espasmos, podemos observar que este fenómeno varia entre os mamíferos, apesar de praticamente todos passarem por ele. Este fenómeno sim, depende da segurança do animal. Um predador tem por norma uma fase de REM superior à de uma presa mas ambos podem passar por esta fase.
O tempo de sono não está apenas relacionado com as condições para este poder ter lugar, pois o tamanho e o metabolismo de um animal também influenciam o tempo de sono que este tem, geralmente quanto maior é o animal(e mais lento o metabolismo) menor a necessidade de sono.
Exemplo: um elefante, que raramente é uma presa, dorme 4 horas, enquanto um porco selvagem, que é mais frequentemente uma presa, dorme 8 horas. Neste caso o factor risco não é inversamente proporcional ao tempo de sono, pelo que se demonstra que não é apenas devido ao habitat que os animais dormem mais, ou menos.