Function load_strand

utils/strand_util.py:10–67

(file, return_strands=False,interp=False,num_interp=100)

from utils.general_util import get_window

def load_strand(file, return_strands=False,interp=False,num_interp=100):
    with open(file, mode='rb')as f:
        num_strand = f.read(4)

        (num_strand,) = struct.unpack('I', num_strand)
        point_count = f.read(4)

        (point_count,) = struct.unpack('I', point_count)

        # print("num_strand:",num_strand)
        segments = f.read(2 * num_strand)
        segments = struct.unpack('H' * num_strand, segments)
        segments = list(segments)

        num_points = sum(segments)

        points = f.read(4 * num_points * 3)
        points = struct.unpack('f' * num_points * 3, points)


    f.close()
    points = list(points)

    points = np.array(points)
    points = np.reshape(points, (-1, 3))
    new_p = []
    if return_strands:
        beg = 0
        strands = []
        oris = []
        for seg in segments:
            end = beg + seg
            strand = points[beg:end]
            if interp:
                try:
                    if strand.shape[0]>3:
                        new_points = B_spline_interpolate(strand,num_interp)
                        strand = np.stack(new_points,1)
                        new_p.append(strand)
                    else:
                        beg+=seg
                        continue
                    # new_p.append(strand)
                except:
                    beg+=seg
                    continue


            strands.append(strand)
            dir = np.concatenate([strand[1:] - strand[:-1], strand[-1:] - strand[-2:-1]], 0)
            dir = dir/np.linalg.norm(dir,2,-1,keepdims=True)
            oris.append(dir)
            beg += seg
        if interp:
            points = np.concatenate(new_p,0)
        return segments, points, strands, oris
    else:
        return segments, points