Water treatment process design and control