iOS размер экрана устройства

Question

Я хочу назначить высоту ячейки табличного представления в соответствии с высотой устройства. Я знаю, что могу получить высоту, но могу ли я назначить, например, высоту ячейки табличного представления, например

func tableView(_ tableView: UITableView, heightForRowAt indexPath: IndexPath) -> CGFloat {
    switch indexPath.section {
      case 0:
        return 0.9 * UIScreen.main.bounds.width + 440 /// i want like lessthan(0.9 * UIScreen.main.bounds.width + 440)
      case 1:
        let episodesCount: CGFloat = 2
        let seasonDetailsHeight: CGFloat = 104
        return 100 * episodesCount + seasonDetailsHeight
      case 2:
        return ((UIScreen.main.bounds.height) / 4)
      default: return 0
    }
  }