# #
# 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.
# #

#
# Implements IGridData and wraps around a Java IGridData.
#  
#    
#     SOFTWARE HISTORY
#    
#    Date            Ticket#       Engineer       Description
#    ------------    ----------    -----------    --------------------------
#    12/10/12                      njensen        Initial Creation.
#    05/01/14        3095          bsteffen       Move numeric data access to new plugin.
#    08/06/14        3185          njensen        Only import Java classes when necessary
#    Apr 23, 2015    4259          njensen        Updated for new JEP API
#    11/02/15        5079          dgilling       Fix typo in getRawData.
#    10/14/16        5916          bsteffen       Correct grid shape, return longitude
#
#

##
# This is a base file that is not intended to be overridden.
##



from ufpy.dataaccess import IGridData
import JData
import numpy as np

class JGridData(IGridData, JData.JData):
    
    def __init__(self, wrappedObject):
        JData.JData.__init__(self, wrappedObject)
        
    def __getitem__(self, key):
        if key == 'parameter':
            return self.getParameter()
        elif key == 'unit':
            return self.getUnit()
        elif key == 'rawData':
            return self.getRawData()
        elif key == 'time':
            return self.getDataTime()
        elif key == 'level':
            return self.getLevel()
    
    def getParameter(self):
        return self.jobj.getParameter()
    
    def getUnit(self):
        return str(self.jobj.getUnit())
    
    def getRawData(self, unit=None):
        # import only the modules that are needed
        from com.raytheon.uf.common.numeric.buffer import FloatBufferWrapper        
        
        nx = self.jobj.getGridGeometry().getGridRange().getSpan(0)
        ny = self.jobj.getGridGeometry().getGridRange().getSpan(1)
        dest = FloatBufferWrapper(nx, ny)
        if unit:
            from tec.uom.se.format import SimpleUnitFormat
            from com.raytheon.uf.common.geospatial.data import UnitConvertingDataFilter
            from com.raytheon.uf.common.numeric.dest import FilteredDataDestination
            from jep import jarray
            from java.text import ParsePosition
            
            unitObj = SimpleUnitFormat.getInstance(SimpleUnitFormat.Flavor.ASCII).parseObject(unit, ParsePosition(0))
            converter = self.jobj.getUnit().getConverterTo(unitObj)
            dataFilter = UnitConvertingDataFilter(converter)
            filters = jarray(1, UnitConvertingDataFilter)
            filters[0] = dataFilter
            unitDest = FilteredDataDestination.addFilters(dest, filters)
            self.jobj.populateData(unitDest)
        else:
            self.jobj.populateData(dest)        
        data = np.array(dest.getArray(), dtype=np.float32)
        data = data.reshape((ny, nx))
        return data
    
    def getLatLonCoords(self):
        """
        Gets the lat/lon coordinates of the grid data.
        
        Returns:
            a tuple where the first element is a numpy array of lons, and the 
            second element is a numpy array of lats
        """
        gridGeometry = self.jobj.getGridGeometry()
        if gridGeometry is None :
            return None
        
        from com.raytheon.uf.common.geospatial import LatLonReprojection
        latlons = LatLonReprojection.getLatLons(gridGeometry)
        nx = gridGeometry.getGridRange().getSpan(0)
        ny = gridGeometry.getGridRange().getSpan(1)        
        latndarray = np.array(latlons.getLats(), dtype=np.float32)
        latndarray = latndarray.reshape((ny, nx))
        lonndarray = np.array(latlons.getLons(), dtype=np.float32)
        lonndarray = lonndarray.reshape((ny, nx))
        return (lonndarray, latndarray)