mirror of
https://github.com/meteoinfo/MeteoInfo.git
synced 2025-12-08 20:36:05 +00:00
150 lines
4.5 KiB
Python
150 lines
4.5 KiB
Python
# coding=utf-8
|
|
#-----------------------------------------------------
|
|
# Author: Yaqiang Wang
|
|
# Date: 2017-5-25
|
|
# Purpose: MeteoInfoLab io in image package
|
|
# Note: Jython
|
|
#-----------------------------------------------------
|
|
|
|
from org.meteoinfo.shape import Graphic
|
|
from org.meteoinfo.image import ImageUtil, GifDecoder, AnimatedGifEncoder
|
|
from mipylib.geolib.milayer import MILayer
|
|
from mipylib.numeric.core import NDArray
|
|
from org.meteoinfo.image.filter import ContrastFilter, SharpenFilter, RGBAdjustFilter, ChannelMixFilter, \
|
|
GainFilter, GammaFilter, GrayFilter, GrayscaleFilter, HSBAdjustFilter, InvertAlphaFilter, \
|
|
InvertFilter, LevelsFilter, MaskFilter, PosterizeFilter, RescaleFilter, SolarizeFilter, \
|
|
ThresholdFilter, FlipFilter, RotateFilter, EmbossFilter, TritoneFilter, LightFilter, OpacityFilter
|
|
from java.awt.image import BufferedImage
|
|
import math
|
|
import os
|
|
|
|
__all__ = [
|
|
'imread','imload','imwrite','gifopen','gifread','gifload','gifanimation','gifaddframe',
|
|
'giffinish','gifwrite'
|
|
]
|
|
|
|
def imread(fname):
|
|
'''
|
|
Read RGB(A) data array from image file.
|
|
|
|
:param fname: (*String*) Image file name.
|
|
|
|
:returns: (*array*) RGB(A) data array.
|
|
'''
|
|
if not os.path.exists(fname):
|
|
raise IOError(fname)
|
|
r = ImageUtil.imageRead(fname)
|
|
return NDArray(r)
|
|
|
|
def imload(fname):
|
|
'''
|
|
Load image from image file.
|
|
|
|
:param fname: (*String*) Image file name.
|
|
|
|
:returns: (*BufferedImage*) Loadded image.
|
|
'''
|
|
if not os.path.exists(fname):
|
|
raise IOError(fname)
|
|
r = ImageUtil.imageLoad(fname)
|
|
return r
|
|
|
|
def imwrite(a, fname):
|
|
'''
|
|
Write RGB(A) data array or image into an image file.
|
|
|
|
:param a: (*array or BufferedImage*) RGB(A) data array or image.
|
|
:param fname: (*String*) Image file name.
|
|
'''
|
|
ImageUtil.imageSave(a, fname)
|
|
|
|
def gifopen(fname):
|
|
'''
|
|
Open a gif image file.
|
|
|
|
:param fname: (*string*) Gif image file name.
|
|
|
|
:returns: (*GifDecoder*) Gif decoder object.
|
|
'''
|
|
if not os.path.exists(fname):
|
|
raise IOError(fname)
|
|
decoder = GifDecoder()
|
|
decoder.read(fname)
|
|
return decoder
|
|
|
|
def gifread(gif, frame=0):
|
|
'''
|
|
Read RGB(A) data array from a gif image file or a gif decoder object.
|
|
|
|
:param gif: (*string or GifDecoder*) Gif image file or gif decoder object.
|
|
:param frame: (*int*) Image frame index.
|
|
|
|
:returns: (*array*) RGB(A) data array.
|
|
'''
|
|
if isinstance(gif, basestring):
|
|
gif = gifopen(gif)
|
|
im = gif.getFrame(frame)
|
|
r = ImageUtil.imageRead(im)
|
|
return NDArray(r)
|
|
|
|
def gifload(gif, frame=0):
|
|
'''
|
|
Load image from a gif image file or a gif decoder object.
|
|
|
|
:param gif: (*string or GifDecoder*) Gif image file or gif decoder object.
|
|
:param frame: (*int*) Image frame index.
|
|
|
|
:returns: (*BufferedImage*) Loadded image.
|
|
'''
|
|
if isinstance(gif, basestring):
|
|
gif = gifopen(gif)
|
|
im = gif.getFrame(frame)
|
|
return im
|
|
|
|
def gifanimation(filename, repeat=0, delay=1000):
|
|
"""
|
|
Create a gif animation file
|
|
|
|
:param: repeat: (*int, Default 0*) Animation repeat time number. 0 means repeat forever.
|
|
:param: delay: (*int, Default 1000*) Animation frame delay time with units of millsecond.
|
|
|
|
:returns: Gif animation object.
|
|
"""
|
|
encoder = AnimatedGifEncoder()
|
|
encoder.setRepeat(repeat)
|
|
encoder.setDelay(delay)
|
|
encoder.start(filename)
|
|
return encoder
|
|
|
|
def gifaddframe(animation, dpi=None):
|
|
"""
|
|
Add a frame to an gif animation object
|
|
|
|
:param animation: Gif animation object
|
|
:param dpi: (*int*) Image resolution
|
|
"""
|
|
#chartpanel.paintGraphics()
|
|
if dpi is None:
|
|
animation.addFrame(chartpanel.paintViewImage())
|
|
else:
|
|
animation.addFrame(chartpanel.paintViewImage(dpi))
|
|
|
|
def giffinish(animation):
|
|
"""
|
|
Finish a gif animation object and write gif animation image file
|
|
|
|
:param animation: Gif animation object
|
|
"""
|
|
animation.finish()
|
|
|
|
def gifwrite(imfns, giffn, repeat=0, delay=1000):
|
|
'''
|
|
Write a gif animation file.
|
|
|
|
:param imfns: (*list*) Input image file names.
|
|
:param giffn: (*string*) Output gif file name.
|
|
:param: repeat: (*int, Default 0*) Animation repeat time number. 0 means repeat forever.
|
|
:param: delay: (*int, Default 1000*) Animation frame delay time with units of millsecond.
|
|
'''
|
|
ImageUtil.createGifAnimator(imfns, giffn, delay, repeat)
|
|
|