Swift3.0 UICollectionView 删除,拖动

UICollectionView实现了一下常见的新闻分类.  附有效果图

近期一直在深入学习swift,实现了CollectionView item的头东与删除,用的都是系统的一些函数方法,看起来比较直观.

第一步:

class HotViewController: UIViewController,UICollectionViewDelegate,UICollectionViewDataSource,UICollectionViewDelegateFlowLayout {

    //声明两个存放字符串的数组
    var nowClassName = [String]()
    var surplusClassName = [String]()

    //是否排序
    var isRank = Bool()

    var collectionView : UICollectionView?
    override func viewDidLoad() {
        super.viewDidLoad()

        self.view.backgroundColor = ColorViewBG

        let layout = UICollectionViewFlowLayout()
        layout.itemSize = CGSize(width:,height:)
        //列间距,行间距,偏移
        layout.minimumInteritemSpacing =
        layout.minimumLineSpacing =
        layout.sectionInset = UIEdgeInsetsMake(, , , )

        collectionView = UICollectionView.init(frame: self.view.bounds, collectionViewLayout: layout)
        collectionView?.delegate = self
        collectionView?.dataSource = self;
        //注册一个cell
        collectionView!.register(HotCell.self, forCellWithReuseIdentifier:"HotCell")
        //注册区头
        collectionView?.register(UICollectionReusableView.self, forSupplementaryViewOfKind:UICollectionElementKindSectionHeader, withReuseIdentifier: "headView")

        collectionView?.backgroundColor = ColorViewBG
        self.view.addSubview(collectionView!)
        let gesture = UILongPressGestureRecognizer(target: self, action: #selector(viewCustom(_ :)))

        collectionView?.addGestureRecognizer(gesture)

        saveData()
    }

func viewCustom(_ longPress:UILongPressGestureRecognizer){

        let point:CGPoint = longPress.location(in: longPress.view)

        let indexPath = self.collectionView?.indexPathForItem(at: point)

        switch longPress.state {
        case .began:

            self.collectionView?.beginInteractiveMovementForItem(at: indexPath!)
            break
        case .changed:
            self.collectionView?.updateInteractiveMovementTargetPosition(point)

            break

        case .ended:
            self.collectionView?.endInteractiveMovement()
            break

        default:
            self.collectionView?.cancelInteractiveMovement()
            break
        }

    }

 //添加数据
    private func saveData() {
        nowClassName += ["A-1","A-2","A-3","A-4","A-5","A-6","A-7","A-8","A-9","A-10","A-11"]
        surplusClassName += ["B-1","B-2","B-3","B-4","B-5","B-6","B-7","B-8","B-9","B-10","B-11"]

    }

// MARK: 代理
    //每个区的item个数
    func collectionView(_ collectionView: UICollectionView, numberOfItemsInSection section: Int) -> Int {

        if section ==  {
            return nowClassName.count
        }else {

            if !isRank {
                return surplusClassName.count
            }else{
                 return
            }
        }

    }

    //分区个数
    func numberOfSections(in collectionView: UICollectionView) -> Int {

        if !isRank {
            return
        }

        return
    }

    //自定义cell
    func collectionView(_ collectionView: UICollectionView, cellForItemAt indexPath: IndexPath) -> UICollectionViewCell {

        let cell = collectionView.dequeueReusableCell(withReuseIdentifier: "HotCell", for: indexPath) as! HotCell
        cell.backgroundColor = UIColor.red
        if indexPath.section ==  {
            cell.label.text = nowClassName[indexPath.item]
            cell.button.addTarget(self, action: #selector(removeItem(_ :)), for: .touchUpInside)
            cell.button.tag = indexPath.row

            cell.button.isHidden = !isRank

        }else{
            cell.label.text = surplusClassName[indexPath.item]
            cell.button.isHidden = true

        }
        return cell

    }

    func removeItem(_ button:UIButton){

        //执行在这里的时候,显示的是有个分区,否则崩溃,报不明错误!,研究过的可以告诉一下,不胜感激!!!
        self.collectionView?.performBatchUpdates({
            //数据变更
            let item = self.nowClassName[button.tag]
            self.nowClassName.remove(at: button.tag)
            self.surplusClassName.append(item)

            let indexPath = IndexPath.init(item: button.tag, section: )
            print(indexPath)

            let arr:[IndexPath] = [indexPath]
            self.collectionView?.deleteItems(at: arr)

        }, completion: { (completion) in

            self.collectionView?.reloadData()
        })

    }

    //是否可以移动
    func collectionView(_ collectionView: UICollectionView, canMoveItemAt indexPath: IndexPath) -> Bool {

        if isRank {
            return true
        }
        return false

    }

    func collectionView(_ collectionView: UICollectionView, didSelectItemAt indexPath: IndexPath) {
        //获取当前 item
        //collectionView.cellForItem(at: indexPath)
        //获取所有的item
        //collectionView.indexPathsForVisibleItems

        if !isRank && indexPath.section == {
            //先把数据更新好,因为移动后界面会自动刷新,否则会崩溃
            nowClassName.append(surplusClassName[indexPath.item])
            surplusClassName.remove(at: indexPath.item)

            let indexPath1 = NSIndexPath.init(item: nowClassName.count-, section: )
            let indexPath2 = NSIndexPath.init(item: indexPath.item, section: )

            //从当前位置移动到新的位置
            collectionView.moveItem(at: indexPath2 as IndexPath, to: indexPath1 as IndexPath)
        }

    }

    //设置拖动(手势拖动触发)
    func collectionView(_ collectionView: UICollectionView, moveItemAt sourceIndexPath: IndexPath, to destinationIndexPath: IndexPath) {

//        if sourceIndexPath.section == 0 && destinationIndexPath.section == 0 {
//            collectionView.exchangeSubview(at: sourceIndexPath.item, withSubviewAt: destinationIndexPath.item)
//        }

    }

    //区头设置
    func collectionView(_ collectionView: UICollectionView, viewForSupplementaryElementOfKind kind: String, at indexPath: IndexPath) -> UICollectionReusableView {

//
        //区头
        var headerView : UICollectionReusableView?

        if kind == UICollectionElementKindSectionHeader{
            headerView = collectionView.dequeueReusableSupplementaryView(ofKind: kind, withReuseIdentifier: "headView", for: indexPath)

            //防止重用,这样很暴力
            for view in (headerView?.subviews)!{
                view.removeFromSuperview()

            }

            let label = UILabel.init(frame: CGRect(x:,y:,width:,height:))
            if indexPath.section ==  {
                label.text = "切换栏目"

                let button = UIButton.init(type: .custom)
                button.frame = CGRect(x:collectionView.frame.size.width - ,y:,width:,height:)
                button.titleLabel?.textColor = UIColor.cyan
                button.backgroundColor = UIColor.white

                let str = isRank ? "完成排序" : "排序删除"
                button.setTitle(str, for: .normal)
                button.setTitleColor(UIColor.red, for: .normal)
                headerView?.addSubview(button)

                button.addTarget(self, action: #selector(click(_ :)), for: .touchUpInside)

            }else if indexPath.section == {
                label.text = "点击添加更多栏目"
            }
            headerView?.addSubview(label)

        }

        return headerView!

    }

    func click(_ btn:UIButton){

        let str = isRank ? "完成排序" : "排序删除"
        btn.setTitle(str, for: .normal)

        isRank = !isRank
        print(isRank)
        self.collectionView?.reloadData()
    }

    //设置HeaderView的宽高
    //MARK: UICollectionViewDelegateFlowLayout
    func collectionView(_ collectionView: UICollectionView, layout collectionViewLayout: UICollectionViewLayout, referenceSizeForHeaderInSection section: Int) -> CGSize {

        return CGSize(width:collectionView.frame.size.width,height:)

    }

HotCell 类的实现

class HotCell: UICollectionViewCell {

    var label = UILabel()
    var button = UIButton()

    override init(frame: CGRect) {
        super.init(frame: frame)

        label = UILabel.init(frame: self.bounds)
        label.textAlignment = .center
        self.addSubview(label)
        button = UIButton.init(type: .custom)
        button.backgroundColor = UIColor.white
        button.frame = CGRect(x:frame.size.width - ,y:,width:,height:)
        self.addSubview(button)

    }

    required init?(coder aDecoder: NSCoder) {
        fatalError("init(coder:) has not been implemented")
    }

}