protocol.py 文件源码

def set_free(self, uid=None):
        ''' reset status of occupied points to zero '''
        self.read()
        data = self.data

        # check for valid id
        if uid in [0,1]:
            print "Error: %s is not a valid ID, returning." %uid
            return None

        if uid is not None:
            # get indices of vertices by id
            vertices = np.nonzero(data["status"] == uid)[0]
        else:
            vertices = np.nonzero(np.logical_and(data["status"] != 0, data["status"] != 1))[0]

        # reset vertices to 0
        data["status"][vertices] = 0
        self.write()