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")
    }
 
}