What's this sci-fi film?

Ok, a man and a woman get separated. I think it's cause' the woman is
pregnant. They're both sent to prisons. In the prison the men have
their dreams watched and aren't allowed to dream.

That's all i can remember really.