awips2/cave/com.raytheon.uf.viz.derivparam.python/localization/derivedParameters/functions/LinTrans.py
Ben Steffensmeier 6d8adaccb2 Issue #2661 Use only u,v to represent vector grids in derived parameters and D2D.
Change-Id: I7ea007f0f919017e8212716cc15712c7ed611e69

Former-commit-id: 8b23f32311 [formerly ee11e5d13c] [formerly 8b23f32311 [formerly ee11e5d13c] [formerly ac33fb9463 [formerly a8bcec4a2fa9ddd49dc300ebd63aebfdfb41723e]]]
Former-commit-id: ac33fb9463
Former-commit-id: b2b2136868 [formerly be163ebf2a]
Former-commit-id: 182a717b00
2014-01-16 09:03:50 -06:00

57 lines
1.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 Add import execute as Add
from Multiply import execute as Multiply
from numpy import zeros_like, 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:
zeroArray = zeros_like(args[0][0])
result = (zeroArray, zeroArray.copy())
else:
for arg in args:
if (type(arg) == ndarray):
result = zeros_like(arg)
break
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