How customize toTree() method?

[adminoid]

I have an ids array of collapsed item ids and want to check if item.id is equal to one of the ids, then return the empty children array, otherwise return normal children array. What is the right way to do that? Thanks in advance!