from    math        import fabs
from    dionysus    import Rips, Filtration, StaticPersistence #, enable_log

# Simple minded pairwise distance functor distance
class Distances:
    def __len__(self):
        return 5

    def __call__(self, x, y):
        return fabs(y-x)

dist = Distances()
r = Rips(dist)
lst = Filtration()
lst2 = Filtration()

#enable_log('rips')

r.generate(1, 3, lst.append)
r.generate(1, 3, lst2.append, [0,2,4])

print "Rips complex on all vertices:", lst
print "Rips complex on vertices [0,2,4]):", lst2

print "Values:", [map(r.eval, lst)]
print "Sorted:", sorted(lst, r.cmp)

cofaces = []
r.vertex_cofaces(2, 1, 3, cofaces.append)
print "Cofaces of vertex 2:", cofaces

cofaces = []
r.vertex_cofaces(2, 1, 3, cofaces.append, [0,2,4])
print "Cofaces of vertex 2 on vertices [0,2,4]:", cofaces

f = lst
f.sort(r.cmp)
p = StaticPersistence(f)
p.pair_simplices()
smap = p.make_simplex_map(f)
for s in p:
    print smap[s], s.sign()