func octreeMesh(octree *gfx.Octant) *gfx.Mesh { m := new(gfx.Mesh) // Slice to zero. m.Vertices = m.Vertices[:0] m.Colors = m.Colors[:0] bci := -1 nextBC := func() gfx.Color { bci++ switch bci % 3 { case 0: return gfx.Color{1, 0, 0, 0} case 1: return gfx.Color{0, 1, 0, 0} case 2: return gfx.Color{0, 0, 1, 0} } panic("never here.") } var add func(o *gfx.Octant) add = func(o *gfx.Octant) { // Add vertices. before := len(m.Vertices) appendCube(m, float32(o.AABB.Size().X/2.0)) center := o.AABB.Center() for i, v := range m.Vertices[before:] { vert := v.Vec3() vert = vert.Add(center) m.Vertices[before+i] = gfx.ConvertVec3(vert) } fmt.Println("Octant", o.Depth, "-", len(o.Objects), "objects") for s := range o.Objects { before := len(m.Vertices) appendCube(m, float32(s.AABB().Size().X/2.0)) center := s.AABB().Center() //fmt.Println(center, s.AABB()) for i, v := range m.Vertices[before:] { vert := v.Vec3() vert = vert.Add(center) m.Vertices[before+i] = gfx.ConvertVec3(vert) } } for _, octant := range o.Octants { if octant != nil { add(octant) } } } add(octree) for _ = range m.Vertices { // Add barycentric coordinates. m.Colors = append(m.Colors, nextBC()) } return m }
func RTreeMesh(tree *rtree.Tree, level int) *gfx.Mesh { m := new(gfx.Mesh) bci := -1 nextBC := func() gfx.Color { bci++ switch bci % 3 { case 0: return gfx.Color{1, 0, 0, 0} case 1: return gfx.Color{0, 1, 0, 0} case 2: return gfx.Color{0, 0, 1, 0} } panic("never here.") } nodes := 0 var add func(n *rtree.Node) add = func(n *rtree.Node) { //if nodes > 100 { // return //} nodes++ if n == nil { return } if level == 0 || nodes == level { // Add vertices. m.Vertices = Rect3LineVerts(n.Bounds(), m.Vertices) } /*for _, s := range n.Objects { c := s.Bounds().Center() sz := 0.005 sb := math.Rect3{ Min: math.Vec3Zero.AddScalar(-sz).Add(c), Max: math.Vec3Zero.AddScalar(sz).Add(c), } m.Vertices = Rect3LineVerts(sb, m.Vertices) }*/ for _, c := range n.Children { add(c) } } add(tree.Root()) fmt.Println(tree.Count(), "objects", nodes, "nodes") for _ = range m.Vertices { // Add barycentric coordinates. m.Colors = append(m.Colors, nextBC()) } return m }