how to traverse a DataTree class

Replies to This Discussion

Permalink Reply by Peter Zock on January 14, 2011 at 5:13am

I found an example of how to traverse the tree, although in my example, I am not sure if it actually goes through the whole tree, a few points seems to be missing ? ...

 

Where I get an error, is where I try to replace the point (tree - branch - item) with the modified one. How can this be done?

 


private void RunScript(DataTree
endPoints, List
magnets, ref object A)
{

// for each magnet ..
foreach (Point3d magnet in magnets)
{
//Loop through all branches/geometry
for(int i = 0; i < endPoints.BranchCount; i++)
{

//The path for this index
GH_Path path = endPoints.Path(i);

//The count of elements in this index
int elementCount = endPoints.Branch(i).Count;

if(elementCount > 0)
{
//The first point in this path
Point3d firstPoint = endPoints[path, 0];
Line connect = new Line(magnet, firstPoint);

// if the geometrypoint is close to the magnet, transform
if(connect.Length < 3.5)
{
connect.Transform(Transform.Scale(magnet, 0.5));
}
returtest.Add(connect);

// change the point
endPoints[path, 0] = new Point3d(connect.ToX, connect.ToY, connect.ToZ);
}

}
}
A = endPoints;

}


Permalink Reply by Peter Zock on January 14, 2011 at 5:32am

Just a quick correction to the previous code, it was only set up to pick the first item on each branch...

for(int k = 0; k < elementCount; k++)
{
// The k point in this path
Point3d kPoint = endPoints[path, k];
Line connect = new Line(magnet, kPoint);

// if the geometrypoint is close to the magnet, transform
if(connect.Length < control)
{
connect.Transform(Transform.Scale(magnet, 1));
returTest.Add(connect);
}

// change the point
//endPoints[path, 0] = new Point3d(connect.ToX, connect.ToY, connect.ToZ);
} 

The question then remains; how can I replace an item in the Tree? 

Permalink Reply by to] on January 14, 2011 at 5:54am

Hi peter

endpoints.Branch(path).RemoveAt(0);
endpoints.Branch(path).Insert(0,new Point3d(connect.ToX, connect.ToY, connect.ToZ);

Permalink Reply by David Rutten on January 14, 2011 at 9:30am

I did it in 8 lines of code (if you don't count curly brackets as separate lines that is).

See attached.

 

--

David Rutten

david@mcneel.com

Poprad, Slovakia

Attachments:

• tree-op.3dm, 38 KB
• tree-op.ghx, 129 KB

Permalink Reply by Peter Zock on January 19, 2011 at 2:53am

Hi to]`s and David,

 

thanks a lot for your reply, it helped a lot :)

 

 

Permalink Reply by Amit Nambiar on April 11, 2018 at 12:04am

This might be useful for someone.

To traverse the path in nested loops I do it like so:

Get the max number of branches at each depth by

int depthLayerOne = tree.Paths[tree.PathCount - 1].Indices[0] + 1;

int depthLayerTwo = tree.Paths[tree.PathCount - 1].Indices[1] + 1;

It gets the last element from the list of paths, which would have the max number of branches at each depth level as indices. I add one to those, to make it easy to loop.

I can do the 0 and 1 index into the 'Indices ' array because in my case I know the number of layers in my tree.

If you don't know that in advance you can use a list to loop through.

Nested looping like so:

for(int i=0; i<depthLayerOne; i++) {

    for(int j=0; j<depthLayerTwo; j++)

        {

            // code

        }

}