e2ecdcfe33
Former-commit-id:a02aeb236c[formerly9f19e3f712] [formerlya02aeb236c[formerly9f19e3f712] [formerly06a8b51d6d[formerly 64fa9254b946eae7e61bbc3f513b7c3696c4f54f]]] Former-commit-id:06a8b51d6dFormer-commit-id:8e80217e59[formerly3360eb6c5f] Former-commit-id:377dcd10b9
80 lines
2.5 KiB
Python
80 lines
2.5 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 Add import execute as Add
|
|
from Multiply import execute as Multiply
|
|
from numpy import zeros, ndarray
|
|
|
|
def execute(*args):
|
|
"""Perform a linear transform
|
|
|
|
@return: result = arg1*arg2 + arg3*arg4 + arg5...
|
|
vecresult = vec1*sca1 + vec2*sca2 + vec3...
|
|
|
|
"""
|
|
|
|
result = None
|
|
|
|
|
|
|
|
if type(args[0]) == tuple:
|
|
targetShape = args[0][0].shape
|
|
targetType = args[0][0].dtype
|
|
zeroArray = zeros(shape=(targetShape), dtype=targetType)
|
|
result = (zeroArray, zeroArray.copy(), zeroArray.copy(), zeroArray.copy())
|
|
else:
|
|
for arg in args:
|
|
if (type(arg) == ndarray):
|
|
targetShape = arg.shape
|
|
targetType = arg.dtype
|
|
result = zeros(shape=targetShape, dtype=targetType)
|
|
|
|
termLength = 2
|
|
|
|
terms = len(args) / termLength
|
|
if len(args) % termLength != 0:
|
|
terms = terms + 1
|
|
|
|
for i in range(terms):
|
|
coefficient = args[i * termLength]
|
|
variable = 1 if i * termLength + 1 >= len(args) else args[i * termLength + 1]
|
|
result = Add(result, Multiply(coefficient, variable))
|
|
|
|
return result
|
|
|
|
def test():
|
|
|
|
from numpy import array
|
|
|
|
if not( all(execute(array([1., 2.]), array([3., 4.]), array([1., 2.]), array([3., 4.])) == array([6., 16.]))):
|
|
raise Exception
|
|
|
|
from Vector import execute as Vector
|
|
|
|
# using meteorological direction
|
|
Vector1 = Vector(array([1., 2.]), array([0., 270.]), True)
|
|
Vector2 = Vector(array([3., 4.]), array([180., 90.]), True)
|
|
(mag, dir, u, v) = execute(Vector1, array([3., 4.]), Vector2, array([3., 4.]))
|
|
if not( all(mag == array([6., 8.])) and all(dir.round(0) == array([180., 90.]))):
|
|
print mag
|
|
print dir
|
|
raise Exception
|
|
|
|
print "LinTrans Test Complete"
|