Function get_strands

utils/strand_util.py:89–117

(points)

def get_strands(points):
    strands = []
    count = 0
    total = points.shape[0]
    oris = []
    while True:
        strand = []
        base_dist = points[count+1] - points[count]
        base_dist = np.linalg.norm(base_dist,2)
        if base_dist==0:
            print('base:0')
            base_dist=0.0025
        for i in range(256):
            strand.append(points[count:count+1])
            if count== total-1:
                count+=1
                break
            dist = points[count+1] - points[count]
            count += 1
            if np.linalg.norm(dist,2)> base_dist*6:
                break
        strand = np.concatenate(strand,0)
        ori = np.concatenate([strand[1:]-strand[:-1],strand[-1:]-strand[-2:-1]],0)
        strands.append(strand)
        oris.append(ori)
        if count==total:
            break
    oris = np.concatenate(oris,0)
    return strands,oris