e2ecdcfe33
Former-commit-id:a02aeb236c[formerly9f19e3f712] [formerlya02aeb236c[formerly9f19e3f712] [formerly06a8b51d6d[formerly 64fa9254b946eae7e61bbc3f513b7c3696c4f54f]]] Former-commit-id:06a8b51d6dFormer-commit-id:8e80217e59[formerly3360eb6c5f] Former-commit-id:377dcd10b9
75 lines
2.7 KiB
Python
75 lines
2.7 KiB
Python
##
|
|
# This software was developed and / or modified by Raytheon Company,
|
|
# pursuant to Contract DG133W-05-CQ-1067 with the US Government.
|
|
#
|
|
# U.S. EXPORT CONTROLLED TECHNICAL DATA
|
|
# This software product contains export-restricted data whose
|
|
# export/transfer/disclosure is restricted by U.S. law. Dissemination
|
|
# to non-U.S. persons whether in the United States or abroad requires
|
|
# an export license or other authorization.
|
|
#
|
|
# Contractor Name: Raytheon Company
|
|
# Contractor Address: 6825 Pine Street, Suite 340
|
|
# Mail Stop B8
|
|
# Omaha, NE 68106
|
|
# 402.291.0100
|
|
#
|
|
# See the AWIPS II Master Rights File ("Master Rights File.pdf") for
|
|
# further licensing information.
|
|
##
|
|
from numpy import zeros, float32, NaN, isnan
|
|
|
|
##
|
|
# Designed to replace interp_up and interp_down in design files for point data
|
|
#
|
|
def execute(paramArray, vertArray, numLevels, vertPoints, maxGap=None):
|
|
#print "paramArray =", paramArray
|
|
#print "vertArray =", vertArray
|
|
#print "numLevels =", numLevels
|
|
#print "vertPoints =", vertPoints
|
|
#print "maxGap =", maxGap
|
|
ret = zeros(numLevels.shape, 'float32')
|
|
ret.fill(NaN)
|
|
for i in range(len(vertArray)):
|
|
count = numLevels[i]
|
|
verts = vertArray[i][:count]
|
|
params = paramArray[i][:count]
|
|
if isinstance(vertPoints, float):
|
|
vertPoint = vertPoints
|
|
elif isinstance(vertPoints, float32):
|
|
vertPoint = vertPoints
|
|
else:
|
|
vertPoint = vertPoints[i]
|
|
below = None
|
|
above = None
|
|
for j in range(len(verts)):
|
|
if isnan(verts[j]) or isnan(params[j]):
|
|
continue
|
|
if (verts[j] > vertPoint and (above == None or verts[above] > verts[j])):
|
|
above = j
|
|
if (verts[j] < vertPoint and (below == None or verts[below] < verts[j])):
|
|
below = j
|
|
if(above == None and below == None):
|
|
ret[i] = NaN
|
|
continue
|
|
if (maxGap != None):
|
|
gap = None
|
|
if(above == None):
|
|
gap = (vertPoint - verts[below])*2
|
|
elif(below == None):
|
|
gap = (verts[above] - vertPoint)*2
|
|
else:
|
|
gap = verts[above] - verts[below]
|
|
if(gap == None or gap > maxGap):
|
|
ret[i] = NaN
|
|
continue
|
|
if(above == None):
|
|
ret[i] = params[below]
|
|
elif(below == None):
|
|
ret[i] = params[above]
|
|
else:
|
|
wgt = (vertPoint-verts[below])/(verts[above] - verts[below])
|
|
ret[i] = params[below] + wgt*(params[above] - params[below])
|
|
#print "ret = ", ret
|
|
#print "----------------------------"
|
|
return ret
|