examples/rips/rips.py
 author Dmitriy Morozov Mon, 05 Jul 2010 23:51:26 -0700 branch dev changeset 211 347b3461965a parent 181 1ee6edc17cb6 permissions -rw-r--r--
Bug in the tutorial: simplices = [] -> simplices = Filtration()
```
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()
```