Skip to content

Commit

Permalink
add data manipulation method
Browse files Browse the repository at this point in the history
  • Loading branch information
limuyang committed Jan 2, 2020
1 parent 414d6ae commit aba5135
Show file tree
Hide file tree
Showing 2 changed files with 122 additions and 40 deletions.
157 changes: 118 additions & 39 deletions library/src/main/java/com/chad/library/adapter/base/BaseNodeAdapter.kt
View file
Original file line number Diff line number Diff line change
Expand Up @@ -9,6 +9,7 @@ import com.chad.library.adapter.base.entity.node.NodeFooterImp
import com.chad.library.adapter.base.provider.BaseItemProvider
import com.chad.library.adapter.base.provider.BaseNodeProvider
import com.chad.library.adapter.base.viewholder.BaseViewHolder
import kotlin.math.max

abstract class BaseNodeAdapter(data: MutableList<BaseNode>? = null)
: BaseProviderMultiAdapter<BaseNode>(data) {
Expand Down Expand Up @@ -102,40 +103,18 @@ abstract class BaseNodeAdapter(data: MutableList<BaseNode>? = null)
}

override fun remove(position: Int) {
if (position >= data.size) {
return
}

//被移除的item数量
var removeCount = 0

val node = this.data[position]
//移除子项
if (!node.childNode.isNullOrEmpty()) {
val items = flatData(node.childNode!!)
this.data.removeAll(items)
removeCount = items.size
}
//移除node自己
this.data.removeAt(position)
removeCount += 1

// 移除脚部
if (node is NodeFooterImp && node.footerNode != null) {
this.data.removeAt(position)
removeCount += 1
}

val removeCount = removeAt(position)
notifyItemRangeRemoved(position + getHeaderLayoutCount(), removeCount)
compatibilityDataSizeChanged(0)
}

override fun setData(index: Int, data: BaseNode) {
val flatData = flatData(arrayListOf(data))
flatData.forEachIndexed { i, baseNode ->
this.data[index + i] = baseNode
}
notifyItemRangeChanged(index + getHeaderLayoutCount(), flatData.size)
val removeCount = removeAt(index)

val newFlatData = flatData(arrayListOf(data))
this.data.addAll(index, newFlatData)

notifyItemRangeChanged(index + getHeaderLayoutCount(), max(removeCount, newFlatData.size))
}

override fun replaceData(newData: Collection<BaseNode>) {
Expand All @@ -161,6 +140,43 @@ abstract class BaseNodeAdapter(data: MutableList<BaseNode>? = null)
super.setDiffNewData(diffResult, flatData(newData))
}

private fun removeAt(position: Int) :Int{
if (position >= data.size) {
return 0
}
//被移除的item数量
var removeCount = 0

val node = this.data[position]
//移除子项
if (!node.childNode.isNullOrEmpty()) {
if (node is BaseExpandNode) {
if (node.isExpanded) {
val items = flatData(node.childNode!!)
this.data.removeAll(items)
removeCount = items.size
}
} else {
val items = flatData(node.childNode!!)
this.data.removeAll(items)
removeCount = items.size
}
}
//移除node自己
this.data.removeAt(position)
removeCount += 1

// 移除脚部
if (node is NodeFooterImp && node.footerNode != null) {
this.data.removeAt(position)
removeCount += 1
}
return removeCount
}

/*************************** 重写数据设置方法 END ***************************/


/*************************** Node 数据操作 ***************************/

/**
Expand All @@ -169,35 +185,98 @@ abstract class BaseNodeAdapter(data: MutableList<BaseNode>? = null)
* @param data BaseNode 子node
*/
fun nodeAddData(parentNode: BaseNode, data: BaseNode) {
val parentIndex = this.data.indexOf(parentNode)
var childIndex = 0
parentNode.childNode?.let {
it.add(data)
childIndex = it.size
val childIndex = it.size

if (parentNode is BaseExpandNode && !parentNode.isExpanded) {
return
}

val parentIndex = this.data.indexOf(parentNode)
addData(parentIndex + childIndex, data)
}
}

/**
* 对指定的父node，在指定位置添加添加子node
* @param parentNode BaseNode 父node
* @param position Int 此位置是相对于其childNodes数据的位置！并不是整个data
* @param childIndex Int 此位置是相对于其childNodes数据的位置！并不是整个data
* @param data BaseNode 添加的数据
*/
fun nodeAddData(parentNode: BaseNode, position: Int, data: BaseNode) {
val parentIndex = this.data.indexOf(parentNode)
var pos = 0
fun nodeAddData(parentNode: BaseNode, childIndex: Int, data: BaseNode) {
parentNode.childNode?.let {
it.add(position, data)
pos = parentIndex + position + 1
it.add(childIndex, data)

if (parentNode is BaseExpandNode && !parentNode.isExpanded) {
return
}

val parentIndex = this.data.indexOf(parentNode)
val pos = parentIndex + 1 + childIndex
addData(parentIndex + pos, data)
}
}

/**
* 对指定的父node下对子node进行移除
* @param parentNode BaseNode 夫node
* @param childIndex Int 此位置是相对于其childNodes数据的位置！并不是整个data
*/
fun nodeRemoveData(parentNode: BaseNode, childIndex: Int) {
parentNode.childNode?.let {
it.removeAt(childIndex)

if (parentNode is BaseExpandNode && !parentNode.isExpanded) {
return
}

val parentIndex = this.data.indexOf(parentNode)
val pos = parentIndex + 1 + childIndex
remove(parentIndex + pos)
}
}

/*************************** 重写数据设置方法 END ***************************/
/**
* 对指定的父node下对子node进行移除
* @param parentNode BaseNode 夫node
* @param childNode BaseNode 子node
*/
fun nodeRemoveData(parentNode: BaseNode, childNode: BaseNode) {
parentNode.childNode?.let {
val isOk = it.remove(childNode)

if (!isOk) {
return
}

if (parentNode is BaseExpandNode && !parentNode.isExpanded) {
return
}
remove(childNode)
}
}

/**
* 改变指定的父node下的对子node数据
* @param parentNode BaseNode
* @param childIndex Int 此位置是相对于其childNodes数据的位置！并不是整个data
* @param data BaseNode 新数据
*/
fun nodeSetData(parentNode: BaseNode, childIndex: Int, data: BaseNode) {
parentNode.childNode?.let {
if (childIndex >= it.size) {
return
}
it[childIndex] = data

val parentIndex = this.data.indexOf(parentNode)
val pos = parentIndex + 1 + childIndex
setData(pos, data)
}
}

/*************************** Node 数据操作 END ***************************/

/**
* 将输入的嵌套类型数组循环递归，在扁平化数据的同时，设置展开状态
Expand Down
5 changes: 4 additions & 1 deletion library/src/main/java/com/chad/library/adapter/base/BaseQuickAdapter.kt
View file
Original file line number Diff line number Diff line change
Expand Up @@ -1101,9 +1101,12 @@ abstract class BaseQuickAdapter<T, VH : BaseViewHolder>

/**
* change data
* 改变数据
* 改变某一位置数据
*/
open fun setData(@IntRange(from = 0) index: Int, data: T) {
if (index >= this.data.size) {
return
}
this.data[index] = data
notifyItemChanged(index + getHeaderLayoutCount())
}
Expand Down

0 comments on commit aba5135

Please sign in to comment.