What is the difference between treatment and rehabilitation when it comes to psychiatric illnesses? 

My understanding is that treatment is just controlling symptoms, whereas rehabilitation is re-integrating back into society. Am I correct?