Are there special rehab centers for people with dementia recovering from a hip surgery?

Could anyone recommend best rehabs near The Villages, FL. for an 81 year old person with dementia who just had a hip surgery?
My mother had a hip surgery and we need to find a rehab that best suits her. At nights she likes to stay awake and move around. She used to be a very active person and a career woman. I'm afraid she will get up at night and take another fall that could be this time very critical. Although rehabs have beds with alarms if she gets up, I am worried that the help will get there too later since she can be very quick once she starts recovering and not realize her condition is still weak from dementia.