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()