*.pyc | |||||
*.sqlite3 | |||||
*.aux |
{ | |||||
"editor.formatOnSave": true, | |||||
"editor.formatOnPaste": true, | |||||
"editor.formatOnType": true, | |||||
"editor.defaultFormatter": "esbenp.prettier-vscode", | |||||
"[html]": { | |||||
"editor.defaultFormatter": "vscode.html-language-features" | |||||
}, | |||||
"[python]": { | |||||
"editor.defaultFormatter": "ms-python.black-formatter" | |||||
}, | |||||
"python.formatting.provider": "none", | |||||
"[latex]": { | |||||
"editor.defaultFormatter": "James-Yu.latex-workshop" | |||||
} | |||||
} |
# Python Basisimage | |||||
FROM python:3.9-alpine | |||||
# Arbeitsverzeichnis | |||||
WORKDIR /app | |||||
# Kopieren des Projekts | |||||
COPY manage.py /app/manage.py | |||||
COPY Webgame /app/Webgame | |||||
COPY mygame /app/mygame | |||||
# Abhängigkeiten installieren | |||||
RUN pip install -r /app/Webgame/requirements.txt | |||||
# Umgebungsvariable setzen | |||||
ENV DJANGO_SETTINGS_MODULE=Webgame.settings | |||||
# Static files in das Image | |||||
RUN python manage.py collectstatic --noinput | |||||
# Datenbank | |||||
RUN python manage.py migrate | |||||
# Port freigeben | |||||
EXPOSE 8000 | |||||
# Anwendung starten mit Daphne | |||||
CMD ["daphne", "Webgame.asgi:application", "--bind", "0.0.0.0", "--port", "8000"] |
""" | |||||
ASGI config for Webgame project. | |||||
It exposes the ASGI callable as a module-level variable named ``application``. | |||||
For more information on this file, see | |||||
https://docs.djangoproject.com/en/4.1/howto/deployment/asgi/ | |||||
""" | |||||
import os | |||||
from django.core.asgi import get_asgi_application | |||||
os.environ.setdefault("DJANGO_SETTINGS_MODULE", "Webgame.settings") | |||||
app = get_asgi_application() | |||||
from channels.auth import AuthMiddlewareStack | |||||
from channels.routing import ProtocolTypeRouter, URLRouter | |||||
from channels.security.websocket import AllowedHostsOriginValidator | |||||
from channels.routing import ProtocolTypeRouter, URLRouter | |||||
from channels.auth import AuthMiddlewareStack | |||||
import mygame.routing | |||||
application = ProtocolTypeRouter( | |||||
{ | |||||
"http": app, | |||||
"websocket": AllowedHostsOriginValidator( | |||||
AuthMiddlewareStack(URLRouter(mygame.routing.websocket_urlpatterns)) | |||||
), | |||||
} | |||||
) |
from django.urls import path, include | |||||
urlpatterns = [ | |||||
# ... andere URLs in deinem Projekt ... | |||||
path("mygame/", include("mygame.routing")), | |||||
] |
""" | |||||
Django settings for Webgame project. | |||||
Generated by 'django-admin startproject' using Django 4.1.7. | |||||
For more information on this file, see | |||||
https://docs.djangoproject.com/en/4.1/topics/settings/ | |||||
For the full list of settings and their values, see | |||||
https://docs.djangoproject.com/en/4.1/ref/settings/ | |||||
""" | |||||
from pathlib import Path | |||||
import os | |||||
# Build paths inside the project like this: BASE_DIR / 'subdir'. | |||||
BASE_DIR = Path(__file__).resolve().parent.parent | |||||
# Quick-start development settings - unsuitable for production | |||||
# See https://docs.djangoproject.com/en/4.1/howto/deployment/checklist/ | |||||
# SECURITY WARNING: keep the secret key used in production secret! | |||||
SECRET_KEY = "django-insecure-mmfrxsf3o6)_4@caak0h_ldedii4x_(&fhg8b7(hd77bfmlo1#" | |||||
# SECURITY WARNING: don't run with debug turned on in production! | |||||
DEBUG = True | |||||
ALLOWED_HOSTS = ["*"] | |||||
# Application definition | |||||
INSTALLED_APPS = [ | |||||
"daphne", | |||||
"channels", | |||||
"polymorphic", | |||||
"django.contrib.admin", | |||||
"django.contrib.auth", | |||||
"django.contrib.contenttypes", | |||||
"django.contrib.sessions", | |||||
"django.contrib.messages", | |||||
"django.contrib.staticfiles", | |||||
"mygame", | |||||
] | |||||
# FÜR CHANNELS WEBSOCKET | |||||
ASGI_APPLICATION = "Webgame.asgi.application" | |||||
CHANNEL_LAYERS = { | |||||
"default": { | |||||
"BACKEND": "channels.layers.InMemoryChannelLayer", | |||||
}, | |||||
} | |||||
CHANNELS_SETTINGS = { | |||||
"CHANNELS_WS_PROTOCOL_TIMEOUT": 5, | |||||
} | |||||
MIDDLEWARE = [ | |||||
"django.middleware.security.SecurityMiddleware", | |||||
"django.contrib.sessions.middleware.SessionMiddleware", | |||||
"django.middleware.common.CommonMiddleware", | |||||
"django.middleware.csrf.CsrfViewMiddleware", | |||||
"django.contrib.auth.middleware.AuthenticationMiddleware", | |||||
"django.contrib.messages.middleware.MessageMiddleware", | |||||
"django.middleware.clickjacking.XFrameOptionsMiddleware", | |||||
] | |||||
ROOT_URLCONF = "Webgame.urls" | |||||
TEMPLATES = [ | |||||
{ | |||||
"BACKEND": "django.template.backends.django.DjangoTemplates", | |||||
"DIRS": [os.path.join(BASE_DIR, "templates")], | |||||
"APP_DIRS": True, | |||||
"OPTIONS": { | |||||
"context_processors": [ | |||||
"django.template.context_processors.debug", | |||||
"django.template.context_processors.request", | |||||
"django.contrib.auth.context_processors.auth", | |||||
"django.contrib.messages.context_processors.messages", | |||||
], | |||||
}, | |||||
}, | |||||
] | |||||
WSGI_APPLICATION = "Webgame.wsgi.application" | |||||
# Database | |||||
# https://docs.djangoproject.com/en/4.1/ref/settings/#databases | |||||
DATABASES = { | |||||
"default": { | |||||
"ENGINE": "django.db.backends.sqlite3", | |||||
"NAME": BASE_DIR / "db.sqlite3", | |||||
} | |||||
} | |||||
# Password validation | |||||
# https://docs.djangoproject.com/en/4.1/ref/settings/#auth-password-validators | |||||
AUTH_PASSWORD_VALIDATORS = [ | |||||
{ | |||||
"NAME": "django.contrib.auth.password_validation.UserAttributeSimilarityValidator", | |||||
}, | |||||
{ | |||||
"NAME": "django.contrib.auth.password_validation.MinimumLengthValidator", | |||||
}, | |||||
{ | |||||
"NAME": "django.contrib.auth.password_validation.CommonPasswordValidator", | |||||
}, | |||||
{ | |||||
"NAME": "django.contrib.auth.password_validation.NumericPasswordValidator", | |||||
}, | |||||
] | |||||
# Internationalization | |||||
# https://docs.djangoproject.com/en/4.1/topics/i18n/ | |||||
LANGUAGE_CODE = "en-us" | |||||
TIME_ZONE = "UTC" | |||||
USE_I18N = True | |||||
USE_TZ = True | |||||
# Static files (CSS, JavaScript, Images) | |||||
# https://docs.djangoproject.com/en/4.1/howto/static-files/ | |||||
STATIC_URL = "/static/" | |||||
STATIC_ROOT = "/app/static/" | |||||
# Default primary key field type | |||||
# https://docs.djangoproject.com/en/4.1/ref/settings/#default-auto-field | |||||
DEFAULT_AUTO_FIELD = "django.db.models.BigAutoField" |
"""Webgame URL Configuration | |||||
The `urlpatterns` list routes URLs to views. For more information please see: | |||||
https://docs.djangoproject.com/en/4.1/topics/http/urls/ | |||||
Examples: | |||||
Function views | |||||
1. Add an import: from my_app import views | |||||
2. Add a URL to urlpatterns: path('', views.home, name='home') | |||||
Class-based views | |||||
1. Add an import: from other_app.views import Home | |||||
2. Add a URL to urlpatterns: path('', Home.as_view(), name='home') | |||||
Including another URLconf | |||||
1. Import the include() function: from django.urls import include, path | |||||
2. Add a URL to urlpatterns: path('blog/', include('blog.urls')) | |||||
""" | |||||
from django.contrib import admin | |||||
from django.urls import include, path | |||||
from django.conf import settings | |||||
from django.conf.urls.static import static | |||||
urlpatterns = [ | |||||
path("admin/", admin.site.urls), | |||||
path("mygame/", include("mygame.urls")), | |||||
] | |||||
if settings.DEBUG: | |||||
urlpatterns += static(settings.STATIC_URL, document_root=settings.STATIC_ROOT) |
""" | |||||
WSGI config for Webgame project. | |||||
It exposes the WSGI callable as a module-level variable named ``application``. | |||||
For more information on this file, see | |||||
https://docs.djangoproject.com/en/4.1/howto/deployment/wsgi/ | |||||
""" | |||||
import os | |||||
from django.core.wsgi import get_wsgi_application | |||||
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'Webgame.settings') | |||||
application = get_wsgi_application() |
pip |
Copyright (c) 2014 | |||||
Rackspace | |||||
Permission is hereby granted, free of charge, to any person obtaining | |||||
a copy of this software and associated documentation files (the | |||||
"Software"), to deal in the Software without restriction, including | |||||
without limitation the rights to use, copy, modify, merge, publish, | |||||
distribute, sublicense, and/or sell copies of the Software, and to | |||||
permit persons to whom the Software is furnished to do so, subject to | |||||
the following conditions: | |||||
The above copyright notice and this permission notice shall be | |||||
included in all copies or substantial portions of the Software. | |||||
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, | |||||
EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF | |||||
MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND | |||||
NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE | |||||
LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION | |||||
OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION | |||||
WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. |
Metadata-Version: 2.1 | |||||
Name: Automat | |||||
Version: 22.10.0 | |||||
Summary: Self-service finite-state machines for the programmer on the go. | |||||
Home-page: https://github.com/glyph/Automat | |||||
Author: Glyph | |||||
Author-email: glyph@twistedmatrix.com | |||||
License: MIT | |||||
Keywords: fsm finite state machine automata | |||||
Classifier: Intended Audience :: Developers | |||||
Classifier: License :: OSI Approved :: MIT License | |||||
Classifier: Operating System :: OS Independent | |||||
Classifier: Programming Language :: Python | |||||
Classifier: Programming Language :: Python :: 2 | |||||
Classifier: Programming Language :: Python :: 2.7 | |||||
Classifier: Programming Language :: Python :: 3 | |||||
Classifier: Programming Language :: Python :: 3.5 | |||||
Classifier: Programming Language :: Python :: 3.6 | |||||
Classifier: Programming Language :: Python :: 3.7 | |||||
Classifier: Programming Language :: Python :: 3.8 | |||||
License-File: LICENSE | |||||
Requires-Dist: attrs (>=19.2.0) | |||||
Requires-Dist: six | |||||
Provides-Extra: visualize | |||||
Requires-Dist: graphviz (>0.5.1) ; extra == 'visualize' | |||||
Requires-Dist: Twisted (>=16.1.1) ; extra == 'visualize' | |||||
../../Scripts/automat-visualize.exe,sha256=roGC4pPXzBoCAMQc0zBGrhxGGJX6KY4msfjS9rSXp50,108444 | |||||
Automat-22.10.0.dist-info/INSTALLER,sha256=zuuue4knoyJ-UwPPXg8fezS7VCrXJQrAP7zeNuwvFQg,4 | |||||
Automat-22.10.0.dist-info/LICENSE,sha256=siATAWeNCpN9k4VDgnyhNgcS6zTiPejuPzv_-9TA43Y,1053 | |||||
Automat-22.10.0.dist-info/METADATA,sha256=Ra-n4-tsp6oC5_GDXvxNs4-r3yf5OaB4mG099Z1_T8o,1024 | |||||
Automat-22.10.0.dist-info/RECORD,, | |||||
Automat-22.10.0.dist-info/WHEEL,sha256=z9j0xAa_JmUKMpmz72K0ZGALSM_n-wQVmGbleXx2VHg,110 | |||||
Automat-22.10.0.dist-info/entry_points.txt,sha256=D5Dc6byHpLWAktM443OHbx0ZGl1ZBZtf6rPNlGi7NbQ,62 | |||||
Automat-22.10.0.dist-info/top_level.txt,sha256=vg4zAOyhP_3YCmpKZLNgFw1uMF3lC_b6TKsdz7jBSpI,8 | |||||
automat/__init__.py,sha256=ec8PILBwt35xyzsstU9kx8p5ADi6KX9d1rBLZsocN_Y,169 | |||||
automat/__pycache__/__init__.cpython-39.pyc,, | |||||
automat/__pycache__/_core.cpython-39.pyc,, | |||||
automat/__pycache__/_discover.cpython-39.pyc,, | |||||
automat/__pycache__/_introspection.cpython-39.pyc,, | |||||
automat/__pycache__/_methodical.cpython-39.pyc,, | |||||
automat/__pycache__/_visualize.cpython-39.pyc,, | |||||
automat/_core.py,sha256=IEtZHq3wsBZPX_VfMHMFy_uNjx1kfE11Qq-nmIXZe28,4819 | |||||
automat/_discover.py,sha256=ye7NHLZkrwYsPmBpTIyJuJ-VRCmwkOUpA0is-A81z04,4367 | |||||
automat/_introspection.py,sha256=L2QjbGut8LLGbgtapvI5t6MoT5v9vd2-N1nHWnaJVm8,1385 | |||||
automat/_methodical.py,sha256=ZxRl_1TzK0RLANBFrX0zlX_fbtEEsVbbCE0XtqG5hz8,15837 | |||||
automat/_test/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0 | |||||
automat/_test/__pycache__/__init__.cpython-39.pyc,, | |||||
automat/_test/__pycache__/test_core.cpython-39.pyc,, | |||||
automat/_test/__pycache__/test_discover.cpython-39.pyc,, | |||||
automat/_test/__pycache__/test_methodical.cpython-39.pyc,, | |||||
automat/_test/__pycache__/test_trace.cpython-39.pyc,, | |||||
automat/_test/__pycache__/test_visualize.cpython-39.pyc,, | |||||
automat/_test/test_core.py,sha256=CDmGBQNi9Pr7ZktfBcOhBvwI7wjLLYRtWZ0P5baXONY,2833 | |||||
automat/_test/test_discover.py,sha256=8QUcW03hVX_amlHDU4MX1O7AZiOiw7UFo3bHvksVcCg,21826 | |||||
automat/_test/test_methodical.py,sha256=0oL6mWc39QJFF0BrDnXGCpviGts8zafHyVnIjSJ9TJg,20210 | |||||
automat/_test/test_trace.py,sha256=Mx1B8QgaE7QFk6blTie2j-Vx95hTV-zySnlxLalt8ek,3279 | |||||
automat/_test/test_visualize.py,sha256=bF_JWkMafGv_Ybi5C5OZq46AinWmms4v5ydZDCaqltw,13996 | |||||
automat/_visualize.py,sha256=jY8HkzaGdMoXB7LavvaneW4GdtBN6PRShl7-4OXDvss,6335 |
Wheel-Version: 1.0 | |||||
Generator: bdist_wheel (0.37.1) | |||||
Root-Is-Purelib: true | |||||
Tag: py2-none-any | |||||
Tag: py3-none-any | |||||
[console_scripts] | |||||
automat-visualize = automat._visualize:tool |
automat |
pip |
Copyright (c) Django Software Foundation and individual contributors. | |||||
All rights reserved. | |||||
Redistribution and use in source and binary forms, with or without modification, | |||||
are permitted provided that the following conditions are met: | |||||
1. Redistributions of source code must retain the above copyright notice, | |||||
this list of conditions and the following disclaimer. | |||||
2. Redistributions in binary form must reproduce the above copyright | |||||
notice, this list of conditions and the following disclaimer in the | |||||
documentation and/or other materials provided with the distribution. | |||||
3. Neither the name of Django nor the names of its contributors may be used | |||||
to endorse or promote products derived from this software without | |||||
specific prior written permission. | |||||
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND | |||||
ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED | |||||
WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE | |||||
DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR | |||||
ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES | |||||
(INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; | |||||
LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON | |||||
ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT | |||||
(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS | |||||
SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
Django is licensed under the three-clause BSD license; see the file | |||||
LICENSE for details. | |||||
Django includes code from the Python standard library, which is licensed under | |||||
the Python license, a permissive open source license. The copyright and license | |||||
is included below for compliance with Python's terms. | |||||
---------------------------------------------------------------------- | |||||
Copyright (c) 2001-present Python Software Foundation; All Rights Reserved | |||||
A. HISTORY OF THE SOFTWARE | |||||
========================== | |||||
Python was created in the early 1990s by Guido van Rossum at Stichting | |||||
Mathematisch Centrum (CWI, see http://www.cwi.nl) in the Netherlands | |||||
as a successor of a language called ABC. Guido remains Python's | |||||
principal author, although it includes many contributions from others. | |||||
In 1995, Guido continued his work on Python at the Corporation for | |||||
National Research Initiatives (CNRI, see http://www.cnri.reston.va.us) | |||||
in Reston, Virginia where he released several versions of the | |||||
software. | |||||
In May 2000, Guido and the Python core development team moved to | |||||
BeOpen.com to form the BeOpen PythonLabs team. In October of the same | |||||
year, the PythonLabs team moved to Digital Creations, which became | |||||
Zope Corporation. In 2001, the Python Software Foundation (PSF, see | |||||
https://www.python.org/psf/) was formed, a non-profit organization | |||||
created specifically to own Python-related Intellectual Property. | |||||
Zope Corporation was a sponsoring member of the PSF. | |||||
All Python releases are Open Source (see http://www.opensource.org for | |||||
the Open Source Definition). Historically, most, but not all, Python | |||||
releases have also been GPL-compatible; the table below summarizes | |||||
the various releases. | |||||
Release Derived Year Owner GPL- | |||||
from compatible? (1) | |||||
0.9.0 thru 1.2 1991-1995 CWI yes | |||||
1.3 thru 1.5.2 1.2 1995-1999 CNRI yes | |||||
1.6 1.5.2 2000 CNRI no | |||||
2.0 1.6 2000 BeOpen.com no | |||||
1.6.1 1.6 2001 CNRI yes (2) | |||||
2.1 2.0+1.6.1 2001 PSF no | |||||
2.0.1 2.0+1.6.1 2001 PSF yes | |||||
2.1.1 2.1+2.0.1 2001 PSF yes | |||||
2.1.2 2.1.1 2002 PSF yes | |||||
2.1.3 2.1.2 2002 PSF yes | |||||
2.2 and above 2.1.1 2001-now PSF yes | |||||
Footnotes: | |||||
(1) GPL-compatible doesn't mean that we're distributing Python under | |||||
the GPL. All Python licenses, unlike the GPL, let you distribute | |||||
a modified version without making your changes open source. The | |||||
GPL-compatible licenses make it possible to combine Python with | |||||
other software that is released under the GPL; the others don't. | |||||
(2) According to Richard Stallman, 1.6.1 is not GPL-compatible, | |||||
because its license has a choice of law clause. According to | |||||
CNRI, however, Stallman's lawyer has told CNRI's lawyer that 1.6.1 | |||||
is "not incompatible" with the GPL. | |||||
Thanks to the many outside volunteers who have worked under Guido's | |||||
direction to make these releases possible. | |||||
B. TERMS AND CONDITIONS FOR ACCESSING OR OTHERWISE USING PYTHON | |||||
=============================================================== | |||||
Python software and documentation are licensed under the | |||||
Python Software Foundation License Version 2. | |||||
Starting with Python 3.8.6, examples, recipes, and other code in | |||||
the documentation are dual licensed under the PSF License Version 2 | |||||
and the Zero-Clause BSD license. | |||||
Some software incorporated into Python is under different licenses. | |||||
The licenses are listed with code falling under that license. | |||||
PYTHON SOFTWARE FOUNDATION LICENSE VERSION 2 | |||||
-------------------------------------------- | |||||
1. This LICENSE AGREEMENT is between the Python Software Foundation | |||||
("PSF"), and the Individual or Organization ("Licensee") accessing and | |||||
otherwise using this software ("Python") in source or binary form and | |||||
its associated documentation. | |||||
2. Subject to the terms and conditions of this License Agreement, PSF hereby | |||||
grants Licensee a nonexclusive, royalty-free, world-wide license to reproduce, | |||||
analyze, test, perform and/or display publicly, prepare derivative works, | |||||
distribute, and otherwise use Python alone or in any derivative version, | |||||
provided, however, that PSF's License Agreement and PSF's notice of copyright, | |||||
i.e., "Copyright (c) 2001, 2002, 2003, 2004, 2005, 2006, 2007, 2008, 2009, 2010, | |||||
2011, 2012, 2013, 2014, 2015, 2016, 2017, 2018, 2019, 2020, 2021, 2022 Python Software Foundation; | |||||
All Rights Reserved" are retained in Python alone or in any derivative version | |||||
prepared by Licensee. | |||||
3. In the event Licensee prepares a derivative work that is based on | |||||
or incorporates Python or any part thereof, and wants to make | |||||
the derivative work available to others as provided herein, then | |||||
Licensee hereby agrees to include in any such work a brief summary of | |||||
the changes made to Python. | |||||
4. PSF is making Python available to Licensee on an "AS IS" | |||||
basis. PSF MAKES NO REPRESENTATIONS OR WARRANTIES, EXPRESS OR | |||||
IMPLIED. BY WAY OF EXAMPLE, BUT NOT LIMITATION, PSF MAKES NO AND | |||||
DISCLAIMS ANY REPRESENTATION OR WARRANTY OF MERCHANTABILITY OR FITNESS | |||||
FOR ANY PARTICULAR PURPOSE OR THAT THE USE OF PYTHON WILL NOT | |||||
INFRINGE ANY THIRD PARTY RIGHTS. | |||||
5. PSF SHALL NOT BE LIABLE TO LICENSEE OR ANY OTHER USERS OF PYTHON | |||||
FOR ANY INCIDENTAL, SPECIAL, OR CONSEQUENTIAL DAMAGES OR LOSS AS | |||||
A RESULT OF MODIFYING, DISTRIBUTING, OR OTHERWISE USING PYTHON, | |||||
OR ANY DERIVATIVE THEREOF, EVEN IF ADVISED OF THE POSSIBILITY THEREOF. | |||||
6. This License Agreement will automatically terminate upon a material | |||||
breach of its terms and conditions. | |||||
7. Nothing in this License Agreement shall be deemed to create any | |||||
relationship of agency, partnership, or joint venture between PSF and | |||||
Licensee. This License Agreement does not grant permission to use PSF | |||||
trademarks or trade name in a trademark sense to endorse or promote | |||||
products or services of Licensee, or any third party. | |||||
8. By copying, installing or otherwise using Python, Licensee | |||||
agrees to be bound by the terms and conditions of this License | |||||
Agreement. | |||||
BEOPEN.COM LICENSE AGREEMENT FOR PYTHON 2.0 | |||||
------------------------------------------- | |||||
BEOPEN PYTHON OPEN SOURCE LICENSE AGREEMENT VERSION 1 | |||||
1. This LICENSE AGREEMENT is between BeOpen.com ("BeOpen"), having an | |||||
office at 160 Saratoga Avenue, Santa Clara, CA 95051, and the | |||||
Individual or Organization ("Licensee") accessing and otherwise using | |||||
this software in source or binary form and its associated | |||||
documentation ("the Software"). | |||||
2. Subject to the terms and conditions of this BeOpen Python License | |||||
Agreement, BeOpen hereby grants Licensee a non-exclusive, | |||||
royalty-free, world-wide license to reproduce, analyze, test, perform | |||||
and/or display publicly, prepare derivative works, distribute, and | |||||
otherwise use the Software alone or in any derivative version, | |||||
provided, however, that the BeOpen Python License is retained in the | |||||
Software, alone or in any derivative version prepared by Licensee. | |||||
3. BeOpen is making the Software available to Licensee on an "AS IS" | |||||
basis. BEOPEN MAKES NO REPRESENTATIONS OR WARRANTIES, EXPRESS OR | |||||
IMPLIED. BY WAY OF EXAMPLE, BUT NOT LIMITATION, BEOPEN MAKES NO AND | |||||
DISCLAIMS ANY REPRESENTATION OR WARRANTY OF MERCHANTABILITY OR FITNESS | |||||
FOR ANY PARTICULAR PURPOSE OR THAT THE USE OF THE SOFTWARE WILL NOT | |||||
INFRINGE ANY THIRD PARTY RIGHTS. | |||||
4. BEOPEN SHALL NOT BE LIABLE TO LICENSEE OR ANY OTHER USERS OF THE | |||||
SOFTWARE FOR ANY INCIDENTAL, SPECIAL, OR CONSEQUENTIAL DAMAGES OR LOSS | |||||
AS A RESULT OF USING, MODIFYING OR DISTRIBUTING THE SOFTWARE, OR ANY | |||||
DERIVATIVE THEREOF, EVEN IF ADVISED OF THE POSSIBILITY THEREOF. | |||||
5. This License Agreement will automatically terminate upon a material | |||||
breach of its terms and conditions. | |||||
6. This License Agreement shall be governed by and interpreted in all | |||||
respects by the law of the State of California, excluding conflict of | |||||
law provisions. Nothing in this License Agreement shall be deemed to | |||||
create any relationship of agency, partnership, or joint venture | |||||
between BeOpen and Licensee. This License Agreement does not grant | |||||
permission to use BeOpen trademarks or trade names in a trademark | |||||
sense to endorse or promote products or services of Licensee, or any | |||||
third party. As an exception, the "BeOpen Python" logos available at | |||||
http://www.pythonlabs.com/logos.html may be used according to the | |||||
permissions granted on that web page. | |||||
7. By copying, installing or otherwise using the software, Licensee | |||||
agrees to be bound by the terms and conditions of this License | |||||
Agreement. | |||||
CNRI LICENSE AGREEMENT FOR PYTHON 1.6.1 | |||||
--------------------------------------- | |||||
1. This LICENSE AGREEMENT is between the Corporation for National | |||||
Research Initiatives, having an office at 1895 Preston White Drive, | |||||
Reston, VA 20191 ("CNRI"), and the Individual or Organization | |||||
("Licensee") accessing and otherwise using Python 1.6.1 software in | |||||
source or binary form and its associated documentation. | |||||
2. Subject to the terms and conditions of this License Agreement, CNRI | |||||
hereby grants Licensee a nonexclusive, royalty-free, world-wide | |||||
license to reproduce, analyze, test, perform and/or display publicly, | |||||
prepare derivative works, distribute, and otherwise use Python 1.6.1 | |||||
alone or in any derivative version, provided, however, that CNRI's | |||||
License Agreement and CNRI's notice of copyright, i.e., "Copyright (c) | |||||
1995-2001 Corporation for National Research Initiatives; All Rights | |||||
Reserved" are retained in Python 1.6.1 alone or in any derivative | |||||
version prepared by Licensee. Alternately, in lieu of CNRI's License | |||||
Agreement, Licensee may substitute the following text (omitting the | |||||
quotes): "Python 1.6.1 is made available subject to the terms and | |||||
conditions in CNRI's License Agreement. This Agreement together with | |||||
Python 1.6.1 may be located on the internet using the following | |||||
unique, persistent identifier (known as a handle): 1895.22/1013. This | |||||
Agreement may also be obtained from a proxy server on the internet | |||||
using the following URL: http://hdl.handle.net/1895.22/1013". | |||||
3. In the event Licensee prepares a derivative work that is based on | |||||
or incorporates Python 1.6.1 or any part thereof, and wants to make | |||||
the derivative work available to others as provided herein, then | |||||
Licensee hereby agrees to include in any such work a brief summary of | |||||
the changes made to Python 1.6.1. | |||||
4. CNRI is making Python 1.6.1 available to Licensee on an "AS IS" | |||||
basis. CNRI MAKES NO REPRESENTATIONS OR WARRANTIES, EXPRESS OR | |||||
IMPLIED. BY WAY OF EXAMPLE, BUT NOT LIMITATION, CNRI MAKES NO AND | |||||
DISCLAIMS ANY REPRESENTATION OR WARRANTY OF MERCHANTABILITY OR FITNESS | |||||
FOR ANY PARTICULAR PURPOSE OR THAT THE USE OF PYTHON 1.6.1 WILL NOT | |||||
INFRINGE ANY THIRD PARTY RIGHTS. | |||||
5. CNRI SHALL NOT BE LIABLE TO LICENSEE OR ANY OTHER USERS OF PYTHON | |||||
1.6.1 FOR ANY INCIDENTAL, SPECIAL, OR CONSEQUENTIAL DAMAGES OR LOSS AS | |||||
A RESULT OF MODIFYING, DISTRIBUTING, OR OTHERWISE USING PYTHON 1.6.1, | |||||
OR ANY DERIVATIVE THEREOF, EVEN IF ADVISED OF THE POSSIBILITY THEREOF. | |||||
6. This License Agreement will automatically terminate upon a material | |||||
breach of its terms and conditions. | |||||
7. This License Agreement shall be governed by the federal | |||||
intellectual property law of the United States, including without | |||||
limitation the federal copyright law, and, to the extent such | |||||
U.S. federal law does not apply, by the law of the Commonwealth of | |||||
Virginia, excluding Virginia's conflict of law provisions. | |||||
Notwithstanding the foregoing, with regard to derivative works based | |||||
on Python 1.6.1 that incorporate non-separable material that was | |||||
previously distributed under the GNU General Public License (GPL), the | |||||
law of the Commonwealth of Virginia shall govern this License | |||||
Agreement only as to issues arising under or with respect to | |||||
Paragraphs 4, 5, and 7 of this License Agreement. Nothing in this | |||||
License Agreement shall be deemed to create any relationship of | |||||
agency, partnership, or joint venture between CNRI and Licensee. This | |||||
License Agreement does not grant permission to use CNRI trademarks or | |||||
trade name in a trademark sense to endorse or promote products or | |||||
services of Licensee, or any third party. | |||||
8. By clicking on the "ACCEPT" button where indicated, or by copying, | |||||
installing or otherwise using Python 1.6.1, Licensee agrees to be | |||||
bound by the terms and conditions of this License Agreement. | |||||
ACCEPT | |||||
CWI LICENSE AGREEMENT FOR PYTHON 0.9.0 THROUGH 1.2 | |||||
-------------------------------------------------- | |||||
Copyright (c) 1991 - 1995, Stichting Mathematisch Centrum Amsterdam, | |||||
The Netherlands. All rights reserved. | |||||
Permission to use, copy, modify, and distribute this software and its | |||||
documentation for any purpose and without fee is hereby granted, | |||||
provided that the above copyright notice appear in all copies and that | |||||
both that copyright notice and this permission notice appear in | |||||
supporting documentation, and that the name of Stichting Mathematisch | |||||
Centrum or CWI not be used in advertising or publicity pertaining to | |||||
distribution of the software without specific, written prior | |||||
permission. | |||||
STICHTING MATHEMATISCH CENTRUM DISCLAIMS ALL WARRANTIES WITH REGARD TO | |||||
THIS SOFTWARE, INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY AND | |||||
FITNESS, IN NO EVENT SHALL STICHTING MATHEMATISCH CENTRUM BE LIABLE | |||||
FOR ANY SPECIAL, INDIRECT OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES | |||||
WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN | |||||
ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT | |||||
OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. | |||||
ZERO-CLAUSE BSD LICENSE FOR CODE IN THE PYTHON DOCUMENTATION | |||||
---------------------------------------------------------------------- | |||||
Permission to use, copy, modify, and/or distribute this software for any | |||||
purpose with or without fee is hereby granted. | |||||
THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH | |||||
REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY | |||||
AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT, | |||||
INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM | |||||
LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR | |||||
OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR | |||||
PERFORMANCE OF THIS SOFTWARE. |
Metadata-Version: 2.1 | |||||
Name: Django | |||||
Version: 4.1.7 | |||||
Summary: A high-level Python web framework that encourages rapid development and clean, pragmatic design. | |||||
Home-page: https://www.djangoproject.com/ | |||||
Author: Django Software Foundation | |||||
Author-email: foundation@djangoproject.com | |||||
License: BSD-3-Clause | |||||
Project-URL: Documentation, https://docs.djangoproject.com/ | |||||
Project-URL: Release notes, https://docs.djangoproject.com/en/stable/releases/ | |||||
Project-URL: Funding, https://www.djangoproject.com/fundraising/ | |||||
Project-URL: Source, https://github.com/django/django | |||||
Project-URL: Tracker, https://code.djangoproject.com/ | |||||
Platform: UNKNOWN | |||||
Classifier: Development Status :: 5 - Production/Stable | |||||
Classifier: Environment :: Web Environment | |||||
Classifier: Framework :: Django | |||||
Classifier: Intended Audience :: Developers | |||||
Classifier: License :: OSI Approved :: BSD License | |||||
Classifier: Operating System :: OS Independent | |||||
Classifier: Programming Language :: Python | |||||
Classifier: Programming Language :: Python :: 3 | |||||
Classifier: Programming Language :: Python :: 3 :: Only | |||||
Classifier: Programming Language :: Python :: 3.8 | |||||
Classifier: Programming Language :: Python :: 3.9 | |||||
Classifier: Programming Language :: Python :: 3.10 | |||||
Classifier: Programming Language :: Python :: 3.11 | |||||
Classifier: Topic :: Internet :: WWW/HTTP | |||||
Classifier: Topic :: Internet :: WWW/HTTP :: Dynamic Content | |||||
Classifier: Topic :: Internet :: WWW/HTTP :: WSGI | |||||
Classifier: Topic :: Software Development :: Libraries :: Application Frameworks | |||||
Classifier: Topic :: Software Development :: Libraries :: Python Modules | |||||
Requires-Python: >=3.8 | |||||
License-File: LICENSE | |||||
License-File: LICENSE.python | |||||
License-File: AUTHORS | |||||
Requires-Dist: asgiref (<4,>=3.5.2) | |||||
Requires-Dist: sqlparse (>=0.2.2) | |||||
Requires-Dist: backports.zoneinfo ; python_version < "3.9" | |||||
Requires-Dist: tzdata ; sys_platform == "win32" | |||||
Provides-Extra: argon2 | |||||
Requires-Dist: argon2-cffi (>=19.1.0) ; extra == 'argon2' | |||||
Provides-Extra: bcrypt | |||||
Requires-Dist: bcrypt ; extra == 'bcrypt' | |||||
====== | |||||
Django | |||||
====== | |||||
Django is a high-level Python web framework that encourages rapid development | |||||
and clean, pragmatic design. Thanks for checking it out. | |||||
All documentation is in the "``docs``" directory and online at | |||||
https://docs.djangoproject.com/en/stable/. If you're just getting started, | |||||
here's how we recommend you read the docs: | |||||
* First, read ``docs/intro/install.txt`` for instructions on installing Django. | |||||
* Next, work through the tutorials in order (``docs/intro/tutorial01.txt``, | |||||
``docs/intro/tutorial02.txt``, etc.). | |||||
* If you want to set up an actual deployment server, read | |||||
``docs/howto/deployment/index.txt`` for instructions. | |||||
* You'll probably want to read through the topical guides (in ``docs/topics``) | |||||
next; from there you can jump to the HOWTOs (in ``docs/howto``) for specific | |||||
problems, and check out the reference (``docs/ref``) for gory details. | |||||
* See ``docs/README`` for instructions on building an HTML version of the docs. | |||||
Docs are updated rigorously. If you find any problems in the docs, or think | |||||
they should be clarified in any way, please take 30 seconds to fill out a | |||||
ticket here: https://code.djangoproject.com/newticket | |||||
To get more help: | |||||
* Join the ``#django`` channel on ``irc.libera.chat``. Lots of helpful people | |||||
hang out there. See https://web.libera.chat if you're new to IRC. | |||||
* Join the django-users mailing list, or read the archives, at | |||||
https://groups.google.com/group/django-users. | |||||
To contribute to Django: | |||||
* Check out https://docs.djangoproject.com/en/dev/internals/contributing/ for | |||||
information about getting involved. | |||||
To run Django's test suite: | |||||
* Follow the instructions in the "Unit tests" section of | |||||
``docs/internals/contributing/writing-code/unit-tests.txt``, published online at | |||||
https://docs.djangoproject.com/en/dev/internals/contributing/writing-code/unit-tests/#running-the-unit-tests | |||||
Supporting the Development of Django | |||||
==================================== | |||||
Django's development depends on your contributions. | |||||
If you depend on Django, remember to support the Django Software Foundation: https://www.djangoproject.com/fundraising/ | |||||
Wheel-Version: 1.0 | |||||
Generator: bdist_wheel (0.37.0) | |||||
Root-Is-Purelib: true | |||||
Tag: py3-none-any | |||||
[console_scripts] | |||||
django-admin = django.core.management:execute_from_command_line | |||||
django |
# Copyright (C) AB Strakt | |||||
# See LICENSE for details. | |||||
""" | |||||
pyOpenSSL - A simple wrapper around the OpenSSL library | |||||
""" | |||||
from OpenSSL import SSL, crypto | |||||
from OpenSSL.version import ( | |||||
__author__, | |||||
__copyright__, | |||||
__email__, | |||||
__license__, | |||||
__summary__, | |||||
__title__, | |||||
__uri__, | |||||
__version__, | |||||
) | |||||
__all__ = [ | |||||
"SSL", | |||||
"crypto", | |||||
"__author__", | |||||
"__copyright__", | |||||
"__email__", | |||||
"__license__", | |||||
"__summary__", | |||||
"__title__", | |||||
"__uri__", | |||||
"__version__", | |||||
] |
import os | |||||
import sys | |||||
import warnings | |||||
from typing import Any, Callable, NoReturn, Type, Union | |||||
from cryptography.hazmat.bindings.openssl.binding import Binding | |||||
StrOrBytesPath = Union[str, bytes, os.PathLike] | |||||
binding = Binding() | |||||
ffi = binding.ffi | |||||
lib = binding.lib | |||||
# This is a special CFFI allocator that does not bother to zero its memory | |||||
# after allocation. This has vastly better performance on large allocations and | |||||
# so should be used whenever we don't need the memory zeroed out. | |||||
no_zero_allocator = ffi.new_allocator(should_clear_after_alloc=False) | |||||
def text(charp: Any) -> str: | |||||
""" | |||||
Get a native string type representing of the given CFFI ``char*`` object. | |||||
:param charp: A C-style string represented using CFFI. | |||||
:return: :class:`str` | |||||
""" | |||||
if not charp: | |||||
return "" | |||||
return ffi.string(charp).decode("utf-8") | |||||
def exception_from_error_queue(exception_type: Type[Exception]) -> NoReturn: | |||||
""" | |||||
Convert an OpenSSL library failure into a Python exception. | |||||
When a call to the native OpenSSL library fails, this is usually signalled | |||||
by the return value, and an error code is stored in an error queue | |||||
associated with the current thread. The err library provides functions to | |||||
obtain these error codes and textual error messages. | |||||
""" | |||||
errors = [] | |||||
while True: | |||||
error = lib.ERR_get_error() | |||||
if error == 0: | |||||
break | |||||
errors.append( | |||||
( | |||||
text(lib.ERR_lib_error_string(error)), | |||||
text(lib.ERR_func_error_string(error)), | |||||
text(lib.ERR_reason_error_string(error)), | |||||
) | |||||
) | |||||
raise exception_type(errors) | |||||
def make_assert(error: Type[Exception]) -> Callable[[bool], Any]: | |||||
""" | |||||
Create an assert function that uses :func:`exception_from_error_queue` to | |||||
raise an exception wrapped by *error*. | |||||
""" | |||||
def openssl_assert(ok: bool) -> None: | |||||
""" | |||||
If *ok* is not True, retrieve the error from OpenSSL and raise it. | |||||
""" | |||||
if ok is not True: | |||||
exception_from_error_queue(error) | |||||
return openssl_assert | |||||
def path_bytes(s: StrOrBytesPath) -> bytes: | |||||
""" | |||||
Convert a Python path to a :py:class:`bytes` for the path which can be | |||||
passed into an OpenSSL API accepting a filename. | |||||
:param s: A path (valid for os.fspath). | |||||
:return: An instance of :py:class:`bytes`. | |||||
""" | |||||
b = os.fspath(s) | |||||
if isinstance(b, str): | |||||
return b.encode(sys.getfilesystemencoding()) | |||||
else: | |||||
return b | |||||
def byte_string(s: str) -> bytes: | |||||
return s.encode("charmap") | |||||
# A marker object to observe whether some optional arguments are passed any | |||||
# value or not. | |||||
UNSPECIFIED = object() | |||||
_TEXT_WARNING = "str for {0} is no longer accepted, use bytes" | |||||
def text_to_bytes_and_warn(label: str, obj: Any) -> Any: | |||||
""" | |||||
If ``obj`` is text, emit a warning that it should be bytes instead and try | |||||
to convert it to bytes automatically. | |||||
:param str label: The name of the parameter from which ``obj`` was taken | |||||
(so a developer can easily find the source of the problem and correct | |||||
it). | |||||
:return: If ``obj`` is the text string type, a ``bytes`` object giving the | |||||
UTF-8 encoding of that text is returned. Otherwise, ``obj`` itself is | |||||
returned. | |||||
""" | |||||
if isinstance(obj, str): | |||||
warnings.warn( | |||||
_TEXT_WARNING.format(label), | |||||
category=DeprecationWarning, | |||||
stacklevel=3, | |||||
) | |||||
return obj.encode("utf-8") | |||||
return obj |
from __future__ import print_function | |||||
import ssl | |||||
import sys | |||||
import cffi | |||||
import cryptography | |||||
import OpenSSL.SSL | |||||
from . import version | |||||
_env_info = """\ | |||||
pyOpenSSL: {pyopenssl} | |||||
cryptography: {cryptography} | |||||
cffi: {cffi} | |||||
cryptography's compiled against OpenSSL: {crypto_openssl_compile} | |||||
cryptography's linked OpenSSL: {crypto_openssl_link} | |||||
Python's OpenSSL: {python_openssl} | |||||
Python executable: {python} | |||||
Python version: {python_version} | |||||
Platform: {platform} | |||||
sys.path: {sys_path}""".format( | |||||
pyopenssl=version.__version__, | |||||
crypto_openssl_compile=OpenSSL._util.ffi.string( | |||||
OpenSSL._util.lib.OPENSSL_VERSION_TEXT, | |||||
).decode("ascii"), | |||||
crypto_openssl_link=OpenSSL.SSL.SSLeay_version( | |||||
OpenSSL.SSL.SSLEAY_VERSION | |||||
).decode("ascii"), | |||||
python_openssl=getattr(ssl, "OPENSSL_VERSION", "n/a"), | |||||
cryptography=cryptography.__version__, | |||||
cffi=cffi.__version__, | |||||
python=sys.executable, | |||||
python_version=sys.version, | |||||
platform=sys.platform, | |||||
sys_path=sys.path, | |||||
) | |||||
if __name__ == "__main__": | |||||
print(_env_info) |
""" | |||||
PRNG management routines, thin wrappers. | |||||
""" | |||||
from OpenSSL._util import lib as _lib | |||||
def add(buffer: bytes, entropy: int) -> None: | |||||
""" | |||||
Mix bytes from *string* into the PRNG state. | |||||
The *entropy* argument is (the lower bound of) an estimate of how much | |||||
randomness is contained in *string*, measured in bytes. | |||||
For more information, see e.g. :rfc:`1750`. | |||||
This function is only relevant if you are forking Python processes and | |||||
need to reseed the CSPRNG after fork. | |||||
:param buffer: Buffer with random data. | |||||
:param entropy: The entropy (in bytes) measurement of the buffer. | |||||
:return: :obj:`None` | |||||
""" | |||||
if not isinstance(buffer, bytes): | |||||
raise TypeError("buffer must be a byte string") | |||||
if not isinstance(entropy, int): | |||||
raise TypeError("entropy must be an integer") | |||||
_lib.RAND_add(buffer, len(buffer), entropy) | |||||
def status() -> int: | |||||
""" | |||||
Check whether the PRNG has been seeded with enough data. | |||||
:return: 1 if the PRNG is seeded enough, 0 otherwise. | |||||
""" | |||||
return _lib.RAND_status() |
# Copyright (C) AB Strakt | |||||
# Copyright (C) Jean-Paul Calderone | |||||
# See LICENSE for details. | |||||
""" | |||||
pyOpenSSL - A simple wrapper around the OpenSSL library | |||||
""" | |||||
__all__ = [ | |||||
"__author__", | |||||
"__copyright__", | |||||
"__email__", | |||||
"__license__", | |||||
"__summary__", | |||||
"__title__", | |||||
"__uri__", | |||||
"__version__", | |||||
] | |||||
__version__ = "23.2.0" | |||||
__title__ = "pyOpenSSL" | |||||
__uri__ = "https://pyopenssl.org/" | |||||
__summary__ = "Python wrapper module around the OpenSSL library" | |||||
__author__ = "The pyOpenSSL developers" | |||||
__email__ = "cryptography-dev@python.org" | |||||
__license__ = "Apache License, Version 2.0" | |||||
__copyright__ = "Copyright 2001-2023 {0}".format(__author__) |
# | |||||
# The Python Imaging Library | |||||
# $Id$ | |||||
# | |||||
# bitmap distribution font (bdf) file parser | |||||
# | |||||
# history: | |||||
# 1996-05-16 fl created (as bdf2pil) | |||||
# 1997-08-25 fl converted to FontFile driver | |||||
# 2001-05-25 fl removed bogus __init__ call | |||||
# 2002-11-20 fl robustification (from Kevin Cazabon, Dmitry Vasiliev) | |||||
# 2003-04-22 fl more robustification (from Graham Dumpleton) | |||||
# | |||||
# Copyright (c) 1997-2003 by Secret Labs AB. | |||||
# Copyright (c) 1997-2003 by Fredrik Lundh. | |||||
# | |||||
# See the README file for information on usage and redistribution. | |||||
# | |||||
""" | |||||
Parse X Bitmap Distribution Format (BDF) | |||||
""" | |||||
from . import FontFile, Image | |||||
bdf_slant = { | |||||
"R": "Roman", | |||||
"I": "Italic", | |||||
"O": "Oblique", | |||||
"RI": "Reverse Italic", | |||||
"RO": "Reverse Oblique", | |||||
"OT": "Other", | |||||
} | |||||
bdf_spacing = {"P": "Proportional", "M": "Monospaced", "C": "Cell"} | |||||
def bdf_char(f): | |||||
# skip to STARTCHAR | |||||
while True: | |||||
s = f.readline() | |||||
if not s: | |||||
return None | |||||
if s[:9] == b"STARTCHAR": | |||||
break | |||||
id = s[9:].strip().decode("ascii") | |||||
# load symbol properties | |||||
props = {} | |||||
while True: | |||||
s = f.readline() | |||||
if not s or s[:6] == b"BITMAP": | |||||
break | |||||
i = s.find(b" ") | |||||
props[s[:i].decode("ascii")] = s[i + 1 : -1].decode("ascii") | |||||
# load bitmap | |||||
bitmap = [] | |||||
while True: | |||||
s = f.readline() | |||||
if not s or s[:7] == b"ENDCHAR": | |||||
break | |||||
bitmap.append(s[:-1]) | |||||
bitmap = b"".join(bitmap) | |||||
# The word BBX | |||||
# followed by the width in x (BBw), height in y (BBh), | |||||
# and x and y displacement (BBxoff0, BByoff0) | |||||
# of the lower left corner from the origin of the character. | |||||
width, height, x_disp, y_disp = [int(p) for p in props["BBX"].split()] | |||||
# The word DWIDTH | |||||
# followed by the width in x and y of the character in device pixels. | |||||
dwx, dwy = [int(p) for p in props["DWIDTH"].split()] | |||||
bbox = ( | |||||
(dwx, dwy), | |||||
(x_disp, -y_disp - height, width + x_disp, -y_disp), | |||||
(0, 0, width, height), | |||||
) | |||||
try: | |||||
im = Image.frombytes("1", (width, height), bitmap, "hex", "1") | |||||
except ValueError: | |||||
# deal with zero-width characters | |||||
im = Image.new("1", (width, height)) | |||||
return id, int(props["ENCODING"]), bbox, im | |||||
class BdfFontFile(FontFile.FontFile): | |||||
"""Font file plugin for the X11 BDF format.""" | |||||
def __init__(self, fp): | |||||
super().__init__() | |||||
s = fp.readline() | |||||
if s[:13] != b"STARTFONT 2.1": | |||||
msg = "not a valid BDF file" | |||||
raise SyntaxError(msg) | |||||
props = {} | |||||
comments = [] | |||||
while True: | |||||
s = fp.readline() | |||||
if not s or s[:13] == b"ENDPROPERTIES": | |||||
break | |||||
i = s.find(b" ") | |||||
props[s[:i].decode("ascii")] = s[i + 1 : -1].decode("ascii") | |||||
if s[:i] in [b"COMMENT", b"COPYRIGHT"]: | |||||
if s.find(b"LogicalFontDescription") < 0: | |||||
comments.append(s[i + 1 : -1].decode("ascii")) | |||||
while True: | |||||
c = bdf_char(fp) | |||||
if not c: | |||||
break | |||||
id, ch, (xy, dst, src), im = c | |||||
if 0 <= ch < len(self.glyph): | |||||
self.glyph[ch] = xy, dst, src, im |
""" | |||||
Blizzard Mipmap Format (.blp) | |||||
Jerome Leclanche <jerome@leclan.ch> | |||||
The contents of this file are hereby released in the public domain (CC0) | |||||
Full text of the CC0 license: | |||||
https://creativecommons.org/publicdomain/zero/1.0/ | |||||
BLP1 files, used mostly in Warcraft III, are not fully supported. | |||||
All types of BLP2 files used in World of Warcraft are supported. | |||||
The BLP file structure consists of a header, up to 16 mipmaps of the | |||||
texture | |||||
Texture sizes must be powers of two, though the two dimensions do | |||||
not have to be equal; 512x256 is valid, but 512x200 is not. | |||||
The first mipmap (mipmap #0) is the full size image; each subsequent | |||||
mipmap halves both dimensions. The final mipmap should be 1x1. | |||||
BLP files come in many different flavours: | |||||
* JPEG-compressed (type == 0) - only supported for BLP1. | |||||
* RAW images (type == 1, encoding == 1). Each mipmap is stored as an | |||||
array of 8-bit values, one per pixel, left to right, top to bottom. | |||||
Each value is an index to the palette. | |||||
* DXT-compressed (type == 1, encoding == 2): | |||||
- DXT1 compression is used if alpha_encoding == 0. | |||||
- An additional alpha bit is used if alpha_depth == 1. | |||||
- DXT3 compression is used if alpha_encoding == 1. | |||||
- DXT5 compression is used if alpha_encoding == 7. | |||||
""" | |||||
import os | |||||
import struct | |||||
from enum import IntEnum | |||||
from io import BytesIO | |||||
from . import Image, ImageFile | |||||
from ._deprecate import deprecate | |||||
class Format(IntEnum): | |||||
JPEG = 0 | |||||
class Encoding(IntEnum): | |||||
UNCOMPRESSED = 1 | |||||
DXT = 2 | |||||
UNCOMPRESSED_RAW_BGRA = 3 | |||||
class AlphaEncoding(IntEnum): | |||||
DXT1 = 0 | |||||
DXT3 = 1 | |||||
DXT5 = 7 | |||||
def __getattr__(name): | |||||
for enum, prefix in { | |||||
Format: "BLP_FORMAT_", | |||||
Encoding: "BLP_ENCODING_", | |||||
AlphaEncoding: "BLP_ALPHA_ENCODING_", | |||||
}.items(): | |||||
if name.startswith(prefix): | |||||
name = name[len(prefix) :] | |||||
if name in enum.__members__: | |||||
deprecate(f"{prefix}{name}", 10, f"{enum.__name__}.{name}") | |||||
return enum[name] | |||||
msg = f"module '{__name__}' has no attribute '{name}'" | |||||
raise AttributeError(msg) | |||||
def unpack_565(i): | |||||
return ((i >> 11) & 0x1F) << 3, ((i >> 5) & 0x3F) << 2, (i & 0x1F) << 3 | |||||
def decode_dxt1(data, alpha=False): | |||||
""" | |||||
input: one "row" of data (i.e. will produce 4*width pixels) | |||||
""" | |||||
blocks = len(data) // 8 # number of blocks in row | |||||
ret = (bytearray(), bytearray(), bytearray(), bytearray()) | |||||
for block in range(blocks): | |||||
# Decode next 8-byte block. | |||||
idx = block * 8 | |||||
color0, color1, bits = struct.unpack_from("<HHI", data, idx) | |||||
r0, g0, b0 = unpack_565(color0) | |||||
r1, g1, b1 = unpack_565(color1) | |||||
# Decode this block into 4x4 pixels | |||||
# Accumulate the results onto our 4 row accumulators | |||||
for j in range(4): | |||||
for i in range(4): | |||||
# get next control op and generate a pixel | |||||
control = bits & 3 | |||||
bits = bits >> 2 | |||||
a = 0xFF | |||||
if control == 0: | |||||
r, g, b = r0, g0, b0 | |||||
elif control == 1: | |||||
r, g, b = r1, g1, b1 | |||||
elif control == 2: | |||||
if color0 > color1: | |||||
r = (2 * r0 + r1) // 3 | |||||
g = (2 * g0 + g1) // 3 | |||||
b = (2 * b0 + b1) // 3 | |||||
else: | |||||
r = (r0 + r1) // 2 | |||||
g = (g0 + g1) // 2 | |||||
b = (b0 + b1) // 2 | |||||
elif control == 3: | |||||
if color0 > color1: | |||||
r = (2 * r1 + r0) // 3 | |||||
g = (2 * g1 + g0) // 3 | |||||
b = (2 * b1 + b0) // 3 | |||||
else: | |||||
r, g, b, a = 0, 0, 0, 0 | |||||
if alpha: | |||||
ret[j].extend([r, g, b, a]) | |||||
else: | |||||
ret[j].extend([r, g, b]) | |||||
return ret | |||||
def decode_dxt3(data): | |||||
""" | |||||
input: one "row" of data (i.e. will produce 4*width pixels) | |||||
""" | |||||
blocks = len(data) // 16 # number of blocks in row | |||||
ret = (bytearray(), bytearray(), bytearray(), bytearray()) | |||||
for block in range(blocks): | |||||
idx = block * 16 | |||||
block = data[idx : idx + 16] | |||||
# Decode next 16-byte block. | |||||
bits = struct.unpack_from("<8B", block) | |||||
color0, color1 = struct.unpack_from("<HH", block, 8) | |||||
(code,) = struct.unpack_from("<I", block, 12) | |||||
r0, g0, b0 = unpack_565(color0) | |||||
r1, g1, b1 = unpack_565(color1) | |||||
for j in range(4): | |||||
high = False # Do we want the higher bits? | |||||
for i in range(4): | |||||
alphacode_index = (4 * j + i) // 2 | |||||
a = bits[alphacode_index] | |||||
if high: | |||||
high = False | |||||
a >>= 4 | |||||
else: | |||||
high = True | |||||
a &= 0xF | |||||
a *= 17 # We get a value between 0 and 15 | |||||
color_code = (code >> 2 * (4 * j + i)) & 0x03 | |||||
if color_code == 0: | |||||
r, g, b = r0, g0, b0 | |||||
elif color_code == 1: | |||||
r, g, b = r1, g1, b1 | |||||
elif color_code == 2: | |||||
r = (2 * r0 + r1) // 3 | |||||
g = (2 * g0 + g1) // 3 | |||||
b = (2 * b0 + b1) // 3 | |||||
elif color_code == 3: | |||||
r = (2 * r1 + r0) // 3 | |||||
g = (2 * g1 + g0) // 3 | |||||
b = (2 * b1 + b0) // 3 | |||||
ret[j].extend([r, g, b, a]) | |||||
return ret | |||||
def decode_dxt5(data): | |||||
""" | |||||
input: one "row" of data (i.e. will produce 4 * width pixels) | |||||
""" | |||||
blocks = len(data) // 16 # number of blocks in row | |||||
ret = (bytearray(), bytearray(), bytearray(), bytearray()) | |||||
for block in range(blocks): | |||||
idx = block * 16 | |||||
block = data[idx : idx + 16] | |||||
# Decode next 16-byte block. | |||||
a0, a1 = struct.unpack_from("<BB", block) | |||||
bits = struct.unpack_from("<6B", block, 2) | |||||
alphacode1 = bits[2] | (bits[3] << 8) | (bits[4] << 16) | (bits[5] << 24) | |||||
alphacode2 = bits[0] | (bits[1] << 8) | |||||
color0, color1 = struct.unpack_from("<HH", block, 8) | |||||
(code,) = struct.unpack_from("<I", block, 12) | |||||
r0, g0, b0 = unpack_565(color0) | |||||
r1, g1, b1 = unpack_565(color1) | |||||
for j in range(4): | |||||
for i in range(4): | |||||
# get next control op and generate a pixel | |||||
alphacode_index = 3 * (4 * j + i) | |||||
if alphacode_index <= 12: | |||||
alphacode = (alphacode2 >> alphacode_index) & 0x07 | |||||
elif alphacode_index == 15: | |||||
alphacode = (alphacode2 >> 15) | ((alphacode1 << 1) & 0x06) | |||||
else: # alphacode_index >= 18 and alphacode_index <= 45 | |||||
alphacode = (alphacode1 >> (alphacode_index - 16)) & 0x07 | |||||
if alphacode == 0: | |||||
a = a0 | |||||
elif alphacode == 1: | |||||
a = a1 | |||||
elif a0 > a1: | |||||
a = ((8 - alphacode) * a0 + (alphacode - 1) * a1) // 7 | |||||
elif alphacode == 6: | |||||
a = 0 | |||||
elif alphacode == 7: | |||||
a = 255 | |||||
else: | |||||
a = ((6 - alphacode) * a0 + (alphacode - 1) * a1) // 5 | |||||
color_code = (code >> 2 * (4 * j + i)) & 0x03 | |||||
if color_code == 0: | |||||
r, g, b = r0, g0, b0 | |||||
elif color_code == 1: | |||||
r, g, b = r1, g1, b1 | |||||
elif color_code == 2: | |||||
r = (2 * r0 + r1) // 3 | |||||
g = (2 * g0 + g1) // 3 | |||||
b = (2 * b0 + b1) // 3 | |||||
elif color_code == 3: | |||||
r = (2 * r1 + r0) // 3 | |||||
g = (2 * g1 + g0) // 3 | |||||
b = (2 * b1 + b0) // 3 | |||||
ret[j].extend([r, g, b, a]) | |||||
return ret | |||||
class BLPFormatError(NotImplementedError): | |||||
pass | |||||
def _accept(prefix): | |||||
return prefix[:4] in (b"BLP1", b"BLP2") | |||||
class BlpImageFile(ImageFile.ImageFile): | |||||
""" | |||||
Blizzard Mipmap Format | |||||
""" | |||||
format = "BLP" | |||||
format_description = "Blizzard Mipmap Format" | |||||
def _open(self): | |||||
self.magic = self.fp.read(4) | |||||
self.fp.seek(5, os.SEEK_CUR) | |||||
(self._blp_alpha_depth,) = struct.unpack("<b", self.fp.read(1)) | |||||
self.fp.seek(2, os.SEEK_CUR) | |||||
self._size = struct.unpack("<II", self.fp.read(8)) | |||||
if self.magic in (b"BLP1", b"BLP2"): | |||||
decoder = self.magic.decode() | |||||
else: | |||||
msg = f"Bad BLP magic {repr(self.magic)}" | |||||
raise BLPFormatError(msg) | |||||
self.mode = "RGBA" if self._blp_alpha_depth else "RGB" | |||||
self.tile = [(decoder, (0, 0) + self.size, 0, (self.mode, 0, 1))] | |||||
class _BLPBaseDecoder(ImageFile.PyDecoder): | |||||
_pulls_fd = True | |||||
def decode(self, buffer): | |||||
try: | |||||
self._read_blp_header() | |||||
self._load() | |||||
except struct.error as e: | |||||
msg = "Truncated BLP file" | |||||
raise OSError(msg) from e | |||||
return -1, 0 | |||||
def _read_blp_header(self): | |||||
self.fd.seek(4) | |||||
(self._blp_compression,) = struct.unpack("<i", self._safe_read(4)) | |||||
(self._blp_encoding,) = struct.unpack("<b", self._safe_read(1)) | |||||
(self._blp_alpha_depth,) = struct.unpack("<b", self._safe_read(1)) | |||||
(self._blp_alpha_encoding,) = struct.unpack("<b", self._safe_read(1)) | |||||
self.fd.seek(1, os.SEEK_CUR) # mips | |||||
self.size = struct.unpack("<II", self._safe_read(8)) | |||||
if isinstance(self, BLP1Decoder): | |||||
# Only present for BLP1 | |||||
(self._blp_encoding,) = struct.unpack("<i", self._safe_read(4)) | |||||
self.fd.seek(4, os.SEEK_CUR) # subtype | |||||
self._blp_offsets = struct.unpack("<16I", self._safe_read(16 * 4)) | |||||
self._blp_lengths = struct.unpack("<16I", self._safe_read(16 * 4)) | |||||
def _safe_read(self, length): | |||||
return ImageFile._safe_read(self.fd, length) | |||||
def _read_palette(self): | |||||
ret = [] | |||||
for i in range(256): | |||||
try: | |||||
b, g, r, a = struct.unpack("<4B", self._safe_read(4)) | |||||
except struct.error: | |||||
break | |||||
ret.append((b, g, r, a)) | |||||
return ret | |||||
def _read_bgra(self, palette): | |||||
data = bytearray() | |||||
_data = BytesIO(self._safe_read(self._blp_lengths[0])) | |||||
while True: | |||||
try: | |||||
(offset,) = struct.unpack("<B", _data.read(1)) | |||||
except struct.error: | |||||
break | |||||
b, g, r, a = palette[offset] | |||||
d = (r, g, b) | |||||
if self._blp_alpha_depth: | |||||
d += (a,) | |||||
data.extend(d) | |||||
return data | |||||
class BLP1Decoder(_BLPBaseDecoder): | |||||
def _load(self): | |||||
if self._blp_compression == Format.JPEG: | |||||
self._decode_jpeg_stream() | |||||
elif self._blp_compression == 1: | |||||
if self._blp_encoding in (4, 5): | |||||
palette = self._read_palette() | |||||
data = self._read_bgra(palette) | |||||
self.set_as_raw(bytes(data)) | |||||
else: | |||||
msg = f"Unsupported BLP encoding {repr(self._blp_encoding)}" | |||||
raise BLPFormatError(msg) | |||||
else: | |||||
msg = f"Unsupported BLP compression {repr(self._blp_encoding)}" | |||||
raise BLPFormatError(msg) | |||||
def _decode_jpeg_stream(self): | |||||
from .JpegImagePlugin import JpegImageFile | |||||
(jpeg_header_size,) = struct.unpack("<I", self._safe_read(4)) | |||||
jpeg_header = self._safe_read(jpeg_header_size) | |||||
self._safe_read(self._blp_offsets[0] - self.fd.tell()) # What IS this? | |||||
data = self._safe_read(self._blp_lengths[0]) | |||||
data = jpeg_header + data | |||||
data = BytesIO(data) | |||||
image = JpegImageFile(data) | |||||
Image._decompression_bomb_check(image.size) | |||||
if image.mode == "CMYK": | |||||
decoder_name, extents, offset, args = image.tile[0] | |||||
image.tile = [(decoder_name, extents, offset, (args[0], "CMYK"))] | |||||
r, g, b = image.convert("RGB").split() | |||||
image = Image.merge("RGB", (b, g, r)) | |||||
self.set_as_raw(image.tobytes()) | |||||
class BLP2Decoder(_BLPBaseDecoder): | |||||
def _load(self): | |||||
palette = self._read_palette() | |||||
self.fd.seek(self._blp_offsets[0]) | |||||
if self._blp_compression == 1: | |||||
# Uncompressed or DirectX compression | |||||
if self._blp_encoding == Encoding.UNCOMPRESSED: | |||||
data = self._read_bgra(palette) | |||||
elif self._blp_encoding == Encoding.DXT: | |||||
data = bytearray() | |||||
if self._blp_alpha_encoding == AlphaEncoding.DXT1: | |||||
linesize = (self.size[0] + 3) // 4 * 8 | |||||
for yb in range((self.size[1] + 3) // 4): | |||||
for d in decode_dxt1( | |||||
self._safe_read(linesize), alpha=bool(self._blp_alpha_depth) | |||||
): | |||||
data += d | |||||
elif self._blp_alpha_encoding == AlphaEncoding.DXT3: | |||||
linesize = (self.size[0] + 3) // 4 * 16 | |||||
for yb in range((self.size[1] + 3) // 4): | |||||
for d in decode_dxt3(self._safe_read(linesize)): | |||||
data += d | |||||
elif self._blp_alpha_encoding == AlphaEncoding.DXT5: | |||||
linesize = (self.size[0] + 3) // 4 * 16 | |||||
for yb in range((self.size[1] + 3) // 4): | |||||
for d in decode_dxt5(self._safe_read(linesize)): | |||||
data += d | |||||
else: | |||||
msg = f"Unsupported alpha encoding {repr(self._blp_alpha_encoding)}" | |||||
raise BLPFormatError(msg) | |||||
else: | |||||
msg = f"Unknown BLP encoding {repr(self._blp_encoding)}" | |||||
raise BLPFormatError(msg) | |||||
else: | |||||
msg = f"Unknown BLP compression {repr(self._blp_compression)}" | |||||
raise BLPFormatError(msg) | |||||
self.set_as_raw(bytes(data)) | |||||
class BLPEncoder(ImageFile.PyEncoder): | |||||
_pushes_fd = True | |||||
def _write_palette(self): | |||||
data = b"" | |||||
palette = self.im.getpalette("RGBA", "RGBA") | |||||
for i in range(256): | |||||
r, g, b, a = palette[i * 4 : (i + 1) * 4] | |||||
data += struct.pack("<4B", b, g, r, a) | |||||
return data | |||||
def encode(self, bufsize): | |||||
palette_data = self._write_palette() | |||||
offset = 20 + 16 * 4 * 2 + len(palette_data) | |||||
data = struct.pack("<16I", offset, *((0,) * 15)) | |||||
w, h = self.im.size | |||||
data += struct.pack("<16I", w * h, *((0,) * 15)) | |||||
data += palette_data | |||||
for y in range(h): | |||||
for x in range(w): | |||||
data += struct.pack("<B", self.im.getpixel((x, y))) | |||||
return len(data), 0, data | |||||
def _save(im, fp, filename, save_all=False): | |||||
if im.mode != "P": | |||||
msg = "Unsupported BLP image mode" | |||||
raise ValueError(msg) | |||||
magic = b"BLP1" if im.encoderinfo.get("blp_version") == "BLP1" else b"BLP2" | |||||
fp.write(magic) | |||||
fp.write(struct.pack("<i", 1)) # Uncompressed or DirectX compression | |||||
fp.write(struct.pack("<b", Encoding.UNCOMPRESSED)) | |||||
fp.write(struct.pack("<b", 1 if im.palette.mode == "RGBA" else 0)) | |||||
fp.write(struct.pack("<b", 0)) # alpha encoding | |||||
fp.write(struct.pack("<b", 0)) # mips | |||||
fp.write(struct.pack("<II", *im.size)) | |||||
if magic == b"BLP1": | |||||
fp.write(struct.pack("<i", 5)) | |||||
fp.write(struct.pack("<i", 0)) | |||||
ImageFile._save(im, fp, [("BLP", (0, 0) + im.size, 0, im.mode)]) | |||||
Image.register_open(BlpImageFile.format, BlpImageFile, _accept) | |||||
Image.register_extension(BlpImageFile.format, ".blp") | |||||
Image.register_decoder("BLP1", BLP1Decoder) | |||||
Image.register_decoder("BLP2", BLP2Decoder) | |||||
Image.register_save(BlpImageFile.format, _save) | |||||
Image.register_encoder("BLP", BLPEncoder) |
# | |||||
# The Python Imaging Library. | |||||
# $Id$ | |||||
# | |||||
# BMP file handler | |||||
# | |||||
# Windows (and OS/2) native bitmap storage format. | |||||
# | |||||
# history: | |||||
# 1995-09-01 fl Created | |||||
# 1996-04-30 fl Added save | |||||
# 1997-08-27 fl Fixed save of 1-bit images | |||||
# 1998-03-06 fl Load P images as L where possible | |||||
# 1998-07-03 fl Load P images as 1 where possible | |||||
# 1998-12-29 fl Handle small palettes | |||||
# 2002-12-30 fl Fixed load of 1-bit palette images | |||||
# 2003-04-21 fl Fixed load of 1-bit monochrome images | |||||
# 2003-04-23 fl Added limited support for BI_BITFIELDS compression | |||||
# | |||||
# Copyright (c) 1997-2003 by Secret Labs AB | |||||
# Copyright (c) 1995-2003 by Fredrik Lundh | |||||
# | |||||
# See the README file for information on usage and redistribution. | |||||
# | |||||
import os | |||||
from . import Image, ImageFile, ImagePalette | |||||
from ._binary import i16le as i16 | |||||
from ._binary import i32le as i32 | |||||
from ._binary import o8 | |||||
from ._binary import o16le as o16 | |||||
from ._binary import o32le as o32 | |||||
# | |||||
# -------------------------------------------------------------------- | |||||
# Read BMP file | |||||
BIT2MODE = { | |||||
# bits => mode, rawmode | |||||
1: ("P", "P;1"), | |||||
4: ("P", "P;4"), | |||||
8: ("P", "P"), | |||||
16: ("RGB", "BGR;15"), | |||||
24: ("RGB", "BGR"), | |||||
32: ("RGB", "BGRX"), | |||||
} | |||||
def _accept(prefix): | |||||
return prefix[:2] == b"BM" | |||||
def _dib_accept(prefix): | |||||
return i32(prefix) in [12, 40, 64, 108, 124] | |||||
# ============================================================================= | |||||
# Image plugin for the Windows BMP format. | |||||
# ============================================================================= | |||||
class BmpImageFile(ImageFile.ImageFile): | |||||
"""Image plugin for the Windows Bitmap format (BMP)""" | |||||
# ------------------------------------------------------------- Description | |||||
format_description = "Windows Bitmap" | |||||
format = "BMP" | |||||
# -------------------------------------------------- BMP Compression values | |||||
COMPRESSIONS = {"RAW": 0, "RLE8": 1, "RLE4": 2, "BITFIELDS": 3, "JPEG": 4, "PNG": 5} | |||||
for k, v in COMPRESSIONS.items(): | |||||
vars()[k] = v | |||||
def _bitmap(self, header=0, offset=0): | |||||
"""Read relevant info about the BMP""" | |||||
read, seek = self.fp.read, self.fp.seek | |||||
if header: | |||||
seek(header) | |||||
# read bmp header size @offset 14 (this is part of the header size) | |||||
file_info = {"header_size": i32(read(4)), "direction": -1} | |||||
# -------------------- If requested, read header at a specific position | |||||
# read the rest of the bmp header, without its size | |||||
header_data = ImageFile._safe_read(self.fp, file_info["header_size"] - 4) | |||||
# -------------------------------------------------- IBM OS/2 Bitmap v1 | |||||
# ----- This format has different offsets because of width/height types | |||||
if file_info["header_size"] == 12: | |||||
file_info["width"] = i16(header_data, 0) | |||||
file_info["height"] = i16(header_data, 2) | |||||
file_info["planes"] = i16(header_data, 4) | |||||
file_info["bits"] = i16(header_data, 6) | |||||
file_info["compression"] = self.RAW | |||||
file_info["palette_padding"] = 3 | |||||
# --------------------------------------------- Windows Bitmap v2 to v5 | |||||
# v3, OS/2 v2, v4, v5 | |||||
elif file_info["header_size"] in (40, 64, 108, 124): | |||||
file_info["y_flip"] = header_data[7] == 0xFF | |||||
file_info["direction"] = 1 if file_info["y_flip"] else -1 | |||||
file_info["width"] = i32(header_data, 0) | |||||
file_info["height"] = ( | |||||
i32(header_data, 4) | |||||
if not file_info["y_flip"] | |||||
else 2**32 - i32(header_data, 4) | |||||
) | |||||
file_info["planes"] = i16(header_data, 8) | |||||
file_info["bits"] = i16(header_data, 10) | |||||
file_info["compression"] = i32(header_data, 12) | |||||
# byte size of pixel data | |||||
file_info["data_size"] = i32(header_data, 16) | |||||
file_info["pixels_per_meter"] = ( | |||||
i32(header_data, 20), | |||||
i32(header_data, 24), | |||||
) | |||||
file_info["colors"] = i32(header_data, 28) | |||||
file_info["palette_padding"] = 4 | |||||
self.info["dpi"] = tuple(x / 39.3701 for x in file_info["pixels_per_meter"]) | |||||
if file_info["compression"] == self.BITFIELDS: | |||||
if len(header_data) >= 52: | |||||
for idx, mask in enumerate( | |||||
["r_mask", "g_mask", "b_mask", "a_mask"] | |||||
): | |||||
file_info[mask] = i32(header_data, 36 + idx * 4) | |||||
else: | |||||
# 40 byte headers only have the three components in the | |||||
# bitfields masks, ref: | |||||
# https://msdn.microsoft.com/en-us/library/windows/desktop/dd183376(v=vs.85).aspx | |||||
# See also | |||||
# https://github.com/python-pillow/Pillow/issues/1293 | |||||
# There is a 4th component in the RGBQuad, in the alpha | |||||
# location, but it is listed as a reserved component, | |||||
# and it is not generally an alpha channel | |||||
file_info["a_mask"] = 0x0 | |||||
for mask in ["r_mask", "g_mask", "b_mask"]: | |||||
file_info[mask] = i32(read(4)) | |||||
file_info["rgb_mask"] = ( | |||||
file_info["r_mask"], | |||||
file_info["g_mask"], | |||||
file_info["b_mask"], | |||||
) | |||||
file_info["rgba_mask"] = ( | |||||
file_info["r_mask"], | |||||
file_info["g_mask"], | |||||
file_info["b_mask"], | |||||
file_info["a_mask"], | |||||
) | |||||
else: | |||||
msg = f"Unsupported BMP header type ({file_info['header_size']})" | |||||
raise OSError(msg) | |||||
# ------------------ Special case : header is reported 40, which | |||||
# ---------------------- is shorter than real size for bpp >= 16 | |||||
self._size = file_info["width"], file_info["height"] | |||||
# ------- If color count was not found in the header, compute from bits | |||||
file_info["colors"] = ( | |||||
file_info["colors"] | |||||
if file_info.get("colors", 0) | |||||
else (1 << file_info["bits"]) | |||||
) | |||||
if offset == 14 + file_info["header_size"] and file_info["bits"] <= 8: | |||||
offset += 4 * file_info["colors"] | |||||
# ---------------------- Check bit depth for unusual unsupported values | |||||
self.mode, raw_mode = BIT2MODE.get(file_info["bits"], (None, None)) | |||||
if self.mode is None: | |||||
msg = f"Unsupported BMP pixel depth ({file_info['bits']})" | |||||
raise OSError(msg) | |||||
# ---------------- Process BMP with Bitfields compression (not palette) | |||||
decoder_name = "raw" | |||||
if file_info["compression"] == self.BITFIELDS: | |||||
SUPPORTED = { | |||||
32: [ | |||||
(0xFF0000, 0xFF00, 0xFF, 0x0), | |||||
(0xFF000000, 0xFF0000, 0xFF00, 0x0), | |||||
(0xFF000000, 0xFF0000, 0xFF00, 0xFF), | |||||
(0xFF, 0xFF00, 0xFF0000, 0xFF000000), | |||||
(0xFF0000, 0xFF00, 0xFF, 0xFF000000), | |||||
(0x0, 0x0, 0x0, 0x0), | |||||
], | |||||
24: [(0xFF0000, 0xFF00, 0xFF)], | |||||
16: [(0xF800, 0x7E0, 0x1F), (0x7C00, 0x3E0, 0x1F)], | |||||
} | |||||
MASK_MODES = { | |||||
(32, (0xFF0000, 0xFF00, 0xFF, 0x0)): "BGRX", | |||||
(32, (0xFF000000, 0xFF0000, 0xFF00, 0x0)): "XBGR", | |||||
(32, (0xFF000000, 0xFF0000, 0xFF00, 0xFF)): "ABGR", | |||||
(32, (0xFF, 0xFF00, 0xFF0000, 0xFF000000)): "RGBA", | |||||
(32, (0xFF0000, 0xFF00, 0xFF, 0xFF000000)): "BGRA", | |||||
(32, (0x0, 0x0, 0x0, 0x0)): "BGRA", | |||||
(24, (0xFF0000, 0xFF00, 0xFF)): "BGR", | |||||
(16, (0xF800, 0x7E0, 0x1F)): "BGR;16", | |||||
(16, (0x7C00, 0x3E0, 0x1F)): "BGR;15", | |||||
} | |||||
if file_info["bits"] in SUPPORTED: | |||||
if ( | |||||
file_info["bits"] == 32 | |||||
and file_info["rgba_mask"] in SUPPORTED[file_info["bits"]] | |||||
): | |||||
raw_mode = MASK_MODES[(file_info["bits"], file_info["rgba_mask"])] | |||||
self.mode = "RGBA" if "A" in raw_mode else self.mode | |||||
elif ( | |||||
file_info["bits"] in (24, 16) | |||||
and file_info["rgb_mask"] in SUPPORTED[file_info["bits"]] | |||||
): | |||||
raw_mode = MASK_MODES[(file_info["bits"], file_info["rgb_mask"])] | |||||
else: | |||||
msg = "Unsupported BMP bitfields layout" | |||||
raise OSError(msg) | |||||
else: | |||||
msg = "Unsupported BMP bitfields layout" | |||||
raise OSError(msg) | |||||
elif file_info["compression"] == self.RAW: | |||||
if file_info["bits"] == 32 and header == 22: # 32-bit .cur offset | |||||
raw_mode, self.mode = "BGRA", "RGBA" | |||||
elif file_info["compression"] in (self.RLE8, self.RLE4): | |||||
decoder_name = "bmp_rle" | |||||
else: | |||||
msg = f"Unsupported BMP compression ({file_info['compression']})" | |||||
raise OSError(msg) | |||||
# --------------- Once the header is processed, process the palette/LUT | |||||
if self.mode == "P": # Paletted for 1, 4 and 8 bit images | |||||
# ---------------------------------------------------- 1-bit images | |||||
if not (0 < file_info["colors"] <= 65536): | |||||
msg = f"Unsupported BMP Palette size ({file_info['colors']})" | |||||
raise OSError(msg) | |||||
else: | |||||
padding = file_info["palette_padding"] | |||||
palette = read(padding * file_info["colors"]) | |||||
greyscale = True | |||||
indices = ( | |||||
(0, 255) | |||||
if file_info["colors"] == 2 | |||||
else list(range(file_info["colors"])) | |||||
) | |||||
# ----------------- Check if greyscale and ignore palette if so | |||||
for ind, val in enumerate(indices): | |||||
rgb = palette[ind * padding : ind * padding + 3] | |||||
if rgb != o8(val) * 3: | |||||
greyscale = False | |||||
# ------- If all colors are grey, white or black, ditch palette | |||||
if greyscale: | |||||
self.mode = "1" if file_info["colors"] == 2 else "L" | |||||
raw_mode = self.mode | |||||
else: | |||||
self.mode = "P" | |||||
self.palette = ImagePalette.raw( | |||||
"BGRX" if padding == 4 else "BGR", palette | |||||
) | |||||
# ---------------------------- Finally set the tile data for the plugin | |||||
self.info["compression"] = file_info["compression"] | |||||
args = [raw_mode] | |||||
if decoder_name == "bmp_rle": | |||||
args.append(file_info["compression"] == self.RLE4) | |||||
else: | |||||
args.append(((file_info["width"] * file_info["bits"] + 31) >> 3) & (~3)) | |||||
args.append(file_info["direction"]) | |||||
self.tile = [ | |||||
( | |||||
decoder_name, | |||||
(0, 0, file_info["width"], file_info["height"]), | |||||
offset or self.fp.tell(), | |||||
tuple(args), | |||||
) | |||||
] | |||||
def _open(self): | |||||
"""Open file, check magic number and read header""" | |||||
# read 14 bytes: magic number, filesize, reserved, header final offset | |||||
head_data = self.fp.read(14) | |||||
# choke if the file does not have the required magic bytes | |||||
if not _accept(head_data): | |||||
msg = "Not a BMP file" | |||||
raise SyntaxError(msg) | |||||
# read the start position of the BMP image data (u32) | |||||
offset = i32(head_data, 10) | |||||
# load bitmap information (offset=raster info) | |||||
self._bitmap(offset=offset) | |||||
class BmpRleDecoder(ImageFile.PyDecoder): | |||||
_pulls_fd = True | |||||
def decode(self, buffer): | |||||
rle4 = self.args[1] | |||||
data = bytearray() | |||||
x = 0 | |||||
while len(data) < self.state.xsize * self.state.ysize: | |||||
pixels = self.fd.read(1) | |||||
byte = self.fd.read(1) | |||||
if not pixels or not byte: | |||||
break | |||||
num_pixels = pixels[0] | |||||
if num_pixels: | |||||
# encoded mode | |||||
if x + num_pixels > self.state.xsize: | |||||
# Too much data for row | |||||
num_pixels = max(0, self.state.xsize - x) | |||||
if rle4: | |||||
first_pixel = o8(byte[0] >> 4) | |||||
second_pixel = o8(byte[0] & 0x0F) | |||||
for index in range(num_pixels): | |||||
if index % 2 == 0: | |||||
data += first_pixel | |||||
else: | |||||
data += second_pixel | |||||
else: | |||||
data += byte * num_pixels | |||||
x += num_pixels | |||||
else: | |||||
if byte[0] == 0: | |||||
# end of line | |||||
while len(data) % self.state.xsize != 0: | |||||
data += b"\x00" | |||||
x = 0 | |||||
elif byte[0] == 1: | |||||
# end of bitmap | |||||
break | |||||
elif byte[0] == 2: | |||||
# delta | |||||
bytes_read = self.fd.read(2) | |||||
if len(bytes_read) < 2: | |||||
break | |||||
right, up = self.fd.read(2) | |||||
data += b"\x00" * (right + up * self.state.xsize) | |||||
x = len(data) % self.state.xsize | |||||
else: | |||||
# absolute mode | |||||
if rle4: | |||||
# 2 pixels per byte | |||||
byte_count = byte[0] // 2 | |||||
bytes_read = self.fd.read(byte_count) | |||||
for byte_read in bytes_read: | |||||
data += o8(byte_read >> 4) | |||||
data += o8(byte_read & 0x0F) | |||||
else: | |||||
byte_count = byte[0] | |||||
bytes_read = self.fd.read(byte_count) | |||||
data += bytes_read | |||||
if len(bytes_read) < byte_count: | |||||
break | |||||
x += byte[0] | |||||
# align to 16-bit word boundary | |||||
if self.fd.tell() % 2 != 0: | |||||
self.fd.seek(1, os.SEEK_CUR) | |||||
rawmode = "L" if self.mode == "L" else "P" | |||||
self.set_as_raw(bytes(data), (rawmode, 0, self.args[-1])) | |||||
return -1, 0 | |||||
# ============================================================================= | |||||
# Image plugin for the DIB format (BMP alias) | |||||
# ============================================================================= | |||||
class DibImageFile(BmpImageFile): | |||||
format = "DIB" | |||||
format_description = "Windows Bitmap" | |||||
def _open(self): | |||||
self._bitmap() | |||||
# | |||||
# -------------------------------------------------------------------- | |||||
# Write BMP file | |||||
SAVE = { | |||||
"1": ("1", 1, 2), | |||||
"L": ("L", 8, 256), | |||||
"P": ("P", 8, 256), | |||||
"RGB": ("BGR", 24, 0), | |||||
"RGBA": ("BGRA", 32, 0), | |||||
} | |||||
def _dib_save(im, fp, filename): | |||||
_save(im, fp, filename, False) | |||||
def _save(im, fp, filename, bitmap_header=True): | |||||
try: | |||||
rawmode, bits, colors = SAVE[im.mode] | |||||
except KeyError as e: | |||||
msg = f"cannot write mode {im.mode} as BMP" | |||||
raise OSError(msg) from e | |||||
info = im.encoderinfo | |||||
dpi = info.get("dpi", (96, 96)) | |||||
# 1 meter == 39.3701 inches | |||||
ppm = tuple(map(lambda x: int(x * 39.3701 + 0.5), dpi)) | |||||
stride = ((im.size[0] * bits + 7) // 8 + 3) & (~3) | |||||
header = 40 # or 64 for OS/2 version 2 | |||||
image = stride * im.size[1] | |||||
if im.mode == "1": | |||||
palette = b"".join(o8(i) * 4 for i in (0, 255)) | |||||
elif im.mode == "L": | |||||
palette = b"".join(o8(i) * 4 for i in range(256)) | |||||
elif im.mode == "P": | |||||
palette = im.im.getpalette("RGB", "BGRX") | |||||
colors = len(palette) // 4 | |||||
else: | |||||
palette = None | |||||
# bitmap header | |||||
if bitmap_header: | |||||
offset = 14 + header + colors * 4 | |||||
file_size = offset + image | |||||
if file_size > 2**32 - 1: | |||||
msg = "File size is too large for the BMP format" | |||||
raise ValueError(msg) | |||||
fp.write( | |||||
b"BM" # file type (magic) | |||||
+ o32(file_size) # file size | |||||
+ o32(0) # reserved | |||||
+ o32(offset) # image data offset | |||||
) | |||||
# bitmap info header | |||||
fp.write( | |||||
o32(header) # info header size | |||||
+ o32(im.size[0]) # width | |||||
+ o32(im.size[1]) # height | |||||
+ o16(1) # planes | |||||
+ o16(bits) # depth | |||||
+ o32(0) # compression (0=uncompressed) | |||||
+ o32(image) # size of bitmap | |||||
+ o32(ppm[0]) # resolution | |||||
+ o32(ppm[1]) # resolution | |||||
+ o32(colors) # colors used | |||||
+ o32(colors) # colors important | |||||
) | |||||
fp.write(b"\0" * (header - 40)) # padding (for OS/2 format) | |||||
if palette: | |||||
fp.write(palette) | |||||
ImageFile._save(im, fp, [("raw", (0, 0) + im.size, 0, (rawmode, stride, -1))]) | |||||
# | |||||
# -------------------------------------------------------------------- | |||||
# Registry | |||||
Image.register_open(BmpImageFile.format, BmpImageFile, _accept) | |||||
Image.register_save(BmpImageFile.format, _save) | |||||
Image.register_extension(BmpImageFile.format, ".bmp") | |||||
Image.register_mime(BmpImageFile.format, "image/bmp") | |||||
Image.register_decoder("bmp_rle", BmpRleDecoder) | |||||
Image.register_open(DibImageFile.format, DibImageFile, _dib_accept) | |||||
Image.register_save(DibImageFile.format, _dib_save) | |||||
Image.register_extension(DibImageFile.format, ".dib") | |||||
Image.register_mime(DibImageFile.format, "image/bmp") |
# | |||||
# The Python Imaging Library | |||||
# $Id$ | |||||
# | |||||
# BUFR stub adapter | |||||
# | |||||
# Copyright (c) 1996-2003 by Fredrik Lundh | |||||
# | |||||
# See the README file for information on usage and redistribution. | |||||
# | |||||
from . import Image, ImageFile | |||||
_handler = None | |||||
def register_handler(handler): | |||||
""" | |||||
Install application-specific BUFR image handler. | |||||
:param handler: Handler object. | |||||
""" | |||||
global _handler | |||||
_handler = handler | |||||
# -------------------------------------------------------------------- | |||||
# Image adapter | |||||
def _accept(prefix): | |||||
return prefix[:4] == b"BUFR" or prefix[:4] == b"ZCZC" | |||||
class BufrStubImageFile(ImageFile.StubImageFile): | |||||
format = "BUFR" | |||||
format_description = "BUFR" | |||||
def _open(self): | |||||
offset = self.fp.tell() | |||||
if not _accept(self.fp.read(4)): | |||||
msg = "Not a BUFR file" | |||||
raise SyntaxError(msg) | |||||
self.fp.seek(offset) | |||||
# make something up | |||||
self.mode = "F" | |||||
self._size = 1, 1 | |||||
loader = self._load() | |||||
if loader: | |||||
loader.open(self) | |||||
def _load(self): | |||||
return _handler | |||||
def _save(im, fp, filename): | |||||
if _handler is None or not hasattr(_handler, "save"): | |||||
msg = "BUFR save handler not installed" | |||||
raise OSError(msg) | |||||
_handler.save(im, fp, filename) | |||||
# -------------------------------------------------------------------- | |||||
# Registry | |||||
Image.register_open(BufrStubImageFile.format, BufrStubImageFile, _accept) | |||||
Image.register_save(BufrStubImageFile.format, _save) | |||||
Image.register_extension(BufrStubImageFile.format, ".bufr") |
# | |||||
# The Python Imaging Library. | |||||
# $Id$ | |||||
# | |||||
# a class to read from a container file | |||||
# | |||||
# History: | |||||
# 1995-06-18 fl Created | |||||
# 1995-09-07 fl Added readline(), readlines() | |||||
# | |||||
# Copyright (c) 1997-2001 by Secret Labs AB | |||||
# Copyright (c) 1995 by Fredrik Lundh | |||||
# | |||||
# See the README file for information on usage and redistribution. | |||||
# | |||||
import io | |||||
class ContainerIO: | |||||
""" | |||||
A file object that provides read access to a part of an existing | |||||
file (for example a TAR file). | |||||
""" | |||||
def __init__(self, file, offset, length): | |||||
""" | |||||
Create file object. | |||||
:param file: Existing file. | |||||
:param offset: Start of region, in bytes. | |||||
:param length: Size of region, in bytes. | |||||
""" | |||||
self.fh = file | |||||
self.pos = 0 | |||||
self.offset = offset | |||||
self.length = length | |||||
self.fh.seek(offset) | |||||
## | |||||
# Always false. | |||||
def isatty(self): | |||||
return False | |||||
def seek(self, offset, mode=io.SEEK_SET): | |||||
""" | |||||
Move file pointer. | |||||
:param offset: Offset in bytes. | |||||
:param mode: Starting position. Use 0 for beginning of region, 1 | |||||
for current offset, and 2 for end of region. You cannot move | |||||
the pointer outside the defined region. | |||||
""" | |||||
if mode == 1: | |||||
self.pos = self.pos + offset | |||||
elif mode == 2: | |||||
self.pos = self.length + offset | |||||
else: | |||||
self.pos = offset | |||||
# clamp | |||||
self.pos = max(0, min(self.pos, self.length)) | |||||
self.fh.seek(self.offset + self.pos) | |||||
def tell(self): | |||||
""" | |||||
Get current file pointer. | |||||
:returns: Offset from start of region, in bytes. | |||||
""" | |||||
return self.pos | |||||
def read(self, n=0): | |||||
""" | |||||
Read data. | |||||
:param n: Number of bytes to read. If omitted or zero, | |||||
read until end of region. | |||||
:returns: An 8-bit string. | |||||
""" | |||||
if n: | |||||
n = min(n, self.length - self.pos) | |||||
else: | |||||
n = self.length - self.pos | |||||
if not n: # EOF | |||||
return b"" if "b" in self.fh.mode else "" | |||||
self.pos = self.pos + n | |||||
return self.fh.read(n) | |||||
def readline(self): | |||||
""" | |||||
Read a line of text. | |||||
:returns: An 8-bit string. | |||||
""" | |||||
s = b"" if "b" in self.fh.mode else "" | |||||
newline_character = b"\n" if "b" in self.fh.mode else "\n" | |||||
while True: | |||||
c = self.read(1) | |||||
if not c: | |||||
break | |||||
s = s + c | |||||
if c == newline_character: | |||||
break | |||||
return s | |||||
def readlines(self): | |||||
""" | |||||
Read multiple lines of text. | |||||
:returns: A list of 8-bit strings. | |||||
""" | |||||
lines = [] | |||||
while True: | |||||
s = self.readline() | |||||
if not s: | |||||
break | |||||
lines.append(s) | |||||
return lines |
# | |||||
# The Python Imaging Library. | |||||
# $Id$ | |||||
# | |||||
# Windows Cursor support for PIL | |||||
# | |||||
# notes: | |||||
# uses BmpImagePlugin.py to read the bitmap data. | |||||
# | |||||
# history: | |||||
# 96-05-27 fl Created | |||||
# | |||||
# Copyright (c) Secret Labs AB 1997. | |||||
# Copyright (c) Fredrik Lundh 1996. | |||||
# | |||||
# See the README file for information on usage and redistribution. | |||||
# | |||||
from . import BmpImagePlugin, Image | |||||
from ._binary import i16le as i16 | |||||
from ._binary import i32le as i32 | |||||
# | |||||
# -------------------------------------------------------------------- | |||||
def _accept(prefix): | |||||
return prefix[:4] == b"\0\0\2\0" | |||||
## | |||||
# Image plugin for Windows Cursor files. | |||||
class CurImageFile(BmpImagePlugin.BmpImageFile): | |||||
format = "CUR" | |||||
format_description = "Windows Cursor" | |||||
def _open(self): | |||||
offset = self.fp.tell() | |||||
# check magic | |||||
s = self.fp.read(6) | |||||
if not _accept(s): | |||||
msg = "not a CUR file" | |||||
raise SyntaxError(msg) | |||||
# pick the largest cursor in the file | |||||
m = b"" | |||||
for i in range(i16(s, 4)): | |||||
s = self.fp.read(16) | |||||
if not m: | |||||
m = s | |||||
elif s[0] > m[0] and s[1] > m[1]: | |||||
m = s | |||||
if not m: | |||||
msg = "No cursors were found" | |||||
raise TypeError(msg) | |||||
# load as bitmap | |||||
self._bitmap(i32(m, 12) + offset) | |||||
# patch up the bitmap height | |||||
self._size = self.size[0], self.size[1] // 2 | |||||
d, e, o, a = self.tile[0] | |||||
self.tile[0] = d, (0, 0) + self.size, o, a | |||||
return | |||||
# | |||||
# -------------------------------------------------------------------- | |||||
Image.register_open(CurImageFile.format, CurImageFile, _accept) | |||||
Image.register_extension(CurImageFile.format, ".cur") |
# | |||||
# The Python Imaging Library. | |||||
# $Id$ | |||||
# | |||||
# DCX file handling | |||||
# | |||||
# DCX is a container file format defined by Intel, commonly used | |||||
# for fax applications. Each DCX file consists of a directory | |||||
# (a list of file offsets) followed by a set of (usually 1-bit) | |||||
# PCX files. | |||||
# | |||||
# History: | |||||
# 1995-09-09 fl Created | |||||
# 1996-03-20 fl Properly derived from PcxImageFile. | |||||
# 1998-07-15 fl Renamed offset attribute to avoid name clash | |||||
# 2002-07-30 fl Fixed file handling | |||||
# | |||||
# Copyright (c) 1997-98 by Secret Labs AB. | |||||
# Copyright (c) 1995-96 by Fredrik Lundh. | |||||
# | |||||
# See the README file for information on usage and redistribution. | |||||
# | |||||
from . import Image | |||||
from ._binary import i32le as i32 | |||||
from .PcxImagePlugin import PcxImageFile | |||||
MAGIC = 0x3ADE68B1 # QUIZ: what's this value, then? | |||||
def _accept(prefix): | |||||
return len(prefix) >= 4 and i32(prefix) == MAGIC | |||||
## | |||||
# Image plugin for the Intel DCX format. | |||||
class DcxImageFile(PcxImageFile): | |||||
format = "DCX" | |||||
format_description = "Intel DCX" | |||||
_close_exclusive_fp_after_loading = False | |||||
def _open(self): | |||||
# Header | |||||
s = self.fp.read(4) | |||||
if not _accept(s): | |||||
msg = "not a DCX file" | |||||
raise SyntaxError(msg) | |||||
# Component directory | |||||
self._offset = [] | |||||
for i in range(1024): | |||||
offset = i32(self.fp.read(4)) | |||||
if not offset: | |||||
break | |||||
self._offset.append(offset) | |||||
self._fp = self.fp | |||||
self.frame = None | |||||
self.n_frames = len(self._offset) | |||||
self.is_animated = self.n_frames > 1 | |||||
self.seek(0) | |||||
def seek(self, frame): | |||||
if not self._seek_check(frame): | |||||
return | |||||
self.frame = frame | |||||
self.fp = self._fp | |||||
self.fp.seek(self._offset[frame]) | |||||
PcxImageFile._open(self) | |||||
def tell(self): | |||||
return self.frame | |||||
Image.register_open(DcxImageFile.format, DcxImageFile, _accept) | |||||
Image.register_extension(DcxImageFile.format, ".dcx") |
""" | |||||
A Pillow loader for .dds files (S3TC-compressed aka DXTC) | |||||
Jerome Leclanche <jerome@leclan.ch> | |||||
Documentation: | |||||
https://web.archive.org/web/20170802060935/http://oss.sgi.com/projects/ogl-sample/registry/EXT/texture_compression_s3tc.txt | |||||
The contents of this file are hereby released in the public domain (CC0) | |||||
Full text of the CC0 license: | |||||
https://creativecommons.org/publicdomain/zero/1.0/ | |||||
""" | |||||
import struct | |||||
from io import BytesIO | |||||
from . import Image, ImageFile | |||||
from ._binary import o32le as o32 | |||||
# Magic ("DDS ") | |||||
DDS_MAGIC = 0x20534444 | |||||
# DDS flags | |||||
DDSD_CAPS = 0x1 | |||||
DDSD_HEIGHT = 0x2 | |||||
DDSD_WIDTH = 0x4 | |||||
DDSD_PITCH = 0x8 | |||||
DDSD_PIXELFORMAT = 0x1000 | |||||
DDSD_MIPMAPCOUNT = 0x20000 | |||||
DDSD_LINEARSIZE = 0x80000 | |||||
DDSD_DEPTH = 0x800000 | |||||
# DDS caps | |||||
DDSCAPS_COMPLEX = 0x8 | |||||
DDSCAPS_TEXTURE = 0x1000 | |||||
DDSCAPS_MIPMAP = 0x400000 | |||||
DDSCAPS2_CUBEMAP = 0x200 | |||||
DDSCAPS2_CUBEMAP_POSITIVEX = 0x400 | |||||
DDSCAPS2_CUBEMAP_NEGATIVEX = 0x800 | |||||
DDSCAPS2_CUBEMAP_POSITIVEY = 0x1000 | |||||
DDSCAPS2_CUBEMAP_NEGATIVEY = 0x2000 | |||||
DDSCAPS2_CUBEMAP_POSITIVEZ = 0x4000 | |||||
DDSCAPS2_CUBEMAP_NEGATIVEZ = 0x8000 | |||||
DDSCAPS2_VOLUME = 0x200000 | |||||
# Pixel Format | |||||
DDPF_ALPHAPIXELS = 0x1 | |||||
DDPF_ALPHA = 0x2 | |||||
DDPF_FOURCC = 0x4 | |||||
DDPF_PALETTEINDEXED8 = 0x20 | |||||
DDPF_RGB = 0x40 | |||||
DDPF_LUMINANCE = 0x20000 | |||||
# dds.h | |||||
DDS_FOURCC = DDPF_FOURCC | |||||
DDS_RGB = DDPF_RGB | |||||
DDS_RGBA = DDPF_RGB | DDPF_ALPHAPIXELS | |||||
DDS_LUMINANCE = DDPF_LUMINANCE | |||||
DDS_LUMINANCEA = DDPF_LUMINANCE | DDPF_ALPHAPIXELS | |||||
DDS_ALPHA = DDPF_ALPHA | |||||
DDS_PAL8 = DDPF_PALETTEINDEXED8 | |||||
DDS_HEADER_FLAGS_TEXTURE = DDSD_CAPS | DDSD_HEIGHT | DDSD_WIDTH | DDSD_PIXELFORMAT | |||||
DDS_HEADER_FLAGS_MIPMAP = DDSD_MIPMAPCOUNT | |||||
DDS_HEADER_FLAGS_VOLUME = DDSD_DEPTH | |||||
DDS_HEADER_FLAGS_PITCH = DDSD_PITCH | |||||
DDS_HEADER_FLAGS_LINEARSIZE = DDSD_LINEARSIZE | |||||
DDS_HEIGHT = DDSD_HEIGHT | |||||
DDS_WIDTH = DDSD_WIDTH | |||||
DDS_SURFACE_FLAGS_TEXTURE = DDSCAPS_TEXTURE | |||||
DDS_SURFACE_FLAGS_MIPMAP = DDSCAPS_COMPLEX | DDSCAPS_MIPMAP | |||||
DDS_SURFACE_FLAGS_CUBEMAP = DDSCAPS_COMPLEX | |||||
DDS_CUBEMAP_POSITIVEX = DDSCAPS2_CUBEMAP | DDSCAPS2_CUBEMAP_POSITIVEX | |||||
DDS_CUBEMAP_NEGATIVEX = DDSCAPS2_CUBEMAP | DDSCAPS2_CUBEMAP_NEGATIVEX | |||||
DDS_CUBEMAP_POSITIVEY = DDSCAPS2_CUBEMAP | DDSCAPS2_CUBEMAP_POSITIVEY | |||||
DDS_CUBEMAP_NEGATIVEY = DDSCAPS2_CUBEMAP | DDSCAPS2_CUBEMAP_NEGATIVEY | |||||
DDS_CUBEMAP_POSITIVEZ = DDSCAPS2_CUBEMAP | DDSCAPS2_CUBEMAP_POSITIVEZ | |||||
DDS_CUBEMAP_NEGATIVEZ = DDSCAPS2_CUBEMAP | DDSCAPS2_CUBEMAP_NEGATIVEZ | |||||
# DXT1 | |||||
DXT1_FOURCC = 0x31545844 | |||||
# DXT3 | |||||
DXT3_FOURCC = 0x33545844 | |||||
# DXT5 | |||||
DXT5_FOURCC = 0x35545844 | |||||
# dxgiformat.h | |||||
DXGI_FORMAT_R8G8B8A8_TYPELESS = 27 | |||||
DXGI_FORMAT_R8G8B8A8_UNORM = 28 | |||||
DXGI_FORMAT_R8G8B8A8_UNORM_SRGB = 29 | |||||
DXGI_FORMAT_BC5_TYPELESS = 82 | |||||
DXGI_FORMAT_BC5_UNORM = 83 | |||||
DXGI_FORMAT_BC5_SNORM = 84 | |||||
DXGI_FORMAT_BC6H_UF16 = 95 | |||||
DXGI_FORMAT_BC6H_SF16 = 96 | |||||
DXGI_FORMAT_BC7_TYPELESS = 97 | |||||
DXGI_FORMAT_BC7_UNORM = 98 | |||||
DXGI_FORMAT_BC7_UNORM_SRGB = 99 | |||||
class DdsImageFile(ImageFile.ImageFile): | |||||
format = "DDS" | |||||
format_description = "DirectDraw Surface" | |||||
def _open(self): | |||||
if not _accept(self.fp.read(4)): | |||||
msg = "not a DDS file" | |||||
raise SyntaxError(msg) | |||||
(header_size,) = struct.unpack("<I", self.fp.read(4)) | |||||
if header_size != 124: | |||||
msg = f"Unsupported header size {repr(header_size)}" | |||||
raise OSError(msg) | |||||
header_bytes = self.fp.read(header_size - 4) | |||||
if len(header_bytes) != 120: | |||||
msg = f"Incomplete header: {len(header_bytes)} bytes" | |||||
raise OSError(msg) | |||||
header = BytesIO(header_bytes) | |||||
flags, height, width = struct.unpack("<3I", header.read(12)) | |||||
self._size = (width, height) | |||||
self.mode = "RGBA" | |||||
pitch, depth, mipmaps = struct.unpack("<3I", header.read(12)) | |||||
struct.unpack("<11I", header.read(44)) # reserved | |||||
# pixel format | |||||
pfsize, pfflags = struct.unpack("<2I", header.read(8)) | |||||
fourcc = header.read(4) | |||||
(bitcount,) = struct.unpack("<I", header.read(4)) | |||||
masks = struct.unpack("<4I", header.read(16)) | |||||
if pfflags & DDPF_LUMINANCE: | |||||
# Texture contains uncompressed L or LA data | |||||
if pfflags & DDPF_ALPHAPIXELS: | |||||
self.mode = "LA" | |||||
else: | |||||
self.mode = "L" | |||||
self.tile = [("raw", (0, 0) + self.size, 0, (self.mode, 0, 1))] | |||||
elif pfflags & DDPF_RGB: | |||||
# Texture contains uncompressed RGB data | |||||
masks = {mask: ["R", "G", "B", "A"][i] for i, mask in enumerate(masks)} | |||||
rawmode = "" | |||||
if pfflags & DDPF_ALPHAPIXELS: | |||||
rawmode += masks[0xFF000000] | |||||
else: | |||||
self.mode = "RGB" | |||||
rawmode += masks[0xFF0000] + masks[0xFF00] + masks[0xFF] | |||||
self.tile = [("raw", (0, 0) + self.size, 0, (rawmode[::-1], 0, 1))] | |||||
else: | |||||
data_start = header_size + 4 | |||||
n = 0 | |||||
if fourcc == b"DXT1": | |||||
self.pixel_format = "DXT1" | |||||
n = 1 | |||||
elif fourcc == b"DXT3": | |||||
self.pixel_format = "DXT3" | |||||
n = 2 | |||||
elif fourcc == b"DXT5": | |||||
self.pixel_format = "DXT5" | |||||
n = 3 | |||||
elif fourcc == b"ATI1": | |||||
self.pixel_format = "BC4" | |||||
n = 4 | |||||
self.mode = "L" | |||||
elif fourcc == b"ATI2": | |||||
self.pixel_format = "BC5" | |||||
n = 5 | |||||
self.mode = "RGB" | |||||
elif fourcc == b"BC5S": | |||||
self.pixel_format = "BC5S" | |||||
n = 5 | |||||
self.mode = "RGB" | |||||
elif fourcc == b"DX10": | |||||
data_start += 20 | |||||
# ignoring flags which pertain to volume textures and cubemaps | |||||
(dxgi_format,) = struct.unpack("<I", self.fp.read(4)) | |||||
self.fp.read(16) | |||||
if dxgi_format in (DXGI_FORMAT_BC5_TYPELESS, DXGI_FORMAT_BC5_UNORM): | |||||
self.pixel_format = "BC5" | |||||
n = 5 | |||||
self.mode = "RGB" | |||||
elif dxgi_format == DXGI_FORMAT_BC5_SNORM: | |||||
self.pixel_format = "BC5S" | |||||
n = 5 | |||||
self.mode = "RGB" | |||||
elif dxgi_format == DXGI_FORMAT_BC6H_UF16: | |||||
self.pixel_format = "BC6H" | |||||
n = 6 | |||||
self.mode = "RGB" | |||||
elif dxgi_format == DXGI_FORMAT_BC6H_SF16: | |||||
self.pixel_format = "BC6HS" | |||||
n = 6 | |||||
self.mode = "RGB" | |||||
elif dxgi_format in (DXGI_FORMAT_BC7_TYPELESS, DXGI_FORMAT_BC7_UNORM): | |||||
self.pixel_format = "BC7" | |||||
n = 7 | |||||
elif dxgi_format == DXGI_FORMAT_BC7_UNORM_SRGB: | |||||
self.pixel_format = "BC7" | |||||
self.info["gamma"] = 1 / 2.2 | |||||
n = 7 | |||||
elif dxgi_format in ( | |||||
DXGI_FORMAT_R8G8B8A8_TYPELESS, | |||||
DXGI_FORMAT_R8G8B8A8_UNORM, | |||||
DXGI_FORMAT_R8G8B8A8_UNORM_SRGB, | |||||
): | |||||
self.tile = [("raw", (0, 0) + self.size, 0, ("RGBA", 0, 1))] | |||||
if dxgi_format == DXGI_FORMAT_R8G8B8A8_UNORM_SRGB: | |||||
self.info["gamma"] = 1 / 2.2 | |||||
return | |||||
else: | |||||
msg = f"Unimplemented DXGI format {dxgi_format}" | |||||
raise NotImplementedError(msg) | |||||
else: | |||||
msg = f"Unimplemented pixel format {repr(fourcc)}" | |||||
raise NotImplementedError(msg) | |||||
self.tile = [ | |||||
("bcn", (0, 0) + self.size, data_start, (n, self.pixel_format)) | |||||
] | |||||
def load_seek(self, pos): | |||||
pass | |||||
def _save(im, fp, filename): | |||||
if im.mode not in ("RGB", "RGBA", "L", "LA"): | |||||
msg = f"cannot write mode {im.mode} as DDS" | |||||
raise OSError(msg) | |||||
rawmode = im.mode | |||||
masks = [0xFF0000, 0xFF00, 0xFF] | |||||
if im.mode in ("L", "LA"): | |||||
pixel_flags = DDPF_LUMINANCE | |||||
else: | |||||
pixel_flags = DDPF_RGB | |||||
rawmode = rawmode[::-1] | |||||
if im.mode in ("LA", "RGBA"): | |||||
pixel_flags |= DDPF_ALPHAPIXELS | |||||
masks.append(0xFF000000) | |||||
bitcount = len(masks) * 8 | |||||
while len(masks) < 4: | |||||
masks.append(0) | |||||
fp.write( | |||||
o32(DDS_MAGIC) | |||||
+ o32(124) # header size | |||||
+ o32( | |||||
DDSD_CAPS | DDSD_HEIGHT | DDSD_WIDTH | DDSD_PITCH | DDSD_PIXELFORMAT | |||||
) # flags | |||||
+ o32(im.height) | |||||
+ o32(im.width) | |||||
+ o32((im.width * bitcount + 7) // 8) # pitch | |||||
+ o32(0) # depth | |||||
+ o32(0) # mipmaps | |||||
+ o32(0) * 11 # reserved | |||||
+ o32(32) # pfsize | |||||
+ o32(pixel_flags) # pfflags | |||||
+ o32(0) # fourcc | |||||
+ o32(bitcount) # bitcount | |||||
+ b"".join(o32(mask) for mask in masks) # rgbabitmask | |||||
+ o32(DDSCAPS_TEXTURE) # dwCaps | |||||
+ o32(0) # dwCaps2 | |||||
+ o32(0) # dwCaps3 | |||||
+ o32(0) # dwCaps4 | |||||
+ o32(0) # dwReserved2 | |||||
) | |||||
if im.mode == "RGBA": | |||||
r, g, b, a = im.split() | |||||
im = Image.merge("RGBA", (a, r, g, b)) | |||||
ImageFile._save(im, fp, [("raw", (0, 0) + im.size, 0, (rawmode, 0, 1))]) | |||||
def _accept(prefix): | |||||
return prefix[:4] == b"DDS " | |||||
Image.register_open(DdsImageFile.format, DdsImageFile, _accept) | |||||
Image.register_save(DdsImageFile.format, _save) | |||||
Image.register_extension(DdsImageFile.format, ".dds") |
# | |||||
# The Python Imaging Library. | |||||
# $Id$ | |||||
# | |||||
# EPS file handling | |||||
# | |||||
# History: | |||||
# 1995-09-01 fl Created (0.1) | |||||
# 1996-05-18 fl Don't choke on "atend" fields, Ghostscript interface (0.2) | |||||
# 1996-08-22 fl Don't choke on floating point BoundingBox values | |||||
# 1996-08-23 fl Handle files from Macintosh (0.3) | |||||
# 2001-02-17 fl Use 're' instead of 'regex' (Python 2.1) (0.4) | |||||
# 2003-09-07 fl Check gs.close status (from Federico Di Gregorio) (0.5) | |||||
# 2014-05-07 e Handling of EPS with binary preview and fixed resolution | |||||
# resizing | |||||
# | |||||
# Copyright (c) 1997-2003 by Secret Labs AB. | |||||
# Copyright (c) 1995-2003 by Fredrik Lundh | |||||
# | |||||
# See the README file for information on usage and redistribution. | |||||
# | |||||
import io | |||||
import os | |||||
import re | |||||
import subprocess | |||||
import sys | |||||
import tempfile | |||||
from . import Image, ImageFile | |||||
from ._binary import i32le as i32 | |||||
from ._deprecate import deprecate | |||||
# -------------------------------------------------------------------- | |||||
split = re.compile(r"^%%([^:]*):[ \t]*(.*)[ \t]*$") | |||||
field = re.compile(r"^%[%!\w]([^:]*)[ \t]*$") | |||||
gs_windows_binary = None | |||||
if sys.platform.startswith("win"): | |||||
import shutil | |||||
for binary in ("gswin32c", "gswin64c", "gs"): | |||||
if shutil.which(binary) is not None: | |||||
gs_windows_binary = binary | |||||
break | |||||
else: | |||||
gs_windows_binary = False | |||||
def has_ghostscript(): | |||||
if gs_windows_binary: | |||||
return True | |||||
if not sys.platform.startswith("win"): | |||||
try: | |||||
subprocess.check_call(["gs", "--version"], stdout=subprocess.DEVNULL) | |||||
return True | |||||
except OSError: | |||||
# No Ghostscript | |||||
pass | |||||
return False | |||||
def Ghostscript(tile, size, fp, scale=1, transparency=False): | |||||
"""Render an image using Ghostscript""" | |||||
# Unpack decoder tile | |||||
decoder, tile, offset, data = tile[0] | |||||
length, bbox = data | |||||
# Hack to support hi-res rendering | |||||
scale = int(scale) or 1 | |||||
# orig_size = size | |||||
# orig_bbox = bbox | |||||
size = (size[0] * scale, size[1] * scale) | |||||
# resolution is dependent on bbox and size | |||||
res = ( | |||||
72.0 * size[0] / (bbox[2] - bbox[0]), | |||||
72.0 * size[1] / (bbox[3] - bbox[1]), | |||||
) | |||||
out_fd, outfile = tempfile.mkstemp() | |||||
os.close(out_fd) | |||||
infile_temp = None | |||||
if hasattr(fp, "name") and os.path.exists(fp.name): | |||||
infile = fp.name | |||||
else: | |||||
in_fd, infile_temp = tempfile.mkstemp() | |||||
os.close(in_fd) | |||||
infile = infile_temp | |||||
# Ignore length and offset! | |||||
# Ghostscript can read it | |||||
# Copy whole file to read in Ghostscript | |||||
with open(infile_temp, "wb") as f: | |||||
# fetch length of fp | |||||
fp.seek(0, io.SEEK_END) | |||||
fsize = fp.tell() | |||||
# ensure start position | |||||
# go back | |||||
fp.seek(0) | |||||
lengthfile = fsize | |||||
while lengthfile > 0: | |||||
s = fp.read(min(lengthfile, 100 * 1024)) | |||||
if not s: | |||||
break | |||||
lengthfile -= len(s) | |||||
f.write(s) | |||||
device = "pngalpha" if transparency else "ppmraw" | |||||
# Build Ghostscript command | |||||
command = [ | |||||
"gs", | |||||
"-q", # quiet mode | |||||
"-g%dx%d" % size, # set output geometry (pixels) | |||||
"-r%fx%f" % res, # set input DPI (dots per inch) | |||||
"-dBATCH", # exit after processing | |||||
"-dNOPAUSE", # don't pause between pages | |||||
"-dSAFER", # safe mode | |||||
f"-sDEVICE={device}", | |||||
f"-sOutputFile={outfile}", # output file | |||||
# adjust for image origin | |||||
"-c", | |||||
f"{-bbox[0]} {-bbox[1]} translate", | |||||
"-f", | |||||
infile, # input file | |||||
# showpage (see https://bugs.ghostscript.com/show_bug.cgi?id=698272) | |||||
"-c", | |||||
"showpage", | |||||
] | |||||
if gs_windows_binary is not None: | |||||
if not gs_windows_binary: | |||||
msg = "Unable to locate Ghostscript on paths" | |||||
raise OSError(msg) | |||||
command[0] = gs_windows_binary | |||||
# push data through Ghostscript | |||||
try: | |||||
startupinfo = None | |||||
if sys.platform.startswith("win"): | |||||
startupinfo = subprocess.STARTUPINFO() | |||||
startupinfo.dwFlags |= subprocess.STARTF_USESHOWWINDOW | |||||
subprocess.check_call(command, startupinfo=startupinfo) | |||||
out_im = Image.open(outfile) | |||||
out_im.load() | |||||
finally: | |||||
try: | |||||
os.unlink(outfile) | |||||
if infile_temp: | |||||
os.unlink(infile_temp) | |||||
except OSError: | |||||
pass | |||||
im = out_im.im.copy() | |||||
out_im.close() | |||||
return im | |||||
class PSFile: | |||||
""" | |||||
Wrapper for bytesio object that treats either CR or LF as end of line. | |||||
This class is no longer used internally, but kept for backwards compatibility. | |||||
""" | |||||
def __init__(self, fp): | |||||
deprecate( | |||||
"PSFile", | |||||
11, | |||||
action="If you need the functionality of this class " | |||||
"you will need to implement it yourself.", | |||||
) | |||||
self.fp = fp | |||||
self.char = None | |||||
def seek(self, offset, whence=io.SEEK_SET): | |||||
self.char = None | |||||
self.fp.seek(offset, whence) | |||||
def readline(self): | |||||
s = [self.char or b""] | |||||
self.char = None | |||||
c = self.fp.read(1) | |||||
while (c not in b"\r\n") and len(c): | |||||
s.append(c) | |||||
c = self.fp.read(1) | |||||
self.char = self.fp.read(1) | |||||
# line endings can be 1 or 2 of \r \n, in either order | |||||
if self.char in b"\r\n": | |||||
self.char = None | |||||
return b"".join(s).decode("latin-1") | |||||
def _accept(prefix): | |||||
return prefix[:4] == b"%!PS" or (len(prefix) >= 4 and i32(prefix) == 0xC6D3D0C5) | |||||
## | |||||
# Image plugin for Encapsulated PostScript. This plugin supports only | |||||
# a few variants of this format. | |||||
class EpsImageFile(ImageFile.ImageFile): | |||||
"""EPS File Parser for the Python Imaging Library""" | |||||
format = "EPS" | |||||
format_description = "Encapsulated Postscript" | |||||
mode_map = {1: "L", 2: "LAB", 3: "RGB", 4: "CMYK"} | |||||
def _open(self): | |||||
(length, offset) = self._find_offset(self.fp) | |||||
# go to offset - start of "%!PS" | |||||
self.fp.seek(offset) | |||||
self.mode = "RGB" | |||||
self._size = None | |||||
byte_arr = bytearray(255) | |||||
bytes_mv = memoryview(byte_arr) | |||||
bytes_read = 0 | |||||
reading_comments = True | |||||
def check_required_header_comments(): | |||||
if "PS-Adobe" not in self.info: | |||||
msg = 'EPS header missing "%!PS-Adobe" comment' | |||||
raise SyntaxError(msg) | |||||
if "BoundingBox" not in self.info: | |||||
msg = 'EPS header missing "%%BoundingBox" comment' | |||||
raise SyntaxError(msg) | |||||
while True: | |||||
byte = self.fp.read(1) | |||||
if byte == b"": | |||||
# if we didn't read a byte we must be at the end of the file | |||||
if bytes_read == 0: | |||||
break | |||||
elif byte in b"\r\n": | |||||
# if we read a line ending character, ignore it and parse what | |||||
# we have already read. if we haven't read any other characters, | |||||
# continue reading | |||||
if bytes_read == 0: | |||||
continue | |||||
else: | |||||
# ASCII/hexadecimal lines in an EPS file must not exceed | |||||
# 255 characters, not including line ending characters | |||||
if bytes_read >= 255: | |||||
# only enforce this for lines starting with a "%", | |||||
# otherwise assume it's binary data | |||||
if byte_arr[0] == ord("%"): | |||||
msg = "not an EPS file" | |||||
raise SyntaxError(msg) | |||||
else: | |||||
if reading_comments: | |||||
check_required_header_comments() | |||||
reading_comments = False | |||||
# reset bytes_read so we can keep reading | |||||
# data until the end of the line | |||||
bytes_read = 0 | |||||
byte_arr[bytes_read] = byte[0] | |||||
bytes_read += 1 | |||||
continue | |||||
if reading_comments: | |||||
# Load EPS header | |||||
# if this line doesn't start with a "%", | |||||
# or does start with "%%EndComments", | |||||
# then we've reached the end of the header/comments | |||||
if byte_arr[0] != ord("%") or bytes_mv[:13] == b"%%EndComments": | |||||
check_required_header_comments() | |||||
reading_comments = False | |||||
continue | |||||
s = str(bytes_mv[:bytes_read], "latin-1") | |||||
try: | |||||
m = split.match(s) | |||||
except re.error as e: | |||||
msg = "not an EPS file" | |||||
raise SyntaxError(msg) from e | |||||
if m: | |||||
k, v = m.group(1, 2) | |||||
self.info[k] = v | |||||
if k == "BoundingBox": | |||||
try: | |||||
# Note: The DSC spec says that BoundingBox | |||||
# fields should be integers, but some drivers | |||||
# put floating point values there anyway. | |||||
box = [int(float(i)) for i in v.split()] | |||||
self._size = box[2] - box[0], box[3] - box[1] | |||||
self.tile = [ | |||||
("eps", (0, 0) + self.size, offset, (length, box)) | |||||
] | |||||
except Exception: | |||||
pass | |||||
else: | |||||
m = field.match(s) | |||||
if m: | |||||
k = m.group(1) | |||||
if k[:8] == "PS-Adobe": | |||||
self.info["PS-Adobe"] = k[9:] | |||||
else: | |||||
self.info[k] = "" | |||||
elif s[0] == "%": | |||||
# handle non-DSC PostScript comments that some | |||||
# tools mistakenly put in the Comments section | |||||
pass | |||||
else: | |||||
msg = "bad EPS header" | |||||
raise OSError(msg) | |||||
elif bytes_mv[:11] == b"%ImageData:": | |||||
# Check for an "ImageData" descriptor | |||||
# https://www.adobe.com/devnet-apps/photoshop/fileformatashtml/#50577413_pgfId-1035096 | |||||
# Values: | |||||
# columns | |||||
# rows | |||||
# bit depth (1 or 8) | |||||
# mode (1: L, 2: LAB, 3: RGB, 4: CMYK) | |||||
# number of padding channels | |||||
# block size (number of bytes per row per channel) | |||||
# binary/ascii (1: binary, 2: ascii) | |||||
# data start identifier (the image data follows after a single line | |||||
# consisting only of this quoted value) | |||||
image_data_values = byte_arr[11:bytes_read].split(None, 7) | |||||
columns, rows, bit_depth, mode_id = [ | |||||
int(value) for value in image_data_values[:4] | |||||
] | |||||
if bit_depth == 1: | |||||
self.mode = "1" | |||||
elif bit_depth == 8: | |||||
try: | |||||
self.mode = self.mode_map[mode_id] | |||||
except ValueError: | |||||
break | |||||
else: | |||||
break | |||||
self._size = columns, rows | |||||
return | |||||
bytes_read = 0 | |||||
check_required_header_comments() | |||||
if not self._size: | |||||
self._size = 1, 1 # errors if this isn't set. why (1,1)? | |||||
msg = "cannot determine EPS bounding box" | |||||
raise OSError(msg) | |||||
def _find_offset(self, fp): | |||||
s = fp.read(4) | |||||
if s == b"%!PS": | |||||
# for HEAD without binary preview | |||||
fp.seek(0, io.SEEK_END) | |||||
length = fp.tell() | |||||
offset = 0 | |||||
elif i32(s) == 0xC6D3D0C5: | |||||
# FIX for: Some EPS file not handled correctly / issue #302 | |||||
# EPS can contain binary data | |||||
# or start directly with latin coding | |||||
# more info see: | |||||
# https://web.archive.org/web/20160528181353/http://partners.adobe.com/public/developer/en/ps/5002.EPSF_Spec.pdf | |||||
s = fp.read(8) | |||||
offset = i32(s) | |||||
length = i32(s, 4) | |||||
else: | |||||
msg = "not an EPS file" | |||||
raise SyntaxError(msg) | |||||
return length, offset | |||||
def load(self, scale=1, transparency=False): | |||||
# Load EPS via Ghostscript | |||||
if self.tile: | |||||
self.im = Ghostscript(self.tile, self.size, self.fp, scale, transparency) | |||||
self.mode = self.im.mode | |||||
self._size = self.im.size | |||||
self.tile = [] | |||||
return Image.Image.load(self) | |||||
def load_seek(self, *args, **kwargs): | |||||
# we can't incrementally load, so force ImageFile.parser to | |||||
# use our custom load method by defining this method. | |||||
pass | |||||
# -------------------------------------------------------------------- | |||||
def _save(im, fp, filename, eps=1): | |||||
"""EPS Writer for the Python Imaging Library.""" | |||||
# make sure image data is available | |||||
im.load() | |||||
# determine PostScript image mode | |||||
if im.mode == "L": | |||||
operator = (8, 1, b"image") | |||||
elif im.mode == "RGB": | |||||
operator = (8, 3, b"false 3 colorimage") | |||||
elif im.mode == "CMYK": | |||||
operator = (8, 4, b"false 4 colorimage") | |||||
else: | |||||
msg = "image mode is not supported" | |||||
raise ValueError(msg) | |||||
if eps: | |||||
# write EPS header | |||||
fp.write(b"%!PS-Adobe-3.0 EPSF-3.0\n") | |||||
fp.write(b"%%Creator: PIL 0.1 EpsEncode\n") | |||||
# fp.write("%%CreationDate: %s"...) | |||||
fp.write(b"%%%%BoundingBox: 0 0 %d %d\n" % im.size) | |||||
fp.write(b"%%Pages: 1\n") | |||||
fp.write(b"%%EndComments\n") | |||||
fp.write(b"%%Page: 1 1\n") | |||||
fp.write(b"%%ImageData: %d %d " % im.size) | |||||
fp.write(b'%d %d 0 1 1 "%s"\n' % operator) | |||||
# image header | |||||
fp.write(b"gsave\n") | |||||
fp.write(b"10 dict begin\n") | |||||
fp.write(b"/buf %d string def\n" % (im.size[0] * operator[1])) | |||||
fp.write(b"%d %d scale\n" % im.size) | |||||
fp.write(b"%d %d 8\n" % im.size) # <= bits | |||||
fp.write(b"[%d 0 0 -%d 0 %d]\n" % (im.size[0], im.size[1], im.size[1])) | |||||
fp.write(b"{ currentfile buf readhexstring pop } bind\n") | |||||
fp.write(operator[2] + b"\n") | |||||
if hasattr(fp, "flush"): | |||||
fp.flush() | |||||
ImageFile._save(im, fp, [("eps", (0, 0) + im.size, 0, None)]) | |||||
fp.write(b"\n%%%%EndBinary\n") | |||||
fp.write(b"grestore end\n") | |||||
if hasattr(fp, "flush"): | |||||
fp.flush() | |||||
# -------------------------------------------------------------------- | |||||
Image.register_open(EpsImageFile.format, EpsImageFile, _accept) | |||||
Image.register_save(EpsImageFile.format, _save) | |||||
Image.register_extensions(EpsImageFile.format, [".ps", ".eps"]) | |||||
Image.register_mime(EpsImageFile.format, "application/postscript") |
# | |||||
# The Python Imaging Library. | |||||
# $Id$ | |||||
# | |||||
# EXIF tags | |||||
# | |||||
# Copyright (c) 2003 by Secret Labs AB | |||||
# | |||||
# See the README file for information on usage and redistribution. | |||||
# | |||||
""" | |||||
This module provides constants and clear-text names for various | |||||
well-known EXIF tags. | |||||
""" | |||||
from enum import IntEnum | |||||
class Base(IntEnum): | |||||
# possibly incomplete | |||||
InteropIndex = 0x0001 | |||||
ProcessingSoftware = 0x000B | |||||
NewSubfileType = 0x00FE | |||||
SubfileType = 0x00FF | |||||
ImageWidth = 0x0100 | |||||
ImageLength = 0x0101 | |||||
BitsPerSample = 0x0102 | |||||
Compression = 0x0103 | |||||
PhotometricInterpretation = 0x0106 | |||||
Thresholding = 0x0107 | |||||
CellWidth = 0x0108 | |||||
CellLength = 0x0109 | |||||
FillOrder = 0x010A | |||||
DocumentName = 0x010D | |||||
ImageDescription = 0x010E | |||||
Make = 0x010F | |||||
Model = 0x0110 | |||||
StripOffsets = 0x0111 | |||||
Orientation = 0x0112 | |||||
SamplesPerPixel = 0x0115 | |||||
RowsPerStrip = 0x0116 | |||||
StripByteCounts = 0x0117 | |||||
MinSampleValue = 0x0118 | |||||
MaxSampleValue = 0x0119 | |||||
XResolution = 0x011A | |||||
YResolution = 0x011B | |||||
PlanarConfiguration = 0x011C | |||||
PageName = 0x011D | |||||
FreeOffsets = 0x0120 | |||||
FreeByteCounts = 0x0121 | |||||
GrayResponseUnit = 0x0122 | |||||
GrayResponseCurve = 0x0123 | |||||
T4Options = 0x0124 | |||||
T6Options = 0x0125 | |||||
ResolutionUnit = 0x0128 | |||||
PageNumber = 0x0129 | |||||
TransferFunction = 0x012D | |||||
Software = 0x0131 | |||||
DateTime = 0x0132 | |||||
Artist = 0x013B | |||||
HostComputer = 0x013C | |||||
Predictor = 0x013D | |||||
WhitePoint = 0x013E | |||||
PrimaryChromaticities = 0x013F | |||||
ColorMap = 0x0140 | |||||
HalftoneHints = 0x0141 | |||||
TileWidth = 0x0142 | |||||
TileLength = 0x0143 | |||||
TileOffsets = 0x0144 | |||||
TileByteCounts = 0x0145 | |||||
SubIFDs = 0x014A | |||||
InkSet = 0x014C | |||||
InkNames = 0x014D | |||||
NumberOfInks = 0x014E | |||||
DotRange = 0x0150 | |||||
TargetPrinter = 0x0151 | |||||
ExtraSamples = 0x0152 | |||||
SampleFormat = 0x0153 | |||||
SMinSampleValue = 0x0154 | |||||
SMaxSampleValue = 0x0155 | |||||
TransferRange = 0x0156 | |||||
ClipPath = 0x0157 | |||||
XClipPathUnits = 0x0158 | |||||
YClipPathUnits = 0x0159 | |||||
Indexed = 0x015A | |||||
JPEGTables = 0x015B | |||||
OPIProxy = 0x015F | |||||
JPEGProc = 0x0200 | |||||
JpegIFOffset = 0x0201 | |||||
JpegIFByteCount = 0x0202 | |||||
JpegRestartInterval = 0x0203 | |||||
JpegLosslessPredictors = 0x0205 | |||||
JpegPointTransforms = 0x0206 | |||||
JpegQTables = 0x0207 | |||||
JpegDCTables = 0x0208 | |||||
JpegACTables = 0x0209 | |||||
YCbCrCoefficients = 0x0211 | |||||
YCbCrSubSampling = 0x0212 | |||||
YCbCrPositioning = 0x0213 | |||||
ReferenceBlackWhite = 0x0214 | |||||
XMLPacket = 0x02BC | |||||
RelatedImageFileFormat = 0x1000 | |||||
RelatedImageWidth = 0x1001 | |||||
RelatedImageLength = 0x1002 | |||||
Rating = 0x4746 | |||||
RatingPercent = 0x4749 | |||||
ImageID = 0x800D | |||||
CFARepeatPatternDim = 0x828D | |||||
BatteryLevel = 0x828F | |||||
Copyright = 0x8298 | |||||
ExposureTime = 0x829A | |||||
FNumber = 0x829D | |||||
IPTCNAA = 0x83BB | |||||
ImageResources = 0x8649 | |||||
ExifOffset = 0x8769 | |||||
InterColorProfile = 0x8773 | |||||
ExposureProgram = 0x8822 | |||||
SpectralSensitivity = 0x8824 | |||||
GPSInfo = 0x8825 | |||||
ISOSpeedRatings = 0x8827 | |||||
OECF = 0x8828 | |||||
Interlace = 0x8829 | |||||
TimeZoneOffset = 0x882A | |||||
SelfTimerMode = 0x882B | |||||
SensitivityType = 0x8830 | |||||
StandardOutputSensitivity = 0x8831 | |||||
RecommendedExposureIndex = 0x8832 | |||||
ISOSpeed = 0x8833 | |||||
ISOSpeedLatitudeyyy = 0x8834 | |||||
ISOSpeedLatitudezzz = 0x8835 | |||||
ExifVersion = 0x9000 | |||||
DateTimeOriginal = 0x9003 | |||||
DateTimeDigitized = 0x9004 | |||||
OffsetTime = 0x9010 | |||||
OffsetTimeOriginal = 0x9011 | |||||
OffsetTimeDigitized = 0x9012 | |||||
ComponentsConfiguration = 0x9101 | |||||
CompressedBitsPerPixel = 0x9102 | |||||
ShutterSpeedValue = 0x9201 | |||||
ApertureValue = 0x9202 | |||||
BrightnessValue = 0x9203 | |||||
ExposureBiasValue = 0x9204 | |||||
MaxApertureValue = 0x9205 | |||||
SubjectDistance = 0x9206 | |||||
MeteringMode = 0x9207 | |||||
LightSource = 0x9208 | |||||
Flash = 0x9209 | |||||
FocalLength = 0x920A | |||||
Noise = 0x920D | |||||
ImageNumber = 0x9211 | |||||
SecurityClassification = 0x9212 | |||||
ImageHistory = 0x9213 | |||||
TIFFEPStandardID = 0x9216 | |||||
MakerNote = 0x927C | |||||
UserComment = 0x9286 | |||||
SubsecTime = 0x9290 | |||||
SubsecTimeOriginal = 0x9291 | |||||
SubsecTimeDigitized = 0x9292 | |||||
AmbientTemperature = 0x9400 | |||||
Humidity = 0x9401 | |||||
Pressure = 0x9402 | |||||
WaterDepth = 0x9403 | |||||
Acceleration = 0x9404 | |||||
CameraElevationAngle = 0x9405 | |||||
XPTitle = 0x9C9B | |||||
XPComment = 0x9C9C | |||||
XPAuthor = 0x9C9D | |||||
XPKeywords = 0x9C9E | |||||
XPSubject = 0x9C9F | |||||
FlashPixVersion = 0xA000 | |||||
ColorSpace = 0xA001 | |||||
ExifImageWidth = 0xA002 | |||||
ExifImageHeight = 0xA003 | |||||
RelatedSoundFile = 0xA004 | |||||
ExifInteroperabilityOffset = 0xA005 | |||||
FlashEnergy = 0xA20B | |||||
SpatialFrequencyResponse = 0xA20C | |||||
FocalPlaneXResolution = 0xA20E | |||||
FocalPlaneYResolution = 0xA20F | |||||
FocalPlaneResolutionUnit = 0xA210 | |||||
SubjectLocation = 0xA214 | |||||
ExposureIndex = 0xA215 | |||||
SensingMethod = 0xA217 | |||||
FileSource = 0xA300 | |||||
SceneType = 0xA301 | |||||
CFAPattern = 0xA302 | |||||
CustomRendered = 0xA401 | |||||
ExposureMode = 0xA402 | |||||
WhiteBalance = 0xA403 | |||||
DigitalZoomRatio = 0xA404 | |||||
FocalLengthIn35mmFilm = 0xA405 | |||||
SceneCaptureType = 0xA406 | |||||
GainControl = 0xA407 | |||||
Contrast = 0xA408 | |||||
Saturation = 0xA409 | |||||
Sharpness = 0xA40A | |||||
DeviceSettingDescription = 0xA40B | |||||
SubjectDistanceRange = 0xA40C | |||||
ImageUniqueID = 0xA420 | |||||
CameraOwnerName = 0xA430 | |||||
BodySerialNumber = 0xA431 | |||||
LensSpecification = 0xA432 | |||||
LensMake = 0xA433 | |||||
LensModel = 0xA434 | |||||
LensSerialNumber = 0xA435 | |||||
CompositeImage = 0xA460 | |||||
CompositeImageCount = 0xA461 | |||||
CompositeImageExposureTimes = 0xA462 | |||||
Gamma = 0xA500 | |||||
PrintImageMatching = 0xC4A5 | |||||
DNGVersion = 0xC612 | |||||
DNGBackwardVersion = 0xC613 | |||||
UniqueCameraModel = 0xC614 | |||||
LocalizedCameraModel = 0xC615 | |||||
CFAPlaneColor = 0xC616 | |||||
CFALayout = 0xC617 | |||||
LinearizationTable = 0xC618 | |||||
BlackLevelRepeatDim = 0xC619 | |||||
BlackLevel = 0xC61A | |||||
BlackLevelDeltaH = 0xC61B | |||||
BlackLevelDeltaV = 0xC61C | |||||
WhiteLevel = 0xC61D | |||||
DefaultScale = 0xC61E | |||||
DefaultCropOrigin = 0xC61F | |||||
DefaultCropSize = 0xC620 | |||||
ColorMatrix1 = 0xC621 | |||||
ColorMatrix2 = 0xC622 | |||||
CameraCalibration1 = 0xC623 | |||||
CameraCalibration2 = 0xC624 | |||||
ReductionMatrix1 = 0xC625 | |||||
ReductionMatrix2 = 0xC626 | |||||
AnalogBalance = 0xC627 | |||||
AsShotNeutral = 0xC628 | |||||
AsShotWhiteXY = 0xC629 | |||||
BaselineExposure = 0xC62A | |||||
BaselineNoise = 0xC62B | |||||
BaselineSharpness = 0xC62C | |||||
BayerGreenSplit = 0xC62D | |||||
LinearResponseLimit = 0xC62E | |||||
CameraSerialNumber = 0xC62F | |||||
LensInfo = 0xC630 | |||||
ChromaBlurRadius = 0xC631 | |||||
AntiAliasStrength = 0xC632 | |||||
ShadowScale = 0xC633 | |||||
DNGPrivateData = 0xC634 | |||||
MakerNoteSafety = 0xC635 | |||||
CalibrationIlluminant1 = 0xC65A | |||||
CalibrationIlluminant2 = 0xC65B | |||||
BestQualityScale = 0xC65C | |||||
RawDataUniqueID = 0xC65D | |||||
OriginalRawFileName = 0xC68B | |||||
OriginalRawFileData = 0xC68C | |||||
ActiveArea = 0xC68D | |||||
MaskedAreas = 0xC68E | |||||
AsShotICCProfile = 0xC68F | |||||
AsShotPreProfileMatrix = 0xC690 | |||||
CurrentICCProfile = 0xC691 | |||||
CurrentPreProfileMatrix = 0xC692 | |||||
ColorimetricReference = 0xC6BF | |||||
CameraCalibrationSignature = 0xC6F3 | |||||
ProfileCalibrationSignature = 0xC6F4 | |||||
AsShotProfileName = 0xC6F6 | |||||
NoiseReductionApplied = 0xC6F7 | |||||
ProfileName = 0xC6F8 | |||||
ProfileHueSatMapDims = 0xC6F9 | |||||
ProfileHueSatMapData1 = 0xC6FA | |||||
ProfileHueSatMapData2 = 0xC6FB | |||||
ProfileToneCurve = 0xC6FC | |||||
ProfileEmbedPolicy = 0xC6FD | |||||
ProfileCopyright = 0xC6FE | |||||
ForwardMatrix1 = 0xC714 | |||||
ForwardMatrix2 = 0xC715 | |||||
PreviewApplicationName = 0xC716 | |||||
PreviewApplicationVersion = 0xC717 | |||||
PreviewSettingsName = 0xC718 | |||||
PreviewSettingsDigest = 0xC719 | |||||
PreviewColorSpace = 0xC71A | |||||
PreviewDateTime = 0xC71B | |||||
RawImageDigest = 0xC71C | |||||
OriginalRawFileDigest = 0xC71D | |||||
SubTileBlockSize = 0xC71E | |||||
RowInterleaveFactor = 0xC71F | |||||
ProfileLookTableDims = 0xC725 | |||||
ProfileLookTableData = 0xC726 | |||||
OpcodeList1 = 0xC740 | |||||
OpcodeList2 = 0xC741 | |||||
OpcodeList3 = 0xC74E | |||||
NoiseProfile = 0xC761 | |||||
"""Maps EXIF tags to tag names.""" | |||||
TAGS = { | |||||
**{i.value: i.name for i in Base}, | |||||
0x920C: "SpatialFrequencyResponse", | |||||
0x9214: "SubjectLocation", | |||||
0x9215: "ExposureIndex", | |||||
0x828E: "CFAPattern", | |||||
0x920B: "FlashEnergy", | |||||
0x9216: "TIFF/EPStandardID", | |||||
} | |||||
class GPS(IntEnum): | |||||
GPSVersionID = 0 | |||||
GPSLatitudeRef = 1 | |||||
GPSLatitude = 2 | |||||
GPSLongitudeRef = 3 | |||||
GPSLongitude = 4 | |||||
GPSAltitudeRef = 5 | |||||
GPSAltitude = 6 | |||||
GPSTimeStamp = 7 | |||||
GPSSatellites = 8 | |||||
GPSStatus = 9 | |||||
GPSMeasureMode = 10 | |||||
GPSDOP = 11 | |||||
GPSSpeedRef = 12 | |||||
GPSSpeed = 13 | |||||
GPSTrackRef = 14 | |||||
GPSTrack = 15 | |||||
GPSImgDirectionRef = 16 | |||||
GPSImgDirection = 17 | |||||
GPSMapDatum = 18 | |||||
GPSDestLatitudeRef = 19 | |||||
GPSDestLatitude = 20 | |||||
GPSDestLongitudeRef = 21 | |||||
GPSDestLongitude = 22 | |||||
GPSDestBearingRef = 23 | |||||
GPSDestBearing = 24 | |||||
GPSDestDistanceRef = 25 | |||||
GPSDestDistance = 26 | |||||
GPSProcessingMethod = 27 | |||||
GPSAreaInformation = 28 | |||||
GPSDateStamp = 29 | |||||
GPSDifferential = 30 | |||||
GPSHPositioningError = 31 | |||||
"""Maps EXIF GPS tags to tag names.""" | |||||
GPSTAGS = {i.value: i.name for i in GPS} | |||||
class Interop(IntEnum): | |||||
InteropIndex = 1 | |||||
InteropVersion = 2 | |||||
RelatedImageFileFormat = 4096 | |||||
RelatedImageWidth = 4097 | |||||
RleatedImageHeight = 4098 | |||||
class IFD(IntEnum): | |||||
Exif = 34665 | |||||
GPSInfo = 34853 | |||||
Makernote = 37500 | |||||
Interop = 40965 | |||||
IFD1 = -1 | |||||
class LightSource(IntEnum): | |||||
Unknown = 0 | |||||
Daylight = 1 | |||||
Fluorescent = 2 | |||||
Tungsten = 3 | |||||
Flash = 4 | |||||
Fine = 9 | |||||
Cloudy = 10 | |||||
Shade = 11 | |||||
DaylightFluorescent = 12 | |||||
DayWhiteFluorescent = 13 | |||||
CoolWhiteFluorescent = 14 | |||||
WhiteFluorescent = 15 | |||||
StandardLightA = 17 | |||||
StandardLightB = 18 | |||||
StandardLightC = 19 | |||||
D55 = 20 | |||||
D65 = 21 | |||||
D75 = 22 | |||||
D50 = 23 | |||||
ISO = 24 | |||||
Other = 255 |
# | |||||
# The Python Imaging Library | |||||
# $Id$ | |||||
# | |||||
# FITS file handling | |||||
# | |||||
# Copyright (c) 1998-2003 by Fredrik Lundh | |||||
# | |||||
# See the README file for information on usage and redistribution. | |||||
# | |||||
import math | |||||
from . import Image, ImageFile | |||||
def _accept(prefix): | |||||
return prefix[:6] == b"SIMPLE" | |||||
class FitsImageFile(ImageFile.ImageFile): | |||||
format = "FITS" | |||||
format_description = "FITS" | |||||
def _open(self): | |||||
headers = {} | |||||
while True: | |||||
header = self.fp.read(80) | |||||
if not header: | |||||
msg = "Truncated FITS file" | |||||
raise OSError(msg) | |||||
keyword = header[:8].strip() | |||||
if keyword == b"END": | |||||
break | |||||
value = header[8:].split(b"/")[0].strip() | |||||
if value.startswith(b"="): | |||||
value = value[1:].strip() | |||||
if not headers and (not _accept(keyword) or value != b"T"): | |||||
msg = "Not a FITS file" | |||||
raise SyntaxError(msg) | |||||
headers[keyword] = value | |||||
naxis = int(headers[b"NAXIS"]) | |||||
if naxis == 0: | |||||
msg = "No image data" | |||||
raise ValueError(msg) | |||||
elif naxis == 1: | |||||
self._size = 1, int(headers[b"NAXIS1"]) | |||||
else: | |||||
self._size = int(headers[b"NAXIS1"]), int(headers[b"NAXIS2"]) | |||||
number_of_bits = int(headers[b"BITPIX"]) | |||||
if number_of_bits == 8: | |||||
self.mode = "L" | |||||
elif number_of_bits == 16: | |||||
self.mode = "I" | |||||
# rawmode = "I;16S" | |||||
elif number_of_bits == 32: | |||||
self.mode = "I" | |||||
elif number_of_bits in (-32, -64): | |||||
self.mode = "F" | |||||
# rawmode = "F" if number_of_bits == -32 else "F;64F" | |||||
offset = math.ceil(self.fp.tell() / 2880) * 2880 | |||||
self.tile = [("raw", (0, 0) + self.size, offset, (self.mode, 0, -1))] | |||||
# -------------------------------------------------------------------- | |||||
# Registry | |||||
Image.register_open(FitsImageFile.format, FitsImageFile, _accept) | |||||
Image.register_extensions(FitsImageFile.format, [".fit", ".fits"]) |
# | |||||
# The Python Imaging Library | |||||
# $Id$ | |||||
# | |||||
# FITS stub adapter | |||||
# | |||||
# Copyright (c) 1998-2003 by Fredrik Lundh | |||||
# | |||||
# See the README file for information on usage and redistribution. | |||||
# | |||||
from . import FitsImagePlugin, Image, ImageFile | |||||
from ._deprecate import deprecate | |||||
_handler = None | |||||
def register_handler(handler): | |||||
""" | |||||
Install application-specific FITS image handler. | |||||
:param handler: Handler object. | |||||
""" | |||||
global _handler | |||||
_handler = handler | |||||
deprecate( | |||||
"FitsStubImagePlugin", | |||||
10, | |||||
action="FITS images can now be read without " | |||||
"a handler through FitsImagePlugin instead", | |||||
) | |||||
# Override FitsImagePlugin with this handler | |||||
# for backwards compatibility | |||||
try: | |||||
Image.ID.remove(FITSStubImageFile.format) | |||||
except ValueError: | |||||
pass | |||||
Image.register_open( | |||||
FITSStubImageFile.format, FITSStubImageFile, FitsImagePlugin._accept | |||||
) | |||||
class FITSStubImageFile(ImageFile.StubImageFile): | |||||
format = FitsImagePlugin.FitsImageFile.format | |||||
format_description = FitsImagePlugin.FitsImageFile.format_description | |||||
def _open(self): | |||||
offset = self.fp.tell() | |||||
im = FitsImagePlugin.FitsImageFile(self.fp) | |||||
self._size = im.size | |||||
self.mode = im.mode | |||||
self.tile = [] | |||||
self.fp.seek(offset) | |||||
loader = self._load() | |||||
if loader: | |||||
loader.open(self) | |||||
def _load(self): | |||||
return _handler | |||||
def _save(im, fp, filename): | |||||
msg = "FITS save handler not installed" | |||||
raise OSError(msg) | |||||
# -------------------------------------------------------------------- | |||||
# Registry | |||||
Image.register_save(FITSStubImageFile.format, _save) |
# | |||||
# The Python Imaging Library. | |||||
# $Id$ | |||||
# | |||||
# FLI/FLC file handling. | |||||
# | |||||
# History: | |||||
# 95-09-01 fl Created | |||||
# 97-01-03 fl Fixed parser, setup decoder tile | |||||
# 98-07-15 fl Renamed offset attribute to avoid name clash | |||||
# | |||||
# Copyright (c) Secret Labs AB 1997-98. | |||||
# Copyright (c) Fredrik Lundh 1995-97. | |||||
# | |||||
# See the README file for information on usage and redistribution. | |||||
# | |||||
import os | |||||
from . import Image, ImageFile, ImagePalette | |||||
from ._binary import i16le as i16 | |||||
from ._binary import i32le as i32 | |||||
from ._binary import o8 | |||||
# | |||||
# decoder | |||||
def _accept(prefix): | |||||
return ( | |||||
len(prefix) >= 6 | |||||
and i16(prefix, 4) in [0xAF11, 0xAF12] | |||||
and i16(prefix, 14) in [0, 3] # flags | |||||
) | |||||
## | |||||
# Image plugin for the FLI/FLC animation format. Use the <b>seek</b> | |||||
# method to load individual frames. | |||||
class FliImageFile(ImageFile.ImageFile): | |||||
format = "FLI" | |||||
format_description = "Autodesk FLI/FLC Animation" | |||||
_close_exclusive_fp_after_loading = False | |||||
def _open(self): | |||||
# HEAD | |||||
s = self.fp.read(128) | |||||
if not (_accept(s) and s[20:22] == b"\x00\x00"): | |||||
msg = "not an FLI/FLC file" | |||||
raise SyntaxError(msg) | |||||
# frames | |||||
self.n_frames = i16(s, 6) | |||||
self.is_animated = self.n_frames > 1 | |||||
# image characteristics | |||||
self.mode = "P" | |||||
self._size = i16(s, 8), i16(s, 10) | |||||
# animation speed | |||||
duration = i32(s, 16) | |||||
magic = i16(s, 4) | |||||
if magic == 0xAF11: | |||||
duration = (duration * 1000) // 70 | |||||
self.info["duration"] = duration | |||||
# look for palette | |||||
palette = [(a, a, a) for a in range(256)] | |||||
s = self.fp.read(16) | |||||
self.__offset = 128 | |||||
if i16(s, 4) == 0xF100: | |||||
# prefix chunk; ignore it | |||||
self.__offset = self.__offset + i32(s) | |||||
s = self.fp.read(16) | |||||
if i16(s, 4) == 0xF1FA: | |||||
# look for palette chunk | |||||
number_of_subchunks = i16(s, 6) | |||||
chunk_size = None | |||||
for _ in range(number_of_subchunks): | |||||
if chunk_size is not None: | |||||
self.fp.seek(chunk_size - 6, os.SEEK_CUR) | |||||
s = self.fp.read(6) | |||||
chunk_type = i16(s, 4) | |||||
if chunk_type in (4, 11): | |||||
self._palette(palette, 2 if chunk_type == 11 else 0) | |||||
break | |||||
chunk_size = i32(s) | |||||
if not chunk_size: | |||||
break | |||||
palette = [o8(r) + o8(g) + o8(b) for (r, g, b) in palette] | |||||
self.palette = ImagePalette.raw("RGB", b"".join(palette)) | |||||
# set things up to decode first frame | |||||
self.__frame = -1 | |||||
self._fp = self.fp | |||||
self.__rewind = self.fp.tell() | |||||
self.seek(0) | |||||
def _palette(self, palette, shift): | |||||
# load palette | |||||
i = 0 | |||||
for e in range(i16(self.fp.read(2))): | |||||
s = self.fp.read(2) | |||||
i = i + s[0] | |||||
n = s[1] | |||||
if n == 0: | |||||
n = 256 | |||||
s = self.fp.read(n * 3) | |||||
for n in range(0, len(s), 3): | |||||
r = s[n] << shift | |||||
g = s[n + 1] << shift | |||||
b = s[n + 2] << shift | |||||
palette[i] = (r, g, b) | |||||
i += 1 | |||||
def seek(self, frame): | |||||
if not self._seek_check(frame): | |||||
return | |||||
if frame < self.__frame: | |||||
self._seek(0) | |||||
for f in range(self.__frame + 1, frame + 1): | |||||
self._seek(f) | |||||
def _seek(self, frame): | |||||
if frame == 0: | |||||
self.__frame = -1 | |||||
self._fp.seek(self.__rewind) | |||||
self.__offset = 128 | |||||
else: | |||||
# ensure that the previous frame was loaded | |||||
self.load() | |||||
if frame != self.__frame + 1: | |||||
msg = f"cannot seek to frame {frame}" | |||||
raise ValueError(msg) | |||||
self.__frame = frame | |||||
# move to next frame | |||||
self.fp = self._fp | |||||
self.fp.seek(self.__offset) | |||||
s = self.fp.read(4) | |||||
if not s: | |||||
raise EOFError | |||||
framesize = i32(s) | |||||
self.decodermaxblock = framesize | |||||
self.tile = [("fli", (0, 0) + self.size, self.__offset, None)] | |||||
self.__offset += framesize | |||||
def tell(self): | |||||
return self.__frame | |||||
# | |||||
# registry | |||||
Image.register_open(FliImageFile.format, FliImageFile, _accept) | |||||
Image.register_extensions(FliImageFile.format, [".fli", ".flc"]) |
# | |||||
# The Python Imaging Library | |||||
# $Id$ | |||||
# | |||||
# base class for raster font file parsers | |||||
# | |||||
# history: | |||||
# 1997-06-05 fl created | |||||
# 1997-08-19 fl restrict image width | |||||
# | |||||
# Copyright (c) 1997-1998 by Secret Labs AB | |||||
# Copyright (c) 1997-1998 by Fredrik Lundh | |||||
# | |||||
# See the README file for information on usage and redistribution. | |||||
# | |||||
import os | |||||
from . import Image, _binary | |||||
WIDTH = 800 | |||||
def puti16(fp, values): | |||||
"""Write network order (big-endian) 16-bit sequence""" | |||||
for v in values: | |||||
if v < 0: | |||||
v += 65536 | |||||
fp.write(_binary.o16be(v)) | |||||
class FontFile: | |||||
"""Base class for raster font file handlers.""" | |||||
bitmap = None | |||||
def __init__(self): | |||||
self.info = {} | |||||
self.glyph = [None] * 256 | |||||
def __getitem__(self, ix): | |||||
return self.glyph[ix] | |||||
def compile(self): | |||||
"""Create metrics and bitmap""" | |||||
if self.bitmap: | |||||
return | |||||
# create bitmap large enough to hold all data | |||||
h = w = maxwidth = 0 | |||||
lines = 1 | |||||
for glyph in self: | |||||
if glyph: | |||||
d, dst, src, im = glyph | |||||
h = max(h, src[3] - src[1]) | |||||
w = w + (src[2] - src[0]) | |||||
if w > WIDTH: | |||||
lines += 1 | |||||
w = src[2] - src[0] | |||||
maxwidth = max(maxwidth, w) | |||||
xsize = maxwidth | |||||
ysize = lines * h | |||||
if xsize == 0 and ysize == 0: | |||||
return "" | |||||
self.ysize = h | |||||
# paste glyphs into bitmap | |||||
self.bitmap = Image.new("1", (xsize, ysize)) | |||||
self.metrics = [None] * 256 | |||||
x = y = 0 | |||||
for i in range(256): | |||||
glyph = self[i] | |||||
if glyph: | |||||
d, dst, src, im = glyph | |||||
xx = src[2] - src[0] | |||||
# yy = src[3] - src[1] | |||||
x0, y0 = x, y | |||||
x = x + xx | |||||
if x > WIDTH: | |||||
x, y = 0, y + h | |||||
x0, y0 = x, y | |||||
x = xx | |||||
s = src[0] + x0, src[1] + y0, src[2] + x0, src[3] + y0 | |||||
self.bitmap.paste(im.crop(src), s) | |||||
self.metrics[i] = d, dst, s | |||||
def save(self, filename): | |||||
"""Save font""" | |||||
self.compile() | |||||
# font data | |||||
self.bitmap.save(os.path.splitext(filename)[0] + ".pbm", "PNG") | |||||
# font metrics | |||||
with open(os.path.splitext(filename)[0] + ".pil", "wb") as fp: | |||||
fp.write(b"PILfont\n") | |||||
fp.write(f";;;;;;{self.ysize};\n".encode("ascii")) # HACK!!! | |||||
fp.write(b"DATA\n") | |||||
for id in range(256): | |||||
m = self.metrics[id] | |||||
if not m: | |||||
puti16(fp, [0] * 10) | |||||
else: | |||||
puti16(fp, m[0] + m[1] + m[2]) |
# | |||||
# THIS IS WORK IN PROGRESS | |||||
# | |||||
# The Python Imaging Library. | |||||
# $Id$ | |||||
# | |||||
# FlashPix support for PIL | |||||
# | |||||
# History: | |||||
# 97-01-25 fl Created (reads uncompressed RGB images only) | |||||
# | |||||
# Copyright (c) Secret Labs AB 1997. | |||||
# Copyright (c) Fredrik Lundh 1997. | |||||
# | |||||
# See the README file for information on usage and redistribution. | |||||
# | |||||
import olefile | |||||
from . import Image, ImageFile | |||||
from ._binary import i32le as i32 | |||||
# we map from colour field tuples to (mode, rawmode) descriptors | |||||
MODES = { | |||||
# opacity | |||||
(0x00007FFE,): ("A", "L"), | |||||
# monochrome | |||||
(0x00010000,): ("L", "L"), | |||||
(0x00018000, 0x00017FFE): ("RGBA", "LA"), | |||||
# photo YCC | |||||
(0x00020000, 0x00020001, 0x00020002): ("RGB", "YCC;P"), | |||||
(0x00028000, 0x00028001, 0x00028002, 0x00027FFE): ("RGBA", "YCCA;P"), | |||||
# standard RGB (NIFRGB) | |||||
(0x00030000, 0x00030001, 0x00030002): ("RGB", "RGB"), | |||||
(0x00038000, 0x00038001, 0x00038002, 0x00037FFE): ("RGBA", "RGBA"), | |||||
} | |||||
# | |||||
# -------------------------------------------------------------------- | |||||
def _accept(prefix): | |||||
return prefix[:8] == olefile.MAGIC | |||||
## | |||||
# Image plugin for the FlashPix images. | |||||
class FpxImageFile(ImageFile.ImageFile): | |||||
format = "FPX" | |||||
format_description = "FlashPix" | |||||
def _open(self): | |||||
# | |||||
# read the OLE directory and see if this is a likely | |||||
# to be a FlashPix file | |||||
try: | |||||
self.ole = olefile.OleFileIO(self.fp) | |||||
except OSError as e: | |||||
msg = "not an FPX file; invalid OLE file" | |||||
raise SyntaxError(msg) from e | |||||
if self.ole.root.clsid != "56616700-C154-11CE-8553-00AA00A1F95B": | |||||
msg = "not an FPX file; bad root CLSID" | |||||
raise SyntaxError(msg) | |||||
self._open_index(1) | |||||
def _open_index(self, index=1): | |||||
# | |||||
# get the Image Contents Property Set | |||||
prop = self.ole.getproperties( | |||||
[f"Data Object Store {index:06d}", "\005Image Contents"] | |||||
) | |||||
# size (highest resolution) | |||||
self._size = prop[0x1000002], prop[0x1000003] | |||||
size = max(self.size) | |||||
i = 1 | |||||
while size > 64: | |||||
size = size / 2 | |||||
i += 1 | |||||
self.maxid = i - 1 | |||||
# mode. instead of using a single field for this, flashpix | |||||
# requires you to specify the mode for each channel in each | |||||
# resolution subimage, and leaves it to the decoder to make | |||||
# sure that they all match. for now, we'll cheat and assume | |||||
# that this is always the case. | |||||
id = self.maxid << 16 | |||||
s = prop[0x2000002 | id] | |||||
colors = [] | |||||
bands = i32(s, 4) | |||||
if bands > 4: | |||||
msg = "Invalid number of bands" | |||||
raise OSError(msg) | |||||
for i in range(bands): | |||||
# note: for now, we ignore the "uncalibrated" flag | |||||
colors.append(i32(s, 8 + i * 4) & 0x7FFFFFFF) | |||||
self.mode, self.rawmode = MODES[tuple(colors)] | |||||
# load JPEG tables, if any | |||||
self.jpeg = {} | |||||
for i in range(256): | |||||
id = 0x3000001 | (i << 16) | |||||
if id in prop: | |||||
self.jpeg[i] = prop[id] | |||||
self._open_subimage(1, self.maxid) | |||||
def _open_subimage(self, index=1, subimage=0): | |||||
# | |||||
# setup tile descriptors for a given subimage | |||||
stream = [ | |||||
f"Data Object Store {index:06d}", | |||||
f"Resolution {subimage:04d}", | |||||
"Subimage 0000 Header", | |||||
] | |||||
fp = self.ole.openstream(stream) | |||||
# skip prefix | |||||
fp.read(28) | |||||
# header stream | |||||
s = fp.read(36) | |||||
size = i32(s, 4), i32(s, 8) | |||||
# tilecount = i32(s, 12) | |||||
tilesize = i32(s, 16), i32(s, 20) | |||||
# channels = i32(s, 24) | |||||
offset = i32(s, 28) | |||||
length = i32(s, 32) | |||||
if size != self.size: | |||||
msg = "subimage mismatch" | |||||
raise OSError(msg) | |||||
# get tile descriptors | |||||
fp.seek(28 + offset) | |||||
s = fp.read(i32(s, 12) * length) | |||||
x = y = 0 | |||||
xsize, ysize = size | |||||
xtile, ytile = tilesize | |||||
self.tile = [] | |||||
for i in range(0, len(s), length): | |||||
x1 = min(xsize, x + xtile) | |||||
y1 = min(ysize, y + ytile) | |||||
compression = i32(s, i + 8) | |||||
if compression == 0: | |||||
self.tile.append( | |||||
( | |||||
"raw", | |||||
(x, y, x1, y1), | |||||
i32(s, i) + 28, | |||||
(self.rawmode,), | |||||
) | |||||
) | |||||
elif compression == 1: | |||||
# FIXME: the fill decoder is not implemented | |||||
self.tile.append( | |||||
( | |||||
"fill", | |||||
(x, y, x1, y1), | |||||
i32(s, i) + 28, | |||||
(self.rawmode, s[12:16]), | |||||
) | |||||
) | |||||
elif compression == 2: | |||||
internal_color_conversion = s[14] | |||||
jpeg_tables = s[15] | |||||
rawmode = self.rawmode | |||||
if internal_color_conversion: | |||||
# The image is stored as usual (usually YCbCr). | |||||
if rawmode == "RGBA": | |||||
# For "RGBA", data is stored as YCbCrA based on | |||||
# negative RGB. The following trick works around | |||||
# this problem : | |||||
jpegmode, rawmode = "YCbCrK", "CMYK" | |||||
else: | |||||
jpegmode = None # let the decoder decide | |||||
else: | |||||
# The image is stored as defined by rawmode | |||||
jpegmode = rawmode | |||||
self.tile.append( | |||||
( | |||||
"jpeg", | |||||
(x, y, x1, y1), | |||||
i32(s, i) + 28, | |||||
(rawmode, jpegmode), | |||||
) | |||||
) | |||||
# FIXME: jpeg tables are tile dependent; the prefix | |||||
# data must be placed in the tile descriptor itself! | |||||
if jpeg_tables: | |||||
self.tile_prefix = self.jpeg[jpeg_tables] | |||||
else: | |||||
msg = "unknown/invalid compression" | |||||
raise OSError(msg) | |||||
x = x + xtile | |||||
if x >= xsize: | |||||
x, y = 0, y + ytile | |||||
if y >= ysize: | |||||
break # isn't really required | |||||
self.stream = stream | |||||
self.fp = None | |||||
def load(self): | |||||
if not self.fp: | |||||
self.fp = self.ole.openstream(self.stream[:2] + ["Subimage 0000 Data"]) | |||||
return ImageFile.ImageFile.load(self) | |||||
def close(self): | |||||
self.ole.close() | |||||
super().close() | |||||
def __exit__(self, *args): | |||||
self.ole.close() | |||||
super().__exit__() | |||||
# | |||||
# -------------------------------------------------------------------- | |||||
Image.register_open(FpxImageFile.format, FpxImageFile, _accept) | |||||
Image.register_extension(FpxImageFile.format, ".fpx") |
""" | |||||
A Pillow loader for .ftc and .ftu files (FTEX) | |||||
Jerome Leclanche <jerome@leclan.ch> | |||||
The contents of this file are hereby released in the public domain (CC0) | |||||
Full text of the CC0 license: | |||||
https://creativecommons.org/publicdomain/zero/1.0/ | |||||
Independence War 2: Edge Of Chaos - Texture File Format - 16 October 2001 | |||||
The textures used for 3D objects in Independence War 2: Edge Of Chaos are in a | |||||
packed custom format called FTEX. This file format uses file extensions FTC | |||||
and FTU. | |||||
* FTC files are compressed textures (using standard texture compression). | |||||
* FTU files are not compressed. | |||||
Texture File Format | |||||
The FTC and FTU texture files both use the same format. This | |||||
has the following structure: | |||||
{header} | |||||
{format_directory} | |||||
{data} | |||||
Where: | |||||
{header} = { | |||||
u32:magic, | |||||
u32:version, | |||||
u32:width, | |||||
u32:height, | |||||
u32:mipmap_count, | |||||
u32:format_count | |||||
} | |||||
* The "magic" number is "FTEX". | |||||
* "width" and "height" are the dimensions of the texture. | |||||
* "mipmap_count" is the number of mipmaps in the texture. | |||||
* "format_count" is the number of texture formats (different versions of the | |||||
same texture) in this file. | |||||
{format_directory} = format_count * { u32:format, u32:where } | |||||
The format value is 0 for DXT1 compressed textures and 1 for 24-bit RGB | |||||
uncompressed textures. | |||||
The texture data for a format starts at the position "where" in the file. | |||||
Each set of texture data in the file has the following structure: | |||||
{data} = format_count * { u32:mipmap_size, mipmap_size * { u8 } } | |||||
* "mipmap_size" is the number of bytes in that mip level. For compressed | |||||
textures this is the size of the texture data compressed with DXT1. For 24 bit | |||||
uncompressed textures, this is 3 * width * height. Following this are the image | |||||
bytes for that mipmap level. | |||||
Note: All data is stored in little-Endian (Intel) byte order. | |||||
""" | |||||
import struct | |||||
from enum import IntEnum | |||||
from io import BytesIO | |||||
from . import Image, ImageFile | |||||
from ._deprecate import deprecate | |||||
MAGIC = b"FTEX" | |||||
class Format(IntEnum): | |||||
DXT1 = 0 | |||||
UNCOMPRESSED = 1 | |||||
def __getattr__(name): | |||||
for enum, prefix in {Format: "FORMAT_"}.items(): | |||||
if name.startswith(prefix): | |||||
name = name[len(prefix) :] | |||||
if name in enum.__members__: | |||||
deprecate(f"{prefix}{name}", 10, f"{enum.__name__}.{name}") | |||||
return enum[name] | |||||
msg = f"module '{__name__}' has no attribute '{name}'" | |||||
raise AttributeError(msg) | |||||
class FtexImageFile(ImageFile.ImageFile): | |||||
format = "FTEX" | |||||
format_description = "Texture File Format (IW2:EOC)" | |||||
def _open(self): | |||||
if not _accept(self.fp.read(4)): | |||||
msg = "not an FTEX file" | |||||
raise SyntaxError(msg) | |||||
struct.unpack("<i", self.fp.read(4)) # version | |||||
self._size = struct.unpack("<2i", self.fp.read(8)) | |||||
mipmap_count, format_count = struct.unpack("<2i", self.fp.read(8)) | |||||
self.mode = "RGB" | |||||
# Only support single-format files. | |||||
# I don't know of any multi-format file. | |||||
assert format_count == 1 | |||||
format, where = struct.unpack("<2i", self.fp.read(8)) | |||||
self.fp.seek(where) | |||||
(mipmap_size,) = struct.unpack("<i", self.fp.read(4)) | |||||
data = self.fp.read(mipmap_size) | |||||
if format == Format.DXT1: | |||||
self.mode = "RGBA" | |||||
self.tile = [("bcn", (0, 0) + self.size, 0, 1)] | |||||
elif format == Format.UNCOMPRESSED: | |||||
self.tile = [("raw", (0, 0) + self.size, 0, ("RGB", 0, 1))] | |||||
else: | |||||
msg = f"Invalid texture compression format: {repr(format)}" | |||||
raise ValueError(msg) | |||||
self.fp.close() | |||||
self.fp = BytesIO(data) | |||||
def load_seek(self, pos): | |||||
pass | |||||
def _accept(prefix): | |||||
return prefix[:4] == MAGIC | |||||
Image.register_open(FtexImageFile.format, FtexImageFile, _accept) | |||||
Image.register_extensions(FtexImageFile.format, [".ftc", ".ftu"]) |
# | |||||
# The Python Imaging Library | |||||
# | |||||
# load a GIMP brush file | |||||
# | |||||
# History: | |||||
# 96-03-14 fl Created | |||||
# 16-01-08 es Version 2 | |||||
# | |||||
# Copyright (c) Secret Labs AB 1997. | |||||
# Copyright (c) Fredrik Lundh 1996. | |||||
# Copyright (c) Eric Soroos 2016. | |||||
# | |||||
# See the README file for information on usage and redistribution. | |||||
# | |||||
# | |||||
# See https://github.com/GNOME/gimp/blob/mainline/devel-docs/gbr.txt for | |||||
# format documentation. | |||||
# | |||||
# This code Interprets version 1 and 2 .gbr files. | |||||
# Version 1 files are obsolete, and should not be used for new | |||||
# brushes. | |||||
# Version 2 files are saved by GIMP v2.8 (at least) | |||||
# Version 3 files have a format specifier of 18 for 16bit floats in | |||||
# the color depth field. This is currently unsupported by Pillow. | |||||
from . import Image, ImageFile | |||||
from ._binary import i32be as i32 | |||||
def _accept(prefix): | |||||
return len(prefix) >= 8 and i32(prefix, 0) >= 20 and i32(prefix, 4) in (1, 2) | |||||
## | |||||
# Image plugin for the GIMP brush format. | |||||
class GbrImageFile(ImageFile.ImageFile): | |||||
format = "GBR" | |||||
format_description = "GIMP brush file" | |||||
def _open(self): | |||||
header_size = i32(self.fp.read(4)) | |||||
if header_size < 20: | |||||
msg = "not a GIMP brush" | |||||
raise SyntaxError(msg) | |||||
version = i32(self.fp.read(4)) | |||||
if version not in (1, 2): | |||||
msg = f"Unsupported GIMP brush version: {version}" | |||||
raise SyntaxError(msg) | |||||
width = i32(self.fp.read(4)) | |||||
height = i32(self.fp.read(4)) | |||||
color_depth = i32(self.fp.read(4)) | |||||
if width <= 0 or height <= 0: | |||||
msg = "not a GIMP brush" | |||||
raise SyntaxError(msg) | |||||
if color_depth not in (1, 4): | |||||
msg = f"Unsupported GIMP brush color depth: {color_depth}" | |||||
raise SyntaxError(msg) | |||||
if version == 1: | |||||
comment_length = header_size - 20 | |||||
else: | |||||
comment_length = header_size - 28 | |||||
magic_number = self.fp.read(4) | |||||
if magic_number != b"GIMP": | |||||
msg = "not a GIMP brush, bad magic number" | |||||
raise SyntaxError(msg) | |||||
self.info["spacing"] = i32(self.fp.read(4)) | |||||
comment = self.fp.read(comment_length)[:-1] | |||||
if color_depth == 1: | |||||
self.mode = "L" | |||||
else: | |||||
self.mode = "RGBA" | |||||
self._size = width, height | |||||
self.info["comment"] = comment | |||||
# Image might not be small | |||||
Image._decompression_bomb_check(self.size) | |||||
# Data is an uncompressed block of w * h * bytes/pixel | |||||
self._data_size = width * height * color_depth | |||||
def load(self): | |||||
if not self.im: | |||||
self.im = Image.core.new(self.mode, self.size) | |||||
self.frombytes(self.fp.read(self._data_size)) | |||||
return Image.Image.load(self) | |||||
# | |||||
# registry | |||||
Image.register_open(GbrImageFile.format, GbrImageFile, _accept) | |||||
Image.register_extension(GbrImageFile.format, ".gbr") |
# | |||||
# The Python Imaging Library. | |||||
# $Id$ | |||||
# | |||||
# GD file handling | |||||
# | |||||
# History: | |||||
# 1996-04-12 fl Created | |||||
# | |||||
# Copyright (c) 1997 by Secret Labs AB. | |||||
# Copyright (c) 1996 by Fredrik Lundh. | |||||
# | |||||
# See the README file for information on usage and redistribution. | |||||
# | |||||
""" | |||||
.. note:: | |||||
This format cannot be automatically recognized, so the | |||||
class is not registered for use with :py:func:`PIL.Image.open()`. To open a | |||||
gd file, use the :py:func:`PIL.GdImageFile.open()` function instead. | |||||
.. warning:: | |||||
THE GD FORMAT IS NOT DESIGNED FOR DATA INTERCHANGE. This | |||||
implementation is provided for convenience and demonstrational | |||||
purposes only. | |||||
""" | |||||
from . import ImageFile, ImagePalette, UnidentifiedImageError | |||||
from ._binary import i16be as i16 | |||||
from ._binary import i32be as i32 | |||||
class GdImageFile(ImageFile.ImageFile): | |||||
""" | |||||
Image plugin for the GD uncompressed format. Note that this format | |||||
is not supported by the standard :py:func:`PIL.Image.open()` function. To use | |||||
this plugin, you have to import the :py:mod:`PIL.GdImageFile` module and | |||||
use the :py:func:`PIL.GdImageFile.open()` function. | |||||
""" | |||||
format = "GD" | |||||
format_description = "GD uncompressed images" | |||||
def _open(self): | |||||
# Header | |||||
s = self.fp.read(1037) | |||||
if not i16(s) in [65534, 65535]: | |||||
msg = "Not a valid GD 2.x .gd file" | |||||
raise SyntaxError(msg) | |||||
self.mode = "L" # FIXME: "P" | |||||
self._size = i16(s, 2), i16(s, 4) | |||||
true_color = s[6] | |||||
true_color_offset = 2 if true_color else 0 | |||||
# transparency index | |||||
tindex = i32(s, 7 + true_color_offset) | |||||
if tindex < 256: | |||||
self.info["transparency"] = tindex | |||||
self.palette = ImagePalette.raw( | |||||
"XBGR", s[7 + true_color_offset + 4 : 7 + true_color_offset + 4 + 256 * 4] | |||||
) | |||||
self.tile = [ | |||||
( | |||||
"raw", | |||||
(0, 0) + self.size, | |||||
7 + true_color_offset + 4 + 256 * 4, | |||||
("L", 0, 1), | |||||
) | |||||
] | |||||
def open(fp, mode="r"): | |||||
""" | |||||
Load texture from a GD image file. | |||||
:param fp: GD file name, or an opened file handle. | |||||
:param mode: Optional mode. In this version, if the mode argument | |||||
is given, it must be "r". | |||||
:returns: An image instance. | |||||
:raises OSError: If the image could not be read. | |||||
""" | |||||
if mode != "r": | |||||
msg = "bad mode" | |||||
raise ValueError(msg) | |||||
try: | |||||
return GdImageFile(fp) | |||||
except SyntaxError as e: | |||||
msg = "cannot identify this image file" | |||||
raise UnidentifiedImageError(msg) from e |
# | |||||
# Python Imaging Library | |||||
# $Id$ | |||||
# | |||||
# stuff to read (and render) GIMP gradient files | |||||
# | |||||
# History: | |||||
# 97-08-23 fl Created | |||||
# | |||||
# Copyright (c) Secret Labs AB 1997. | |||||
# Copyright (c) Fredrik Lundh 1997. | |||||
# | |||||
# See the README file for information on usage and redistribution. | |||||
# | |||||
""" | |||||
Stuff to translate curve segments to palette values (derived from | |||||
the corresponding code in GIMP, written by Federico Mena Quintero. | |||||
See the GIMP distribution for more information.) | |||||
""" | |||||
from math import log, pi, sin, sqrt | |||||
from ._binary import o8 | |||||
EPSILON = 1e-10 | |||||
"""""" # Enable auto-doc for data member | |||||
def linear(middle, pos): | |||||
if pos <= middle: | |||||
if middle < EPSILON: | |||||
return 0.0 | |||||
else: | |||||
return 0.5 * pos / middle | |||||
else: | |||||
pos = pos - middle | |||||
middle = 1.0 - middle | |||||
if middle < EPSILON: | |||||
return 1.0 | |||||
else: | |||||
return 0.5 + 0.5 * pos / middle | |||||
def curved(middle, pos): | |||||
return pos ** (log(0.5) / log(max(middle, EPSILON))) | |||||
def sine(middle, pos): | |||||
return (sin((-pi / 2.0) + pi * linear(middle, pos)) + 1.0) / 2.0 | |||||
def sphere_increasing(middle, pos): | |||||
return sqrt(1.0 - (linear(middle, pos) - 1.0) ** 2) | |||||
def sphere_decreasing(middle, pos): | |||||
return 1.0 - sqrt(1.0 - linear(middle, pos) ** 2) | |||||
SEGMENTS = [linear, curved, sine, sphere_increasing, sphere_decreasing] | |||||
"""""" # Enable auto-doc for data member | |||||
class GradientFile: | |||||
gradient = None | |||||
def getpalette(self, entries=256): | |||||
palette = [] | |||||
ix = 0 | |||||
x0, x1, xm, rgb0, rgb1, segment = self.gradient[ix] | |||||
for i in range(entries): | |||||
x = i / (entries - 1) | |||||
while x1 < x: | |||||
ix += 1 | |||||
x0, x1, xm, rgb0, rgb1, segment = self.gradient[ix] | |||||
w = x1 - x0 | |||||
if w < EPSILON: | |||||
scale = segment(0.5, 0.5) | |||||
else: | |||||
scale = segment((xm - x0) / w, (x - x0) / w) | |||||
# expand to RGBA | |||||
r = o8(int(255 * ((rgb1[0] - rgb0[0]) * scale + rgb0[0]) + 0.5)) | |||||
g = o8(int(255 * ((rgb1[1] - rgb0[1]) * scale + rgb0[1]) + 0.5)) | |||||
b = o8(int(255 * ((rgb1[2] - rgb0[2]) * scale + rgb0[2]) + 0.5)) | |||||
a = o8(int(255 * ((rgb1[3] - rgb0[3]) * scale + rgb0[3]) + 0.5)) | |||||
# add to palette | |||||
palette.append(r + g + b + a) | |||||
return b"".join(palette), "RGBA" | |||||
class GimpGradientFile(GradientFile): | |||||
"""File handler for GIMP's gradient format.""" | |||||
def __init__(self, fp): | |||||
if fp.readline()[:13] != b"GIMP Gradient": | |||||
msg = "not a GIMP gradient file" | |||||
raise SyntaxError(msg) | |||||
line = fp.readline() | |||||
# GIMP 1.2 gradient files don't contain a name, but GIMP 1.3 files do | |||||
if line.startswith(b"Name: "): | |||||
line = fp.readline().strip() | |||||
count = int(line) | |||||
gradient = [] | |||||
for i in range(count): | |||||
s = fp.readline().split() | |||||
w = [float(x) for x in s[:11]] | |||||
x0, x1 = w[0], w[2] | |||||
xm = w[1] | |||||
rgb0 = w[3:7] | |||||
rgb1 = w[7:11] | |||||
segment = SEGMENTS[int(s[11])] | |||||
cspace = int(s[12]) | |||||
if cspace != 0: | |||||
msg = "cannot handle HSV colour space" | |||||
raise OSError(msg) | |||||
gradient.append((x0, x1, xm, rgb0, rgb1, segment)) | |||||
self.gradient = gradient |
# | |||||
# Python Imaging Library | |||||
# $Id$ | |||||
# | |||||
# stuff to read GIMP palette files | |||||
# | |||||
# History: | |||||
# 1997-08-23 fl Created | |||||
# 2004-09-07 fl Support GIMP 2.0 palette files. | |||||
# | |||||
# Copyright (c) Secret Labs AB 1997-2004. All rights reserved. | |||||
# Copyright (c) Fredrik Lundh 1997-2004. | |||||
# | |||||
# See the README file for information on usage and redistribution. | |||||
# | |||||
import re | |||||
from ._binary import o8 | |||||
class GimpPaletteFile: | |||||
"""File handler for GIMP's palette format.""" | |||||
rawmode = "RGB" | |||||
def __init__(self, fp): | |||||
self.palette = [o8(i) * 3 for i in range(256)] | |||||
if fp.readline()[:12] != b"GIMP Palette": | |||||
msg = "not a GIMP palette file" | |||||
raise SyntaxError(msg) | |||||
for i in range(256): | |||||
s = fp.readline() | |||||
if not s: | |||||
break | |||||
# skip fields and comment lines | |||||
if re.match(rb"\w+:|#", s): | |||||
continue | |||||
if len(s) > 100: | |||||
msg = "bad palette file" | |||||
raise SyntaxError(msg) | |||||
v = tuple(map(int, s.split()[:3])) | |||||
if len(v) != 3: | |||||
msg = "bad palette entry" | |||||
raise ValueError(msg) | |||||
self.palette[i] = o8(v[0]) + o8(v[1]) + o8(v[2]) | |||||
self.palette = b"".join(self.palette) | |||||
def getpalette(self): | |||||
return self.palette, self.rawmode |
# | |||||
# The Python Imaging Library | |||||
# $Id$ | |||||
# | |||||
# GRIB stub adapter | |||||
# | |||||
# Copyright (c) 1996-2003 by Fredrik Lundh | |||||
# | |||||
# See the README file for information on usage and redistribution. | |||||
# | |||||
from . import Image, ImageFile | |||||
_handler = None | |||||
def register_handler(handler): | |||||
""" | |||||
Install application-specific GRIB image handler. | |||||
:param handler: Handler object. | |||||
""" | |||||
global _handler | |||||
_handler = handler | |||||
# -------------------------------------------------------------------- | |||||
# Image adapter | |||||
def _accept(prefix): | |||||
return prefix[:4] == b"GRIB" and prefix[7] == 1 | |||||
class GribStubImageFile(ImageFile.StubImageFile): | |||||
format = "GRIB" | |||||
format_description = "GRIB" | |||||
def _open(self): | |||||
offset = self.fp.tell() | |||||
if not _accept(self.fp.read(8)): | |||||
msg = "Not a GRIB file" | |||||
raise SyntaxError(msg) | |||||
self.fp.seek(offset) | |||||
# make something up | |||||
self.mode = "F" | |||||
self._size = 1, 1 | |||||
loader = self._load() | |||||
if loader: | |||||
loader.open(self) | |||||
def _load(self): | |||||
return _handler | |||||
def _save(im, fp, filename): | |||||
if _handler is None or not hasattr(_handler, "save"): | |||||
msg = "GRIB save handler not installed" | |||||
raise OSError(msg) | |||||
_handler.save(im, fp, filename) | |||||
# -------------------------------------------------------------------- | |||||
# Registry | |||||
Image.register_open(GribStubImageFile.format, GribStubImageFile, _accept) | |||||
Image.register_save(GribStubImageFile.format, _save) | |||||
Image.register_extension(GribStubImageFile.format, ".grib") |
# | |||||
# The Python Imaging Library | |||||
# $Id$ | |||||
# | |||||
# HDF5 stub adapter | |||||
# | |||||
# Copyright (c) 2000-2003 by Fredrik Lundh | |||||
# | |||||
# See the README file for information on usage and redistribution. | |||||
# | |||||
from . import Image, ImageFile | |||||
_handler = None | |||||
def register_handler(handler): | |||||
""" | |||||
Install application-specific HDF5 image handler. | |||||
:param handler: Handler object. | |||||
""" | |||||
global _handler | |||||
_handler = handler | |||||
# -------------------------------------------------------------------- | |||||
# Image adapter | |||||
def _accept(prefix): | |||||
return prefix[:8] == b"\x89HDF\r\n\x1a\n" | |||||
class HDF5StubImageFile(ImageFile.StubImageFile): | |||||
format = "HDF5" | |||||
format_description = "HDF5" | |||||
def _open(self): | |||||
offset = self.fp.tell() | |||||
if not _accept(self.fp.read(8)): | |||||
msg = "Not an HDF file" | |||||
raise SyntaxError(msg) | |||||
self.fp.seek(offset) | |||||
# make something up | |||||
self.mode = "F" | |||||
self._size = 1, 1 | |||||
loader = self._load() | |||||
if loader: | |||||
loader.open(self) | |||||
def _load(self): | |||||
return _handler | |||||
def _save(im, fp, filename): | |||||
if _handler is None or not hasattr(_handler, "save"): | |||||
msg = "HDF5 save handler not installed" | |||||
raise OSError(msg) | |||||
_handler.save(im, fp, filename) | |||||
# -------------------------------------------------------------------- | |||||
# Registry | |||||
Image.register_open(HDF5StubImageFile.format, HDF5StubImageFile, _accept) | |||||
Image.register_save(HDF5StubImageFile.format, _save) | |||||
Image.register_extensions(HDF5StubImageFile.format, [".h5", ".hdf"]) |
# | |||||
# The Python Imaging Library. | |||||
# $Id$ | |||||
# | |||||
# macOS icns file decoder, based on icns.py by Bob Ippolito. | |||||
# | |||||
# history: | |||||
# 2004-10-09 fl Turned into a PIL plugin; removed 2.3 dependencies. | |||||
# 2020-04-04 Allow saving on all operating systems. | |||||
# | |||||
# Copyright (c) 2004 by Bob Ippolito. | |||||
# Copyright (c) 2004 by Secret Labs. | |||||
# Copyright (c) 2004 by Fredrik Lundh. | |||||
# Copyright (c) 2014 by Alastair Houghton. | |||||
# Copyright (c) 2020 by Pan Jing. | |||||
# | |||||
# See the README file for information on usage and redistribution. | |||||
# | |||||
import io | |||||
import os | |||||
import struct | |||||
import sys | |||||
from PIL import Image, ImageFile, PngImagePlugin, features | |||||
enable_jpeg2k = features.check_codec("jpg_2000") | |||||
if enable_jpeg2k: | |||||
from PIL import Jpeg2KImagePlugin | |||||
MAGIC = b"icns" | |||||
HEADERSIZE = 8 | |||||
def nextheader(fobj): | |||||
return struct.unpack(">4sI", fobj.read(HEADERSIZE)) | |||||
def read_32t(fobj, start_length, size): | |||||
# The 128x128 icon seems to have an extra header for some reason. | |||||
(start, length) = start_length | |||||
fobj.seek(start) | |||||
sig = fobj.read(4) | |||||
if sig != b"\x00\x00\x00\x00": | |||||
msg = "Unknown signature, expecting 0x00000000" | |||||
raise SyntaxError(msg) | |||||
return read_32(fobj, (start + 4, length - 4), size) | |||||
def read_32(fobj, start_length, size): | |||||
""" | |||||
Read a 32bit RGB icon resource. Seems to be either uncompressed or | |||||
an RLE packbits-like scheme. | |||||
""" | |||||
(start, length) = start_length | |||||
fobj.seek(start) | |||||
pixel_size = (size[0] * size[2], size[1] * size[2]) | |||||
sizesq = pixel_size[0] * pixel_size[1] | |||||
if length == sizesq * 3: | |||||
# uncompressed ("RGBRGBGB") | |||||
indata = fobj.read(length) | |||||
im = Image.frombuffer("RGB", pixel_size, indata, "raw", "RGB", 0, 1) | |||||
else: | |||||
# decode image | |||||
im = Image.new("RGB", pixel_size, None) | |||||
for band_ix in range(3): | |||||
data = [] | |||||
bytesleft = sizesq | |||||
while bytesleft > 0: | |||||
byte = fobj.read(1) | |||||
if not byte: | |||||
break | |||||
byte = byte[0] | |||||
if byte & 0x80: | |||||
blocksize = byte - 125 | |||||
byte = fobj.read(1) | |||||
for i in range(blocksize): | |||||
data.append(byte) | |||||
else: | |||||
blocksize = byte + 1 | |||||
data.append(fobj.read(blocksize)) | |||||
bytesleft -= blocksize | |||||
if bytesleft <= 0: | |||||
break | |||||
if bytesleft != 0: | |||||
msg = f"Error reading channel [{repr(bytesleft)} left]" | |||||
raise SyntaxError(msg) | |||||
band = Image.frombuffer("L", pixel_size, b"".join(data), "raw", "L", 0, 1) | |||||
im.im.putband(band.im, band_ix) | |||||
return {"RGB": im} | |||||
def read_mk(fobj, start_length, size): | |||||
# Alpha masks seem to be uncompressed | |||||
start = start_length[0] | |||||
fobj.seek(start) | |||||
pixel_size = (size[0] * size[2], size[1] * size[2]) | |||||
sizesq = pixel_size[0] * pixel_size[1] | |||||
band = Image.frombuffer("L", pixel_size, fobj.read(sizesq), "raw", "L", 0, 1) | |||||
return {"A": band} | |||||
def read_png_or_jpeg2000(fobj, start_length, size): | |||||
(start, length) = start_length | |||||
fobj.seek(start) | |||||
sig = fobj.read(12) | |||||
if sig[:8] == b"\x89PNG\x0d\x0a\x1a\x0a": | |||||
fobj.seek(start) | |||||
im = PngImagePlugin.PngImageFile(fobj) | |||||
Image._decompression_bomb_check(im.size) | |||||
return {"RGBA": im} | |||||
elif ( | |||||
sig[:4] == b"\xff\x4f\xff\x51" | |||||
or sig[:4] == b"\x0d\x0a\x87\x0a" | |||||
or sig == b"\x00\x00\x00\x0cjP \x0d\x0a\x87\x0a" | |||||
): | |||||
if not enable_jpeg2k: | |||||
msg = ( | |||||
"Unsupported icon subimage format (rebuild PIL " | |||||
"with JPEG 2000 support to fix this)" | |||||
) | |||||
raise ValueError(msg) | |||||
# j2k, jpc or j2c | |||||
fobj.seek(start) | |||||
jp2kstream = fobj.read(length) | |||||
f = io.BytesIO(jp2kstream) | |||||
im = Jpeg2KImagePlugin.Jpeg2KImageFile(f) | |||||
Image._decompression_bomb_check(im.size) | |||||
if im.mode != "RGBA": | |||||
im = im.convert("RGBA") | |||||
return {"RGBA": im} | |||||
else: | |||||
msg = "Unsupported icon subimage format" | |||||
raise ValueError(msg) | |||||
class IcnsFile: | |||||
SIZES = { | |||||
(512, 512, 2): [(b"ic10", read_png_or_jpeg2000)], | |||||
(512, 512, 1): [(b"ic09", read_png_or_jpeg2000)], | |||||
(256, 256, 2): [(b"ic14", read_png_or_jpeg2000)], | |||||
(256, 256, 1): [(b"ic08", read_png_or_jpeg2000)], | |||||
(128, 128, 2): [(b"ic13", read_png_or_jpeg2000)], | |||||
(128, 128, 1): [ | |||||
(b"ic07", read_png_or_jpeg2000), | |||||
(b"it32", read_32t), | |||||
(b"t8mk", read_mk), | |||||
], | |||||
(64, 64, 1): [(b"icp6", read_png_or_jpeg2000)], | |||||
(32, 32, 2): [(b"ic12", read_png_or_jpeg2000)], | |||||
(48, 48, 1): [(b"ih32", read_32), (b"h8mk", read_mk)], | |||||
(32, 32, 1): [ | |||||
(b"icp5", read_png_or_jpeg2000), | |||||
(b"il32", read_32), | |||||
(b"l8mk", read_mk), | |||||
], | |||||
(16, 16, 2): [(b"ic11", read_png_or_jpeg2000)], | |||||
(16, 16, 1): [ | |||||
(b"icp4", read_png_or_jpeg2000), | |||||
(b"is32", read_32), | |||||
(b"s8mk", read_mk), | |||||
], | |||||
} | |||||
def __init__(self, fobj): | |||||
""" | |||||
fobj is a file-like object as an icns resource | |||||
""" | |||||
# signature : (start, length) | |||||
self.dct = dct = {} | |||||
self.fobj = fobj | |||||
sig, filesize = nextheader(fobj) | |||||
if not _accept(sig): | |||||
msg = "not an icns file" | |||||
raise SyntaxError(msg) | |||||
i = HEADERSIZE | |||||
while i < filesize: | |||||
sig, blocksize = nextheader(fobj) | |||||
if blocksize <= 0: | |||||
msg = "invalid block header" | |||||
raise SyntaxError(msg) | |||||
i += HEADERSIZE | |||||
blocksize -= HEADERSIZE | |||||
dct[sig] = (i, blocksize) | |||||
fobj.seek(blocksize, io.SEEK_CUR) | |||||
i += blocksize | |||||
def itersizes(self): | |||||
sizes = [] | |||||
for size, fmts in self.SIZES.items(): | |||||
for fmt, reader in fmts: | |||||
if fmt in self.dct: | |||||
sizes.append(size) | |||||
break | |||||
return sizes | |||||
def bestsize(self): | |||||
sizes = self.itersizes() | |||||
if not sizes: | |||||
msg = "No 32bit icon resources found" | |||||
raise SyntaxError(msg) | |||||
return max(sizes) | |||||
def dataforsize(self, size): | |||||
""" | |||||
Get an icon resource as {channel: array}. Note that | |||||
the arrays are bottom-up like windows bitmaps and will likely | |||||
need to be flipped or transposed in some way. | |||||
""" | |||||
dct = {} | |||||
for code, reader in self.SIZES[size]: | |||||
desc = self.dct.get(code) | |||||
if desc is not None: | |||||
dct.update(reader(self.fobj, desc, size)) | |||||
return dct | |||||
def getimage(self, size=None): | |||||
if size is None: | |||||
size = self.bestsize() | |||||
if len(size) == 2: | |||||
size = (size[0], size[1], 1) | |||||
channels = self.dataforsize(size) | |||||
im = channels.get("RGBA", None) | |||||
if im: | |||||
return im | |||||
im = channels.get("RGB").copy() | |||||
try: | |||||
im.putalpha(channels["A"]) | |||||
except KeyError: | |||||
pass | |||||
return im | |||||
## | |||||
# Image plugin for Mac OS icons. | |||||
class IcnsImageFile(ImageFile.ImageFile): | |||||
""" | |||||
PIL image support for Mac OS .icns files. | |||||
Chooses the best resolution, but will possibly load | |||||
a different size image if you mutate the size attribute | |||||
before calling 'load'. | |||||
The info dictionary has a key 'sizes' that is a list | |||||
of sizes that the icns file has. | |||||
""" | |||||
format = "ICNS" | |||||
format_description = "Mac OS icns resource" | |||||
def _open(self): | |||||
self.icns = IcnsFile(self.fp) | |||||
self.mode = "RGBA" | |||||
self.info["sizes"] = self.icns.itersizes() | |||||
self.best_size = self.icns.bestsize() | |||||
self.size = ( | |||||
self.best_size[0] * self.best_size[2], | |||||
self.best_size[1] * self.best_size[2], | |||||
) | |||||
@property | |||||
def size(self): | |||||
return self._size | |||||
@size.setter | |||||
def size(self, value): | |||||
info_size = value | |||||
if info_size not in self.info["sizes"] and len(info_size) == 2: | |||||
info_size = (info_size[0], info_size[1], 1) | |||||
if ( | |||||
info_size not in self.info["sizes"] | |||||
and len(info_size) == 3 | |||||
and info_size[2] == 1 | |||||
): | |||||
simple_sizes = [ | |||||
(size[0] * size[2], size[1] * size[2]) for size in self.info["sizes"] | |||||
] | |||||
if value in simple_sizes: | |||||
info_size = self.info["sizes"][simple_sizes.index(value)] | |||||
if info_size not in self.info["sizes"]: | |||||
msg = "This is not one of the allowed sizes of this image" | |||||
raise ValueError(msg) | |||||
self._size = value | |||||
def load(self): | |||||
if len(self.size) == 3: | |||||
self.best_size = self.size | |||||
self.size = ( | |||||
self.best_size[0] * self.best_size[2], | |||||
self.best_size[1] * self.best_size[2], | |||||
) | |||||
px = Image.Image.load(self) | |||||
if self.im is not None and self.im.size == self.size: | |||||
# Already loaded | |||||
return px | |||||
self.load_prepare() | |||||
# This is likely NOT the best way to do it, but whatever. | |||||
im = self.icns.getimage(self.best_size) | |||||
# If this is a PNG or JPEG 2000, it won't be loaded yet | |||||
px = im.load() | |||||
self.im = im.im | |||||
self.mode = im.mode | |||||
self.size = im.size | |||||
return px | |||||
def _save(im, fp, filename): | |||||
""" | |||||
Saves the image as a series of PNG files, | |||||
that are then combined into a .icns file. | |||||
""" | |||||
if hasattr(fp, "flush"): | |||||
fp.flush() | |||||
sizes = { | |||||
b"ic07": 128, | |||||
b"ic08": 256, | |||||
b"ic09": 512, | |||||
b"ic10": 1024, | |||||
b"ic11": 32, | |||||
b"ic12": 64, | |||||
b"ic13": 256, | |||||
b"ic14": 512, | |||||
} | |||||
provided_images = {im.width: im for im in im.encoderinfo.get("append_images", [])} | |||||
size_streams = {} | |||||
for size in set(sizes.values()): | |||||
image = ( | |||||
provided_images[size] | |||||
if size in provided_images | |||||
else im.resize((size, size)) | |||||
) | |||||
temp = io.BytesIO() | |||||
image.save(temp, "png") | |||||
size_streams[size] = temp.getvalue() | |||||
entries = [] | |||||
for type, size in sizes.items(): | |||||
stream = size_streams[size] | |||||
entries.append( | |||||
{"type": type, "size": HEADERSIZE + len(stream), "stream": stream} | |||||
) | |||||
# Header | |||||
fp.write(MAGIC) | |||||
file_length = HEADERSIZE # Header | |||||
file_length += HEADERSIZE + 8 * len(entries) # TOC | |||||
file_length += sum(entry["size"] for entry in entries) | |||||
fp.write(struct.pack(">i", file_length)) | |||||
# TOC | |||||
fp.write(b"TOC ") | |||||
fp.write(struct.pack(">i", HEADERSIZE + len(entries) * HEADERSIZE)) | |||||
for entry in entries: | |||||
fp.write(entry["type"]) | |||||
fp.write(struct.pack(">i", entry["size"])) | |||||
# Data | |||||
for entry in entries: | |||||
fp.write(entry["type"]) | |||||
fp.write(struct.pack(">i", entry["size"])) | |||||
fp.write(entry["stream"]) | |||||
if hasattr(fp, "flush"): | |||||
fp.flush() | |||||
def _accept(prefix): | |||||
return prefix[:4] == MAGIC | |||||
Image.register_open(IcnsImageFile.format, IcnsImageFile, _accept) | |||||
Image.register_extension(IcnsImageFile.format, ".icns") | |||||
Image.register_save(IcnsImageFile.format, _save) | |||||
Image.register_mime(IcnsImageFile.format, "image/icns") | |||||
if __name__ == "__main__": | |||||
if len(sys.argv) < 2: | |||||
print("Syntax: python3 IcnsImagePlugin.py [file]") | |||||
sys.exit() | |||||
with open(sys.argv[1], "rb") as fp: | |||||
imf = IcnsImageFile(fp) | |||||
for size in imf.info["sizes"]: | |||||
imf.size = size | |||||
imf.save("out-%s-%s-%s.png" % size) | |||||
with Image.open(sys.argv[1]) as im: | |||||
im.save("out.png") | |||||
if sys.platform == "windows": | |||||
os.startfile("out.png") |
# | |||||
# The Python Imaging Library. | |||||
# $Id$ | |||||
# | |||||
# Windows Icon support for PIL | |||||
# | |||||
# History: | |||||
# 96-05-27 fl Created | |||||
# | |||||
# Copyright (c) Secret Labs AB 1997. | |||||
# Copyright (c) Fredrik Lundh 1996. | |||||
# | |||||
# See the README file for information on usage and redistribution. | |||||
# | |||||
# This plugin is a refactored version of Win32IconImagePlugin by Bryan Davis | |||||
# <casadebender@gmail.com>. | |||||
# https://code.google.com/archive/p/casadebender/wikis/Win32IconImagePlugin.wiki | |||||
# | |||||
# Icon format references: | |||||
# * https://en.wikipedia.org/wiki/ICO_(file_format) | |||||
# * https://msdn.microsoft.com/en-us/library/ms997538.aspx | |||||
import warnings | |||||
from io import BytesIO | |||||
from math import ceil, log | |||||
from . import BmpImagePlugin, Image, ImageFile, PngImagePlugin | |||||
from ._binary import i16le as i16 | |||||
from ._binary import i32le as i32 | |||||
from ._binary import o8 | |||||
from ._binary import o16le as o16 | |||||
from ._binary import o32le as o32 | |||||
# | |||||
# -------------------------------------------------------------------- | |||||
_MAGIC = b"\0\0\1\0" | |||||
def _save(im, fp, filename): | |||||
fp.write(_MAGIC) # (2+2) | |||||
bmp = im.encoderinfo.get("bitmap_format") == "bmp" | |||||
sizes = im.encoderinfo.get( | |||||
"sizes", | |||||
[(16, 16), (24, 24), (32, 32), (48, 48), (64, 64), (128, 128), (256, 256)], | |||||
) | |||||
frames = [] | |||||
provided_ims = [im] + im.encoderinfo.get("append_images", []) | |||||
width, height = im.size | |||||
for size in sorted(set(sizes)): | |||||
if size[0] > width or size[1] > height or size[0] > 256 or size[1] > 256: | |||||
continue | |||||
for provided_im in provided_ims: | |||||
if provided_im.size != size: | |||||
continue | |||||
frames.append(provided_im) | |||||
if bmp: | |||||
bits = BmpImagePlugin.SAVE[provided_im.mode][1] | |||||
bits_used = [bits] | |||||
for other_im in provided_ims: | |||||
if other_im.size != size: | |||||
continue | |||||
bits = BmpImagePlugin.SAVE[other_im.mode][1] | |||||
if bits not in bits_used: | |||||
# Another image has been supplied for this size | |||||
# with a different bit depth | |||||
frames.append(other_im) | |||||
bits_used.append(bits) | |||||
break | |||||
else: | |||||
# TODO: invent a more convenient method for proportional scalings | |||||
frame = provided_im.copy() | |||||
frame.thumbnail(size, Image.Resampling.LANCZOS, reducing_gap=None) | |||||
frames.append(frame) | |||||
fp.write(o16(len(frames))) # idCount(2) | |||||
offset = fp.tell() + len(frames) * 16 | |||||
for frame in frames: | |||||
width, height = frame.size | |||||
# 0 means 256 | |||||
fp.write(o8(width if width < 256 else 0)) # bWidth(1) | |||||
fp.write(o8(height if height < 256 else 0)) # bHeight(1) | |||||
bits, colors = BmpImagePlugin.SAVE[frame.mode][1:] if bmp else (32, 0) | |||||
fp.write(o8(colors)) # bColorCount(1) | |||||
fp.write(b"\0") # bReserved(1) | |||||
fp.write(b"\0\0") # wPlanes(2) | |||||
fp.write(o16(bits)) # wBitCount(2) | |||||
image_io = BytesIO() | |||||
if bmp: | |||||
frame.save(image_io, "dib") | |||||
if bits != 32: | |||||
and_mask = Image.new("1", size) | |||||
ImageFile._save( | |||||
and_mask, image_io, [("raw", (0, 0) + size, 0, ("1", 0, -1))] | |||||
) | |||||
else: | |||||
frame.save(image_io, "png") | |||||
image_io.seek(0) | |||||
image_bytes = image_io.read() | |||||
if bmp: | |||||
image_bytes = image_bytes[:8] + o32(height * 2) + image_bytes[12:] | |||||
bytes_len = len(image_bytes) | |||||
fp.write(o32(bytes_len)) # dwBytesInRes(4) | |||||
fp.write(o32(offset)) # dwImageOffset(4) | |||||
current = fp.tell() | |||||
fp.seek(offset) | |||||
fp.write(image_bytes) | |||||
offset = offset + bytes_len | |||||
fp.seek(current) | |||||
def _accept(prefix): | |||||
return prefix[:4] == _MAGIC | |||||
class IcoFile: | |||||
def __init__(self, buf): | |||||
""" | |||||
Parse image from file-like object containing ico file data | |||||
""" | |||||
# check magic | |||||
s = buf.read(6) | |||||
if not _accept(s): | |||||
msg = "not an ICO file" | |||||
raise SyntaxError(msg) | |||||
self.buf = buf | |||||
self.entry = [] | |||||
# Number of items in file | |||||
self.nb_items = i16(s, 4) | |||||
# Get headers for each item | |||||
for i in range(self.nb_items): | |||||
s = buf.read(16) | |||||
icon_header = { | |||||
"width": s[0], | |||||
"height": s[1], | |||||
"nb_color": s[2], # No. of colors in image (0 if >=8bpp) | |||||
"reserved": s[3], | |||||
"planes": i16(s, 4), | |||||
"bpp": i16(s, 6), | |||||
"size": i32(s, 8), | |||||
"offset": i32(s, 12), | |||||
} | |||||
# See Wikipedia | |||||
for j in ("width", "height"): | |||||
if not icon_header[j]: | |||||
icon_header[j] = 256 | |||||
# See Wikipedia notes about color depth. | |||||
# We need this just to differ images with equal sizes | |||||
icon_header["color_depth"] = ( | |||||
icon_header["bpp"] | |||||
or ( | |||||
icon_header["nb_color"] != 0 | |||||
and ceil(log(icon_header["nb_color"], 2)) | |||||
) | |||||
or 256 | |||||
) | |||||
icon_header["dim"] = (icon_header["width"], icon_header["height"]) | |||||
icon_header["square"] = icon_header["width"] * icon_header["height"] | |||||
self.entry.append(icon_header) | |||||
self.entry = sorted(self.entry, key=lambda x: x["color_depth"]) | |||||
# ICO images are usually squares | |||||
# self.entry = sorted(self.entry, key=lambda x: x['width']) | |||||
self.entry = sorted(self.entry, key=lambda x: x["square"]) | |||||
self.entry.reverse() | |||||
def sizes(self): | |||||
""" | |||||
Get a list of all available icon sizes and color depths. | |||||
""" | |||||
return {(h["width"], h["height"]) for h in self.entry} | |||||
def getentryindex(self, size, bpp=False): | |||||
for i, h in enumerate(self.entry): | |||||
if size == h["dim"] and (bpp is False or bpp == h["color_depth"]): | |||||
return i | |||||
return 0 | |||||
def getimage(self, size, bpp=False): | |||||
""" | |||||
Get an image from the icon | |||||
""" | |||||
return self.frame(self.getentryindex(size, bpp)) | |||||
def frame(self, idx): | |||||
""" | |||||
Get an image from frame idx | |||||
""" | |||||
header = self.entry[idx] | |||||
self.buf.seek(header["offset"]) | |||||
data = self.buf.read(8) | |||||
self.buf.seek(header["offset"]) | |||||
if data[:8] == PngImagePlugin._MAGIC: | |||||
# png frame | |||||
im = PngImagePlugin.PngImageFile(self.buf) | |||||
Image._decompression_bomb_check(im.size) | |||||
else: | |||||
# XOR + AND mask bmp frame | |||||
im = BmpImagePlugin.DibImageFile(self.buf) | |||||
Image._decompression_bomb_check(im.size) | |||||
# change tile dimension to only encompass XOR image | |||||
im._size = (im.size[0], int(im.size[1] / 2)) | |||||
d, e, o, a = im.tile[0] | |||||
im.tile[0] = d, (0, 0) + im.size, o, a | |||||
# figure out where AND mask image starts | |||||
bpp = header["bpp"] | |||||
if 32 == bpp: | |||||
# 32-bit color depth icon image allows semitransparent areas | |||||
# PIL's DIB format ignores transparency bits, recover them. | |||||
# The DIB is packed in BGRX byte order where X is the alpha | |||||
# channel. | |||||
# Back up to start of bmp data | |||||
self.buf.seek(o) | |||||
# extract every 4th byte (eg. 3,7,11,15,...) | |||||
alpha_bytes = self.buf.read(im.size[0] * im.size[1] * 4)[3::4] | |||||
# convert to an 8bpp grayscale image | |||||
mask = Image.frombuffer( | |||||
"L", # 8bpp | |||||
im.size, # (w, h) | |||||
alpha_bytes, # source chars | |||||
"raw", # raw decoder | |||||
("L", 0, -1), # 8bpp inverted, unpadded, reversed | |||||
) | |||||
else: | |||||
# get AND image from end of bitmap | |||||
w = im.size[0] | |||||
if (w % 32) > 0: | |||||
# bitmap row data is aligned to word boundaries | |||||
w += 32 - (im.size[0] % 32) | |||||
# the total mask data is | |||||
# padded row size * height / bits per char | |||||
total_bytes = int((w * im.size[1]) / 8) | |||||
and_mask_offset = header["offset"] + header["size"] - total_bytes | |||||
self.buf.seek(and_mask_offset) | |||||
mask_data = self.buf.read(total_bytes) | |||||
# convert raw data to image | |||||
mask = Image.frombuffer( | |||||
"1", # 1 bpp | |||||
im.size, # (w, h) | |||||
mask_data, # source chars | |||||
"raw", # raw decoder | |||||
("1;I", int(w / 8), -1), # 1bpp inverted, padded, reversed | |||||
) | |||||
# now we have two images, im is XOR image and mask is AND image | |||||
# apply mask image as alpha channel | |||||
im = im.convert("RGBA") | |||||
im.putalpha(mask) | |||||
return im | |||||
## | |||||
# Image plugin for Windows Icon files. | |||||
class IcoImageFile(ImageFile.ImageFile): | |||||
""" | |||||
PIL read-only image support for Microsoft Windows .ico files. | |||||
By default the largest resolution image in the file will be loaded. This | |||||
can be changed by altering the 'size' attribute before calling 'load'. | |||||
The info dictionary has a key 'sizes' that is a list of the sizes available | |||||
in the icon file. | |||||
Handles classic, XP and Vista icon formats. | |||||
When saving, PNG compression is used. Support for this was only added in | |||||
Windows Vista. If you are unable to view the icon in Windows, convert the | |||||
image to "RGBA" mode before saving. | |||||
This plugin is a refactored version of Win32IconImagePlugin by Bryan Davis | |||||
<casadebender@gmail.com>. | |||||
https://code.google.com/archive/p/casadebender/wikis/Win32IconImagePlugin.wiki | |||||
""" | |||||
format = "ICO" | |||||
format_description = "Windows Icon" | |||||
def _open(self): | |||||
self.ico = IcoFile(self.fp) | |||||
self.info["sizes"] = self.ico.sizes() | |||||
self.size = self.ico.entry[0]["dim"] | |||||
self.load() | |||||
@property | |||||
def size(self): | |||||
return self._size | |||||
@size.setter | |||||
def size(self, value): | |||||
if value not in self.info["sizes"]: | |||||
msg = "This is not one of the allowed sizes of this image" | |||||
raise ValueError(msg) | |||||
self._size = value | |||||
def load(self): | |||||
if self.im is not None and self.im.size == self.size: | |||||
# Already loaded | |||||
return Image.Image.load(self) | |||||
im = self.ico.getimage(self.size) | |||||
# if tile is PNG, it won't really be loaded yet | |||||
im.load() | |||||
self.im = im.im | |||||
self.pyaccess = None | |||||
self.mode = im.mode | |||||
if im.size != self.size: | |||||
warnings.warn("Image was not the expected size") | |||||
index = self.ico.getentryindex(self.size) | |||||
sizes = list(self.info["sizes"]) | |||||
sizes[index] = im.size | |||||
self.info["sizes"] = set(sizes) | |||||
self.size = im.size | |||||
def load_seek(self): | |||||
# Flag the ImageFile.Parser so that it | |||||
# just does all the decode at the end. | |||||
pass | |||||
# | |||||
# -------------------------------------------------------------------- | |||||
Image.register_open(IcoImageFile.format, IcoImageFile, _accept) | |||||
Image.register_save(IcoImageFile.format, _save) | |||||
Image.register_extension(IcoImageFile.format, ".ico") | |||||
Image.register_mime(IcoImageFile.format, "image/x-icon") |
# | |||||
# The Python Imaging Library. | |||||
# $Id$ | |||||
# | |||||
# IFUNC IM file handling for PIL | |||||
# | |||||
# history: | |||||
# 1995-09-01 fl Created. | |||||
# 1997-01-03 fl Save palette images | |||||
# 1997-01-08 fl Added sequence support | |||||
# 1997-01-23 fl Added P and RGB save support | |||||
# 1997-05-31 fl Read floating point images | |||||
# 1997-06-22 fl Save floating point images | |||||
# 1997-08-27 fl Read and save 1-bit images | |||||
# 1998-06-25 fl Added support for RGB+LUT images | |||||
# 1998-07-02 fl Added support for YCC images | |||||
# 1998-07-15 fl Renamed offset attribute to avoid name clash | |||||
# 1998-12-29 fl Added I;16 support | |||||
# 2001-02-17 fl Use 're' instead of 'regex' (Python 2.1) (0.7) | |||||
# 2003-09-26 fl Added LA/PA support | |||||
# | |||||
# Copyright (c) 1997-2003 by Secret Labs AB. | |||||
# Copyright (c) 1995-2001 by Fredrik Lundh. | |||||
# | |||||
# See the README file for information on usage and redistribution. | |||||
# | |||||
import os | |||||
import re | |||||
from . import Image, ImageFile, ImagePalette | |||||
# -------------------------------------------------------------------- | |||||
# Standard tags | |||||
COMMENT = "Comment" | |||||
DATE = "Date" | |||||
EQUIPMENT = "Digitalization equipment" | |||||
FRAMES = "File size (no of images)" | |||||
LUT = "Lut" | |||||
NAME = "Name" | |||||
SCALE = "Scale (x,y)" | |||||
SIZE = "Image size (x*y)" | |||||
MODE = "Image type" | |||||
TAGS = { | |||||
COMMENT: 0, | |||||
DATE: 0, | |||||
EQUIPMENT: 0, | |||||
FRAMES: 0, | |||||
LUT: 0, | |||||
NAME: 0, | |||||
SCALE: 0, | |||||
SIZE: 0, | |||||
MODE: 0, | |||||
} | |||||
OPEN = { | |||||
# ifunc93/p3cfunc formats | |||||
"0 1 image": ("1", "1"), | |||||
"L 1 image": ("1", "1"), | |||||
"Greyscale image": ("L", "L"), | |||||
"Grayscale image": ("L", "L"), | |||||
"RGB image": ("RGB", "RGB;L"), | |||||
"RLB image": ("RGB", "RLB"), | |||||
"RYB image": ("RGB", "RLB"), | |||||
"B1 image": ("1", "1"), | |||||
"B2 image": ("P", "P;2"), | |||||
"B4 image": ("P", "P;4"), | |||||
"X 24 image": ("RGB", "RGB"), | |||||
"L 32 S image": ("I", "I;32"), | |||||
"L 32 F image": ("F", "F;32"), | |||||
# old p3cfunc formats | |||||
"RGB3 image": ("RGB", "RGB;T"), | |||||
"RYB3 image": ("RGB", "RYB;T"), | |||||
# extensions | |||||
"LA image": ("LA", "LA;L"), | |||||
"PA image": ("LA", "PA;L"), | |||||
"RGBA image": ("RGBA", "RGBA;L"), | |||||
"RGBX image": ("RGBX", "RGBX;L"), | |||||
"CMYK image": ("CMYK", "CMYK;L"), | |||||
"YCC image": ("YCbCr", "YCbCr;L"), | |||||
} | |||||
# ifunc95 extensions | |||||
for i in ["8", "8S", "16", "16S", "32", "32F"]: | |||||
OPEN[f"L {i} image"] = ("F", f"F;{i}") | |||||
OPEN[f"L*{i} image"] = ("F", f"F;{i}") | |||||
for i in ["16", "16L", "16B"]: | |||||
OPEN[f"L {i} image"] = (f"I;{i}", f"I;{i}") | |||||
OPEN[f"L*{i} image"] = (f"I;{i}", f"I;{i}") | |||||
for i in ["32S"]: | |||||
OPEN[f"L {i} image"] = ("I", f"I;{i}") | |||||
OPEN[f"L*{i} image"] = ("I", f"I;{i}") | |||||
for i in range(2, 33): | |||||
OPEN[f"L*{i} image"] = ("F", f"F;{i}") | |||||
# -------------------------------------------------------------------- | |||||
# Read IM directory | |||||
split = re.compile(rb"^([A-Za-z][^:]*):[ \t]*(.*)[ \t]*$") | |||||
def number(s): | |||||
try: | |||||
return int(s) | |||||
except ValueError: | |||||
return float(s) | |||||
## | |||||
# Image plugin for the IFUNC IM file format. | |||||
class ImImageFile(ImageFile.ImageFile): | |||||
format = "IM" | |||||
format_description = "IFUNC Image Memory" | |||||
_close_exclusive_fp_after_loading = False | |||||
def _open(self): | |||||
# Quick rejection: if there's not an LF among the first | |||||
# 100 bytes, this is (probably) not a text header. | |||||
if b"\n" not in self.fp.read(100): | |||||
msg = "not an IM file" | |||||
raise SyntaxError(msg) | |||||
self.fp.seek(0) | |||||
n = 0 | |||||
# Default values | |||||
self.info[MODE] = "L" | |||||
self.info[SIZE] = (512, 512) | |||||
self.info[FRAMES] = 1 | |||||
self.rawmode = "L" | |||||
while True: | |||||
s = self.fp.read(1) | |||||
# Some versions of IFUNC uses \n\r instead of \r\n... | |||||
if s == b"\r": | |||||
continue | |||||
if not s or s == b"\0" or s == b"\x1A": | |||||
break | |||||
# FIXME: this may read whole file if not a text file | |||||
s = s + self.fp.readline() | |||||
if len(s) > 100: | |||||
msg = "not an IM file" | |||||
raise SyntaxError(msg) | |||||
if s[-2:] == b"\r\n": | |||||
s = s[:-2] | |||||
elif s[-1:] == b"\n": | |||||
s = s[:-1] | |||||
try: | |||||
m = split.match(s) | |||||
except re.error as e: | |||||
msg = "not an IM file" | |||||
raise SyntaxError(msg) from e | |||||
if m: | |||||
k, v = m.group(1, 2) | |||||
# Don't know if this is the correct encoding, | |||||
# but a decent guess (I guess) | |||||
k = k.decode("latin-1", "replace") | |||||
v = v.decode("latin-1", "replace") | |||||
# Convert value as appropriate | |||||
if k in [FRAMES, SCALE, SIZE]: | |||||
v = v.replace("*", ",") | |||||
v = tuple(map(number, v.split(","))) | |||||
if len(v) == 1: | |||||
v = v[0] | |||||
elif k == MODE and v in OPEN: | |||||
v, self.rawmode = OPEN[v] | |||||
# Add to dictionary. Note that COMMENT tags are | |||||
# combined into a list of strings. | |||||
if k == COMMENT: | |||||
if k in self.info: | |||||
self.info[k].append(v) | |||||
else: | |||||
self.info[k] = [v] | |||||
else: | |||||
self.info[k] = v | |||||
if k in TAGS: | |||||
n += 1 | |||||
else: | |||||
msg = "Syntax error in IM header: " + s.decode("ascii", "replace") | |||||
raise SyntaxError(msg) | |||||
if not n: | |||||
msg = "Not an IM file" | |||||
raise SyntaxError(msg) | |||||
# Basic attributes | |||||
self._size = self.info[SIZE] | |||||
self.mode = self.info[MODE] | |||||
# Skip forward to start of image data | |||||
while s and s[:1] != b"\x1A": | |||||
s = self.fp.read(1) | |||||
if not s: | |||||
msg = "File truncated" | |||||
raise SyntaxError(msg) | |||||
if LUT in self.info: | |||||
# convert lookup table to palette or lut attribute | |||||
palette = self.fp.read(768) | |||||
greyscale = 1 # greyscale palette | |||||
linear = 1 # linear greyscale palette | |||||
for i in range(256): | |||||
if palette[i] == palette[i + 256] == palette[i + 512]: | |||||
if palette[i] != i: | |||||
linear = 0 | |||||
else: | |||||
greyscale = 0 | |||||
if self.mode in ["L", "LA", "P", "PA"]: | |||||
if greyscale: | |||||
if not linear: | |||||
self.lut = list(palette[:256]) | |||||
else: | |||||
if self.mode in ["L", "P"]: | |||||
self.mode = self.rawmode = "P" | |||||
elif self.mode in ["LA", "PA"]: | |||||
self.mode = "PA" | |||||
self.rawmode = "PA;L" | |||||
self.palette = ImagePalette.raw("RGB;L", palette) | |||||
elif self.mode == "RGB": | |||||
if not greyscale or not linear: | |||||
self.lut = list(palette) | |||||
self.frame = 0 | |||||
self.__offset = offs = self.fp.tell() | |||||
self._fp = self.fp # FIXME: hack | |||||
if self.rawmode[:2] == "F;": | |||||
# ifunc95 formats | |||||
try: | |||||
# use bit decoder (if necessary) | |||||
bits = int(self.rawmode[2:]) | |||||
if bits not in [8, 16, 32]: | |||||
self.tile = [("bit", (0, 0) + self.size, offs, (bits, 8, 3, 0, -1))] | |||||
return | |||||
except ValueError: | |||||
pass | |||||
if self.rawmode in ["RGB;T", "RYB;T"]: | |||||
# Old LabEye/3PC files. Would be very surprised if anyone | |||||
# ever stumbled upon such a file ;-) | |||||
size = self.size[0] * self.size[1] | |||||
self.tile = [ | |||||
("raw", (0, 0) + self.size, offs, ("G", 0, -1)), | |||||
("raw", (0, 0) + self.size, offs + size, ("R", 0, -1)), | |||||
("raw", (0, 0) + self.size, offs + 2 * size, ("B", 0, -1)), | |||||
] | |||||
else: | |||||
# LabEye/IFUNC files | |||||
self.tile = [("raw", (0, 0) + self.size, offs, (self.rawmode, 0, -1))] | |||||
@property | |||||
def n_frames(self): | |||||
return self.info[FRAMES] | |||||
@property | |||||
def is_animated(self): | |||||
return self.info[FRAMES] > 1 | |||||
def seek(self, frame): | |||||
if not self._seek_check(frame): | |||||
return | |||||
self.frame = frame | |||||
if self.mode == "1": | |||||
bits = 1 | |||||
else: | |||||
bits = 8 * len(self.mode) | |||||
size = ((self.size[0] * bits + 7) // 8) * self.size[1] | |||||
offs = self.__offset + frame * size | |||||
self.fp = self._fp | |||||
self.tile = [("raw", (0, 0) + self.size, offs, (self.rawmode, 0, -1))] | |||||
def tell(self): | |||||
return self.frame | |||||
# | |||||
# -------------------------------------------------------------------- | |||||
# Save IM files | |||||
SAVE = { | |||||
# mode: (im type, raw mode) | |||||
"1": ("0 1", "1"), | |||||
"L": ("Greyscale", "L"), | |||||
"LA": ("LA", "LA;L"), | |||||
"P": ("Greyscale", "P"), | |||||
"PA": ("LA", "PA;L"), | |||||
"I": ("L 32S", "I;32S"), | |||||
"I;16": ("L 16", "I;16"), | |||||
"I;16L": ("L 16L", "I;16L"), | |||||
"I;16B": ("L 16B", "I;16B"), | |||||
"F": ("L 32F", "F;32F"), | |||||
"RGB": ("RGB", "RGB;L"), | |||||
"RGBA": ("RGBA", "RGBA;L"), | |||||
"RGBX": ("RGBX", "RGBX;L"), | |||||
"CMYK": ("CMYK", "CMYK;L"), | |||||
"YCbCr": ("YCC", "YCbCr;L"), | |||||
} | |||||
def _save(im, fp, filename): | |||||
try: | |||||
image_type, rawmode = SAVE[im.mode] | |||||
except KeyError as e: | |||||
msg = f"Cannot save {im.mode} images as IM" | |||||
raise ValueError(msg) from e | |||||
frames = im.encoderinfo.get("frames", 1) | |||||
fp.write(f"Image type: {image_type} image\r\n".encode("ascii")) | |||||
if filename: | |||||
# Each line must be 100 characters or less, | |||||
# or: SyntaxError("not an IM file") | |||||
# 8 characters are used for "Name: " and "\r\n" | |||||
# Keep just the filename, ditch the potentially overlong path | |||||
name, ext = os.path.splitext(os.path.basename(filename)) | |||||
name = "".join([name[: 92 - len(ext)], ext]) | |||||
fp.write(f"Name: {name}\r\n".encode("ascii")) | |||||
fp.write(("Image size (x*y): %d*%d\r\n" % im.size).encode("ascii")) | |||||
fp.write(f"File size (no of images): {frames}\r\n".encode("ascii")) | |||||
if im.mode in ["P", "PA"]: | |||||
fp.write(b"Lut: 1\r\n") | |||||
fp.write(b"\000" * (511 - fp.tell()) + b"\032") | |||||
if im.mode in ["P", "PA"]: | |||||
im_palette = im.im.getpalette("RGB", "RGB;L") | |||||
colors = len(im_palette) // 3 | |||||
palette = b"" | |||||
for i in range(3): | |||||
palette += im_palette[colors * i : colors * (i + 1)] | |||||
palette += b"\x00" * (256 - colors) | |||||
fp.write(palette) # 768 bytes | |||||
ImageFile._save(im, fp, [("raw", (0, 0) + im.size, 0, (rawmode, 0, -1))]) | |||||
# | |||||
# -------------------------------------------------------------------- | |||||
# Registry | |||||
Image.register_open(ImImageFile.format, ImImageFile) | |||||
Image.register_save(ImImageFile.format, _save) | |||||
Image.register_extension(ImImageFile.format, ".im") |
# | |||||
# The Python Imaging Library. | |||||
# $Id$ | |||||
# | |||||
# standard channel operations | |||||
# | |||||
# History: | |||||
# 1996-03-24 fl Created | |||||
# 1996-08-13 fl Added logical operations (for "1" images) | |||||
# 2000-10-12 fl Added offset method (from Image.py) | |||||
# | |||||
# Copyright (c) 1997-2000 by Secret Labs AB | |||||
# Copyright (c) 1996-2000 by Fredrik Lundh | |||||
# | |||||
# See the README file for information on usage and redistribution. | |||||
# | |||||
from . import Image | |||||
def constant(image, value): | |||||
"""Fill a channel with a given grey level. | |||||
:rtype: :py:class:`~PIL.Image.Image` | |||||
""" | |||||
return Image.new("L", image.size, value) | |||||
def duplicate(image): | |||||
"""Copy a channel. Alias for :py:meth:`PIL.Image.Image.copy`. | |||||
:rtype: :py:class:`~PIL.Image.Image` | |||||
""" | |||||
return image.copy() | |||||
def invert(image): | |||||
""" | |||||
Invert an image (channel). :: | |||||
out = MAX - image | |||||
:rtype: :py:class:`~PIL.Image.Image` | |||||
""" | |||||
image.load() | |||||
return image._new(image.im.chop_invert()) | |||||
def lighter(image1, image2): | |||||
""" | |||||
Compares the two images, pixel by pixel, and returns a new image containing | |||||
the lighter values. :: | |||||
out = max(image1, image2) | |||||
:rtype: :py:class:`~PIL.Image.Image` | |||||
""" | |||||
image1.load() | |||||
image2.load() | |||||
return image1._new(image1.im.chop_lighter(image2.im)) | |||||
def darker(image1, image2): | |||||
""" | |||||
Compares the two images, pixel by pixel, and returns a new image containing | |||||
the darker values. :: | |||||
out = min(image1, image2) | |||||
:rtype: :py:class:`~PIL.Image.Image` | |||||
""" | |||||
image1.load() | |||||
image2.load() | |||||
return image1._new(image1.im.chop_darker(image2.im)) | |||||
def difference(image1, image2): | |||||
""" | |||||
Returns the absolute value of the pixel-by-pixel difference between the two | |||||
images. :: | |||||
out = abs(image1 - image2) | |||||
:rtype: :py:class:`~PIL.Image.Image` | |||||
""" | |||||
image1.load() | |||||
image2.load() | |||||
return image1._new(image1.im.chop_difference(image2.im)) | |||||
def multiply(image1, image2): | |||||
""" | |||||
Superimposes two images on top of each other. | |||||
If you multiply an image with a solid black image, the result is black. If | |||||
you multiply with a solid white image, the image is unaffected. :: | |||||
out = image1 * image2 / MAX | |||||
:rtype: :py:class:`~PIL.Image.Image` | |||||
""" | |||||
image1.load() | |||||
image2.load() | |||||
return image1._new(image1.im.chop_multiply(image2.im)) | |||||
def screen(image1, image2): | |||||
""" | |||||
Superimposes two inverted images on top of each other. :: | |||||
out = MAX - ((MAX - image1) * (MAX - image2) / MAX) | |||||
:rtype: :py:class:`~PIL.Image.Image` | |||||
""" | |||||
image1.load() | |||||
image2.load() | |||||
return image1._new(image1.im.chop_screen(image2.im)) | |||||
def soft_light(image1, image2): | |||||
""" | |||||
Superimposes two images on top of each other using the Soft Light algorithm | |||||
:rtype: :py:class:`~PIL.Image.Image` | |||||
""" | |||||
image1.load() | |||||
image2.load() | |||||
return image1._new(image1.im.chop_soft_light(image2.im)) | |||||
def hard_light(image1, image2): | |||||
""" | |||||
Superimposes two images on top of each other using the Hard Light algorithm | |||||
:rtype: :py:class:`~PIL.Image.Image` | |||||
""" | |||||
image1.load() | |||||
image2.load() | |||||
return image1._new(image1.im.chop_hard_light(image2.im)) | |||||
def overlay(image1, image2): | |||||
""" | |||||
Superimposes two images on top of each other using the Overlay algorithm | |||||
:rtype: :py:class:`~PIL.Image.Image` | |||||
""" | |||||
image1.load() | |||||
image2.load() | |||||
return image1._new(image1.im.chop_overlay(image2.im)) | |||||
def add(image1, image2, scale=1.0, offset=0): | |||||
""" | |||||
Adds two images, dividing the result by scale and adding the | |||||
offset. If omitted, scale defaults to 1.0, and offset to 0.0. :: | |||||
out = ((image1 + image2) / scale + offset) | |||||
:rtype: :py:class:`~PIL.Image.Image` | |||||
""" | |||||
image1.load() | |||||
image2.load() | |||||
return image1._new(image1.im.chop_add(image2.im, scale, offset)) | |||||
def subtract(image1, image2, scale=1.0, offset=0): | |||||
""" | |||||
Subtracts two images, dividing the result by scale and adding the offset. | |||||
If omitted, scale defaults to 1.0, and offset to 0.0. :: | |||||
out = ((image1 - image2) / scale + offset) | |||||
:rtype: :py:class:`~PIL.Image.Image` | |||||
""" | |||||
image1.load() | |||||
image2.load() | |||||
return image1._new(image1.im.chop_subtract(image2.im, scale, offset)) | |||||
def add_modulo(image1, image2): | |||||
"""Add two images, without clipping the result. :: | |||||
out = ((image1 + image2) % MAX) | |||||
:rtype: :py:class:`~PIL.Image.Image` | |||||
""" | |||||
image1.load() | |||||
image2.load() | |||||
return image1._new(image1.im.chop_add_modulo(image2.im)) | |||||
def subtract_modulo(image1, image2): | |||||
"""Subtract two images, without clipping the result. :: | |||||
out = ((image1 - image2) % MAX) | |||||
:rtype: :py:class:`~PIL.Image.Image` | |||||
""" | |||||
image1.load() | |||||
image2.load() | |||||
return image1._new(image1.im.chop_subtract_modulo(image2.im)) | |||||
def logical_and(image1, image2): | |||||
"""Logical AND between two images. | |||||
Both of the images must have mode "1". If you would like to perform a | |||||
logical AND on an image with a mode other than "1", try | |||||
:py:meth:`~PIL.ImageChops.multiply` instead, using a black-and-white mask | |||||
as the second image. :: | |||||
out = ((image1 and image2) % MAX) | |||||
:rtype: :py:class:`~PIL.Image.Image` | |||||
""" | |||||
image1.load() | |||||
image2.load() | |||||
return image1._new(image1.im.chop_and(image2.im)) | |||||
def logical_or(image1, image2): | |||||
"""Logical OR between two images. | |||||
Both of the images must have mode "1". :: | |||||
out = ((image1 or image2) % MAX) | |||||
:rtype: :py:class:`~PIL.Image.Image` | |||||
""" | |||||
image1.load() | |||||
image2.load() | |||||
return image1._new(image1.im.chop_or(image2.im)) | |||||
def logical_xor(image1, image2): | |||||
"""Logical XOR between two images. | |||||
Both of the images must have mode "1". :: | |||||
out = ((bool(image1) != bool(image2)) % MAX) | |||||
:rtype: :py:class:`~PIL.Image.Image` | |||||
""" | |||||
image1.load() | |||||
image2.load() | |||||
return image1._new(image1.im.chop_xor(image2.im)) | |||||
def blend(image1, image2, alpha): | |||||
"""Blend images using constant transparency weight. Alias for | |||||
:py:func:`PIL.Image.blend`. | |||||
:rtype: :py:class:`~PIL.Image.Image` | |||||
""" | |||||
return Image.blend(image1, image2, alpha) | |||||
def composite(image1, image2, mask): | |||||
"""Create composite using transparency mask. Alias for | |||||
:py:func:`PIL.Image.composite`. | |||||
:rtype: :py:class:`~PIL.Image.Image` | |||||
""" | |||||
return Image.composite(image1, image2, mask) | |||||
def offset(image, xoffset, yoffset=None): | |||||
"""Returns a copy of the image where data has been offset by the given | |||||
distances. Data wraps around the edges. If ``yoffset`` is omitted, it | |||||
is assumed to be equal to ``xoffset``. | |||||
:param image: Input image. | |||||
:param xoffset: The horizontal distance. | |||||
:param yoffset: The vertical distance. If omitted, both | |||||
distances are set to the same value. | |||||
:rtype: :py:class:`~PIL.Image.Image` | |||||
""" | |||||
if yoffset is None: | |||||
yoffset = xoffset | |||||
image.load() | |||||
return image._new(image.im.offset(xoffset, yoffset)) |
# | |||||
# The Python Imaging Library | |||||
# $Id$ | |||||
# | |||||
# map CSS3-style colour description strings to RGB | |||||
# | |||||
# History: | |||||
# 2002-10-24 fl Added support for CSS-style color strings | |||||
# 2002-12-15 fl Added RGBA support | |||||
# 2004-03-27 fl Fixed remaining int() problems for Python 1.5.2 | |||||
# 2004-07-19 fl Fixed gray/grey spelling issues | |||||
# 2009-03-05 fl Fixed rounding error in grayscale calculation | |||||
# | |||||
# Copyright (c) 2002-2004 by Secret Labs AB | |||||
# Copyright (c) 2002-2004 by Fredrik Lundh | |||||
# | |||||
# See the README file for information on usage and redistribution. | |||||
# | |||||
import re | |||||
from . import Image | |||||
def getrgb(color): | |||||
""" | |||||
Convert a color string to an RGB or RGBA tuple. If the string cannot be | |||||
parsed, this function raises a :py:exc:`ValueError` exception. | |||||
.. versionadded:: 1.1.4 | |||||
:param color: A color string | |||||
:return: ``(red, green, blue[, alpha])`` | |||||
""" | |||||
if len(color) > 100: | |||||
msg = "color specifier is too long" | |||||
raise ValueError(msg) | |||||
color = color.lower() | |||||
rgb = colormap.get(color, None) | |||||
if rgb: | |||||
if isinstance(rgb, tuple): | |||||
return rgb | |||||
colormap[color] = rgb = getrgb(rgb) | |||||
return rgb | |||||
# check for known string formats | |||||
if re.match("#[a-f0-9]{3}$", color): | |||||
return int(color[1] * 2, 16), int(color[2] * 2, 16), int(color[3] * 2, 16) | |||||
if re.match("#[a-f0-9]{4}$", color): | |||||
return ( | |||||
int(color[1] * 2, 16), | |||||
int(color[2] * 2, 16), | |||||
int(color[3] * 2, 16), | |||||
int(color[4] * 2, 16), | |||||
) | |||||
if re.match("#[a-f0-9]{6}$", color): | |||||
return int(color[1:3], 16), int(color[3:5], 16), int(color[5:7], 16) | |||||
if re.match("#[a-f0-9]{8}$", color): | |||||
return ( | |||||
int(color[1:3], 16), | |||||
int(color[3:5], 16), | |||||
int(color[5:7], 16), | |||||
int(color[7:9], 16), | |||||
) | |||||
m = re.match(r"rgb\(\s*(\d+)\s*,\s*(\d+)\s*,\s*(\d+)\s*\)$", color) | |||||
if m: | |||||
return int(m.group(1)), int(m.group(2)), int(m.group(3)) | |||||
m = re.match(r"rgb\(\s*(\d+)%\s*,\s*(\d+)%\s*,\s*(\d+)%\s*\)$", color) | |||||
if m: | |||||
return ( | |||||
int((int(m.group(1)) * 255) / 100.0 + 0.5), | |||||
int((int(m.group(2)) * 255) / 100.0 + 0.5), | |||||
int((int(m.group(3)) * 255) / 100.0 + 0.5), | |||||
) | |||||
m = re.match( | |||||
r"hsl\(\s*(\d+\.?\d*)\s*,\s*(\d+\.?\d*)%\s*,\s*(\d+\.?\d*)%\s*\)$", color | |||||
) | |||||
if m: | |||||
from colorsys import hls_to_rgb | |||||
rgb = hls_to_rgb( | |||||
float(m.group(1)) / 360.0, | |||||
float(m.group(3)) / 100.0, | |||||
float(m.group(2)) / 100.0, | |||||
) | |||||
return ( | |||||
int(rgb[0] * 255 + 0.5), | |||||
int(rgb[1] * 255 + 0.5), | |||||
int(rgb[2] * 255 + 0.5), | |||||
) | |||||
m = re.match( | |||||
r"hs[bv]\(\s*(\d+\.?\d*)\s*,\s*(\d+\.?\d*)%\s*,\s*(\d+\.?\d*)%\s*\)$", color | |||||
) | |||||
if m: | |||||
from colorsys import hsv_to_rgb | |||||
rgb = hsv_to_rgb( | |||||
float(m.group(1)) / 360.0, | |||||
float(m.group(2)) / 100.0, | |||||
float(m.group(3)) / 100.0, | |||||
) | |||||
return ( | |||||
int(rgb[0] * 255 + 0.5), | |||||
int(rgb[1] * 255 + 0.5), | |||||
int(rgb[2] * 255 + 0.5), | |||||
) | |||||
m = re.match(r"rgba\(\s*(\d+)\s*,\s*(\d+)\s*,\s*(\d+)\s*,\s*(\d+)\s*\)$", color) | |||||
if m: | |||||
return int(m.group(1)), int(m.group(2)), int(m.group(3)), int(m.group(4)) | |||||
msg = f"unknown color specifier: {repr(color)}" | |||||
raise ValueError(msg) | |||||
def getcolor(color, mode): | |||||
""" | |||||
Same as :py:func:`~PIL.ImageColor.getrgb`, but converts the RGB value to a | |||||
greyscale value if ``mode`` is not color or a palette image. If the string | |||||
cannot be parsed, this function raises a :py:exc:`ValueError` exception. | |||||
.. versionadded:: 1.1.4 | |||||
:param color: A color string | |||||
:param mode: Convert result to this mode | |||||
:return: ``(graylevel[, alpha]) or (red, green, blue[, alpha])`` | |||||
""" | |||||
# same as getrgb, but converts the result to the given mode | |||||
color, alpha = getrgb(color), 255 | |||||
if len(color) == 4: | |||||
color, alpha = color[:3], color[3] | |||||
if Image.getmodebase(mode) == "L": | |||||
r, g, b = color | |||||
# ITU-R Recommendation 601-2 for nonlinear RGB | |||||
# scaled to 24 bits to match the convert's implementation. | |||||
color = (r * 19595 + g * 38470 + b * 7471 + 0x8000) >> 16 | |||||
if mode[-1] == "A": | |||||
return color, alpha | |||||
else: | |||||
if mode[-1] == "A": | |||||
return color + (alpha,) | |||||
return color | |||||
colormap = { | |||||
# X11 colour table from https://drafts.csswg.org/css-color-4/, with | |||||
# gray/grey spelling issues fixed. This is a superset of HTML 4.0 | |||||
# colour names used in CSS 1. | |||||
"aliceblue": "#f0f8ff", | |||||
"antiquewhite": "#faebd7", | |||||
"aqua": "#00ffff", | |||||
"aquamarine": "#7fffd4", | |||||
"azure": "#f0ffff", | |||||
"beige": "#f5f5dc", | |||||
"bisque": "#ffe4c4", | |||||
"black": "#000000", | |||||
"blanchedalmond": "#ffebcd", | |||||
"blue": "#0000ff", | |||||
"blueviolet": "#8a2be2", | |||||
"brown": "#a52a2a", | |||||
"burlywood": "#deb887", | |||||
"cadetblue": "#5f9ea0", | |||||
"chartreuse": "#7fff00", | |||||
"chocolate": "#d2691e", | |||||
"coral": "#ff7f50", | |||||
"cornflowerblue": "#6495ed", | |||||
"cornsilk": "#fff8dc", | |||||
"crimson": "#dc143c", | |||||
"cyan": "#00ffff", | |||||
"darkblue": "#00008b", | |||||
"darkcyan": "#008b8b", | |||||
"darkgoldenrod": "#b8860b", | |||||
"darkgray": "#a9a9a9", | |||||
"darkgrey": "#a9a9a9", | |||||
"darkgreen": "#006400", | |||||
"darkkhaki": "#bdb76b", | |||||
"darkmagenta": "#8b008b", | |||||
"darkolivegreen": "#556b2f", | |||||
"darkorange": "#ff8c00", | |||||
"darkorchid": "#9932cc", | |||||
"darkred": "#8b0000", | |||||
"darksalmon": "#e9967a", | |||||
"darkseagreen": "#8fbc8f", | |||||
"darkslateblue": "#483d8b", | |||||
"darkslategray": "#2f4f4f", | |||||
"darkslategrey": "#2f4f4f", | |||||
"darkturquoise": "#00ced1", | |||||
"darkviolet": "#9400d3", | |||||
"deeppink": "#ff1493", | |||||
"deepskyblue": "#00bfff", | |||||
"dimgray": "#696969", | |||||
"dimgrey": "#696969", | |||||
"dodgerblue": "#1e90ff", | |||||
"firebrick": "#b22222", | |||||
"floralwhite": "#fffaf0", | |||||
"forestgreen": "#228b22", | |||||
"fuchsia": "#ff00ff", | |||||
"gainsboro": "#dcdcdc", | |||||
"ghostwhite": "#f8f8ff", | |||||
"gold": "#ffd700", | |||||
"goldenrod": "#daa520", | |||||
"gray": "#808080", | |||||
"grey": "#808080", | |||||
"green": "#008000", | |||||
"greenyellow": "#adff2f", | |||||
"honeydew": "#f0fff0", | |||||
"hotpink": "#ff69b4", | |||||
"indianred": "#cd5c5c", | |||||
"indigo": "#4b0082", | |||||
"ivory": "#fffff0", | |||||
"khaki": "#f0e68c", | |||||
"lavender": "#e6e6fa", | |||||
"lavenderblush": "#fff0f5", | |||||
"lawngreen": "#7cfc00", | |||||
"lemonchiffon": "#fffacd", | |||||
"lightblue": "#add8e6", | |||||
"lightcoral": "#f08080", | |||||
"lightcyan": "#e0ffff", | |||||
"lightgoldenrodyellow": "#fafad2", | |||||
"lightgreen": "#90ee90", | |||||
"lightgray": "#d3d3d3", | |||||
"lightgrey": "#d3d3d3", | |||||
"lightpink": "#ffb6c1", | |||||
"lightsalmon": "#ffa07a", | |||||
"lightseagreen": "#20b2aa", | |||||
"lightskyblue": "#87cefa", | |||||
"lightslategray": "#778899", | |||||
"lightslategrey": "#778899", | |||||
"lightsteelblue": "#b0c4de", | |||||
"lightyellow": "#ffffe0", | |||||
"lime": "#00ff00", | |||||
"limegreen": "#32cd32", | |||||
"linen": "#faf0e6", | |||||
"magenta": "#ff00ff", | |||||
"maroon": "#800000", | |||||
"mediumaquamarine": "#66cdaa", | |||||
"mediumblue": "#0000cd", | |||||
"mediumorchid": "#ba55d3", | |||||
"mediumpurple": "#9370db", | |||||
"mediumseagreen": "#3cb371", | |||||
"mediumslateblue": "#7b68ee", | |||||
"mediumspringgreen": "#00fa9a", | |||||
"mediumturquoise": "#48d1cc", | |||||
"mediumvioletred": "#c71585", | |||||
"midnightblue": "#191970", | |||||
"mintcream": "#f5fffa", | |||||
"mistyrose": "#ffe4e1", | |||||
"moccasin": "#ffe4b5", | |||||
"navajowhite": "#ffdead", | |||||
"navy": "#000080", | |||||
"oldlace": "#fdf5e6", | |||||
"olive": "#808000", | |||||
"olivedrab": "#6b8e23", | |||||
"orange": "#ffa500", | |||||
"orangered": "#ff4500", | |||||
"orchid": "#da70d6", | |||||
"palegoldenrod": "#eee8aa", | |||||
"palegreen": "#98fb98", | |||||
"paleturquoise": "#afeeee", | |||||
"palevioletred": "#db7093", | |||||
"papayawhip": "#ffefd5", | |||||
"peachpuff": "#ffdab9", | |||||
"peru": "#cd853f", | |||||
"pink": "#ffc0cb", | |||||
"plum": "#dda0dd", | |||||
"powderblue": "#b0e0e6", | |||||
"purple": "#800080", | |||||
"rebeccapurple": "#663399", | |||||
"red": "#ff0000", | |||||
"rosybrown": "#bc8f8f", | |||||
"royalblue": "#4169e1", | |||||
"saddlebrown": "#8b4513", | |||||
"salmon": "#fa8072", | |||||
"sandybrown": "#f4a460", | |||||
"seagreen": "#2e8b57", | |||||
"seashell": "#fff5ee", | |||||
"sienna": "#a0522d", | |||||
"silver": "#c0c0c0", | |||||
"skyblue": "#87ceeb", | |||||
"slateblue": "#6a5acd", | |||||
"slategray": "#708090", | |||||
"slategrey": "#708090", | |||||
"snow": "#fffafa", | |||||
"springgreen": "#00ff7f", | |||||
"steelblue": "#4682b4", | |||||
"tan": "#d2b48c", | |||||
"teal": "#008080", | |||||
"thistle": "#d8bfd8", | |||||
"tomato": "#ff6347", | |||||
"turquoise": "#40e0d0", | |||||
"violet": "#ee82ee", | |||||
"wheat": "#f5deb3", | |||||
"white": "#ffffff", | |||||
"whitesmoke": "#f5f5f5", | |||||
"yellow": "#ffff00", | |||||
"yellowgreen": "#9acd32", | |||||
} |
# | |||||
# The Python Imaging Library | |||||
# $Id$ | |||||
# | |||||
# WCK-style drawing interface operations | |||||
# | |||||
# History: | |||||
# 2003-12-07 fl created | |||||
# 2005-05-15 fl updated; added to PIL as ImageDraw2 | |||||
# 2005-05-15 fl added text support | |||||
# 2005-05-20 fl added arc/chord/pieslice support | |||||
# | |||||
# Copyright (c) 2003-2005 by Secret Labs AB | |||||
# Copyright (c) 2003-2005 by Fredrik Lundh | |||||
# | |||||
# See the README file for information on usage and redistribution. | |||||
# | |||||
""" | |||||
(Experimental) WCK-style drawing interface operations | |||||
.. seealso:: :py:mod:`PIL.ImageDraw` | |||||
""" | |||||
import warnings | |||||
from . import Image, ImageColor, ImageDraw, ImageFont, ImagePath | |||||
from ._deprecate import deprecate | |||||
class Pen: | |||||
"""Stores an outline color and width.""" | |||||
def __init__(self, color, width=1, opacity=255): | |||||
self.color = ImageColor.getrgb(color) | |||||
self.width = width | |||||
class Brush: | |||||
"""Stores a fill color""" | |||||
def __init__(self, color, opacity=255): | |||||
self.color = ImageColor.getrgb(color) | |||||
class Font: | |||||
"""Stores a TrueType font and color""" | |||||
def __init__(self, color, file, size=12): | |||||
# FIXME: add support for bitmap fonts | |||||
self.color = ImageColor.getrgb(color) | |||||
self.font = ImageFont.truetype(file, size) | |||||
class Draw: | |||||
""" | |||||
(Experimental) WCK-style drawing interface | |||||
""" | |||||
def __init__(self, image, size=None, color=None): | |||||
if not hasattr(image, "im"): | |||||
image = Image.new(image, size, color) | |||||
self.draw = ImageDraw.Draw(image) | |||||
self.image = image | |||||
self.transform = None | |||||
def flush(self): | |||||
return self.image | |||||
def render(self, op, xy, pen, brush=None): | |||||
# handle color arguments | |||||
outline = fill = None | |||||
width = 1 | |||||
if isinstance(pen, Pen): | |||||
outline = pen.color | |||||
width = pen.width | |||||
elif isinstance(brush, Pen): | |||||
outline = brush.color | |||||
width = brush.width | |||||
if isinstance(brush, Brush): | |||||
fill = brush.color | |||||
elif isinstance(pen, Brush): | |||||
fill = pen.color | |||||
# handle transformation | |||||
if self.transform: | |||||
xy = ImagePath.Path(xy) | |||||
xy.transform(self.transform) | |||||
# render the item | |||||
if op == "line": | |||||
self.draw.line(xy, fill=outline, width=width) | |||||
else: | |||||
getattr(self.draw, op)(xy, fill=fill, outline=outline) | |||||
def settransform(self, offset): | |||||
"""Sets a transformation offset.""" | |||||
(xoffset, yoffset) = offset | |||||
self.transform = (1, 0, xoffset, 0, 1, yoffset) | |||||
def arc(self, xy, start, end, *options): | |||||
""" | |||||
Draws an arc (a portion of a circle outline) between the start and end | |||||
angles, inside the given bounding box. | |||||
.. seealso:: :py:meth:`PIL.ImageDraw.ImageDraw.arc` | |||||
""" | |||||
self.render("arc", xy, start, end, *options) | |||||
def chord(self, xy, start, end, *options): | |||||
""" | |||||
Same as :py:meth:`~PIL.ImageDraw2.Draw.arc`, but connects the end points | |||||
with a straight line. | |||||
.. seealso:: :py:meth:`PIL.ImageDraw.ImageDraw.chord` | |||||
""" | |||||
self.render("chord", xy, start, end, *options) | |||||
def ellipse(self, xy, *options): | |||||
""" | |||||
Draws an ellipse inside the given bounding box. | |||||
.. seealso:: :py:meth:`PIL.ImageDraw.ImageDraw.ellipse` | |||||
""" | |||||
self.render("ellipse", xy, *options) | |||||
def line(self, xy, *options): | |||||
""" | |||||
Draws a line between the coordinates in the ``xy`` list. | |||||
.. seealso:: :py:meth:`PIL.ImageDraw.ImageDraw.line` | |||||
""" | |||||
self.render("line", xy, *options) | |||||
def pieslice(self, xy, start, end, *options): | |||||
""" | |||||
Same as arc, but also draws straight lines between the end points and the | |||||
center of the bounding box. | |||||
.. seealso:: :py:meth:`PIL.ImageDraw.ImageDraw.pieslice` | |||||
""" | |||||
self.render("pieslice", xy, start, end, *options) | |||||
def polygon(self, xy, *options): | |||||
""" | |||||
Draws a polygon. | |||||
The polygon outline consists of straight lines between the given | |||||
coordinates, plus a straight line between the last and the first | |||||
coordinate. | |||||
.. seealso:: :py:meth:`PIL.ImageDraw.ImageDraw.polygon` | |||||
""" | |||||
self.render("polygon", xy, *options) | |||||
def rectangle(self, xy, *options): | |||||
""" | |||||
Draws a rectangle. | |||||
.. seealso:: :py:meth:`PIL.ImageDraw.ImageDraw.rectangle` | |||||
""" | |||||
self.render("rectangle", xy, *options) | |||||
def text(self, xy, text, font): | |||||
""" | |||||
Draws the string at the given position. | |||||
.. seealso:: :py:meth:`PIL.ImageDraw.ImageDraw.text` | |||||
""" | |||||
if self.transform: | |||||
xy = ImagePath.Path(xy) | |||||
xy.transform(self.transform) | |||||
self.draw.text(xy, text, font=font.font, fill=font.color) | |||||
def textsize(self, text, font): | |||||
""" | |||||
.. deprecated:: 9.2.0 | |||||
Return the size of the given string, in pixels. | |||||
.. seealso:: :py:meth:`PIL.ImageDraw.ImageDraw.textsize` | |||||
""" | |||||
deprecate("textsize", 10, "textbbox or textlength") | |||||
with warnings.catch_warnings(): | |||||
warnings.filterwarnings("ignore", category=DeprecationWarning) | |||||
return self.draw.textsize(text, font=font.font) | |||||
def textbbox(self, xy, text, font): | |||||
""" | |||||
Returns bounding box (in pixels) of given text. | |||||
:return: ``(left, top, right, bottom)`` bounding box | |||||
.. seealso:: :py:meth:`PIL.ImageDraw.ImageDraw.textbbox` | |||||
""" | |||||
if self.transform: | |||||
xy = ImagePath.Path(xy) | |||||
xy.transform(self.transform) | |||||
return self.draw.textbbox(xy, text, font=font.font) | |||||
def textlength(self, text, font): | |||||
""" | |||||
Returns length (in pixels) of given text. | |||||
This is the amount by which following text should be offset. | |||||
.. seealso:: :py:meth:`PIL.ImageDraw.ImageDraw.textlength` | |||||
""" | |||||
return self.draw.textlength(text, font=font.font) |
# | |||||
# The Python Imaging Library. | |||||
# $Id$ | |||||
# | |||||
# image enhancement classes | |||||
# | |||||
# For a background, see "Image Processing By Interpolation and | |||||
# Extrapolation", Paul Haeberli and Douglas Voorhies. Available | |||||
# at http://www.graficaobscura.com/interp/index.html | |||||
# | |||||
# History: | |||||
# 1996-03-23 fl Created | |||||
# 2009-06-16 fl Fixed mean calculation | |||||
# | |||||
# Copyright (c) Secret Labs AB 1997. | |||||
# Copyright (c) Fredrik Lundh 1996. | |||||
# | |||||
# See the README file for information on usage and redistribution. | |||||
# | |||||
from . import Image, ImageFilter, ImageStat | |||||
class _Enhance: | |||||
def enhance(self, factor): | |||||
""" | |||||
Returns an enhanced image. | |||||
:param factor: A floating point value controlling the enhancement. | |||||
Factor 1.0 always returns a copy of the original image, | |||||
lower factors mean less color (brightness, contrast, | |||||
etc), and higher values more. There are no restrictions | |||||
on this value. | |||||
:rtype: :py:class:`~PIL.Image.Image` | |||||
""" | |||||
return Image.blend(self.degenerate, self.image, factor) | |||||
class Color(_Enhance): | |||||
"""Adjust image color balance. | |||||
This class can be used to adjust the colour balance of an image, in | |||||
a manner similar to the controls on a colour TV set. An enhancement | |||||
factor of 0.0 gives a black and white image. A factor of 1.0 gives | |||||
the original image. | |||||
""" | |||||
def __init__(self, image): | |||||
self.image = image | |||||
self.intermediate_mode = "L" | |||||
if "A" in image.getbands(): | |||||
self.intermediate_mode = "LA" | |||||
self.degenerate = image.convert(self.intermediate_mode).convert(image.mode) | |||||
class Contrast(_Enhance): | |||||
"""Adjust image contrast. | |||||
This class can be used to control the contrast of an image, similar | |||||
to the contrast control on a TV set. An enhancement factor of 0.0 | |||||
gives a solid grey image. A factor of 1.0 gives the original image. | |||||
""" | |||||
def __init__(self, image): | |||||
self.image = image | |||||
mean = int(ImageStat.Stat(image.convert("L")).mean[0] + 0.5) | |||||
self.degenerate = Image.new("L", image.size, mean).convert(image.mode) | |||||
if "A" in image.getbands(): | |||||
self.degenerate.putalpha(image.getchannel("A")) | |||||
class Brightness(_Enhance): | |||||
"""Adjust image brightness. | |||||
This class can be used to control the brightness of an image. An | |||||
enhancement factor of 0.0 gives a black image. A factor of 1.0 gives the | |||||
original image. | |||||
""" | |||||
def __init__(self, image): | |||||
self.image = image | |||||
self.degenerate = Image.new(image.mode, image.size, 0) | |||||
if "A" in image.getbands(): | |||||
self.degenerate.putalpha(image.getchannel("A")) | |||||
class Sharpness(_Enhance): | |||||
"""Adjust image sharpness. | |||||
This class can be used to adjust the sharpness of an image. An | |||||
enhancement factor of 0.0 gives a blurred image, a factor of 1.0 gives the | |||||
original image, and a factor of 2.0 gives a sharpened image. | |||||
""" | |||||
def __init__(self, image): | |||||
self.image = image | |||||
self.degenerate = image.filter(ImageFilter.SMOOTH) | |||||
if "A" in image.getbands(): | |||||
self.degenerate.putalpha(image.getchannel("A")) |
# | |||||
# The Python Imaging Library. | |||||
# $Id$ | |||||
# | |||||
# base class for image file handlers | |||||
# | |||||
# history: | |||||
# 1995-09-09 fl Created | |||||
# 1996-03-11 fl Fixed load mechanism. | |||||
# 1996-04-15 fl Added pcx/xbm decoders. | |||||
# 1996-04-30 fl Added encoders. | |||||
# 1996-12-14 fl Added load helpers | |||||
# 1997-01-11 fl Use encode_to_file where possible | |||||
# 1997-08-27 fl Flush output in _save | |||||
# 1998-03-05 fl Use memory mapping for some modes | |||||
# 1999-02-04 fl Use memory mapping also for "I;16" and "I;16B" | |||||
# 1999-05-31 fl Added image parser | |||||
# 2000-10-12 fl Set readonly flag on memory-mapped images | |||||
# 2002-03-20 fl Use better messages for common decoder errors | |||||
# 2003-04-21 fl Fall back on mmap/map_buffer if map is not available | |||||
# 2003-10-30 fl Added StubImageFile class | |||||
# 2004-02-25 fl Made incremental parser more robust | |||||
# | |||||
# Copyright (c) 1997-2004 by Secret Labs AB | |||||
# Copyright (c) 1995-2004 by Fredrik Lundh | |||||
# | |||||
# See the README file for information on usage and redistribution. | |||||
# | |||||
import io | |||||
import itertools | |||||
import struct | |||||
import sys | |||||
from . import Image | |||||
from ._util import is_path | |||||
MAXBLOCK = 65536 | |||||
SAFEBLOCK = 1024 * 1024 | |||||
LOAD_TRUNCATED_IMAGES = False | |||||
"""Whether or not to load truncated image files. User code may change this.""" | |||||
ERRORS = { | |||||
-1: "image buffer overrun error", | |||||
-2: "decoding error", | |||||
-3: "unknown error", | |||||
-8: "bad configuration", | |||||
-9: "out of memory error", | |||||
} | |||||
""" | |||||
Dict of known error codes returned from :meth:`.PyDecoder.decode`, | |||||
:meth:`.PyEncoder.encode` :meth:`.PyEncoder.encode_to_pyfd` and | |||||
:meth:`.PyEncoder.encode_to_file`. | |||||
""" | |||||
# | |||||
# -------------------------------------------------------------------- | |||||
# Helpers | |||||
def raise_oserror(error): | |||||
try: | |||||
msg = Image.core.getcodecstatus(error) | |||||
except AttributeError: | |||||
msg = ERRORS.get(error) | |||||
if not msg: | |||||
msg = f"decoder error {error}" | |||||
msg += " when reading image file" | |||||
raise OSError(msg) | |||||
def _tilesort(t): | |||||
# sort on offset | |||||
return t[2] | |||||
# | |||||
# -------------------------------------------------------------------- | |||||
# ImageFile base class | |||||
class ImageFile(Image.Image): | |||||
"""Base class for image file format handlers.""" | |||||
def __init__(self, fp=None, filename=None): | |||||
super().__init__() | |||||
self._min_frame = 0 | |||||
self.custom_mimetype = None | |||||
self.tile = None | |||||
""" A list of tile descriptors, or ``None`` """ | |||||
self.readonly = 1 # until we know better | |||||
self.decoderconfig = () | |||||
self.decodermaxblock = MAXBLOCK | |||||
if is_path(fp): | |||||
# filename | |||||
self.fp = open(fp, "rb") | |||||
self.filename = fp | |||||
self._exclusive_fp = True | |||||
else: | |||||
# stream | |||||
self.fp = fp | |||||
self.filename = filename | |||||
# can be overridden | |||||
self._exclusive_fp = None | |||||
try: | |||||
try: | |||||
self._open() | |||||
except ( | |||||
IndexError, # end of data | |||||
TypeError, # end of data (ord) | |||||
KeyError, # unsupported mode | |||||
EOFError, # got header but not the first frame | |||||
struct.error, | |||||
) as v: | |||||
raise SyntaxError(v) from v | |||||
if not self.mode or self.size[0] <= 0 or self.size[1] <= 0: | |||||
msg = "not identified by this driver" | |||||
raise SyntaxError(msg) | |||||
except BaseException: | |||||
# close the file only if we have opened it this constructor | |||||
if self._exclusive_fp: | |||||
self.fp.close() | |||||
raise | |||||
def get_format_mimetype(self): | |||||
if self.custom_mimetype: | |||||
return self.custom_mimetype | |||||
if self.format is not None: | |||||
return Image.MIME.get(self.format.upper()) | |||||
def __setstate__(self, state): | |||||
self.tile = [] | |||||
super().__setstate__(state) | |||||
def verify(self): | |||||
"""Check file integrity""" | |||||
# raise exception if something's wrong. must be called | |||||
# directly after open, and closes file when finished. | |||||
if self._exclusive_fp: | |||||
self.fp.close() | |||||
self.fp = None | |||||
def load(self): | |||||
"""Load image data based on tile list""" | |||||
if self.tile is None: | |||||
msg = "cannot load this image" | |||||
raise OSError(msg) | |||||
pixel = Image.Image.load(self) | |||||
if not self.tile: | |||||
return pixel | |||||
self.map = None | |||||
use_mmap = self.filename and len(self.tile) == 1 | |||||
# As of pypy 2.1.0, memory mapping was failing here. | |||||
use_mmap = use_mmap and not hasattr(sys, "pypy_version_info") | |||||
readonly = 0 | |||||
# look for read/seek overrides | |||||
try: | |||||
read = self.load_read | |||||
# don't use mmap if there are custom read/seek functions | |||||
use_mmap = False | |||||
except AttributeError: | |||||
read = self.fp.read | |||||
try: | |||||
seek = self.load_seek | |||||
use_mmap = False | |||||
except AttributeError: | |||||
seek = self.fp.seek | |||||
if use_mmap: | |||||
# try memory mapping | |||||
decoder_name, extents, offset, args = self.tile[0] | |||||
if ( | |||||
decoder_name == "raw" | |||||
and len(args) >= 3 | |||||
and args[0] == self.mode | |||||
and args[0] in Image._MAPMODES | |||||
): | |||||
try: | |||||
# use mmap, if possible | |||||
import mmap | |||||
with open(self.filename) as fp: | |||||
self.map = mmap.mmap(fp.fileno(), 0, access=mmap.ACCESS_READ) | |||||
if offset + self.size[1] * args[1] > self.map.size(): | |||||
# buffer is not large enough | |||||
raise OSError | |||||
self.im = Image.core.map_buffer( | |||||
self.map, self.size, decoder_name, offset, args | |||||
) | |||||
readonly = 1 | |||||
# After trashing self.im, | |||||
# we might need to reload the palette data. | |||||
if self.palette: | |||||
self.palette.dirty = 1 | |||||
except (AttributeError, OSError, ImportError): | |||||
self.map = None | |||||
self.load_prepare() | |||||
err_code = -3 # initialize to unknown error | |||||
if not self.map: | |||||
# sort tiles in file order | |||||
self.tile.sort(key=_tilesort) | |||||
try: | |||||
# FIXME: This is a hack to handle TIFF's JpegTables tag. | |||||
prefix = self.tile_prefix | |||||
except AttributeError: | |||||
prefix = b"" | |||||
# Remove consecutive duplicates that only differ by their offset | |||||
self.tile = [ | |||||
list(tiles)[-1] | |||||
for _, tiles in itertools.groupby( | |||||
self.tile, lambda tile: (tile[0], tile[1], tile[3]) | |||||
) | |||||
] | |||||
for decoder_name, extents, offset, args in self.tile: | |||||
seek(offset) | |||||
decoder = Image._getdecoder( | |||||
self.mode, decoder_name, args, self.decoderconfig | |||||
) | |||||
try: | |||||
decoder.setimage(self.im, extents) | |||||
if decoder.pulls_fd: | |||||
decoder.setfd(self.fp) | |||||
err_code = decoder.decode(b"")[1] | |||||
else: | |||||
b = prefix | |||||
while True: | |||||
try: | |||||
s = read(self.decodermaxblock) | |||||
except (IndexError, struct.error) as e: | |||||
# truncated png/gif | |||||
if LOAD_TRUNCATED_IMAGES: | |||||
break | |||||
else: | |||||
msg = "image file is truncated" | |||||
raise OSError(msg) from e | |||||
if not s: # truncated jpeg | |||||
if LOAD_TRUNCATED_IMAGES: | |||||
break | |||||
else: | |||||
msg = ( | |||||
"image file is truncated " | |||||
f"({len(b)} bytes not processed)" | |||||
) | |||||
raise OSError(msg) | |||||
b = b + s | |||||
n, err_code = decoder.decode(b) | |||||
if n < 0: | |||||
break | |||||
b = b[n:] | |||||
finally: | |||||
# Need to cleanup here to prevent leaks | |||||
decoder.cleanup() | |||||
self.tile = [] | |||||
self.readonly = readonly | |||||
self.load_end() | |||||
if self._exclusive_fp and self._close_exclusive_fp_after_loading: | |||||
self.fp.close() | |||||
self.fp = None | |||||
if not self.map and not LOAD_TRUNCATED_IMAGES and err_code < 0: | |||||
# still raised if decoder fails to return anything | |||||
raise_oserror(err_code) | |||||
return Image.Image.load(self) | |||||
def load_prepare(self): | |||||
# create image memory if necessary | |||||
if not self.im or self.im.mode != self.mode or self.im.size != self.size: | |||||
self.im = Image.core.new(self.mode, self.size) | |||||
# create palette (optional) | |||||
if self.mode == "P": | |||||
Image.Image.load(self) | |||||
def load_end(self): | |||||
# may be overridden | |||||
pass | |||||
# may be defined for contained formats | |||||
# def load_seek(self, pos): | |||||
# pass | |||||
# may be defined for blocked formats (e.g. PNG) | |||||
# def load_read(self, bytes): | |||||
# pass | |||||
def _seek_check(self, frame): | |||||
if ( | |||||
frame < self._min_frame | |||||
# Only check upper limit on frames if additional seek operations | |||||
# are not required to do so | |||||
or ( | |||||
not (hasattr(self, "_n_frames") and self._n_frames is None) | |||||
and frame >= self.n_frames + self._min_frame | |||||
) | |||||
): | |||||
msg = "attempt to seek outside sequence" | |||||
raise EOFError(msg) | |||||
return self.tell() != frame | |||||
class StubImageFile(ImageFile): | |||||
""" | |||||
Base class for stub image loaders. | |||||
A stub loader is an image loader that can identify files of a | |||||
certain format, but relies on external code to load the file. | |||||
""" | |||||
def _open(self): | |||||
msg = "StubImageFile subclass must implement _open" | |||||
raise NotImplementedError(msg) | |||||
def load(self): | |||||
loader = self._load() | |||||
if loader is None: | |||||
msg = f"cannot find loader for this {self.format} file" | |||||
raise OSError(msg) | |||||
image = loader.load(self) | |||||
assert image is not None | |||||
# become the other object (!) | |||||
self.__class__ = image.__class__ | |||||
self.__dict__ = image.__dict__ | |||||
return image.load() | |||||
def _load(self): | |||||
"""(Hook) Find actual image loader.""" | |||||
msg = "StubImageFile subclass must implement _load" | |||||
raise NotImplementedError(msg) | |||||
class Parser: | |||||
""" | |||||
Incremental image parser. This class implements the standard | |||||
feed/close consumer interface. | |||||
""" | |||||
incremental = None | |||||
image = None | |||||
data = None | |||||
decoder = None | |||||
offset = 0 | |||||
finished = 0 | |||||
def reset(self): | |||||
""" | |||||
(Consumer) Reset the parser. Note that you can only call this | |||||
method immediately after you've created a parser; parser | |||||
instances cannot be reused. | |||||
""" | |||||
assert self.data is None, "cannot reuse parsers" | |||||
def feed(self, data): | |||||
""" | |||||
(Consumer) Feed data to the parser. | |||||
:param data: A string buffer. | |||||
:exception OSError: If the parser failed to parse the image file. | |||||
""" | |||||
# collect data | |||||
if self.finished: | |||||
return | |||||
if self.data is None: | |||||
self.data = data | |||||
else: | |||||
self.data = self.data + data | |||||
# parse what we have | |||||
if self.decoder: | |||||
if self.offset > 0: | |||||
# skip header | |||||
skip = min(len(self.data), self.offset) | |||||
self.data = self.data[skip:] | |||||
self.offset = self.offset - skip | |||||
if self.offset > 0 or not self.data: | |||||
return | |||||
n, e = self.decoder.decode(self.data) | |||||
if n < 0: | |||||
# end of stream | |||||
self.data = None | |||||
self.finished = 1 | |||||
if e < 0: | |||||
# decoding error | |||||
self.image = None | |||||
raise_oserror(e) | |||||
else: | |||||
# end of image | |||||
return | |||||
self.data = self.data[n:] | |||||
elif self.image: | |||||
# if we end up here with no decoder, this file cannot | |||||
# be incrementally parsed. wait until we've gotten all | |||||
# available data | |||||
pass | |||||
else: | |||||
# attempt to open this file | |||||
try: | |||||
with io.BytesIO(self.data) as fp: | |||||
im = Image.open(fp) | |||||
except OSError: | |||||
# traceback.print_exc() | |||||
pass # not enough data | |||||
else: | |||||
flag = hasattr(im, "load_seek") or hasattr(im, "load_read") | |||||
if flag or len(im.tile) != 1: | |||||
# custom load code, or multiple tiles | |||||
self.decode = None | |||||
else: | |||||
# initialize decoder | |||||
im.load_prepare() | |||||
d, e, o, a = im.tile[0] | |||||
im.tile = [] | |||||
self.decoder = Image._getdecoder(im.mode, d, a, im.decoderconfig) | |||||
self.decoder.setimage(im.im, e) | |||||
# calculate decoder offset | |||||
self.offset = o | |||||
if self.offset <= len(self.data): | |||||
self.data = self.data[self.offset :] | |||||
self.offset = 0 | |||||
self.image = im | |||||
def __enter__(self): | |||||
return self | |||||
def __exit__(self, *args): | |||||
self.close() | |||||
def close(self): | |||||
""" | |||||
(Consumer) Close the stream. | |||||
:returns: An image object. | |||||
:exception OSError: If the parser failed to parse the image file either | |||||
because it cannot be identified or cannot be | |||||
decoded. | |||||
""" | |||||
# finish decoding | |||||
if self.decoder: | |||||
# get rid of what's left in the buffers | |||||
self.feed(b"") | |||||
self.data = self.decoder = None | |||||
if not self.finished: | |||||
msg = "image was incomplete" | |||||
raise OSError(msg) | |||||
if not self.image: | |||||
msg = "cannot parse this image" | |||||
raise OSError(msg) | |||||
if self.data: | |||||
# incremental parsing not possible; reopen the file | |||||
# not that we have all data | |||||
with io.BytesIO(self.data) as fp: | |||||
try: | |||||
self.image = Image.open(fp) | |||||
finally: | |||||
self.image.load() | |||||
return self.image | |||||
# -------------------------------------------------------------------- | |||||
def _save(im, fp, tile, bufsize=0): | |||||
"""Helper to save image based on tile list | |||||
:param im: Image object. | |||||
:param fp: File object. | |||||
:param tile: Tile list. | |||||
:param bufsize: Optional buffer size | |||||
""" | |||||
im.load() | |||||
if not hasattr(im, "encoderconfig"): | |||||
im.encoderconfig = () | |||||
tile.sort(key=_tilesort) | |||||
# FIXME: make MAXBLOCK a configuration parameter | |||||
# It would be great if we could have the encoder specify what it needs | |||||
# But, it would need at least the image size in most cases. RawEncode is | |||||
# a tricky case. | |||||
bufsize = max(MAXBLOCK, bufsize, im.size[0] * 4) # see RawEncode.c | |||||
try: | |||||
fh = fp.fileno() | |||||
fp.flush() | |||||
_encode_tile(im, fp, tile, bufsize, fh) | |||||
except (AttributeError, io.UnsupportedOperation) as exc: | |||||
_encode_tile(im, fp, tile, bufsize, None, exc) | |||||
if hasattr(fp, "flush"): | |||||
fp.flush() | |||||
def _encode_tile(im, fp, tile, bufsize, fh, exc=None): | |||||
for e, b, o, a in tile: | |||||
if o > 0: | |||||
fp.seek(o) | |||||
encoder = Image._getencoder(im.mode, e, a, im.encoderconfig) | |||||
try: | |||||
encoder.setimage(im.im, b) | |||||
if encoder.pushes_fd: | |||||
encoder.setfd(fp) | |||||
errcode = encoder.encode_to_pyfd()[1] | |||||
else: | |||||
if exc: | |||||
# compress to Python file-compatible object | |||||
while True: | |||||
errcode, data = encoder.encode(bufsize)[1:] | |||||
fp.write(data) | |||||
if errcode: | |||||
break | |||||
else: | |||||
# slight speedup: compress to real file object | |||||
errcode = encoder.encode_to_file(fh, bufsize) | |||||
if errcode < 0: | |||||
msg = f"encoder error {errcode} when writing image file" | |||||
raise OSError(msg) from exc | |||||
finally: | |||||
encoder.cleanup() | |||||
def _safe_read(fp, size): | |||||
""" | |||||
Reads large blocks in a safe way. Unlike fp.read(n), this function | |||||
doesn't trust the user. If the requested size is larger than | |||||
SAFEBLOCK, the file is read block by block. | |||||
:param fp: File handle. Must implement a <b>read</b> method. | |||||
:param size: Number of bytes to read. | |||||
:returns: A string containing <i>size</i> bytes of data. | |||||
Raises an OSError if the file is truncated and the read cannot be completed | |||||
""" | |||||
if size <= 0: | |||||
return b"" | |||||
if size <= SAFEBLOCK: | |||||
data = fp.read(size) | |||||
if len(data) < size: | |||||
msg = "Truncated File Read" | |||||
raise OSError(msg) | |||||
return data | |||||
data = [] | |||||
remaining_size = size | |||||
while remaining_size > 0: | |||||
block = fp.read(min(remaining_size, SAFEBLOCK)) | |||||
if not block: | |||||
break | |||||
data.append(block) | |||||
remaining_size -= len(block) | |||||
if sum(len(d) for d in data) < size: | |||||
msg = "Truncated File Read" | |||||
raise OSError(msg) | |||||
return b"".join(data) | |||||
class PyCodecState: | |||||
def __init__(self): | |||||
self.xsize = 0 | |||||
self.ysize = 0 | |||||
self.xoff = 0 | |||||
self.yoff = 0 | |||||
def extents(self): | |||||
return self.xoff, self.yoff, self.xoff + self.xsize, self.yoff + self.ysize | |||||
class PyCodec: | |||||
def __init__(self, mode, *args): | |||||
self.im = None | |||||
self.state = PyCodecState() | |||||
self.fd = None | |||||
self.mode = mode | |||||
self.init(args) | |||||
def init(self, args): | |||||
""" | |||||
Override to perform codec specific initialization | |||||
:param args: Array of args items from the tile entry | |||||
:returns: None | |||||
""" | |||||
self.args = args | |||||
def cleanup(self): | |||||
""" | |||||
Override to perform codec specific cleanup | |||||
:returns: None | |||||
""" | |||||
pass | |||||
def setfd(self, fd): | |||||
""" | |||||
Called from ImageFile to set the Python file-like object | |||||
:param fd: A Python file-like object | |||||
:returns: None | |||||
""" | |||||
self.fd = fd | |||||
def setimage(self, im, extents=None): | |||||
""" | |||||
Called from ImageFile to set the core output image for the codec | |||||
:param im: A core image object | |||||
:param extents: a 4 tuple of (x0, y0, x1, y1) defining the rectangle | |||||
for this tile | |||||
:returns: None | |||||
""" | |||||
# following c code | |||||
self.im = im | |||||
if extents: | |||||
(x0, y0, x1, y1) = extents | |||||
else: | |||||
(x0, y0, x1, y1) = (0, 0, 0, 0) | |||||
if x0 == 0 and x1 == 0: | |||||
self.state.xsize, self.state.ysize = self.im.size | |||||
else: | |||||
self.state.xoff = x0 | |||||
self.state.yoff = y0 | |||||
self.state.xsize = x1 - x0 | |||||
self.state.ysize = y1 - y0 | |||||
if self.state.xsize <= 0 or self.state.ysize <= 0: | |||||
msg = "Size cannot be negative" | |||||
raise ValueError(msg) | |||||
if ( | |||||
self.state.xsize + self.state.xoff > self.im.size[0] | |||||
or self.state.ysize + self.state.yoff > self.im.size[1] | |||||
): | |||||
msg = "Tile cannot extend outside image" | |||||
raise ValueError(msg) | |||||
class PyDecoder(PyCodec): | |||||
""" | |||||
Python implementation of a format decoder. Override this class and | |||||
add the decoding logic in the :meth:`decode` method. | |||||
See :ref:`Writing Your Own File Codec in Python<file-codecs-py>` | |||||
""" | |||||
_pulls_fd = False | |||||
@property | |||||
def pulls_fd(self): | |||||
return self._pulls_fd | |||||
def decode(self, buffer): | |||||
""" | |||||
Override to perform the decoding process. | |||||
:param buffer: A bytes object with the data to be decoded. | |||||
:returns: A tuple of ``(bytes consumed, errcode)``. | |||||
If finished with decoding return -1 for the bytes consumed. | |||||
Err codes are from :data:`.ImageFile.ERRORS`. | |||||
""" | |||||
raise NotImplementedError() | |||||
def set_as_raw(self, data, rawmode=None): | |||||
""" | |||||
Convenience method to set the internal image from a stream of raw data | |||||
:param data: Bytes to be set | |||||
:param rawmode: The rawmode to be used for the decoder. | |||||
If not specified, it will default to the mode of the image | |||||
:returns: None | |||||
""" | |||||
if not rawmode: | |||||
rawmode = self.mode | |||||
d = Image._getdecoder(self.mode, "raw", rawmode) | |||||
d.setimage(self.im, self.state.extents()) | |||||
s = d.decode(data) | |||||
if s[0] >= 0: | |||||
msg = "not enough image data" | |||||
raise ValueError(msg) | |||||
if s[1] != 0: | |||||
msg = "cannot decode image data" | |||||
raise ValueError(msg) | |||||
class PyEncoder(PyCodec): | |||||
""" | |||||
Python implementation of a format encoder. Override this class and | |||||
add the decoding logic in the :meth:`encode` method. | |||||
See :ref:`Writing Your Own File Codec in Python<file-codecs-py>` | |||||
""" | |||||
_pushes_fd = False | |||||
@property | |||||
def pushes_fd(self): | |||||
return self._pushes_fd | |||||
def encode(self, bufsize): | |||||
""" | |||||
Override to perform the encoding process. | |||||
:param bufsize: Buffer size. | |||||
:returns: A tuple of ``(bytes encoded, errcode, bytes)``. | |||||
If finished with encoding return 1 for the error code. | |||||
Err codes are from :data:`.ImageFile.ERRORS`. | |||||
""" | |||||
raise NotImplementedError() | |||||
def encode_to_pyfd(self): | |||||
""" | |||||
If ``pushes_fd`` is ``True``, then this method will be used, | |||||
and ``encode()`` will only be called once. | |||||
:returns: A tuple of ``(bytes consumed, errcode)``. | |||||
Err codes are from :data:`.ImageFile.ERRORS`. | |||||
""" | |||||
if not self.pushes_fd: | |||||
return 0, -8 # bad configuration | |||||
bytes_consumed, errcode, data = self.encode(0) | |||||
if data: | |||||
self.fd.write(data) | |||||
return bytes_consumed, errcode | |||||
def encode_to_file(self, fh, bufsize): | |||||
""" | |||||
:param fh: File handle. | |||||
:param bufsize: Buffer size. | |||||
:returns: If finished successfully, return 0. | |||||
Otherwise, return an error code. Err codes are from | |||||
:data:`.ImageFile.ERRORS`. | |||||
""" | |||||
errcode = 0 | |||||
while errcode == 0: | |||||
status, errcode, buf = self.encode(bufsize) | |||||
if status > 0: | |||||
fh.write(buf[status:]) | |||||
return errcode |
# | |||||
# The Python Imaging Library. | |||||
# $Id$ | |||||
# | |||||
# standard filters | |||||
# | |||||
# History: | |||||
# 1995-11-27 fl Created | |||||
# 2002-06-08 fl Added rank and mode filters | |||||
# 2003-09-15 fl Fixed rank calculation in rank filter; added expand call | |||||
# | |||||
# Copyright (c) 1997-2003 by Secret Labs AB. | |||||
# Copyright (c) 1995-2002 by Fredrik Lundh. | |||||
# | |||||
# See the README file for information on usage and redistribution. | |||||
# | |||||
import functools | |||||
class Filter: | |||||
pass | |||||
class MultibandFilter(Filter): | |||||
pass | |||||
class BuiltinFilter(MultibandFilter): | |||||
def filter(self, image): | |||||
if image.mode == "P": | |||||
msg = "cannot filter palette images" | |||||
raise ValueError(msg) | |||||
return image.filter(*self.filterargs) | |||||
class Kernel(BuiltinFilter): | |||||
""" | |||||
Create a convolution kernel. The current version only | |||||
supports 3x3 and 5x5 integer and floating point kernels. | |||||
In the current version, kernels can only be applied to | |||||
"L" and "RGB" images. | |||||
:param size: Kernel size, given as (width, height). In the current | |||||
version, this must be (3,3) or (5,5). | |||||
:param kernel: A sequence containing kernel weights. | |||||
:param scale: Scale factor. If given, the result for each pixel is | |||||
divided by this value. The default is the sum of the | |||||
kernel weights. | |||||
:param offset: Offset. If given, this value is added to the result, | |||||
after it has been divided by the scale factor. | |||||
""" | |||||
name = "Kernel" | |||||
def __init__(self, size, kernel, scale=None, offset=0): | |||||
if scale is None: | |||||
# default scale is sum of kernel | |||||
scale = functools.reduce(lambda a, b: a + b, kernel) | |||||
if size[0] * size[1] != len(kernel): | |||||
msg = "not enough coefficients in kernel" | |||||
raise ValueError(msg) | |||||
self.filterargs = size, scale, offset, kernel | |||||
class RankFilter(Filter): | |||||
""" | |||||
Create a rank filter. The rank filter sorts all pixels in | |||||
a window of the given size, and returns the ``rank``'th value. | |||||
:param size: The kernel size, in pixels. | |||||
:param rank: What pixel value to pick. Use 0 for a min filter, | |||||
``size * size / 2`` for a median filter, ``size * size - 1`` | |||||
for a max filter, etc. | |||||
""" | |||||
name = "Rank" | |||||
def __init__(self, size, rank): | |||||
self.size = size | |||||
self.rank = rank | |||||
def filter(self, image): | |||||
if image.mode == "P": | |||||
msg = "cannot filter palette images" | |||||
raise ValueError(msg) | |||||
image = image.expand(self.size // 2, self.size // 2) | |||||
return image.rankfilter(self.size, self.rank) | |||||
class MedianFilter(RankFilter): | |||||
""" | |||||
Create a median filter. Picks the median pixel value in a window with the | |||||
given size. | |||||
:param size: The kernel size, in pixels. | |||||
""" | |||||
name = "Median" | |||||
def __init__(self, size=3): | |||||
self.size = size | |||||
self.rank = size * size // 2 | |||||
class MinFilter(RankFilter): | |||||
""" | |||||
Create a min filter. Picks the lowest pixel value in a window with the | |||||
given size. | |||||
:param size: The kernel size, in pixels. | |||||
""" | |||||
name = "Min" | |||||
def __init__(self, size=3): | |||||
self.size = size | |||||
self.rank = 0 | |||||
class MaxFilter(RankFilter): | |||||
""" | |||||
Create a max filter. Picks the largest pixel value in a window with the | |||||
given size. | |||||
:param size: The kernel size, in pixels. | |||||
""" | |||||
name = "Max" | |||||
def __init__(self, size=3): | |||||
self.size = size | |||||
self.rank = size * size - 1 | |||||
class ModeFilter(Filter): | |||||
""" | |||||
Create a mode filter. Picks the most frequent pixel value in a box with the | |||||
given size. Pixel values that occur only once or twice are ignored; if no | |||||
pixel value occurs more than twice, the original pixel value is preserved. | |||||
:param size: The kernel size, in pixels. | |||||
""" | |||||
name = "Mode" | |||||
def __init__(self, size=3): | |||||
self.size = size | |||||
def filter(self, image): | |||||
return image.modefilter(self.size) | |||||
class GaussianBlur(MultibandFilter): | |||||
"""Blurs the image with a sequence of extended box filters, which | |||||
approximates a Gaussian kernel. For details on accuracy see | |||||
<https://www.mia.uni-saarland.de/Publications/gwosdek-ssvm11.pdf> | |||||
:param radius: Standard deviation of the Gaussian kernel. | |||||
""" | |||||
name = "GaussianBlur" | |||||
def __init__(self, radius=2): | |||||
self.radius = radius | |||||
def filter(self, image): | |||||
return image.gaussian_blur(self.radius) | |||||
class BoxBlur(MultibandFilter): | |||||
"""Blurs the image by setting each pixel to the average value of the pixels | |||||
in a square box extending radius pixels in each direction. | |||||
Supports float radius of arbitrary size. Uses an optimized implementation | |||||
which runs in linear time relative to the size of the image | |||||
for any radius value. | |||||
:param radius: Size of the box in one direction. Radius 0 does not blur, | |||||
returns an identical image. Radius 1 takes 1 pixel | |||||
in each direction, i.e. 9 pixels in total. | |||||
""" | |||||
name = "BoxBlur" | |||||
def __init__(self, radius): | |||||
if radius < 0: | |||||
msg = "radius must be >= 0" | |||||
raise ValueError(msg) | |||||
self.radius = radius | |||||
def filter(self, image): | |||||
return image.box_blur(self.radius) | |||||
class UnsharpMask(MultibandFilter): | |||||
"""Unsharp mask filter. | |||||
See Wikipedia's entry on `digital unsharp masking`_ for an explanation of | |||||
the parameters. | |||||
:param radius: Blur Radius | |||||
:param percent: Unsharp strength, in percent | |||||
:param threshold: Threshold controls the minimum brightness change that | |||||
will be sharpened | |||||
.. _digital unsharp masking: https://en.wikipedia.org/wiki/Unsharp_masking#Digital_unsharp_masking | |||||
""" # noqa: E501 | |||||
name = "UnsharpMask" | |||||
def __init__(self, radius=2, percent=150, threshold=3): | |||||
self.radius = radius | |||||
self.percent = percent | |||||
self.threshold = threshold | |||||
def filter(self, image): | |||||
return image.unsharp_mask(self.radius, self.percent, self.threshold) | |||||
class BLUR(BuiltinFilter): | |||||
name = "Blur" | |||||
# fmt: off | |||||
filterargs = (5, 5), 16, 0, ( | |||||
1, 1, 1, 1, 1, | |||||
1, 0, 0, 0, 1, | |||||
1, 0, 0, 0, 1, | |||||
1, 0, 0, 0, 1, | |||||
1, 1, 1, 1, 1, | |||||
) | |||||
# fmt: on | |||||
class CONTOUR(BuiltinFilter): | |||||
name = "Contour" | |||||
# fmt: off | |||||
filterargs = (3, 3), 1, 255, ( | |||||
-1, -1, -1, | |||||
-1, 8, -1, | |||||
-1, -1, -1, | |||||
) | |||||
# fmt: on | |||||
class DETAIL(BuiltinFilter): | |||||
name = "Detail" | |||||
# fmt: off | |||||
filterargs = (3, 3), 6, 0, ( | |||||
0, -1, 0, | |||||
-1, 10, -1, | |||||
0, -1, 0, | |||||
) | |||||
# fmt: on | |||||
class EDGE_ENHANCE(BuiltinFilter): | |||||
name = "Edge-enhance" | |||||
# fmt: off | |||||
filterargs = (3, 3), 2, 0, ( | |||||
-1, -1, -1, | |||||
-1, 10, -1, | |||||
-1, -1, -1, | |||||
) | |||||
# fmt: on | |||||
class EDGE_ENHANCE_MORE(BuiltinFilter): | |||||
name = "Edge-enhance More" | |||||
# fmt: off | |||||
filterargs = (3, 3), 1, 0, ( | |||||
-1, -1, -1, | |||||
-1, 9, -1, | |||||
-1, -1, -1, | |||||
) | |||||
# fmt: on | |||||
class EMBOSS(BuiltinFilter): | |||||
name = "Emboss" | |||||
# fmt: off | |||||
filterargs = (3, 3), 1, 128, ( | |||||
-1, 0, 0, | |||||
0, 1, 0, | |||||
0, 0, 0, | |||||
) | |||||
# fmt: on | |||||
class FIND_EDGES(BuiltinFilter): | |||||
name = "Find Edges" | |||||
# fmt: off | |||||
filterargs = (3, 3), 1, 0, ( | |||||
-1, -1, -1, | |||||
-1, 8, -1, | |||||
-1, -1, -1, | |||||
) | |||||
# fmt: on | |||||
class SHARPEN(BuiltinFilter): | |||||
name = "Sharpen" | |||||
# fmt: off | |||||
filterargs = (3, 3), 16, 0, ( | |||||
-2, -2, -2, | |||||
-2, 32, -2, | |||||
-2, -2, -2, | |||||
) | |||||
# fmt: on | |||||
class SMOOTH(BuiltinFilter): | |||||
name = "Smooth" | |||||
# fmt: off | |||||
filterargs = (3, 3), 13, 0, ( | |||||
1, 1, 1, | |||||
1, 5, 1, | |||||
1, 1, 1, | |||||
) | |||||
# fmt: on | |||||
class SMOOTH_MORE(BuiltinFilter): | |||||
name = "Smooth More" | |||||
# fmt: off | |||||
filterargs = (5, 5), 100, 0, ( | |||||
1, 1, 1, 1, 1, | |||||
1, 5, 5, 5, 1, | |||||
1, 5, 44, 5, 1, | |||||
1, 5, 5, 5, 1, | |||||
1, 1, 1, 1, 1, | |||||
) | |||||
# fmt: on | |||||
class Color3DLUT(MultibandFilter): | |||||
"""Three-dimensional color lookup table. | |||||
Transforms 3-channel pixels using the values of the channels as coordinates | |||||
in the 3D lookup table and interpolating the nearest elements. | |||||
This method allows you to apply almost any color transformation | |||||
in constant time by using pre-calculated decimated tables. | |||||
.. versionadded:: 5.2.0 | |||||
:param size: Size of the table. One int or tuple of (int, int, int). | |||||
Minimal size in any dimension is 2, maximum is 65. | |||||
:param table: Flat lookup table. A list of ``channels * size**3`` | |||||
float elements or a list of ``size**3`` channels-sized | |||||
tuples with floats. Channels are changed first, | |||||
then first dimension, then second, then third. | |||||
Value 0.0 corresponds lowest value of output, 1.0 highest. | |||||
:param channels: Number of channels in the table. Could be 3 or 4. | |||||
Default is 3. | |||||
:param target_mode: A mode for the result image. Should have not less | |||||
than ``channels`` channels. Default is ``None``, | |||||
which means that mode wouldn't be changed. | |||||
""" | |||||
name = "Color 3D LUT" | |||||
def __init__(self, size, table, channels=3, target_mode=None, **kwargs): | |||||
if channels not in (3, 4): | |||||
msg = "Only 3 or 4 output channels are supported" | |||||
raise ValueError(msg) | |||||
self.size = size = self._check_size(size) | |||||
self.channels = channels | |||||
self.mode = target_mode | |||||
# Hidden flag `_copy_table=False` could be used to avoid extra copying | |||||
# of the table if the table is specially made for the constructor. | |||||
copy_table = kwargs.get("_copy_table", True) | |||||
items = size[0] * size[1] * size[2] | |||||
wrong_size = False | |||||
numpy = None | |||||
if hasattr(table, "shape"): | |||||
try: | |||||
import numpy | |||||
except ImportError: # pragma: no cover | |||||
pass | |||||
if numpy and isinstance(table, numpy.ndarray): | |||||
if copy_table: | |||||
table = table.copy() | |||||
if table.shape in [ | |||||
(items * channels,), | |||||
(items, channels), | |||||
(size[2], size[1], size[0], channels), | |||||
]: | |||||
table = table.reshape(items * channels) | |||||
else: | |||||
wrong_size = True | |||||
else: | |||||
if copy_table: | |||||
table = list(table) | |||||
# Convert to a flat list | |||||
if table and isinstance(table[0], (list, tuple)): | |||||
table, raw_table = [], table | |||||
for pixel in raw_table: | |||||
if len(pixel) != channels: | |||||
msg = ( | |||||
"The elements of the table should " | |||||
f"have a length of {channels}." | |||||
) | |||||
raise ValueError(msg) | |||||
table.extend(pixel) | |||||
if wrong_size or len(table) != items * channels: | |||||
msg = ( | |||||
"The table should have either channels * size**3 float items " | |||||
"or size**3 items of channels-sized tuples with floats. " | |||||
f"Table should be: {channels}x{size[0]}x{size[1]}x{size[2]}. " | |||||
f"Actual length: {len(table)}" | |||||
) | |||||
raise ValueError(msg) | |||||
self.table = table | |||||
@staticmethod | |||||
def _check_size(size): | |||||
try: | |||||
_, _, _ = size | |||||
except ValueError as e: | |||||
msg = "Size should be either an integer or a tuple of three integers." | |||||
raise ValueError(msg) from e | |||||
except TypeError: | |||||
size = (size, size, size) | |||||
size = [int(x) for x in size] | |||||
for size_1d in size: | |||||
if not 2 <= size_1d <= 65: | |||||
msg = "Size should be in [2, 65] range." | |||||
raise ValueError(msg) | |||||
return size | |||||
@classmethod | |||||
def generate(cls, size, callback, channels=3, target_mode=None): | |||||
"""Generates new LUT using provided callback. | |||||
:param size: Size of the table. Passed to the constructor. | |||||
:param callback: Function with three parameters which correspond | |||||
three color channels. Will be called ``size**3`` | |||||
times with values from 0.0 to 1.0 and should return | |||||
a tuple with ``channels`` elements. | |||||
:param channels: The number of channels which should return callback. | |||||
:param target_mode: Passed to the constructor of the resulting | |||||
lookup table. | |||||
""" | |||||
size_1d, size_2d, size_3d = cls._check_size(size) | |||||
if channels not in (3, 4): | |||||
msg = "Only 3 or 4 output channels are supported" | |||||
raise ValueError(msg) | |||||
table = [0] * (size_1d * size_2d * size_3d * channels) | |||||
idx_out = 0 | |||||
for b in range(size_3d): | |||||
for g in range(size_2d): | |||||
for r in range(size_1d): | |||||
table[idx_out : idx_out + channels] = callback( | |||||
r / (size_1d - 1), g / (size_2d - 1), b / (size_3d - 1) | |||||
) | |||||
idx_out += channels | |||||
return cls( | |||||
(size_1d, size_2d, size_3d), | |||||
table, | |||||
channels=channels, | |||||
target_mode=target_mode, | |||||
_copy_table=False, | |||||
) | |||||
def transform(self, callback, with_normals=False, channels=None, target_mode=None): | |||||
"""Transforms the table values using provided callback and returns | |||||
a new LUT with altered values. | |||||
:param callback: A function which takes old lookup table values | |||||
and returns a new set of values. The number | |||||
of arguments which function should take is | |||||
``self.channels`` or ``3 + self.channels`` | |||||
if ``with_normals`` flag is set. | |||||
Should return a tuple of ``self.channels`` or | |||||
``channels`` elements if it is set. | |||||
:param with_normals: If true, ``callback`` will be called with | |||||
coordinates in the color cube as the first | |||||
three arguments. Otherwise, ``callback`` | |||||
will be called only with actual color values. | |||||
:param channels: The number of channels in the resulting lookup table. | |||||
:param target_mode: Passed to the constructor of the resulting | |||||
lookup table. | |||||
""" | |||||
if channels not in (None, 3, 4): | |||||
msg = "Only 3 or 4 output channels are supported" | |||||
raise ValueError(msg) | |||||
ch_in = self.channels | |||||
ch_out = channels or ch_in | |||||
size_1d, size_2d, size_3d = self.size | |||||
table = [0] * (size_1d * size_2d * size_3d * ch_out) | |||||
idx_in = 0 | |||||
idx_out = 0 | |||||
for b in range(size_3d): | |||||
for g in range(size_2d): | |||||
for r in range(size_1d): | |||||
values = self.table[idx_in : idx_in + ch_in] | |||||
if with_normals: | |||||
values = callback( | |||||
r / (size_1d - 1), | |||||
g / (size_2d - 1), | |||||
b / (size_3d - 1), | |||||
*values, | |||||
) | |||||
else: | |||||
values = callback(*values) | |||||
table[idx_out : idx_out + ch_out] = values | |||||
idx_in += ch_in | |||||
idx_out += ch_out | |||||
return type(self)( | |||||
self.size, | |||||
table, | |||||
channels=ch_out, | |||||
target_mode=target_mode or self.mode, | |||||
_copy_table=False, | |||||
) | |||||
def __repr__(self): | |||||
r = [ | |||||
f"{self.__class__.__name__} from {self.table.__class__.__name__}", | |||||
"size={:d}x{:d}x{:d}".format(*self.size), | |||||
f"channels={self.channels:d}", | |||||
] | |||||
if self.mode: | |||||
r.append(f"target_mode={self.mode}") | |||||
return "<{}>".format(" ".join(r)) | |||||
def filter(self, image): | |||||
from . import Image | |||||
return image.color_lut_3d( | |||||
self.mode or image.mode, | |||||
Image.Resampling.BILINEAR, | |||||
self.channels, | |||||
self.size[0], | |||||
self.size[1], | |||||
self.size[2], | |||||
self.table, | |||||
) |
# | |||||
# The Python Imaging Library | |||||
# $Id$ | |||||
# | |||||
# screen grabber | |||||
# | |||||
# History: | |||||
# 2001-04-26 fl created | |||||
# 2001-09-17 fl use builtin driver, if present | |||||
# 2002-11-19 fl added grabclipboard support | |||||
# | |||||
# Copyright (c) 2001-2002 by Secret Labs AB | |||||
# Copyright (c) 2001-2002 by Fredrik Lundh | |||||
# | |||||
# See the README file for information on usage and redistribution. | |||||
# | |||||
import os | |||||
import shutil | |||||
import subprocess | |||||
import sys | |||||
import tempfile | |||||
from . import Image | |||||
def grab(bbox=None, include_layered_windows=False, all_screens=False, xdisplay=None): | |||||
if xdisplay is None: | |||||
if sys.platform == "darwin": | |||||
fh, filepath = tempfile.mkstemp(".png") | |||||
os.close(fh) | |||||
args = ["screencapture"] | |||||
if bbox: | |||||
left, top, right, bottom = bbox | |||||
args += ["-R", f"{left},{top},{right-left},{bottom-top}"] | |||||
subprocess.call(args + ["-x", filepath]) | |||||
im = Image.open(filepath) | |||||
im.load() | |||||
os.unlink(filepath) | |||||
if bbox: | |||||
im_resized = im.resize((right - left, bottom - top)) | |||||
im.close() | |||||
return im_resized | |||||
return im | |||||
elif sys.platform == "win32": | |||||
offset, size, data = Image.core.grabscreen_win32( | |||||
include_layered_windows, all_screens | |||||
) | |||||
im = Image.frombytes( | |||||
"RGB", | |||||
size, | |||||
data, | |||||
# RGB, 32-bit line padding, origin lower left corner | |||||
"raw", | |||||
"BGR", | |||||
(size[0] * 3 + 3) & -4, | |||||
-1, | |||||
) | |||||
if bbox: | |||||
x0, y0 = offset | |||||
left, top, right, bottom = bbox | |||||
im = im.crop((left - x0, top - y0, right - x0, bottom - y0)) | |||||
return im | |||||
elif shutil.which("gnome-screenshot"): | |||||
fh, filepath = tempfile.mkstemp(".png") | |||||
os.close(fh) | |||||
subprocess.call(["gnome-screenshot", "-f", filepath]) | |||||
im = Image.open(filepath) | |||||
im.load() | |||||
os.unlink(filepath) | |||||
if bbox: | |||||
im_cropped = im.crop(bbox) | |||||
im.close() | |||||
return im_cropped | |||||
return im | |||||
# use xdisplay=None for default display on non-win32/macOS systems | |||||
if not Image.core.HAVE_XCB: | |||||
msg = "Pillow was built without XCB support" | |||||
raise OSError(msg) | |||||
size, data = Image.core.grabscreen_x11(xdisplay) | |||||
im = Image.frombytes("RGB", size, data, "raw", "BGRX", size[0] * 4, 1) | |||||
if bbox: | |||||
im = im.crop(bbox) | |||||
return im | |||||
def grabclipboard(): | |||||
if sys.platform == "darwin": | |||||
fh, filepath = tempfile.mkstemp(".jpg") | |||||
os.close(fh) | |||||
commands = [ | |||||
'set theFile to (open for access POSIX file "' | |||||
+ filepath | |||||
+ '" with write permission)', | |||||
"try", | |||||
" write (the clipboard as JPEG picture) to theFile", | |||||
"end try", | |||||
"close access theFile", | |||||
] | |||||
script = ["osascript"] | |||||
for command in commands: | |||||
script += ["-e", command] | |||||
subprocess.call(script) | |||||
im = None | |||||
if os.stat(filepath).st_size != 0: | |||||
im = Image.open(filepath) | |||||
im.load() | |||||
os.unlink(filepath) | |||||
return im | |||||
elif sys.platform == "win32": | |||||
fmt, data = Image.core.grabclipboard_win32() | |||||
if fmt == "file": # CF_HDROP | |||||
import struct | |||||
o = struct.unpack_from("I", data)[0] | |||||
if data[16] != 0: | |||||
files = data[o:].decode("utf-16le").split("\0") | |||||
else: | |||||
files = data[o:].decode("mbcs").split("\0") | |||||
return files[: files.index("")] | |||||
if isinstance(data, bytes): | |||||
import io | |||||
data = io.BytesIO(data) | |||||
if fmt == "png": | |||||
from . import PngImagePlugin | |||||
return PngImagePlugin.PngImageFile(data) | |||||
elif fmt == "DIB": | |||||
from . import BmpImagePlugin | |||||
return BmpImagePlugin.DibImageFile(data) | |||||
return None | |||||
else: | |||||
if shutil.which("wl-paste"): | |||||
args = ["wl-paste"] | |||||
elif shutil.which("xclip"): | |||||
args = ["xclip", "-selection", "clipboard", "-t", "image/png", "-o"] | |||||
else: | |||||
msg = "wl-paste or xclip is required for ImageGrab.grabclipboard() on Linux" | |||||
raise NotImplementedError(msg) | |||||
fh, filepath = tempfile.mkstemp() | |||||
subprocess.call(args, stdout=fh) | |||||
os.close(fh) | |||||
im = Image.open(filepath) | |||||
im.load() | |||||
os.unlink(filepath) | |||||
return im |
# | |||||
# The Python Imaging Library | |||||
# $Id$ | |||||
# | |||||
# a simple math add-on for the Python Imaging Library | |||||
# | |||||
# History: | |||||
# 1999-02-15 fl Original PIL Plus release | |||||
# 2005-05-05 fl Simplified and cleaned up for PIL 1.1.6 | |||||
# 2005-09-12 fl Fixed int() and float() for Python 2.4.1 | |||||
# | |||||
# Copyright (c) 1999-2005 by Secret Labs AB | |||||
# Copyright (c) 2005 by Fredrik Lundh | |||||
# | |||||
# See the README file for information on usage and redistribution. | |||||
# | |||||
import builtins | |||||
from . import Image, _imagingmath | |||||
def _isconstant(v): | |||||
return isinstance(v, (int, float)) | |||||
class _Operand: | |||||
"""Wraps an image operand, providing standard operators""" | |||||
def __init__(self, im): | |||||
self.im = im | |||||
def __fixup(self, im1): | |||||
# convert image to suitable mode | |||||
if isinstance(im1, _Operand): | |||||
# argument was an image. | |||||
if im1.im.mode in ("1", "L"): | |||||
return im1.im.convert("I") | |||||
elif im1.im.mode in ("I", "F"): | |||||
return im1.im | |||||
else: | |||||
msg = f"unsupported mode: {im1.im.mode}" | |||||
raise ValueError(msg) | |||||
else: | |||||
# argument was a constant | |||||
if _isconstant(im1) and self.im.mode in ("1", "L", "I"): | |||||
return Image.new("I", self.im.size, im1) | |||||
else: | |||||
return Image.new("F", self.im.size, im1) | |||||
def apply(self, op, im1, im2=None, mode=None): | |||||
im1 = self.__fixup(im1) | |||||
if im2 is None: | |||||
# unary operation | |||||
out = Image.new(mode or im1.mode, im1.size, None) | |||||
im1.load() | |||||
try: | |||||
op = getattr(_imagingmath, op + "_" + im1.mode) | |||||
except AttributeError as e: | |||||
msg = f"bad operand type for '{op}'" | |||||
raise TypeError(msg) from e | |||||
_imagingmath.unop(op, out.im.id, im1.im.id) | |||||
else: | |||||
# binary operation | |||||
im2 = self.__fixup(im2) | |||||
if im1.mode != im2.mode: | |||||
# convert both arguments to floating point | |||||
if im1.mode != "F": | |||||
im1 = im1.convert("F") | |||||
if im2.mode != "F": | |||||
im2 = im2.convert("F") | |||||
if im1.size != im2.size: | |||||
# crop both arguments to a common size | |||||
size = (min(im1.size[0], im2.size[0]), min(im1.size[1], im2.size[1])) | |||||
if im1.size != size: | |||||
im1 = im1.crop((0, 0) + size) | |||||
if im2.size != size: | |||||
im2 = im2.crop((0, 0) + size) | |||||
out = Image.new(mode or im1.mode, im1.size, None) | |||||
im1.load() | |||||
im2.load() | |||||
try: | |||||
op = getattr(_imagingmath, op + "_" + im1.mode) | |||||
except AttributeError as e: | |||||
msg = f"bad operand type for '{op}'" | |||||
raise TypeError(msg) from e | |||||
_imagingmath.binop(op, out.im.id, im1.im.id, im2.im.id) | |||||
return _Operand(out) | |||||
# unary operators | |||||
def __bool__(self): | |||||
# an image is "true" if it contains at least one non-zero pixel | |||||
return self.im.getbbox() is not None | |||||
def __abs__(self): | |||||
return self.apply("abs", self) | |||||
def __pos__(self): | |||||
return self | |||||
def __neg__(self): | |||||
return self.apply("neg", self) | |||||
# binary operators | |||||
def __add__(self, other): | |||||
return self.apply("add", self, other) | |||||
def __radd__(self, other): | |||||
return self.apply("add", other, self) | |||||
def __sub__(self, other): | |||||
return self.apply("sub", self, other) | |||||
def __rsub__(self, other): | |||||
return self.apply("sub", other, self) | |||||
def __mul__(self, other): | |||||
return self.apply("mul", self, other) | |||||
def __rmul__(self, other): | |||||
return self.apply("mul", other, self) | |||||
def __truediv__(self, other): | |||||
return self.apply("div", self, other) | |||||
def __rtruediv__(self, other): | |||||
return self.apply("div", other, self) | |||||
def __mod__(self, other): | |||||
return self.apply("mod", self, other) | |||||
def __rmod__(self, other): | |||||
return self.apply("mod", other, self) | |||||
def __pow__(self, other): | |||||
return self.apply("pow", self, other) | |||||
def __rpow__(self, other): | |||||
return self.apply("pow", other, self) | |||||
# bitwise | |||||
def __invert__(self): | |||||
return self.apply("invert", self) | |||||
def __and__(self, other): | |||||
return self.apply("and", self, other) | |||||
def __rand__(self, other): | |||||
return self.apply("and", other, self) | |||||
def __or__(self, other): | |||||
return self.apply("or", self, other) | |||||
def __ror__(self, other): | |||||
return self.apply("or", other, self) | |||||
def __xor__(self, other): | |||||
return self.apply("xor", self, other) | |||||
def __rxor__(self, other): | |||||
return self.apply("xor", other, self) | |||||
def __lshift__(self, other): | |||||
return self.apply("lshift", self, other) | |||||
def __rshift__(self, other): | |||||
return self.apply("rshift", self, other) | |||||
# logical | |||||
def __eq__(self, other): | |||||
return self.apply("eq", self, other) | |||||
def __ne__(self, other): | |||||
return self.apply("ne", self, other) | |||||
def __lt__(self, other): | |||||
return self.apply("lt", self, other) | |||||
def __le__(self, other): | |||||
return self.apply("le", self, other) | |||||
def __gt__(self, other): | |||||
return self.apply("gt", self, other) | |||||
def __ge__(self, other): | |||||
return self.apply("ge", self, other) | |||||
# conversions | |||||
def imagemath_int(self): | |||||
return _Operand(self.im.convert("I")) | |||||
def imagemath_float(self): | |||||
return _Operand(self.im.convert("F")) | |||||
# logical | |||||
def imagemath_equal(self, other): | |||||
return self.apply("eq", self, other, mode="I") | |||||
def imagemath_notequal(self, other): | |||||
return self.apply("ne", self, other, mode="I") | |||||
def imagemath_min(self, other): | |||||
return self.apply("min", self, other) | |||||
def imagemath_max(self, other): | |||||
return self.apply("max", self, other) | |||||
def imagemath_convert(self, mode): | |||||
return _Operand(self.im.convert(mode)) | |||||
ops = {} | |||||
for k, v in list(globals().items()): | |||||
if k[:10] == "imagemath_": | |||||
ops[k[10:]] = v | |||||
def eval(expression, _dict={}, **kw): | |||||
""" | |||||
Evaluates an image expression. | |||||
:param expression: A string containing a Python-style expression. | |||||
:param options: Values to add to the evaluation context. You | |||||
can either use a dictionary, or one or more keyword | |||||
arguments. | |||||
:return: The evaluated expression. This is usually an image object, but can | |||||
also be an integer, a floating point value, or a pixel tuple, | |||||
depending on the expression. | |||||
""" | |||||
# build execution namespace | |||||
args = ops.copy() | |||||
args.update(_dict) | |||||
args.update(kw) | |||||
for k, v in list(args.items()): | |||||
if hasattr(v, "im"): | |||||
args[k] = _Operand(v) | |||||
compiled_code = compile(expression, "<string>", "eval") | |||||
def scan(code): | |||||
for const in code.co_consts: | |||||
if type(const) == type(compiled_code): | |||||
scan(const) | |||||
for name in code.co_names: | |||||
if name not in args and name != "abs": | |||||
msg = f"'{name}' not allowed" | |||||
raise ValueError(msg) | |||||
scan(compiled_code) | |||||
out = builtins.eval(expression, {"__builtins": {"abs": abs}}, args) | |||||
try: | |||||
return out.im | |||||
except AttributeError: | |||||
return out |
# | |||||
# The Python Imaging Library. | |||||
# $Id$ | |||||
# | |||||
# standard mode descriptors | |||||
# | |||||
# History: | |||||
# 2006-03-20 fl Added | |||||
# | |||||
# Copyright (c) 2006 by Secret Labs AB. | |||||
# Copyright (c) 2006 by Fredrik Lundh. | |||||
# | |||||
# See the README file for information on usage and redistribution. | |||||
# | |||||
import sys | |||||
# mode descriptor cache | |||||
_modes = None | |||||
class ModeDescriptor: | |||||
"""Wrapper for mode strings.""" | |||||
def __init__(self, mode, bands, basemode, basetype, typestr): | |||||
self.mode = mode | |||||
self.bands = bands | |||||
self.basemode = basemode | |||||
self.basetype = basetype | |||||
self.typestr = typestr | |||||
def __str__(self): | |||||
return self.mode | |||||
def getmode(mode): | |||||
"""Gets a mode descriptor for the given mode.""" | |||||
global _modes | |||||
if not _modes: | |||||
# initialize mode cache | |||||
modes = {} | |||||
endian = "<" if sys.byteorder == "little" else ">" | |||||
for m, (basemode, basetype, bands, typestr) in { | |||||
# core modes | |||||
# Bits need to be extended to bytes | |||||
"1": ("L", "L", ("1",), "|b1"), | |||||
"L": ("L", "L", ("L",), "|u1"), | |||||
"I": ("L", "I", ("I",), endian + "i4"), | |||||
"F": ("L", "F", ("F",), endian + "f4"), | |||||
"P": ("P", "L", ("P",), "|u1"), | |||||
"RGB": ("RGB", "L", ("R", "G", "B"), "|u1"), | |||||
"RGBX": ("RGB", "L", ("R", "G", "B", "X"), "|u1"), | |||||
"RGBA": ("RGB", "L", ("R", "G", "B", "A"), "|u1"), | |||||
"CMYK": ("RGB", "L", ("C", "M", "Y", "K"), "|u1"), | |||||
"YCbCr": ("RGB", "L", ("Y", "Cb", "Cr"), "|u1"), | |||||
# UNDONE - unsigned |u1i1i1 | |||||
"LAB": ("RGB", "L", ("L", "A", "B"), "|u1"), | |||||
"HSV": ("RGB", "L", ("H", "S", "V"), "|u1"), | |||||
# extra experimental modes | |||||
"RGBa": ("RGB", "L", ("R", "G", "B", "a"), "|u1"), | |||||
"BGR;15": ("RGB", "L", ("B", "G", "R"), "|u1"), | |||||
"BGR;16": ("RGB", "L", ("B", "G", "R"), "|u1"), | |||||
"BGR;24": ("RGB", "L", ("B", "G", "R"), "|u1"), | |||||
"LA": ("L", "L", ("L", "A"), "|u1"), | |||||
"La": ("L", "L", ("L", "a"), "|u1"), | |||||
"PA": ("RGB", "L", ("P", "A"), "|u1"), | |||||
}.items(): | |||||
modes[m] = ModeDescriptor(m, bands, basemode, basetype, typestr) | |||||
# mapping modes | |||||
for i16mode, typestr in { | |||||
# I;16 == I;16L, and I;32 == I;32L | |||||
"I;16": "<u2", | |||||
"I;16S": "<i2", | |||||
"I;16L": "<u2", | |||||
"I;16LS": "<i2", | |||||
"I;16B": ">u2", | |||||
"I;16BS": ">i2", | |||||
"I;16N": endian + "u2", | |||||
"I;16NS": endian + "i2", | |||||
"I;32": "<u4", | |||||
"I;32B": ">u4", | |||||
"I;32L": "<u4", | |||||
"I;32S": "<i4", | |||||
"I;32BS": ">i4", | |||||
"I;32LS": "<i4", | |||||
}.items(): | |||||
modes[i16mode] = ModeDescriptor(i16mode, ("I",), "L", "L", typestr) | |||||
# set global mode cache atomically | |||||
_modes = modes | |||||
return _modes[mode] |
# A binary morphology add-on for the Python Imaging Library | |||||
# | |||||
# History: | |||||
# 2014-06-04 Initial version. | |||||
# | |||||
# Copyright (c) 2014 Dov Grobgeld <dov.grobgeld@gmail.com> | |||||
import re | |||||
from . import Image, _imagingmorph | |||||
LUT_SIZE = 1 << 9 | |||||
# fmt: off | |||||
ROTATION_MATRIX = [ | |||||
6, 3, 0, | |||||
7, 4, 1, | |||||
8, 5, 2, | |||||
] | |||||
MIRROR_MATRIX = [ | |||||
2, 1, 0, | |||||
5, 4, 3, | |||||
8, 7, 6, | |||||
] | |||||
# fmt: on | |||||
class LutBuilder: | |||||
"""A class for building a MorphLut from a descriptive language | |||||
The input patterns is a list of a strings sequences like these:: | |||||
4:(... | |||||
.1. | |||||
111)->1 | |||||
(whitespaces including linebreaks are ignored). The option 4 | |||||
describes a series of symmetry operations (in this case a | |||||
4-rotation), the pattern is described by: | |||||
- . or X - Ignore | |||||
- 1 - Pixel is on | |||||
- 0 - Pixel is off | |||||
The result of the operation is described after "->" string. | |||||
The default is to return the current pixel value, which is | |||||
returned if no other match is found. | |||||
Operations: | |||||
- 4 - 4 way rotation | |||||
- N - Negate | |||||
- 1 - Dummy op for no other operation (an op must always be given) | |||||
- M - Mirroring | |||||
Example:: | |||||
lb = LutBuilder(patterns = ["4:(... .1. 111)->1"]) | |||||
lut = lb.build_lut() | |||||
""" | |||||
def __init__(self, patterns=None, op_name=None): | |||||
if patterns is not None: | |||||
self.patterns = patterns | |||||
else: | |||||
self.patterns = [] | |||||
self.lut = None | |||||
if op_name is not None: | |||||
known_patterns = { | |||||
"corner": ["1:(... ... ...)->0", "4:(00. 01. ...)->1"], | |||||
"dilation4": ["4:(... .0. .1.)->1"], | |||||
"dilation8": ["4:(... .0. .1.)->1", "4:(... .0. ..1)->1"], | |||||
"erosion4": ["4:(... .1. .0.)->0"], | |||||
"erosion8": ["4:(... .1. .0.)->0", "4:(... .1. ..0)->0"], | |||||
"edge": [ | |||||
"1:(... ... ...)->0", | |||||
"4:(.0. .1. ...)->1", | |||||
"4:(01. .1. ...)->1", | |||||
], | |||||
} | |||||
if op_name not in known_patterns: | |||||
msg = "Unknown pattern " + op_name + "!" | |||||
raise Exception(msg) | |||||
self.patterns = known_patterns[op_name] | |||||
def add_patterns(self, patterns): | |||||
self.patterns += patterns | |||||
def build_default_lut(self): | |||||
symbols = [0, 1] | |||||
m = 1 << 4 # pos of current pixel | |||||
self.lut = bytearray(symbols[(i & m) > 0] for i in range(LUT_SIZE)) | |||||
def get_lut(self): | |||||
return self.lut | |||||
def _string_permute(self, pattern, permutation): | |||||
"""string_permute takes a pattern and a permutation and returns the | |||||
string permuted according to the permutation list. | |||||
""" | |||||
assert len(permutation) == 9 | |||||
return "".join(pattern[p] for p in permutation) | |||||
def _pattern_permute(self, basic_pattern, options, basic_result): | |||||
"""pattern_permute takes a basic pattern and its result and clones | |||||
the pattern according to the modifications described in the $options | |||||
parameter. It returns a list of all cloned patterns.""" | |||||
patterns = [(basic_pattern, basic_result)] | |||||
# rotations | |||||
if "4" in options: | |||||
res = patterns[-1][1] | |||||
for i in range(4): | |||||
patterns.append( | |||||
(self._string_permute(patterns[-1][0], ROTATION_MATRIX), res) | |||||
) | |||||
# mirror | |||||
if "M" in options: | |||||
n = len(patterns) | |||||
for pattern, res in patterns[:n]: | |||||
patterns.append((self._string_permute(pattern, MIRROR_MATRIX), res)) | |||||
# negate | |||||
if "N" in options: | |||||
n = len(patterns) | |||||
for pattern, res in patterns[:n]: | |||||
# Swap 0 and 1 | |||||
pattern = pattern.replace("0", "Z").replace("1", "0").replace("Z", "1") | |||||
res = 1 - int(res) | |||||
patterns.append((pattern, res)) | |||||
return patterns | |||||
def build_lut(self): | |||||
"""Compile all patterns into a morphology lut. | |||||
TBD :Build based on (file) morphlut:modify_lut | |||||
""" | |||||
self.build_default_lut() | |||||
patterns = [] | |||||
# Parse and create symmetries of the patterns strings | |||||
for p in self.patterns: | |||||
m = re.search(r"(\w*):?\s*\((.+?)\)\s*->\s*(\d)", p.replace("\n", "")) | |||||
if not m: | |||||
msg = 'Syntax error in pattern "' + p + '"' | |||||
raise Exception(msg) | |||||
options = m.group(1) | |||||
pattern = m.group(2) | |||||
result = int(m.group(3)) | |||||
# Get rid of spaces | |||||
pattern = pattern.replace(" ", "").replace("\n", "") | |||||
patterns += self._pattern_permute(pattern, options, result) | |||||
# compile the patterns into regular expressions for speed | |||||
for i, pattern in enumerate(patterns): | |||||
p = pattern[0].replace(".", "X").replace("X", "[01]") | |||||
p = re.compile(p) | |||||
patterns[i] = (p, pattern[1]) | |||||
# Step through table and find patterns that match. | |||||
# Note that all the patterns are searched. The last one | |||||
# caught overrides | |||||
for i in range(LUT_SIZE): | |||||
# Build the bit pattern | |||||
bitpattern = bin(i)[2:] | |||||
bitpattern = ("0" * (9 - len(bitpattern)) + bitpattern)[::-1] | |||||
for p, r in patterns: | |||||
if p.match(bitpattern): | |||||
self.lut[i] = [0, 1][r] | |||||
return self.lut | |||||
class MorphOp: | |||||
"""A class for binary morphological operators""" | |||||
def __init__(self, lut=None, op_name=None, patterns=None): | |||||
"""Create a binary morphological operator""" | |||||
self.lut = lut | |||||
if op_name is not None: | |||||
self.lut = LutBuilder(op_name=op_name).build_lut() | |||||
elif patterns is not None: | |||||
self.lut = LutBuilder(patterns=patterns).build_lut() | |||||
def apply(self, image): | |||||
"""Run a single morphological operation on an image | |||||
Returns a tuple of the number of changed pixels and the | |||||
morphed image""" | |||||
if self.lut is None: | |||||
msg = "No operator loaded" | |||||
raise Exception(msg) | |||||
if image.mode != "L": | |||||
msg = "Image mode must be L" | |||||
raise ValueError(msg) | |||||
outimage = Image.new(image.mode, image.size, None) | |||||
count = _imagingmorph.apply(bytes(self.lut), image.im.id, outimage.im.id) | |||||
return count, outimage | |||||
def match(self, image): | |||||
"""Get a list of coordinates matching the morphological operation on | |||||
an image. | |||||
Returns a list of tuples of (x,y) coordinates | |||||
of all matching pixels. See :ref:`coordinate-system`.""" | |||||
if self.lut is None: | |||||
msg = "No operator loaded" | |||||
raise Exception(msg) | |||||
if image.mode != "L": | |||||
msg = "Image mode must be L" | |||||
raise ValueError(msg) | |||||
return _imagingmorph.match(bytes(self.lut), image.im.id) | |||||
def get_on_pixels(self, image): | |||||
"""Get a list of all turned on pixels in a binary image | |||||
Returns a list of tuples of (x,y) coordinates | |||||
of all matching pixels. See :ref:`coordinate-system`.""" | |||||
if image.mode != "L": | |||||
msg = "Image mode must be L" | |||||
raise ValueError(msg) | |||||
return _imagingmorph.get_on_pixels(image.im.id) | |||||
def load_lut(self, filename): | |||||
"""Load an operator from an mrl file""" | |||||
with open(filename, "rb") as f: | |||||
self.lut = bytearray(f.read()) | |||||
if len(self.lut) != LUT_SIZE: | |||||
self.lut = None | |||||
msg = "Wrong size operator file!" | |||||
raise Exception(msg) | |||||
def save_lut(self, filename): | |||||
"""Save an operator to an mrl file""" | |||||
if self.lut is None: | |||||
msg = "No operator loaded" | |||||
raise Exception(msg) | |||||
with open(filename, "wb") as f: | |||||
f.write(self.lut) | |||||
def set_lut(self, lut): | |||||
"""Set the lut from an external source""" | |||||
self.lut = lut |
# | |||||
# The Python Imaging Library. | |||||
# $Id$ | |||||
# | |||||
# standard image operations | |||||
# | |||||
# History: | |||||
# 2001-10-20 fl Created | |||||
# 2001-10-23 fl Added autocontrast operator | |||||
# 2001-12-18 fl Added Kevin's fit operator | |||||
# 2004-03-14 fl Fixed potential division by zero in equalize | |||||
# 2005-05-05 fl Fixed equalize for low number of values | |||||
# | |||||
# Copyright (c) 2001-2004 by Secret Labs AB | |||||
# Copyright (c) 2001-2004 by Fredrik Lundh | |||||
# | |||||
# See the README file for information on usage and redistribution. | |||||
# | |||||
import functools | |||||
import operator | |||||
import re | |||||
from . import Image, ImagePalette | |||||
# | |||||
# helpers | |||||
def _border(border): | |||||
if isinstance(border, tuple): | |||||
if len(border) == 2: | |||||
left, top = right, bottom = border | |||||
elif len(border) == 4: | |||||
left, top, right, bottom = border | |||||
else: | |||||
left = top = right = bottom = border | |||||
return left, top, right, bottom | |||||
def _color(color, mode): | |||||
if isinstance(color, str): | |||||
from . import ImageColor | |||||
color = ImageColor.getcolor(color, mode) | |||||
return color | |||||
def _lut(image, lut): | |||||
if image.mode == "P": | |||||
# FIXME: apply to lookup table, not image data | |||||
msg = "mode P support coming soon" | |||||
raise NotImplementedError(msg) | |||||
elif image.mode in ("L", "RGB"): | |||||
if image.mode == "RGB" and len(lut) == 256: | |||||
lut = lut + lut + lut | |||||
return image.point(lut) | |||||
else: | |||||
msg = "not supported for this image mode" | |||||
raise OSError(msg) | |||||
# | |||||
# actions | |||||
def autocontrast(image, cutoff=0, ignore=None, mask=None, preserve_tone=False): | |||||
""" | |||||
Maximize (normalize) image contrast. This function calculates a | |||||
histogram of the input image (or mask region), removes ``cutoff`` percent of the | |||||
lightest and darkest pixels from the histogram, and remaps the image | |||||
so that the darkest pixel becomes black (0), and the lightest | |||||
becomes white (255). | |||||
:param image: The image to process. | |||||
:param cutoff: The percent to cut off from the histogram on the low and | |||||
high ends. Either a tuple of (low, high), or a single | |||||
number for both. | |||||
:param ignore: The background pixel value (use None for no background). | |||||
:param mask: Histogram used in contrast operation is computed using pixels | |||||
within the mask. If no mask is given the entire image is used | |||||
for histogram computation. | |||||
:param preserve_tone: Preserve image tone in Photoshop-like style autocontrast. | |||||
.. versionadded:: 8.2.0 | |||||
:return: An image. | |||||
""" | |||||
if preserve_tone: | |||||
histogram = image.convert("L").histogram(mask) | |||||
else: | |||||
histogram = image.histogram(mask) | |||||
lut = [] | |||||
for layer in range(0, len(histogram), 256): | |||||
h = histogram[layer : layer + 256] | |||||
if ignore is not None: | |||||
# get rid of outliers | |||||
try: | |||||
h[ignore] = 0 | |||||
except TypeError: | |||||
# assume sequence | |||||
for ix in ignore: | |||||
h[ix] = 0 | |||||
if cutoff: | |||||
# cut off pixels from both ends of the histogram | |||||
if not isinstance(cutoff, tuple): | |||||
cutoff = (cutoff, cutoff) | |||||
# get number of pixels | |||||
n = 0 | |||||
for ix in range(256): | |||||
n = n + h[ix] | |||||
# remove cutoff% pixels from the low end | |||||
cut = n * cutoff[0] // 100 | |||||
for lo in range(256): | |||||
if cut > h[lo]: | |||||
cut = cut - h[lo] | |||||
h[lo] = 0 | |||||
else: | |||||
h[lo] -= cut | |||||
cut = 0 | |||||
if cut <= 0: | |||||
break | |||||
# remove cutoff% samples from the high end | |||||
cut = n * cutoff[1] // 100 | |||||
for hi in range(255, -1, -1): | |||||
if cut > h[hi]: | |||||
cut = cut - h[hi] | |||||
h[hi] = 0 | |||||
else: | |||||
h[hi] -= cut | |||||
cut = 0 | |||||
if cut <= 0: | |||||
break | |||||
# find lowest/highest samples after preprocessing | |||||
for lo in range(256): | |||||
if h[lo]: | |||||
break | |||||
for hi in range(255, -1, -1): | |||||
if h[hi]: | |||||
break | |||||
if hi <= lo: | |||||
# don't bother | |||||
lut.extend(list(range(256))) | |||||
else: | |||||
scale = 255.0 / (hi - lo) | |||||
offset = -lo * scale | |||||
for ix in range(256): | |||||
ix = int(ix * scale + offset) | |||||
if ix < 0: | |||||
ix = 0 | |||||
elif ix > 255: | |||||
ix = 255 | |||||
lut.append(ix) | |||||
return _lut(image, lut) | |||||
def colorize(image, black, white, mid=None, blackpoint=0, whitepoint=255, midpoint=127): | |||||
""" | |||||
Colorize grayscale image. | |||||
This function calculates a color wedge which maps all black pixels in | |||||
the source image to the first color and all white pixels to the | |||||
second color. If ``mid`` is specified, it uses three-color mapping. | |||||
The ``black`` and ``white`` arguments should be RGB tuples or color names; | |||||
optionally you can use three-color mapping by also specifying ``mid``. | |||||
Mapping positions for any of the colors can be specified | |||||
(e.g. ``blackpoint``), where these parameters are the integer | |||||
value corresponding to where the corresponding color should be mapped. | |||||
These parameters must have logical order, such that | |||||
``blackpoint <= midpoint <= whitepoint`` (if ``mid`` is specified). | |||||
:param image: The image to colorize. | |||||
:param black: The color to use for black input pixels. | |||||
:param white: The color to use for white input pixels. | |||||
:param mid: The color to use for midtone input pixels. | |||||
:param blackpoint: an int value [0, 255] for the black mapping. | |||||
:param whitepoint: an int value [0, 255] for the white mapping. | |||||
:param midpoint: an int value [0, 255] for the midtone mapping. | |||||
:return: An image. | |||||
""" | |||||
# Initial asserts | |||||
assert image.mode == "L" | |||||
if mid is None: | |||||
assert 0 <= blackpoint <= whitepoint <= 255 | |||||
else: | |||||
assert 0 <= blackpoint <= midpoint <= whitepoint <= 255 | |||||
# Define colors from arguments | |||||
black = _color(black, "RGB") | |||||
white = _color(white, "RGB") | |||||
if mid is not None: | |||||
mid = _color(mid, "RGB") | |||||
# Empty lists for the mapping | |||||
red = [] | |||||
green = [] | |||||
blue = [] | |||||
# Create the low-end values | |||||
for i in range(0, blackpoint): | |||||
red.append(black[0]) | |||||
green.append(black[1]) | |||||
blue.append(black[2]) | |||||
# Create the mapping (2-color) | |||||
if mid is None: | |||||
range_map = range(0, whitepoint - blackpoint) | |||||
for i in range_map: | |||||
red.append(black[0] + i * (white[0] - black[0]) // len(range_map)) | |||||
green.append(black[1] + i * (white[1] - black[1]) // len(range_map)) | |||||
blue.append(black[2] + i * (white[2] - black[2]) // len(range_map)) | |||||
# Create the mapping (3-color) | |||||
else: | |||||
range_map1 = range(0, midpoint - blackpoint) | |||||
range_map2 = range(0, whitepoint - midpoint) | |||||
for i in range_map1: | |||||
red.append(black[0] + i * (mid[0] - black[0]) // len(range_map1)) | |||||
green.append(black[1] + i * (mid[1] - black[1]) // len(range_map1)) | |||||
blue.append(black[2] + i * (mid[2] - black[2]) // len(range_map1)) | |||||
for i in range_map2: | |||||
red.append(mid[0] + i * (white[0] - mid[0]) // len(range_map2)) | |||||
green.append(mid[1] + i * (white[1] - mid[1]) // len(range_map2)) | |||||
blue.append(mid[2] + i * (white[2] - mid[2]) // len(range_map2)) | |||||
# Create the high-end values | |||||
for i in range(0, 256 - whitepoint): | |||||
red.append(white[0]) | |||||
green.append(white[1]) | |||||
blue.append(white[2]) | |||||
# Return converted image | |||||
image = image.convert("RGB") | |||||
return _lut(image, red + green + blue) | |||||
def contain(image, size, method=Image.Resampling.BICUBIC): | |||||
""" | |||||
Returns a resized version of the image, set to the maximum width and height | |||||
within the requested size, while maintaining the original aspect ratio. | |||||
:param image: The image to resize and crop. | |||||
:param size: The requested output size in pixels, given as a | |||||
(width, height) tuple. | |||||
:param method: Resampling method to use. Default is | |||||
:py:attr:`~PIL.Image.Resampling.BICUBIC`. | |||||
See :ref:`concept-filters`. | |||||
:return: An image. | |||||
""" | |||||
im_ratio = image.width / image.height | |||||
dest_ratio = size[0] / size[1] | |||||
if im_ratio != dest_ratio: | |||||
if im_ratio > dest_ratio: | |||||
new_height = round(image.height / image.width * size[0]) | |||||
if new_height != size[1]: | |||||
size = (size[0], new_height) | |||||
else: | |||||
new_width = round(image.width / image.height * size[1]) | |||||
if new_width != size[0]: | |||||
size = (new_width, size[1]) | |||||
return image.resize(size, resample=method) | |||||
def pad(image, size, method=Image.Resampling.BICUBIC, color=None, centering=(0.5, 0.5)): | |||||
""" | |||||
Returns a resized and padded version of the image, expanded to fill the | |||||
requested aspect ratio and size. | |||||
:param image: The image to resize and crop. | |||||
:param size: The requested output size in pixels, given as a | |||||
(width, height) tuple. | |||||
:param method: Resampling method to use. Default is | |||||
:py:attr:`~PIL.Image.Resampling.BICUBIC`. | |||||
See :ref:`concept-filters`. | |||||
:param color: The background color of the padded image. | |||||
:param centering: Control the position of the original image within the | |||||
padded version. | |||||
(0.5, 0.5) will keep the image centered | |||||
(0, 0) will keep the image aligned to the top left | |||||
(1, 1) will keep the image aligned to the bottom | |||||
right | |||||
:return: An image. | |||||
""" | |||||
resized = contain(image, size, method) | |||||
if resized.size == size: | |||||
out = resized | |||||
else: | |||||
out = Image.new(image.mode, size, color) | |||||
if resized.palette: | |||||
out.putpalette(resized.getpalette()) | |||||
if resized.width != size[0]: | |||||
x = round((size[0] - resized.width) * max(0, min(centering[0], 1))) | |||||
out.paste(resized, (x, 0)) | |||||
else: | |||||
y = round((size[1] - resized.height) * max(0, min(centering[1], 1))) | |||||
out.paste(resized, (0, y)) | |||||
return out | |||||
def crop(image, border=0): | |||||
""" | |||||
Remove border from image. The same amount of pixels are removed | |||||
from all four sides. This function works on all image modes. | |||||
.. seealso:: :py:meth:`~PIL.Image.Image.crop` | |||||
:param image: The image to crop. | |||||
:param border: The number of pixels to remove. | |||||
:return: An image. | |||||
""" | |||||
left, top, right, bottom = _border(border) | |||||
return image.crop((left, top, image.size[0] - right, image.size[1] - bottom)) | |||||
def scale(image, factor, resample=Image.Resampling.BICUBIC): | |||||
""" | |||||
Returns a rescaled image by a specific factor given in parameter. | |||||
A factor greater than 1 expands the image, between 0 and 1 contracts the | |||||
image. | |||||
:param image: The image to rescale. | |||||
:param factor: The expansion factor, as a float. | |||||
:param resample: Resampling method to use. Default is | |||||
:py:attr:`~PIL.Image.Resampling.BICUBIC`. | |||||
See :ref:`concept-filters`. | |||||
:returns: An :py:class:`~PIL.Image.Image` object. | |||||
""" | |||||
if factor == 1: | |||||
return image.copy() | |||||
elif factor <= 0: | |||||
msg = "the factor must be greater than 0" | |||||
raise ValueError(msg) | |||||
else: | |||||
size = (round(factor * image.width), round(factor * image.height)) | |||||
return image.resize(size, resample) | |||||
def deform(image, deformer, resample=Image.Resampling.BILINEAR): | |||||
""" | |||||
Deform the image. | |||||
:param image: The image to deform. | |||||
:param deformer: A deformer object. Any object that implements a | |||||
``getmesh`` method can be used. | |||||
:param resample: An optional resampling filter. Same values possible as | |||||
in the PIL.Image.transform function. | |||||
:return: An image. | |||||
""" | |||||
return image.transform( | |||||
image.size, Image.Transform.MESH, deformer.getmesh(image), resample | |||||
) | |||||
def equalize(image, mask=None): | |||||
""" | |||||
Equalize the image histogram. This function applies a non-linear | |||||
mapping to the input image, in order to create a uniform | |||||
distribution of grayscale values in the output image. | |||||
:param image: The image to equalize. | |||||
:param mask: An optional mask. If given, only the pixels selected by | |||||
the mask are included in the analysis. | |||||
:return: An image. | |||||
""" | |||||
if image.mode == "P": | |||||
image = image.convert("RGB") | |||||
h = image.histogram(mask) | |||||
lut = [] | |||||
for b in range(0, len(h), 256): | |||||
histo = [_f for _f in h[b : b + 256] if _f] | |||||
if len(histo) <= 1: | |||||
lut.extend(list(range(256))) | |||||
else: | |||||
step = (functools.reduce(operator.add, histo) - histo[-1]) // 255 | |||||
if not step: | |||||
lut.extend(list(range(256))) | |||||
else: | |||||
n = step // 2 | |||||
for i in range(256): | |||||
lut.append(n // step) | |||||
n = n + h[i + b] | |||||
return _lut(image, lut) | |||||
def expand(image, border=0, fill=0): | |||||
""" | |||||
Add border to the image | |||||
:param image: The image to expand. | |||||
:param border: Border width, in pixels. | |||||
:param fill: Pixel fill value (a color value). Default is 0 (black). | |||||
:return: An image. | |||||
""" | |||||
left, top, right, bottom = _border(border) | |||||
width = left + image.size[0] + right | |||||
height = top + image.size[1] + bottom | |||||
color = _color(fill, image.mode) | |||||
if image.palette: | |||||
palette = ImagePalette.ImagePalette(palette=image.getpalette()) | |||||
if isinstance(color, tuple): | |||||
color = palette.getcolor(color) | |||||
else: | |||||
palette = None | |||||
out = Image.new(image.mode, (width, height), color) | |||||
if palette: | |||||
out.putpalette(palette.palette) | |||||
out.paste(image, (left, top)) | |||||
return out | |||||
def fit(image, size, method=Image.Resampling.BICUBIC, bleed=0.0, centering=(0.5, 0.5)): | |||||
""" | |||||
Returns a resized and cropped version of the image, cropped to the | |||||
requested aspect ratio and size. | |||||
This function was contributed by Kevin Cazabon. | |||||
:param image: The image to resize and crop. | |||||
:param size: The requested output size in pixels, given as a | |||||
(width, height) tuple. | |||||
:param method: Resampling method to use. Default is | |||||
:py:attr:`~PIL.Image.Resampling.BICUBIC`. | |||||
See :ref:`concept-filters`. | |||||
:param bleed: Remove a border around the outside of the image from all | |||||
four edges. The value is a decimal percentage (use 0.01 for | |||||
one percent). The default value is 0 (no border). | |||||
Cannot be greater than or equal to 0.5. | |||||
:param centering: Control the cropping position. Use (0.5, 0.5) for | |||||
center cropping (e.g. if cropping the width, take 50% off | |||||
of the left side, and therefore 50% off the right side). | |||||
(0.0, 0.0) will crop from the top left corner (i.e. if | |||||
cropping the width, take all of the crop off of the right | |||||
side, and if cropping the height, take all of it off the | |||||
bottom). (1.0, 0.0) will crop from the bottom left | |||||
corner, etc. (i.e. if cropping the width, take all of the | |||||
crop off the left side, and if cropping the height take | |||||
none from the top, and therefore all off the bottom). | |||||
:return: An image. | |||||
""" | |||||
# by Kevin Cazabon, Feb 17/2000 | |||||
# kevin@cazabon.com | |||||
# https://www.cazabon.com | |||||
# ensure centering is mutable | |||||
centering = list(centering) | |||||
if not 0.0 <= centering[0] <= 1.0: | |||||
centering[0] = 0.5 | |||||
if not 0.0 <= centering[1] <= 1.0: | |||||
centering[1] = 0.5 | |||||
if not 0.0 <= bleed < 0.5: | |||||
bleed = 0.0 | |||||
# calculate the area to use for resizing and cropping, subtracting | |||||
# the 'bleed' around the edges | |||||
# number of pixels to trim off on Top and Bottom, Left and Right | |||||
bleed_pixels = (bleed * image.size[0], bleed * image.size[1]) | |||||
live_size = ( | |||||
image.size[0] - bleed_pixels[0] * 2, | |||||
image.size[1] - bleed_pixels[1] * 2, | |||||
) | |||||
# calculate the aspect ratio of the live_size | |||||
live_size_ratio = live_size[0] / live_size[1] | |||||
# calculate the aspect ratio of the output image | |||||
output_ratio = size[0] / size[1] | |||||
# figure out if the sides or top/bottom will be cropped off | |||||
if live_size_ratio == output_ratio: | |||||
# live_size is already the needed ratio | |||||
crop_width = live_size[0] | |||||
crop_height = live_size[1] | |||||
elif live_size_ratio >= output_ratio: | |||||
# live_size is wider than what's needed, crop the sides | |||||
crop_width = output_ratio * live_size[1] | |||||
crop_height = live_size[1] | |||||
else: | |||||
# live_size is taller than what's needed, crop the top and bottom | |||||
crop_width = live_size[0] | |||||
crop_height = live_size[0] / output_ratio | |||||
# make the crop | |||||
crop_left = bleed_pixels[0] + (live_size[0] - crop_width) * centering[0] | |||||
crop_top = bleed_pixels[1] + (live_size[1] - crop_height) * centering[1] | |||||
crop = (crop_left, crop_top, crop_left + crop_width, crop_top + crop_height) | |||||
# resize the image and return it | |||||
return image.resize(size, method, box=crop) | |||||
def flip(image): | |||||
""" | |||||
Flip the image vertically (top to bottom). | |||||
:param image: The image to flip. | |||||
:return: An image. | |||||
""" | |||||
return image.transpose(Image.Transpose.FLIP_TOP_BOTTOM) | |||||
def grayscale(image): | |||||
""" | |||||
Convert the image to grayscale. | |||||
:param image: The image to convert. | |||||
:return: An image. | |||||
""" | |||||
return image.convert("L") | |||||
def invert(image): | |||||
""" | |||||
Invert (negate) the image. | |||||
:param image: The image to invert. | |||||
:return: An image. | |||||
""" | |||||
lut = [] | |||||
for i in range(256): | |||||
lut.append(255 - i) | |||||
return image.point(lut) if image.mode == "1" else _lut(image, lut) | |||||
def mirror(image): | |||||
""" | |||||
Flip image horizontally (left to right). | |||||
:param image: The image to mirror. | |||||
:return: An image. | |||||
""" | |||||
return image.transpose(Image.Transpose.FLIP_LEFT_RIGHT) | |||||
def posterize(image, bits): | |||||
""" | |||||
Reduce the number of bits for each color channel. | |||||
:param image: The image to posterize. | |||||
:param bits: The number of bits to keep for each channel (1-8). | |||||
:return: An image. | |||||
""" | |||||
lut = [] | |||||
mask = ~(2 ** (8 - bits) - 1) | |||||
for i in range(256): | |||||
lut.append(i & mask) | |||||
return _lut(image, lut) | |||||
def solarize(image, threshold=128): | |||||
""" | |||||
Invert all pixel values above a threshold. | |||||
:param image: The image to solarize. | |||||
:param threshold: All pixels above this greyscale level are inverted. | |||||
:return: An image. | |||||
""" | |||||
lut = [] | |||||
for i in range(256): | |||||
if i < threshold: | |||||
lut.append(i) | |||||
else: | |||||
lut.append(255 - i) | |||||
return _lut(image, lut) | |||||
def exif_transpose(image): | |||||
""" | |||||
If an image has an EXIF Orientation tag, other than 1, return a new image | |||||
that is transposed accordingly. The new image will have the orientation | |||||
data removed. | |||||
Otherwise, return a copy of the image. | |||||
:param image: The image to transpose. | |||||
:return: An image. | |||||
""" | |||||
exif = image.getexif() | |||||
orientation = exif.get(0x0112) | |||||
method = { | |||||
2: Image.Transpose.FLIP_LEFT_RIGHT, | |||||
3: Image.Transpose.ROTATE_180, | |||||
4: Image.Transpose.FLIP_TOP_BOTTOM, | |||||
5: Image.Transpose.TRANSPOSE, | |||||
6: Image.Transpose.ROTATE_270, | |||||
7: Image.Transpose.TRANSVERSE, | |||||
8: Image.Transpose.ROTATE_90, | |||||
}.get(orientation) | |||||
if method is not None: | |||||
transposed_image = image.transpose(method) | |||||
transposed_exif = transposed_image.getexif() | |||||
if 0x0112 in transposed_exif: | |||||
del transposed_exif[0x0112] | |||||
if "exif" in transposed_image.info: | |||||
transposed_image.info["exif"] = transposed_exif.tobytes() | |||||
elif "Raw profile type exif" in transposed_image.info: | |||||
transposed_image.info[ | |||||
"Raw profile type exif" | |||||
] = transposed_exif.tobytes().hex() | |||||
elif "XML:com.adobe.xmp" in transposed_image.info: | |||||
for pattern in ( | |||||
r'tiff:Orientation="([0-9])"', | |||||
r"<tiff:Orientation>([0-9])</tiff:Orientation>", | |||||
): | |||||
transposed_image.info["XML:com.adobe.xmp"] = re.sub( | |||||
pattern, "", transposed_image.info["XML:com.adobe.xmp"] | |||||
) | |||||
return transposed_image | |||||
return image.copy() |
# | |||||
# The Python Imaging Library. | |||||
# $Id$ | |||||
# | |||||
# image palette object | |||||
# | |||||
# History: | |||||
# 1996-03-11 fl Rewritten. | |||||
# 1997-01-03 fl Up and running. | |||||
# 1997-08-23 fl Added load hack | |||||
# 2001-04-16 fl Fixed randint shadow bug in random() | |||||
# | |||||
# Copyright (c) 1997-2001 by Secret Labs AB | |||||
# Copyright (c) 1996-1997 by Fredrik Lundh | |||||
# | |||||
# See the README file for information on usage and redistribution. | |||||
# | |||||
import array | |||||
from . import GimpGradientFile, GimpPaletteFile, ImageColor, PaletteFile | |||||
from ._deprecate import deprecate | |||||
class ImagePalette: | |||||
""" | |||||
Color palette for palette mapped images | |||||
:param mode: The mode to use for the palette. See: | |||||
:ref:`concept-modes`. Defaults to "RGB" | |||||
:param palette: An optional palette. If given, it must be a bytearray, | |||||
an array or a list of ints between 0-255. The list must consist of | |||||
all channels for one color followed by the next color (e.g. RGBRGBRGB). | |||||
Defaults to an empty palette. | |||||
""" | |||||
def __init__(self, mode="RGB", palette=None, size=0): | |||||
self.mode = mode | |||||
self.rawmode = None # if set, palette contains raw data | |||||
self.palette = palette or bytearray() | |||||
self.dirty = None | |||||
if size != 0: | |||||
deprecate("The size parameter", 10, None) | |||||
if size != len(self.palette): | |||||
msg = "wrong palette size" | |||||
raise ValueError(msg) | |||||
@property | |||||
def palette(self): | |||||
return self._palette | |||||
@palette.setter | |||||
def palette(self, palette): | |||||
self._colors = None | |||||
self._palette = palette | |||||
@property | |||||
def colors(self): | |||||
if self._colors is None: | |||||
mode_len = len(self.mode) | |||||
self._colors = {} | |||||
for i in range(0, len(self.palette), mode_len): | |||||
color = tuple(self.palette[i : i + mode_len]) | |||||
if color in self._colors: | |||||
continue | |||||
self._colors[color] = i // mode_len | |||||
return self._colors | |||||
@colors.setter | |||||
def colors(self, colors): | |||||
self._colors = colors | |||||
def copy(self): | |||||
new = ImagePalette() | |||||
new.mode = self.mode | |||||
new.rawmode = self.rawmode | |||||
if self.palette is not None: | |||||
new.palette = self.palette[:] | |||||
new.dirty = self.dirty | |||||
return new | |||||
def getdata(self): | |||||
""" | |||||
Get palette contents in format suitable for the low-level | |||||
``im.putpalette`` primitive. | |||||
.. warning:: This method is experimental. | |||||
""" | |||||
if self.rawmode: | |||||
return self.rawmode, self.palette | |||||
return self.mode, self.tobytes() | |||||
def tobytes(self): | |||||
"""Convert palette to bytes. | |||||
.. warning:: This method is experimental. | |||||
""" | |||||
if self.rawmode: | |||||
msg = "palette contains raw palette data" | |||||
raise ValueError(msg) | |||||
if isinstance(self.palette, bytes): | |||||
return self.palette | |||||
arr = array.array("B", self.palette) | |||||
return arr.tobytes() | |||||
# Declare tostring as an alias for tobytes | |||||
tostring = tobytes | |||||
def getcolor(self, color, image=None): | |||||
"""Given an rgb tuple, allocate palette entry. | |||||
.. warning:: This method is experimental. | |||||
""" | |||||
if self.rawmode: | |||||
msg = "palette contains raw palette data" | |||||
raise ValueError(msg) | |||||
if isinstance(color, tuple): | |||||
if self.mode == "RGB": | |||||
if len(color) == 4: | |||||
if color[3] != 255: | |||||
msg = "cannot add non-opaque RGBA color to RGB palette" | |||||
raise ValueError(msg) | |||||
color = color[:3] | |||||
elif self.mode == "RGBA": | |||||
if len(color) == 3: | |||||
color += (255,) | |||||
try: | |||||
return self.colors[color] | |||||
except KeyError as e: | |||||
# allocate new color slot | |||||
if not isinstance(self.palette, bytearray): | |||||
self._palette = bytearray(self.palette) | |||||
index = len(self.palette) // 3 | |||||
special_colors = () | |||||
if image: | |||||
special_colors = ( | |||||
image.info.get("background"), | |||||
image.info.get("transparency"), | |||||
) | |||||
while index in special_colors: | |||||
index += 1 | |||||
if index >= 256: | |||||
if image: | |||||
# Search for an unused index | |||||
for i, count in reversed(list(enumerate(image.histogram()))): | |||||
if count == 0 and i not in special_colors: | |||||
index = i | |||||
break | |||||
if index >= 256: | |||||
msg = "cannot allocate more than 256 colors" | |||||
raise ValueError(msg) from e | |||||
self.colors[color] = index | |||||
if index * 3 < len(self.palette): | |||||
self._palette = ( | |||||
self.palette[: index * 3] | |||||
+ bytes(color) | |||||
+ self.palette[index * 3 + 3 :] | |||||
) | |||||
else: | |||||
self._palette += bytes(color) | |||||
self.dirty = 1 | |||||
return index | |||||
else: | |||||
msg = f"unknown color specifier: {repr(color)}" | |||||
raise ValueError(msg) | |||||
def save(self, fp): | |||||
"""Save palette to text file. | |||||
.. warning:: This method is experimental. | |||||
""" | |||||
if self.rawmode: | |||||
msg = "palette contains raw palette data" | |||||
raise ValueError(msg) | |||||
if isinstance(fp, str): | |||||
fp = open(fp, "w") | |||||
fp.write("# Palette\n") | |||||
fp.write(f"# Mode: {self.mode}\n") | |||||
for i in range(256): | |||||
fp.write(f"{i}") | |||||
for j in range(i * len(self.mode), (i + 1) * len(self.mode)): | |||||
try: | |||||
fp.write(f" {self.palette[j]}") | |||||
except IndexError: | |||||
fp.write(" 0") | |||||
fp.write("\n") | |||||
fp.close() | |||||
# -------------------------------------------------------------------- | |||||
# Internal | |||||
def raw(rawmode, data): | |||||
palette = ImagePalette() | |||||
palette.rawmode = rawmode | |||||
palette.palette = data | |||||
palette.dirty = 1 | |||||
return palette | |||||
# -------------------------------------------------------------------- | |||||
# Factories | |||||
def make_linear_lut(black, white): | |||||
lut = [] | |||||
if black == 0: | |||||
for i in range(256): | |||||
lut.append(white * i // 255) | |||||
else: | |||||
raise NotImplementedError # FIXME | |||||
return lut | |||||
def make_gamma_lut(exp): | |||||
lut = [] | |||||
for i in range(256): | |||||
lut.append(int(((i / 255.0) ** exp) * 255.0 + 0.5)) | |||||
return lut | |||||
def negative(mode="RGB"): | |||||
palette = list(range(256 * len(mode))) | |||||
palette.reverse() | |||||
return ImagePalette(mode, [i // len(mode) for i in palette]) | |||||
def random(mode="RGB"): | |||||
from random import randint | |||||
palette = [] | |||||
for i in range(256 * len(mode)): | |||||
palette.append(randint(0, 255)) | |||||
return ImagePalette(mode, palette) | |||||
def sepia(white="#fff0c0"): | |||||
bands = [make_linear_lut(0, band) for band in ImageColor.getrgb(white)] | |||||
return ImagePalette("RGB", [bands[i % 3][i // 3] for i in range(256 * 3)]) | |||||
def wedge(mode="RGB"): | |||||
palette = list(range(256 * len(mode))) | |||||
return ImagePalette(mode, [i // len(mode) for i in palette]) | |||||
def load(filename): | |||||
# FIXME: supports GIMP gradients only | |||||
with open(filename, "rb") as fp: | |||||
for paletteHandler in [ | |||||
GimpPaletteFile.GimpPaletteFile, | |||||
GimpGradientFile.GimpGradientFile, | |||||
PaletteFile.PaletteFile, | |||||
]: | |||||
try: | |||||
fp.seek(0) | |||||
lut = paletteHandler(fp).getpalette() | |||||
if lut: | |||||
break | |||||
except (SyntaxError, ValueError): | |||||
# import traceback | |||||
# traceback.print_exc() | |||||
pass | |||||
else: | |||||
msg = "cannot load palette" | |||||
raise OSError(msg) | |||||
return lut # data, rawmode |
# | |||||
# The Python Imaging Library | |||||
# $Id$ | |||||
# | |||||
# path interface | |||||
# | |||||
# History: | |||||
# 1996-11-04 fl Created | |||||
# 2002-04-14 fl Added documentation stub class | |||||
# | |||||
# Copyright (c) Secret Labs AB 1997. | |||||
# Copyright (c) Fredrik Lundh 1996. | |||||
# | |||||
# See the README file for information on usage and redistribution. | |||||
# | |||||
from . import Image | |||||
Path = Image.core.path |
# | |||||
# The Python Imaging Library. | |||||
# $Id$ | |||||
# | |||||
# a simple Qt image interface. | |||||
# | |||||
# history: | |||||
# 2006-06-03 fl: created | |||||
# 2006-06-04 fl: inherit from QImage instead of wrapping it | |||||
# 2006-06-05 fl: removed toimage helper; move string support to ImageQt | |||||
# 2013-11-13 fl: add support for Qt5 (aurelien.ballier@cyclonit.com) | |||||
# | |||||
# Copyright (c) 2006 by Secret Labs AB | |||||
# Copyright (c) 2006 by Fredrik Lundh | |||||
# | |||||
# See the README file for information on usage and redistribution. | |||||
# | |||||
import sys | |||||
from io import BytesIO | |||||
from . import Image | |||||
from ._deprecate import deprecate | |||||
from ._util import is_path | |||||
qt_versions = [ | |||||
["6", "PyQt6"], | |||||
["side6", "PySide6"], | |||||
["5", "PyQt5"], | |||||
["side2", "PySide2"], | |||||
] | |||||
# If a version has already been imported, attempt it first | |||||
qt_versions.sort(key=lambda qt_version: qt_version[1] in sys.modules, reverse=True) | |||||
for qt_version, qt_module in qt_versions: | |||||
try: | |||||
if qt_module == "PyQt6": | |||||
from PyQt6.QtCore import QBuffer, QIODevice | |||||
from PyQt6.QtGui import QImage, QPixmap, qRgba | |||||
elif qt_module == "PySide6": | |||||
from PySide6.QtCore import QBuffer, QIODevice | |||||
from PySide6.QtGui import QImage, QPixmap, qRgba | |||||
elif qt_module == "PyQt5": | |||||
from PyQt5.QtCore import QBuffer, QIODevice | |||||
from PyQt5.QtGui import QImage, QPixmap, qRgba | |||||
deprecate("Support for PyQt5", 10, "PyQt6 or PySide6") | |||||
elif qt_module == "PySide2": | |||||
from PySide2.QtCore import QBuffer, QIODevice | |||||
from PySide2.QtGui import QImage, QPixmap, qRgba | |||||
deprecate("Support for PySide2", 10, "PyQt6 or PySide6") | |||||
except (ImportError, RuntimeError): | |||||
continue | |||||
qt_is_installed = True | |||||
break | |||||
else: | |||||
qt_is_installed = False | |||||
qt_version = None | |||||
def rgb(r, g, b, a=255): | |||||
"""(Internal) Turns an RGB color into a Qt compatible color integer.""" | |||||
# use qRgb to pack the colors, and then turn the resulting long | |||||
# into a negative integer with the same bitpattern. | |||||
return qRgba(r, g, b, a) & 0xFFFFFFFF | |||||
def fromqimage(im): | |||||
""" | |||||
:param im: QImage or PIL ImageQt object | |||||
""" | |||||
buffer = QBuffer() | |||||
if qt_version == "6": | |||||
try: | |||||
qt_openmode = QIODevice.OpenModeFlag | |||||
except AttributeError: | |||||
qt_openmode = QIODevice.OpenMode | |||||
else: | |||||
qt_openmode = QIODevice | |||||
buffer.open(qt_openmode.ReadWrite) | |||||
# preserve alpha channel with png | |||||
# otherwise ppm is more friendly with Image.open | |||||
if im.hasAlphaChannel(): | |||||
im.save(buffer, "png") | |||||
else: | |||||
im.save(buffer, "ppm") | |||||
b = BytesIO() | |||||
b.write(buffer.data()) | |||||
buffer.close() | |||||
b.seek(0) | |||||
return Image.open(b) | |||||
def fromqpixmap(im): | |||||
return fromqimage(im) | |||||
# buffer = QBuffer() | |||||
# buffer.open(QIODevice.ReadWrite) | |||||
# # im.save(buffer) | |||||
# # What if png doesn't support some image features like animation? | |||||
# im.save(buffer, 'ppm') | |||||
# bytes_io = BytesIO() | |||||
# bytes_io.write(buffer.data()) | |||||
# buffer.close() | |||||
# bytes_io.seek(0) | |||||
# return Image.open(bytes_io) | |||||
def align8to32(bytes, width, mode): | |||||
""" | |||||
converts each scanline of data from 8 bit to 32 bit aligned | |||||
""" | |||||
bits_per_pixel = {"1": 1, "L": 8, "P": 8, "I;16": 16}[mode] | |||||
# calculate bytes per line and the extra padding if needed | |||||
bits_per_line = bits_per_pixel * width | |||||
full_bytes_per_line, remaining_bits_per_line = divmod(bits_per_line, 8) | |||||
bytes_per_line = full_bytes_per_line + (1 if remaining_bits_per_line else 0) | |||||
extra_padding = -bytes_per_line % 4 | |||||
# already 32 bit aligned by luck | |||||
if not extra_padding: | |||||
return bytes | |||||
new_data = [] | |||||
for i in range(len(bytes) // bytes_per_line): | |||||
new_data.append( | |||||
bytes[i * bytes_per_line : (i + 1) * bytes_per_line] | |||||
+ b"\x00" * extra_padding | |||||
) | |||||
return b"".join(new_data) | |||||
def _toqclass_helper(im): | |||||
data = None | |||||
colortable = None | |||||
exclusive_fp = False | |||||
# handle filename, if given instead of image name | |||||
if hasattr(im, "toUtf8"): | |||||
# FIXME - is this really the best way to do this? | |||||
im = str(im.toUtf8(), "utf-8") | |||||
if is_path(im): | |||||
im = Image.open(im) | |||||
exclusive_fp = True | |||||
qt_format = QImage.Format if qt_version == "6" else QImage | |||||
if im.mode == "1": | |||||
format = qt_format.Format_Mono | |||||
elif im.mode == "L": | |||||
format = qt_format.Format_Indexed8 | |||||
colortable = [] | |||||
for i in range(256): | |||||
colortable.append(rgb(i, i, i)) | |||||
elif im.mode == "P": | |||||
format = qt_format.Format_Indexed8 | |||||
colortable = [] | |||||
palette = im.getpalette() | |||||
for i in range(0, len(palette), 3): | |||||
colortable.append(rgb(*palette[i : i + 3])) | |||||
elif im.mode == "RGB": | |||||
# Populate the 4th channel with 255 | |||||
im = im.convert("RGBA") | |||||
data = im.tobytes("raw", "BGRA") | |||||
format = qt_format.Format_RGB32 | |||||
elif im.mode == "RGBA": | |||||
data = im.tobytes("raw", "BGRA") | |||||
format = qt_format.Format_ARGB32 | |||||
elif im.mode == "I;16" and hasattr(qt_format, "Format_Grayscale16"): # Qt 5.13+ | |||||
im = im.point(lambda i: i * 256) | |||||
format = qt_format.Format_Grayscale16 | |||||
else: | |||||
if exclusive_fp: | |||||
im.close() | |||||
msg = f"unsupported image mode {repr(im.mode)}" | |||||
raise ValueError(msg) | |||||
size = im.size | |||||
__data = data or align8to32(im.tobytes(), size[0], im.mode) | |||||
if exclusive_fp: | |||||
im.close() | |||||
return {"data": __data, "size": size, "format": format, "colortable": colortable} | |||||
if qt_is_installed: | |||||
class ImageQt(QImage): | |||||
def __init__(self, im): | |||||
""" | |||||
An PIL image wrapper for Qt. This is a subclass of PyQt's QImage | |||||
class. | |||||
:param im: A PIL Image object, or a file name (given either as | |||||
Python string or a PyQt string object). | |||||
""" | |||||
im_data = _toqclass_helper(im) | |||||
# must keep a reference, or Qt will crash! | |||||
# All QImage constructors that take data operate on an existing | |||||
# buffer, so this buffer has to hang on for the life of the image. | |||||
# Fixes https://github.com/python-pillow/Pillow/issues/1370 | |||||
self.__data = im_data["data"] | |||||
super().__init__( | |||||
self.__data, | |||||
im_data["size"][0], | |||||
im_data["size"][1], | |||||
im_data["format"], | |||||
) | |||||
if im_data["colortable"]: | |||||
self.setColorTable(im_data["colortable"]) | |||||
def toqimage(im): | |||||
return ImageQt(im) | |||||
def toqpixmap(im): | |||||
# # This doesn't work. For now using a dumb approach. | |||||
# im_data = _toqclass_helper(im) | |||||
# result = QPixmap(im_data["size"][0], im_data["size"][1]) | |||||
# result.loadFromData(im_data["data"]) | |||||
qimage = toqimage(im) | |||||
return QPixmap.fromImage(qimage) |
# | |||||
# The Python Imaging Library. | |||||
# $Id$ | |||||
# | |||||
# sequence support classes | |||||
# | |||||
# history: | |||||
# 1997-02-20 fl Created | |||||
# | |||||
# Copyright (c) 1997 by Secret Labs AB. | |||||
# Copyright (c) 1997 by Fredrik Lundh. | |||||
# | |||||
# See the README file for information on usage and redistribution. | |||||
# | |||||
## | |||||
class Iterator: | |||||
""" | |||||
This class implements an iterator object that can be used to loop | |||||
over an image sequence. | |||||
You can use the ``[]`` operator to access elements by index. This operator | |||||
will raise an :py:exc:`IndexError` if you try to access a nonexistent | |||||
frame. | |||||
:param im: An image object. | |||||
""" | |||||
def __init__(self, im): | |||||
if not hasattr(im, "seek"): | |||||
msg = "im must have seek method" | |||||
raise AttributeError(msg) | |||||
self.im = im | |||||
self.position = getattr(self.im, "_min_frame", 0) | |||||
def __getitem__(self, ix): | |||||
try: | |||||
self.im.seek(ix) | |||||
return self.im | |||||
except EOFError as e: | |||||
raise IndexError from e # end of sequence | |||||
def __iter__(self): | |||||
return self | |||||
def __next__(self): | |||||
try: | |||||
self.im.seek(self.position) | |||||
self.position += 1 | |||||
return self.im | |||||
except EOFError as e: | |||||
raise StopIteration from e | |||||
def all_frames(im, func=None): | |||||
""" | |||||
Applies a given function to all frames in an image or a list of images. | |||||
The frames are returned as a list of separate images. | |||||
:param im: An image, or a list of images. | |||||
:param func: The function to apply to all of the image frames. | |||||
:returns: A list of images. | |||||
""" | |||||
if not isinstance(im, list): | |||||
im = [im] | |||||
ims = [] | |||||
for imSequence in im: | |||||
current = imSequence.tell() | |||||
ims += [im_frame.copy() for im_frame in Iterator(imSequence)] | |||||
imSequence.seek(current) | |||||
return [func(im) for im in ims] if func else ims |
# | |||||
# The Python Imaging Library. | |||||
# $Id$ | |||||
# | |||||
# im.show() drivers | |||||
# | |||||
# History: | |||||
# 2008-04-06 fl Created | |||||
# | |||||
# Copyright (c) Secret Labs AB 2008. | |||||
# | |||||
# See the README file for information on usage and redistribution. | |||||
# | |||||
import os | |||||
import shutil | |||||
import subprocess | |||||
import sys | |||||
from shlex import quote | |||||
from PIL import Image | |||||
from ._deprecate import deprecate | |||||
_viewers = [] | |||||
def register(viewer, order=1): | |||||
""" | |||||
The :py:func:`register` function is used to register additional viewers:: | |||||
from PIL import ImageShow | |||||
ImageShow.register(MyViewer()) # MyViewer will be used as a last resort | |||||
ImageShow.register(MySecondViewer(), 0) # MySecondViewer will be prioritised | |||||
ImageShow.register(ImageShow.XVViewer(), 0) # XVViewer will be prioritised | |||||
:param viewer: The viewer to be registered. | |||||
:param order: | |||||
Zero or a negative integer to prepend this viewer to the list, | |||||
a positive integer to append it. | |||||
""" | |||||
try: | |||||
if issubclass(viewer, Viewer): | |||||
viewer = viewer() | |||||
except TypeError: | |||||
pass # raised if viewer wasn't a class | |||||
if order > 0: | |||||
_viewers.append(viewer) | |||||
else: | |||||
_viewers.insert(0, viewer) | |||||
def show(image, title=None, **options): | |||||
r""" | |||||
Display a given image. | |||||
:param image: An image object. | |||||
:param title: Optional title. Not all viewers can display the title. | |||||
:param \**options: Additional viewer options. | |||||
:returns: ``True`` if a suitable viewer was found, ``False`` otherwise. | |||||
""" | |||||
for viewer in _viewers: | |||||
if viewer.show(image, title=title, **options): | |||||
return True | |||||
return False | |||||
class Viewer: | |||||
"""Base class for viewers.""" | |||||
# main api | |||||
def show(self, image, **options): | |||||
""" | |||||
The main function for displaying an image. | |||||
Converts the given image to the target format and displays it. | |||||
""" | |||||
if not ( | |||||
image.mode in ("1", "RGBA") | |||||
or (self.format == "PNG" and image.mode in ("I;16", "LA")) | |||||
): | |||||
base = Image.getmodebase(image.mode) | |||||
if image.mode != base: | |||||
image = image.convert(base) | |||||
return self.show_image(image, **options) | |||||
# hook methods | |||||
format = None | |||||
"""The format to convert the image into.""" | |||||
options = {} | |||||
"""Additional options used to convert the image.""" | |||||
def get_format(self, image): | |||||
"""Return format name, or ``None`` to save as PGM/PPM.""" | |||||
return self.format | |||||
def get_command(self, file, **options): | |||||
""" | |||||
Returns the command used to display the file. | |||||
Not implemented in the base class. | |||||
""" | |||||
raise NotImplementedError | |||||
def save_image(self, image): | |||||
"""Save to temporary file and return filename.""" | |||||
return image._dump(format=self.get_format(image), **self.options) | |||||
def show_image(self, image, **options): | |||||
"""Display the given image.""" | |||||
return self.show_file(self.save_image(image), **options) | |||||
def show_file(self, path=None, **options): | |||||
""" | |||||
Display given file. | |||||
Before Pillow 9.1.0, the first argument was ``file``. This is now deprecated, | |||||
and will be removed in Pillow 10.0.0 (2023-07-01). ``path`` should be used | |||||
instead. | |||||
""" | |||||
if path is None: | |||||
if "file" in options: | |||||
deprecate("The 'file' argument", 10, "'path'") | |||||
path = options.pop("file") | |||||
else: | |||||
msg = "Missing required argument: 'path'" | |||||
raise TypeError(msg) | |||||
os.system(self.get_command(path, **options)) # nosec | |||||
return 1 | |||||
# -------------------------------------------------------------------- | |||||
class WindowsViewer(Viewer): | |||||
"""The default viewer on Windows is the default system application for PNG files.""" | |||||
format = "PNG" | |||||
options = {"compress_level": 1, "save_all": True} | |||||
def get_command(self, file, **options): | |||||
return ( | |||||
f'start "Pillow" /WAIT "{file}" ' | |||||
"&& ping -n 4 127.0.0.1 >NUL " | |||||
f'&& del /f "{file}"' | |||||
) | |||||
if sys.platform == "win32": | |||||
register(WindowsViewer) | |||||
class MacViewer(Viewer): | |||||
"""The default viewer on macOS using ``Preview.app``.""" | |||||
format = "PNG" | |||||
options = {"compress_level": 1, "save_all": True} | |||||
def get_command(self, file, **options): | |||||
# on darwin open returns immediately resulting in the temp | |||||
# file removal while app is opening | |||||
command = "open -a Preview.app" | |||||
command = f"({command} {quote(file)}; sleep 20; rm -f {quote(file)})&" | |||||
return command | |||||
def show_file(self, path=None, **options): | |||||
""" | |||||
Display given file. | |||||
Before Pillow 9.1.0, the first argument was ``file``. This is now deprecated, | |||||
and will be removed in Pillow 10.0.0 (2023-07-01). ``path`` should be used | |||||
instead. | |||||
""" | |||||
if path is None: | |||||
if "file" in options: | |||||
deprecate("The 'file' argument", 10, "'path'") | |||||
path = options.pop("file") | |||||
else: | |||||
msg = "Missing required argument: 'path'" | |||||
raise TypeError(msg) | |||||
subprocess.call(["open", "-a", "Preview.app", path]) | |||||
executable = sys.executable or shutil.which("python3") | |||||
if executable: | |||||
subprocess.Popen( | |||||
[ | |||||
executable, | |||||
"-c", | |||||
"import os, sys, time; time.sleep(20); os.remove(sys.argv[1])", | |||||
path, | |||||
] | |||||
) | |||||
return 1 | |||||
if sys.platform == "darwin": | |||||
register(MacViewer) | |||||
class UnixViewer(Viewer): | |||||
format = "PNG" | |||||
options = {"compress_level": 1, "save_all": True} | |||||
def get_command(self, file, **options): | |||||
command = self.get_command_ex(file, **options)[0] | |||||
return f"({command} {quote(file)}" | |||||
class XDGViewer(UnixViewer): | |||||
""" | |||||
The freedesktop.org ``xdg-open`` command. | |||||
""" | |||||
def get_command_ex(self, file, **options): | |||||
command = executable = "xdg-open" | |||||
return command, executable | |||||
def show_file(self, path=None, **options): | |||||
""" | |||||
Display given file. | |||||
Before Pillow 9.1.0, the first argument was ``file``. This is now deprecated, | |||||
and will be removed in Pillow 10.0.0 (2023-07-01). ``path`` should be used | |||||
instead. | |||||
""" | |||||
if path is None: | |||||
if "file" in options: | |||||
deprecate("The 'file' argument", 10, "'path'") | |||||
path = options.pop("file") | |||||
else: | |||||
msg = "Missing required argument: 'path'" | |||||
raise TypeError(msg) | |||||
subprocess.Popen(["xdg-open", path]) | |||||
return 1 | |||||
class DisplayViewer(UnixViewer): | |||||
""" | |||||
The ImageMagick ``display`` command. | |||||
This viewer supports the ``title`` parameter. | |||||
""" | |||||
def get_command_ex(self, file, title=None, **options): | |||||
command = executable = "display" | |||||
if title: | |||||
command += f" -title {quote(title)}" | |||||
return command, executable | |||||
def show_file(self, path=None, **options): | |||||
""" | |||||
Display given file. | |||||
Before Pillow 9.1.0, the first argument was ``file``. This is now deprecated, | |||||
and ``path`` should be used instead. | |||||
""" | |||||
if path is None: | |||||
if "file" in options: | |||||
deprecate("The 'file' argument", 10, "'path'") | |||||
path = options.pop("file") | |||||
else: | |||||
msg = "Missing required argument: 'path'" | |||||
raise TypeError(msg) | |||||
args = ["display"] | |||||
title = options.get("title") | |||||
if title: | |||||
args += ["-title", title] | |||||
args.append(path) | |||||
subprocess.Popen(args) | |||||
return 1 | |||||
class GmDisplayViewer(UnixViewer): | |||||
"""The GraphicsMagick ``gm display`` command.""" | |||||
def get_command_ex(self, file, **options): | |||||
executable = "gm" | |||||
command = "gm display" | |||||
return command, executable | |||||
def show_file(self, path=None, **options): | |||||
""" | |||||
Display given file. | |||||
Before Pillow 9.1.0, the first argument was ``file``. This is now deprecated, | |||||
and ``path`` should be used instead. | |||||
""" | |||||
if path is None: | |||||
if "file" in options: | |||||
deprecate("The 'file' argument", 10, "'path'") | |||||
path = options.pop("file") | |||||
else: | |||||
msg = "Missing required argument: 'path'" | |||||
raise TypeError(msg) | |||||
subprocess.Popen(["gm", "display", path]) | |||||
return 1 | |||||
class EogViewer(UnixViewer): | |||||
"""The GNOME Image Viewer ``eog`` command.""" | |||||
def get_command_ex(self, file, **options): | |||||
executable = "eog" | |||||
command = "eog -n" | |||||
return command, executable | |||||
def show_file(self, path=None, **options): | |||||
""" | |||||
Display given file. | |||||
Before Pillow 9.1.0, the first argument was ``file``. This is now deprecated, | |||||
and ``path`` should be used instead. | |||||
""" | |||||
if path is None: | |||||
if "file" in options: | |||||
deprecate("The 'file' argument", 10, "'path'") | |||||
path = options.pop("file") | |||||
else: | |||||
msg = "Missing required argument: 'path'" | |||||
raise TypeError(msg) | |||||
subprocess.Popen(["eog", "-n", path]) | |||||
return 1 | |||||
class XVViewer(UnixViewer): | |||||
""" | |||||
The X Viewer ``xv`` command. | |||||
This viewer supports the ``title`` parameter. | |||||
""" | |||||
def get_command_ex(self, file, title=None, **options): | |||||
# note: xv is pretty outdated. most modern systems have | |||||
# imagemagick's display command instead. | |||||
command = executable = "xv" | |||||
if title: | |||||
command += f" -name {quote(title)}" | |||||
return command, executable | |||||
def show_file(self, path=None, **options): | |||||
""" | |||||
Display given file. | |||||
Before Pillow 9.1.0, the first argument was ``file``. This is now deprecated, | |||||
and ``path`` should be used instead. | |||||
""" | |||||
if path is None: | |||||
if "file" in options: | |||||
deprecate("The 'file' argument", 10, "'path'") | |||||
path = options.pop("file") | |||||
else: | |||||
msg = "Missing required argument: 'path'" | |||||
raise TypeError(msg) | |||||
args = ["xv"] | |||||
title = options.get("title") | |||||
if title: | |||||
args += ["-name", title] | |||||
args.append(path) | |||||
subprocess.Popen(args) | |||||
return 1 | |||||
if sys.platform not in ("win32", "darwin"): # unixoids | |||||
if shutil.which("xdg-open"): | |||||
register(XDGViewer) | |||||
if shutil.which("display"): | |||||
register(DisplayViewer) | |||||
if shutil.which("gm"): | |||||
register(GmDisplayViewer) | |||||
if shutil.which("eog"): | |||||
register(EogViewer) | |||||
if shutil.which("xv"): | |||||
register(XVViewer) | |||||
class IPythonViewer(Viewer): | |||||
"""The viewer for IPython frontends.""" | |||||
def show_image(self, image, **options): | |||||
ipython_display(image) | |||||
return 1 | |||||
try: | |||||
from IPython.display import display as ipython_display | |||||
except ImportError: | |||||
pass | |||||
else: | |||||
register(IPythonViewer) | |||||
if __name__ == "__main__": | |||||
if len(sys.argv) < 2: | |||||
print("Syntax: python3 ImageShow.py imagefile [title]") | |||||
sys.exit() | |||||
with Image.open(sys.argv[1]) as im: | |||||
print(show(im, *sys.argv[2:])) |
# | |||||
# The Python Imaging Library. | |||||
# $Id$ | |||||
# | |||||
# global image statistics | |||||
# | |||||
# History: | |||||
# 1996-04-05 fl Created | |||||
# 1997-05-21 fl Added mask; added rms, var, stddev attributes | |||||
# 1997-08-05 fl Added median | |||||
# 1998-07-05 hk Fixed integer overflow error | |||||
# | |||||
# Notes: | |||||
# This class shows how to implement delayed evaluation of attributes. | |||||
# To get a certain value, simply access the corresponding attribute. | |||||
# The __getattr__ dispatcher takes care of the rest. | |||||
# | |||||
# Copyright (c) Secret Labs AB 1997. | |||||
# Copyright (c) Fredrik Lundh 1996-97. | |||||
# | |||||
# See the README file for information on usage and redistribution. | |||||
# | |||||
import functools | |||||
import math | |||||
import operator | |||||
class Stat: | |||||
def __init__(self, image_or_list, mask=None): | |||||
try: | |||||
if mask: | |||||
self.h = image_or_list.histogram(mask) | |||||
else: | |||||
self.h = image_or_list.histogram() | |||||
except AttributeError: | |||||
self.h = image_or_list # assume it to be a histogram list | |||||
if not isinstance(self.h, list): | |||||
msg = "first argument must be image or list" | |||||
raise TypeError(msg) | |||||
self.bands = list(range(len(self.h) // 256)) | |||||
def __getattr__(self, id): | |||||
"""Calculate missing attribute""" | |||||
if id[:4] == "_get": | |||||
raise AttributeError(id) | |||||
# calculate missing attribute | |||||
v = getattr(self, "_get" + id)() | |||||
setattr(self, id, v) | |||||
return v | |||||
def _getextrema(self): | |||||
"""Get min/max values for each band in the image""" | |||||
def minmax(histogram): | |||||
n = 255 | |||||
x = 0 | |||||
for i in range(256): | |||||
if histogram[i]: | |||||
n = min(n, i) | |||||
x = max(x, i) | |||||
return n, x # returns (255, 0) if there's no data in the histogram | |||||
v = [] | |||||
for i in range(0, len(self.h), 256): | |||||
v.append(minmax(self.h[i:])) | |||||
return v | |||||
def _getcount(self): | |||||
"""Get total number of pixels in each layer""" | |||||
v = [] | |||||
for i in range(0, len(self.h), 256): | |||||
v.append(functools.reduce(operator.add, self.h[i : i + 256])) | |||||
return v | |||||
def _getsum(self): | |||||
"""Get sum of all pixels in each layer""" | |||||
v = [] | |||||
for i in range(0, len(self.h), 256): | |||||
layer_sum = 0.0 | |||||
for j in range(256): | |||||
layer_sum += j * self.h[i + j] | |||||
v.append(layer_sum) | |||||
return v | |||||
def _getsum2(self): | |||||
"""Get squared sum of all pixels in each layer""" | |||||
v = [] | |||||
for i in range(0, len(self.h), 256): | |||||
sum2 = 0.0 | |||||
for j in range(256): | |||||
sum2 += (j**2) * float(self.h[i + j]) | |||||
v.append(sum2) | |||||
return v | |||||
def _getmean(self): | |||||
"""Get average pixel level for each layer""" | |||||
v = [] | |||||
for i in self.bands: | |||||
v.append(self.sum[i] / self.count[i]) | |||||
return v | |||||
def _getmedian(self): | |||||
"""Get median pixel level for each layer""" | |||||
v = [] | |||||
for i in self.bands: | |||||
s = 0 | |||||
half = self.count[i] // 2 | |||||
b = i * 256 | |||||
for j in range(256): | |||||
s = s + self.h[b + j] | |||||
if s > half: | |||||
break | |||||
v.append(j) | |||||
return v | |||||
def _getrms(self): | |||||
"""Get RMS for each layer""" | |||||
v = [] | |||||
for i in self.bands: | |||||
v.append(math.sqrt(self.sum2[i] / self.count[i])) | |||||
return v | |||||
def _getvar(self): | |||||
"""Get variance for each layer""" | |||||
v = [] | |||||
for i in self.bands: | |||||
n = self.count[i] | |||||
v.append((self.sum2[i] - (self.sum[i] ** 2.0) / n) / n) | |||||
return v | |||||
def _getstddev(self): | |||||
"""Get standard deviation for each layer""" | |||||
v = [] | |||||
for i in self.bands: | |||||
v.append(math.sqrt(self.var[i])) | |||||
return v | |||||
Global = Stat # compatibility |
# | |||||
# The Python Imaging Library. | |||||
# $Id$ | |||||
# | |||||
# a Tk display interface | |||||
# | |||||
# History: | |||||
# 96-04-08 fl Created | |||||
# 96-09-06 fl Added getimage method | |||||
# 96-11-01 fl Rewritten, removed image attribute and crop method | |||||
# 97-05-09 fl Use PyImagingPaste method instead of image type | |||||
# 97-05-12 fl Minor tweaks to match the IFUNC95 interface | |||||
# 97-05-17 fl Support the "pilbitmap" booster patch | |||||
# 97-06-05 fl Added file= and data= argument to image constructors | |||||
# 98-03-09 fl Added width and height methods to Image classes | |||||
# 98-07-02 fl Use default mode for "P" images without palette attribute | |||||
# 98-07-02 fl Explicitly destroy Tkinter image objects | |||||
# 99-07-24 fl Support multiple Tk interpreters (from Greg Couch) | |||||
# 99-07-26 fl Automatically hook into Tkinter (if possible) | |||||
# 99-08-15 fl Hook uses _imagingtk instead of _imaging | |||||
# | |||||
# Copyright (c) 1997-1999 by Secret Labs AB | |||||
# Copyright (c) 1996-1997 by Fredrik Lundh | |||||
# | |||||
# See the README file for information on usage and redistribution. | |||||
# | |||||
import tkinter | |||||
from io import BytesIO | |||||
from . import Image | |||||
from ._deprecate import deprecate | |||||
# -------------------------------------------------------------------- | |||||
# Check for Tkinter interface hooks | |||||
_pilbitmap_ok = None | |||||
def _pilbitmap_check(): | |||||
global _pilbitmap_ok | |||||
if _pilbitmap_ok is None: | |||||
try: | |||||
im = Image.new("1", (1, 1)) | |||||
tkinter.BitmapImage(data=f"PIL:{im.im.id}") | |||||
_pilbitmap_ok = 1 | |||||
except tkinter.TclError: | |||||
_pilbitmap_ok = 0 | |||||
return _pilbitmap_ok | |||||
def _get_image_from_kw(kw): | |||||
source = None | |||||
if "file" in kw: | |||||
source = kw.pop("file") | |||||
elif "data" in kw: | |||||
source = BytesIO(kw.pop("data")) | |||||
if source: | |||||
return Image.open(source) | |||||
def _pyimagingtkcall(command, photo, id): | |||||
tk = photo.tk | |||||
try: | |||||
tk.call(command, photo, id) | |||||
except tkinter.TclError: | |||||
# activate Tkinter hook | |||||
# may raise an error if it cannot attach to Tkinter | |||||
from . import _imagingtk | |||||
_imagingtk.tkinit(tk.interpaddr()) | |||||
tk.call(command, photo, id) | |||||
# -------------------------------------------------------------------- | |||||
# PhotoImage | |||||
class PhotoImage: | |||||
""" | |||||
A Tkinter-compatible photo image. This can be used | |||||
everywhere Tkinter expects an image object. If the image is an RGBA | |||||
image, pixels having alpha 0 are treated as transparent. | |||||
The constructor takes either a PIL image, or a mode and a size. | |||||
Alternatively, you can use the ``file`` or ``data`` options to initialize | |||||
the photo image object. | |||||
:param image: Either a PIL image, or a mode string. If a mode string is | |||||
used, a size must also be given. | |||||
:param size: If the first argument is a mode string, this defines the size | |||||
of the image. | |||||
:keyword file: A filename to load the image from (using | |||||
``Image.open(file)``). | |||||
:keyword data: An 8-bit string containing image data (as loaded from an | |||||
image file). | |||||
""" | |||||
def __init__(self, image=None, size=None, **kw): | |||||
# Tk compatibility: file or data | |||||
if image is None: | |||||
image = _get_image_from_kw(kw) | |||||
if hasattr(image, "mode") and hasattr(image, "size"): | |||||
# got an image instead of a mode | |||||
mode = image.mode | |||||
if mode == "P": | |||||
# palette mapped data | |||||
image.apply_transparency() | |||||
image.load() | |||||
try: | |||||
mode = image.palette.mode | |||||
except AttributeError: | |||||
mode = "RGB" # default | |||||
size = image.size | |||||
kw["width"], kw["height"] = size | |||||
else: | |||||
mode = image | |||||
image = None | |||||
if mode not in ["1", "L", "RGB", "RGBA"]: | |||||
mode = Image.getmodebase(mode) | |||||
self.__mode = mode | |||||
self.__size = size | |||||
self.__photo = tkinter.PhotoImage(**kw) | |||||
self.tk = self.__photo.tk | |||||
if image: | |||||
self.paste(image) | |||||
def __del__(self): | |||||
name = self.__photo.name | |||||
self.__photo.name = None | |||||
try: | |||||
self.__photo.tk.call("image", "delete", name) | |||||
except Exception: | |||||
pass # ignore internal errors | |||||
def __str__(self): | |||||
""" | |||||
Get the Tkinter photo image identifier. This method is automatically | |||||
called by Tkinter whenever a PhotoImage object is passed to a Tkinter | |||||
method. | |||||
:return: A Tkinter photo image identifier (a string). | |||||
""" | |||||
return str(self.__photo) | |||||
def width(self): | |||||
""" | |||||
Get the width of the image. | |||||
:return: The width, in pixels. | |||||
""" | |||||
return self.__size[0] | |||||
def height(self): | |||||
""" | |||||
Get the height of the image. | |||||
:return: The height, in pixels. | |||||
""" | |||||
return self.__size[1] | |||||
def paste(self, im, box=None): | |||||
""" | |||||
Paste a PIL image into the photo image. Note that this can | |||||
be very slow if the photo image is displayed. | |||||
:param im: A PIL image. The size must match the target region. If the | |||||
mode does not match, the image is converted to the mode of | |||||
the bitmap image. | |||||
:param box: Deprecated. This parameter will be removed in Pillow 10 | |||||
(2023-07-01). | |||||
""" | |||||
if box is not None: | |||||
deprecate("The box parameter", 10, None) | |||||
# convert to blittable | |||||
im.load() | |||||
image = im.im | |||||
if image.isblock() and im.mode == self.__mode: | |||||
block = image | |||||
else: | |||||
block = image.new_block(self.__mode, im.size) | |||||
image.convert2(block, image) # convert directly between buffers | |||||
_pyimagingtkcall("PyImagingPhoto", self.__photo, block.id) | |||||
# -------------------------------------------------------------------- | |||||
# BitmapImage | |||||
class BitmapImage: | |||||
""" | |||||
A Tkinter-compatible bitmap image. This can be used everywhere Tkinter | |||||
expects an image object. | |||||
The given image must have mode "1". Pixels having value 0 are treated as | |||||
transparent. Options, if any, are passed on to Tkinter. The most commonly | |||||
used option is ``foreground``, which is used to specify the color for the | |||||
non-transparent parts. See the Tkinter documentation for information on | |||||
how to specify colours. | |||||
:param image: A PIL image. | |||||
""" | |||||
def __init__(self, image=None, **kw): | |||||
# Tk compatibility: file or data | |||||
if image is None: | |||||
image = _get_image_from_kw(kw) | |||||
self.__mode = image.mode | |||||
self.__size = image.size | |||||
if _pilbitmap_check(): | |||||
# fast way (requires the pilbitmap booster patch) | |||||
image.load() | |||||
kw["data"] = f"PIL:{image.im.id}" | |||||
self.__im = image # must keep a reference | |||||
else: | |||||
# slow but safe way | |||||
kw["data"] = image.tobitmap() | |||||
self.__photo = tkinter.BitmapImage(**kw) | |||||
def __del__(self): | |||||
name = self.__photo.name | |||||
self.__photo.name = None | |||||
try: | |||||
self.__photo.tk.call("image", "delete", name) | |||||
except Exception: | |||||
pass # ignore internal errors | |||||
def width(self): | |||||
""" | |||||
Get the width of the image. | |||||
:return: The width, in pixels. | |||||
""" | |||||
return self.__size[0] | |||||
def height(self): | |||||
""" | |||||
Get the height of the image. | |||||
:return: The height, in pixels. | |||||
""" | |||||
return self.__size[1] | |||||
def __str__(self): | |||||
""" | |||||
Get the Tkinter bitmap image identifier. This method is automatically | |||||
called by Tkinter whenever a BitmapImage object is passed to a Tkinter | |||||
method. | |||||
:return: A Tkinter bitmap image identifier (a string). | |||||
""" | |||||
return str(self.__photo) | |||||
def getimage(photo): | |||||
"""Copies the contents of a PhotoImage to a PIL image memory.""" | |||||
im = Image.new("RGBA", (photo.width(), photo.height())) | |||||
block = im.im | |||||
_pyimagingtkcall("PyImagingPhotoGet", photo, block.id) | |||||
return im | |||||
def _show(image, title): | |||||
"""Helper for the Image.show method.""" | |||||
class UI(tkinter.Label): | |||||
def __init__(self, master, im): | |||||
if im.mode == "1": | |||||
self.image = BitmapImage(im, foreground="white", master=master) | |||||
else: | |||||
self.image = PhotoImage(im, master=master) | |||||
super().__init__(master, image=self.image, bg="black", bd=0) | |||||
if not tkinter._default_root: | |||||
msg = "tkinter not initialized" | |||||
raise OSError(msg) | |||||
top = tkinter.Toplevel() | |||||
if title: | |||||
top.title(title) | |||||
UI(top, image).pack() |
# | |||||
# The Python Imaging Library. | |||||
# $Id$ | |||||
# | |||||
# transform wrappers | |||||
# | |||||
# History: | |||||
# 2002-04-08 fl Created | |||||
# | |||||
# Copyright (c) 2002 by Secret Labs AB | |||||
# Copyright (c) 2002 by Fredrik Lundh | |||||
# | |||||
# See the README file for information on usage and redistribution. | |||||
# | |||||
from . import Image | |||||
class Transform(Image.ImageTransformHandler): | |||||
def __init__(self, data): | |||||
self.data = data | |||||
def getdata(self): | |||||
return self.method, self.data | |||||
def transform(self, size, image, **options): | |||||
# can be overridden | |||||
method, data = self.getdata() | |||||
return image.transform(size, method, data, **options) | |||||
class AffineTransform(Transform): | |||||
""" | |||||
Define an affine image transform. | |||||
This function takes a 6-tuple (a, b, c, d, e, f) which contain the first | |||||
two rows from an affine transform matrix. For each pixel (x, y) in the | |||||
output image, the new value is taken from a position (a x + b y + c, | |||||
d x + e y + f) in the input image, rounded to nearest pixel. | |||||
This function can be used to scale, translate, rotate, and shear the | |||||
original image. | |||||
See :py:meth:`~PIL.Image.Image.transform` | |||||
:param matrix: A 6-tuple (a, b, c, d, e, f) containing the first two rows | |||||
from an affine transform matrix. | |||||
""" | |||||
method = Image.Transform.AFFINE | |||||
class ExtentTransform(Transform): | |||||
""" | |||||
Define a transform to extract a subregion from an image. | |||||
Maps a rectangle (defined by two corners) from the image to a rectangle of | |||||
the given size. The resulting image will contain data sampled from between | |||||
the corners, such that (x0, y0) in the input image will end up at (0,0) in | |||||
the output image, and (x1, y1) at size. | |||||
This method can be used to crop, stretch, shrink, or mirror an arbitrary | |||||
rectangle in the current image. It is slightly slower than crop, but about | |||||
as fast as a corresponding resize operation. | |||||
See :py:meth:`~PIL.Image.Image.transform` | |||||
:param bbox: A 4-tuple (x0, y0, x1, y1) which specifies two points in the | |||||
input image's coordinate system. See :ref:`coordinate-system`. | |||||
""" | |||||
method = Image.Transform.EXTENT | |||||
class QuadTransform(Transform): | |||||
""" | |||||
Define a quad image transform. | |||||
Maps a quadrilateral (a region defined by four corners) from the image to a | |||||
rectangle of the given size. | |||||
See :py:meth:`~PIL.Image.Image.transform` | |||||
:param xy: An 8-tuple (x0, y0, x1, y1, x2, y2, x3, y3) which contain the | |||||
upper left, lower left, lower right, and upper right corner of the | |||||
source quadrilateral. | |||||
""" | |||||
method = Image.Transform.QUAD | |||||
class MeshTransform(Transform): | |||||
""" | |||||
Define a mesh image transform. A mesh transform consists of one or more | |||||
individual quad transforms. | |||||
See :py:meth:`~PIL.Image.Image.transform` | |||||
:param data: A list of (bbox, quad) tuples. | |||||
""" | |||||
method = Image.Transform.MESH |
# | |||||
# The Python Imaging Library. | |||||
# $Id$ | |||||
# | |||||
# a Windows DIB display interface | |||||
# | |||||
# History: | |||||
# 1996-05-20 fl Created | |||||
# 1996-09-20 fl Fixed subregion exposure | |||||
# 1997-09-21 fl Added draw primitive (for tzPrint) | |||||
# 2003-05-21 fl Added experimental Window/ImageWindow classes | |||||
# 2003-09-05 fl Added fromstring/tostring methods | |||||
# | |||||
# Copyright (c) Secret Labs AB 1997-2003. | |||||
# Copyright (c) Fredrik Lundh 1996-2003. | |||||
# | |||||
# See the README file for information on usage and redistribution. | |||||
# | |||||
from . import Image | |||||
class HDC: | |||||
""" | |||||
Wraps an HDC integer. The resulting object can be passed to the | |||||
:py:meth:`~PIL.ImageWin.Dib.draw` and :py:meth:`~PIL.ImageWin.Dib.expose` | |||||
methods. | |||||
""" | |||||
def __init__(self, dc): | |||||
self.dc = dc | |||||
def __int__(self): | |||||
return self.dc | |||||
class HWND: | |||||
""" | |||||
Wraps an HWND integer. The resulting object can be passed to the | |||||
:py:meth:`~PIL.ImageWin.Dib.draw` and :py:meth:`~PIL.ImageWin.Dib.expose` | |||||
methods, instead of a DC. | |||||
""" | |||||
def __init__(self, wnd): | |||||
self.wnd = wnd | |||||
def __int__(self): | |||||
return self.wnd | |||||
class Dib: | |||||
""" | |||||
A Windows bitmap with the given mode and size. The mode can be one of "1", | |||||
"L", "P", or "RGB". | |||||
If the display requires a palette, this constructor creates a suitable | |||||
palette and associates it with the image. For an "L" image, 128 greylevels | |||||
are allocated. For an "RGB" image, a 6x6x6 colour cube is used, together | |||||
with 20 greylevels. | |||||
To make sure that palettes work properly under Windows, you must call the | |||||
``palette`` method upon certain events from Windows. | |||||
:param image: Either a PIL image, or a mode string. If a mode string is | |||||
used, a size must also be given. The mode can be one of "1", | |||||
"L", "P", or "RGB". | |||||
:param size: If the first argument is a mode string, this | |||||
defines the size of the image. | |||||
""" | |||||
def __init__(self, image, size=None): | |||||
if hasattr(image, "mode") and hasattr(image, "size"): | |||||
mode = image.mode | |||||
size = image.size | |||||
else: | |||||
mode = image | |||||
image = None | |||||
if mode not in ["1", "L", "P", "RGB"]: | |||||
mode = Image.getmodebase(mode) | |||||
self.image = Image.core.display(mode, size) | |||||
self.mode = mode | |||||
self.size = size | |||||
if image: | |||||
self.paste(image) | |||||
def expose(self, handle): | |||||
""" | |||||
Copy the bitmap contents to a device context. | |||||
:param handle: Device context (HDC), cast to a Python integer, or an | |||||
HDC or HWND instance. In PythonWin, you can use | |||||
``CDC.GetHandleAttrib()`` to get a suitable handle. | |||||
""" | |||||
if isinstance(handle, HWND): | |||||
dc = self.image.getdc(handle) | |||||
try: | |||||
result = self.image.expose(dc) | |||||
finally: | |||||
self.image.releasedc(handle, dc) | |||||
else: | |||||
result = self.image.expose(handle) | |||||
return result | |||||
def draw(self, handle, dst, src=None): | |||||
""" | |||||
Same as expose, but allows you to specify where to draw the image, and | |||||
what part of it to draw. | |||||
The destination and source areas are given as 4-tuple rectangles. If | |||||
the source is omitted, the entire image is copied. If the source and | |||||
the destination have different sizes, the image is resized as | |||||
necessary. | |||||
""" | |||||
if not src: | |||||
src = (0, 0) + self.size | |||||
if isinstance(handle, HWND): | |||||
dc = self.image.getdc(handle) | |||||
try: | |||||
result = self.image.draw(dc, dst, src) | |||||
finally: | |||||
self.image.releasedc(handle, dc) | |||||
else: | |||||
result = self.image.draw(handle, dst, src) | |||||
return result | |||||
def query_palette(self, handle): | |||||
""" | |||||
Installs the palette associated with the image in the given device | |||||
context. | |||||
This method should be called upon **QUERYNEWPALETTE** and | |||||
**PALETTECHANGED** events from Windows. If this method returns a | |||||
non-zero value, one or more display palette entries were changed, and | |||||
the image should be redrawn. | |||||
:param handle: Device context (HDC), cast to a Python integer, or an | |||||
HDC or HWND instance. | |||||
:return: A true value if one or more entries were changed (this | |||||
indicates that the image should be redrawn). | |||||
""" | |||||
if isinstance(handle, HWND): | |||||
handle = self.image.getdc(handle) | |||||
try: | |||||
result = self.image.query_palette(handle) | |||||
finally: | |||||
self.image.releasedc(handle, handle) | |||||
else: | |||||
result = self.image.query_palette(handle) | |||||
return result | |||||
def paste(self, im, box=None): | |||||
""" | |||||
Paste a PIL image into the bitmap image. | |||||
:param im: A PIL image. The size must match the target region. | |||||
If the mode does not match, the image is converted to the | |||||
mode of the bitmap image. | |||||
:param box: A 4-tuple defining the left, upper, right, and | |||||
lower pixel coordinate. See :ref:`coordinate-system`. If | |||||
None is given instead of a tuple, all of the image is | |||||
assumed. | |||||
""" | |||||
im.load() | |||||
if self.mode != im.mode: | |||||
im = im.convert(self.mode) | |||||
if box: | |||||
self.image.paste(im.im, box) | |||||
else: | |||||
self.image.paste(im.im) | |||||
def frombytes(self, buffer): | |||||
""" | |||||
Load display memory contents from byte data. | |||||
:param buffer: A buffer containing display data (usually | |||||
data returned from :py:func:`~PIL.ImageWin.Dib.tobytes`) | |||||
""" | |||||
return self.image.frombytes(buffer) | |||||
def tobytes(self): | |||||
""" | |||||
Copy display memory contents to bytes object. | |||||
:return: A bytes object containing display data. | |||||
""" | |||||
return self.image.tobytes() | |||||
class Window: | |||||
"""Create a Window with the given title size.""" | |||||
def __init__(self, title="PIL", width=None, height=None): | |||||
self.hwnd = Image.core.createwindow( | |||||
title, self.__dispatcher, width or 0, height or 0 | |||||
) | |||||
def __dispatcher(self, action, *args): | |||||
return getattr(self, "ui_handle_" + action)(*args) | |||||
def ui_handle_clear(self, dc, x0, y0, x1, y1): | |||||
pass | |||||
def ui_handle_damage(self, x0, y0, x1, y1): | |||||
pass | |||||
def ui_handle_destroy(self): | |||||
pass | |||||
def ui_handle_repair(self, dc, x0, y0, x1, y1): | |||||
pass | |||||
def ui_handle_resize(self, width, height): | |||||
pass | |||||
def mainloop(self): | |||||
Image.core.eventloop() | |||||
class ImageWindow(Window): | |||||
"""Create an image window which displays the given image.""" | |||||
def __init__(self, image, title="PIL"): | |||||
if not isinstance(image, Dib): | |||||
image = Dib(image) | |||||
self.image = image | |||||
width, height = image.size | |||||
super().__init__(title, width=width, height=height) | |||||
def ui_handle_repair(self, dc, x0, y0, x1, y1): | |||||
self.image.draw(dc, (x0, y0, x1, y1)) |
# | |||||
# The Python Imaging Library. | |||||
# $Id$ | |||||
# | |||||
# IM Tools support for PIL | |||||
# | |||||
# history: | |||||
# 1996-05-27 fl Created (read 8-bit images only) | |||||
# 2001-02-17 fl Use 're' instead of 'regex' (Python 2.1) (0.2) | |||||
# | |||||
# Copyright (c) Secret Labs AB 1997-2001. | |||||
# Copyright (c) Fredrik Lundh 1996-2001. | |||||
# | |||||
# See the README file for information on usage and redistribution. | |||||
# | |||||
import re | |||||
from . import Image, ImageFile | |||||
# | |||||
# -------------------------------------------------------------------- | |||||
field = re.compile(rb"([a-z]*) ([^ \r\n]*)") | |||||
## | |||||
# Image plugin for IM Tools images. | |||||
class ImtImageFile(ImageFile.ImageFile): | |||||
format = "IMT" | |||||
format_description = "IM Tools" | |||||
def _open(self): | |||||
# Quick rejection: if there's not a LF among the first | |||||
# 100 bytes, this is (probably) not a text header. | |||||
buffer = self.fp.read(100) | |||||
if b"\n" not in buffer: | |||||
msg = "not an IM file" | |||||
raise SyntaxError(msg) | |||||
xsize = ysize = 0 | |||||
while True: | |||||
if buffer: | |||||
s = buffer[:1] | |||||
buffer = buffer[1:] | |||||
else: | |||||
s = self.fp.read(1) | |||||
if not s: | |||||
break | |||||
if s == b"\x0C": | |||||
# image data begins | |||||
self.tile = [ | |||||
( | |||||
"raw", | |||||
(0, 0) + self.size, | |||||
self.fp.tell() - len(buffer), | |||||
(self.mode, 0, 1), | |||||
) | |||||
] | |||||
break | |||||
else: | |||||
# read key/value pair | |||||
if b"\n" not in buffer: | |||||
buffer += self.fp.read(100) | |||||
lines = buffer.split(b"\n") | |||||
s += lines.pop(0) | |||||
buffer = b"\n".join(lines) | |||||
if len(s) == 1 or len(s) > 100: | |||||
break | |||||
if s[0] == ord(b"*"): | |||||
continue # comment | |||||
m = field.match(s) | |||||
if not m: | |||||
break | |||||
k, v = m.group(1, 2) | |||||
if k == b"width": | |||||
xsize = int(v) | |||||
self._size = xsize, ysize | |||||
elif k == b"height": | |||||
ysize = int(v) | |||||
self._size = xsize, ysize | |||||
elif k == b"pixel" and v == b"n8": | |||||
self.mode = "L" | |||||
# | |||||
# -------------------------------------------------------------------- | |||||
Image.register_open(ImtImageFile.format, ImtImageFile) | |||||
# | |||||
# no extension registered (".im" is simply too common) |
# | |||||
# The Python Imaging Library. | |||||
# $Id$ | |||||
# | |||||
# IPTC/NAA file handling | |||||
# | |||||
# history: | |||||
# 1995-10-01 fl Created | |||||
# 1998-03-09 fl Cleaned up and added to PIL | |||||
# 2002-06-18 fl Added getiptcinfo helper | |||||
# | |||||
# Copyright (c) Secret Labs AB 1997-2002. | |||||
# Copyright (c) Fredrik Lundh 1995. | |||||
# | |||||
# See the README file for information on usage and redistribution. | |||||
# | |||||
import os | |||||
import tempfile | |||||
from . import Image, ImageFile | |||||
from ._binary import i8 | |||||
from ._binary import i16be as i16 | |||||
from ._binary import i32be as i32 | |||||
from ._binary import o8 | |||||
COMPRESSION = {1: "raw", 5: "jpeg"} | |||||
PAD = o8(0) * 4 | |||||
# | |||||
# Helpers | |||||
def i(c): | |||||
return i32((PAD + c)[-4:]) | |||||
def dump(c): | |||||
for i in c: | |||||
print("%02x" % i8(i), end=" ") | |||||
print() | |||||
## | |||||
# Image plugin for IPTC/NAA datastreams. To read IPTC/NAA fields | |||||
# from TIFF and JPEG files, use the <b>getiptcinfo</b> function. | |||||
class IptcImageFile(ImageFile.ImageFile): | |||||
format = "IPTC" | |||||
format_description = "IPTC/NAA" | |||||
def getint(self, key): | |||||
return i(self.info[key]) | |||||
def field(self): | |||||
# | |||||
# get a IPTC field header | |||||
s = self.fp.read(5) | |||||
if not len(s): | |||||
return None, 0 | |||||
tag = s[1], s[2] | |||||
# syntax | |||||
if s[0] != 0x1C or tag[0] < 1 or tag[0] > 9: | |||||
msg = "invalid IPTC/NAA file" | |||||
raise SyntaxError(msg) | |||||
# field size | |||||
size = s[3] | |||||
if size > 132: | |||||
msg = "illegal field length in IPTC/NAA file" | |||||
raise OSError(msg) | |||||
elif size == 128: | |||||
size = 0 | |||||
elif size > 128: | |||||
size = i(self.fp.read(size - 128)) | |||||
else: | |||||
size = i16(s, 3) | |||||
return tag, size | |||||
def _open(self): | |||||
# load descriptive fields | |||||
while True: | |||||
offset = self.fp.tell() | |||||
tag, size = self.field() | |||||
if not tag or tag == (8, 10): | |||||
break | |||||
if size: | |||||
tagdata = self.fp.read(size) | |||||
else: | |||||
tagdata = None | |||||
if tag in self.info: | |||||
if isinstance(self.info[tag], list): | |||||
self.info[tag].append(tagdata) | |||||
else: | |||||
self.info[tag] = [self.info[tag], tagdata] | |||||
else: | |||||
self.info[tag] = tagdata | |||||
# mode | |||||
layers = i8(self.info[(3, 60)][0]) | |||||
component = i8(self.info[(3, 60)][1]) | |||||
if (3, 65) in self.info: | |||||
id = i8(self.info[(3, 65)][0]) - 1 | |||||
else: | |||||
id = 0 | |||||
if layers == 1 and not component: | |||||
self.mode = "L" | |||||
elif layers == 3 and component: | |||||
self.mode = "RGB"[id] | |||||
elif layers == 4 and component: | |||||
self.mode = "CMYK"[id] | |||||
# size | |||||
self._size = self.getint((3, 20)), self.getint((3, 30)) | |||||
# compression | |||||
try: | |||||
compression = COMPRESSION[self.getint((3, 120))] | |||||
except KeyError as e: | |||||
msg = "Unknown IPTC image compression" | |||||
raise OSError(msg) from e | |||||
# tile | |||||
if tag == (8, 10): | |||||
self.tile = [ | |||||
("iptc", (compression, offset), (0, 0, self.size[0], self.size[1])) | |||||
] | |||||
def load(self): | |||||
if len(self.tile) != 1 or self.tile[0][0] != "iptc": | |||||
return ImageFile.ImageFile.load(self) | |||||
type, tile, box = self.tile[0] | |||||
encoding, offset = tile | |||||
self.fp.seek(offset) | |||||
# Copy image data to temporary file | |||||
o_fd, outfile = tempfile.mkstemp(text=False) | |||||
o = os.fdopen(o_fd) | |||||
if encoding == "raw": | |||||
# To simplify access to the extracted file, | |||||
# prepend a PPM header | |||||
o.write("P5\n%d %d\n255\n" % self.size) | |||||
while True: | |||||
type, size = self.field() | |||||
if type != (8, 10): | |||||
break | |||||
while size > 0: | |||||
s = self.fp.read(min(size, 8192)) | |||||
if not s: | |||||
break | |||||
o.write(s) | |||||
size -= len(s) | |||||
o.close() | |||||
try: | |||||
with Image.open(outfile) as _im: | |||||
_im.load() | |||||
self.im = _im.im | |||||
finally: | |||||
try: | |||||
os.unlink(outfile) | |||||
except OSError: | |||||
pass | |||||
Image.register_open(IptcImageFile.format, IptcImageFile) | |||||
Image.register_extension(IptcImageFile.format, ".iim") | |||||
def getiptcinfo(im): | |||||
""" | |||||
Get IPTC information from TIFF, JPEG, or IPTC file. | |||||
:param im: An image containing IPTC data. | |||||
:returns: A dictionary containing IPTC information, or None if | |||||
no IPTC information block was found. | |||||
""" | |||||
import io | |||||
from . import JpegImagePlugin, TiffImagePlugin | |||||
data = None | |||||
if isinstance(im, IptcImageFile): | |||||
# return info dictionary right away | |||||
return im.info | |||||
elif isinstance(im, JpegImagePlugin.JpegImageFile): | |||||
# extract the IPTC/NAA resource | |||||
photoshop = im.info.get("photoshop") | |||||
if photoshop: | |||||
data = photoshop.get(0x0404) | |||||
elif isinstance(im, TiffImagePlugin.TiffImageFile): | |||||
# get raw data from the IPTC/NAA tag (PhotoShop tags the data | |||||
# as 4-byte integers, so we cannot use the get method...) | |||||
try: | |||||
data = im.tag.tagdata[TiffImagePlugin.IPTC_NAA_CHUNK] | |||||
except (AttributeError, KeyError): | |||||
pass | |||||
if data is None: | |||||
return None # no properties | |||||
# create an IptcImagePlugin object without initializing it | |||||
class FakeImage: | |||||
pass | |||||
im = FakeImage() | |||||
im.__class__ = IptcImageFile | |||||
# parse the IPTC information chunk | |||||
im.info = {} | |||||
im.fp = io.BytesIO(data) | |||||
try: | |||||
im._open() | |||||
except (IndexError, KeyError): | |||||
pass # expected failure | |||||
return im.info |
# | |||||
# The Python Imaging Library | |||||
# $Id$ | |||||
# | |||||
# JPEG2000 file handling | |||||
# | |||||
# History: | |||||
# 2014-03-12 ajh Created | |||||
# 2021-06-30 rogermb Extract dpi information from the 'resc' header box | |||||
# | |||||
# Copyright (c) 2014 Coriolis Systems Limited | |||||
# Copyright (c) 2014 Alastair Houghton | |||||
# | |||||
# See the README file for information on usage and redistribution. | |||||
# | |||||
import io | |||||
import os | |||||
import struct | |||||
from . import Image, ImageFile, _binary | |||||
class BoxReader: | |||||
""" | |||||
A small helper class to read fields stored in JPEG2000 header boxes | |||||
and to easily step into and read sub-boxes. | |||||
""" | |||||
def __init__(self, fp, length=-1): | |||||
self.fp = fp | |||||
self.has_length = length >= 0 | |||||
self.length = length | |||||
self.remaining_in_box = -1 | |||||
def _can_read(self, num_bytes): | |||||
if self.has_length and self.fp.tell() + num_bytes > self.length: | |||||
# Outside box: ensure we don't read past the known file length | |||||
return False | |||||
if self.remaining_in_box >= 0: | |||||
# Inside box contents: ensure read does not go past box boundaries | |||||
return num_bytes <= self.remaining_in_box | |||||
else: | |||||
return True # No length known, just read | |||||
def _read_bytes(self, num_bytes): | |||||
if not self._can_read(num_bytes): | |||||
msg = "Not enough data in header" | |||||
raise SyntaxError(msg) | |||||
data = self.fp.read(num_bytes) | |||||
if len(data) < num_bytes: | |||||
msg = f"Expected to read {num_bytes} bytes but only got {len(data)}." | |||||
raise OSError(msg) | |||||
if self.remaining_in_box > 0: | |||||
self.remaining_in_box -= num_bytes | |||||
return data | |||||
def read_fields(self, field_format): | |||||
size = struct.calcsize(field_format) | |||||
data = self._read_bytes(size) | |||||
return struct.unpack(field_format, data) | |||||
def read_boxes(self): | |||||
size = self.remaining_in_box | |||||
data = self._read_bytes(size) | |||||
return BoxReader(io.BytesIO(data), size) | |||||
def has_next_box(self): | |||||
if self.has_length: | |||||
return self.fp.tell() + self.remaining_in_box < self.length | |||||
else: | |||||
return True | |||||
def next_box_type(self): | |||||
# Skip the rest of the box if it has not been read | |||||
if self.remaining_in_box > 0: | |||||
self.fp.seek(self.remaining_in_box, os.SEEK_CUR) | |||||
self.remaining_in_box = -1 | |||||
# Read the length and type of the next box | |||||
lbox, tbox = self.read_fields(">I4s") | |||||
if lbox == 1: | |||||
lbox = self.read_fields(">Q")[0] | |||||
hlen = 16 | |||||
else: | |||||
hlen = 8 | |||||
if lbox < hlen or not self._can_read(lbox - hlen): | |||||
msg = "Invalid header length" | |||||
raise SyntaxError(msg) | |||||
self.remaining_in_box = lbox - hlen | |||||
return tbox | |||||
def _parse_codestream(fp): | |||||
"""Parse the JPEG 2000 codestream to extract the size and component | |||||
count from the SIZ marker segment, returning a PIL (size, mode) tuple.""" | |||||
hdr = fp.read(2) | |||||
lsiz = _binary.i16be(hdr) | |||||
siz = hdr + fp.read(lsiz - 2) | |||||
lsiz, rsiz, xsiz, ysiz, xosiz, yosiz, _, _, _, _, csiz = struct.unpack_from( | |||||
">HHIIIIIIIIH", siz | |||||
) | |||||
ssiz = [None] * csiz | |||||
xrsiz = [None] * csiz | |||||
yrsiz = [None] * csiz | |||||
for i in range(csiz): | |||||
ssiz[i], xrsiz[i], yrsiz[i] = struct.unpack_from(">BBB", siz, 36 + 3 * i) | |||||
size = (xsiz - xosiz, ysiz - yosiz) | |||||
if csiz == 1: | |||||
if (yrsiz[0] & 0x7F) > 8: | |||||
mode = "I;16" | |||||
else: | |||||
mode = "L" | |||||
elif csiz == 2: | |||||
mode = "LA" | |||||
elif csiz == 3: | |||||
mode = "RGB" | |||||
elif csiz == 4: | |||||
mode = "RGBA" | |||||
else: | |||||
mode = None | |||||
return size, mode | |||||
def _res_to_dpi(num, denom, exp): | |||||
"""Convert JPEG2000's (numerator, denominator, exponent-base-10) resolution, | |||||
calculated as (num / denom) * 10^exp and stored in dots per meter, | |||||
to floating-point dots per inch.""" | |||||
if denom != 0: | |||||
return (254 * num * (10**exp)) / (10000 * denom) | |||||
def _parse_jp2_header(fp): | |||||
"""Parse the JP2 header box to extract size, component count, | |||||
color space information, and optionally DPI information, | |||||
returning a (size, mode, mimetype, dpi) tuple.""" | |||||
# Find the JP2 header box | |||||
reader = BoxReader(fp) | |||||
header = None | |||||
mimetype = None | |||||
while reader.has_next_box(): | |||||
tbox = reader.next_box_type() | |||||
if tbox == b"jp2h": | |||||
header = reader.read_boxes() | |||||
break | |||||
elif tbox == b"ftyp": | |||||
if reader.read_fields(">4s")[0] == b"jpx ": | |||||
mimetype = "image/jpx" | |||||
size = None | |||||
mode = None | |||||
bpc = None | |||||
nc = None | |||||
dpi = None # 2-tuple of DPI info, or None | |||||
while header.has_next_box(): | |||||
tbox = header.next_box_type() | |||||
if tbox == b"ihdr": | |||||
height, width, nc, bpc = header.read_fields(">IIHB") | |||||
size = (width, height) | |||||
if nc == 1 and (bpc & 0x7F) > 8: | |||||
mode = "I;16" | |||||
elif nc == 1: | |||||
mode = "L" | |||||
elif nc == 2: | |||||
mode = "LA" | |||||
elif nc == 3: | |||||
mode = "RGB" | |||||
elif nc == 4: | |||||
mode = "RGBA" | |||||
elif tbox == b"res ": | |||||
res = header.read_boxes() | |||||
while res.has_next_box(): | |||||
tres = res.next_box_type() | |||||
if tres == b"resc": | |||||
vrcn, vrcd, hrcn, hrcd, vrce, hrce = res.read_fields(">HHHHBB") | |||||
hres = _res_to_dpi(hrcn, hrcd, hrce) | |||||
vres = _res_to_dpi(vrcn, vrcd, vrce) | |||||
if hres is not None and vres is not None: | |||||
dpi = (hres, vres) | |||||
break | |||||
if size is None or mode is None: | |||||
msg = "Malformed JP2 header" | |||||
raise SyntaxError(msg) | |||||
return size, mode, mimetype, dpi | |||||
## | |||||
# Image plugin for JPEG2000 images. | |||||
class Jpeg2KImageFile(ImageFile.ImageFile): | |||||
format = "JPEG2000" | |||||
format_description = "JPEG 2000 (ISO 15444)" | |||||
def _open(self): | |||||
sig = self.fp.read(4) | |||||
if sig == b"\xff\x4f\xff\x51": | |||||
self.codec = "j2k" | |||||
self._size, self.mode = _parse_codestream(self.fp) | |||||
else: | |||||
sig = sig + self.fp.read(8) | |||||
if sig == b"\x00\x00\x00\x0cjP \x0d\x0a\x87\x0a": | |||||
self.codec = "jp2" | |||||
header = _parse_jp2_header(self.fp) | |||||
self._size, self.mode, self.custom_mimetype, dpi = header | |||||
if dpi is not None: | |||||
self.info["dpi"] = dpi | |||||
if self.fp.read(12).endswith(b"jp2c\xff\x4f\xff\x51"): | |||||
self._parse_comment() | |||||
else: | |||||
msg = "not a JPEG 2000 file" | |||||
raise SyntaxError(msg) | |||||
if self.size is None or self.mode is None: | |||||
msg = "unable to determine size/mode" | |||||
raise SyntaxError(msg) | |||||
self._reduce = 0 | |||||
self.layers = 0 | |||||
fd = -1 | |||||
length = -1 | |||||
try: | |||||
fd = self.fp.fileno() | |||||
length = os.fstat(fd).st_size | |||||
except Exception: | |||||
fd = -1 | |||||
try: | |||||
pos = self.fp.tell() | |||||
self.fp.seek(0, io.SEEK_END) | |||||
length = self.fp.tell() | |||||
self.fp.seek(pos) | |||||
except Exception: | |||||
length = -1 | |||||
self.tile = [ | |||||
( | |||||
"jpeg2k", | |||||
(0, 0) + self.size, | |||||
0, | |||||
(self.codec, self._reduce, self.layers, fd, length), | |||||
) | |||||
] | |||||
def _parse_comment(self): | |||||
hdr = self.fp.read(2) | |||||
length = _binary.i16be(hdr) | |||||
self.fp.seek(length - 2, os.SEEK_CUR) | |||||
while True: | |||||
marker = self.fp.read(2) | |||||
if not marker: | |||||
break | |||||
typ = marker[1] | |||||
if typ in (0x90, 0xD9): | |||||
# Start of tile or end of codestream | |||||
break | |||||
hdr = self.fp.read(2) | |||||
length = _binary.i16be(hdr) | |||||
if typ == 0x64: | |||||
# Comment | |||||
self.info["comment"] = self.fp.read(length - 2)[2:] | |||||
break | |||||
else: | |||||
self.fp.seek(length - 2, os.SEEK_CUR) | |||||
@property | |||||
def reduce(self): | |||||
# https://github.com/python-pillow/Pillow/issues/4343 found that the | |||||
# new Image 'reduce' method was shadowed by this plugin's 'reduce' | |||||
# property. This attempts to allow for both scenarios | |||||
return self._reduce or super().reduce | |||||
@reduce.setter | |||||
def reduce(self, value): | |||||
self._reduce = value | |||||
def load(self): | |||||
if self.tile and self._reduce: | |||||
power = 1 << self._reduce | |||||
adjust = power >> 1 | |||||
self._size = ( | |||||
int((self.size[0] + adjust) / power), | |||||
int((self.size[1] + adjust) / power), | |||||
) | |||||
# Update the reduce and layers settings | |||||
t = self.tile[0] | |||||
t3 = (t[3][0], self._reduce, self.layers, t[3][3], t[3][4]) | |||||
self.tile = [(t[0], (0, 0) + self.size, t[2], t3)] | |||||
return ImageFile.ImageFile.load(self) | |||||
def _accept(prefix): | |||||
return ( | |||||
prefix[:4] == b"\xff\x4f\xff\x51" | |||||
or prefix[:12] == b"\x00\x00\x00\x0cjP \x0d\x0a\x87\x0a" | |||||
) | |||||
# ------------------------------------------------------------ | |||||
# Save support | |||||
def _save(im, fp, filename): | |||||
# Get the keyword arguments | |||||
info = im.encoderinfo | |||||
if filename.endswith(".j2k") or info.get("no_jp2", False): | |||||
kind = "j2k" | |||||
else: | |||||
kind = "jp2" | |||||
offset = info.get("offset", None) | |||||
tile_offset = info.get("tile_offset", None) | |||||
tile_size = info.get("tile_size", None) | |||||
quality_mode = info.get("quality_mode", "rates") | |||||
quality_layers = info.get("quality_layers", None) | |||||
if quality_layers is not None and not ( | |||||
isinstance(quality_layers, (list, tuple)) | |||||
and all( | |||||
[ | |||||
isinstance(quality_layer, (int, float)) | |||||
for quality_layer in quality_layers | |||||
] | |||||
) | |||||
): | |||||
msg = "quality_layers must be a sequence of numbers" | |||||
raise ValueError(msg) | |||||
num_resolutions = info.get("num_resolutions", 0) | |||||
cblk_size = info.get("codeblock_size", None) | |||||
precinct_size = info.get("precinct_size", None) | |||||
irreversible = info.get("irreversible", False) | |||||
progression = info.get("progression", "LRCP") | |||||
cinema_mode = info.get("cinema_mode", "no") | |||||
mct = info.get("mct", 0) | |||||
signed = info.get("signed", False) | |||||
comment = info.get("comment") | |||||
if isinstance(comment, str): | |||||
comment = comment.encode() | |||||
plt = info.get("plt", False) | |||||
fd = -1 | |||||
if hasattr(fp, "fileno"): | |||||
try: | |||||
fd = fp.fileno() | |||||
except Exception: | |||||
fd = -1 | |||||
im.encoderconfig = ( | |||||
offset, | |||||
tile_offset, | |||||
tile_size, | |||||
quality_mode, | |||||
quality_layers, | |||||
num_resolutions, | |||||
cblk_size, | |||||
precinct_size, | |||||
irreversible, | |||||
progression, | |||||
cinema_mode, | |||||
mct, | |||||
signed, | |||||
fd, | |||||
comment, | |||||
plt, | |||||
) | |||||
ImageFile._save(im, fp, [("jpeg2k", (0, 0) + im.size, 0, kind)]) | |||||
# ------------------------------------------------------------ | |||||
# Registry stuff | |||||
Image.register_open(Jpeg2KImageFile.format, Jpeg2KImageFile, _accept) | |||||
Image.register_save(Jpeg2KImageFile.format, _save) | |||||
Image.register_extensions( | |||||
Jpeg2KImageFile.format, [".jp2", ".j2k", ".jpc", ".jpf", ".jpx", ".j2c"] | |||||
) | |||||
Image.register_mime(Jpeg2KImageFile.format, "image/jp2") |
# | |||||
# The Python Imaging Library. | |||||
# $Id$ | |||||
# | |||||
# JPEG (JFIF) file handling | |||||
# | |||||
# See "Digital Compression and Coding of Continuous-Tone Still Images, | |||||
# Part 1, Requirements and Guidelines" (CCITT T.81 / ISO 10918-1) | |||||
# | |||||
# History: | |||||
# 1995-09-09 fl Created | |||||
# 1995-09-13 fl Added full parser | |||||
# 1996-03-25 fl Added hack to use the IJG command line utilities | |||||
# 1996-05-05 fl Workaround Photoshop 2.5 CMYK polarity bug | |||||
# 1996-05-28 fl Added draft support, JFIF version (0.1) | |||||
# 1996-12-30 fl Added encoder options, added progression property (0.2) | |||||
# 1997-08-27 fl Save mode 1 images as BW (0.3) | |||||
# 1998-07-12 fl Added YCbCr to draft and save methods (0.4) | |||||
# 1998-10-19 fl Don't hang on files using 16-bit DQT's (0.4.1) | |||||
# 2001-04-16 fl Extract DPI settings from JFIF files (0.4.2) | |||||
# 2002-07-01 fl Skip pad bytes before markers; identify Exif files (0.4.3) | |||||
# 2003-04-25 fl Added experimental EXIF decoder (0.5) | |||||
# 2003-06-06 fl Added experimental EXIF GPSinfo decoder | |||||
# 2003-09-13 fl Extract COM markers | |||||
# 2009-09-06 fl Added icc_profile support (from Florian Hoech) | |||||
# 2009-03-06 fl Changed CMYK handling; always use Adobe polarity (0.6) | |||||
# 2009-03-08 fl Added subsampling support (from Justin Huff). | |||||
# | |||||
# Copyright (c) 1997-2003 by Secret Labs AB. | |||||
# Copyright (c) 1995-1996 by Fredrik Lundh. | |||||
# | |||||
# See the README file for information on usage and redistribution. | |||||
# | |||||
import array | |||||
import io | |||||
import math | |||||
import os | |||||
import struct | |||||
import subprocess | |||||
import sys | |||||
import tempfile | |||||
import warnings | |||||
from . import Image, ImageFile | |||||
from ._binary import i16be as i16 | |||||
from ._binary import i32be as i32 | |||||
from ._binary import o8 | |||||
from ._binary import o16be as o16 | |||||
from ._deprecate import deprecate | |||||
from .JpegPresets import presets | |||||
# | |||||
# Parser | |||||
def Skip(self, marker): | |||||
n = i16(self.fp.read(2)) - 2 | |||||
ImageFile._safe_read(self.fp, n) | |||||
def APP(self, marker): | |||||
# | |||||
# Application marker. Store these in the APP dictionary. | |||||
# Also look for well-known application markers. | |||||
n = i16(self.fp.read(2)) - 2 | |||||
s = ImageFile._safe_read(self.fp, n) | |||||
app = "APP%d" % (marker & 15) | |||||
self.app[app] = s # compatibility | |||||
self.applist.append((app, s)) | |||||
if marker == 0xFFE0 and s[:4] == b"JFIF": | |||||
# extract JFIF information | |||||
self.info["jfif"] = version = i16(s, 5) # version | |||||
self.info["jfif_version"] = divmod(version, 256) | |||||
# extract JFIF properties | |||||
try: | |||||
jfif_unit = s[7] | |||||
jfif_density = i16(s, 8), i16(s, 10) | |||||
except Exception: | |||||
pass | |||||
else: | |||||
if jfif_unit == 1: | |||||
self.info["dpi"] = jfif_density | |||||
self.info["jfif_unit"] = jfif_unit | |||||
self.info["jfif_density"] = jfif_density | |||||
elif marker == 0xFFE1 and s[:5] == b"Exif\0": | |||||
if "exif" not in self.info: | |||||
# extract EXIF information (incomplete) | |||||
self.info["exif"] = s # FIXME: value will change | |||||
self._exif_offset = self.fp.tell() - n + 6 | |||||
elif marker == 0xFFE2 and s[:5] == b"FPXR\0": | |||||
# extract FlashPix information (incomplete) | |||||
self.info["flashpix"] = s # FIXME: value will change | |||||
elif marker == 0xFFE2 and s[:12] == b"ICC_PROFILE\0": | |||||
# Since an ICC profile can be larger than the maximum size of | |||||
# a JPEG marker (64K), we need provisions to split it into | |||||
# multiple markers. The format defined by the ICC specifies | |||||
# one or more APP2 markers containing the following data: | |||||
# Identifying string ASCII "ICC_PROFILE\0" (12 bytes) | |||||
# Marker sequence number 1, 2, etc (1 byte) | |||||
# Number of markers Total of APP2's used (1 byte) | |||||
# Profile data (remainder of APP2 data) | |||||
# Decoders should use the marker sequence numbers to | |||||
# reassemble the profile, rather than assuming that the APP2 | |||||
# markers appear in the correct sequence. | |||||
self.icclist.append(s) | |||||
elif marker == 0xFFED and s[:14] == b"Photoshop 3.0\x00": | |||||
# parse the image resource block | |||||
offset = 14 | |||||
photoshop = self.info.setdefault("photoshop", {}) | |||||
while s[offset : offset + 4] == b"8BIM": | |||||
try: | |||||
offset += 4 | |||||
# resource code | |||||
code = i16(s, offset) | |||||
offset += 2 | |||||
# resource name (usually empty) | |||||
name_len = s[offset] | |||||
# name = s[offset+1:offset+1+name_len] | |||||
offset += 1 + name_len | |||||
offset += offset & 1 # align | |||||
# resource data block | |||||
size = i32(s, offset) | |||||
offset += 4 | |||||
data = s[offset : offset + size] | |||||
if code == 0x03ED: # ResolutionInfo | |||||
data = { | |||||
"XResolution": i32(data, 0) / 65536, | |||||
"DisplayedUnitsX": i16(data, 4), | |||||
"YResolution": i32(data, 8) / 65536, | |||||
"DisplayedUnitsY": i16(data, 12), | |||||
} | |||||
photoshop[code] = data | |||||
offset += size | |||||
offset += offset & 1 # align | |||||
except struct.error: | |||||
break # insufficient data | |||||
elif marker == 0xFFEE and s[:5] == b"Adobe": | |||||
self.info["adobe"] = i16(s, 5) | |||||
# extract Adobe custom properties | |||||
try: | |||||
adobe_transform = s[11] | |||||
except IndexError: | |||||
pass | |||||
else: | |||||
self.info["adobe_transform"] = adobe_transform | |||||
elif marker == 0xFFE2 and s[:4] == b"MPF\0": | |||||
# extract MPO information | |||||
self.info["mp"] = s[4:] | |||||
# offset is current location minus buffer size | |||||
# plus constant header size | |||||
self.info["mpoffset"] = self.fp.tell() - n + 4 | |||||
# If DPI isn't in JPEG header, fetch from EXIF | |||||
if "dpi" not in self.info and "exif" in self.info: | |||||
try: | |||||
exif = self.getexif() | |||||
resolution_unit = exif[0x0128] | |||||
x_resolution = exif[0x011A] | |||||
try: | |||||
dpi = float(x_resolution[0]) / x_resolution[1] | |||||
except TypeError: | |||||
dpi = x_resolution | |||||
if math.isnan(dpi): | |||||
raise ValueError | |||||
if resolution_unit == 3: # cm | |||||
# 1 dpcm = 2.54 dpi | |||||
dpi *= 2.54 | |||||
self.info["dpi"] = dpi, dpi | |||||
except (TypeError, KeyError, SyntaxError, ValueError, ZeroDivisionError): | |||||
# SyntaxError for invalid/unreadable EXIF | |||||
# KeyError for dpi not included | |||||
# ZeroDivisionError for invalid dpi rational value | |||||
# ValueError or TypeError for dpi being an invalid float | |||||
self.info["dpi"] = 72, 72 | |||||
def COM(self, marker): | |||||
# | |||||
# Comment marker. Store these in the APP dictionary. | |||||
n = i16(self.fp.read(2)) - 2 | |||||
s = ImageFile._safe_read(self.fp, n) | |||||
self.info["comment"] = s | |||||
self.app["COM"] = s # compatibility | |||||
self.applist.append(("COM", s)) | |||||
def SOF(self, marker): | |||||
# | |||||
# Start of frame marker. Defines the size and mode of the | |||||
# image. JPEG is colour blind, so we use some simple | |||||
# heuristics to map the number of layers to an appropriate | |||||
# mode. Note that this could be made a bit brighter, by | |||||
# looking for JFIF and Adobe APP markers. | |||||
n = i16(self.fp.read(2)) - 2 | |||||
s = ImageFile._safe_read(self.fp, n) | |||||
self._size = i16(s, 3), i16(s, 1) | |||||
self.bits = s[0] | |||||
if self.bits != 8: | |||||
msg = f"cannot handle {self.bits}-bit layers" | |||||
raise SyntaxError(msg) | |||||
self.layers = s[5] | |||||
if self.layers == 1: | |||||
self.mode = "L" | |||||
elif self.layers == 3: | |||||
self.mode = "RGB" | |||||
elif self.layers == 4: | |||||
self.mode = "CMYK" | |||||
else: | |||||
msg = f"cannot handle {self.layers}-layer images" | |||||
raise SyntaxError(msg) | |||||
if marker in [0xFFC2, 0xFFC6, 0xFFCA, 0xFFCE]: | |||||
self.info["progressive"] = self.info["progression"] = 1 | |||||
if self.icclist: | |||||
# fixup icc profile | |||||
self.icclist.sort() # sort by sequence number | |||||
if self.icclist[0][13] == len(self.icclist): | |||||
profile = [] | |||||
for p in self.icclist: | |||||
profile.append(p[14:]) | |||||
icc_profile = b"".join(profile) | |||||
else: | |||||
icc_profile = None # wrong number of fragments | |||||
self.info["icc_profile"] = icc_profile | |||||
self.icclist = [] | |||||
for i in range(6, len(s), 3): | |||||
t = s[i : i + 3] | |||||
# 4-tuples: id, vsamp, hsamp, qtable | |||||
self.layer.append((t[0], t[1] // 16, t[1] & 15, t[2])) | |||||
def DQT(self, marker): | |||||
# | |||||
# Define quantization table. Note that there might be more | |||||
# than one table in each marker. | |||||
# FIXME: The quantization tables can be used to estimate the | |||||
# compression quality. | |||||
n = i16(self.fp.read(2)) - 2 | |||||
s = ImageFile._safe_read(self.fp, n) | |||||
while len(s): | |||||
v = s[0] | |||||
precision = 1 if (v // 16 == 0) else 2 # in bytes | |||||
qt_length = 1 + precision * 64 | |||||
if len(s) < qt_length: | |||||
msg = "bad quantization table marker" | |||||
raise SyntaxError(msg) | |||||
data = array.array("B" if precision == 1 else "H", s[1:qt_length]) | |||||
if sys.byteorder == "little" and precision > 1: | |||||
data.byteswap() # the values are always big-endian | |||||
self.quantization[v & 15] = [data[i] for i in zigzag_index] | |||||
s = s[qt_length:] | |||||
# | |||||
# JPEG marker table | |||||
MARKER = { | |||||
0xFFC0: ("SOF0", "Baseline DCT", SOF), | |||||
0xFFC1: ("SOF1", "Extended Sequential DCT", SOF), | |||||
0xFFC2: ("SOF2", "Progressive DCT", SOF), | |||||
0xFFC3: ("SOF3", "Spatial lossless", SOF), | |||||
0xFFC4: ("DHT", "Define Huffman table", Skip), | |||||
0xFFC5: ("SOF5", "Differential sequential DCT", SOF), | |||||
0xFFC6: ("SOF6", "Differential progressive DCT", SOF), | |||||
0xFFC7: ("SOF7", "Differential spatial", SOF), | |||||
0xFFC8: ("JPG", "Extension", None), | |||||
0xFFC9: ("SOF9", "Extended sequential DCT (AC)", SOF), | |||||
0xFFCA: ("SOF10", "Progressive DCT (AC)", SOF), | |||||
0xFFCB: ("SOF11", "Spatial lossless DCT (AC)", SOF), | |||||
0xFFCC: ("DAC", "Define arithmetic coding conditioning", Skip), | |||||
0xFFCD: ("SOF13", "Differential sequential DCT (AC)", SOF), | |||||
0xFFCE: ("SOF14", "Differential progressive DCT (AC)", SOF), | |||||
0xFFCF: ("SOF15", "Differential spatial (AC)", SOF), | |||||
0xFFD0: ("RST0", "Restart 0", None), | |||||
0xFFD1: ("RST1", "Restart 1", None), | |||||
0xFFD2: ("RST2", "Restart 2", None), | |||||
0xFFD3: ("RST3", "Restart 3", None), | |||||
0xFFD4: ("RST4", "Restart 4", None), | |||||
0xFFD5: ("RST5", "Restart 5", None), | |||||
0xFFD6: ("RST6", "Restart 6", None), | |||||
0xFFD7: ("RST7", "Restart 7", None), | |||||
0xFFD8: ("SOI", "Start of image", None), | |||||
0xFFD9: ("EOI", "End of image", None), | |||||
0xFFDA: ("SOS", "Start of scan", Skip), | |||||
0xFFDB: ("DQT", "Define quantization table", DQT), | |||||
0xFFDC: ("DNL", "Define number of lines", Skip), | |||||
0xFFDD: ("DRI", "Define restart interval", Skip), | |||||
0xFFDE: ("DHP", "Define hierarchical progression", SOF), | |||||
0xFFDF: ("EXP", "Expand reference component", Skip), | |||||
0xFFE0: ("APP0", "Application segment 0", APP), | |||||
0xFFE1: ("APP1", "Application segment 1", APP), | |||||
0xFFE2: ("APP2", "Application segment 2", APP), | |||||
0xFFE3: ("APP3", "Application segment 3", APP), | |||||
0xFFE4: ("APP4", "Application segment 4", APP), | |||||
0xFFE5: ("APP5", "Application segment 5", APP), | |||||
0xFFE6: ("APP6", "Application segment 6", APP), | |||||
0xFFE7: ("APP7", "Application segment 7", APP), | |||||
0xFFE8: ("APP8", "Application segment 8", APP), | |||||
0xFFE9: ("APP9", "Application segment 9", APP), | |||||
0xFFEA: ("APP10", "Application segment 10", APP), | |||||
0xFFEB: ("APP11", "Application segment 11", APP), | |||||
0xFFEC: ("APP12", "Application segment 12", APP), | |||||
0xFFED: ("APP13", "Application segment 13", APP), | |||||
0xFFEE: ("APP14", "Application segment 14", APP), | |||||
0xFFEF: ("APP15", "Application segment 15", APP), | |||||
0xFFF0: ("JPG0", "Extension 0", None), | |||||
0xFFF1: ("JPG1", "Extension 1", None), | |||||
0xFFF2: ("JPG2", "Extension 2", None), | |||||
0xFFF3: ("JPG3", "Extension 3", None), | |||||
0xFFF4: ("JPG4", "Extension 4", None), | |||||
0xFFF5: ("JPG5", "Extension 5", None), | |||||
0xFFF6: ("JPG6", "Extension 6", None), | |||||
0xFFF7: ("JPG7", "Extension 7", None), | |||||
0xFFF8: ("JPG8", "Extension 8", None), | |||||
0xFFF9: ("JPG9", "Extension 9", None), | |||||
0xFFFA: ("JPG10", "Extension 10", None), | |||||
0xFFFB: ("JPG11", "Extension 11", None), | |||||
0xFFFC: ("JPG12", "Extension 12", None), | |||||
0xFFFD: ("JPG13", "Extension 13", None), | |||||
0xFFFE: ("COM", "Comment", COM), | |||||
} | |||||
def _accept(prefix): | |||||
# Magic number was taken from https://en.wikipedia.org/wiki/JPEG | |||||
return prefix[:3] == b"\xFF\xD8\xFF" | |||||
## | |||||
# Image plugin for JPEG and JFIF images. | |||||
class JpegImageFile(ImageFile.ImageFile): | |||||
format = "JPEG" | |||||
format_description = "JPEG (ISO 10918)" | |||||
def _open(self): | |||||
s = self.fp.read(3) | |||||
if not _accept(s): | |||||
msg = "not a JPEG file" | |||||
raise SyntaxError(msg) | |||||
s = b"\xFF" | |||||
# Create attributes | |||||
self.bits = self.layers = 0 | |||||
# JPEG specifics (internal) | |||||
self.layer = [] | |||||
self.huffman_dc = {} | |||||
self.huffman_ac = {} | |||||
self.quantization = {} | |||||
self.app = {} # compatibility | |||||
self.applist = [] | |||||
self.icclist = [] | |||||
while True: | |||||
i = s[0] | |||||
if i == 0xFF: | |||||
s = s + self.fp.read(1) | |||||
i = i16(s) | |||||
else: | |||||
# Skip non-0xFF junk | |||||
s = self.fp.read(1) | |||||
continue | |||||
if i in MARKER: | |||||
name, description, handler = MARKER[i] | |||||
if handler is not None: | |||||
handler(self, i) | |||||
if i == 0xFFDA: # start of scan | |||||
rawmode = self.mode | |||||
if self.mode == "CMYK": | |||||
rawmode = "CMYK;I" # assume adobe conventions | |||||
self.tile = [("jpeg", (0, 0) + self.size, 0, (rawmode, ""))] | |||||
# self.__offset = self.fp.tell() | |||||
break | |||||
s = self.fp.read(1) | |||||
elif i == 0 or i == 0xFFFF: | |||||
# padded marker or junk; move on | |||||
s = b"\xff" | |||||
elif i == 0xFF00: # Skip extraneous data (escaped 0xFF) | |||||
s = self.fp.read(1) | |||||
else: | |||||
msg = "no marker found" | |||||
raise SyntaxError(msg) | |||||
def load_read(self, read_bytes): | |||||
""" | |||||
internal: read more image data | |||||
For premature EOF and LOAD_TRUNCATED_IMAGES adds EOI marker | |||||
so libjpeg can finish decoding | |||||
""" | |||||
s = self.fp.read(read_bytes) | |||||
if not s and ImageFile.LOAD_TRUNCATED_IMAGES and not hasattr(self, "_ended"): | |||||
# Premature EOF. | |||||
# Pretend file is finished adding EOI marker | |||||
self._ended = True | |||||
return b"\xFF\xD9" | |||||
return s | |||||
def draft(self, mode, size): | |||||
if len(self.tile) != 1: | |||||
return | |||||
# Protect from second call | |||||
if self.decoderconfig: | |||||
return | |||||
d, e, o, a = self.tile[0] | |||||
scale = 1 | |||||
original_size = self.size | |||||
if a[0] == "RGB" and mode in ["L", "YCbCr"]: | |||||
self.mode = mode | |||||
a = mode, "" | |||||
if size: | |||||
scale = min(self.size[0] // size[0], self.size[1] // size[1]) | |||||
for s in [8, 4, 2, 1]: | |||||
if scale >= s: | |||||
break | |||||
e = ( | |||||
e[0], | |||||
e[1], | |||||
(e[2] - e[0] + s - 1) // s + e[0], | |||||
(e[3] - e[1] + s - 1) // s + e[1], | |||||
) | |||||
self._size = ((self.size[0] + s - 1) // s, (self.size[1] + s - 1) // s) | |||||
scale = s | |||||
self.tile = [(d, e, o, a)] | |||||
self.decoderconfig = (scale, 0) | |||||
box = (0, 0, original_size[0] / scale, original_size[1] / scale) | |||||
return self.mode, box | |||||
def load_djpeg(self): | |||||
# ALTERNATIVE: handle JPEGs via the IJG command line utilities | |||||
f, path = tempfile.mkstemp() | |||||
os.close(f) | |||||
if os.path.exists(self.filename): | |||||
subprocess.check_call(["djpeg", "-outfile", path, self.filename]) | |||||
else: | |||||
msg = "Invalid Filename" | |||||
raise ValueError(msg) | |||||
try: | |||||
with Image.open(path) as _im: | |||||
_im.load() | |||||
self.im = _im.im | |||||
finally: | |||||
try: | |||||
os.unlink(path) | |||||
except OSError: | |||||
pass | |||||
self.mode = self.im.mode | |||||
self._size = self.im.size | |||||
self.tile = [] | |||||
def _getexif(self): | |||||
return _getexif(self) | |||||
def _getmp(self): | |||||
return _getmp(self) | |||||
def getxmp(self): | |||||
""" | |||||
Returns a dictionary containing the XMP tags. | |||||
Requires defusedxml to be installed. | |||||
:returns: XMP tags in a dictionary. | |||||
""" | |||||
for segment, content in self.applist: | |||||
if segment == "APP1": | |||||
marker, xmp_tags = content.rsplit(b"\x00", 1) | |||||
if marker == b"http://ns.adobe.com/xap/1.0/": | |||||
return self._getxmp(xmp_tags) | |||||
return {} | |||||
def _getexif(self): | |||||
if "exif" not in self.info: | |||||
return None | |||||
return self.getexif()._get_merged_dict() | |||||
def _getmp(self): | |||||
# Extract MP information. This method was inspired by the "highly | |||||
# experimental" _getexif version that's been in use for years now, | |||||
# itself based on the ImageFileDirectory class in the TIFF plugin. | |||||
# The MP record essentially consists of a TIFF file embedded in a JPEG | |||||
# application marker. | |||||
try: | |||||
data = self.info["mp"] | |||||
except KeyError: | |||||
return None | |||||
file_contents = io.BytesIO(data) | |||||
head = file_contents.read(8) | |||||
endianness = ">" if head[:4] == b"\x4d\x4d\x00\x2a" else "<" | |||||
# process dictionary | |||||
from . import TiffImagePlugin | |||||
try: | |||||
info = TiffImagePlugin.ImageFileDirectory_v2(head) | |||||
file_contents.seek(info.next) | |||||
info.load(file_contents) | |||||
mp = dict(info) | |||||
except Exception as e: | |||||
msg = "malformed MP Index (unreadable directory)" | |||||
raise SyntaxError(msg) from e | |||||
# it's an error not to have a number of images | |||||
try: | |||||
quant = mp[0xB001] | |||||
except KeyError as e: | |||||
msg = "malformed MP Index (no number of images)" | |||||
raise SyntaxError(msg) from e | |||||
# get MP entries | |||||
mpentries = [] | |||||
try: | |||||
rawmpentries = mp[0xB002] | |||||
for entrynum in range(0, quant): | |||||
unpackedentry = struct.unpack_from( | |||||
f"{endianness}LLLHH", rawmpentries, entrynum * 16 | |||||
) | |||||
labels = ("Attribute", "Size", "DataOffset", "EntryNo1", "EntryNo2") | |||||
mpentry = dict(zip(labels, unpackedentry)) | |||||
mpentryattr = { | |||||
"DependentParentImageFlag": bool(mpentry["Attribute"] & (1 << 31)), | |||||
"DependentChildImageFlag": bool(mpentry["Attribute"] & (1 << 30)), | |||||
"RepresentativeImageFlag": bool(mpentry["Attribute"] & (1 << 29)), | |||||
"Reserved": (mpentry["Attribute"] & (3 << 27)) >> 27, | |||||
"ImageDataFormat": (mpentry["Attribute"] & (7 << 24)) >> 24, | |||||
"MPType": mpentry["Attribute"] & 0x00FFFFFF, | |||||
} | |||||
if mpentryattr["ImageDataFormat"] == 0: | |||||
mpentryattr["ImageDataFormat"] = "JPEG" | |||||
else: | |||||
msg = "unsupported picture format in MPO" | |||||
raise SyntaxError(msg) | |||||
mptypemap = { | |||||
0x000000: "Undefined", | |||||
0x010001: "Large Thumbnail (VGA Equivalent)", | |||||
0x010002: "Large Thumbnail (Full HD Equivalent)", | |||||
0x020001: "Multi-Frame Image (Panorama)", | |||||
0x020002: "Multi-Frame Image: (Disparity)", | |||||
0x020003: "Multi-Frame Image: (Multi-Angle)", | |||||
0x030000: "Baseline MP Primary Image", | |||||
} | |||||
mpentryattr["MPType"] = mptypemap.get(mpentryattr["MPType"], "Unknown") | |||||
mpentry["Attribute"] = mpentryattr | |||||
mpentries.append(mpentry) | |||||
mp[0xB002] = mpentries | |||||
except KeyError as e: | |||||
msg = "malformed MP Index (bad MP Entry)" | |||||
raise SyntaxError(msg) from e | |||||
# Next we should try and parse the individual image unique ID list; | |||||
# we don't because I've never seen this actually used in a real MPO | |||||
# file and so can't test it. | |||||
return mp | |||||
# -------------------------------------------------------------------- | |||||
# stuff to save JPEG files | |||||
RAWMODE = { | |||||
"1": "L", | |||||
"L": "L", | |||||
"RGB": "RGB", | |||||
"RGBX": "RGB", | |||||
"CMYK": "CMYK;I", # assume adobe conventions | |||||
"YCbCr": "YCbCr", | |||||
} | |||||
# fmt: off | |||||
zigzag_index = ( | |||||
0, 1, 5, 6, 14, 15, 27, 28, | |||||
2, 4, 7, 13, 16, 26, 29, 42, | |||||
3, 8, 12, 17, 25, 30, 41, 43, | |||||
9, 11, 18, 24, 31, 40, 44, 53, | |||||
10, 19, 23, 32, 39, 45, 52, 54, | |||||
20, 22, 33, 38, 46, 51, 55, 60, | |||||
21, 34, 37, 47, 50, 56, 59, 61, | |||||
35, 36, 48, 49, 57, 58, 62, 63, | |||||
) | |||||
samplings = { | |||||
(1, 1, 1, 1, 1, 1): 0, | |||||
(2, 1, 1, 1, 1, 1): 1, | |||||
(2, 2, 1, 1, 1, 1): 2, | |||||
} | |||||
# fmt: on | |||||
def convert_dict_qtables(qtables): | |||||
deprecate("convert_dict_qtables", 10, action="Conversion is no longer needed") | |||||
return qtables | |||||
def get_sampling(im): | |||||
# There's no subsampling when images have only 1 layer | |||||
# (grayscale images) or when they are CMYK (4 layers), | |||||
# so set subsampling to the default value. | |||||
# | |||||
# NOTE: currently Pillow can't encode JPEG to YCCK format. | |||||
# If YCCK support is added in the future, subsampling code will have | |||||
# to be updated (here and in JpegEncode.c) to deal with 4 layers. | |||||
if not hasattr(im, "layers") or im.layers in (1, 4): | |||||
return -1 | |||||
sampling = im.layer[0][1:3] + im.layer[1][1:3] + im.layer[2][1:3] | |||||
return samplings.get(sampling, -1) | |||||
def _save(im, fp, filename): | |||||
if im.width == 0 or im.height == 0: | |||||
msg = "cannot write empty image as JPEG" | |||||
raise ValueError(msg) | |||||
try: | |||||
rawmode = RAWMODE[im.mode] | |||||
except KeyError as e: | |||||
msg = f"cannot write mode {im.mode} as JPEG" | |||||
raise OSError(msg) from e | |||||
info = im.encoderinfo | |||||
dpi = [round(x) for x in info.get("dpi", (0, 0))] | |||||
quality = info.get("quality", -1) | |||||
subsampling = info.get("subsampling", -1) | |||||
qtables = info.get("qtables") | |||||
if quality == "keep": | |||||
quality = -1 | |||||
subsampling = "keep" | |||||
qtables = "keep" | |||||
elif quality in presets: | |||||
preset = presets[quality] | |||||
quality = -1 | |||||
subsampling = preset.get("subsampling", -1) | |||||
qtables = preset.get("quantization") | |||||
elif not isinstance(quality, int): | |||||
msg = "Invalid quality setting" | |||||
raise ValueError(msg) | |||||
else: | |||||
if subsampling in presets: | |||||
subsampling = presets[subsampling].get("subsampling", -1) | |||||
if isinstance(qtables, str) and qtables in presets: | |||||
qtables = presets[qtables].get("quantization") | |||||
if subsampling == "4:4:4": | |||||
subsampling = 0 | |||||
elif subsampling == "4:2:2": | |||||
subsampling = 1 | |||||
elif subsampling == "4:2:0": | |||||
subsampling = 2 | |||||
elif subsampling == "4:1:1": | |||||
# For compatibility. Before Pillow 4.3, 4:1:1 actually meant 4:2:0. | |||||
# Set 4:2:0 if someone is still using that value. | |||||
subsampling = 2 | |||||
elif subsampling == "keep": | |||||
if im.format != "JPEG": | |||||
msg = "Cannot use 'keep' when original image is not a JPEG" | |||||
raise ValueError(msg) | |||||
subsampling = get_sampling(im) | |||||
def validate_qtables(qtables): | |||||
if qtables is None: | |||||
return qtables | |||||
if isinstance(qtables, str): | |||||
try: | |||||
lines = [ | |||||
int(num) | |||||
for line in qtables.splitlines() | |||||
for num in line.split("#", 1)[0].split() | |||||
] | |||||
except ValueError as e: | |||||
msg = "Invalid quantization table" | |||||
raise ValueError(msg) from e | |||||
else: | |||||
qtables = [lines[s : s + 64] for s in range(0, len(lines), 64)] | |||||
if isinstance(qtables, (tuple, list, dict)): | |||||
if isinstance(qtables, dict): | |||||
qtables = [ | |||||
qtables[key] for key in range(len(qtables)) if key in qtables | |||||
] | |||||
elif isinstance(qtables, tuple): | |||||
qtables = list(qtables) | |||||
if not (0 < len(qtables) < 5): | |||||
msg = "None or too many quantization tables" | |||||
raise ValueError(msg) | |||||
for idx, table in enumerate(qtables): | |||||
try: | |||||
if len(table) != 64: | |||||
raise TypeError | |||||
table = array.array("H", table) | |||||
except TypeError as e: | |||||
msg = "Invalid quantization table" | |||||
raise ValueError(msg) from e | |||||
else: | |||||
qtables[idx] = list(table) | |||||
return qtables | |||||
if qtables == "keep": | |||||
if im.format != "JPEG": | |||||
msg = "Cannot use 'keep' when original image is not a JPEG" | |||||
raise ValueError(msg) | |||||
qtables = getattr(im, "quantization", None) | |||||
qtables = validate_qtables(qtables) | |||||
extra = info.get("extra", b"") | |||||
MAX_BYTES_IN_MARKER = 65533 | |||||
icc_profile = info.get("icc_profile") | |||||
if icc_profile: | |||||
ICC_OVERHEAD_LEN = 14 | |||||
MAX_DATA_BYTES_IN_MARKER = MAX_BYTES_IN_MARKER - ICC_OVERHEAD_LEN | |||||
markers = [] | |||||
while icc_profile: | |||||
markers.append(icc_profile[:MAX_DATA_BYTES_IN_MARKER]) | |||||
icc_profile = icc_profile[MAX_DATA_BYTES_IN_MARKER:] | |||||
i = 1 | |||||
for marker in markers: | |||||
size = o16(2 + ICC_OVERHEAD_LEN + len(marker)) | |||||
extra += ( | |||||
b"\xFF\xE2" | |||||
+ size | |||||
+ b"ICC_PROFILE\0" | |||||
+ o8(i) | |||||
+ o8(len(markers)) | |||||
+ marker | |||||
) | |||||
i += 1 | |||||
comment = info.get("comment", im.info.get("comment")) | |||||
# "progressive" is the official name, but older documentation | |||||
# says "progression" | |||||
# FIXME: issue a warning if the wrong form is used (post-1.1.7) | |||||
progressive = info.get("progressive", False) or info.get("progression", False) | |||||
optimize = info.get("optimize", False) | |||||
exif = info.get("exif", b"") | |||||
if isinstance(exif, Image.Exif): | |||||
exif = exif.tobytes() | |||||
if len(exif) > MAX_BYTES_IN_MARKER: | |||||
msg = "EXIF data is too long" | |||||
raise ValueError(msg) | |||||
# get keyword arguments | |||||
im.encoderconfig = ( | |||||
quality, | |||||
progressive, | |||||
info.get("smooth", 0), | |||||
optimize, | |||||
info.get("streamtype", 0), | |||||
dpi[0], | |||||
dpi[1], | |||||
subsampling, | |||||
qtables, | |||||
comment, | |||||
extra, | |||||
exif, | |||||
) | |||||
# if we optimize, libjpeg needs a buffer big enough to hold the whole image | |||||
# in a shot. Guessing on the size, at im.size bytes. (raw pixel size is | |||||
# channels*size, this is a value that's been used in a django patch. | |||||
# https://github.com/matthewwithanm/django-imagekit/issues/50 | |||||
bufsize = 0 | |||||
if optimize or progressive: | |||||
# CMYK can be bigger | |||||
if im.mode == "CMYK": | |||||
bufsize = 4 * im.size[0] * im.size[1] | |||||
# keep sets quality to -1, but the actual value may be high. | |||||
elif quality >= 95 or quality == -1: | |||||
bufsize = 2 * im.size[0] * im.size[1] | |||||
else: | |||||
bufsize = im.size[0] * im.size[1] | |||||
# The EXIF info needs to be written as one block, + APP1, + one spare byte. | |||||
# Ensure that our buffer is big enough. Same with the icc_profile block. | |||||
bufsize = max(ImageFile.MAXBLOCK, bufsize, len(exif) + 5, len(extra) + 1) | |||||
ImageFile._save(im, fp, [("jpeg", (0, 0) + im.size, 0, rawmode)], bufsize) | |||||
def _save_cjpeg(im, fp, filename): | |||||
# ALTERNATIVE: handle JPEGs via the IJG command line utilities. | |||||
tempfile = im._dump() | |||||
subprocess.check_call(["cjpeg", "-outfile", filename, tempfile]) | |||||
try: | |||||
os.unlink(tempfile) | |||||
except OSError: | |||||
pass | |||||
## | |||||
# Factory for making JPEG and MPO instances | |||||
def jpeg_factory(fp=None, filename=None): | |||||
im = JpegImageFile(fp, filename) | |||||
try: | |||||
mpheader = im._getmp() | |||||
if mpheader[45057] > 1: | |||||
# It's actually an MPO | |||||
from .MpoImagePlugin import MpoImageFile | |||||
# Don't reload everything, just convert it. | |||||
im = MpoImageFile.adopt(im, mpheader) | |||||
except (TypeError, IndexError): | |||||
# It is really a JPEG | |||||
pass | |||||
except SyntaxError: | |||||
warnings.warn( | |||||
"Image appears to be a malformed MPO file, it will be " | |||||
"interpreted as a base JPEG file" | |||||
) | |||||
return im | |||||
# --------------------------------------------------------------------- | |||||
# Registry stuff | |||||
Image.register_open(JpegImageFile.format, jpeg_factory, _accept) | |||||
Image.register_save(JpegImageFile.format, _save) | |||||
Image.register_extensions(JpegImageFile.format, [".jfif", ".jpe", ".jpg", ".jpeg"]) | |||||
Image.register_mime(JpegImageFile.format, "image/jpeg") |
""" | |||||
JPEG quality settings equivalent to the Photoshop settings. | |||||
Can be used when saving JPEG files. | |||||
The following presets are available by default: | |||||
``web_low``, ``web_medium``, ``web_high``, ``web_very_high``, ``web_maximum``, | |||||
``low``, ``medium``, ``high``, ``maximum``. | |||||
More presets can be added to the :py:data:`presets` dict if needed. | |||||
To apply the preset, specify:: | |||||
quality="preset_name" | |||||
To apply only the quantization table:: | |||||
qtables="preset_name" | |||||
To apply only the subsampling setting:: | |||||
subsampling="preset_name" | |||||
Example:: | |||||
im.save("image_name.jpg", quality="web_high") | |||||
Subsampling | |||||
----------- | |||||
Subsampling is the practice of encoding images by implementing less resolution | |||||
for chroma information than for luma information. | |||||
(ref.: https://en.wikipedia.org/wiki/Chroma_subsampling) | |||||
Possible subsampling values are 0, 1 and 2 that correspond to 4:4:4, 4:2:2 and | |||||
4:2:0. | |||||
You can get the subsampling of a JPEG with the | |||||
:func:`.JpegImagePlugin.get_sampling` function. | |||||
In JPEG compressed data a JPEG marker is used instead of an EXIF tag. | |||||
(ref.: https://exiv2.org/tags.html) | |||||
Quantization tables | |||||
------------------- | |||||
They are values use by the DCT (Discrete cosine transform) to remove | |||||
*unnecessary* information from the image (the lossy part of the compression). | |||||
(ref.: https://en.wikipedia.org/wiki/Quantization_matrix#Quantization_matrices, | |||||
https://en.wikipedia.org/wiki/JPEG#Quantization) | |||||
You can get the quantization tables of a JPEG with:: | |||||
im.quantization | |||||
This will return a dict with a number of lists. You can pass this dict | |||||
directly as the qtables argument when saving a JPEG. | |||||
The quantization table format in presets is a list with sublists. These formats | |||||
are interchangeable. | |||||
Libjpeg ref.: | |||||
https://web.archive.org/web/20120328125543/http://www.jpegcameras.com/libjpeg/libjpeg-3.html | |||||
""" | |||||
# fmt: off | |||||
presets = { | |||||
'web_low': {'subsampling': 2, # "4:2:0" | |||||
'quantization': [ | |||||
[20, 16, 25, 39, 50, 46, 62, 68, | |||||
16, 18, 23, 38, 38, 53, 65, 68, | |||||
25, 23, 31, 38, 53, 65, 68, 68, | |||||
39, 38, 38, 53, 65, 68, 68, 68, | |||||
50, 38, 53, 65, 68, 68, 68, 68, | |||||
46, 53, 65, 68, 68, 68, 68, 68, | |||||
62, 65, 68, 68, 68, 68, 68, 68, | |||||
68, 68, 68, 68, 68, 68, 68, 68], | |||||
[21, 25, 32, 38, 54, 68, 68, 68, | |||||
25, 28, 24, 38, 54, 68, 68, 68, | |||||
32, 24, 32, 43, 66, 68, 68, 68, | |||||
38, 38, 43, 53, 68, 68, 68, 68, | |||||
54, 54, 66, 68, 68, 68, 68, 68, | |||||
68, 68, 68, 68, 68, 68, 68, 68, | |||||
68, 68, 68, 68, 68, 68, 68, 68, | |||||
68, 68, 68, 68, 68, 68, 68, 68] | |||||
]}, | |||||
'web_medium': {'subsampling': 2, # "4:2:0" | |||||
'quantization': [ | |||||
[16, 11, 11, 16, 23, 27, 31, 30, | |||||
11, 12, 12, 15, 20, 23, 23, 30, | |||||
11, 12, 13, 16, 23, 26, 35, 47, | |||||
16, 15, 16, 23, 26, 37, 47, 64, | |||||
23, 20, 23, 26, 39, 51, 64, 64, | |||||
27, 23, 26, 37, 51, 64, 64, 64, | |||||
31, 23, 35, 47, 64, 64, 64, 64, | |||||
30, 30, 47, 64, 64, 64, 64, 64], | |||||
[17, 15, 17, 21, 20, 26, 38, 48, | |||||
15, 19, 18, 17, 20, 26, 35, 43, | |||||
17, 18, 20, 22, 26, 30, 46, 53, | |||||
21, 17, 22, 28, 30, 39, 53, 64, | |||||
20, 20, 26, 30, 39, 48, 64, 64, | |||||
26, 26, 30, 39, 48, 63, 64, 64, | |||||
38, 35, 46, 53, 64, 64, 64, 64, | |||||
48, 43, 53, 64, 64, 64, 64, 64] | |||||
]}, | |||||
'web_high': {'subsampling': 0, # "4:4:4" | |||||
'quantization': [ | |||||
[6, 4, 4, 6, 9, 11, 12, 16, | |||||
4, 5, 5, 6, 8, 10, 12, 12, | |||||
4, 5, 5, 6, 10, 12, 14, 19, | |||||
6, 6, 6, 11, 12, 15, 19, 28, | |||||
9, 8, 10, 12, 16, 20, 27, 31, | |||||
11, 10, 12, 15, 20, 27, 31, 31, | |||||
12, 12, 14, 19, 27, 31, 31, 31, | |||||
16, 12, 19, 28, 31, 31, 31, 31], | |||||
[7, 7, 13, 24, 26, 31, 31, 31, | |||||
7, 12, 16, 21, 31, 31, 31, 31, | |||||
13, 16, 17, 31, 31, 31, 31, 31, | |||||
24, 21, 31, 31, 31, 31, 31, 31, | |||||
26, 31, 31, 31, 31, 31, 31, 31, | |||||
31, 31, 31, 31, 31, 31, 31, 31, | |||||
31, 31, 31, 31, 31, 31, 31, 31, | |||||
31, 31, 31, 31, 31, 31, 31, 31] | |||||
]}, | |||||
'web_very_high': {'subsampling': 0, # "4:4:4" | |||||
'quantization': [ | |||||
[2, 2, 2, 2, 3, 4, 5, 6, | |||||
2, 2, 2, 2, 3, 4, 5, 6, | |||||
2, 2, 2, 2, 4, 5, 7, 9, | |||||
2, 2, 2, 4, 5, 7, 9, 12, | |||||
3, 3, 4, 5, 8, 10, 12, 12, | |||||
4, 4, 5, 7, 10, 12, 12, 12, | |||||
5, 5, 7, 9, 12, 12, 12, 12, | |||||
6, 6, 9, 12, 12, 12, 12, 12], | |||||
[3, 3, 5, 9, 13, 15, 15, 15, | |||||
3, 4, 6, 11, 14, 12, 12, 12, | |||||
5, 6, 9, 14, 12, 12, 12, 12, | |||||
9, 11, 14, 12, 12, 12, 12, 12, | |||||
13, 14, 12, 12, 12, 12, 12, 12, | |||||
15, 12, 12, 12, 12, 12, 12, 12, | |||||
15, 12, 12, 12, 12, 12, 12, 12, | |||||
15, 12, 12, 12, 12, 12, 12, 12] | |||||
]}, | |||||
'web_maximum': {'subsampling': 0, # "4:4:4" | |||||
'quantization': [ | |||||
[1, 1, 1, 1, 1, 1, 1, 1, | |||||
1, 1, 1, 1, 1, 1, 1, 1, | |||||
1, 1, 1, 1, 1, 1, 1, 2, | |||||
1, 1, 1, 1, 1, 1, 2, 2, | |||||
1, 1, 1, 1, 1, 2, 2, 3, | |||||
1, 1, 1, 1, 2, 2, 3, 3, | |||||
1, 1, 1, 2, 2, 3, 3, 3, | |||||
1, 1, 2, 2, 3, 3, 3, 3], | |||||
[1, 1, 1, 2, 2, 3, 3, 3, | |||||
1, 1, 1, 2, 3, 3, 3, 3, | |||||
1, 1, 1, 3, 3, 3, 3, 3, | |||||
2, 2, 3, 3, 3, 3, 3, 3, | |||||
2, 3, 3, 3, 3, 3, 3, 3, | |||||
3, 3, 3, 3, 3, 3, 3, 3, | |||||
3, 3, 3, 3, 3, 3, 3, 3, | |||||
3, 3, 3, 3, 3, 3, 3, 3] | |||||
]}, | |||||
'low': {'subsampling': 2, # "4:2:0" | |||||
'quantization': [ | |||||
[18, 14, 14, 21, 30, 35, 34, 17, | |||||
14, 16, 16, 19, 26, 23, 12, 12, | |||||
14, 16, 17, 21, 23, 12, 12, 12, | |||||
21, 19, 21, 23, 12, 12, 12, 12, | |||||
30, 26, 23, 12, 12, 12, 12, 12, | |||||
35, 23, 12, 12, 12, 12, 12, 12, | |||||
34, 12, 12, 12, 12, 12, 12, 12, | |||||
17, 12, 12, 12, 12, 12, 12, 12], | |||||
[20, 19, 22, 27, 20, 20, 17, 17, | |||||
19, 25, 23, 14, 14, 12, 12, 12, | |||||
22, 23, 14, 14, 12, 12, 12, 12, | |||||
27, 14, 14, 12, 12, 12, 12, 12, | |||||
20, 14, 12, 12, 12, 12, 12, 12, | |||||
20, 12, 12, 12, 12, 12, 12, 12, | |||||
17, 12, 12, 12, 12, 12, 12, 12, | |||||
17, 12, 12, 12, 12, 12, 12, 12] | |||||
]}, | |||||
'medium': {'subsampling': 2, # "4:2:0" | |||||
'quantization': [ | |||||
[12, 8, 8, 12, 17, 21, 24, 17, | |||||
8, 9, 9, 11, 15, 19, 12, 12, | |||||
8, 9, 10, 12, 19, 12, 12, 12, | |||||
12, 11, 12, 21, 12, 12, 12, 12, | |||||
17, 15, 19, 12, 12, 12, 12, 12, | |||||
21, 19, 12, 12, 12, 12, 12, 12, | |||||
24, 12, 12, 12, 12, 12, 12, 12, | |||||
17, 12, 12, 12, 12, 12, 12, 12], | |||||
[13, 11, 13, 16, 20, 20, 17, 17, | |||||
11, 14, 14, 14, 14, 12, 12, 12, | |||||
13, 14, 14, 14, 12, 12, 12, 12, | |||||
16, 14, 14, 12, 12, 12, 12, 12, | |||||
20, 14, 12, 12, 12, 12, 12, 12, | |||||
20, 12, 12, 12, 12, 12, 12, 12, | |||||
17, 12, 12, 12, 12, 12, 12, 12, | |||||
17, 12, 12, 12, 12, 12, 12, 12] | |||||
]}, | |||||
'high': {'subsampling': 0, # "4:4:4" | |||||
'quantization': [ | |||||
[6, 4, 4, 6, 9, 11, 12, 16, | |||||
4, 5, 5, 6, 8, 10, 12, 12, | |||||
4, 5, 5, 6, 10, 12, 12, 12, | |||||
6, 6, 6, 11, 12, 12, 12, 12, | |||||
9, 8, 10, 12, 12, 12, 12, 12, | |||||
11, 10, 12, 12, 12, 12, 12, 12, | |||||
12, 12, 12, 12, 12, 12, 12, 12, | |||||
16, 12, 12, 12, 12, 12, 12, 12], | |||||
[7, 7, 13, 24, 20, 20, 17, 17, | |||||
7, 12, 16, 14, 14, 12, 12, 12, | |||||
13, 16, 14, 14, 12, 12, 12, 12, | |||||
24, 14, 14, 12, 12, 12, 12, 12, | |||||
20, 14, 12, 12, 12, 12, 12, 12, | |||||
20, 12, 12, 12, 12, 12, 12, 12, | |||||
17, 12, 12, 12, 12, 12, 12, 12, | |||||
17, 12, 12, 12, 12, 12, 12, 12] | |||||
]}, | |||||
'maximum': {'subsampling': 0, # "4:4:4" | |||||
'quantization': [ | |||||
[2, 2, 2, 2, 3, 4, 5, 6, | |||||
2, 2, 2, 2, 3, 4, 5, 6, | |||||
2, 2, 2, 2, 4, 5, 7, 9, | |||||
2, 2, 2, 4, 5, 7, 9, 12, | |||||
3, 3, 4, 5, 8, 10, 12, 12, | |||||
4, 4, 5, 7, 10, 12, 12, 12, | |||||
5, 5, 7, 9, 12, 12, 12, 12, | |||||
6, 6, 9, 12, 12, 12, 12, 12], | |||||
[3, 3, 5, 9, 13, 15, 15, 15, | |||||
3, 4, 6, 10, 14, 12, 12, 12, | |||||
5, 6, 9, 14, 12, 12, 12, 12, | |||||
9, 10, 14, 12, 12, 12, 12, 12, | |||||
13, 14, 12, 12, 12, 12, 12, 12, | |||||
15, 12, 12, 12, 12, 12, 12, 12, | |||||
15, 12, 12, 12, 12, 12, 12, 12, | |||||
15, 12, 12, 12, 12, 12, 12, 12] | |||||
]}, | |||||
} | |||||
# fmt: on |
# | |||||
# The Python Imaging Library. | |||||
# $Id$ | |||||
# | |||||
# Basic McIdas support for PIL | |||||
# | |||||
# History: | |||||
# 1997-05-05 fl Created (8-bit images only) | |||||
# 2009-03-08 fl Added 16/32-bit support. | |||||
# | |||||
# Thanks to Richard Jones and Craig Swank for specs and samples. | |||||
# | |||||
# Copyright (c) Secret Labs AB 1997. | |||||
# Copyright (c) Fredrik Lundh 1997. | |||||
# | |||||
# See the README file for information on usage and redistribution. | |||||
# | |||||
import struct | |||||
from . import Image, ImageFile | |||||
def _accept(s): | |||||
return s[:8] == b"\x00\x00\x00\x00\x00\x00\x00\x04" | |||||
## | |||||
# Image plugin for McIdas area images. | |||||
class McIdasImageFile(ImageFile.ImageFile): | |||||
format = "MCIDAS" | |||||
format_description = "McIdas area file" | |||||
def _open(self): | |||||
# parse area file directory | |||||
s = self.fp.read(256) | |||||
if not _accept(s) or len(s) != 256: | |||||
msg = "not an McIdas area file" | |||||
raise SyntaxError(msg) | |||||
self.area_descriptor_raw = s | |||||
self.area_descriptor = w = [0] + list(struct.unpack("!64i", s)) | |||||
# get mode | |||||
if w[11] == 1: | |||||
mode = rawmode = "L" | |||||
elif w[11] == 2: | |||||
# FIXME: add memory map support | |||||
mode = "I" | |||||
rawmode = "I;16B" | |||||
elif w[11] == 4: | |||||
# FIXME: add memory map support | |||||
mode = "I" | |||||
rawmode = "I;32B" | |||||
else: | |||||
msg = "unsupported McIdas format" | |||||
raise SyntaxError(msg) | |||||
self.mode = mode | |||||
self._size = w[10], w[9] | |||||
offset = w[34] + w[15] | |||||
stride = w[15] + w[10] * w[11] * w[14] | |||||
self.tile = [("raw", (0, 0) + self.size, offset, (rawmode, stride, 1))] | |||||
# -------------------------------------------------------------------- | |||||
# registry | |||||
Image.register_open(McIdasImageFile.format, McIdasImageFile, _accept) | |||||
# no default extension |
# | |||||
# The Python Imaging Library. | |||||
# $Id$ | |||||
# | |||||
# Microsoft Image Composer support for PIL | |||||
# | |||||
# Notes: | |||||
# uses TiffImagePlugin.py to read the actual image streams | |||||
# | |||||
# History: | |||||
# 97-01-20 fl Created | |||||
# | |||||
# Copyright (c) Secret Labs AB 1997. | |||||
# Copyright (c) Fredrik Lundh 1997. | |||||
# | |||||
# See the README file for information on usage and redistribution. | |||||
# | |||||
import olefile | |||||
from . import Image, TiffImagePlugin | |||||
# | |||||
# -------------------------------------------------------------------- | |||||
def _accept(prefix): | |||||
return prefix[:8] == olefile.MAGIC | |||||
## | |||||
# Image plugin for Microsoft's Image Composer file format. | |||||
class MicImageFile(TiffImagePlugin.TiffImageFile): | |||||
format = "MIC" | |||||
format_description = "Microsoft Image Composer" | |||||
_close_exclusive_fp_after_loading = False | |||||
def _open(self): | |||||
# read the OLE directory and see if this is a likely | |||||
# to be a Microsoft Image Composer file | |||||
try: | |||||
self.ole = olefile.OleFileIO(self.fp) | |||||
except OSError as e: | |||||
msg = "not an MIC file; invalid OLE file" | |||||
raise SyntaxError(msg) from e | |||||
# find ACI subfiles with Image members (maybe not the | |||||
# best way to identify MIC files, but what the... ;-) | |||||
self.images = [] | |||||
for path in self.ole.listdir(): | |||||
if path[1:] and path[0][-4:] == ".ACI" and path[1] == "Image": | |||||
self.images.append(path) | |||||
# if we didn't find any images, this is probably not | |||||
# an MIC file. | |||||
if not self.images: | |||||
msg = "not an MIC file; no image entries" | |||||
raise SyntaxError(msg) | |||||
self.frame = None | |||||
self._n_frames = len(self.images) | |||||
self.is_animated = self._n_frames > 1 | |||||
if len(self.images) > 1: | |||||
self._category = Image.CONTAINER | |||||
self.seek(0) | |||||
def seek(self, frame): | |||||
if not self._seek_check(frame): | |||||
return | |||||
try: | |||||
filename = self.images[frame] | |||||
except IndexError as e: | |||||
msg = "no such frame" | |||||
raise EOFError(msg) from e | |||||
self.fp = self.ole.openstream(filename) | |||||
TiffImagePlugin.TiffImageFile._open(self) | |||||
self.frame = frame | |||||
def tell(self): | |||||
return self.frame | |||||
def close(self): | |||||
self.ole.close() | |||||
super().close() | |||||
def __exit__(self, *args): | |||||
self.ole.close() | |||||
super().__exit__() | |||||
# | |||||
# -------------------------------------------------------------------- | |||||
Image.register_open(MicImageFile.format, MicImageFile, _accept) | |||||
Image.register_extension(MicImageFile.format, ".mic") |
# | |||||
# The Python Imaging Library. | |||||
# $Id$ | |||||
# | |||||
# MPEG file handling | |||||
# | |||||
# History: | |||||
# 95-09-09 fl Created | |||||
# | |||||
# Copyright (c) Secret Labs AB 1997. | |||||
# Copyright (c) Fredrik Lundh 1995. | |||||
# | |||||
# See the README file for information on usage and redistribution. | |||||
# | |||||
from . import Image, ImageFile | |||||
from ._binary import i8 | |||||
# | |||||
# Bitstream parser | |||||
class BitStream: | |||||
def __init__(self, fp): | |||||
self.fp = fp | |||||
self.bits = 0 | |||||
self.bitbuffer = 0 | |||||
def next(self): | |||||
return i8(self.fp.read(1)) | |||||
def peek(self, bits): | |||||
while self.bits < bits: | |||||
c = self.next() | |||||
if c < 0: | |||||
self.bits = 0 | |||||
continue | |||||
self.bitbuffer = (self.bitbuffer << 8) + c | |||||
self.bits += 8 | |||||
return self.bitbuffer >> (self.bits - bits) & (1 << bits) - 1 | |||||
def skip(self, bits): | |||||
while self.bits < bits: | |||||
self.bitbuffer = (self.bitbuffer << 8) + i8(self.fp.read(1)) | |||||
self.bits += 8 | |||||
self.bits = self.bits - bits | |||||
def read(self, bits): | |||||
v = self.peek(bits) | |||||
self.bits = self.bits - bits | |||||
return v | |||||
## | |||||
# Image plugin for MPEG streams. This plugin can identify a stream, | |||||
# but it cannot read it. | |||||
class MpegImageFile(ImageFile.ImageFile): | |||||
format = "MPEG" | |||||
format_description = "MPEG" | |||||
def _open(self): | |||||
s = BitStream(self.fp) | |||||
if s.read(32) != 0x1B3: | |||||
msg = "not an MPEG file" | |||||
raise SyntaxError(msg) | |||||
self.mode = "RGB" | |||||
self._size = s.read(12), s.read(12) | |||||
# -------------------------------------------------------------------- | |||||
# Registry stuff | |||||
Image.register_open(MpegImageFile.format, MpegImageFile) | |||||
Image.register_extensions(MpegImageFile.format, [".mpg", ".mpeg"]) | |||||
Image.register_mime(MpegImageFile.format, "video/mpeg") |
# | |||||
# The Python Imaging Library. | |||||
# $Id$ | |||||
# | |||||
# MPO file handling | |||||
# | |||||
# See "Multi-Picture Format" (CIPA DC-007-Translation 2009, Standard of the | |||||
# Camera & Imaging Products Association) | |||||
# | |||||
# The multi-picture object combines multiple JPEG images (with a modified EXIF | |||||
# data format) into a single file. While it can theoretically be used much like | |||||
# a GIF animation, it is commonly used to represent 3D photographs and is (as | |||||
# of this writing) the most commonly used format by 3D cameras. | |||||
# | |||||
# History: | |||||
# 2014-03-13 Feneric Created | |||||
# | |||||
# See the README file for information on usage and redistribution. | |||||
# | |||||
import itertools | |||||
import os | |||||
import struct | |||||
from . import ( | |||||
ExifTags, | |||||
Image, | |||||
ImageFile, | |||||
ImageSequence, | |||||
JpegImagePlugin, | |||||
TiffImagePlugin, | |||||
) | |||||
from ._binary import i16be as i16 | |||||
from ._binary import o32le | |||||
# def _accept(prefix): | |||||
# return JpegImagePlugin._accept(prefix) | |||||
def _save(im, fp, filename): | |||||
JpegImagePlugin._save(im, fp, filename) | |||||
def _save_all(im, fp, filename): | |||||
append_images = im.encoderinfo.get("append_images", []) | |||||
if not append_images: | |||||
try: | |||||
animated = im.is_animated | |||||
except AttributeError: | |||||
animated = False | |||||
if not animated: | |||||
_save(im, fp, filename) | |||||
return | |||||
mpf_offset = 28 | |||||
offsets = [] | |||||
for imSequence in itertools.chain([im], append_images): | |||||
for im_frame in ImageSequence.Iterator(imSequence): | |||||
if not offsets: | |||||
# APP2 marker | |||||
im_frame.encoderinfo["extra"] = ( | |||||
b"\xFF\xE2" + struct.pack(">H", 6 + 82) + b"MPF\0" + b" " * 82 | |||||
) | |||||
exif = im_frame.encoderinfo.get("exif") | |||||
if isinstance(exif, Image.Exif): | |||||
exif = exif.tobytes() | |||||
im_frame.encoderinfo["exif"] = exif | |||||
if exif: | |||||
mpf_offset += 4 + len(exif) | |||||
JpegImagePlugin._save(im_frame, fp, filename) | |||||
offsets.append(fp.tell()) | |||||
else: | |||||
im_frame.save(fp, "JPEG") | |||||
offsets.append(fp.tell() - offsets[-1]) | |||||
ifd = TiffImagePlugin.ImageFileDirectory_v2() | |||||
ifd[0xB000] = b"0100" | |||||
ifd[0xB001] = len(offsets) | |||||
mpentries = b"" | |||||
data_offset = 0 | |||||
for i, size in enumerate(offsets): | |||||
if i == 0: | |||||
mptype = 0x030000 # Baseline MP Primary Image | |||||
else: | |||||
mptype = 0x000000 # Undefined | |||||
mpentries += struct.pack("<LLLHH", mptype, size, data_offset, 0, 0) | |||||
if i == 0: | |||||
data_offset -= mpf_offset | |||||
data_offset += size | |||||
ifd[0xB002] = mpentries | |||||
fp.seek(mpf_offset) | |||||
fp.write(b"II\x2A\x00" + o32le(8) + ifd.tobytes(8)) | |||||
fp.seek(0, os.SEEK_END) | |||||
## | |||||
# Image plugin for MPO images. | |||||
class MpoImageFile(JpegImagePlugin.JpegImageFile): | |||||
format = "MPO" | |||||
format_description = "MPO (CIPA DC-007)" | |||||
_close_exclusive_fp_after_loading = False | |||||
def _open(self): | |||||
self.fp.seek(0) # prep the fp in order to pass the JPEG test | |||||
JpegImagePlugin.JpegImageFile._open(self) | |||||
self._after_jpeg_open() | |||||
def _after_jpeg_open(self, mpheader=None): | |||||
self._initial_size = self.size | |||||
self.mpinfo = mpheader if mpheader is not None else self._getmp() | |||||
self.n_frames = self.mpinfo[0xB001] | |||||
self.__mpoffsets = [ | |||||
mpent["DataOffset"] + self.info["mpoffset"] for mpent in self.mpinfo[0xB002] | |||||
] | |||||
self.__mpoffsets[0] = 0 | |||||
# Note that the following assertion will only be invalid if something | |||||
# gets broken within JpegImagePlugin. | |||||
assert self.n_frames == len(self.__mpoffsets) | |||||
del self.info["mpoffset"] # no longer needed | |||||
self.is_animated = self.n_frames > 1 | |||||
self._fp = self.fp # FIXME: hack | |||||
self._fp.seek(self.__mpoffsets[0]) # get ready to read first frame | |||||
self.__frame = 0 | |||||
self.offset = 0 | |||||
# for now we can only handle reading and individual frame extraction | |||||
self.readonly = 1 | |||||
def load_seek(self, pos): | |||||
self._fp.seek(pos) | |||||
def seek(self, frame): | |||||
if not self._seek_check(frame): | |||||
return | |||||
self.fp = self._fp | |||||
self.offset = self.__mpoffsets[frame] | |||||
self.fp.seek(self.offset + 2) # skip SOI marker | |||||
segment = self.fp.read(2) | |||||
if not segment: | |||||
msg = "No data found for frame" | |||||
raise ValueError(msg) | |||||
self._size = self._initial_size | |||||
if i16(segment) == 0xFFE1: # APP1 | |||||
n = i16(self.fp.read(2)) - 2 | |||||
self.info["exif"] = ImageFile._safe_read(self.fp, n) | |||||
self._reload_exif() | |||||
mptype = self.mpinfo[0xB002][frame]["Attribute"]["MPType"] | |||||
if mptype.startswith("Large Thumbnail"): | |||||
exif = self.getexif().get_ifd(ExifTags.IFD.Exif) | |||||
if 40962 in exif and 40963 in exif: | |||||
self._size = (exif[40962], exif[40963]) | |||||
elif "exif" in self.info: | |||||
del self.info["exif"] | |||||
self._reload_exif() | |||||
self.tile = [("jpeg", (0, 0) + self.size, self.offset, (self.mode, ""))] | |||||
self.__frame = frame | |||||
def tell(self): | |||||
return self.__frame | |||||
@staticmethod | |||||
def adopt(jpeg_instance, mpheader=None): | |||||
""" | |||||
Transform the instance of JpegImageFile into | |||||
an instance of MpoImageFile. | |||||
After the call, the JpegImageFile is extended | |||||
to be an MpoImageFile. | |||||
This is essentially useful when opening a JPEG | |||||
file that reveals itself as an MPO, to avoid | |||||
double call to _open. | |||||
""" | |||||
jpeg_instance.__class__ = MpoImageFile | |||||
jpeg_instance._after_jpeg_open(mpheader) | |||||
return jpeg_instance | |||||
# --------------------------------------------------------------------- | |||||
# Registry stuff | |||||
# Note that since MPO shares a factory with JPEG, we do not need to do a | |||||
# separate registration for it here. | |||||
# Image.register_open(MpoImageFile.format, | |||||
# JpegImagePlugin.jpeg_factory, _accept) | |||||
Image.register_save(MpoImageFile.format, _save) | |||||
Image.register_save_all(MpoImageFile.format, _save_all) | |||||
Image.register_extension(MpoImageFile.format, ".mpo") | |||||
Image.register_mime(MpoImageFile.format, "image/mpo") |
# | |||||
# The Python Imaging Library. | |||||
# | |||||
# MSP file handling | |||||
# | |||||
# This is the format used by the Paint program in Windows 1 and 2. | |||||
# | |||||
# History: | |||||
# 95-09-05 fl Created | |||||
# 97-01-03 fl Read/write MSP images | |||||
# 17-02-21 es Fixed RLE interpretation | |||||
# | |||||
# Copyright (c) Secret Labs AB 1997. | |||||
# Copyright (c) Fredrik Lundh 1995-97. | |||||
# Copyright (c) Eric Soroos 2017. | |||||
# | |||||
# See the README file for information on usage and redistribution. | |||||
# | |||||
# More info on this format: https://archive.org/details/gg243631 | |||||
# Page 313: | |||||
# Figure 205. Windows Paint Version 1: "DanM" Format | |||||
# Figure 206. Windows Paint Version 2: "LinS" Format. Used in Windows V2.03 | |||||
# | |||||
# See also: https://www.fileformat.info/format/mspaint/egff.htm | |||||
import io | |||||
import struct | |||||
from . import Image, ImageFile | |||||
from ._binary import i16le as i16 | |||||
from ._binary import o16le as o16 | |||||
# | |||||
# read MSP files | |||||
def _accept(prefix): | |||||
return prefix[:4] in [b"DanM", b"LinS"] | |||||
## | |||||
# Image plugin for Windows MSP images. This plugin supports both | |||||
# uncompressed (Windows 1.0). | |||||
class MspImageFile(ImageFile.ImageFile): | |||||
format = "MSP" | |||||
format_description = "Windows Paint" | |||||
def _open(self): | |||||
# Header | |||||
s = self.fp.read(32) | |||||
if not _accept(s): | |||||
msg = "not an MSP file" | |||||
raise SyntaxError(msg) | |||||
# Header checksum | |||||
checksum = 0 | |||||
for i in range(0, 32, 2): | |||||
checksum = checksum ^ i16(s, i) | |||||
if checksum != 0: | |||||
msg = "bad MSP checksum" | |||||
raise SyntaxError(msg) | |||||
self.mode = "1" | |||||
self._size = i16(s, 4), i16(s, 6) | |||||
if s[:4] == b"DanM": | |||||
self.tile = [("raw", (0, 0) + self.size, 32, ("1", 0, 1))] | |||||
else: | |||||
self.tile = [("MSP", (0, 0) + self.size, 32, None)] | |||||
class MspDecoder(ImageFile.PyDecoder): | |||||
# The algo for the MSP decoder is from | |||||
# https://www.fileformat.info/format/mspaint/egff.htm | |||||
# cc-by-attribution -- That page references is taken from the | |||||
# Encyclopedia of Graphics File Formats and is licensed by | |||||
# O'Reilly under the Creative Common/Attribution license | |||||
# | |||||
# For RLE encoded files, the 32byte header is followed by a scan | |||||
# line map, encoded as one 16bit word of encoded byte length per | |||||
# line. | |||||
# | |||||
# NOTE: the encoded length of the line can be 0. This was not | |||||
# handled in the previous version of this encoder, and there's no | |||||
# mention of how to handle it in the documentation. From the few | |||||
# examples I've seen, I've assumed that it is a fill of the | |||||
# background color, in this case, white. | |||||
# | |||||
# | |||||
# Pseudocode of the decoder: | |||||
# Read a BYTE value as the RunType | |||||
# If the RunType value is zero | |||||
# Read next byte as the RunCount | |||||
# Read the next byte as the RunValue | |||||
# Write the RunValue byte RunCount times | |||||
# If the RunType value is non-zero | |||||
# Use this value as the RunCount | |||||
# Read and write the next RunCount bytes literally | |||||
# | |||||
# e.g.: | |||||
# 0x00 03 ff 05 00 01 02 03 04 | |||||
# would yield the bytes: | |||||
# 0xff ff ff 00 01 02 03 04 | |||||
# | |||||
# which are then interpreted as a bit packed mode '1' image | |||||
_pulls_fd = True | |||||
def decode(self, buffer): | |||||
img = io.BytesIO() | |||||
blank_line = bytearray((0xFF,) * ((self.state.xsize + 7) // 8)) | |||||
try: | |||||
self.fd.seek(32) | |||||
rowmap = struct.unpack_from( | |||||
f"<{self.state.ysize}H", self.fd.read(self.state.ysize * 2) | |||||
) | |||||
except struct.error as e: | |||||
msg = "Truncated MSP file in row map" | |||||
raise OSError(msg) from e | |||||
for x, rowlen in enumerate(rowmap): | |||||
try: | |||||
if rowlen == 0: | |||||
img.write(blank_line) | |||||
continue | |||||
row = self.fd.read(rowlen) | |||||
if len(row) != rowlen: | |||||
msg = f"Truncated MSP file, expected {rowlen} bytes on row {x}" | |||||
raise OSError(msg) | |||||
idx = 0 | |||||
while idx < rowlen: | |||||
runtype = row[idx] | |||||
idx += 1 | |||||
if runtype == 0: | |||||
(runcount, runval) = struct.unpack_from("Bc", row, idx) | |||||
img.write(runval * runcount) | |||||
idx += 2 | |||||
else: | |||||
runcount = runtype | |||||
img.write(row[idx : idx + runcount]) | |||||
idx += runcount | |||||
except struct.error as e: | |||||
msg = f"Corrupted MSP file in row {x}" | |||||
raise OSError(msg) from e | |||||
self.set_as_raw(img.getvalue(), ("1", 0, 1)) | |||||
return -1, 0 | |||||
Image.register_decoder("MSP", MspDecoder) | |||||
# | |||||
# write MSP files (uncompressed only) | |||||
def _save(im, fp, filename): | |||||
if im.mode != "1": | |||||
msg = f"cannot write mode {im.mode} as MSP" | |||||
raise OSError(msg) | |||||
# create MSP header | |||||
header = [0] * 16 | |||||
header[0], header[1] = i16(b"Da"), i16(b"nM") # version 1 | |||||
header[2], header[3] = im.size | |||||
header[4], header[5] = 1, 1 | |||||
header[6], header[7] = 1, 1 | |||||
header[8], header[9] = im.size | |||||
checksum = 0 | |||||
for h in header: | |||||
checksum = checksum ^ h | |||||
header[12] = checksum # FIXME: is this the right field? | |||||
# header | |||||
for h in header: | |||||
fp.write(o16(h)) | |||||
# image body | |||||
ImageFile._save(im, fp, [("raw", (0, 0) + im.size, 32, ("1", 0, 1))]) | |||||
# | |||||
# registry | |||||
Image.register_open(MspImageFile.format, MspImageFile, _accept) | |||||
Image.register_save(MspImageFile.format, _save) | |||||
Image.register_extension(MspImageFile.format, ".msp") |
# | |||||
# The Python Imaging Library | |||||
# $Id$ | |||||
# | |||||
# Simple PostScript graphics interface | |||||
# | |||||
# History: | |||||
# 1996-04-20 fl Created | |||||
# 1999-01-10 fl Added gsave/grestore to image method | |||||
# 2005-05-04 fl Fixed floating point issue in image (from Eric Etheridge) | |||||
# | |||||
# Copyright (c) 1997-2005 by Secret Labs AB. All rights reserved. | |||||
# Copyright (c) 1996 by Fredrik Lundh. | |||||
# | |||||
# See the README file for information on usage and redistribution. | |||||
# | |||||
import sys | |||||
from . import EpsImagePlugin | |||||
## | |||||
# Simple PostScript graphics interface. | |||||
class PSDraw: | |||||
""" | |||||
Sets up printing to the given file. If ``fp`` is omitted, | |||||
``sys.stdout.buffer`` or ``sys.stdout`` is assumed. | |||||
""" | |||||
def __init__(self, fp=None): | |||||
if not fp: | |||||
try: | |||||
fp = sys.stdout.buffer | |||||
except AttributeError: | |||||
fp = sys.stdout | |||||
self.fp = fp | |||||
def begin_document(self, id=None): | |||||
"""Set up printing of a document. (Write PostScript DSC header.)""" | |||||
# FIXME: incomplete | |||||
self.fp.write( | |||||
b"%!PS-Adobe-3.0\n" | |||||
b"save\n" | |||||
b"/showpage { } def\n" | |||||
b"%%EndComments\n" | |||||
b"%%BeginDocument\n" | |||||
) | |||||
# self.fp.write(ERROR_PS) # debugging! | |||||
self.fp.write(EDROFF_PS) | |||||
self.fp.write(VDI_PS) | |||||
self.fp.write(b"%%EndProlog\n") | |||||
self.isofont = {} | |||||
def end_document(self): | |||||
"""Ends printing. (Write PostScript DSC footer.)""" | |||||
self.fp.write(b"%%EndDocument\nrestore showpage\n%%End\n") | |||||
if hasattr(self.fp, "flush"): | |||||
self.fp.flush() | |||||
def setfont(self, font, size): | |||||
""" | |||||
Selects which font to use. | |||||
:param font: A PostScript font name | |||||
:param size: Size in points. | |||||
""" | |||||
font = bytes(font, "UTF-8") | |||||
if font not in self.isofont: | |||||
# reencode font | |||||
self.fp.write(b"/PSDraw-%s ISOLatin1Encoding /%s E\n" % (font, font)) | |||||
self.isofont[font] = 1 | |||||
# rough | |||||
self.fp.write(b"/F0 %d /PSDraw-%s F\n" % (size, font)) | |||||
def line(self, xy0, xy1): | |||||
""" | |||||
Draws a line between the two points. Coordinates are given in | |||||
PostScript point coordinates (72 points per inch, (0, 0) is the lower | |||||
left corner of the page). | |||||
""" | |||||
self.fp.write(b"%d %d %d %d Vl\n" % (*xy0, *xy1)) | |||||
def rectangle(self, box): | |||||
""" | |||||
Draws a rectangle. | |||||
:param box: A tuple of four integers, specifying left, bottom, width and | |||||
height. | |||||
""" | |||||
self.fp.write(b"%d %d M 0 %d %d Vr\n" % box) | |||||
def text(self, xy, text): | |||||
""" | |||||
Draws text at the given position. You must use | |||||
:py:meth:`~PIL.PSDraw.PSDraw.setfont` before calling this method. | |||||
""" | |||||
text = bytes(text, "UTF-8") | |||||
text = b"\\(".join(text.split(b"(")) | |||||
text = b"\\)".join(text.split(b")")) | |||||
xy += (text,) | |||||
self.fp.write(b"%d %d M (%s) S\n" % xy) | |||||
def image(self, box, im, dpi=None): | |||||
"""Draw a PIL image, centered in the given box.""" | |||||
# default resolution depends on mode | |||||
if not dpi: | |||||
if im.mode == "1": | |||||
dpi = 200 # fax | |||||
else: | |||||
dpi = 100 # greyscale | |||||
# image size (on paper) | |||||
x = im.size[0] * 72 / dpi | |||||
y = im.size[1] * 72 / dpi | |||||
# max allowed size | |||||
xmax = float(box[2] - box[0]) | |||||
ymax = float(box[3] - box[1]) | |||||
if x > xmax: | |||||
y = y * xmax / x | |||||
x = xmax | |||||
if y > ymax: | |||||
x = x * ymax / y | |||||
y = ymax | |||||
dx = (xmax - x) / 2 + box[0] | |||||
dy = (ymax - y) / 2 + box[1] | |||||
self.fp.write(b"gsave\n%f %f translate\n" % (dx, dy)) | |||||
if (x, y) != im.size: | |||||
# EpsImagePlugin._save prints the image at (0,0,xsize,ysize) | |||||
sx = x / im.size[0] | |||||
sy = y / im.size[1] | |||||
self.fp.write(b"%f %f scale\n" % (sx, sy)) | |||||
EpsImagePlugin._save(im, self.fp, None, 0) | |||||
self.fp.write(b"\ngrestore\n") | |||||
# -------------------------------------------------------------------- | |||||
# PostScript driver | |||||
# | |||||
# EDROFF.PS -- PostScript driver for Edroff 2 | |||||
# | |||||
# History: | |||||
# 94-01-25 fl: created (edroff 2.04) | |||||
# | |||||
# Copyright (c) Fredrik Lundh 1994. | |||||
# | |||||
EDROFF_PS = b"""\ | |||||
/S { show } bind def | |||||
/P { moveto show } bind def | |||||
/M { moveto } bind def | |||||
/X { 0 rmoveto } bind def | |||||
/Y { 0 exch rmoveto } bind def | |||||
/E { findfont | |||||
dup maxlength dict begin | |||||
{ | |||||
1 index /FID ne { def } { pop pop } ifelse | |||||
} forall | |||||
/Encoding exch def | |||||
dup /FontName exch def | |||||
currentdict end definefont pop | |||||
} bind def | |||||
/F { findfont exch scalefont dup setfont | |||||
[ exch /setfont cvx ] cvx bind def | |||||
} bind def | |||||
""" | |||||
# | |||||
# VDI.PS -- PostScript driver for VDI meta commands | |||||
# | |||||
# History: | |||||
# 94-01-25 fl: created (edroff 2.04) | |||||
# | |||||
# Copyright (c) Fredrik Lundh 1994. | |||||
# | |||||
VDI_PS = b"""\ | |||||
/Vm { moveto } bind def | |||||
/Va { newpath arcn stroke } bind def | |||||
/Vl { moveto lineto stroke } bind def | |||||
/Vc { newpath 0 360 arc closepath } bind def | |||||
/Vr { exch dup 0 rlineto | |||||
exch dup 0 exch rlineto | |||||
exch neg 0 rlineto | |||||
0 exch neg rlineto | |||||
setgray fill } bind def | |||||
/Tm matrix def | |||||
/Ve { Tm currentmatrix pop | |||||
translate scale newpath 0 0 .5 0 360 arc closepath | |||||
Tm setmatrix | |||||
} bind def | |||||
/Vf { currentgray exch setgray fill setgray } bind def | |||||
""" | |||||
# | |||||
# ERROR.PS -- Error handler | |||||
# | |||||
# History: | |||||
# 89-11-21 fl: created (pslist 1.10) | |||||
# | |||||
ERROR_PS = b"""\ | |||||
/landscape false def | |||||
/errorBUF 200 string def | |||||
/errorNL { currentpoint 10 sub exch pop 72 exch moveto } def | |||||
errordict begin /handleerror { | |||||
initmatrix /Courier findfont 10 scalefont setfont | |||||
newpath 72 720 moveto $error begin /newerror false def | |||||
(PostScript Error) show errorNL errorNL | |||||
(Error: ) show | |||||
/errorname load errorBUF cvs show errorNL errorNL | |||||
(Command: ) show | |||||
/command load dup type /stringtype ne { errorBUF cvs } if show | |||||
errorNL errorNL | |||||
(VMstatus: ) show | |||||
vmstatus errorBUF cvs show ( bytes available, ) show | |||||
errorBUF cvs show ( bytes used at level ) show | |||||
errorBUF cvs show errorNL errorNL | |||||
(Operand stargck: ) show errorNL /ostargck load { | |||||
dup type /stringtype ne { errorBUF cvs } if 72 0 rmoveto show errorNL | |||||
} forall errorNL | |||||
(Execution stargck: ) show errorNL /estargck load { | |||||
dup type /stringtype ne { errorBUF cvs } if 72 0 rmoveto show errorNL | |||||
} forall | |||||
end showpage | |||||
} def end | |||||
""" |
# | |||||
# Python Imaging Library | |||||
# $Id$ | |||||
# | |||||
# stuff to read simple, teragon-style palette files | |||||
# | |||||
# History: | |||||
# 97-08-23 fl Created | |||||
# | |||||
# Copyright (c) Secret Labs AB 1997. | |||||
# Copyright (c) Fredrik Lundh 1997. | |||||
# | |||||
# See the README file for information on usage and redistribution. | |||||
# | |||||
from ._binary import o8 | |||||
class PaletteFile: | |||||
"""File handler for Teragon-style palette files.""" | |||||
rawmode = "RGB" | |||||
def __init__(self, fp): | |||||
self.palette = [(i, i, i) for i in range(256)] | |||||
while True: | |||||
s = fp.readline() | |||||
if not s: | |||||
break | |||||
if s[:1] == b"#": | |||||
continue | |||||
if len(s) > 100: | |||||
msg = "bad palette file" | |||||
raise SyntaxError(msg) | |||||
v = [int(x) for x in s.split()] | |||||
try: | |||||
[i, r, g, b] = v | |||||
except ValueError: | |||||
[i, r] = v | |||||
g = b = r | |||||
if 0 <= i <= 255: | |||||
self.palette[i] = o8(r) + o8(g) + o8(b) | |||||
self.palette = b"".join(self.palette) | |||||
def getpalette(self): | |||||
return self.palette, self.rawmode |
# | |||||
# The Python Imaging Library. | |||||
# $Id$ | |||||
# | |||||
## | |||||
# Image plugin for Palm pixmap images (output only). | |||||
## | |||||
from . import Image, ImageFile | |||||
from ._binary import o8 | |||||
from ._binary import o16be as o16b | |||||
# fmt: off | |||||
_Palm8BitColormapValues = ( | |||||
(255, 255, 255), (255, 204, 255), (255, 153, 255), (255, 102, 255), | |||||
(255, 51, 255), (255, 0, 255), (255, 255, 204), (255, 204, 204), | |||||
(255, 153, 204), (255, 102, 204), (255, 51, 204), (255, 0, 204), | |||||
(255, 255, 153), (255, 204, 153), (255, 153, 153), (255, 102, 153), | |||||
(255, 51, 153), (255, 0, 153), (204, 255, 255), (204, 204, 255), | |||||
(204, 153, 255), (204, 102, 255), (204, 51, 255), (204, 0, 255), | |||||
(204, 255, 204), (204, 204, 204), (204, 153, 204), (204, 102, 204), | |||||
(204, 51, 204), (204, 0, 204), (204, 255, 153), (204, 204, 153), | |||||
(204, 153, 153), (204, 102, 153), (204, 51, 153), (204, 0, 153), | |||||
(153, 255, 255), (153, 204, 255), (153, 153, 255), (153, 102, 255), | |||||
(153, 51, 255), (153, 0, 255), (153, 255, 204), (153, 204, 204), | |||||
(153, 153, 204), (153, 102, 204), (153, 51, 204), (153, 0, 204), | |||||
(153, 255, 153), (153, 204, 153), (153, 153, 153), (153, 102, 153), | |||||
(153, 51, 153), (153, 0, 153), (102, 255, 255), (102, 204, 255), | |||||
(102, 153, 255), (102, 102, 255), (102, 51, 255), (102, 0, 255), | |||||
(102, 255, 204), (102, 204, 204), (102, 153, 204), (102, 102, 204), | |||||
(102, 51, 204), (102, 0, 204), (102, 255, 153), (102, 204, 153), | |||||
(102, 153, 153), (102, 102, 153), (102, 51, 153), (102, 0, 153), | |||||
(51, 255, 255), (51, 204, 255), (51, 153, 255), (51, 102, 255), | |||||
(51, 51, 255), (51, 0, 255), (51, 255, 204), (51, 204, 204), | |||||
(51, 153, 204), (51, 102, 204), (51, 51, 204), (51, 0, 204), | |||||
(51, 255, 153), (51, 204, 153), (51, 153, 153), (51, 102, 153), | |||||
(51, 51, 153), (51, 0, 153), (0, 255, 255), (0, 204, 255), | |||||
(0, 153, 255), (0, 102, 255), (0, 51, 255), (0, 0, 255), | |||||
(0, 255, 204), (0, 204, 204), (0, 153, 204), (0, 102, 204), | |||||
(0, 51, 204), (0, 0, 204), (0, 255, 153), (0, 204, 153), | |||||
(0, 153, 153), (0, 102, 153), (0, 51, 153), (0, 0, 153), | |||||
(255, 255, 102), (255, 204, 102), (255, 153, 102), (255, 102, 102), | |||||
(255, 51, 102), (255, 0, 102), (255, 255, 51), (255, 204, 51), | |||||
(255, 153, 51), (255, 102, 51), (255, 51, 51), (255, 0, 51), | |||||
(255, 255, 0), (255, 204, 0), (255, 153, 0), (255, 102, 0), | |||||
(255, 51, 0), (255, 0, 0), (204, 255, 102), (204, 204, 102), | |||||
(204, 153, 102), (204, 102, 102), (204, 51, 102), (204, 0, 102), | |||||
(204, 255, 51), (204, 204, 51), (204, 153, 51), (204, 102, 51), | |||||
(204, 51, 51), (204, 0, 51), (204, 255, 0), (204, 204, 0), | |||||
(204, 153, 0), (204, 102, 0), (204, 51, 0), (204, 0, 0), | |||||
(153, 255, 102), (153, 204, 102), (153, 153, 102), (153, 102, 102), | |||||
(153, 51, 102), (153, 0, 102), (153, 255, 51), (153, 204, 51), | |||||
(153, 153, 51), (153, 102, 51), (153, 51, 51), (153, 0, 51), | |||||
(153, 255, 0), (153, 204, 0), (153, 153, 0), (153, 102, 0), | |||||
(153, 51, 0), (153, 0, 0), (102, 255, 102), (102, 204, 102), | |||||
(102, 153, 102), (102, 102, 102), (102, 51, 102), (102, 0, 102), | |||||
(102, 255, 51), (102, 204, 51), (102, 153, 51), (102, 102, 51), | |||||
(102, 51, 51), (102, 0, 51), (102, 255, 0), (102, 204, 0), | |||||
(102, 153, 0), (102, 102, 0), (102, 51, 0), (102, 0, 0), | |||||
(51, 255, 102), (51, 204, 102), (51, 153, 102), (51, 102, 102), | |||||
(51, 51, 102), (51, 0, 102), (51, 255, 51), (51, 204, 51), | |||||
(51, 153, 51), (51, 102, 51), (51, 51, 51), (51, 0, 51), | |||||
(51, 255, 0), (51, 204, 0), (51, 153, 0), (51, 102, 0), | |||||
(51, 51, 0), (51, 0, 0), (0, 255, 102), (0, 204, 102), | |||||
(0, 153, 102), (0, 102, 102), (0, 51, 102), (0, 0, 102), | |||||
(0, 255, 51), (0, 204, 51), (0, 153, 51), (0, 102, 51), | |||||
(0, 51, 51), (0, 0, 51), (0, 255, 0), (0, 204, 0), | |||||
(0, 153, 0), (0, 102, 0), (0, 51, 0), (17, 17, 17), | |||||
(34, 34, 34), (68, 68, 68), (85, 85, 85), (119, 119, 119), | |||||
(136, 136, 136), (170, 170, 170), (187, 187, 187), (221, 221, 221), | |||||
(238, 238, 238), (192, 192, 192), (128, 0, 0), (128, 0, 128), | |||||
(0, 128, 0), (0, 128, 128), (0, 0, 0), (0, 0, 0), | |||||
(0, 0, 0), (0, 0, 0), (0, 0, 0), (0, 0, 0), | |||||
(0, 0, 0), (0, 0, 0), (0, 0, 0), (0, 0, 0), | |||||
(0, 0, 0), (0, 0, 0), (0, 0, 0), (0, 0, 0), | |||||
(0, 0, 0), (0, 0, 0), (0, 0, 0), (0, 0, 0), | |||||
(0, 0, 0), (0, 0, 0), (0, 0, 0), (0, 0, 0), | |||||
(0, 0, 0), (0, 0, 0), (0, 0, 0), (0, 0, 0)) | |||||
# fmt: on | |||||
# so build a prototype image to be used for palette resampling | |||||
def build_prototype_image(): | |||||
image = Image.new("L", (1, len(_Palm8BitColormapValues))) | |||||
image.putdata(list(range(len(_Palm8BitColormapValues)))) | |||||
palettedata = () | |||||
for colormapValue in _Palm8BitColormapValues: | |||||
palettedata += colormapValue | |||||
palettedata += (0, 0, 0) * (256 - len(_Palm8BitColormapValues)) | |||||
image.putpalette(palettedata) | |||||
return image | |||||
Palm8BitColormapImage = build_prototype_image() | |||||
# OK, we now have in Palm8BitColormapImage, | |||||
# a "P"-mode image with the right palette | |||||
# | |||||
# -------------------------------------------------------------------- | |||||
_FLAGS = {"custom-colormap": 0x4000, "is-compressed": 0x8000, "has-transparent": 0x2000} | |||||
_COMPRESSION_TYPES = {"none": 0xFF, "rle": 0x01, "scanline": 0x00} | |||||
# | |||||
# -------------------------------------------------------------------- | |||||
## | |||||
# (Internal) Image save plugin for the Palm format. | |||||
def _save(im, fp, filename): | |||||
if im.mode == "P": | |||||
# we assume this is a color Palm image with the standard colormap, | |||||
# unless the "info" dict has a "custom-colormap" field | |||||
rawmode = "P" | |||||
bpp = 8 | |||||
version = 1 | |||||
elif im.mode == "L": | |||||
if im.encoderinfo.get("bpp") in (1, 2, 4): | |||||
# this is 8-bit grayscale, so we shift it to get the high-order bits, | |||||
# and invert it because | |||||
# Palm does greyscale from white (0) to black (1) | |||||
bpp = im.encoderinfo["bpp"] | |||||
im = im.point( | |||||
lambda x, shift=8 - bpp, maxval=(1 << bpp) - 1: maxval - (x >> shift) | |||||
) | |||||
elif im.info.get("bpp") in (1, 2, 4): | |||||
# here we assume that even though the inherent mode is 8-bit grayscale, | |||||
# only the lower bpp bits are significant. | |||||
# We invert them to match the Palm. | |||||
bpp = im.info["bpp"] | |||||
im = im.point(lambda x, maxval=(1 << bpp) - 1: maxval - (x & maxval)) | |||||
else: | |||||
msg = f"cannot write mode {im.mode} as Palm" | |||||
raise OSError(msg) | |||||
# we ignore the palette here | |||||
im.mode = "P" | |||||
rawmode = "P;" + str(bpp) | |||||
version = 1 | |||||
elif im.mode == "1": | |||||
# monochrome -- write it inverted, as is the Palm standard | |||||
rawmode = "1;I" | |||||
bpp = 1 | |||||
version = 0 | |||||
else: | |||||
msg = f"cannot write mode {im.mode} as Palm" | |||||
raise OSError(msg) | |||||
# | |||||
# make sure image data is available | |||||
im.load() | |||||
# write header | |||||
cols = im.size[0] | |||||
rows = im.size[1] | |||||
rowbytes = int((cols + (16 // bpp - 1)) / (16 // bpp)) * 2 | |||||
transparent_index = 0 | |||||
compression_type = _COMPRESSION_TYPES["none"] | |||||
flags = 0 | |||||
if im.mode == "P" and "custom-colormap" in im.info: | |||||
flags = flags & _FLAGS["custom-colormap"] | |||||
colormapsize = 4 * 256 + 2 | |||||
colormapmode = im.palette.mode | |||||
colormap = im.getdata().getpalette() | |||||
else: | |||||
colormapsize = 0 | |||||
if "offset" in im.info: | |||||
offset = (rowbytes * rows + 16 + 3 + colormapsize) // 4 | |||||
else: | |||||
offset = 0 | |||||
fp.write(o16b(cols) + o16b(rows) + o16b(rowbytes) + o16b(flags)) | |||||
fp.write(o8(bpp)) | |||||
fp.write(o8(version)) | |||||
fp.write(o16b(offset)) | |||||
fp.write(o8(transparent_index)) | |||||
fp.write(o8(compression_type)) | |||||
fp.write(o16b(0)) # reserved by Palm | |||||
# now write colormap if necessary | |||||
if colormapsize > 0: | |||||
fp.write(o16b(256)) | |||||
for i in range(256): | |||||
fp.write(o8(i)) | |||||
if colormapmode == "RGB": | |||||
fp.write( | |||||
o8(colormap[3 * i]) | |||||
+ o8(colormap[3 * i + 1]) | |||||
+ o8(colormap[3 * i + 2]) | |||||
) | |||||
elif colormapmode == "RGBA": | |||||
fp.write( | |||||
o8(colormap[4 * i]) | |||||
+ o8(colormap[4 * i + 1]) | |||||
+ o8(colormap[4 * i + 2]) | |||||
) | |||||
# now convert data to raw form | |||||
ImageFile._save(im, fp, [("raw", (0, 0) + im.size, 0, (rawmode, rowbytes, 1))]) | |||||
if hasattr(fp, "flush"): | |||||
fp.flush() | |||||
# | |||||
# -------------------------------------------------------------------- | |||||
Image.register_save("Palm", _save) | |||||
Image.register_extension("Palm", ".palm") | |||||
Image.register_mime("Palm", "image/palm") |