bindings/python/dionysus/__init__.py
author Christos Mantoulidis <cmad@stanford.edu>
Mon, 27 Jul 2009 21:42:51 -0700
branchdev
changeset 152 fdc2835b3e88
parent 132 2a737609b8bf
child 147 d39a20acb253
permissions -rw-r--r--
added logging to WeightedRips (still needs debugging)

from    _dionysus   import *
from    distances   import *
from    zigzag      import *


def init_with_data(self, v, d = None):
    self._cpp_init_(v)
    if d is not None:
        self.data = d

def repr_with_data(self):
    str = self._cpp_repr_()
    if hasattr(self, 'data'):
        str += ' %f' % self.data
    return str

Simplex._cpp_init_ =    Simplex.__init__
Simplex.__init__ =      init_with_data
Simplex._cpp_repr_ =    Simplex.__repr__
Simplex.__repr__  =     repr_with_data


def data_cmp(s1, s2):
    return cmp(s1.data, s2.data)

def data_dim_cmp(s1,s2):
    return cmp(s1.dimension(), s2.dimension()) or data_cmp(s1,s2)

def vertex_dim_cmp(s1, s2):
    return cmp(s1.dimension(), s2.dimension()) or vertex_cmp(s1, s2)