Пример #1
0
"""Test :func:`~vacumm.misc.plot.section2` in quiver mode"""

# Imports
from vcmq import create_lon, N, MV2, create_dep, os, code_file_name, section2
from vacumm.misc.phys.units import deg2m

# Init data with z 1D
nz = 11
nx = 11
x = create_lon(N.arange(nx))
xm = deg2m(x[:],lat=45.) # meters
dx = xm[:].ptp()
z = create_dep((-nz+1, 1.), units='m', long_name='Depth')
dz = z[:].ptp()
scale = dz/dx
u = MV2.ones((nz,nx)) # 1 m/s
w = u*scale           # 1 m/s * scale
for var in u,w:
    var.setAxis(0, z)
    var.setAxis(1, x)
    var.units = 'm/s'

# Plot
figfile = code_file_name(ext='png')
if os.path.exists(figfile): os.remove(figfile)
s = section2((u,w), quiver_norm=1, fill=False, show=False,
    savefig=figfile, close=False, axes_aspect=1)

# Result
result = dict(files=figfile)
Пример #2
0
    def _search(self, nom=None, regexp=True, nmax=5, **kwargs):
        """ Generic search within stations """

        stations = []
        if nom is not None:
            # Search within names using regular expression of strict comparisons
            nom = nom.strip()
            if regexp:
                # Regular expression
                import re
                this_re = re.compile(nom, re.I)
                for station in self._stations:
                    if this_re.search(station['nom']) is not None:
                        stations.append(station)
                        if len(stations) == nmax:
                            break
            else:
                # Strict equality
                for station in self._stations:
                    if station['nom'].lower() == nom.lower():
                        stations.append(station)
                        if len(stations) == nmax:
                            break

        else:
            # Search using other specific arguments
            for key, val in kwargs.items():

                if key in self._ids:
                    # Use ids
                    for station in self._stations:
                        if station in stations:
                            continue
                        if station[key] is not None and \
                               station[key].lower() == val.lower():
                            stations.append(station)
                            break

                elif key == 'position' and len(val) == 2:
                    # Find the closest station
                    import vacumm.misc as M, math
                    distances = []
                    x0 = deg2m(*val)
                    y0 = deg2m(val[1])
                    for station in self._stations:
                        if station in stations:
                            continue
                        x = deg2m(station['longitude'], station['latitude'])
                        y = deg2m(station['latitude'])
                        distances.append(math.sqrt((x - x0)**2 + (y - y0)**2))
                    stations.append(self._stations[N.argmin(distances)])

                if len(stations) == nmax:
                    break

        if len(stations):
            if nmax == 1:
                return stations[0]
            else:
                return stations
        else:
            return None
Пример #3
0
    def _search(self,nom=None,regexp=True,nmax=5,**kwargs):
        """ Generic search within stations """

        stations = []
        if nom is not None:
            # Search within names using regular expression of strict comparisons
            nom = nom.strip()
            if regexp:
                # Regular expression
                import re
                this_re = re.compile(nom, re.I)
                for station in self._stations:
                    if this_re.search(station['nom']) is not None:
                        stations.append(station)
                        if len(stations) == nmax:
                            break
            else:
                # Strict equality
                for station in self._stations:
                    if station['nom'].lower() == nom.lower():
                        stations.append(station)
                        if len(stations) == nmax:
                            break

        else:
            # Search using other specific arguments
            for key,val in kwargs.items():

                if key in self._ids:
                    # Use ids
                    for station in self._stations:
                        if station in stations:
                            continue
                        if station[key] is not None and \
                               station[key].lower() == val.lower():
                            stations.append(station)
                            break

                elif key == 'position' and len(val) == 2:
                    # Find the closest station
                    import vacumm.misc as M,math
                    distances = []
                    x0 = deg2m(*val)
                    y0 = deg2m(val[1])
                    for station in self._stations:
                        if station in stations:
                            continue
                        x = deg2m(station['longitude'], station['latitude'])
                        y = deg2m(station['latitude'])
                        distances.append(math.sqrt((x-x0)**2+(y-y0)**2))
                    stations.append(self._stations[N.argmin(distances)])

                if len(stations) == nmax:
                    break

        if len(stations):
            if nmax == 1:
                return stations[0]
            else:
                return stations
        else:
            return None
Пример #4
0
"""Test :func:`~vacumm.misc.plot.section2` in quiver mode"""

# Imports
from vcmq import create_lon, N, MV2, create_dep, os, code_file_name, section2
from vacumm.misc.phys.units import deg2m

# Init data with z 1D
nz = 11
nx = 11
x = create_lon(N.arange(nx))
xm = deg2m(x[:], lat=45.)  # meters
dx = xm[:].ptp()
z = create_dep((-nz + 1, 1.), units='m', long_name='Depth')
dz = z[:].ptp()
scale = dz / dx
u = MV2.ones((nz, nx))  # 1 m/s
w = u * scale  # 1 m/s * scale
for var in u, w:
    var.setAxis(0, z)
    var.setAxis(1, x)
    var.units = 'm/s'

# Plot
figfile = code_file_name(ext='png')
if os.path.exists(figfile): os.remove(figfile)
s = section2((u, w),
             quiver_norm=1,
             fill=False,
             show=False,
             savefig=figfile,
             close=True,