from fibertree import Fiber
ab = Fiber([0, 1, 10, 20], [(0, 1), (1, 2), (10, 11), (20, 21)])
(a, b) = ab.unzip()
ab.print("ab\n")
a.print("a\n")
b.print("b\n")
from fibertree import Fiber
a = Fiber([0, 1, 10, 20], [1, 2, 11, 21])
ap = a.project(lambda c: c + 1)
a.print("a\n")
ap.print("ap\n")
from fibertree import Fiber
f = Fiber([0, 1, 2, 10, 12, 31, 41], [0, 10, 20, 100, 120, 310, 410])
print("Original fiber\n")
f.print()
#
# Unform coordiate-based split
#
coords = 10
print("Uniform coordinate split (groups of %s coordinates)\n" % coords)
fibers = f.splitUniform(coords)
for c, s in fibers:
s.print()
#
# Non-unform coordiate-based split
#
splits = [0, 12, 31]
print(f"NonUniform coordinate split (splits at {splits})\n")
fibers = f.splitNonUniform(splits)
for c, s in fibers:
s.print()
#
# Equal position-based split
from fibertree import Fiber, Tensor, TensorImage
a = Fiber([0, 2, 8], [5, 6, 7])
a.print("Fiber")
i = TensorImage(a)
i.show()
