awips2/cave/com.raytheon.uf.viz.derivparam.python/localization/derivedParameters/functions/Poly.py
root e2ecdcfe33 Initial revision of AWIPS2 11.9.0-7p5
Former-commit-id: a02aeb236c [formerly 9f19e3f712] [formerly a02aeb236c [formerly 9f19e3f712] [formerly 06a8b51d6d [formerly 64fa9254b946eae7e61bbc3f513b7c3696c4f54f]]]
Former-commit-id: 06a8b51d6d
Former-commit-id: 8e80217e59 [formerly 3360eb6c5f]
Former-commit-id: 377dcd10b9
2012-01-06 08:55:05 -06:00

60 lines
No EOL
1.8 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 power
from numpy import multiply
def execute(*args):
"""Combine the arguments into a polynomial result
@return result of arg1*arg2^arg3 + arg4*arg5^arg6 + arg7*arg8
"""
result = 0
terms = len(args) / 3
if len(args) % 3 != 0:
terms = terms + 1
for i in range(terms):
coefficient = args[i * 3]
variable = 1 if i * 3 + 1 >= len(args) else args[i * 3 + 1]
exponent = 1 if i * 3 + 2 >= len(args) else args[i * 3 + 2]
result += multiply(coefficient, power(variable, exponent))
return result
def test():
from numpy import array
if not(all(execute(array([1., 2.])) == array([1., 2.]))):
raise Exception
if not(all(execute(array([1., 2.]), array([3., 4.])) == array([3., 8.]))):
raise Exception
if not(all(execute(array([1., 2.]), array([3., 4.]), array([5., 6.])) == array([243., 8192.]))):
raise Exception
print "Poly Test Complete"