Trying to pick GH referenced objects by their "original" layer

Hi everyone, I'm trying to pick some objects according to the layer of the rhino object from which they were referenced into GH, but I'm not succeeding.  

I've been looking for a while for a GH definition or scripting component that does this, but I haven't been able to find one. So what I have is:

A lot of groups referenced into grasshopper. The groups (and the geometry they contain) are being moved in the viewport, and then ungrouped. At that point, I would like to know to what layer those objects used to belong (when they were referenced from rhino), and then, group those objects according to that.

Is that possible? If not, I'll try to find a way to get the objects's IDS at the beginning of my GH definition (right after the pipeline component I'm using), and then somehow sort those IDS and group them in the same way the geometry is being grouped and sorted. Unfortunately, all the grouping and sorting is now being done by some VB components I found on the forum, and I don't know how to tweak them to make them work in order to do the same they are doing with the geometry, but with their IDs. 

Well, I've attached the files, I hope somebody can help me. 

Thanks!