Swift开发使用RATreeView搭建树形视图

项目中需要有树形图结构的视图，研究了一下RATreeView,搭建完成。用到的方法拿出来跟大家分享一下，希望可以帮到大家。

先看一下效果图

效果图

项目地址

github

swift使用RATreeView方法说明

1.cocopods集成项目
2.在桥文件引入RATreeView
3.具体方法使用

声明视图

let treeView = RATreeView()
treeView.frame = CGRect(x: 0, y: 40, width: SCREEN_WIDTH, height: SCREEN_HEIGHT - 64)
treeView.treeFooterView = UIView()
treeView.delegate = self
treeView.dataSource = self
treeView.register(AgentTreeTableViewCell.self,forCellReuseIdentifier: "cell")

主要方法

树形图的行数

 func treeView(_ treeView: RATreeView, numberOfChildrenOfItem item: Any?) -> Int {
        if let item = item as? 数据类型 {
            return item.children!.count //子数目
        } else {
            return self.data.count//根数目
        }
}

每行返回的对应的数据

func treeView(_ treeView: RATreeView, child index: Int, ofItem item: Any?) -> Any {
      if let item = item as? 数据类型  {
            return item.children![index]
        } else {
            return data[index] as AnyObject
        }
}

cell视图

func treeView(_ treeView: RATreeView, cellForItem item: Any?) -> UITableViewCell {
        //声明cell
        let cell = treeView.dequeueReusableCell(withIdentifier: "cell") as! AgentTreeTableViewCell 
        //获取数据
        let thisItem = item as! 数据类型 
        //获取当前cell的层级，可进行缩进或其他操作
        let level = treeView.levelForCell(forItem: thisItem)
        cell.selectionStyle = .none
        cell.setUpData(thisItem.agentName!, level: level,children:thisItem.children!)
        
        cell.leftIcon.agentInfo = thisItem
        cell.leftIcon.addTarget(self, action: #selector(toClickOkBtn(_:)), for: .touchUpInside)
        return cell
 }

cell点击事件

func treeView(_ treeView: RATreeView, didSelectRowForItem item: Any) {
}

项目中其余有用方法

关闭侧滑cell出现编辑（默认是开启的）

func treeView(_ treeView: RATreeView, canEditRowForItem item: Any) -> Bool {
     return false
}

展开与关闭数据

1.全部展开

self.treeView.reloadData()//一定要写着reloadData()之后
self.treeView.expandRow(forItem: self.data[0], expandChildren: true, with: RATreeViewRowAnimationNone)

2.关闭点击cell展开与关闭
treeview默认的是点击cell展开或关闭该组数据，如果不想使用cell默认的点击事件进行展开或关闭,想自己重写，或者把展开关闭事件放到指定的按钮上怎么办

1）第一种方法

treeView.allowsSelection = false

设置之后cell点击事件被禁止，如果自己在didSelectRowForItem方法里设置事件也会被禁止。

2）第二种方法

func treeView(_ treeView: RATreeView, shouldExpandRowForItem item: Any) -> Bool {
     return false
}

这种方法会禁止默认的展开与关闭事件，而且不会禁止didSelectRowForItem方法。

禁止之后怎么将展开与关闭事件放到按钮或其他控件呢

let isExpanded = treeView.isCell(forItemExpanded:点击当前行的item)//可以判断当前行是否展开
if isExpanded {
  treeView.collapseRow(forItem: item, collapseChildren: true, with: RATreeViewRowAnimationAutomatic)
} else {
  reeView.expandRow(forItem: item, expandChildren: false, with: RATreeViewRowAnimationAutomatic)
}

项目中用到的就这些，如果大家还有什么补充或疑惑，欢迎私信或评论，点个赞吧。谢谢大家