awips2/cave/com.raytheon.uf.viz.derivparam.python/localization/derivedParameters/functions/Add.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

47 lines
1.4 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 add,array,zeros_like
import Vector
def execute(*args):
""" Perform scalar or vector addition
"""
if len(args) == 1 and isinstance(args[0], list):
return execute(*args[0])
elif isinstance(args[0], tuple):
return vectorAddition(args)
else:
return scalarAddition(args)
def scalarAddition(args):
return reduce(add, args)
def vectorAddition(args):
uResult = zeros_like(args[0][0])
vResult = zeros_like(args[0][0])
for u, v in args:
uResult += u
vResult += v
return Vector.componentsTo(uResult, vResult)