31cc3a237b
Former-commit-id:2807ec9b07[formerly2230164e1f] [formerly641d4acd15] [formerly2807ec9b07[formerly2230164e1f] [formerly641d4acd15] [formerly9fcb730858[formerly641d4acd15[formerly 37c406a9945bfb6ccb99d2472b44d649f2ce18d0]]]] Former-commit-id:9fcb730858Former-commit-id:3613b62fe1[formerlyb9b5068329] [formerly 614b30500251a3eee6fb9711755f92aabf706b56 [formerly1d2d1e20f5]] Former-commit-id: 48d6ac2942a8a7ef98a1259ca7320bedf1f46da9 [formerly4f01f96cf4] Former-commit-id:9d757d79e2
122 lines
4.9 KiB
Python
122 lines
4.9 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.
|
|
# #
|
|
|
|
#
|
|
# Routes requests to the Data Access Framework through Python Thrift.
|
|
#
|
|
#
|
|
#
|
|
# SOFTWARE HISTORY
|
|
#
|
|
# Date Ticket# Engineer Description
|
|
# ------------ ---------- ----------- --------------------------
|
|
# 05/21/13 #2023 dgilling Initial Creation.
|
|
# 01/06/14 #2537 bsteffen Share geometry WKT.
|
|
# 03/03/14 #2673 bsteffen Add ability to query only ref times.
|
|
#
|
|
|
|
|
|
import numpy
|
|
import shapely.wkt
|
|
|
|
from dynamicserialize.dstypes.com.raytheon.uf.common.dataaccess.impl import DefaultDataRequest
|
|
from dynamicserialize.dstypes.com.raytheon.uf.common.dataaccess.request import GetAvailableLocationNamesRequest
|
|
from dynamicserialize.dstypes.com.raytheon.uf.common.dataaccess.request import GetAvailableTimesRequest
|
|
from dynamicserialize.dstypes.com.raytheon.uf.common.dataaccess.request import GetGeometryDataRequest
|
|
from dynamicserialize.dstypes.com.raytheon.uf.common.dataaccess.request import GetGridDataRequest
|
|
|
|
from ufpy import ThriftClient
|
|
from ufpy.dataaccess import PyGeometryData
|
|
from ufpy.dataaccess import PyGridData
|
|
|
|
|
|
class ThriftClientRouter(object):
|
|
|
|
def __init__(self, host='localhost'):
|
|
self._client = ThriftClient.ThriftClient(host)
|
|
|
|
def getAvailableTimes(self, request, refTimeOnly):
|
|
timesRequest = GetAvailableTimesRequest()
|
|
timesRequest.setRequestParameters(request)
|
|
timesRequest.setRefTimeOnly(refTimeOnly)
|
|
response = self._client.sendRequest(timesRequest)
|
|
return response
|
|
|
|
def getGridData(self, request, times):
|
|
gridDataRequest = GetGridDataRequest()
|
|
gridDataRequest.setRequestParameters(request)
|
|
# if we have an iterable times instance, then the user must have asked
|
|
# for grid data with the List of DataTime objects
|
|
# else, we assume it was a single TimeRange that was meant for the
|
|
# request
|
|
try:
|
|
iter(times)
|
|
gridDataRequest.setRequestedTimes(times)
|
|
except TypeError:
|
|
gridDataRequest.setRequestedPeriod(times)
|
|
response = self._client.sendRequest(gridDataRequest)
|
|
|
|
locSpecificData = {}
|
|
locNames = response.getSiteNxValues().keys()
|
|
for location in locNames:
|
|
nx = response.getSiteNxValues()[location]
|
|
ny = response.getSiteNyValues()[location]
|
|
latData = numpy.reshape(numpy.array(response.getSiteLatGrids()[location]), (nx, ny))
|
|
lonData = numpy.reshape(numpy.array(response.getSiteLonGrids()[location]), (nx, ny))
|
|
locSpecificData[location] = (nx, ny, (lonData, latData))
|
|
|
|
retVal = []
|
|
for gridDataRecord in response.getGridData():
|
|
locationName = gridDataRecord.getLocationName()
|
|
locData = locSpecificData[locationName]
|
|
retVal.append(PyGridData.PyGridData(gridDataRecord, locData[0], locData[1], locData[2]))
|
|
return retVal
|
|
|
|
def getGeometryData(self, request, times):
|
|
geoDataRequest = GetGeometryDataRequest()
|
|
geoDataRequest.setRequestParameters(request)
|
|
# if we have an iterable times instance, then the user must have asked
|
|
# for geometry data with the List of DataTime objects
|
|
# else, we assume it was a single TimeRange that was meant for the
|
|
# request
|
|
try:
|
|
iter(times)
|
|
geoDataRequest.setRequestedTimes(times)
|
|
except TypeError:
|
|
geoDataRequest.setRequestedPeriod(times)
|
|
response = self._client.sendRequest(geoDataRequest)
|
|
geometries = []
|
|
for wkt in response.getGeometryWKTs():
|
|
geometries.append(shapely.wkt.loads(wkt))
|
|
|
|
retVal = []
|
|
for geoDataRecord in response.getGeoData():
|
|
geom = geometries[geoDataRecord.getGeometryWKTindex()]
|
|
retVal.append(PyGeometryData.PyGeometryData(geoDataRecord, geom))
|
|
return retVal
|
|
|
|
def getAvailableLocationNames(self, request):
|
|
locNamesRequest = GetAvailableLocationNamesRequest()
|
|
locNamesRequest.setRequestParameters(request)
|
|
response = self._client.sendRequest(locNamesRequest)
|
|
return response
|
|
|
|
def newDataRequest(self):
|
|
return DefaultDataRequest()
|