Кнопка удаления не отображается на UITableViewCell с закругленными углами

Question

В моем представлении таблицы у меня есть логика в моем tableView(tableView: UITableView, willDisplayCell cell: UITableViewCell, forRowAtIndexPath indexPath: NSIndexPath), чтобы определить, имеет ли ячейка закругленные углы сверху, снизу или без закругленных углов.

override func tableView(tableView: UITableView, willDisplayCell cell: UITableViewCell, forRowAtIndexPath indexPath: NSIndexPath) { 

     if indexPath.row == 0 { 
      // round top 2 corners 
      let maskLayer = CAShapeLayer() 
      maskLayer.path = UIBezierPath(roundedRect: cell.bounds, byRoundingCorners: [UIRectCorner .TopLeft, UIRectCorner .TopRight], cornerRadii: CGSizeMake(4, 4)).CGPath 
      cell.layer.mask = maskLayer 
     } 

     if indexPath.section != 1 { 

      if indexPath.row == self.tableView.numberOfRowsInSection(indexPath.section) - 1 { 
       let maskLayer = CAShapeLayer() 
       maskLayer.path = UIBezierPath(roundedRect: cell.bounds, byRoundingCorners: [UIRectCorner .BottomLeft, UIRectCorner .BottomRight], cornerRadii: CGSizeMake(4, 4)).CGPath 
       cell.layer.mask = maskLayer 
      } 
     } 

    } 

Это работает, но после того, как я внедрил редактирование в виде таблицы, с кнопкой удаления происходят странные вещи.

С закругленными углами код:

Когда я удалить скругленные углы:

Это действительно сбивает с толку меня, потому что я не уверен, что я делаю неправильно. Может ли кто-нибудь помочь?

Answer 1

override func tableView(tableView: UITableView, willDisplayCell cell: UITableViewCell, forRowAtIndexPath indexPath: NSIndexPath) { 

     if indexPath.row == 0 { 
      // round top 2 corners 
      let maskLayer = CAShapeLayer() 
      maskLayer.path = UIBezierPath(roundedRect: cell.bounds, byRoundingCorners: [UIRectCorner .TopLeft, UIRectCorner .TopRight], cornerRadii: CGSizeMake(4, 4)).CGPath 
      cell.contentView.layer.mask = maskLayer 
     } 

     if indexPath.section != 1 { 

      if indexPath.row == self.tableView.numberOfRowsInSection(indexPath.section) - 1 { 
       let maskLayer = CAShapeLayer() 
       maskLayer.path = UIBezierPath(roundedRect: cell.bounds, byRoundingCorners: [UIRectCorner .BottomLeft, UIRectCorner .BottomRight], cornerRadii: CGSizeMake(4, 4)).CGPath 
       cell.contentView.layer.mask = maskLayer 
      } 
     } 

    } 

использование cell.contentView.layer.mask = maskLayer