NDH_PythonTools/line_fill.py at main · nholschuh/NDH_PythonTools · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
import numpy as np
import scipy.interpolate as interp

################## NDH Tools self imports
###########################################################
from .distance_vector import distance_vector
from .heading import heading
###########################################################

def line_fill(segmat,value,density0_or_distance1, start=0, stop=0, keep_vertices=0):
    """
    % (C) Nick Holschuh - Amherst College -- 2022 (Nick.Holschuh@gmail.com)
    %
    % Fills in data points with given 'density' along provided segments, withdata
    % provided in an nxm matrix, where n is the number of data, with thefirst 2
    % columns containing the x and y coordinates. The later columns can be other
    % data that you wish to linearly interpolate between points.
    %
    %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
    % The inputs are:
    %
    %     segmat -- This is an Nx2 matrix that contains x and y values
    %     value -- either the density or the distance between points
    %     density0_or_distance1 -- this defines either the number of points
    %               between the supplied values in the segmat, or the spacing
    %               between them.
    %
    %%%%%%%%%%%%%%%
    % The outputs are:
    %
    %      linematrix -- array with the additional values inserted
    %
    %%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%%
    """


    if isinstance(segmat,type(np.array)) == 0:
        segmat = np.array(segmat)

    s_dim = segmat.shape
    turnflag = 2

    if np.min(s_dim) == 1:
        remove_flag = 1;
        if s_dim(1) == 1:
            segmat = [segmat.T,np.zeros(segmat.T.shape)];
            turnflag = 1
        else:
            segmat = [segmat,zeros(segmat.shape)];
            turnflag = 2
    else:
        remove_flag = 0
        if s_dim[0] == 2:
            segmat = segmat.T
            turnflag = 1


    ################### the section for interpolating a fixed number of points for each group
    if density0_or_distance1 == 0:
        linematrix = np.zeros((value*(len(segmat[:,1])-1)+1,len(segmat[0,:]))) #Creates an empty vector with the value*# of original values

        for i in np.arange(len(segmat[:,1])-1):
            rangevec = segmat[i+1,:] - segmat[i,:]
            incrementvec = rangevec/value

            startingindex = value*i
            linematrix[startingindex,:] = segmat[i,:]
            for j in np.arange(value-1):
                linematrix[startingindex+j+1,:] = linematrix[startingindex+j,:]+incrementvec

        linematrix[len(linematrix[:,1])-1,:] = segmat[len(segmat[:,1])-1,:]


    ###################
    else:

        dist_vec2 = ndh.distance_vector(segmat[:,0],segmat[:,1],1)
        remove_rows = np.where(dist_vec2 == 0)[0]

        segmat = np.delete(segmat,remove_rows,0)
        dist_vec = ndh.distance_vector(segmat[:,0],segmat[:,1])

        if stop == 0:
            new_dist = np.arange(start,np.max(dist_vec),value)
        elif stop > np.max(dist_vec):
            new_dist = np.arange(start,np.max(dist_vec),value)
        else:
            new_dist = np.arange(start,stop,value)


        ############## This section maintains the vertices in the data
        if keep_vertices == 1:
            angle_thresh = 0.1
            headings = ndh.heading(segmat[:,0],segmat[:,1])
            headings_change = np.abs(np.diff(headings))
            keep_inds = np.where(headings_change > angle_thresh)[0]
            add_dists = dist_vec[keep_inds]
            new_dist = np.unique(sorted(np.concatenate([np.array([0]),new_dist,add_dists,np.array([np.max(dist_vec)])])))

        linematrix = np.zeros((len(new_dist),len(segmat[0,:])))


        for i in np.arange(len(segmat[0,:])):
            f =  interp.interp1d(dist_vec,segmat[:,i])
            linematrix[:,i] = f(new_dist)


    if remove_flag == 1:
        linematrix = linematrix[:,:-2]

    if turnflag == 1:
        segmat = segmat.T
        linematrix = linematrix.T
    elif turnflag == 2:
        linematrix = linematrix

    return linematrix