168 lines
4.7 KiB
Python
168 lines
4.7 KiB
Python
# 生成线元要素表
|
||
|
||
import copy
|
||
import numpy
|
||
|
||
#直曲表
|
||
sheet_align = sheet("alignment_table")
|
||
|
||
#线元数据表
|
||
sheet_linear = sheet("linear_feature")
|
||
|
||
#生成线元数据
|
||
sheet_linear.clear_content()
|
||
|
||
row1 = 3
|
||
row2 = 0
|
||
epsilon = 0.0005
|
||
|
||
sheet_linear[row2, 0].value = sheet_align[row1, 3].value
|
||
sheet_linear[row2, 6].value = sheet_align[row1 + 1, 20].value
|
||
sheet_linear[row2, 7].value = sheet_align[row1, 1].value
|
||
sheet_linear[row2, 8].value = sheet_align[row1, 2].value
|
||
|
||
row1 += 1
|
||
row2 += 1
|
||
|
||
while not sheet_align[row1, 6].empty():
|
||
sheet_linear[row2, 0].value = sheet_align[row1, 13].value
|
||
if abs(sheet_linear[row2, 0].value - sheet_linear[row2 - 1, 0].value) > epsilon:
|
||
sheet_linear[row2, 2].value = 0
|
||
sheet_linear[row2, 3].value = 0
|
||
row2 += 1
|
||
|
||
sheet_linear[row2, 0].value = sheet_align[row1, 14].value
|
||
if abs(sheet_linear[row2, 0].value - sheet_linear[row2 - 1, 0].value) > epsilon:
|
||
sheet_linear[row2, 2].value = 0
|
||
sheet_linear[row2, 3].value = sheet_align[row1, 6].value * sheet_align[row1, 21].value
|
||
row2 += 1
|
||
|
||
sheet_linear[row2, 0].value = sheet_align[row1, 16].value
|
||
if abs(sheet_linear[row2, 0].value - sheet_linear[row2 - 1, 0].value) > epsilon:
|
||
sheet_linear[row2, 2].value = sheet_align[row1, 6].value * sheet_align[row1, 21].value
|
||
sheet_linear[row2, 3].value = sheet_linear[row2, 2].value
|
||
row2 += 1
|
||
|
||
sheet_linear[row2, 0].value = sheet_align[row1, 17].value
|
||
if abs(sheet_linear[row2, 0].value - sheet_linear[row2 - 1, 0].value) > epsilon:
|
||
sheet_linear[row2, 2].value = sheet_align[row1, 6].value * sheet_align[row1, 21].value
|
||
sheet_linear[row2, 3].value = 0
|
||
row2 += 1
|
||
|
||
row1 += 1
|
||
|
||
sheet_linear[row2, 0].value = sheet_align[row1, 3].value
|
||
sheet_linear[row2, 2].value = 0
|
||
sheet_linear[row2, 3].value = 0
|
||
|
||
#线元要素计算
|
||
|
||
def calc_linear_type(x, y):
|
||
#0-圆曲线，1-直线，2-缓和曲线
|
||
epsilon = 0.00001
|
||
if abs(x - y) >= epsilon:
|
||
return 2
|
||
if abs(x) < epsilon:
|
||
return 1
|
||
return 0
|
||
|
||
# 定义桩点
|
||
class stake_point:
|
||
def __init__(
|
||
self, chainage = 0, x = 0, y = 0, z = 0,
|
||
alpha = 0, beta = 0, rho = 0
|
||
):
|
||
self.chainage = chainage #里程
|
||
self.x = x #x坐标
|
||
self.y = y #y坐标
|
||
self.z = z #高程
|
||
self.alpha = alpha #切线方位角
|
||
self.beta = beta #
|
||
self.rho = rho #曲率
|
||
|
||
# 缓和曲线
|
||
# 起点里程、曲率、坐标、切线方位角以及终点里程、曲率确定一条缓和曲线
|
||
class curve:
|
||
def __init__(self, start = None, end = None):
|
||
if start is None:
|
||
self.start = stake_point()
|
||
else:
|
||
self.start = copy.copy(start) #起始桩点
|
||
if end is None:
|
||
self.end = stake_point()
|
||
else:
|
||
self.end = copy.copy(end) #终止桩点
|
||
|
||
# 求缓和曲线曲率变化参数
|
||
def get_ratio(self):
|
||
return math.sqrt((self.end.chainage - self.start.chainage) / abs(self.end.rho - self.start.rho))
|
||
|
||
# 求里程为chainage的一点处的曲率、坐标及切线方位角，采用Gauss-Legendre公式计算
|
||
def get_point(self, chainage):
|
||
|
||
l = chainage - self.start.chainage
|
||
ll = self.end.chainage - self.start.chainage
|
||
if abs(ll) < 0.001:
|
||
return copy.copy(self.start)
|
||
|
||
sp = stake_point(chainage = chainage)
|
||
# 求点的切线方位角及曲率
|
||
sp.rho = self.start.rho + (self.end.rho - self.start.rho) * l / ll # 曲率线性变化
|
||
sp.alpha = self.start.alpha + (self.start.rho + sp.rho) * l / 2
|
||
sp.alpha %= 2 * math.pi
|
||
|
||
f = np.zeros((2, 4), dtype = np.float64)
|
||
f[0, 0] = 0.0694318442
|
||
f[0, 1] = 0.3300094782
|
||
f[0, 2] = 1 - f[0, 1]
|
||
f[0, 3] = 1 - f[0, 0]
|
||
f[1, 0] = 0.1739274226
|
||
f[1, 1] = 0.3260725774
|
||
f[1, 2] = f[1, 1]
|
||
f[1, 3] = f[1, 0]
|
||
|
||
s1 = 0
|
||
s2 = 0
|
||
a0 = self.start.alpha
|
||
c0 = self.start.rho
|
||
c1 = self.end.rho
|
||
h = (c1 - c0) * l * l / (2 * ll)
|
||
for i in range(4):
|
||
s1 = s1 + f[1, i] * math.cos(a0 + c0 * l * f[0, i] + h * f[0, i] * f[0, i])
|
||
s2 = s2 + f[1, i] * math.sin(a0 + c0 * l * f[0, i] + h * f[0, i] * f[0, i])
|
||
sp.x = self.start.x + l * s1
|
||
sp.y = self.start.y + l * s2
|
||
|
||
return sp
|
||
|
||
row = 1
|
||
with sheet_linear:
|
||
while not cell(row, 0).empty():
|
||
cell(row, 1).value = cell(row, 0).value - cell(row - 1, 0).value
|
||
if cell(row, 2).value != 0:
|
||
cell(row, 4).value = 1 / cell(row, 2).value
|
||
if cell(row, 3).value != 0:
|
||
cell(row, 5).value = 1 / cell(row, 3).value
|
||
cell(row, 9).value = calc_linear_type(cell(row, 2).value, cell(row, 3).value)
|
||
|
||
cv = curve()
|
||
cv.start.chainage = cell(row - 1, 0).value
|
||
cv.start.rho = (cell(row, 4).value or 0)
|
||
cv.start.alpha = cell(row - 1, 6).value
|
||
cv.start.x = cell(row - 1, 7).value
|
||
cv.start.y = cell(row - 1, 8).value
|
||
cv.end.chainage = cell(row, 0).value
|
||
cv.end.rho = (cell(row, 5).value or 0)
|
||
|
||
if cell(row, 9).value == 2:
|
||
cell(row, 10).value = cv.get_ratio()
|
||
|
||
sp = cv.get_point(cv.end.chainage)
|
||
cell(row, 6).value = sp.alpha
|
||
cell(row, 7).value = sp.x
|
||
cell(row, 8).value = sp.y
|
||
|
||
row += 1
|
||
|
||
sheet_linear.fit_content()
|