AK51 commited on
Commit
b610d23
·
verified ·
1 Parent(s): 35ae7d4

Upload 13308 files

Browse files
This view is limited to 50 files because it contains too many changes.   See raw diff
Files changed (50) hide show
  1. .gitattributes +176 -0
  2. .kiro/specs/nasa-solar-image-downloader/design.md +231 -0
  3. .kiro/specs/nasa-solar-image-downloader/requirements.md +80 -0
  4. .kiro/specs/nasa-solar-image-downloader/tasks.md +182 -0
  5. .pytest_cache/.gitignore +2 -0
  6. .pytest_cache/CACHEDIR.TAG +4 -0
  7. .pytest_cache/README.md +8 -0
  8. .pytest_cache/v/cache/lastfailed +1 -0
  9. .pytest_cache/v/cache/nodeids +85 -0
  10. .venv/.gitignore +1 -0
  11. .venv/Lib/site-packages/PIL/AvifImagePlugin.py +291 -0
  12. .venv/Lib/site-packages/PIL/BdfFontFile.py +122 -0
  13. .venv/Lib/site-packages/PIL/BlpImagePlugin.py +497 -0
  14. .venv/Lib/site-packages/PIL/BmpImagePlugin.py +515 -0
  15. .venv/Lib/site-packages/PIL/BufrStubImagePlugin.py +75 -0
  16. .venv/Lib/site-packages/PIL/ContainerIO.py +173 -0
  17. .venv/Lib/site-packages/PIL/CurImagePlugin.py +75 -0
  18. .venv/Lib/site-packages/PIL/DcxImagePlugin.py +83 -0
  19. .venv/Lib/site-packages/PIL/DdsImagePlugin.py +624 -0
  20. .venv/Lib/site-packages/PIL/EpsImagePlugin.py +479 -0
  21. .venv/Lib/site-packages/PIL/ExifTags.py +382 -0
  22. .venv/Lib/site-packages/PIL/FitsImagePlugin.py +152 -0
  23. .venv/Lib/site-packages/PIL/FliImagePlugin.py +184 -0
  24. .venv/Lib/site-packages/PIL/FontFile.py +134 -0
  25. .venv/Lib/site-packages/PIL/FpxImagePlugin.py +257 -0
  26. .venv/Lib/site-packages/PIL/FtexImagePlugin.py +114 -0
  27. .venv/Lib/site-packages/PIL/GbrImagePlugin.py +101 -0
  28. .venv/Lib/site-packages/PIL/GdImageFile.py +102 -0
  29. .venv/Lib/site-packages/PIL/GifImagePlugin.py +1215 -0
  30. .venv/Lib/site-packages/PIL/GimpGradientFile.py +153 -0
  31. .venv/Lib/site-packages/PIL/GimpPaletteFile.py +75 -0
  32. .venv/Lib/site-packages/PIL/GribStubImagePlugin.py +75 -0
  33. .venv/Lib/site-packages/PIL/Hdf5StubImagePlugin.py +75 -0
  34. .venv/Lib/site-packages/PIL/IcnsImagePlugin.py +401 -0
  35. .venv/Lib/site-packages/PIL/IcoImagePlugin.py +381 -0
  36. .venv/Lib/site-packages/PIL/ImImagePlugin.py +389 -0
  37. .venv/Lib/site-packages/PIL/Image.py +0 -0
  38. .venv/Lib/site-packages/PIL/ImageChops.py +311 -0
  39. .venv/Lib/site-packages/PIL/ImageCms.py +1076 -0
  40. .venv/Lib/site-packages/PIL/ImageColor.py +320 -0
  41. .venv/Lib/site-packages/PIL/ImageDraw.py +1036 -0
  42. .venv/Lib/site-packages/PIL/ImageDraw2.py +243 -0
  43. .venv/Lib/site-packages/PIL/ImageEnhance.py +113 -0
  44. .venv/Lib/site-packages/PIL/ImageFile.py +926 -0
  45. .venv/Lib/site-packages/PIL/ImageFilter.py +607 -0
  46. .venv/Lib/site-packages/PIL/ImageFont.py +1312 -0
  47. .venv/Lib/site-packages/PIL/ImageGrab.py +196 -0
  48. .venv/Lib/site-packages/PIL/ImageMath.py +314 -0
  49. .venv/Lib/site-packages/PIL/ImageMode.py +85 -0
  50. .venv/Lib/site-packages/PIL/ImageMorph.py +265 -0
.gitattributes CHANGED
@@ -33,3 +33,179 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ .venv/Lib/site-packages/__pycache__/typing_extensions.cpython-310.pyc filter=lfs diff=lfs merge=lfs -text
37
+ .venv/Lib/site-packages/_brotli.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
38
+ .venv/Lib/site-packages/charset_normalizer/md__mypyc.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
39
+ .venv/Lib/site-packages/cv2/cv2.pyd filter=lfs diff=lfs merge=lfs -text
40
+ .venv/Lib/site-packages/cv2/opencv_videoio_ffmpeg4120_64.dll filter=lfs diff=lfs merge=lfs -text
41
+ .venv/Lib/site-packages/gradio/__pycache__/blocks.cpython-310.pyc filter=lfs diff=lfs merge=lfs -text
42
+ .venv/Lib/site-packages/gradio/media_assets/audio/cantina.wav filter=lfs diff=lfs merge=lfs -text
43
+ .venv/Lib/site-packages/gradio/media_assets/audio/recording1.wav filter=lfs diff=lfs merge=lfs -text
44
+ .venv/Lib/site-packages/gradio/media_assets/audio/sax.wav filter=lfs diff=lfs merge=lfs -text
45
+ .venv/Lib/site-packages/gradio/media_assets/images/groot.jpeg filter=lfs diff=lfs merge=lfs -text
46
+ .venv/Lib/site-packages/gradio/media_assets/images/tower.jpg filter=lfs diff=lfs merge=lfs -text
47
+ .venv/Lib/site-packages/gradio/media_assets/models3d/Duck.glb filter=lfs diff=lfs merge=lfs -text
48
+ .venv/Lib/site-packages/gradio/media_assets/models3d/sofia.stl filter=lfs diff=lfs merge=lfs -text
49
+ .venv/Lib/site-packages/gradio/media_assets/videos/a.mp4 filter=lfs diff=lfs merge=lfs -text
50
+ .venv/Lib/site-packages/gradio/media_assets/videos/b.mp4 filter=lfs diff=lfs merge=lfs -text
51
+ .venv/Lib/site-packages/gradio/media_assets/videos/world.mp4 filter=lfs diff=lfs merge=lfs -text
52
+ .venv/Lib/site-packages/gradio/templates/frontend/static/fonts/Source[[:space:]]Sans[[:space:]]Pro/SourceSansPro-Bold.woff2 filter=lfs diff=lfs merge=lfs -text
53
+ .venv/Lib/site-packages/gradio/templates/frontend/static/fonts/Source[[:space:]]Sans[[:space:]]Pro/SourceSansPro-Regular.woff2 filter=lfs diff=lfs merge=lfs -text
54
+ .venv/Lib/site-packages/gradio/templates/node/build/client/_app/immutable/assets/Index.BEZDBfgO.css.br filter=lfs diff=lfs merge=lfs -text
55
+ .venv/Lib/site-packages/gradio/templates/node/build/client/_app/immutable/chunks/BnGCFZm2.js.br filter=lfs diff=lfs merge=lfs -text
56
+ .venv/Lib/site-packages/gradio/templates/node/build/client/_app/immutable/chunks/BnkdMOzK.js.br filter=lfs diff=lfs merge=lfs -text
57
+ .venv/Lib/site-packages/gradio/templates/node/build/client/_app/immutable/chunks/BpXYcn3B.js.br filter=lfs diff=lfs merge=lfs -text
58
+ .venv/Lib/site-packages/gradio/templates/node/build/client/_app/immutable/chunks/BRl6J_3q.js.br filter=lfs diff=lfs merge=lfs -text
59
+ .venv/Lib/site-packages/gradio/templates/node/build/client/_app/immutable/chunks/CNFd7YEL.js.br filter=lfs diff=lfs merge=lfs -text
60
+ .venv/Lib/site-packages/gradio/templates/node/build/client/_app/immutable/chunks/CwDjrd1a.js.br filter=lfs diff=lfs merge=lfs -text
61
+ .venv/Lib/site-packages/gradio/templates/node/build/client/_app/immutable/chunks/CXISBPh7.js.br filter=lfs diff=lfs merge=lfs -text
62
+ .venv/Lib/site-packages/gradio/templates/node/build/client/_app/immutable/chunks/CXwQHv0N.js.br filter=lfs diff=lfs merge=lfs -text
63
+ .venv/Lib/site-packages/gradio/templates/node/build/client/_app/immutable/chunks/CZjDJdHJ.js.br filter=lfs diff=lfs merge=lfs -text
64
+ .venv/Lib/site-packages/gradio/templates/node/build/client/_app/immutable/chunks/D4Fs8SLG.js.br filter=lfs diff=lfs merge=lfs -text
65
+ .venv/Lib/site-packages/gradio/templates/node/build/client/_app/immutable/chunks/DQN0kXwA.js.br filter=lfs diff=lfs merge=lfs -text
66
+ .venv/Lib/site-packages/gradio/templates/node/build/client/_app/immutable/chunks/DsEewaC8.js.br filter=lfs diff=lfs merge=lfs -text
67
+ .venv/Lib/site-packages/gradio/templates/node/build/client/_app/immutable/chunks/DxmqXz3e.js.br filter=lfs diff=lfs merge=lfs -text
68
+ .venv/Lib/site-packages/gradio/templates/node/build/client/_app/immutable/chunks/ubPmiu88.js.br filter=lfs diff=lfs merge=lfs -text
69
+ .venv/Lib/site-packages/hf_xet/hf_xet.pyd filter=lfs diff=lfs merge=lfs -text
70
+ .venv/Lib/site-packages/huggingface_hub/__pycache__/hf_api.cpython-310.pyc filter=lfs diff=lfs merge=lfs -text
71
+ .venv/Lib/site-packages/huggingface_hub/inference/__pycache__/_client.cpython-310.pyc filter=lfs diff=lfs merge=lfs -text
72
+ .venv/Lib/site-packages/huggingface_hub/inference/_generated/__pycache__/_async_client.cpython-310.pyc filter=lfs diff=lfs merge=lfs -text
73
+ .venv/Lib/site-packages/idna/__pycache__/idnadata.cpython-310.pyc filter=lfs diff=lfs merge=lfs -text
74
+ .venv/Lib/site-packages/idna/__pycache__/uts46data.cpython-310.pyc filter=lfs diff=lfs merge=lfs -text
75
+ .venv/Lib/site-packages/numpy.libs/libscipy_openblas64_-13e2df515630b4a41f92893938845698.dll filter=lfs diff=lfs merge=lfs -text
76
+ .venv/Lib/site-packages/numpy.libs/msvcp140-263139962577ecda4cd9469ca360a746.dll filter=lfs diff=lfs merge=lfs -text
77
+ .venv/Lib/site-packages/numpy/_core/__pycache__/_add_newdocs.cpython-310.pyc filter=lfs diff=lfs merge=lfs -text
78
+ .venv/Lib/site-packages/numpy/_core/__pycache__/fromnumeric.cpython-310.pyc filter=lfs diff=lfs merge=lfs -text
79
+ .venv/Lib/site-packages/numpy/_core/_multiarray_umath.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
80
+ .venv/Lib/site-packages/numpy/_core/_simd.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
81
+ .venv/Lib/site-packages/numpy/_core/lib/npymath.lib filter=lfs diff=lfs merge=lfs -text
82
+ .venv/Lib/site-packages/numpy/_core/tests/__pycache__/test_multiarray.cpython-310.pyc filter=lfs diff=lfs merge=lfs -text
83
+ .venv/Lib/site-packages/numpy/_core/tests/__pycache__/test_numeric.cpython-310.pyc filter=lfs diff=lfs merge=lfs -text
84
+ .venv/Lib/site-packages/numpy/_core/tests/__pycache__/test_regression.cpython-310.pyc filter=lfs diff=lfs merge=lfs -text
85
+ .venv/Lib/site-packages/numpy/_core/tests/__pycache__/test_ufunc.cpython-310.pyc filter=lfs diff=lfs merge=lfs -text
86
+ .venv/Lib/site-packages/numpy/_core/tests/__pycache__/test_umath.cpython-310.pyc filter=lfs diff=lfs merge=lfs -text
87
+ .venv/Lib/site-packages/numpy/fft/_pocketfft_umath.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
88
+ .venv/Lib/site-packages/numpy/lib/__pycache__/_function_base_impl.cpython-310.pyc filter=lfs diff=lfs merge=lfs -text
89
+ .venv/Lib/site-packages/numpy/lib/tests/__pycache__/test_function_base.cpython-310.pyc filter=lfs diff=lfs merge=lfs -text
90
+ .venv/Lib/site-packages/numpy/linalg/__pycache__/_linalg.cpython-310.pyc filter=lfs diff=lfs merge=lfs -text
91
+ .venv/Lib/site-packages/numpy/linalg/_umath_linalg.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
92
+ .venv/Lib/site-packages/numpy/ma/__pycache__/core.cpython-310.pyc filter=lfs diff=lfs merge=lfs -text
93
+ .venv/Lib/site-packages/numpy/ma/tests/__pycache__/test_core.cpython-310.pyc filter=lfs diff=lfs merge=lfs -text
94
+ .venv/Lib/site-packages/numpy/random/_bounded_integers.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
95
+ .venv/Lib/site-packages/numpy/random/_common.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
96
+ .venv/Lib/site-packages/numpy/random/_generator.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
97
+ .venv/Lib/site-packages/numpy/random/bit_generator.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
98
+ .venv/Lib/site-packages/numpy/random/lib/npyrandom.lib filter=lfs diff=lfs merge=lfs -text
99
+ .venv/Lib/site-packages/numpy/random/mtrand.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
100
+ .venv/Lib/site-packages/orjson/orjson.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
101
+ .venv/Lib/site-packages/pandas.libs/msvcp140-a4c2229bdc2a2a630acdc095b4d86008.dll filter=lfs diff=lfs merge=lfs -text
102
+ .venv/Lib/site-packages/pandas/_libs/algos.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
103
+ .venv/Lib/site-packages/pandas/_libs/groupby.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
104
+ .venv/Lib/site-packages/pandas/_libs/hashing.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
105
+ .venv/Lib/site-packages/pandas/_libs/hashtable.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
106
+ .venv/Lib/site-packages/pandas/_libs/index.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
107
+ .venv/Lib/site-packages/pandas/_libs/internals.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
108
+ .venv/Lib/site-packages/pandas/_libs/interval.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
109
+ .venv/Lib/site-packages/pandas/_libs/join.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
110
+ .venv/Lib/site-packages/pandas/_libs/lib.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
111
+ .venv/Lib/site-packages/pandas/_libs/missing.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
112
+ .venv/Lib/site-packages/pandas/_libs/ops.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
113
+ .venv/Lib/site-packages/pandas/_libs/parsers.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
114
+ .venv/Lib/site-packages/pandas/_libs/reshape.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
115
+ .venv/Lib/site-packages/pandas/_libs/sas.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
116
+ .venv/Lib/site-packages/pandas/_libs/sparse.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
117
+ .venv/Lib/site-packages/pandas/_libs/tslib.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
118
+ .venv/Lib/site-packages/pandas/_libs/tslibs/conversion.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
119
+ .venv/Lib/site-packages/pandas/_libs/tslibs/dtypes.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
120
+ .venv/Lib/site-packages/pandas/_libs/tslibs/fields.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
121
+ .venv/Lib/site-packages/pandas/_libs/tslibs/nattype.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
122
+ .venv/Lib/site-packages/pandas/_libs/tslibs/np_datetime.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
123
+ .venv/Lib/site-packages/pandas/_libs/tslibs/offsets.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
124
+ .venv/Lib/site-packages/pandas/_libs/tslibs/parsing.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
125
+ .venv/Lib/site-packages/pandas/_libs/tslibs/period.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
126
+ .venv/Lib/site-packages/pandas/_libs/tslibs/strptime.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
127
+ .venv/Lib/site-packages/pandas/_libs/tslibs/timedeltas.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
128
+ .venv/Lib/site-packages/pandas/_libs/tslibs/timestamps.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
129
+ .venv/Lib/site-packages/pandas/_libs/tslibs/timezones.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
130
+ .venv/Lib/site-packages/pandas/_libs/tslibs/tzconversion.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
131
+ .venv/Lib/site-packages/pandas/_libs/tslibs/vectorized.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
132
+ .venv/Lib/site-packages/pandas/_libs/window/aggregations.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
133
+ .venv/Lib/site-packages/pandas/_libs/window/indexers.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
134
+ .venv/Lib/site-packages/pandas/_libs/writers.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
135
+ .venv/Lib/site-packages/pandas/core/__pycache__/frame.cpython-310.pyc filter=lfs diff=lfs merge=lfs -text
136
+ .venv/Lib/site-packages/pandas/core/__pycache__/generic.cpython-310.pyc filter=lfs diff=lfs merge=lfs -text
137
+ .venv/Lib/site-packages/pandas/core/__pycache__/series.cpython-310.pyc filter=lfs diff=lfs merge=lfs -text
138
+ .venv/Lib/site-packages/pandas/core/groupby/__pycache__/groupby.cpython-310.pyc filter=lfs diff=lfs merge=lfs -text
139
+ .venv/Lib/site-packages/pandas/core/indexes/__pycache__/base.cpython-310.pyc filter=lfs diff=lfs merge=lfs -text
140
+ .venv/Lib/site-packages/pandas/core/indexes/__pycache__/multi.cpython-310.pyc filter=lfs diff=lfs merge=lfs -text
141
+ .venv/Lib/site-packages/pandas/core/strings/__pycache__/accessor.cpython-310.pyc filter=lfs diff=lfs merge=lfs -text
142
+ .venv/Lib/site-packages/pandas/io/__pycache__/pytables.cpython-310.pyc filter=lfs diff=lfs merge=lfs -text
143
+ .venv/Lib/site-packages/pandas/io/__pycache__/stata.cpython-310.pyc filter=lfs diff=lfs merge=lfs -text
144
+ .venv/Lib/site-packages/pandas/io/formats/__pycache__/style.cpython-310.pyc filter=lfs diff=lfs merge=lfs -text
145
+ .venv/Lib/site-packages/pandas/tests/frame/__pycache__/test_constructors.cpython-310.pyc filter=lfs diff=lfs merge=lfs -text
146
+ .venv/Lib/site-packages/pandas/tests/indexing/__pycache__/test_loc.cpython-310.pyc filter=lfs diff=lfs merge=lfs -text
147
+ .venv/Lib/site-packages/pandas/tests/io/__pycache__/test_sql.cpython-310.pyc filter=lfs diff=lfs merge=lfs -text
148
+ .venv/Lib/site-packages/pandas/tests/tools/__pycache__/test_to_datetime.cpython-310.pyc filter=lfs diff=lfs merge=lfs -text
149
+ .venv/Lib/site-packages/PIL/__pycache__/Image.cpython-310.pyc filter=lfs diff=lfs merge=lfs -text
150
+ .venv/Lib/site-packages/PIL/_avif.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
151
+ .venv/Lib/site-packages/PIL/_imaging.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
152
+ .venv/Lib/site-packages/PIL/_imagingcms.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
153
+ .venv/Lib/site-packages/PIL/_imagingft.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
154
+ .venv/Lib/site-packages/PIL/_webp.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
155
+ .venv/Lib/site-packages/pip/_vendor/distlib/t64-arm.exe filter=lfs diff=lfs merge=lfs -text
156
+ .venv/Lib/site-packages/pip/_vendor/distlib/t64.exe filter=lfs diff=lfs merge=lfs -text
157
+ .venv/Lib/site-packages/pip/_vendor/distlib/w64-arm.exe filter=lfs diff=lfs merge=lfs -text
158
+ .venv/Lib/site-packages/pip/_vendor/distlib/w64.exe filter=lfs diff=lfs merge=lfs -text
159
+ .venv/Lib/site-packages/pip/_vendor/idna/__pycache__/idnadata.cpython-310.pyc filter=lfs diff=lfs merge=lfs -text
160
+ .venv/Lib/site-packages/pip/_vendor/idna/__pycache__/uts46data.cpython-310.pyc filter=lfs diff=lfs merge=lfs -text
161
+ .venv/Lib/site-packages/pip/_vendor/pkg_resources/__pycache__/__init__.cpython-310.pyc filter=lfs diff=lfs merge=lfs -text
162
+ .venv/Lib/site-packages/pip/_vendor/rich/__pycache__/_emoji_codes.cpython-310.pyc filter=lfs diff=lfs merge=lfs -text
163
+ .venv/Lib/site-packages/pkg_resources/__pycache__/__init__.cpython-310.pyc filter=lfs diff=lfs merge=lfs -text
164
+ .venv/Lib/site-packages/pkg_resources/_vendor/more_itertools/__pycache__/more.cpython-310.pyc filter=lfs diff=lfs merge=lfs -text
165
+ .venv/Lib/site-packages/pkg_resources/_vendor/pyparsing/__pycache__/core.cpython-310.pyc filter=lfs diff=lfs merge=lfs -text
166
+ .venv/Lib/site-packages/pydantic_core/__pycache__/core_schema.cpython-310.pyc filter=lfs diff=lfs merge=lfs -text
167
+ .venv/Lib/site-packages/pydantic_core/_pydantic_core.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
168
+ .venv/Lib/site-packages/pygments/lexers/__pycache__/lisp.cpython-310.pyc filter=lfs diff=lfs merge=lfs -text
169
+ .venv/Lib/site-packages/rich/__pycache__/_emoji_codes.cpython-310.pyc filter=lfs diff=lfs merge=lfs -text
170
+ .venv/Lib/site-packages/setuptools/_vendor/more_itertools/__pycache__/more.cpython-310.pyc filter=lfs diff=lfs merge=lfs -text
171
+ .venv/Lib/site-packages/setuptools/_vendor/pyparsing/__pycache__/core.cpython-310.pyc filter=lfs diff=lfs merge=lfs -text
172
+ .venv/Lib/site-packages/setuptools/cli-arm64.exe filter=lfs diff=lfs merge=lfs -text
173
+ .venv/Lib/site-packages/setuptools/gui-arm64.exe filter=lfs diff=lfs merge=lfs -text
174
+ .venv/Lib/site-packages/yaml/_yaml.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
175
+ .venv/Scripts/f2py.exe filter=lfs diff=lfs merge=lfs -text
176
+ .venv/Scripts/fastapi.exe filter=lfs diff=lfs merge=lfs -text
177
+ .venv/Scripts/gradio.exe filter=lfs diff=lfs merge=lfs -text
178
+ .venv/Scripts/hf.exe filter=lfs diff=lfs merge=lfs -text
179
+ .venv/Scripts/httpx.exe filter=lfs diff=lfs merge=lfs -text
180
+ .venv/Scripts/hypothesis.exe filter=lfs diff=lfs merge=lfs -text
181
+ .venv/Scripts/markdown-it.exe filter=lfs diff=lfs merge=lfs -text
182
+ .venv/Scripts/normalizer.exe filter=lfs diff=lfs merge=lfs -text
183
+ .venv/Scripts/numpy-config.exe filter=lfs diff=lfs merge=lfs -text
184
+ .venv/Scripts/pip.exe filter=lfs diff=lfs merge=lfs -text
185
+ .venv/Scripts/pip3.10.exe filter=lfs diff=lfs merge=lfs -text
186
+ .venv/Scripts/pip3.exe filter=lfs diff=lfs merge=lfs -text
187
+ .venv/Scripts/py.test.exe filter=lfs diff=lfs merge=lfs -text
188
+ .venv/Scripts/pygmentize.exe filter=lfs diff=lfs merge=lfs -text
189
+ .venv/Scripts/pytest.exe filter=lfs diff=lfs merge=lfs -text
190
+ .venv/Scripts/python.exe filter=lfs diff=lfs merge=lfs -text
191
+ .venv/Scripts/pythonw.exe filter=lfs diff=lfs merge=lfs -text
192
+ .venv/Scripts/tiny-agents.exe filter=lfs diff=lfs merge=lfs -text
193
+ .venv/Scripts/tqdm.exe filter=lfs diff=lfs merge=lfs -text
194
+ .venv/Scripts/typer.exe filter=lfs diff=lfs merge=lfs -text
195
+ .venv/Scripts/upload_theme.exe filter=lfs diff=lfs merge=lfs -text
196
+ .venv/Scripts/uvicorn.exe filter=lfs diff=lfs merge=lfs -text
197
+ background_solar.jpg filter=lfs diff=lfs merge=lfs -text
198
+ src/ui_img/20251219_000311_1024_094335193.jpg filter=lfs diff=lfs merge=lfs -text
199
+ src/ui_img/20251219_000311_1024_304211171.jpg filter=lfs diff=lfs merge=lfs -text
200
+ src/ui_img/20251219_001633_1024_211193171.jpg filter=lfs diff=lfs merge=lfs -text
201
+ src/ui_img/20251220_000035_1024_0211.jpg filter=lfs diff=lfs merge=lfs -text
202
+ src/ui_img/20251220_000114_1024_0335.jpg filter=lfs diff=lfs merge=lfs -text
203
+ src/ui_img/20251220_000151_1024_1600.jpg filter=lfs diff=lfs merge=lfs -text
204
+ src/ui_img/20251220_000317_1024_1700.jpg filter=lfs diff=lfs merge=lfs -text
205
+ src/ui_img/20251220_000600_1024_0094.jpg filter=lfs diff=lfs merge=lfs -text
206
+ src/ui_img/20251220_000644_1024_0131.jpg filter=lfs diff=lfs merge=lfs -text
207
+ src/ui_img/20251220_000658_1024_0171.jpg filter=lfs diff=lfs merge=lfs -text
208
+ src/ui_img/20251220_000753_1024_0193.jpg filter=lfs diff=lfs merge=lfs -text
209
+ src/ui_img/20251220_000854_1024_0304.jpg filter=lfs diff=lfs merge=lfs -text
210
+ src/ui_img/background.png filter=lfs diff=lfs merge=lfs -text
211
+ test_data/2025/12/17/20251217_001959_4096_0211.jpg filter=lfs diff=lfs merge=lfs -text
.kiro/specs/nasa-solar-image-downloader/design.md ADDED
@@ -0,0 +1,231 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Design Document
2
+
3
+ ## Overview
4
+
5
+ The NASA Solar Image Downloader is a Python application that automatically downloads high-resolution solar images from NASA's Solar Dynamics Observatory (SDO) and provides a video-like interface for viewing image sequences. The system operates continuously, checking for new images every 5 minutes and maintaining a local repository of the last month's solar data.
6
+
7
+ The application consists of three main components: an image downloader that interfaces with NASA's web repository, a local storage manager that organizes images by date, and a video interface that allows users to view image sequences with playback controls.
8
+
9
+ ## Architecture
10
+
11
+ The system follows a modular architecture with clear separation of concerns:
12
+
13
+ ```
14
+ ┌─────────────────┐ ┌─────────────────┐ ┌─────────────────┐
15
+ │ Video UI │ │ Image Manager │ │ NASA Downloader │
16
+ │ │ │ │ │ │
17
+ │ - Playback │◄──►│ - File Storage │◄──►│ - URL Generation│
18
+ │ - Navigation │ │ - Organization │ │ - HTTP Requests │
19
+ │ - Controls │ │ - Validation │ │ - Error Handling│
20
+ └─────────────────┘ └─────────────────┘ └─────────────────┘
21
+ │ │ │
22
+ └───────────────────────┼───────────────────────┘
23
+
24
+ ┌─────────────────┐
25
+ │ Scheduler │
26
+ │ │
27
+ │ - 5min Timer │
28
+ │ - Monitoring │
29
+ │ - Coordination │
30
+ └─────────────────┘
31
+ ```
32
+
33
+ ## Components and Interfaces
34
+
35
+ ### NASA Downloader Component
36
+ - **URLGenerator**: Constructs NASA SDO URLs based on date and time patterns
37
+ - **ImageFetcher**: Downloads images using HTTP requests with retry logic
38
+ - **ValidationService**: Verifies image URLs and file integrity
39
+
40
+ ### Image Manager Component
41
+ - **StorageOrganizer**: Creates and manages date-based folder structure
42
+ - **FileValidator**: Checks for duplicate files and validates image integrity
43
+ - **MetadataTracker**: Maintains records of downloaded images and timestamps
44
+
45
+ ### Video Interface Component
46
+ - **ImageSequencer**: Loads and orders images for playback
47
+ - **PlaybackController**: Manages play/pause/stop/speed controls
48
+ - **NavigationHandler**: Provides frame-by-frame stepping and seeking
49
+
50
+ ### Scheduler Component
51
+ - **MonitoringLoop**: Executes 5-minute check cycles
52
+ - **TaskCoordinator**: Manages download tasks and UI updates
53
+ - **StatusReporter**: Logs monitoring activity and download results
54
+
55
+ ## Data Models
56
+
57
+ ### Image Metadata
58
+ ```python
59
+ @dataclass
60
+ class ImageMetadata:
61
+ date: datetime
62
+ time_sequence: str # HHMMSS format
63
+ filename: str
64
+ local_path: Path
65
+ file_size: int
66
+ download_timestamp: datetime
67
+ url: str
68
+ ```
69
+
70
+ ### Download Task
71
+ ```python
72
+ @dataclass
73
+ class DownloadTask:
74
+ url: str
75
+ target_path: Path
76
+ retry_count: int
77
+ status: TaskStatus # PENDING, DOWNLOADING, COMPLETED, FAILED
78
+ error_message: Optional[str]
79
+ ```
80
+
81
+ ### Playback State
82
+ ```python
83
+ @dataclass
84
+ class PlaybackState:
85
+ current_frame: int
86
+ total_frames: int
87
+ is_playing: bool
88
+ playback_speed: float # frames per second
89
+ selected_date_range: Tuple[datetime, datetime]
90
+ ```
91
+ ## Correctness Properties
92
+
93
+ *A property is a characteristic or behavior that should hold true across all valid executions of a system-essentially, a formal statement about what the system should do. Properties serve as the bridge between human-readable specifications and machine-verifiable correctness guarantees.*
94
+
95
+ ### Property Reflection
96
+
97
+ After reviewing all testable properties from the prework analysis, several redundancies were identified:
98
+ - Properties 1.2 and 2.1 both test date path structure creation - consolidated into Property 2
99
+ - Properties related to error logging (1.3, 4.3, 5.2) share similar validation patterns - kept separate as they test different error scenarios
100
+ - Download completion reporting (1.4) and incremental downloading (4.2) test different aspects and remain separate
101
+
102
+ ### Property 1: Last month date range validation
103
+ *For any* system startup date, the Image_Downloader should generate URLs only for dates within the last 30 days and with "_4096_0211" suffix
104
+ **Validates: Requirements 1.1**
105
+
106
+ ### Property 2: Date path structure consistency
107
+ *For any* valid date, the Local_Storage should create folder structure following "data/YYYY/MM/DD/" pattern
108
+ **Validates: Requirements 1.2, 2.1**
109
+
110
+ ### Property 3: Error resilience during downloads
111
+ *For any* download failure, the Image_Downloader should log the error and continue processing remaining downloads without stopping
112
+ **Validates: Requirements 1.3**
113
+
114
+ ### Property 4: Download count accuracy
115
+ *For any* completed download session, the reported count should equal the actual number of successfully downloaded files
116
+ **Validates: Requirements 1.4**
117
+
118
+ ### Property 5: URL validation consistency
119
+ *For any* generated NASA SDO URL, the validation should accept well-formed URLs and reject malformed ones consistently
120
+ **Validates: Requirements 1.5**
121
+
122
+ ### Property 6: Filename preservation
123
+ *For any* downloaded image, the local filename should exactly match the original NASA filename format
124
+ **Validates: Requirements 2.2**
125
+
126
+ ### Property 7: Duplicate detection
127
+ *For any* image that already exists locally, attempting to download it again should skip the download and leave the existing file unchanged
128
+ **Validates: Requirements 2.3**
129
+
130
+ ### Property 8: File integrity verification
131
+ *For any* downloaded image, the local file size should match the expected size from the HTTP response headers
132
+ **Validates: Requirements 2.4**
133
+
134
+ ### Property 9: Time sequence ordering
135
+ *For any* collection of images with timestamps, the Video_Interface should display them in chronological order based on their time sequence
136
+ **Validates: Requirements 3.1**
137
+
138
+ ### Property 10: Frame navigation consistency
139
+ *For any* current frame position, stepping forward then backward should return to the original frame
140
+ **Validates: Requirements 3.3**
141
+
142
+ ### Property 11: Metadata display accuracy
143
+ *For any* currently displayed image, the shown timestamp and position should match the actual image metadata
144
+ **Validates: Requirements 3.4**
145
+
146
+ ### Property 12: Missing image graceful handling
147
+ *For any* image sequence with gaps, the Video_Interface should continue functioning without crashes when encountering missing files
148
+ **Validates: Requirements 3.5**
149
+
150
+ ### Property 13: Incremental download behavior
151
+ *For any* monitoring check, only images not already present in Local_Storage should be added to the download queue
152
+ **Validates: Requirements 4.2**
153
+
154
+ ### Property 14: Monitoring cycle logging
155
+ *For any* 5-minute check cycle, appropriate log entries should be created documenting the check time and results
156
+ **Validates: Requirements 4.3**
157
+
158
+ ### Property 15: Empty check handling
159
+ *For any* monitoring cycle that finds no new images, the system should continue to the next cycle without errors
160
+ **Validates: Requirements 4.5**
161
+
162
+ ### Property 16: Exponential backoff retry pattern
163
+ *For any* network timeout, subsequent retry attempts should follow exponential backoff timing (1s, 2s, 4s, 8s, etc.)
164
+ **Validates: Requirements 5.1**
165
+
166
+ ### Property 17: HTTP error logging and continuation
167
+ *For any* HTTP error response, the specific error code should be logged and processing should continue with remaining requests
168
+ **Validates: Requirements 5.2**
169
+
170
+ ### Property 18: Failed request queuing
171
+ *For any* server unavailability scenario, failed download requests should be added to a retry queue for later processing
172
+ **Validates: Requirements 5.3**
173
+
174
+ ### Property 19: Rate limiting compliance
175
+ *For any* sequence of download requests, the time interval between requests should respect the configured rate limit
176
+ **Validates: Requirements 5.4**
177
+
178
+ ### Property 20: Connection recovery resumption
179
+ *For any* queued downloads when connection is restored, all pending downloads should automatically resume processing
180
+ **Validates: Requirements 5.5**
181
+
182
+ ## Error Handling
183
+
184
+ The system implements comprehensive error handling across all components:
185
+
186
+ ### Network Error Handling
187
+ - **Connection timeouts**: Exponential backoff retry with maximum 5 attempts
188
+ - **HTTP errors**: Log specific error codes (404, 500, etc.) and continue processing
189
+ - **Rate limiting**: Implement delays between requests to respect NASA server limits
190
+ - **Server unavailability**: Queue failed requests for automatic retry when service resumes
191
+
192
+ ### File System Error Handling
193
+ - **Disk space**: Check available space before downloads and notify user if insufficient
194
+ - **Permission errors**: Handle read/write permission issues with clear error messages
195
+ - **Corrupted files**: Validate file integrity and re-download if corruption detected
196
+ - **Missing directories**: Automatically create required folder structure
197
+
198
+ ### UI Error Handling
199
+ - **Missing images**: Display placeholder or skip gracefully in video sequence
200
+ - **Invalid date ranges**: Validate user input and provide helpful error messages
201
+ - **Playback errors**: Handle corrupted or unreadable image files without crashing
202
+ - **Memory issues**: Implement image caching and cleanup for large sequences
203
+
204
+ ## Testing Strategy
205
+
206
+ The testing approach combines unit testing for specific functionality with property-based testing for universal behaviors:
207
+
208
+ ### Unit Testing Framework
209
+ - **Framework**: pytest for Python unit testing
210
+ - **Coverage**: Focus on specific examples, edge cases, and integration points
211
+ - **Mock objects**: Use unittest.mock for external dependencies (NASA servers, file system)
212
+ - **Test data**: Create sample image files and mock HTTP responses for testing
213
+
214
+ ### Property-Based Testing Framework
215
+ - **Framework**: Hypothesis for Python property-based testing
216
+ - **Configuration**: Minimum 100 iterations per property test to ensure thorough coverage
217
+ - **Generators**: Custom generators for dates, URLs, file paths, and image metadata
218
+ - **Shrinking**: Leverage Hypothesis shrinking to find minimal failing examples
219
+
220
+ ### Test Organization
221
+ - Unit tests verify specific examples and integration behavior
222
+ - Property tests verify universal correctness properties across all inputs
223
+ - Each property-based test tagged with format: **Feature: nasa-solar-image-downloader, Property {number}: {property_text}**
224
+ - Tests organized by component (downloader, storage, ui, scheduler)
225
+ - Continuous integration runs both unit and property tests on every commit
226
+
227
+ ### Test Data Management
228
+ - Mock NASA server responses for reliable testing
229
+ - Generate test images with known metadata for validation
230
+ - Create temporary directories for file system testing
231
+ - Use time mocking for scheduler and timing-dependent tests
.kiro/specs/nasa-solar-image-downloader/requirements.md ADDED
@@ -0,0 +1,80 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Requirements Document
2
+
3
+ ## Introduction
4
+
5
+ A Python data mining application that automatically downloads NASA Solar Dynamics Observatory (SDO) images from the web and provides a user interface for viewing sequences of images like a video. The system focuses on downloading high-resolution (4096) images with specific instrument code (0211) and organizing them by date for sequential viewing.
6
+
7
+ ## Glossary
8
+
9
+ - **NASA_SDO_System**: The NASA Solar Dynamics Observatory image repository system
10
+ - **Image_Downloader**: The component responsible for fetching images from NASA servers
11
+ - **Local_Storage**: The local "data" folder where downloaded images are stored
12
+ - **Video_Interface**: The user interface component that displays image sequences
13
+ - **Date_Path**: The hierarchical folder structure based on date (YYYY/MM/DD format)
14
+ - **Time_Sequence**: The sequential numbering system used by NASA for images (HHMMSS format)
15
+ - **Resolution_Code**: The image resolution identifier (4096 for high resolution)
16
+ - **Instrument_Code**: The specific NASA instrument identifier (0211 for AIA 211 Angstrom)
17
+ - **Monitoring_Cycle**: The 5-minute interval system for checking new images
18
+ - **Last_Month_Period**: The 30-day period from current date backwards for initial data collection
19
+
20
+ ## Requirements
21
+
22
+ ### Requirement 1
23
+
24
+ **User Story:** As a researcher, I want to automatically download NASA solar images from the last month, so that I can analyze recent solar activity patterns.
25
+
26
+ #### Acceptance Criteria
27
+
28
+ 1. WHEN the system starts, THE Image_Downloader SHALL fetch all available images with "_4096_0211" suffix from the last 30 days from the NASA_SDO_System
29
+ 2. WHEN downloading images, THE Image_Downloader SHALL create the appropriate Date_Path structure in Local_Storage
30
+ 3. WHEN an image download fails, THE Image_Downloader SHALL log the error and continue with remaining downloads
31
+ 4. WHEN all downloads complete, THE Image_Downloader SHALL report the total number of successfully downloaded images
32
+ 5. WHERE network connectivity is available, THE Image_Downloader SHALL validate image URLs before attempting downloads
33
+
34
+ ### Requirement 2
35
+
36
+ **User Story:** As a user, I want images to be organized in a local data folder by date, so that I can easily locate and manage downloaded content.
37
+
38
+ #### Acceptance Criteria
39
+
40
+ 1. THE Local_Storage SHALL organize images using the Date_Path structure (data/YYYY/MM/DD/)
41
+ 2. WHEN storing images, THE Local_Storage SHALL preserve the original NASA filename format
42
+ 3. WHEN duplicate images are detected, THE Local_Storage SHALL skip re-downloading existing files
43
+ 4. THE Local_Storage SHALL maintain file integrity by verifying downloaded image sizes
44
+ 5. WHEN storage space is insufficient, THE Local_Storage SHALL notify the user before attempting downloads
45
+
46
+ ### Requirement 3
47
+
48
+ **User Story:** As a researcher, I want to view sequences of downloaded images like a video, so that I can observe solar phenomena over time.
49
+
50
+ #### Acceptance Criteria
51
+
52
+ 1. WHEN a user selects a date range, THE Video_Interface SHALL display images in Time_Sequence order
53
+ 2. WHEN playing the sequence, THE Video_Interface SHALL provide playback controls (play, pause, stop, speed adjustment)
54
+ 3. WHEN navigating the sequence, THE Video_Interface SHALL allow frame-by-frame stepping forward and backward
55
+ 4. THE Video_Interface SHALL display current image timestamp and sequence position
56
+ 5. WHERE images are missing from the sequence, THE Video_Interface SHALL handle gaps gracefully without crashing
57
+
58
+ ### Requirement 4
59
+
60
+ **User Story:** As a user, I want the system to automatically check for new images every 5 minutes, so that I always have the latest solar data available.
61
+
62
+ #### Acceptance Criteria
63
+
64
+ 1. THE Image_Downloader SHALL check the NASA_SDO_System for new images every 5 minutes
65
+ 2. WHEN new images are detected, THE Image_Downloader SHALL download only the images not already present in Local_Storage
66
+ 3. WHEN the 5-minute check cycle runs, THE Image_Downloader SHALL log the check time and any new images found
67
+ 4. THE Image_Downloader SHALL maintain a continuous monitoring loop that runs until manually stopped
68
+ 5. WHERE no new images are found during a check, THE Image_Downloader SHALL wait for the next 5-minute interval without error
69
+
70
+ ### Requirement 5
71
+
72
+ **User Story:** As a system administrator, I want the application to handle network errors gracefully, so that the download process remains robust and reliable.
73
+
74
+ #### Acceptance Criteria
75
+
76
+ 1. WHEN network timeouts occur, THE Image_Downloader SHALL implement exponential backoff retry logic
77
+ 2. WHEN HTTP errors are encountered, THE Image_Downloader SHALL log specific error codes and continue processing
78
+ 3. IF the NASA server is unavailable, THEN THE Image_Downloader SHALL queue failed requests for later retry
79
+ 4. THE Image_Downloader SHALL respect rate limiting to avoid overwhelming the NASA_SDO_System
80
+ 5. WHEN connection is restored after failure, THE Image_Downloader SHALL automatically resume pending downloads
.kiro/specs/nasa-solar-image-downloader/tasks.md ADDED
@@ -0,0 +1,182 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Implementation Plan
2
+
3
+ - [x] 1. Set up project structure and core interfaces
4
+
5
+
6
+ - Create directory structure for components (downloader, storage, ui, scheduler)
7
+ - Set up Python project with requirements.txt and main entry point
8
+ - Define core data models (ImageMetadata, DownloadTask, PlaybackState)
9
+ - Install and configure testing frameworks (pytest, hypothesis)
10
+ - _Requirements: 1.1, 2.1, 3.1, 4.1_
11
+
12
+ - [ ]* 1.1 Write property test for date range validation
13
+ - **Property 1: Last month date range validation**
14
+ - **Validates: Requirements 1.1**
15
+
16
+
17
+
18
+
19
+ - [ ] 2. Implement NASA URL generation and validation
20
+ - Create URLGenerator class to construct NASA SDO URLs from dates
21
+ - Implement URL validation logic for NASA image format
22
+ - Add date range calculation for last 30 days
23
+ - _Requirements: 1.1, 1.5_
24
+
25
+
26
+
27
+
28
+ - [ ]* 2.1 Write property test for URL validation consistency
29
+ - **Property 5: URL validation consistency**
30
+ - **Validates: Requirements 1.5**
31
+
32
+ - [ ] 3. Implement local storage and file organization
33
+ - Create StorageOrganizer class for date-based folder structure
34
+ - Implement file validation and duplicate detection
35
+ - Add filename preservation logic
36
+ - _Requirements: 2.1, 2.2, 2.3, 2.4_
37
+
38
+ - [ ]* 3.1 Write property test for date path structure
39
+ - **Property 2: Date path structure consistency**
40
+ - **Validates: Requirements 1.2, 2.1**
41
+
42
+ - [x]* 3.2 Write property test for filename preservation
43
+
44
+
45
+ - **Property 6: Filename preservation**
46
+ - **Validates: Requirements 2.2**
47
+
48
+ - [ ]* 3.3 Write property test for duplicate detection
49
+ - **Property 7: Duplicate detection**
50
+ - **Validates: Requirements 2.3**
51
+
52
+ - [ ] 4. Implement HTTP downloader with error handling
53
+ - Create ImageFetcher class with requests library
54
+ - Implement exponential backoff retry logic
55
+ - Add HTTP error handling and logging
56
+ - Implement rate limiting to respect NASA servers
57
+ - _Requirements: 1.3, 1.4, 5.1, 5.2, 5.4_
58
+
59
+ - [ ]* 4.1 Write property test for error resilience
60
+ - **Property 3: Error resilience during downloads**
61
+ - **Validates: Requirements 1.3**
62
+
63
+ - [ ]* 4.2 Write property test for download count accuracy
64
+ - **Property 4: Download count accuracy**
65
+ - **Validates: Requirements 1.4**
66
+
67
+ - [ ]* 4.3 Write property test for exponential backoff
68
+ - **Property 16: Exponential backoff retry pattern**
69
+ - **Validates: Requirements 5.1**
70
+
71
+
72
+
73
+
74
+
75
+ - [ ]* 4.4 Write property test for HTTP error handling
76
+ - **Property 17: HTTP error logging and continuation**
77
+ - **Validates: Requirements 5.2**
78
+
79
+ - [x]* 4.5 Write property test for rate limiting
80
+
81
+
82
+ - **Property 19: Rate limiting compliance**
83
+
84
+
85
+
86
+
87
+ - **Validates: Requirements 5.4**
88
+
89
+ - [ ] 5. Implement file integrity and validation
90
+ - Add file size verification during downloads
91
+ - Implement file integrity checking
92
+ - Create validation service for downloaded images
93
+ - _Requirements: 2.4_
94
+
95
+ - [ ]* 5.1 Write property test for file integrity
96
+ - **Property 8: File integrity verification**
97
+ - **Validates: Requirements 2.4**
98
+
99
+ - [ ] 6. Checkpoint - Ensure all tests pass
100
+ - Ensure all tests pass, ask the user if questions arise.
101
+
102
+ - [ ] 7. Implement monitoring scheduler
103
+ - Create MonitoringLoop class with 5-minute intervals
104
+ - Implement task coordination and status reporting
105
+ - Add logging for monitoring cycles
106
+ - _Requirements: 4.1, 4.2, 4.3, 4.5_
107
+
108
+ - [ ]* 7.1 Write property test for incremental downloads
109
+ - **Property 13: Incremental download behavior**
110
+ - **Validates: Requirements 4.2**
111
+
112
+ - [ ]* 7.2 Write property test for monitoring logging
113
+ - **Property 14: Monitoring cycle logging**
114
+ - **Validates: Requirements 4.3**
115
+
116
+ - [ ]* 7.3 Write property test for empty check handling
117
+ - **Property 15: Empty check handling**
118
+ - **Validates: Requirements 4.5**
119
+
120
+ - [ ] 8. Implement request queuing and recovery
121
+ - Create failed request queue for server unavailability
122
+ - Implement automatic retry when connection restored
123
+ - Add queue management and persistence
124
+ - _Requirements: 5.3, 5.5_
125
+
126
+ - [ ]* 8.1 Write property test for request queuing
127
+ - **Property 18: Failed request queuing**
128
+ - **Validates: Requirements 5.3**
129
+
130
+ - [ ]* 8.2 Write property test for connection recovery
131
+ - **Property 20: Connection recovery resumption**
132
+ - **Validates: Requirements 5.5**
133
+
134
+ - [ ] 9. Implement image sequence management
135
+ - Create ImageSequencer class to load and order images
136
+ - Implement time sequence sorting logic
137
+ - Add metadata extraction from filenames
138
+ - _Requirements: 3.1, 3.4_
139
+
140
+ - [ ]* 9.1 Write property test for time sequence ordering
141
+ - **Property 9: Time sequence ordering**
142
+ - **Validates: Requirements 3.1**
143
+
144
+ - [ ]* 9.2 Write property test for metadata display
145
+ - **Property 11: Metadata display accuracy**
146
+ - **Validates: Requirements 3.4**
147
+
148
+ - [ ] 10. Implement video interface UI
149
+ - Create main window with tkinter or pygame
150
+ - Implement image display and playback controls
151
+ - Add frame navigation (play, pause, stop, step)
152
+ - Add speed adjustment controls
153
+ - _Requirements: 3.2, 3.3_
154
+
155
+ - [ ]* 10.1 Write property test for frame navigation
156
+ - **Property 10: Frame navigation consistency**
157
+ - **Validates: Requirements 3.3**
158
+
159
+ - [ ] 11. Implement graceful error handling in UI
160
+ - Add missing image placeholder handling
161
+ - Implement error recovery for corrupted files
162
+ - Add user notifications for errors
163
+ - _Requirements: 3.5_
164
+
165
+ - [ ]* 11.1 Write property test for missing image handling
166
+ - **Property 12: Missing image graceful handling**
167
+ - **Validates: Requirements 3.5**
168
+
169
+ - [ ] 12. Integrate all components
170
+ - Create main application entry point
171
+ - Wire together downloader, storage, scheduler, and UI
172
+ - Implement application lifecycle management
173
+ - Add configuration management
174
+ - _Requirements: All requirements integration_
175
+
176
+ - [ ]* 12.1 Write integration tests for complete workflow
177
+ - Test end-to-end download and display workflow
178
+ - Test monitoring cycle integration
179
+ - Test error recovery across components
180
+
181
+ - [ ] 13. Final checkpoint - Ensure all tests pass
182
+ - Ensure all tests pass, ask the user if questions arise.
.pytest_cache/.gitignore ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ # Created by pytest automatically.
2
+ *
.pytest_cache/CACHEDIR.TAG ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ Signature: 8a477f597d28d172789f06886806bc55
2
+ # This file is a cache directory tag created by pytest.
3
+ # For information about cache directory tags, see:
4
+ # https://bford.info/cachedir/spec.html
.pytest_cache/README.md ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ # pytest cache directory #
2
+
3
+ This directory contains data from the pytest's cache plugin,
4
+ which provides the `--lf` and `--ff` options, as well as the `cache` fixture.
5
+
6
+ **Do not** commit this to version control.
7
+
8
+ See [the docs](https://docs.pytest.org/en/stable/how-to/cache.html) for more information.
.pytest_cache/v/cache/lastfailed ADDED
@@ -0,0 +1 @@
 
 
1
+ {}
.pytest_cache/v/cache/nodeids ADDED
@@ -0,0 +1,85 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ [
2
+ "tests/test_image_fetcher.py::TestDownloadManager::test_download_and_save_download_failure",
3
+ "tests/test_image_fetcher.py::TestDownloadManager::test_download_and_save_duplicate_skip",
4
+ "tests/test_image_fetcher.py::TestDownloadManager::test_download_and_save_integrity_failure",
5
+ "tests/test_image_fetcher.py::TestDownloadManager::test_download_and_save_success",
6
+ "tests/test_image_fetcher.py::TestDownloadManager::test_reset_counters",
7
+ "tests/test_image_fetcher.py::TestImageFetcher::test_check_image_exists_false",
8
+ "tests/test_image_fetcher.py::TestImageFetcher::test_check_image_exists_true",
9
+ "tests/test_image_fetcher.py::TestImageFetcher::test_download_image_404",
10
+ "tests/test_image_fetcher.py::TestImageFetcher::test_download_image_http_error_with_retry",
11
+ "tests/test_image_fetcher.py::TestImageFetcher::test_download_image_max_retries_exceeded",
12
+ "tests/test_image_fetcher.py::TestImageFetcher::test_download_image_success",
13
+ "tests/test_image_fetcher.py::TestImageFetcher::test_download_image_timeout_retry",
14
+ "tests/test_image_fetcher.py::TestImageFetcher::test_exponential_backoff",
15
+ "tests/test_image_fetcher.py::TestImageFetcher::test_get_image_size",
16
+ "tests/test_image_fetcher.py::TestImageFetcher::test_rate_limiting",
17
+ "tests/test_models.py::test_download_task_defaults",
18
+ "tests/test_models.py::test_image_metadata_creation",
19
+ "tests/test_models.py::test_playback_state_defaults",
20
+ "tests/test_monitoring_loop.py::TestMonitoringLoop::test_download_new_images_failure",
21
+ "tests/test_monitoring_loop.py::TestMonitoringLoop::test_download_new_images_success",
22
+ "tests/test_monitoring_loop.py::TestMonitoringLoop::test_filter_new_images_all_new",
23
+ "tests/test_monitoring_loop.py::TestMonitoringLoop::test_filter_new_images_invalid_url",
24
+ "tests/test_monitoring_loop.py::TestMonitoringLoop::test_filter_new_images_some_exist",
25
+ "tests/test_monitoring_loop.py::TestMonitoringLoop::test_force_check_not_running",
26
+ "tests/test_monitoring_loop.py::TestMonitoringLoop::test_get_monitoring_range",
27
+ "tests/test_monitoring_loop.py::TestMonitoringLoop::test_get_status_initial",
28
+ "tests/test_monitoring_loop.py::TestMonitoringLoop::test_initialization",
29
+ "tests/test_monitoring_loop.py::TestMonitoringLoop::test_set_monitoring_range",
30
+ "tests/test_monitoring_loop.py::TestMonitoringLoop::test_set_monitoring_range_invalid",
31
+ "tests/test_monitoring_loop.py::TestMonitoringLoop::test_start_monitoring",
32
+ "tests/test_monitoring_loop.py::TestMonitoringLoop::test_stop_monitoring_not_running",
33
+ "tests/test_monitoring_loop.py::TestStatusReporter::test_initialization",
34
+ "tests/test_monitoring_loop.py::TestStatusReporter::test_log_periodic_status",
35
+ "tests/test_monitoring_loop.py::TestStatusReporter::test_print_status",
36
+ "tests/test_monitoring_loop.py::TestStatusReporter::test_print_status_no_last_check",
37
+ "tests/test_monitoring_loop.py::TestTaskCoordinator::test_initialization",
38
+ "tests/test_monitoring_loop.py::TestTaskCoordinator::test_on_check_complete",
39
+ "tests/test_monitoring_loop.py::TestTaskCoordinator::test_on_check_start",
40
+ "tests/test_monitoring_loop.py::TestTaskCoordinator::test_on_new_images_found",
41
+ "tests/test_storage_organizer.py::TestStorageOrganizer::test_check_sufficient_space",
42
+ "tests/test_storage_organizer.py::TestStorageOrganizer::test_cleanup_corrupted_files",
43
+ "tests/test_storage_organizer.py::TestStorageOrganizer::test_create_date_structure",
44
+ "tests/test_storage_organizer.py::TestStorageOrganizer::test_file_exists_false",
45
+ "tests/test_storage_organizer.py::TestStorageOrganizer::test_get_date_path",
46
+ "tests/test_storage_organizer.py::TestStorageOrganizer::test_get_file_size",
47
+ "tests/test_storage_organizer.py::TestStorageOrganizer::test_get_image_metadata",
48
+ "tests/test_storage_organizer.py::TestStorageOrganizer::test_get_local_path",
49
+ "tests/test_storage_organizer.py::TestStorageOrganizer::test_list_local_images_empty",
50
+ "tests/test_storage_organizer.py::TestStorageOrganizer::test_list_local_images_with_files",
51
+ "tests/test_storage_organizer.py::TestStorageOrganizer::test_save_and_check_image",
52
+ "tests/test_storage_organizer.py::TestStorageOrganizer::test_validate_file_integrity_failure",
53
+ "tests/test_storage_organizer.py::TestStorageOrganizer::test_validate_file_integrity_success",
54
+ "tests/test_url_generator.py::TestURLGenerator::test_construct_url",
55
+ "tests/test_url_generator.py::TestURLGenerator::test_extract_metadata_from_url",
56
+ "tests/test_url_generator.py::TestURLGenerator::test_extract_metadata_invalid_url",
57
+ "tests/test_url_generator.py::TestURLGenerator::test_generate_daily_urls",
58
+ "tests/test_url_generator.py::TestURLGenerator::test_generate_date_range_urls",
59
+ "tests/test_url_generator.py::TestURLGenerator::test_generate_date_range_urls_custom",
60
+ "tests/test_url_generator.py::TestURLGenerator::test_generate_date_range_urls_single_day",
61
+ "tests/test_url_generator.py::TestURLGenerator::test_generate_default_urls",
62
+ "tests/test_url_generator.py::TestURLGenerator::test_generate_last_month_urls",
63
+ "tests/test_url_generator.py::TestURLGenerator::test_last_month_date_range",
64
+ "tests/test_url_generator.py::TestURLGenerator::test_validate_url_invalid",
65
+ "tests/test_url_generator.py::TestURLGenerator::test_validate_url_valid",
66
+ "tests/test_validation_service.py::TestValidationService::test_calculate_file_hash",
67
+ "tests/test_validation_service.py::TestValidationService::test_calculate_file_hash_different_algorithms",
68
+ "tests/test_validation_service.py::TestValidationService::test_comprehensive_validation_empty_file",
69
+ "tests/test_validation_service.py::TestValidationService::test_comprehensive_validation_missing_file",
70
+ "tests/test_validation_service.py::TestValidationService::test_comprehensive_validation_multiple_errors",
71
+ "tests/test_validation_service.py::TestValidationService::test_comprehensive_validation_success",
72
+ "tests/test_validation_service.py::TestValidationService::test_get_image_info_corrupted_file",
73
+ "tests/test_validation_service.py::TestValidationService::test_get_image_info_success",
74
+ "tests/test_validation_service.py::TestValidationService::test_repair_corrupted_image_failure",
75
+ "tests/test_validation_service.py::TestValidationService::test_repair_corrupted_image_success",
76
+ "tests/test_validation_service.py::TestValidationService::test_validate_file_size_exact_match",
77
+ "tests/test_validation_service.py::TestValidationService::test_validate_file_size_outside_tolerance",
78
+ "tests/test_validation_service.py::TestValidationService::test_validate_file_size_within_tolerance",
79
+ "tests/test_validation_service.py::TestValidationService::test_validate_image_content_too_large",
80
+ "tests/test_validation_service.py::TestValidationService::test_validate_image_content_too_small",
81
+ "tests/test_validation_service.py::TestValidationService::test_validate_image_content_valid",
82
+ "tests/test_validation_service.py::TestValidationService::test_validate_image_format_corrupted_file",
83
+ "tests/test_validation_service.py::TestValidationService::test_validate_image_format_invalid_format",
84
+ "tests/test_validation_service.py::TestValidationService::test_validate_image_format_valid_jpeg"
85
+ ]
.venv/.gitignore ADDED
@@ -0,0 +1 @@
 
 
1
+ *
.venv/Lib/site-packages/PIL/AvifImagePlugin.py ADDED
@@ -0,0 +1,291 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from __future__ import annotations
2
+
3
+ import os
4
+ from io import BytesIO
5
+ from typing import IO
6
+
7
+ from . import ExifTags, Image, ImageFile
8
+
9
+ try:
10
+ from . import _avif
11
+
12
+ SUPPORTED = True
13
+ except ImportError:
14
+ SUPPORTED = False
15
+
16
+ # Decoder options as module globals, until there is a way to pass parameters
17
+ # to Image.open (see https://github.com/python-pillow/Pillow/issues/569)
18
+ DECODE_CODEC_CHOICE = "auto"
19
+ DEFAULT_MAX_THREADS = 0
20
+
21
+
22
+ def get_codec_version(codec_name: str) -> str | None:
23
+ versions = _avif.codec_versions()
24
+ for version in versions.split(", "):
25
+ if version.split(" [")[0] == codec_name:
26
+ return version.split(":")[-1].split(" ")[0]
27
+ return None
28
+
29
+
30
+ def _accept(prefix: bytes) -> bool | str:
31
+ if prefix[4:8] != b"ftyp":
32
+ return False
33
+ major_brand = prefix[8:12]
34
+ if major_brand in (
35
+ # coding brands
36
+ b"avif",
37
+ b"avis",
38
+ # We accept files with AVIF container brands; we can't yet know if
39
+ # the ftyp box has the correct compatible brands, but if it doesn't
40
+ # then the plugin will raise a SyntaxError which Pillow will catch
41
+ # before moving on to the next plugin that accepts the file.
42
+ #
43
+ # Also, because this file might not actually be an AVIF file, we
44
+ # don't raise an error if AVIF support isn't properly compiled.
45
+ b"mif1",
46
+ b"msf1",
47
+ ):
48
+ if not SUPPORTED:
49
+ return (
50
+ "image file could not be identified because AVIF support not installed"
51
+ )
52
+ return True
53
+ return False
54
+
55
+
56
+ def _get_default_max_threads() -> int:
57
+ if DEFAULT_MAX_THREADS:
58
+ return DEFAULT_MAX_THREADS
59
+ if hasattr(os, "sched_getaffinity"):
60
+ return len(os.sched_getaffinity(0))
61
+ else:
62
+ return os.cpu_count() or 1
63
+
64
+
65
+ class AvifImageFile(ImageFile.ImageFile):
66
+ format = "AVIF"
67
+ format_description = "AVIF image"
68
+ __frame = -1
69
+
70
+ def _open(self) -> None:
71
+ if not SUPPORTED:
72
+ msg = "image file could not be opened because AVIF support not installed"
73
+ raise SyntaxError(msg)
74
+
75
+ if DECODE_CODEC_CHOICE != "auto" and not _avif.decoder_codec_available(
76
+ DECODE_CODEC_CHOICE
77
+ ):
78
+ msg = "Invalid opening codec"
79
+ raise ValueError(msg)
80
+ self._decoder = _avif.AvifDecoder(
81
+ self.fp.read(),
82
+ DECODE_CODEC_CHOICE,
83
+ _get_default_max_threads(),
84
+ )
85
+
86
+ # Get info from decoder
87
+ self._size, self.n_frames, self._mode, icc, exif, exif_orientation, xmp = (
88
+ self._decoder.get_info()
89
+ )
90
+ self.is_animated = self.n_frames > 1
91
+
92
+ if icc:
93
+ self.info["icc_profile"] = icc
94
+ if xmp:
95
+ self.info["xmp"] = xmp
96
+
97
+ if exif_orientation != 1 or exif:
98
+ exif_data = Image.Exif()
99
+ if exif:
100
+ exif_data.load(exif)
101
+ original_orientation = exif_data.get(ExifTags.Base.Orientation, 1)
102
+ else:
103
+ original_orientation = 1
104
+ if exif_orientation != original_orientation:
105
+ exif_data[ExifTags.Base.Orientation] = exif_orientation
106
+ exif = exif_data.tobytes()
107
+ if exif:
108
+ self.info["exif"] = exif
109
+ self.seek(0)
110
+
111
+ def seek(self, frame: int) -> None:
112
+ if not self._seek_check(frame):
113
+ return
114
+
115
+ # Set tile
116
+ self.__frame = frame
117
+ self.tile = [ImageFile._Tile("raw", (0, 0) + self.size, 0, self.mode)]
118
+
119
+ def load(self) -> Image.core.PixelAccess | None:
120
+ if self.tile:
121
+ # We need to load the image data for this frame
122
+ data, timescale, pts_in_timescales, duration_in_timescales = (
123
+ self._decoder.get_frame(self.__frame)
124
+ )
125
+ self.info["timestamp"] = round(1000 * (pts_in_timescales / timescale))
126
+ self.info["duration"] = round(1000 * (duration_in_timescales / timescale))
127
+
128
+ if self.fp and self._exclusive_fp:
129
+ self.fp.close()
130
+ self.fp = BytesIO(data)
131
+
132
+ return super().load()
133
+
134
+ def load_seek(self, pos: int) -> None:
135
+ pass
136
+
137
+ def tell(self) -> int:
138
+ return self.__frame
139
+
140
+
141
+ def _save_all(im: Image.Image, fp: IO[bytes], filename: str | bytes) -> None:
142
+ _save(im, fp, filename, save_all=True)
143
+
144
+
145
+ def _save(
146
+ im: Image.Image, fp: IO[bytes], filename: str | bytes, save_all: bool = False
147
+ ) -> None:
148
+ info = im.encoderinfo.copy()
149
+ if save_all:
150
+ append_images = list(info.get("append_images", []))
151
+ else:
152
+ append_images = []
153
+
154
+ total = 0
155
+ for ims in [im] + append_images:
156
+ total += getattr(ims, "n_frames", 1)
157
+
158
+ quality = info.get("quality", 75)
159
+ if not isinstance(quality, int) or quality < 0 or quality > 100:
160
+ msg = "Invalid quality setting"
161
+ raise ValueError(msg)
162
+
163
+ duration = info.get("duration", 0)
164
+ subsampling = info.get("subsampling", "4:2:0")
165
+ speed = info.get("speed", 6)
166
+ max_threads = info.get("max_threads", _get_default_max_threads())
167
+ codec = info.get("codec", "auto")
168
+ if codec != "auto" and not _avif.encoder_codec_available(codec):
169
+ msg = "Invalid saving codec"
170
+ raise ValueError(msg)
171
+ range_ = info.get("range", "full")
172
+ tile_rows_log2 = info.get("tile_rows", 0)
173
+ tile_cols_log2 = info.get("tile_cols", 0)
174
+ alpha_premultiplied = bool(info.get("alpha_premultiplied", False))
175
+ autotiling = bool(info.get("autotiling", tile_rows_log2 == tile_cols_log2 == 0))
176
+
177
+ icc_profile = info.get("icc_profile", im.info.get("icc_profile"))
178
+ exif_orientation = 1
179
+ if exif := info.get("exif"):
180
+ if isinstance(exif, Image.Exif):
181
+ exif_data = exif
182
+ else:
183
+ exif_data = Image.Exif()
184
+ exif_data.load(exif)
185
+ if ExifTags.Base.Orientation in exif_data:
186
+ exif_orientation = exif_data.pop(ExifTags.Base.Orientation)
187
+ exif = exif_data.tobytes() if exif_data else b""
188
+ elif isinstance(exif, Image.Exif):
189
+ exif = exif_data.tobytes()
190
+
191
+ xmp = info.get("xmp")
192
+
193
+ if isinstance(xmp, str):
194
+ xmp = xmp.encode("utf-8")
195
+
196
+ advanced = info.get("advanced")
197
+ if advanced is not None:
198
+ if isinstance(advanced, dict):
199
+ advanced = advanced.items()
200
+ try:
201
+ advanced = tuple(advanced)
202
+ except TypeError:
203
+ invalid = True
204
+ else:
205
+ invalid = any(not isinstance(v, tuple) or len(v) != 2 for v in advanced)
206
+ if invalid:
207
+ msg = (
208
+ "advanced codec options must be a dict of key-value string "
209
+ "pairs or a series of key-value two-tuples"
210
+ )
211
+ raise ValueError(msg)
212
+
213
+ # Setup the AVIF encoder
214
+ enc = _avif.AvifEncoder(
215
+ im.size,
216
+ subsampling,
217
+ quality,
218
+ speed,
219
+ max_threads,
220
+ codec,
221
+ range_,
222
+ tile_rows_log2,
223
+ tile_cols_log2,
224
+ alpha_premultiplied,
225
+ autotiling,
226
+ icc_profile or b"",
227
+ exif or b"",
228
+ exif_orientation,
229
+ xmp or b"",
230
+ advanced,
231
+ )
232
+
233
+ # Add each frame
234
+ frame_idx = 0
235
+ frame_duration = 0
236
+ cur_idx = im.tell()
237
+ is_single_frame = total == 1
238
+ try:
239
+ for ims in [im] + append_images:
240
+ # Get number of frames in this image
241
+ nfr = getattr(ims, "n_frames", 1)
242
+
243
+ for idx in range(nfr):
244
+ ims.seek(idx)
245
+
246
+ # Make sure image mode is supported
247
+ frame = ims
248
+ rawmode = ims.mode
249
+ if ims.mode not in {"RGB", "RGBA"}:
250
+ rawmode = "RGBA" if ims.has_transparency_data else "RGB"
251
+ frame = ims.convert(rawmode)
252
+
253
+ # Update frame duration
254
+ if isinstance(duration, (list, tuple)):
255
+ frame_duration = duration[frame_idx]
256
+ else:
257
+ frame_duration = duration
258
+
259
+ # Append the frame to the animation encoder
260
+ enc.add(
261
+ frame.tobytes("raw", rawmode),
262
+ frame_duration,
263
+ frame.size,
264
+ rawmode,
265
+ is_single_frame,
266
+ )
267
+
268
+ # Update frame index
269
+ frame_idx += 1
270
+
271
+ if not save_all:
272
+ break
273
+
274
+ finally:
275
+ im.seek(cur_idx)
276
+
277
+ # Get the final output from the encoder
278
+ data = enc.finish()
279
+ if data is None:
280
+ msg = "cannot write file as AVIF (encoder returned None)"
281
+ raise OSError(msg)
282
+
283
+ fp.write(data)
284
+
285
+
286
+ Image.register_open(AvifImageFile.format, AvifImageFile, _accept)
287
+ if SUPPORTED:
288
+ Image.register_save(AvifImageFile.format, _save)
289
+ Image.register_save_all(AvifImageFile.format, _save_all)
290
+ Image.register_extensions(AvifImageFile.format, [".avif", ".avifs"])
291
+ Image.register_mime(AvifImageFile.format, "image/avif")
.venv/Lib/site-packages/PIL/BdfFontFile.py ADDED
@@ -0,0 +1,122 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #
2
+ # The Python Imaging Library
3
+ # $Id$
4
+ #
5
+ # bitmap distribution font (bdf) file parser
6
+ #
7
+ # history:
8
+ # 1996-05-16 fl created (as bdf2pil)
9
+ # 1997-08-25 fl converted to FontFile driver
10
+ # 2001-05-25 fl removed bogus __init__ call
11
+ # 2002-11-20 fl robustification (from Kevin Cazabon, Dmitry Vasiliev)
12
+ # 2003-04-22 fl more robustification (from Graham Dumpleton)
13
+ #
14
+ # Copyright (c) 1997-2003 by Secret Labs AB.
15
+ # Copyright (c) 1997-2003 by Fredrik Lundh.
16
+ #
17
+ # See the README file for information on usage and redistribution.
18
+ #
19
+
20
+ """
21
+ Parse X Bitmap Distribution Format (BDF)
22
+ """
23
+ from __future__ import annotations
24
+
25
+ from typing import BinaryIO
26
+
27
+ from . import FontFile, Image
28
+
29
+
30
+ def bdf_char(
31
+ f: BinaryIO,
32
+ ) -> (
33
+ tuple[
34
+ str,
35
+ int,
36
+ tuple[tuple[int, int], tuple[int, int, int, int], tuple[int, int, int, int]],
37
+ Image.Image,
38
+ ]
39
+ | None
40
+ ):
41
+ # skip to STARTCHAR
42
+ while True:
43
+ s = f.readline()
44
+ if not s:
45
+ return None
46
+ if s.startswith(b"STARTCHAR"):
47
+ break
48
+ id = s[9:].strip().decode("ascii")
49
+
50
+ # load symbol properties
51
+ props = {}
52
+ while True:
53
+ s = f.readline()
54
+ if not s or s.startswith(b"BITMAP"):
55
+ break
56
+ i = s.find(b" ")
57
+ props[s[:i].decode("ascii")] = s[i + 1 : -1].decode("ascii")
58
+
59
+ # load bitmap
60
+ bitmap = bytearray()
61
+ while True:
62
+ s = f.readline()
63
+ if not s or s.startswith(b"ENDCHAR"):
64
+ break
65
+ bitmap += s[:-1]
66
+
67
+ # The word BBX
68
+ # followed by the width in x (BBw), height in y (BBh),
69
+ # and x and y displacement (BBxoff0, BByoff0)
70
+ # of the lower left corner from the origin of the character.
71
+ width, height, x_disp, y_disp = (int(p) for p in props["BBX"].split())
72
+
73
+ # The word DWIDTH
74
+ # followed by the width in x and y of the character in device pixels.
75
+ dwx, dwy = (int(p) for p in props["DWIDTH"].split())
76
+
77
+ bbox = (
78
+ (dwx, dwy),
79
+ (x_disp, -y_disp - height, width + x_disp, -y_disp),
80
+ (0, 0, width, height),
81
+ )
82
+
83
+ try:
84
+ im = Image.frombytes("1", (width, height), bitmap, "hex", "1")
85
+ except ValueError:
86
+ # deal with zero-width characters
87
+ im = Image.new("1", (width, height))
88
+
89
+ return id, int(props["ENCODING"]), bbox, im
90
+
91
+
92
+ class BdfFontFile(FontFile.FontFile):
93
+ """Font file plugin for the X11 BDF format."""
94
+
95
+ def __init__(self, fp: BinaryIO) -> None:
96
+ super().__init__()
97
+
98
+ s = fp.readline()
99
+ if not s.startswith(b"STARTFONT 2.1"):
100
+ msg = "not a valid BDF file"
101
+ raise SyntaxError(msg)
102
+
103
+ props = {}
104
+ comments = []
105
+
106
+ while True:
107
+ s = fp.readline()
108
+ if not s or s.startswith(b"ENDPROPERTIES"):
109
+ break
110
+ i = s.find(b" ")
111
+ props[s[:i].decode("ascii")] = s[i + 1 : -1].decode("ascii")
112
+ if s[:i] in [b"COMMENT", b"COPYRIGHT"]:
113
+ if s.find(b"LogicalFontDescription") < 0:
114
+ comments.append(s[i + 1 : -1].decode("ascii"))
115
+
116
+ while True:
117
+ c = bdf_char(fp)
118
+ if not c:
119
+ break
120
+ id, ch, (xy, dst, src), im = c
121
+ if 0 <= ch < len(self.glyph):
122
+ self.glyph[ch] = xy, dst, src, im
.venv/Lib/site-packages/PIL/BlpImagePlugin.py ADDED
@@ -0,0 +1,497 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ """
2
+ Blizzard Mipmap Format (.blp)
3
+ Jerome Leclanche <jerome@leclan.ch>
4
+
5
+ The contents of this file are hereby released in the public domain (CC0)
6
+ Full text of the CC0 license:
7
+ https://creativecommons.org/publicdomain/zero/1.0/
8
+
9
+ BLP1 files, used mostly in Warcraft III, are not fully supported.
10
+ All types of BLP2 files used in World of Warcraft are supported.
11
+
12
+ The BLP file structure consists of a header, up to 16 mipmaps of the
13
+ texture
14
+
15
+ Texture sizes must be powers of two, though the two dimensions do
16
+ not have to be equal; 512x256 is valid, but 512x200 is not.
17
+ The first mipmap (mipmap #0) is the full size image; each subsequent
18
+ mipmap halves both dimensions. The final mipmap should be 1x1.
19
+
20
+ BLP files come in many different flavours:
21
+ * JPEG-compressed (type == 0) - only supported for BLP1.
22
+ * RAW images (type == 1, encoding == 1). Each mipmap is stored as an
23
+ array of 8-bit values, one per pixel, left to right, top to bottom.
24
+ Each value is an index to the palette.
25
+ * DXT-compressed (type == 1, encoding == 2):
26
+ - DXT1 compression is used if alpha_encoding == 0.
27
+ - An additional alpha bit is used if alpha_depth == 1.
28
+ - DXT3 compression is used if alpha_encoding == 1.
29
+ - DXT5 compression is used if alpha_encoding == 7.
30
+ """
31
+
32
+ from __future__ import annotations
33
+
34
+ import abc
35
+ import os
36
+ import struct
37
+ from enum import IntEnum
38
+ from io import BytesIO
39
+ from typing import IO
40
+
41
+ from . import Image, ImageFile
42
+
43
+
44
+ class Format(IntEnum):
45
+ JPEG = 0
46
+
47
+
48
+ class Encoding(IntEnum):
49
+ UNCOMPRESSED = 1
50
+ DXT = 2
51
+ UNCOMPRESSED_RAW_BGRA = 3
52
+
53
+
54
+ class AlphaEncoding(IntEnum):
55
+ DXT1 = 0
56
+ DXT3 = 1
57
+ DXT5 = 7
58
+
59
+
60
+ def unpack_565(i: int) -> tuple[int, int, int]:
61
+ return ((i >> 11) & 0x1F) << 3, ((i >> 5) & 0x3F) << 2, (i & 0x1F) << 3
62
+
63
+
64
+ def decode_dxt1(
65
+ data: bytes, alpha: bool = False
66
+ ) -> tuple[bytearray, bytearray, bytearray, bytearray]:
67
+ """
68
+ input: one "row" of data (i.e. will produce 4*width pixels)
69
+ """
70
+
71
+ blocks = len(data) // 8 # number of blocks in row
72
+ ret = (bytearray(), bytearray(), bytearray(), bytearray())
73
+
74
+ for block_index in range(blocks):
75
+ # Decode next 8-byte block.
76
+ idx = block_index * 8
77
+ color0, color1, bits = struct.unpack_from("<HHI", data, idx)
78
+
79
+ r0, g0, b0 = unpack_565(color0)
80
+ r1, g1, b1 = unpack_565(color1)
81
+
82
+ # Decode this block into 4x4 pixels
83
+ # Accumulate the results onto our 4 row accumulators
84
+ for j in range(4):
85
+ for i in range(4):
86
+ # get next control op and generate a pixel
87
+
88
+ control = bits & 3
89
+ bits = bits >> 2
90
+
91
+ a = 0xFF
92
+ if control == 0:
93
+ r, g, b = r0, g0, b0
94
+ elif control == 1:
95
+ r, g, b = r1, g1, b1
96
+ elif control == 2:
97
+ if color0 > color1:
98
+ r = (2 * r0 + r1) // 3
99
+ g = (2 * g0 + g1) // 3
100
+ b = (2 * b0 + b1) // 3
101
+ else:
102
+ r = (r0 + r1) // 2
103
+ g = (g0 + g1) // 2
104
+ b = (b0 + b1) // 2
105
+ elif control == 3:
106
+ if color0 > color1:
107
+ r = (2 * r1 + r0) // 3
108
+ g = (2 * g1 + g0) // 3
109
+ b = (2 * b1 + b0) // 3
110
+ else:
111
+ r, g, b, a = 0, 0, 0, 0
112
+
113
+ if alpha:
114
+ ret[j].extend([r, g, b, a])
115
+ else:
116
+ ret[j].extend([r, g, b])
117
+
118
+ return ret
119
+
120
+
121
+ def decode_dxt3(data: bytes) -> tuple[bytearray, bytearray, bytearray, bytearray]:
122
+ """
123
+ input: one "row" of data (i.e. will produce 4*width pixels)
124
+ """
125
+
126
+ blocks = len(data) // 16 # number of blocks in row
127
+ ret = (bytearray(), bytearray(), bytearray(), bytearray())
128
+
129
+ for block_index in range(blocks):
130
+ idx = block_index * 16
131
+ block = data[idx : idx + 16]
132
+ # Decode next 16-byte block.
133
+ bits = struct.unpack_from("<8B", block)
134
+ color0, color1 = struct.unpack_from("<HH", block, 8)
135
+
136
+ (code,) = struct.unpack_from("<I", block, 12)
137
+
138
+ r0, g0, b0 = unpack_565(color0)
139
+ r1, g1, b1 = unpack_565(color1)
140
+
141
+ for j in range(4):
142
+ high = False # Do we want the higher bits?
143
+ for i in range(4):
144
+ alphacode_index = (4 * j + i) // 2
145
+ a = bits[alphacode_index]
146
+ if high:
147
+ high = False
148
+ a >>= 4
149
+ else:
150
+ high = True
151
+ a &= 0xF
152
+ a *= 17 # We get a value between 0 and 15
153
+
154
+ color_code = (code >> 2 * (4 * j + i)) & 0x03
155
+
156
+ if color_code == 0:
157
+ r, g, b = r0, g0, b0
158
+ elif color_code == 1:
159
+ r, g, b = r1, g1, b1
160
+ elif color_code == 2:
161
+ r = (2 * r0 + r1) // 3
162
+ g = (2 * g0 + g1) // 3
163
+ b = (2 * b0 + b1) // 3
164
+ elif color_code == 3:
165
+ r = (2 * r1 + r0) // 3
166
+ g = (2 * g1 + g0) // 3
167
+ b = (2 * b1 + b0) // 3
168
+
169
+ ret[j].extend([r, g, b, a])
170
+
171
+ return ret
172
+
173
+
174
+ def decode_dxt5(data: bytes) -> tuple[bytearray, bytearray, bytearray, bytearray]:
175
+ """
176
+ input: one "row" of data (i.e. will produce 4 * width pixels)
177
+ """
178
+
179
+ blocks = len(data) // 16 # number of blocks in row
180
+ ret = (bytearray(), bytearray(), bytearray(), bytearray())
181
+
182
+ for block_index in range(blocks):
183
+ idx = block_index * 16
184
+ block = data[idx : idx + 16]
185
+ # Decode next 16-byte block.
186
+ a0, a1 = struct.unpack_from("<BB", block)
187
+
188
+ bits = struct.unpack_from("<6B", block, 2)
189
+ alphacode1 = bits[2] | (bits[3] << 8) | (bits[4] << 16) | (bits[5] << 24)
190
+ alphacode2 = bits[0] | (bits[1] << 8)
191
+
192
+ color0, color1 = struct.unpack_from("<HH", block, 8)
193
+
194
+ (code,) = struct.unpack_from("<I", block, 12)
195
+
196
+ r0, g0, b0 = unpack_565(color0)
197
+ r1, g1, b1 = unpack_565(color1)
198
+
199
+ for j in range(4):
200
+ for i in range(4):
201
+ # get next control op and generate a pixel
202
+ alphacode_index = 3 * (4 * j + i)
203
+
204
+ if alphacode_index <= 12:
205
+ alphacode = (alphacode2 >> alphacode_index) & 0x07
206
+ elif alphacode_index == 15:
207
+ alphacode = (alphacode2 >> 15) | ((alphacode1 << 1) & 0x06)
208
+ else: # alphacode_index >= 18 and alphacode_index <= 45
209
+ alphacode = (alphacode1 >> (alphacode_index - 16)) & 0x07
210
+
211
+ if alphacode == 0:
212
+ a = a0
213
+ elif alphacode == 1:
214
+ a = a1
215
+ elif a0 > a1:
216
+ a = ((8 - alphacode) * a0 + (alphacode - 1) * a1) // 7
217
+ elif alphacode == 6:
218
+ a = 0
219
+ elif alphacode == 7:
220
+ a = 255
221
+ else:
222
+ a = ((6 - alphacode) * a0 + (alphacode - 1) * a1) // 5
223
+
224
+ color_code = (code >> 2 * (4 * j + i)) & 0x03
225
+
226
+ if color_code == 0:
227
+ r, g, b = r0, g0, b0
228
+ elif color_code == 1:
229
+ r, g, b = r1, g1, b1
230
+ elif color_code == 2:
231
+ r = (2 * r0 + r1) // 3
232
+ g = (2 * g0 + g1) // 3
233
+ b = (2 * b0 + b1) // 3
234
+ elif color_code == 3:
235
+ r = (2 * r1 + r0) // 3
236
+ g = (2 * g1 + g0) // 3
237
+ b = (2 * b1 + b0) // 3
238
+
239
+ ret[j].extend([r, g, b, a])
240
+
241
+ return ret
242
+
243
+
244
+ class BLPFormatError(NotImplementedError):
245
+ pass
246
+
247
+
248
+ def _accept(prefix: bytes) -> bool:
249
+ return prefix.startswith((b"BLP1", b"BLP2"))
250
+
251
+
252
+ class BlpImageFile(ImageFile.ImageFile):
253
+ """
254
+ Blizzard Mipmap Format
255
+ """
256
+
257
+ format = "BLP"
258
+ format_description = "Blizzard Mipmap Format"
259
+
260
+ def _open(self) -> None:
261
+ self.magic = self.fp.read(4)
262
+ if not _accept(self.magic):
263
+ msg = f"Bad BLP magic {repr(self.magic)}"
264
+ raise BLPFormatError(msg)
265
+
266
+ compression = struct.unpack("<i", self.fp.read(4))[0]
267
+ if self.magic == b"BLP1":
268
+ alpha = struct.unpack("<I", self.fp.read(4))[0] != 0
269
+ else:
270
+ encoding = struct.unpack("<b", self.fp.read(1))[0]
271
+ alpha = struct.unpack("<b", self.fp.read(1))[0] != 0
272
+ alpha_encoding = struct.unpack("<b", self.fp.read(1))[0]
273
+ self.fp.seek(1, os.SEEK_CUR) # mips
274
+
275
+ self._size = struct.unpack("<II", self.fp.read(8))
276
+
277
+ args: tuple[int, int, bool] | tuple[int, int, bool, int]
278
+ if self.magic == b"BLP1":
279
+ encoding = struct.unpack("<i", self.fp.read(4))[0]
280
+ self.fp.seek(4, os.SEEK_CUR) # subtype
281
+
282
+ args = (compression, encoding, alpha)
283
+ offset = 28
284
+ else:
285
+ args = (compression, encoding, alpha, alpha_encoding)
286
+ offset = 20
287
+
288
+ decoder = self.magic.decode()
289
+
290
+ self._mode = "RGBA" if alpha else "RGB"
291
+ self.tile = [ImageFile._Tile(decoder, (0, 0) + self.size, offset, args)]
292
+
293
+
294
+ class _BLPBaseDecoder(abc.ABC, ImageFile.PyDecoder):
295
+ _pulls_fd = True
296
+
297
+ def decode(self, buffer: bytes | Image.SupportsArrayInterface) -> tuple[int, int]:
298
+ try:
299
+ self._read_header()
300
+ self._load()
301
+ except struct.error as e:
302
+ msg = "Truncated BLP file"
303
+ raise OSError(msg) from e
304
+ return -1, 0
305
+
306
+ @abc.abstractmethod
307
+ def _load(self) -> None:
308
+ pass
309
+
310
+ def _read_header(self) -> None:
311
+ self._offsets = struct.unpack("<16I", self._safe_read(16 * 4))
312
+ self._lengths = struct.unpack("<16I", self._safe_read(16 * 4))
313
+
314
+ def _safe_read(self, length: int) -> bytes:
315
+ assert self.fd is not None
316
+ return ImageFile._safe_read(self.fd, length)
317
+
318
+ def _read_palette(self) -> list[tuple[int, int, int, int]]:
319
+ ret = []
320
+ for i in range(256):
321
+ try:
322
+ b, g, r, a = struct.unpack("<4B", self._safe_read(4))
323
+ except struct.error:
324
+ break
325
+ ret.append((b, g, r, a))
326
+ return ret
327
+
328
+ def _read_bgra(
329
+ self, palette: list[tuple[int, int, int, int]], alpha: bool
330
+ ) -> bytearray:
331
+ data = bytearray()
332
+ _data = BytesIO(self._safe_read(self._lengths[0]))
333
+ while True:
334
+ try:
335
+ (offset,) = struct.unpack("<B", _data.read(1))
336
+ except struct.error:
337
+ break
338
+ b, g, r, a = palette[offset]
339
+ d: tuple[int, ...] = (r, g, b)
340
+ if alpha:
341
+ d += (a,)
342
+ data.extend(d)
343
+ return data
344
+
345
+
346
+ class BLP1Decoder(_BLPBaseDecoder):
347
+ def _load(self) -> None:
348
+ self._compression, self._encoding, alpha = self.args
349
+
350
+ if self._compression == Format.JPEG:
351
+ self._decode_jpeg_stream()
352
+
353
+ elif self._compression == 1:
354
+ if self._encoding in (4, 5):
355
+ palette = self._read_palette()
356
+ data = self._read_bgra(palette, alpha)
357
+ self.set_as_raw(data)
358
+ else:
359
+ msg = f"Unsupported BLP encoding {repr(self._encoding)}"
360
+ raise BLPFormatError(msg)
361
+ else:
362
+ msg = f"Unsupported BLP compression {repr(self._encoding)}"
363
+ raise BLPFormatError(msg)
364
+
365
+ def _decode_jpeg_stream(self) -> None:
366
+ from .JpegImagePlugin import JpegImageFile
367
+
368
+ (jpeg_header_size,) = struct.unpack("<I", self._safe_read(4))
369
+ jpeg_header = self._safe_read(jpeg_header_size)
370
+ assert self.fd is not None
371
+ self._safe_read(self._offsets[0] - self.fd.tell()) # What IS this?
372
+ data = self._safe_read(self._lengths[0])
373
+ data = jpeg_header + data
374
+ image = JpegImageFile(BytesIO(data))
375
+ Image._decompression_bomb_check(image.size)
376
+ if image.mode == "CMYK":
377
+ args = image.tile[0].args
378
+ assert isinstance(args, tuple)
379
+ image.tile = [image.tile[0]._replace(args=(args[0], "CMYK"))]
380
+ self.set_as_raw(image.convert("RGB").tobytes(), "BGR")
381
+
382
+
383
+ class BLP2Decoder(_BLPBaseDecoder):
384
+ def _load(self) -> None:
385
+ self._compression, self._encoding, alpha, self._alpha_encoding = self.args
386
+
387
+ palette = self._read_palette()
388
+
389
+ assert self.fd is not None
390
+ self.fd.seek(self._offsets[0])
391
+
392
+ if self._compression == 1:
393
+ # Uncompressed or DirectX compression
394
+
395
+ if self._encoding == Encoding.UNCOMPRESSED:
396
+ data = self._read_bgra(palette, alpha)
397
+
398
+ elif self._encoding == Encoding.DXT:
399
+ data = bytearray()
400
+ if self._alpha_encoding == AlphaEncoding.DXT1:
401
+ linesize = (self.state.xsize + 3) // 4 * 8
402
+ for yb in range((self.state.ysize + 3) // 4):
403
+ for d in decode_dxt1(self._safe_read(linesize), alpha):
404
+ data += d
405
+
406
+ elif self._alpha_encoding == AlphaEncoding.DXT3:
407
+ linesize = (self.state.xsize + 3) // 4 * 16
408
+ for yb in range((self.state.ysize + 3) // 4):
409
+ for d in decode_dxt3(self._safe_read(linesize)):
410
+ data += d
411
+
412
+ elif self._alpha_encoding == AlphaEncoding.DXT5:
413
+ linesize = (self.state.xsize + 3) // 4 * 16
414
+ for yb in range((self.state.ysize + 3) // 4):
415
+ for d in decode_dxt5(self._safe_read(linesize)):
416
+ data += d
417
+ else:
418
+ msg = f"Unsupported alpha encoding {repr(self._alpha_encoding)}"
419
+ raise BLPFormatError(msg)
420
+ else:
421
+ msg = f"Unknown BLP encoding {repr(self._encoding)}"
422
+ raise BLPFormatError(msg)
423
+
424
+ else:
425
+ msg = f"Unknown BLP compression {repr(self._compression)}"
426
+ raise BLPFormatError(msg)
427
+
428
+ self.set_as_raw(data)
429
+
430
+
431
+ class BLPEncoder(ImageFile.PyEncoder):
432
+ _pushes_fd = True
433
+
434
+ def _write_palette(self) -> bytes:
435
+ data = b""
436
+ assert self.im is not None
437
+ palette = self.im.getpalette("RGBA", "RGBA")
438
+ for i in range(len(palette) // 4):
439
+ r, g, b, a = palette[i * 4 : (i + 1) * 4]
440
+ data += struct.pack("<4B", b, g, r, a)
441
+ while len(data) < 256 * 4:
442
+ data += b"\x00" * 4
443
+ return data
444
+
445
+ def encode(self, bufsize: int) -> tuple[int, int, bytes]:
446
+ palette_data = self._write_palette()
447
+
448
+ offset = 20 + 16 * 4 * 2 + len(palette_data)
449
+ data = struct.pack("<16I", offset, *((0,) * 15))
450
+
451
+ assert self.im is not None
452
+ w, h = self.im.size
453
+ data += struct.pack("<16I", w * h, *((0,) * 15))
454
+
455
+ data += palette_data
456
+
457
+ for y in range(h):
458
+ for x in range(w):
459
+ data += struct.pack("<B", self.im.getpixel((x, y)))
460
+
461
+ return len(data), 0, data
462
+
463
+
464
+ def _save(im: Image.Image, fp: IO[bytes], filename: str | bytes) -> None:
465
+ if im.mode != "P":
466
+ msg = "Unsupported BLP image mode"
467
+ raise ValueError(msg)
468
+
469
+ magic = b"BLP1" if im.encoderinfo.get("blp_version") == "BLP1" else b"BLP2"
470
+ fp.write(magic)
471
+
472
+ assert im.palette is not None
473
+ fp.write(struct.pack("<i", 1)) # Uncompressed or DirectX compression
474
+
475
+ alpha_depth = 1 if im.palette.mode == "RGBA" else 0
476
+ if magic == b"BLP1":
477
+ fp.write(struct.pack("<L", alpha_depth))
478
+ else:
479
+ fp.write(struct.pack("<b", Encoding.UNCOMPRESSED))
480
+ fp.write(struct.pack("<b", alpha_depth))
481
+ fp.write(struct.pack("<b", 0)) # alpha encoding
482
+ fp.write(struct.pack("<b", 0)) # mips
483
+ fp.write(struct.pack("<II", *im.size))
484
+ if magic == b"BLP1":
485
+ fp.write(struct.pack("<i", 5))
486
+ fp.write(struct.pack("<i", 0))
487
+
488
+ ImageFile._save(im, fp, [ImageFile._Tile("BLP", (0, 0) + im.size, 0, im.mode)])
489
+
490
+
491
+ Image.register_open(BlpImageFile.format, BlpImageFile, _accept)
492
+ Image.register_extension(BlpImageFile.format, ".blp")
493
+ Image.register_decoder("BLP1", BLP1Decoder)
494
+ Image.register_decoder("BLP2", BLP2Decoder)
495
+
496
+ Image.register_save(BlpImageFile.format, _save)
497
+ Image.register_encoder("BLP", BLPEncoder)
.venv/Lib/site-packages/PIL/BmpImagePlugin.py ADDED
@@ -0,0 +1,515 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #
2
+ # The Python Imaging Library.
3
+ # $Id$
4
+ #
5
+ # BMP file handler
6
+ #
7
+ # Windows (and OS/2) native bitmap storage format.
8
+ #
9
+ # history:
10
+ # 1995-09-01 fl Created
11
+ # 1996-04-30 fl Added save
12
+ # 1997-08-27 fl Fixed save of 1-bit images
13
+ # 1998-03-06 fl Load P images as L where possible
14
+ # 1998-07-03 fl Load P images as 1 where possible
15
+ # 1998-12-29 fl Handle small palettes
16
+ # 2002-12-30 fl Fixed load of 1-bit palette images
17
+ # 2003-04-21 fl Fixed load of 1-bit monochrome images
18
+ # 2003-04-23 fl Added limited support for BI_BITFIELDS compression
19
+ #
20
+ # Copyright (c) 1997-2003 by Secret Labs AB
21
+ # Copyright (c) 1995-2003 by Fredrik Lundh
22
+ #
23
+ # See the README file for information on usage and redistribution.
24
+ #
25
+ from __future__ import annotations
26
+
27
+ import os
28
+ from typing import IO, Any
29
+
30
+ from . import Image, ImageFile, ImagePalette
31
+ from ._binary import i16le as i16
32
+ from ._binary import i32le as i32
33
+ from ._binary import o8
34
+ from ._binary import o16le as o16
35
+ from ._binary import o32le as o32
36
+
37
+ #
38
+ # --------------------------------------------------------------------
39
+ # Read BMP file
40
+
41
+ BIT2MODE = {
42
+ # bits => mode, rawmode
43
+ 1: ("P", "P;1"),
44
+ 4: ("P", "P;4"),
45
+ 8: ("P", "P"),
46
+ 16: ("RGB", "BGR;15"),
47
+ 24: ("RGB", "BGR"),
48
+ 32: ("RGB", "BGRX"),
49
+ }
50
+
51
+ USE_RAW_ALPHA = False
52
+
53
+
54
+ def _accept(prefix: bytes) -> bool:
55
+ return prefix.startswith(b"BM")
56
+
57
+
58
+ def _dib_accept(prefix: bytes) -> bool:
59
+ return i32(prefix) in [12, 40, 52, 56, 64, 108, 124]
60
+
61
+
62
+ # =============================================================================
63
+ # Image plugin for the Windows BMP format.
64
+ # =============================================================================
65
+ class BmpImageFile(ImageFile.ImageFile):
66
+ """Image plugin for the Windows Bitmap format (BMP)"""
67
+
68
+ # ------------------------------------------------------------- Description
69
+ format_description = "Windows Bitmap"
70
+ format = "BMP"
71
+
72
+ # -------------------------------------------------- BMP Compression values
73
+ COMPRESSIONS = {"RAW": 0, "RLE8": 1, "RLE4": 2, "BITFIELDS": 3, "JPEG": 4, "PNG": 5}
74
+ for k, v in COMPRESSIONS.items():
75
+ vars()[k] = v
76
+
77
+ def _bitmap(self, header: int = 0, offset: int = 0) -> None:
78
+ """Read relevant info about the BMP"""
79
+ read, seek = self.fp.read, self.fp.seek
80
+ if header:
81
+ seek(header)
82
+ # read bmp header size @offset 14 (this is part of the header size)
83
+ file_info: dict[str, bool | int | tuple[int, ...]] = {
84
+ "header_size": i32(read(4)),
85
+ "direction": -1,
86
+ }
87
+
88
+ # -------------------- If requested, read header at a specific position
89
+ # read the rest of the bmp header, without its size
90
+ assert isinstance(file_info["header_size"], int)
91
+ header_data = ImageFile._safe_read(self.fp, file_info["header_size"] - 4)
92
+
93
+ # ------------------------------- Windows Bitmap v2, IBM OS/2 Bitmap v1
94
+ # ----- This format has different offsets because of width/height types
95
+ # 12: BITMAPCOREHEADER/OS21XBITMAPHEADER
96
+ if file_info["header_size"] == 12:
97
+ file_info["width"] = i16(header_data, 0)
98
+ file_info["height"] = i16(header_data, 2)
99
+ file_info["planes"] = i16(header_data, 4)
100
+ file_info["bits"] = i16(header_data, 6)
101
+ file_info["compression"] = self.COMPRESSIONS["RAW"]
102
+ file_info["palette_padding"] = 3
103
+
104
+ # --------------------------------------------- Windows Bitmap v3 to v5
105
+ # 40: BITMAPINFOHEADER
106
+ # 52: BITMAPV2HEADER
107
+ # 56: BITMAPV3HEADER
108
+ # 64: BITMAPCOREHEADER2/OS22XBITMAPHEADER
109
+ # 108: BITMAPV4HEADER
110
+ # 124: BITMAPV5HEADER
111
+ elif file_info["header_size"] in (40, 52, 56, 64, 108, 124):
112
+ file_info["y_flip"] = header_data[7] == 0xFF
113
+ file_info["direction"] = 1 if file_info["y_flip"] else -1
114
+ file_info["width"] = i32(header_data, 0)
115
+ file_info["height"] = (
116
+ i32(header_data, 4)
117
+ if not file_info["y_flip"]
118
+ else 2**32 - i32(header_data, 4)
119
+ )
120
+ file_info["planes"] = i16(header_data, 8)
121
+ file_info["bits"] = i16(header_data, 10)
122
+ file_info["compression"] = i32(header_data, 12)
123
+ # byte size of pixel data
124
+ file_info["data_size"] = i32(header_data, 16)
125
+ file_info["pixels_per_meter"] = (
126
+ i32(header_data, 20),
127
+ i32(header_data, 24),
128
+ )
129
+ file_info["colors"] = i32(header_data, 28)
130
+ file_info["palette_padding"] = 4
131
+ assert isinstance(file_info["pixels_per_meter"], tuple)
132
+ self.info["dpi"] = tuple(x / 39.3701 for x in file_info["pixels_per_meter"])
133
+ if file_info["compression"] == self.COMPRESSIONS["BITFIELDS"]:
134
+ masks = ["r_mask", "g_mask", "b_mask"]
135
+ if len(header_data) >= 48:
136
+ if len(header_data) >= 52:
137
+ masks.append("a_mask")
138
+ else:
139
+ file_info["a_mask"] = 0x0
140
+ for idx, mask in enumerate(masks):
141
+ file_info[mask] = i32(header_data, 36 + idx * 4)
142
+ else:
143
+ # 40 byte headers only have the three components in the
144
+ # bitfields masks, ref:
145
+ # https://msdn.microsoft.com/en-us/library/windows/desktop/dd183376(v=vs.85).aspx
146
+ # See also
147
+ # https://github.com/python-pillow/Pillow/issues/1293
148
+ # There is a 4th component in the RGBQuad, in the alpha
149
+ # location, but it is listed as a reserved component,
150
+ # and it is not generally an alpha channel
151
+ file_info["a_mask"] = 0x0
152
+ for mask in masks:
153
+ file_info[mask] = i32(read(4))
154
+ assert isinstance(file_info["r_mask"], int)
155
+ assert isinstance(file_info["g_mask"], int)
156
+ assert isinstance(file_info["b_mask"], int)
157
+ assert isinstance(file_info["a_mask"], int)
158
+ file_info["rgb_mask"] = (
159
+ file_info["r_mask"],
160
+ file_info["g_mask"],
161
+ file_info["b_mask"],
162
+ )
163
+ file_info["rgba_mask"] = (
164
+ file_info["r_mask"],
165
+ file_info["g_mask"],
166
+ file_info["b_mask"],
167
+ file_info["a_mask"],
168
+ )
169
+ else:
170
+ msg = f"Unsupported BMP header type ({file_info['header_size']})"
171
+ raise OSError(msg)
172
+
173
+ # ------------------ Special case : header is reported 40, which
174
+ # ---------------------- is shorter than real size for bpp >= 16
175
+ assert isinstance(file_info["width"], int)
176
+ assert isinstance(file_info["height"], int)
177
+ self._size = file_info["width"], file_info["height"]
178
+
179
+ # ------- If color count was not found in the header, compute from bits
180
+ assert isinstance(file_info["bits"], int)
181
+ file_info["colors"] = (
182
+ file_info["colors"]
183
+ if file_info.get("colors", 0)
184
+ else (1 << file_info["bits"])
185
+ )
186
+ assert isinstance(file_info["colors"], int)
187
+ if offset == 14 + file_info["header_size"] and file_info["bits"] <= 8:
188
+ offset += 4 * file_info["colors"]
189
+
190
+ # ---------------------- Check bit depth for unusual unsupported values
191
+ self._mode, raw_mode = BIT2MODE.get(file_info["bits"], ("", ""))
192
+ if not self.mode:
193
+ msg = f"Unsupported BMP pixel depth ({file_info['bits']})"
194
+ raise OSError(msg)
195
+
196
+ # ---------------- Process BMP with Bitfields compression (not palette)
197
+ decoder_name = "raw"
198
+ if file_info["compression"] == self.COMPRESSIONS["BITFIELDS"]:
199
+ SUPPORTED: dict[int, list[tuple[int, ...]]] = {
200
+ 32: [
201
+ (0xFF0000, 0xFF00, 0xFF, 0x0),
202
+ (0xFF000000, 0xFF0000, 0xFF00, 0x0),
203
+ (0xFF000000, 0xFF00, 0xFF, 0x0),
204
+ (0xFF000000, 0xFF0000, 0xFF00, 0xFF),
205
+ (0xFF, 0xFF00, 0xFF0000, 0xFF000000),
206
+ (0xFF0000, 0xFF00, 0xFF, 0xFF000000),
207
+ (0xFF000000, 0xFF00, 0xFF, 0xFF0000),
208
+ (0x0, 0x0, 0x0, 0x0),
209
+ ],
210
+ 24: [(0xFF0000, 0xFF00, 0xFF)],
211
+ 16: [(0xF800, 0x7E0, 0x1F), (0x7C00, 0x3E0, 0x1F)],
212
+ }
213
+ MASK_MODES = {
214
+ (32, (0xFF0000, 0xFF00, 0xFF, 0x0)): "BGRX",
215
+ (32, (0xFF000000, 0xFF0000, 0xFF00, 0x0)): "XBGR",
216
+ (32, (0xFF000000, 0xFF00, 0xFF, 0x0)): "BGXR",
217
+ (32, (0xFF000000, 0xFF0000, 0xFF00, 0xFF)): "ABGR",
218
+ (32, (0xFF, 0xFF00, 0xFF0000, 0xFF000000)): "RGBA",
219
+ (32, (0xFF0000, 0xFF00, 0xFF, 0xFF000000)): "BGRA",
220
+ (32, (0xFF000000, 0xFF00, 0xFF, 0xFF0000)): "BGAR",
221
+ (32, (0x0, 0x0, 0x0, 0x0)): "BGRA",
222
+ (24, (0xFF0000, 0xFF00, 0xFF)): "BGR",
223
+ (16, (0xF800, 0x7E0, 0x1F)): "BGR;16",
224
+ (16, (0x7C00, 0x3E0, 0x1F)): "BGR;15",
225
+ }
226
+ if file_info["bits"] in SUPPORTED:
227
+ if (
228
+ file_info["bits"] == 32
229
+ and file_info["rgba_mask"] in SUPPORTED[file_info["bits"]]
230
+ ):
231
+ assert isinstance(file_info["rgba_mask"], tuple)
232
+ raw_mode = MASK_MODES[(file_info["bits"], file_info["rgba_mask"])]
233
+ self._mode = "RGBA" if "A" in raw_mode else self.mode
234
+ elif (
235
+ file_info["bits"] in (24, 16)
236
+ and file_info["rgb_mask"] in SUPPORTED[file_info["bits"]]
237
+ ):
238
+ assert isinstance(file_info["rgb_mask"], tuple)
239
+ raw_mode = MASK_MODES[(file_info["bits"], file_info["rgb_mask"])]
240
+ else:
241
+ msg = "Unsupported BMP bitfields layout"
242
+ raise OSError(msg)
243
+ else:
244
+ msg = "Unsupported BMP bitfields layout"
245
+ raise OSError(msg)
246
+ elif file_info["compression"] == self.COMPRESSIONS["RAW"]:
247
+ if file_info["bits"] == 32 and (
248
+ header == 22 or USE_RAW_ALPHA # 32-bit .cur offset
249
+ ):
250
+ raw_mode, self._mode = "BGRA", "RGBA"
251
+ elif file_info["compression"] in (
252
+ self.COMPRESSIONS["RLE8"],
253
+ self.COMPRESSIONS["RLE4"],
254
+ ):
255
+ decoder_name = "bmp_rle"
256
+ else:
257
+ msg = f"Unsupported BMP compression ({file_info['compression']})"
258
+ raise OSError(msg)
259
+
260
+ # --------------- Once the header is processed, process the palette/LUT
261
+ if self.mode == "P": # Paletted for 1, 4 and 8 bit images
262
+ # ---------------------------------------------------- 1-bit images
263
+ if not (0 < file_info["colors"] <= 65536):
264
+ msg = f"Unsupported BMP Palette size ({file_info['colors']})"
265
+ raise OSError(msg)
266
+ else:
267
+ assert isinstance(file_info["palette_padding"], int)
268
+ padding = file_info["palette_padding"]
269
+ palette = read(padding * file_info["colors"])
270
+ grayscale = True
271
+ indices = (
272
+ (0, 255)
273
+ if file_info["colors"] == 2
274
+ else list(range(file_info["colors"]))
275
+ )
276
+
277
+ # ----------------- Check if grayscale and ignore palette if so
278
+ for ind, val in enumerate(indices):
279
+ rgb = palette[ind * padding : ind * padding + 3]
280
+ if rgb != o8(val) * 3:
281
+ grayscale = False
282
+
283
+ # ------- If all colors are gray, white or black, ditch palette
284
+ if grayscale:
285
+ self._mode = "1" if file_info["colors"] == 2 else "L"
286
+ raw_mode = self.mode
287
+ else:
288
+ self._mode = "P"
289
+ self.palette = ImagePalette.raw(
290
+ "BGRX" if padding == 4 else "BGR", palette
291
+ )
292
+
293
+ # ---------------------------- Finally set the tile data for the plugin
294
+ self.info["compression"] = file_info["compression"]
295
+ args: list[Any] = [raw_mode]
296
+ if decoder_name == "bmp_rle":
297
+ args.append(file_info["compression"] == self.COMPRESSIONS["RLE4"])
298
+ else:
299
+ assert isinstance(file_info["width"], int)
300
+ args.append(((file_info["width"] * file_info["bits"] + 31) >> 3) & (~3))
301
+ args.append(file_info["direction"])
302
+ self.tile = [
303
+ ImageFile._Tile(
304
+ decoder_name,
305
+ (0, 0, file_info["width"], file_info["height"]),
306
+ offset or self.fp.tell(),
307
+ tuple(args),
308
+ )
309
+ ]
310
+
311
+ def _open(self) -> None:
312
+ """Open file, check magic number and read header"""
313
+ # read 14 bytes: magic number, filesize, reserved, header final offset
314
+ head_data = self.fp.read(14)
315
+ # choke if the file does not have the required magic bytes
316
+ if not _accept(head_data):
317
+ msg = "Not a BMP file"
318
+ raise SyntaxError(msg)
319
+ # read the start position of the BMP image data (u32)
320
+ offset = i32(head_data, 10)
321
+ # load bitmap information (offset=raster info)
322
+ self._bitmap(offset=offset)
323
+
324
+
325
+ class BmpRleDecoder(ImageFile.PyDecoder):
326
+ _pulls_fd = True
327
+
328
+ def decode(self, buffer: bytes | Image.SupportsArrayInterface) -> tuple[int, int]:
329
+ assert self.fd is not None
330
+ rle4 = self.args[1]
331
+ data = bytearray()
332
+ x = 0
333
+ dest_length = self.state.xsize * self.state.ysize
334
+ while len(data) < dest_length:
335
+ pixels = self.fd.read(1)
336
+ byte = self.fd.read(1)
337
+ if not pixels or not byte:
338
+ break
339
+ num_pixels = pixels[0]
340
+ if num_pixels:
341
+ # encoded mode
342
+ if x + num_pixels > self.state.xsize:
343
+ # Too much data for row
344
+ num_pixels = max(0, self.state.xsize - x)
345
+ if rle4:
346
+ first_pixel = o8(byte[0] >> 4)
347
+ second_pixel = o8(byte[0] & 0x0F)
348
+ for index in range(num_pixels):
349
+ if index % 2 == 0:
350
+ data += first_pixel
351
+ else:
352
+ data += second_pixel
353
+ else:
354
+ data += byte * num_pixels
355
+ x += num_pixels
356
+ else:
357
+ if byte[0] == 0:
358
+ # end of line
359
+ while len(data) % self.state.xsize != 0:
360
+ data += b"\x00"
361
+ x = 0
362
+ elif byte[0] == 1:
363
+ # end of bitmap
364
+ break
365
+ elif byte[0] == 2:
366
+ # delta
367
+ bytes_read = self.fd.read(2)
368
+ if len(bytes_read) < 2:
369
+ break
370
+ right, up = self.fd.read(2)
371
+ data += b"\x00" * (right + up * self.state.xsize)
372
+ x = len(data) % self.state.xsize
373
+ else:
374
+ # absolute mode
375
+ if rle4:
376
+ # 2 pixels per byte
377
+ byte_count = byte[0] // 2
378
+ bytes_read = self.fd.read(byte_count)
379
+ for byte_read in bytes_read:
380
+ data += o8(byte_read >> 4)
381
+ data += o8(byte_read & 0x0F)
382
+ else:
383
+ byte_count = byte[0]
384
+ bytes_read = self.fd.read(byte_count)
385
+ data += bytes_read
386
+ if len(bytes_read) < byte_count:
387
+ break
388
+ x += byte[0]
389
+
390
+ # align to 16-bit word boundary
391
+ if self.fd.tell() % 2 != 0:
392
+ self.fd.seek(1, os.SEEK_CUR)
393
+ rawmode = "L" if self.mode == "L" else "P"
394
+ self.set_as_raw(bytes(data), rawmode, (0, self.args[-1]))
395
+ return -1, 0
396
+
397
+
398
+ # =============================================================================
399
+ # Image plugin for the DIB format (BMP alias)
400
+ # =============================================================================
401
+ class DibImageFile(BmpImageFile):
402
+ format = "DIB"
403
+ format_description = "Windows Bitmap"
404
+
405
+ def _open(self) -> None:
406
+ self._bitmap()
407
+
408
+
409
+ #
410
+ # --------------------------------------------------------------------
411
+ # Write BMP file
412
+
413
+
414
+ SAVE = {
415
+ "1": ("1", 1, 2),
416
+ "L": ("L", 8, 256),
417
+ "P": ("P", 8, 256),
418
+ "RGB": ("BGR", 24, 0),
419
+ "RGBA": ("BGRA", 32, 0),
420
+ }
421
+
422
+
423
+ def _dib_save(im: Image.Image, fp: IO[bytes], filename: str | bytes) -> None:
424
+ _save(im, fp, filename, False)
425
+
426
+
427
+ def _save(
428
+ im: Image.Image, fp: IO[bytes], filename: str | bytes, bitmap_header: bool = True
429
+ ) -> None:
430
+ try:
431
+ rawmode, bits, colors = SAVE[im.mode]
432
+ except KeyError as e:
433
+ msg = f"cannot write mode {im.mode} as BMP"
434
+ raise OSError(msg) from e
435
+
436
+ info = im.encoderinfo
437
+
438
+ dpi = info.get("dpi", (96, 96))
439
+
440
+ # 1 meter == 39.3701 inches
441
+ ppm = tuple(int(x * 39.3701 + 0.5) for x in dpi)
442
+
443
+ stride = ((im.size[0] * bits + 7) // 8 + 3) & (~3)
444
+ header = 40 # or 64 for OS/2 version 2
445
+ image = stride * im.size[1]
446
+
447
+ if im.mode == "1":
448
+ palette = b"".join(o8(i) * 3 + b"\x00" for i in (0, 255))
449
+ elif im.mode == "L":
450
+ palette = b"".join(o8(i) * 3 + b"\x00" for i in range(256))
451
+ elif im.mode == "P":
452
+ palette = im.im.getpalette("RGB", "BGRX")
453
+ colors = len(palette) // 4
454
+ else:
455
+ palette = None
456
+
457
+ # bitmap header
458
+ if bitmap_header:
459
+ offset = 14 + header + colors * 4
460
+ file_size = offset + image
461
+ if file_size > 2**32 - 1:
462
+ msg = "File size is too large for the BMP format"
463
+ raise ValueError(msg)
464
+ fp.write(
465
+ b"BM" # file type (magic)
466
+ + o32(file_size) # file size
467
+ + o32(0) # reserved
468
+ + o32(offset) # image data offset
469
+ )
470
+
471
+ # bitmap info header
472
+ fp.write(
473
+ o32(header) # info header size
474
+ + o32(im.size[0]) # width
475
+ + o32(im.size[1]) # height
476
+ + o16(1) # planes
477
+ + o16(bits) # depth
478
+ + o32(0) # compression (0=uncompressed)
479
+ + o32(image) # size of bitmap
480
+ + o32(ppm[0]) # resolution
481
+ + o32(ppm[1]) # resolution
482
+ + o32(colors) # colors used
483
+ + o32(colors) # colors important
484
+ )
485
+
486
+ fp.write(b"\0" * (header - 40)) # padding (for OS/2 format)
487
+
488
+ if palette:
489
+ fp.write(palette)
490
+
491
+ ImageFile._save(
492
+ im, fp, [ImageFile._Tile("raw", (0, 0) + im.size, 0, (rawmode, stride, -1))]
493
+ )
494
+
495
+
496
+ #
497
+ # --------------------------------------------------------------------
498
+ # Registry
499
+
500
+
501
+ Image.register_open(BmpImageFile.format, BmpImageFile, _accept)
502
+ Image.register_save(BmpImageFile.format, _save)
503
+
504
+ Image.register_extension(BmpImageFile.format, ".bmp")
505
+
506
+ Image.register_mime(BmpImageFile.format, "image/bmp")
507
+
508
+ Image.register_decoder("bmp_rle", BmpRleDecoder)
509
+
510
+ Image.register_open(DibImageFile.format, DibImageFile, _dib_accept)
511
+ Image.register_save(DibImageFile.format, _dib_save)
512
+
513
+ Image.register_extension(DibImageFile.format, ".dib")
514
+
515
+ Image.register_mime(DibImageFile.format, "image/bmp")
.venv/Lib/site-packages/PIL/BufrStubImagePlugin.py ADDED
@@ -0,0 +1,75 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #
2
+ # The Python Imaging Library
3
+ # $Id$
4
+ #
5
+ # BUFR stub adapter
6
+ #
7
+ # Copyright (c) 1996-2003 by Fredrik Lundh
8
+ #
9
+ # See the README file for information on usage and redistribution.
10
+ #
11
+ from __future__ import annotations
12
+
13
+ import os
14
+ from typing import IO
15
+
16
+ from . import Image, ImageFile
17
+
18
+ _handler = None
19
+
20
+
21
+ def register_handler(handler: ImageFile.StubHandler | None) -> None:
22
+ """
23
+ Install application-specific BUFR image handler.
24
+
25
+ :param handler: Handler object.
26
+ """
27
+ global _handler
28
+ _handler = handler
29
+
30
+
31
+ # --------------------------------------------------------------------
32
+ # Image adapter
33
+
34
+
35
+ def _accept(prefix: bytes) -> bool:
36
+ return prefix.startswith((b"BUFR", b"ZCZC"))
37
+
38
+
39
+ class BufrStubImageFile(ImageFile.StubImageFile):
40
+ format = "BUFR"
41
+ format_description = "BUFR"
42
+
43
+ def _open(self) -> None:
44
+ if not _accept(self.fp.read(4)):
45
+ msg = "Not a BUFR file"
46
+ raise SyntaxError(msg)
47
+
48
+ self.fp.seek(-4, os.SEEK_CUR)
49
+
50
+ # make something up
51
+ self._mode = "F"
52
+ self._size = 1, 1
53
+
54
+ loader = self._load()
55
+ if loader:
56
+ loader.open(self)
57
+
58
+ def _load(self) -> ImageFile.StubHandler | None:
59
+ return _handler
60
+
61
+
62
+ def _save(im: Image.Image, fp: IO[bytes], filename: str | bytes) -> None:
63
+ if _handler is None or not hasattr(_handler, "save"):
64
+ msg = "BUFR save handler not installed"
65
+ raise OSError(msg)
66
+ _handler.save(im, fp, filename)
67
+
68
+
69
+ # --------------------------------------------------------------------
70
+ # Registry
71
+
72
+ Image.register_open(BufrStubImageFile.format, BufrStubImageFile, _accept)
73
+ Image.register_save(BufrStubImageFile.format, _save)
74
+
75
+ Image.register_extension(BufrStubImageFile.format, ".bufr")
.venv/Lib/site-packages/PIL/ContainerIO.py ADDED
@@ -0,0 +1,173 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #
2
+ # The Python Imaging Library.
3
+ # $Id$
4
+ #
5
+ # a class to read from a container file
6
+ #
7
+ # History:
8
+ # 1995-06-18 fl Created
9
+ # 1995-09-07 fl Added readline(), readlines()
10
+ #
11
+ # Copyright (c) 1997-2001 by Secret Labs AB
12
+ # Copyright (c) 1995 by Fredrik Lundh
13
+ #
14
+ # See the README file for information on usage and redistribution.
15
+ #
16
+ from __future__ import annotations
17
+
18
+ import io
19
+ from collections.abc import Iterable
20
+ from typing import IO, AnyStr, NoReturn
21
+
22
+
23
+ class ContainerIO(IO[AnyStr]):
24
+ """
25
+ A file object that provides read access to a part of an existing
26
+ file (for example a TAR file).
27
+ """
28
+
29
+ def __init__(self, file: IO[AnyStr], offset: int, length: int) -> None:
30
+ """
31
+ Create file object.
32
+
33
+ :param file: Existing file.
34
+ :param offset: Start of region, in bytes.
35
+ :param length: Size of region, in bytes.
36
+ """
37
+ self.fh: IO[AnyStr] = file
38
+ self.pos = 0
39
+ self.offset = offset
40
+ self.length = length
41
+ self.fh.seek(offset)
42
+
43
+ ##
44
+ # Always false.
45
+
46
+ def isatty(self) -> bool:
47
+ return False
48
+
49
+ def seekable(self) -> bool:
50
+ return True
51
+
52
+ def seek(self, offset: int, mode: int = io.SEEK_SET) -> int:
53
+ """
54
+ Move file pointer.
55
+
56
+ :param offset: Offset in bytes.
57
+ :param mode: Starting position. Use 0 for beginning of region, 1
58
+ for current offset, and 2 for end of region. You cannot move
59
+ the pointer outside the defined region.
60
+ :returns: Offset from start of region, in bytes.
61
+ """
62
+ if mode == 1:
63
+ self.pos = self.pos + offset
64
+ elif mode == 2:
65
+ self.pos = self.length + offset
66
+ else:
67
+ self.pos = offset
68
+ # clamp
69
+ self.pos = max(0, min(self.pos, self.length))
70
+ self.fh.seek(self.offset + self.pos)
71
+ return self.pos
72
+
73
+ def tell(self) -> int:
74
+ """
75
+ Get current file pointer.
76
+
77
+ :returns: Offset from start of region, in bytes.
78
+ """
79
+ return self.pos
80
+
81
+ def readable(self) -> bool:
82
+ return True
83
+
84
+ def read(self, n: int = -1) -> AnyStr:
85
+ """
86
+ Read data.
87
+
88
+ :param n: Number of bytes to read. If omitted, zero or negative,
89
+ read until end of region.
90
+ :returns: An 8-bit string.
91
+ """
92
+ if n > 0:
93
+ n = min(n, self.length - self.pos)
94
+ else:
95
+ n = self.length - self.pos
96
+ if n <= 0: # EOF
97
+ return b"" if "b" in self.fh.mode else "" # type: ignore[return-value]
98
+ self.pos = self.pos + n
99
+ return self.fh.read(n)
100
+
101
+ def readline(self, n: int = -1) -> AnyStr:
102
+ """
103
+ Read a line of text.
104
+
105
+ :param n: Number of bytes to read. If omitted, zero or negative,
106
+ read until end of line.
107
+ :returns: An 8-bit string.
108
+ """
109
+ s: AnyStr = b"" if "b" in self.fh.mode else "" # type: ignore[assignment]
110
+ newline_character = b"\n" if "b" in self.fh.mode else "\n"
111
+ while True:
112
+ c = self.read(1)
113
+ if not c:
114
+ break
115
+ s = s + c
116
+ if c == newline_character or len(s) == n:
117
+ break
118
+ return s
119
+
120
+ def readlines(self, n: int | None = -1) -> list[AnyStr]:
121
+ """
122
+ Read multiple lines of text.
123
+
124
+ :param n: Number of lines to read. If omitted, zero, negative or None,
125
+ read until end of region.
126
+ :returns: A list of 8-bit strings.
127
+ """
128
+ lines = []
129
+ while True:
130
+ s = self.readline()
131
+ if not s:
132
+ break
133
+ lines.append(s)
134
+ if len(lines) == n:
135
+ break
136
+ return lines
137
+
138
+ def writable(self) -> bool:
139
+ return False
140
+
141
+ def write(self, b: AnyStr) -> NoReturn:
142
+ raise NotImplementedError()
143
+
144
+ def writelines(self, lines: Iterable[AnyStr]) -> NoReturn:
145
+ raise NotImplementedError()
146
+
147
+ def truncate(self, size: int | None = None) -> int:
148
+ raise NotImplementedError()
149
+
150
+ def __enter__(self) -> ContainerIO[AnyStr]:
151
+ return self
152
+
153
+ def __exit__(self, *args: object) -> None:
154
+ self.close()
155
+
156
+ def __iter__(self) -> ContainerIO[AnyStr]:
157
+ return self
158
+
159
+ def __next__(self) -> AnyStr:
160
+ line = self.readline()
161
+ if not line:
162
+ msg = "end of region"
163
+ raise StopIteration(msg)
164
+ return line
165
+
166
+ def fileno(self) -> int:
167
+ return self.fh.fileno()
168
+
169
+ def flush(self) -> None:
170
+ self.fh.flush()
171
+
172
+ def close(self) -> None:
173
+ self.fh.close()
.venv/Lib/site-packages/PIL/CurImagePlugin.py ADDED
@@ -0,0 +1,75 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #
2
+ # The Python Imaging Library.
3
+ # $Id$
4
+ #
5
+ # Windows Cursor support for PIL
6
+ #
7
+ # notes:
8
+ # uses BmpImagePlugin.py to read the bitmap data.
9
+ #
10
+ # history:
11
+ # 96-05-27 fl Created
12
+ #
13
+ # Copyright (c) Secret Labs AB 1997.
14
+ # Copyright (c) Fredrik Lundh 1996.
15
+ #
16
+ # See the README file for information on usage and redistribution.
17
+ #
18
+ from __future__ import annotations
19
+
20
+ from . import BmpImagePlugin, Image
21
+ from ._binary import i16le as i16
22
+ from ._binary import i32le as i32
23
+
24
+ #
25
+ # --------------------------------------------------------------------
26
+
27
+
28
+ def _accept(prefix: bytes) -> bool:
29
+ return prefix.startswith(b"\0\0\2\0")
30
+
31
+
32
+ ##
33
+ # Image plugin for Windows Cursor files.
34
+
35
+
36
+ class CurImageFile(BmpImagePlugin.BmpImageFile):
37
+ format = "CUR"
38
+ format_description = "Windows Cursor"
39
+
40
+ def _open(self) -> None:
41
+ assert self.fp is not None
42
+ offset = self.fp.tell()
43
+
44
+ # check magic
45
+ s = self.fp.read(6)
46
+ if not _accept(s):
47
+ msg = "not a CUR file"
48
+ raise SyntaxError(msg)
49
+
50
+ # pick the largest cursor in the file
51
+ m = b""
52
+ for i in range(i16(s, 4)):
53
+ s = self.fp.read(16)
54
+ if not m:
55
+ m = s
56
+ elif s[0] > m[0] and s[1] > m[1]:
57
+ m = s
58
+ if not m:
59
+ msg = "No cursors were found"
60
+ raise TypeError(msg)
61
+
62
+ # load as bitmap
63
+ self._bitmap(i32(m, 12) + offset)
64
+
65
+ # patch up the bitmap height
66
+ self._size = self.size[0], self.size[1] // 2
67
+ self.tile = [self.tile[0]._replace(extents=(0, 0) + self.size)]
68
+
69
+
70
+ #
71
+ # --------------------------------------------------------------------
72
+
73
+ Image.register_open(CurImageFile.format, CurImageFile, _accept)
74
+
75
+ Image.register_extension(CurImageFile.format, ".cur")
.venv/Lib/site-packages/PIL/DcxImagePlugin.py ADDED
@@ -0,0 +1,83 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #
2
+ # The Python Imaging Library.
3
+ # $Id$
4
+ #
5
+ # DCX file handling
6
+ #
7
+ # DCX is a container file format defined by Intel, commonly used
8
+ # for fax applications. Each DCX file consists of a directory
9
+ # (a list of file offsets) followed by a set of (usually 1-bit)
10
+ # PCX files.
11
+ #
12
+ # History:
13
+ # 1995-09-09 fl Created
14
+ # 1996-03-20 fl Properly derived from PcxImageFile.
15
+ # 1998-07-15 fl Renamed offset attribute to avoid name clash
16
+ # 2002-07-30 fl Fixed file handling
17
+ #
18
+ # Copyright (c) 1997-98 by Secret Labs AB.
19
+ # Copyright (c) 1995-96 by Fredrik Lundh.
20
+ #
21
+ # See the README file for information on usage and redistribution.
22
+ #
23
+ from __future__ import annotations
24
+
25
+ from . import Image
26
+ from ._binary import i32le as i32
27
+ from ._util import DeferredError
28
+ from .PcxImagePlugin import PcxImageFile
29
+
30
+ MAGIC = 0x3ADE68B1 # QUIZ: what's this value, then?
31
+
32
+
33
+ def _accept(prefix: bytes) -> bool:
34
+ return len(prefix) >= 4 and i32(prefix) == MAGIC
35
+
36
+
37
+ ##
38
+ # Image plugin for the Intel DCX format.
39
+
40
+
41
+ class DcxImageFile(PcxImageFile):
42
+ format = "DCX"
43
+ format_description = "Intel DCX"
44
+ _close_exclusive_fp_after_loading = False
45
+
46
+ def _open(self) -> None:
47
+ # Header
48
+ s = self.fp.read(4)
49
+ if not _accept(s):
50
+ msg = "not a DCX file"
51
+ raise SyntaxError(msg)
52
+
53
+ # Component directory
54
+ self._offset = []
55
+ for i in range(1024):
56
+ offset = i32(self.fp.read(4))
57
+ if not offset:
58
+ break
59
+ self._offset.append(offset)
60
+
61
+ self._fp = self.fp
62
+ self.frame = -1
63
+ self.n_frames = len(self._offset)
64
+ self.is_animated = self.n_frames > 1
65
+ self.seek(0)
66
+
67
+ def seek(self, frame: int) -> None:
68
+ if not self._seek_check(frame):
69
+ return
70
+ if isinstance(self._fp, DeferredError):
71
+ raise self._fp.ex
72
+ self.frame = frame
73
+ self.fp = self._fp
74
+ self.fp.seek(self._offset[frame])
75
+ PcxImageFile._open(self)
76
+
77
+ def tell(self) -> int:
78
+ return self.frame
79
+
80
+
81
+ Image.register_open(DcxImageFile.format, DcxImageFile, _accept)
82
+
83
+ Image.register_extension(DcxImageFile.format, ".dcx")
.venv/Lib/site-packages/PIL/DdsImagePlugin.py ADDED
@@ -0,0 +1,624 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ """
2
+ A Pillow plugin for .dds files (S3TC-compressed aka DXTC)
3
+ Jerome Leclanche <jerome@leclan.ch>
4
+
5
+ Documentation:
6
+ https://web.archive.org/web/20170802060935/http://oss.sgi.com/projects/ogl-sample/registry/EXT/texture_compression_s3tc.txt
7
+
8
+ The contents of this file are hereby released in the public domain (CC0)
9
+ Full text of the CC0 license:
10
+ https://creativecommons.org/publicdomain/zero/1.0/
11
+ """
12
+
13
+ from __future__ import annotations
14
+
15
+ import io
16
+ import struct
17
+ import sys
18
+ from enum import IntEnum, IntFlag
19
+ from typing import IO
20
+
21
+ from . import Image, ImageFile, ImagePalette
22
+ from ._binary import i32le as i32
23
+ from ._binary import o8
24
+ from ._binary import o32le as o32
25
+
26
+ # Magic ("DDS ")
27
+ DDS_MAGIC = 0x20534444
28
+
29
+
30
+ # DDS flags
31
+ class DDSD(IntFlag):
32
+ CAPS = 0x1
33
+ HEIGHT = 0x2
34
+ WIDTH = 0x4
35
+ PITCH = 0x8
36
+ PIXELFORMAT = 0x1000
37
+ MIPMAPCOUNT = 0x20000
38
+ LINEARSIZE = 0x80000
39
+ DEPTH = 0x800000
40
+
41
+
42
+ # DDS caps
43
+ class DDSCAPS(IntFlag):
44
+ COMPLEX = 0x8
45
+ TEXTURE = 0x1000
46
+ MIPMAP = 0x400000
47
+
48
+
49
+ class DDSCAPS2(IntFlag):
50
+ CUBEMAP = 0x200
51
+ CUBEMAP_POSITIVEX = 0x400
52
+ CUBEMAP_NEGATIVEX = 0x800
53
+ CUBEMAP_POSITIVEY = 0x1000
54
+ CUBEMAP_NEGATIVEY = 0x2000
55
+ CUBEMAP_POSITIVEZ = 0x4000
56
+ CUBEMAP_NEGATIVEZ = 0x8000
57
+ VOLUME = 0x200000
58
+
59
+
60
+ # Pixel Format
61
+ class DDPF(IntFlag):
62
+ ALPHAPIXELS = 0x1
63
+ ALPHA = 0x2
64
+ FOURCC = 0x4
65
+ PALETTEINDEXED8 = 0x20
66
+ RGB = 0x40
67
+ LUMINANCE = 0x20000
68
+
69
+
70
+ # dxgiformat.h
71
+ class DXGI_FORMAT(IntEnum):
72
+ UNKNOWN = 0
73
+ R32G32B32A32_TYPELESS = 1
74
+ R32G32B32A32_FLOAT = 2
75
+ R32G32B32A32_UINT = 3
76
+ R32G32B32A32_SINT = 4
77
+ R32G32B32_TYPELESS = 5
78
+ R32G32B32_FLOAT = 6
79
+ R32G32B32_UINT = 7
80
+ R32G32B32_SINT = 8
81
+ R16G16B16A16_TYPELESS = 9
82
+ R16G16B16A16_FLOAT = 10
83
+ R16G16B16A16_UNORM = 11
84
+ R16G16B16A16_UINT = 12
85
+ R16G16B16A16_SNORM = 13
86
+ R16G16B16A16_SINT = 14
87
+ R32G32_TYPELESS = 15
88
+ R32G32_FLOAT = 16
89
+ R32G32_UINT = 17
90
+ R32G32_SINT = 18
91
+ R32G8X24_TYPELESS = 19
92
+ D32_FLOAT_S8X24_UINT = 20
93
+ R32_FLOAT_X8X24_TYPELESS = 21
94
+ X32_TYPELESS_G8X24_UINT = 22
95
+ R10G10B10A2_TYPELESS = 23
96
+ R10G10B10A2_UNORM = 24
97
+ R10G10B10A2_UINT = 25
98
+ R11G11B10_FLOAT = 26
99
+ R8G8B8A8_TYPELESS = 27
100
+ R8G8B8A8_UNORM = 28
101
+ R8G8B8A8_UNORM_SRGB = 29
102
+ R8G8B8A8_UINT = 30
103
+ R8G8B8A8_SNORM = 31
104
+ R8G8B8A8_SINT = 32
105
+ R16G16_TYPELESS = 33
106
+ R16G16_FLOAT = 34
107
+ R16G16_UNORM = 35
108
+ R16G16_UINT = 36
109
+ R16G16_SNORM = 37
110
+ R16G16_SINT = 38
111
+ R32_TYPELESS = 39
112
+ D32_FLOAT = 40
113
+ R32_FLOAT = 41
114
+ R32_UINT = 42
115
+ R32_SINT = 43
116
+ R24G8_TYPELESS = 44
117
+ D24_UNORM_S8_UINT = 45
118
+ R24_UNORM_X8_TYPELESS = 46
119
+ X24_TYPELESS_G8_UINT = 47
120
+ R8G8_TYPELESS = 48
121
+ R8G8_UNORM = 49
122
+ R8G8_UINT = 50
123
+ R8G8_SNORM = 51
124
+ R8G8_SINT = 52
125
+ R16_TYPELESS = 53
126
+ R16_FLOAT = 54
127
+ D16_UNORM = 55
128
+ R16_UNORM = 56
129
+ R16_UINT = 57
130
+ R16_SNORM = 58
131
+ R16_SINT = 59
132
+ R8_TYPELESS = 60
133
+ R8_UNORM = 61
134
+ R8_UINT = 62
135
+ R8_SNORM = 63
136
+ R8_SINT = 64
137
+ A8_UNORM = 65
138
+ R1_UNORM = 66
139
+ R9G9B9E5_SHAREDEXP = 67
140
+ R8G8_B8G8_UNORM = 68
141
+ G8R8_G8B8_UNORM = 69
142
+ BC1_TYPELESS = 70
143
+ BC1_UNORM = 71
144
+ BC1_UNORM_SRGB = 72
145
+ BC2_TYPELESS = 73
146
+ BC2_UNORM = 74
147
+ BC2_UNORM_SRGB = 75
148
+ BC3_TYPELESS = 76
149
+ BC3_UNORM = 77
150
+ BC3_UNORM_SRGB = 78
151
+ BC4_TYPELESS = 79
152
+ BC4_UNORM = 80
153
+ BC4_SNORM = 81
154
+ BC5_TYPELESS = 82
155
+ BC5_UNORM = 83
156
+ BC5_SNORM = 84
157
+ B5G6R5_UNORM = 85
158
+ B5G5R5A1_UNORM = 86
159
+ B8G8R8A8_UNORM = 87
160
+ B8G8R8X8_UNORM = 88
161
+ R10G10B10_XR_BIAS_A2_UNORM = 89
162
+ B8G8R8A8_TYPELESS = 90
163
+ B8G8R8A8_UNORM_SRGB = 91
164
+ B8G8R8X8_TYPELESS = 92
165
+ B8G8R8X8_UNORM_SRGB = 93
166
+ BC6H_TYPELESS = 94
167
+ BC6H_UF16 = 95
168
+ BC6H_SF16 = 96
169
+ BC7_TYPELESS = 97
170
+ BC7_UNORM = 98
171
+ BC7_UNORM_SRGB = 99
172
+ AYUV = 100
173
+ Y410 = 101
174
+ Y416 = 102
175
+ NV12 = 103
176
+ P010 = 104
177
+ P016 = 105
178
+ OPAQUE_420 = 106
179
+ YUY2 = 107
180
+ Y210 = 108
181
+ Y216 = 109
182
+ NV11 = 110
183
+ AI44 = 111
184
+ IA44 = 112
185
+ P8 = 113
186
+ A8P8 = 114
187
+ B4G4R4A4_UNORM = 115
188
+ P208 = 130
189
+ V208 = 131
190
+ V408 = 132
191
+ SAMPLER_FEEDBACK_MIN_MIP_OPAQUE = 189
192
+ SAMPLER_FEEDBACK_MIP_REGION_USED_OPAQUE = 190
193
+
194
+
195
+ class D3DFMT(IntEnum):
196
+ UNKNOWN = 0
197
+ R8G8B8 = 20
198
+ A8R8G8B8 = 21
199
+ X8R8G8B8 = 22
200
+ R5G6B5 = 23
201
+ X1R5G5B5 = 24
202
+ A1R5G5B5 = 25
203
+ A4R4G4B4 = 26
204
+ R3G3B2 = 27
205
+ A8 = 28
206
+ A8R3G3B2 = 29
207
+ X4R4G4B4 = 30
208
+ A2B10G10R10 = 31
209
+ A8B8G8R8 = 32
210
+ X8B8G8R8 = 33
211
+ G16R16 = 34
212
+ A2R10G10B10 = 35
213
+ A16B16G16R16 = 36
214
+ A8P8 = 40
215
+ P8 = 41
216
+ L8 = 50
217
+ A8L8 = 51
218
+ A4L4 = 52
219
+ V8U8 = 60
220
+ L6V5U5 = 61
221
+ X8L8V8U8 = 62
222
+ Q8W8V8U8 = 63
223
+ V16U16 = 64
224
+ A2W10V10U10 = 67
225
+ D16_LOCKABLE = 70
226
+ D32 = 71
227
+ D15S1 = 73
228
+ D24S8 = 75
229
+ D24X8 = 77
230
+ D24X4S4 = 79
231
+ D16 = 80
232
+ D32F_LOCKABLE = 82
233
+ D24FS8 = 83
234
+ D32_LOCKABLE = 84
235
+ S8_LOCKABLE = 85
236
+ L16 = 81
237
+ VERTEXDATA = 100
238
+ INDEX16 = 101
239
+ INDEX32 = 102
240
+ Q16W16V16U16 = 110
241
+ R16F = 111
242
+ G16R16F = 112
243
+ A16B16G16R16F = 113
244
+ R32F = 114
245
+ G32R32F = 115
246
+ A32B32G32R32F = 116
247
+ CxV8U8 = 117
248
+ A1 = 118
249
+ A2B10G10R10_XR_BIAS = 119
250
+ BINARYBUFFER = 199
251
+
252
+ UYVY = i32(b"UYVY")
253
+ R8G8_B8G8 = i32(b"RGBG")
254
+ YUY2 = i32(b"YUY2")
255
+ G8R8_G8B8 = i32(b"GRGB")
256
+ DXT1 = i32(b"DXT1")
257
+ DXT2 = i32(b"DXT2")
258
+ DXT3 = i32(b"DXT3")
259
+ DXT4 = i32(b"DXT4")
260
+ DXT5 = i32(b"DXT5")
261
+ DX10 = i32(b"DX10")
262
+ BC4S = i32(b"BC4S")
263
+ BC4U = i32(b"BC4U")
264
+ BC5S = i32(b"BC5S")
265
+ BC5U = i32(b"BC5U")
266
+ ATI1 = i32(b"ATI1")
267
+ ATI2 = i32(b"ATI2")
268
+ MULTI2_ARGB8 = i32(b"MET1")
269
+
270
+
271
+ # Backward compatibility layer
272
+ module = sys.modules[__name__]
273
+ for item in DDSD:
274
+ assert item.name is not None
275
+ setattr(module, f"DDSD_{item.name}", item.value)
276
+ for item1 in DDSCAPS:
277
+ assert item1.name is not None
278
+ setattr(module, f"DDSCAPS_{item1.name}", item1.value)
279
+ for item2 in DDSCAPS2:
280
+ assert item2.name is not None
281
+ setattr(module, f"DDSCAPS2_{item2.name}", item2.value)
282
+ for item3 in DDPF:
283
+ assert item3.name is not None
284
+ setattr(module, f"DDPF_{item3.name}", item3.value)
285
+
286
+ DDS_FOURCC = DDPF.FOURCC
287
+ DDS_RGB = DDPF.RGB
288
+ DDS_RGBA = DDPF.RGB | DDPF.ALPHAPIXELS
289
+ DDS_LUMINANCE = DDPF.LUMINANCE
290
+ DDS_LUMINANCEA = DDPF.LUMINANCE | DDPF.ALPHAPIXELS
291
+ DDS_ALPHA = DDPF.ALPHA
292
+ DDS_PAL8 = DDPF.PALETTEINDEXED8
293
+
294
+ DDS_HEADER_FLAGS_TEXTURE = DDSD.CAPS | DDSD.HEIGHT | DDSD.WIDTH | DDSD.PIXELFORMAT
295
+ DDS_HEADER_FLAGS_MIPMAP = DDSD.MIPMAPCOUNT
296
+ DDS_HEADER_FLAGS_VOLUME = DDSD.DEPTH
297
+ DDS_HEADER_FLAGS_PITCH = DDSD.PITCH
298
+ DDS_HEADER_FLAGS_LINEARSIZE = DDSD.LINEARSIZE
299
+
300
+ DDS_HEIGHT = DDSD.HEIGHT
301
+ DDS_WIDTH = DDSD.WIDTH
302
+
303
+ DDS_SURFACE_FLAGS_TEXTURE = DDSCAPS.TEXTURE
304
+ DDS_SURFACE_FLAGS_MIPMAP = DDSCAPS.COMPLEX | DDSCAPS.MIPMAP
305
+ DDS_SURFACE_FLAGS_CUBEMAP = DDSCAPS.COMPLEX
306
+
307
+ DDS_CUBEMAP_POSITIVEX = DDSCAPS2.CUBEMAP | DDSCAPS2.CUBEMAP_POSITIVEX
308
+ DDS_CUBEMAP_NEGATIVEX = DDSCAPS2.CUBEMAP | DDSCAPS2.CUBEMAP_NEGATIVEX
309
+ DDS_CUBEMAP_POSITIVEY = DDSCAPS2.CUBEMAP | DDSCAPS2.CUBEMAP_POSITIVEY
310
+ DDS_CUBEMAP_NEGATIVEY = DDSCAPS2.CUBEMAP | DDSCAPS2.CUBEMAP_NEGATIVEY
311
+ DDS_CUBEMAP_POSITIVEZ = DDSCAPS2.CUBEMAP | DDSCAPS2.CUBEMAP_POSITIVEZ
312
+ DDS_CUBEMAP_NEGATIVEZ = DDSCAPS2.CUBEMAP | DDSCAPS2.CUBEMAP_NEGATIVEZ
313
+
314
+ DXT1_FOURCC = D3DFMT.DXT1
315
+ DXT3_FOURCC = D3DFMT.DXT3
316
+ DXT5_FOURCC = D3DFMT.DXT5
317
+
318
+ DXGI_FORMAT_R8G8B8A8_TYPELESS = DXGI_FORMAT.R8G8B8A8_TYPELESS
319
+ DXGI_FORMAT_R8G8B8A8_UNORM = DXGI_FORMAT.R8G8B8A8_UNORM
320
+ DXGI_FORMAT_R8G8B8A8_UNORM_SRGB = DXGI_FORMAT.R8G8B8A8_UNORM_SRGB
321
+ DXGI_FORMAT_BC5_TYPELESS = DXGI_FORMAT.BC5_TYPELESS
322
+ DXGI_FORMAT_BC5_UNORM = DXGI_FORMAT.BC5_UNORM
323
+ DXGI_FORMAT_BC5_SNORM = DXGI_FORMAT.BC5_SNORM
324
+ DXGI_FORMAT_BC6H_UF16 = DXGI_FORMAT.BC6H_UF16
325
+ DXGI_FORMAT_BC6H_SF16 = DXGI_FORMAT.BC6H_SF16
326
+ DXGI_FORMAT_BC7_TYPELESS = DXGI_FORMAT.BC7_TYPELESS
327
+ DXGI_FORMAT_BC7_UNORM = DXGI_FORMAT.BC7_UNORM
328
+ DXGI_FORMAT_BC7_UNORM_SRGB = DXGI_FORMAT.BC7_UNORM_SRGB
329
+
330
+
331
+ class DdsImageFile(ImageFile.ImageFile):
332
+ format = "DDS"
333
+ format_description = "DirectDraw Surface"
334
+
335
+ def _open(self) -> None:
336
+ if not _accept(self.fp.read(4)):
337
+ msg = "not a DDS file"
338
+ raise SyntaxError(msg)
339
+ (header_size,) = struct.unpack("<I", self.fp.read(4))
340
+ if header_size != 124:
341
+ msg = f"Unsupported header size {repr(header_size)}"
342
+ raise OSError(msg)
343
+ header_bytes = self.fp.read(header_size - 4)
344
+ if len(header_bytes) != 120:
345
+ msg = f"Incomplete header: {len(header_bytes)} bytes"
346
+ raise OSError(msg)
347
+ header = io.BytesIO(header_bytes)
348
+
349
+ flags, height, width = struct.unpack("<3I", header.read(12))
350
+ self._size = (width, height)
351
+ extents = (0, 0) + self.size
352
+
353
+ pitch, depth, mipmaps = struct.unpack("<3I", header.read(12))
354
+ struct.unpack("<11I", header.read(44)) # reserved
355
+
356
+ # pixel format
357
+ pfsize, pfflags, fourcc, bitcount = struct.unpack("<4I", header.read(16))
358
+ n = 0
359
+ rawmode = None
360
+ if pfflags & DDPF.RGB:
361
+ # Texture contains uncompressed RGB data
362
+ if pfflags & DDPF.ALPHAPIXELS:
363
+ self._mode = "RGBA"
364
+ mask_count = 4
365
+ else:
366
+ self._mode = "RGB"
367
+ mask_count = 3
368
+
369
+ masks = struct.unpack(f"<{mask_count}I", header.read(mask_count * 4))
370
+ self.tile = [ImageFile._Tile("dds_rgb", extents, 0, (bitcount, masks))]
371
+ return
372
+ elif pfflags & DDPF.LUMINANCE:
373
+ if bitcount == 8:
374
+ self._mode = "L"
375
+ elif bitcount == 16 and pfflags & DDPF.ALPHAPIXELS:
376
+ self._mode = "LA"
377
+ else:
378
+ msg = f"Unsupported bitcount {bitcount} for {pfflags}"
379
+ raise OSError(msg)
380
+ elif pfflags & DDPF.PALETTEINDEXED8:
381
+ self._mode = "P"
382
+ self.palette = ImagePalette.raw("RGBA", self.fp.read(1024))
383
+ self.palette.mode = "RGBA"
384
+ elif pfflags & DDPF.FOURCC:
385
+ offset = header_size + 4
386
+ if fourcc == D3DFMT.DXT1:
387
+ self._mode = "RGBA"
388
+ self.pixel_format = "DXT1"
389
+ n = 1
390
+ elif fourcc == D3DFMT.DXT3:
391
+ self._mode = "RGBA"
392
+ self.pixel_format = "DXT3"
393
+ n = 2
394
+ elif fourcc == D3DFMT.DXT5:
395
+ self._mode = "RGBA"
396
+ self.pixel_format = "DXT5"
397
+ n = 3
398
+ elif fourcc in (D3DFMT.BC4U, D3DFMT.ATI1):
399
+ self._mode = "L"
400
+ self.pixel_format = "BC4"
401
+ n = 4
402
+ elif fourcc == D3DFMT.BC5S:
403
+ self._mode = "RGB"
404
+ self.pixel_format = "BC5S"
405
+ n = 5
406
+ elif fourcc in (D3DFMT.BC5U, D3DFMT.ATI2):
407
+ self._mode = "RGB"
408
+ self.pixel_format = "BC5"
409
+ n = 5
410
+ elif fourcc == D3DFMT.DX10:
411
+ offset += 20
412
+ # ignoring flags which pertain to volume textures and cubemaps
413
+ (dxgi_format,) = struct.unpack("<I", self.fp.read(4))
414
+ self.fp.read(16)
415
+ if dxgi_format in (
416
+ DXGI_FORMAT.BC1_UNORM,
417
+ DXGI_FORMAT.BC1_TYPELESS,
418
+ ):
419
+ self._mode = "RGBA"
420
+ self.pixel_format = "BC1"
421
+ n = 1
422
+ elif dxgi_format in (DXGI_FORMAT.BC2_TYPELESS, DXGI_FORMAT.BC2_UNORM):
423
+ self._mode = "RGBA"
424
+ self.pixel_format = "BC2"
425
+ n = 2
426
+ elif dxgi_format in (DXGI_FORMAT.BC3_TYPELESS, DXGI_FORMAT.BC3_UNORM):
427
+ self._mode = "RGBA"
428
+ self.pixel_format = "BC3"
429
+ n = 3
430
+ elif dxgi_format in (DXGI_FORMAT.BC4_TYPELESS, DXGI_FORMAT.BC4_UNORM):
431
+ self._mode = "L"
432
+ self.pixel_format = "BC4"
433
+ n = 4
434
+ elif dxgi_format in (DXGI_FORMAT.BC5_TYPELESS, DXGI_FORMAT.BC5_UNORM):
435
+ self._mode = "RGB"
436
+ self.pixel_format = "BC5"
437
+ n = 5
438
+ elif dxgi_format == DXGI_FORMAT.BC5_SNORM:
439
+ self._mode = "RGB"
440
+ self.pixel_format = "BC5S"
441
+ n = 5
442
+ elif dxgi_format == DXGI_FORMAT.BC6H_UF16:
443
+ self._mode = "RGB"
444
+ self.pixel_format = "BC6H"
445
+ n = 6
446
+ elif dxgi_format == DXGI_FORMAT.BC6H_SF16:
447
+ self._mode = "RGB"
448
+ self.pixel_format = "BC6HS"
449
+ n = 6
450
+ elif dxgi_format in (
451
+ DXGI_FORMAT.BC7_TYPELESS,
452
+ DXGI_FORMAT.BC7_UNORM,
453
+ DXGI_FORMAT.BC7_UNORM_SRGB,
454
+ ):
455
+ self._mode = "RGBA"
456
+ self.pixel_format = "BC7"
457
+ n = 7
458
+ if dxgi_format == DXGI_FORMAT.BC7_UNORM_SRGB:
459
+ self.info["gamma"] = 1 / 2.2
460
+ elif dxgi_format in (
461
+ DXGI_FORMAT.R8G8B8A8_TYPELESS,
462
+ DXGI_FORMAT.R8G8B8A8_UNORM,
463
+ DXGI_FORMAT.R8G8B8A8_UNORM_SRGB,
464
+ ):
465
+ self._mode = "RGBA"
466
+ if dxgi_format == DXGI_FORMAT.R8G8B8A8_UNORM_SRGB:
467
+ self.info["gamma"] = 1 / 2.2
468
+ else:
469
+ msg = f"Unimplemented DXGI format {dxgi_format}"
470
+ raise NotImplementedError(msg)
471
+ else:
472
+ msg = f"Unimplemented pixel format {repr(fourcc)}"
473
+ raise NotImplementedError(msg)
474
+ else:
475
+ msg = f"Unknown pixel format flags {pfflags}"
476
+ raise NotImplementedError(msg)
477
+
478
+ if n:
479
+ self.tile = [
480
+ ImageFile._Tile("bcn", extents, offset, (n, self.pixel_format))
481
+ ]
482
+ else:
483
+ self.tile = [ImageFile._Tile("raw", extents, 0, rawmode or self.mode)]
484
+
485
+ def load_seek(self, pos: int) -> None:
486
+ pass
487
+
488
+
489
+ class DdsRgbDecoder(ImageFile.PyDecoder):
490
+ _pulls_fd = True
491
+
492
+ def decode(self, buffer: bytes | Image.SupportsArrayInterface) -> tuple[int, int]:
493
+ assert self.fd is not None
494
+ bitcount, masks = self.args
495
+
496
+ # Some masks will be padded with zeros, e.g. R 0b11 G 0b1100
497
+ # Calculate how many zeros each mask is padded with
498
+ mask_offsets = []
499
+ # And the maximum value of each channel without the padding
500
+ mask_totals = []
501
+ for mask in masks:
502
+ offset = 0
503
+ if mask != 0:
504
+ while mask >> (offset + 1) << (offset + 1) == mask:
505
+ offset += 1
506
+ mask_offsets.append(offset)
507
+ mask_totals.append(mask >> offset)
508
+
509
+ data = bytearray()
510
+ bytecount = bitcount // 8
511
+ dest_length = self.state.xsize * self.state.ysize * len(masks)
512
+ while len(data) < dest_length:
513
+ value = int.from_bytes(self.fd.read(bytecount), "little")
514
+ for i, mask in enumerate(masks):
515
+ masked_value = value & mask
516
+ # Remove the zero padding, and scale it to 8 bits
517
+ data += o8(
518
+ int(((masked_value >> mask_offsets[i]) / mask_totals[i]) * 255)
519
+ )
520
+ self.set_as_raw(data)
521
+ return -1, 0
522
+
523
+
524
+ def _save(im: Image.Image, fp: IO[bytes], filename: str | bytes) -> None:
525
+ if im.mode not in ("RGB", "RGBA", "L", "LA"):
526
+ msg = f"cannot write mode {im.mode} as DDS"
527
+ raise OSError(msg)
528
+
529
+ flags = DDSD.CAPS | DDSD.HEIGHT | DDSD.WIDTH | DDSD.PIXELFORMAT
530
+ bitcount = len(im.getbands()) * 8
531
+ pixel_format = im.encoderinfo.get("pixel_format")
532
+ args: tuple[int] | str
533
+ if pixel_format:
534
+ codec_name = "bcn"
535
+ flags |= DDSD.LINEARSIZE
536
+ pitch = (im.width + 3) * 4
537
+ rgba_mask = [0, 0, 0, 0]
538
+ pixel_flags = DDPF.FOURCC
539
+ if pixel_format == "DXT1":
540
+ fourcc = D3DFMT.DXT1
541
+ args = (1,)
542
+ elif pixel_format == "DXT3":
543
+ fourcc = D3DFMT.DXT3
544
+ args = (2,)
545
+ elif pixel_format == "DXT5":
546
+ fourcc = D3DFMT.DXT5
547
+ args = (3,)
548
+ else:
549
+ fourcc = D3DFMT.DX10
550
+ if pixel_format == "BC2":
551
+ args = (2,)
552
+ dxgi_format = DXGI_FORMAT.BC2_TYPELESS
553
+ elif pixel_format == "BC3":
554
+ args = (3,)
555
+ dxgi_format = DXGI_FORMAT.BC3_TYPELESS
556
+ elif pixel_format == "BC5":
557
+ args = (5,)
558
+ dxgi_format = DXGI_FORMAT.BC5_TYPELESS
559
+ if im.mode != "RGB":
560
+ msg = "only RGB mode can be written as BC5"
561
+ raise OSError(msg)
562
+ else:
563
+ msg = f"cannot write pixel format {pixel_format}"
564
+ raise OSError(msg)
565
+ else:
566
+ codec_name = "raw"
567
+ flags |= DDSD.PITCH
568
+ pitch = (im.width * bitcount + 7) // 8
569
+
570
+ alpha = im.mode[-1] == "A"
571
+ if im.mode[0] == "L":
572
+ pixel_flags = DDPF.LUMINANCE
573
+ args = im.mode
574
+ if alpha:
575
+ rgba_mask = [0x000000FF, 0x000000FF, 0x000000FF]
576
+ else:
577
+ rgba_mask = [0xFF000000, 0xFF000000, 0xFF000000]
578
+ else:
579
+ pixel_flags = DDPF.RGB
580
+ args = im.mode[::-1]
581
+ rgba_mask = [0x00FF0000, 0x0000FF00, 0x000000FF]
582
+
583
+ if alpha:
584
+ r, g, b, a = im.split()
585
+ im = Image.merge("RGBA", (a, r, g, b))
586
+ if alpha:
587
+ pixel_flags |= DDPF.ALPHAPIXELS
588
+ rgba_mask.append(0xFF000000 if alpha else 0)
589
+
590
+ fourcc = D3DFMT.UNKNOWN
591
+ fp.write(
592
+ o32(DDS_MAGIC)
593
+ + struct.pack(
594
+ "<7I",
595
+ 124, # header size
596
+ flags, # flags
597
+ im.height,
598
+ im.width,
599
+ pitch,
600
+ 0, # depth
601
+ 0, # mipmaps
602
+ )
603
+ + struct.pack("11I", *((0,) * 11)) # reserved
604
+ # pfsize, pfflags, fourcc, bitcount
605
+ + struct.pack("<4I", 32, pixel_flags, fourcc, bitcount)
606
+ + struct.pack("<4I", *rgba_mask) # dwRGBABitMask
607
+ + struct.pack("<5I", DDSCAPS.TEXTURE, 0, 0, 0, 0)
608
+ )
609
+ if fourcc == D3DFMT.DX10:
610
+ fp.write(
611
+ # dxgi_format, 2D resource, misc, array size, straight alpha
612
+ struct.pack("<5I", dxgi_format, 3, 0, 0, 1)
613
+ )
614
+ ImageFile._save(im, fp, [ImageFile._Tile(codec_name, (0, 0) + im.size, 0, args)])
615
+
616
+
617
+ def _accept(prefix: bytes) -> bool:
618
+ return prefix.startswith(b"DDS ")
619
+
620
+
621
+ Image.register_open(DdsImageFile.format, DdsImageFile, _accept)
622
+ Image.register_decoder("dds_rgb", DdsRgbDecoder)
623
+ Image.register_save(DdsImageFile.format, _save)
624
+ Image.register_extension(DdsImageFile.format, ".dds")
.venv/Lib/site-packages/PIL/EpsImagePlugin.py ADDED
@@ -0,0 +1,479 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #
2
+ # The Python Imaging Library.
3
+ # $Id$
4
+ #
5
+ # EPS file handling
6
+ #
7
+ # History:
8
+ # 1995-09-01 fl Created (0.1)
9
+ # 1996-05-18 fl Don't choke on "atend" fields, Ghostscript interface (0.2)
10
+ # 1996-08-22 fl Don't choke on floating point BoundingBox values
11
+ # 1996-08-23 fl Handle files from Macintosh (0.3)
12
+ # 2001-02-17 fl Use 're' instead of 'regex' (Python 2.1) (0.4)
13
+ # 2003-09-07 fl Check gs.close status (from Federico Di Gregorio) (0.5)
14
+ # 2014-05-07 e Handling of EPS with binary preview and fixed resolution
15
+ # resizing
16
+ #
17
+ # Copyright (c) 1997-2003 by Secret Labs AB.
18
+ # Copyright (c) 1995-2003 by Fredrik Lundh
19
+ #
20
+ # See the README file for information on usage and redistribution.
21
+ #
22
+ from __future__ import annotations
23
+
24
+ import io
25
+ import os
26
+ import re
27
+ import subprocess
28
+ import sys
29
+ import tempfile
30
+ from typing import IO
31
+
32
+ from . import Image, ImageFile
33
+ from ._binary import i32le as i32
34
+
35
+ # --------------------------------------------------------------------
36
+
37
+
38
+ split = re.compile(r"^%%([^:]*):[ \t]*(.*)[ \t]*$")
39
+ field = re.compile(r"^%[%!\w]([^:]*)[ \t]*$")
40
+
41
+ gs_binary: str | bool | None = None
42
+ gs_windows_binary = None
43
+
44
+
45
+ def has_ghostscript() -> bool:
46
+ global gs_binary, gs_windows_binary
47
+ if gs_binary is None:
48
+ if sys.platform.startswith("win"):
49
+ if gs_windows_binary is None:
50
+ import shutil
51
+
52
+ for binary in ("gswin32c", "gswin64c", "gs"):
53
+ if shutil.which(binary) is not None:
54
+ gs_windows_binary = binary
55
+ break
56
+ else:
57
+ gs_windows_binary = False
58
+ gs_binary = gs_windows_binary
59
+ else:
60
+ try:
61
+ subprocess.check_call(["gs", "--version"], stdout=subprocess.DEVNULL)
62
+ gs_binary = "gs"
63
+ except OSError:
64
+ gs_binary = False
65
+ return gs_binary is not False
66
+
67
+
68
+ def Ghostscript(
69
+ tile: list[ImageFile._Tile],
70
+ size: tuple[int, int],
71
+ fp: IO[bytes],
72
+ scale: int = 1,
73
+ transparency: bool = False,
74
+ ) -> Image.core.ImagingCore:
75
+ """Render an image using Ghostscript"""
76
+ global gs_binary
77
+ if not has_ghostscript():
78
+ msg = "Unable to locate Ghostscript on paths"
79
+ raise OSError(msg)
80
+ assert isinstance(gs_binary, str)
81
+
82
+ # Unpack decoder tile
83
+ args = tile[0].args
84
+ assert isinstance(args, tuple)
85
+ length, bbox = args
86
+
87
+ # Hack to support hi-res rendering
88
+ scale = int(scale) or 1
89
+ width = size[0] * scale
90
+ height = size[1] * scale
91
+ # resolution is dependent on bbox and size
92
+ res_x = 72.0 * width / (bbox[2] - bbox[0])
93
+ res_y = 72.0 * height / (bbox[3] - bbox[1])
94
+
95
+ out_fd, outfile = tempfile.mkstemp()
96
+ os.close(out_fd)
97
+
98
+ infile_temp = None
99
+ if hasattr(fp, "name") and os.path.exists(fp.name):
100
+ infile = fp.name
101
+ else:
102
+ in_fd, infile_temp = tempfile.mkstemp()
103
+ os.close(in_fd)
104
+ infile = infile_temp
105
+
106
+ # Ignore length and offset!
107
+ # Ghostscript can read it
108
+ # Copy whole file to read in Ghostscript
109
+ with open(infile_temp, "wb") as f:
110
+ # fetch length of fp
111
+ fp.seek(0, io.SEEK_END)
112
+ fsize = fp.tell()
113
+ # ensure start position
114
+ # go back
115
+ fp.seek(0)
116
+ lengthfile = fsize
117
+ while lengthfile > 0:
118
+ s = fp.read(min(lengthfile, 100 * 1024))
119
+ if not s:
120
+ break
121
+ lengthfile -= len(s)
122
+ f.write(s)
123
+
124
+ if transparency:
125
+ # "RGBA"
126
+ device = "pngalpha"
127
+ else:
128
+ # "pnmraw" automatically chooses between
129
+ # PBM ("1"), PGM ("L"), and PPM ("RGB").
130
+ device = "pnmraw"
131
+
132
+ # Build Ghostscript command
133
+ command = [
134
+ gs_binary,
135
+ "-q", # quiet mode
136
+ f"-g{width:d}x{height:d}", # set output geometry (pixels)
137
+ f"-r{res_x:f}x{res_y:f}", # set input DPI (dots per inch)
138
+ "-dBATCH", # exit after processing
139
+ "-dNOPAUSE", # don't pause between pages
140
+ "-dSAFER", # safe mode
141
+ f"-sDEVICE={device}",
142
+ f"-sOutputFile={outfile}", # output file
143
+ # adjust for image origin
144
+ "-c",
145
+ f"{-bbox[0]} {-bbox[1]} translate",
146
+ "-f",
147
+ infile, # input file
148
+ # showpage (see https://bugs.ghostscript.com/show_bug.cgi?id=698272)
149
+ "-c",
150
+ "showpage",
151
+ ]
152
+
153
+ # push data through Ghostscript
154
+ try:
155
+ startupinfo = None
156
+ if sys.platform.startswith("win"):
157
+ startupinfo = subprocess.STARTUPINFO()
158
+ startupinfo.dwFlags |= subprocess.STARTF_USESHOWWINDOW
159
+ subprocess.check_call(command, startupinfo=startupinfo)
160
+ with Image.open(outfile) as out_im:
161
+ out_im.load()
162
+ return out_im.im.copy()
163
+ finally:
164
+ try:
165
+ os.unlink(outfile)
166
+ if infile_temp:
167
+ os.unlink(infile_temp)
168
+ except OSError:
169
+ pass
170
+
171
+
172
+ def _accept(prefix: bytes) -> bool:
173
+ return prefix.startswith(b"%!PS") or (
174
+ len(prefix) >= 4 and i32(prefix) == 0xC6D3D0C5
175
+ )
176
+
177
+
178
+ ##
179
+ # Image plugin for Encapsulated PostScript. This plugin supports only
180
+ # a few variants of this format.
181
+
182
+
183
+ class EpsImageFile(ImageFile.ImageFile):
184
+ """EPS File Parser for the Python Imaging Library"""
185
+
186
+ format = "EPS"
187
+ format_description = "Encapsulated Postscript"
188
+
189
+ mode_map = {1: "L", 2: "LAB", 3: "RGB", 4: "CMYK"}
190
+
191
+ def _open(self) -> None:
192
+ (length, offset) = self._find_offset(self.fp)
193
+
194
+ # go to offset - start of "%!PS"
195
+ self.fp.seek(offset)
196
+
197
+ self._mode = "RGB"
198
+
199
+ # When reading header comments, the first comment is used.
200
+ # When reading trailer comments, the last comment is used.
201
+ bounding_box: list[int] | None = None
202
+ imagedata_size: tuple[int, int] | None = None
203
+
204
+ byte_arr = bytearray(255)
205
+ bytes_mv = memoryview(byte_arr)
206
+ bytes_read = 0
207
+ reading_header_comments = True
208
+ reading_trailer_comments = False
209
+ trailer_reached = False
210
+
211
+ def check_required_header_comments() -> None:
212
+ """
213
+ The EPS specification requires that some headers exist.
214
+ This should be checked when the header comments formally end,
215
+ when image data starts, or when the file ends, whichever comes first.
216
+ """
217
+ if "PS-Adobe" not in self.info:
218
+ msg = 'EPS header missing "%!PS-Adobe" comment'
219
+ raise SyntaxError(msg)
220
+ if "BoundingBox" not in self.info:
221
+ msg = 'EPS header missing "%%BoundingBox" comment'
222
+ raise SyntaxError(msg)
223
+
224
+ def read_comment(s: str) -> bool:
225
+ nonlocal bounding_box, reading_trailer_comments
226
+ try:
227
+ m = split.match(s)
228
+ except re.error as e:
229
+ msg = "not an EPS file"
230
+ raise SyntaxError(msg) from e
231
+
232
+ if not m:
233
+ return False
234
+
235
+ k, v = m.group(1, 2)
236
+ self.info[k] = v
237
+ if k == "BoundingBox":
238
+ if v == "(atend)":
239
+ reading_trailer_comments = True
240
+ elif not bounding_box or (trailer_reached and reading_trailer_comments):
241
+ try:
242
+ # Note: The DSC spec says that BoundingBox
243
+ # fields should be integers, but some drivers
244
+ # put floating point values there anyway.
245
+ bounding_box = [int(float(i)) for i in v.split()]
246
+ except Exception:
247
+ pass
248
+ return True
249
+
250
+ while True:
251
+ byte = self.fp.read(1)
252
+ if byte == b"":
253
+ # if we didn't read a byte we must be at the end of the file
254
+ if bytes_read == 0:
255
+ if reading_header_comments:
256
+ check_required_header_comments()
257
+ break
258
+ elif byte in b"\r\n":
259
+ # if we read a line ending character, ignore it and parse what
260
+ # we have already read. if we haven't read any other characters,
261
+ # continue reading
262
+ if bytes_read == 0:
263
+ continue
264
+ else:
265
+ # ASCII/hexadecimal lines in an EPS file must not exceed
266
+ # 255 characters, not including line ending characters
267
+ if bytes_read >= 255:
268
+ # only enforce this for lines starting with a "%",
269
+ # otherwise assume it's binary data
270
+ if byte_arr[0] == ord("%"):
271
+ msg = "not an EPS file"
272
+ raise SyntaxError(msg)
273
+ else:
274
+ if reading_header_comments:
275
+ check_required_header_comments()
276
+ reading_header_comments = False
277
+ # reset bytes_read so we can keep reading
278
+ # data until the end of the line
279
+ bytes_read = 0
280
+ byte_arr[bytes_read] = byte[0]
281
+ bytes_read += 1
282
+ continue
283
+
284
+ if reading_header_comments:
285
+ # Load EPS header
286
+
287
+ # if this line doesn't start with a "%",
288
+ # or does start with "%%EndComments",
289
+ # then we've reached the end of the header/comments
290
+ if byte_arr[0] != ord("%") or bytes_mv[:13] == b"%%EndComments":
291
+ check_required_header_comments()
292
+ reading_header_comments = False
293
+ continue
294
+
295
+ s = str(bytes_mv[:bytes_read], "latin-1")
296
+ if not read_comment(s):
297
+ m = field.match(s)
298
+ if m:
299
+ k = m.group(1)
300
+ if k.startswith("PS-Adobe"):
301
+ self.info["PS-Adobe"] = k[9:]
302
+ else:
303
+ self.info[k] = ""
304
+ elif s[0] == "%":
305
+ # handle non-DSC PostScript comments that some
306
+ # tools mistakenly put in the Comments section
307
+ pass
308
+ else:
309
+ msg = "bad EPS header"
310
+ raise OSError(msg)
311
+ elif bytes_mv[:11] == b"%ImageData:":
312
+ # Check for an "ImageData" descriptor
313
+ # https://www.adobe.com/devnet-apps/photoshop/fileformatashtml/#50577413_pgfId-1035096
314
+
315
+ # If we've already read an "ImageData" descriptor,
316
+ # don't read another one.
317
+ if imagedata_size:
318
+ bytes_read = 0
319
+ continue
320
+
321
+ # Values:
322
+ # columns
323
+ # rows
324
+ # bit depth (1 or 8)
325
+ # mode (1: L, 2: LAB, 3: RGB, 4: CMYK)
326
+ # number of padding channels
327
+ # block size (number of bytes per row per channel)
328
+ # binary/ascii (1: binary, 2: ascii)
329
+ # data start identifier (the image data follows after a single line
330
+ # consisting only of this quoted value)
331
+ image_data_values = byte_arr[11:bytes_read].split(None, 7)
332
+ columns, rows, bit_depth, mode_id = (
333
+ int(value) for value in image_data_values[:4]
334
+ )
335
+
336
+ if bit_depth == 1:
337
+ self._mode = "1"
338
+ elif bit_depth == 8:
339
+ try:
340
+ self._mode = self.mode_map[mode_id]
341
+ except ValueError:
342
+ break
343
+ else:
344
+ break
345
+
346
+ # Parse the columns and rows after checking the bit depth and mode
347
+ # in case the bit depth and/or mode are invalid.
348
+ imagedata_size = columns, rows
349
+ elif bytes_mv[:5] == b"%%EOF":
350
+ break
351
+ elif trailer_reached and reading_trailer_comments:
352
+ # Load EPS trailer
353
+ s = str(bytes_mv[:bytes_read], "latin-1")
354
+ read_comment(s)
355
+ elif bytes_mv[:9] == b"%%Trailer":
356
+ trailer_reached = True
357
+ elif bytes_mv[:14] == b"%%BeginBinary:":
358
+ bytecount = int(byte_arr[14:bytes_read])
359
+ self.fp.seek(bytecount, os.SEEK_CUR)
360
+ bytes_read = 0
361
+
362
+ # A "BoundingBox" is always required,
363
+ # even if an "ImageData" descriptor size exists.
364
+ if not bounding_box:
365
+ msg = "cannot determine EPS bounding box"
366
+ raise OSError(msg)
367
+
368
+ # An "ImageData" size takes precedence over the "BoundingBox".
369
+ self._size = imagedata_size or (
370
+ bounding_box[2] - bounding_box[0],
371
+ bounding_box[3] - bounding_box[1],
372
+ )
373
+
374
+ self.tile = [
375
+ ImageFile._Tile("eps", (0, 0) + self.size, offset, (length, bounding_box))
376
+ ]
377
+
378
+ def _find_offset(self, fp: IO[bytes]) -> tuple[int, int]:
379
+ s = fp.read(4)
380
+
381
+ if s == b"%!PS":
382
+ # for HEAD without binary preview
383
+ fp.seek(0, io.SEEK_END)
384
+ length = fp.tell()
385
+ offset = 0
386
+ elif i32(s) == 0xC6D3D0C5:
387
+ # FIX for: Some EPS file not handled correctly / issue #302
388
+ # EPS can contain binary data
389
+ # or start directly with latin coding
390
+ # more info see:
391
+ # https://web.archive.org/web/20160528181353/http://partners.adobe.com/public/developer/en/ps/5002.EPSF_Spec.pdf
392
+ s = fp.read(8)
393
+ offset = i32(s)
394
+ length = i32(s, 4)
395
+ else:
396
+ msg = "not an EPS file"
397
+ raise SyntaxError(msg)
398
+
399
+ return length, offset
400
+
401
+ def load(
402
+ self, scale: int = 1, transparency: bool = False
403
+ ) -> Image.core.PixelAccess | None:
404
+ # Load EPS via Ghostscript
405
+ if self.tile:
406
+ self.im = Ghostscript(self.tile, self.size, self.fp, scale, transparency)
407
+ self._mode = self.im.mode
408
+ self._size = self.im.size
409
+ self.tile = []
410
+ return Image.Image.load(self)
411
+
412
+ def load_seek(self, pos: int) -> None:
413
+ # we can't incrementally load, so force ImageFile.parser to
414
+ # use our custom load method by defining this method.
415
+ pass
416
+
417
+
418
+ # --------------------------------------------------------------------
419
+
420
+
421
+ def _save(im: Image.Image, fp: IO[bytes], filename: str | bytes, eps: int = 1) -> None:
422
+ """EPS Writer for the Python Imaging Library."""
423
+
424
+ # make sure image data is available
425
+ im.load()
426
+
427
+ # determine PostScript image mode
428
+ if im.mode == "L":
429
+ operator = (8, 1, b"image")
430
+ elif im.mode == "RGB":
431
+ operator = (8, 3, b"false 3 colorimage")
432
+ elif im.mode == "CMYK":
433
+ operator = (8, 4, b"false 4 colorimage")
434
+ else:
435
+ msg = "image mode is not supported"
436
+ raise ValueError(msg)
437
+
438
+ if eps:
439
+ # write EPS header
440
+ fp.write(b"%!PS-Adobe-3.0 EPSF-3.0\n")
441
+ fp.write(b"%%Creator: PIL 0.1 EpsEncode\n")
442
+ # fp.write("%%CreationDate: %s"...)
443
+ fp.write(b"%%%%BoundingBox: 0 0 %d %d\n" % im.size)
444
+ fp.write(b"%%Pages: 1\n")
445
+ fp.write(b"%%EndComments\n")
446
+ fp.write(b"%%Page: 1 1\n")
447
+ fp.write(b"%%ImageData: %d %d " % im.size)
448
+ fp.write(b'%d %d 0 1 1 "%s"\n' % operator)
449
+
450
+ # image header
451
+ fp.write(b"gsave\n")
452
+ fp.write(b"10 dict begin\n")
453
+ fp.write(b"/buf %d string def\n" % (im.size[0] * operator[1]))
454
+ fp.write(b"%d %d scale\n" % im.size)
455
+ fp.write(b"%d %d 8\n" % im.size) # <= bits
456
+ fp.write(b"[%d 0 0 -%d 0 %d]\n" % (im.size[0], im.size[1], im.size[1]))
457
+ fp.write(b"{ currentfile buf readhexstring pop } bind\n")
458
+ fp.write(operator[2] + b"\n")
459
+ if hasattr(fp, "flush"):
460
+ fp.flush()
461
+
462
+ ImageFile._save(im, fp, [ImageFile._Tile("eps", (0, 0) + im.size)])
463
+
464
+ fp.write(b"\n%%%%EndBinary\n")
465
+ fp.write(b"grestore end\n")
466
+ if hasattr(fp, "flush"):
467
+ fp.flush()
468
+
469
+
470
+ # --------------------------------------------------------------------
471
+
472
+
473
+ Image.register_open(EpsImageFile.format, EpsImageFile, _accept)
474
+
475
+ Image.register_save(EpsImageFile.format, _save)
476
+
477
+ Image.register_extensions(EpsImageFile.format, [".ps", ".eps"])
478
+
479
+ Image.register_mime(EpsImageFile.format, "application/postscript")
.venv/Lib/site-packages/PIL/ExifTags.py ADDED
@@ -0,0 +1,382 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #
2
+ # The Python Imaging Library.
3
+ # $Id$
4
+ #
5
+ # EXIF tags
6
+ #
7
+ # Copyright (c) 2003 by Secret Labs AB
8
+ #
9
+ # See the README file for information on usage and redistribution.
10
+ #
11
+
12
+ """
13
+ This module provides constants and clear-text names for various
14
+ well-known EXIF tags.
15
+ """
16
+ from __future__ import annotations
17
+
18
+ from enum import IntEnum
19
+
20
+
21
+ class Base(IntEnum):
22
+ # possibly incomplete
23
+ InteropIndex = 0x0001
24
+ ProcessingSoftware = 0x000B
25
+ NewSubfileType = 0x00FE
26
+ SubfileType = 0x00FF
27
+ ImageWidth = 0x0100
28
+ ImageLength = 0x0101
29
+ BitsPerSample = 0x0102
30
+ Compression = 0x0103
31
+ PhotometricInterpretation = 0x0106
32
+ Thresholding = 0x0107
33
+ CellWidth = 0x0108
34
+ CellLength = 0x0109
35
+ FillOrder = 0x010A
36
+ DocumentName = 0x010D
37
+ ImageDescription = 0x010E
38
+ Make = 0x010F
39
+ Model = 0x0110
40
+ StripOffsets = 0x0111
41
+ Orientation = 0x0112
42
+ SamplesPerPixel = 0x0115
43
+ RowsPerStrip = 0x0116
44
+ StripByteCounts = 0x0117
45
+ MinSampleValue = 0x0118
46
+ MaxSampleValue = 0x0119
47
+ XResolution = 0x011A
48
+ YResolution = 0x011B
49
+ PlanarConfiguration = 0x011C
50
+ PageName = 0x011D
51
+ FreeOffsets = 0x0120
52
+ FreeByteCounts = 0x0121
53
+ GrayResponseUnit = 0x0122
54
+ GrayResponseCurve = 0x0123
55
+ T4Options = 0x0124
56
+ T6Options = 0x0125
57
+ ResolutionUnit = 0x0128
58
+ PageNumber = 0x0129
59
+ TransferFunction = 0x012D
60
+ Software = 0x0131
61
+ DateTime = 0x0132
62
+ Artist = 0x013B
63
+ HostComputer = 0x013C
64
+ Predictor = 0x013D
65
+ WhitePoint = 0x013E
66
+ PrimaryChromaticities = 0x013F
67
+ ColorMap = 0x0140
68
+ HalftoneHints = 0x0141
69
+ TileWidth = 0x0142
70
+ TileLength = 0x0143
71
+ TileOffsets = 0x0144
72
+ TileByteCounts = 0x0145
73
+ SubIFDs = 0x014A
74
+ InkSet = 0x014C
75
+ InkNames = 0x014D
76
+ NumberOfInks = 0x014E
77
+ DotRange = 0x0150
78
+ TargetPrinter = 0x0151
79
+ ExtraSamples = 0x0152
80
+ SampleFormat = 0x0153
81
+ SMinSampleValue = 0x0154
82
+ SMaxSampleValue = 0x0155
83
+ TransferRange = 0x0156
84
+ ClipPath = 0x0157
85
+ XClipPathUnits = 0x0158
86
+ YClipPathUnits = 0x0159
87
+ Indexed = 0x015A
88
+ JPEGTables = 0x015B
89
+ OPIProxy = 0x015F
90
+ JPEGProc = 0x0200
91
+ JpegIFOffset = 0x0201
92
+ JpegIFByteCount = 0x0202
93
+ JpegRestartInterval = 0x0203
94
+ JpegLosslessPredictors = 0x0205
95
+ JpegPointTransforms = 0x0206
96
+ JpegQTables = 0x0207
97
+ JpegDCTables = 0x0208
98
+ JpegACTables = 0x0209
99
+ YCbCrCoefficients = 0x0211
100
+ YCbCrSubSampling = 0x0212
101
+ YCbCrPositioning = 0x0213
102
+ ReferenceBlackWhite = 0x0214
103
+ XMLPacket = 0x02BC
104
+ RelatedImageFileFormat = 0x1000
105
+ RelatedImageWidth = 0x1001
106
+ RelatedImageLength = 0x1002
107
+ Rating = 0x4746
108
+ RatingPercent = 0x4749
109
+ ImageID = 0x800D
110
+ CFARepeatPatternDim = 0x828D
111
+ BatteryLevel = 0x828F
112
+ Copyright = 0x8298
113
+ ExposureTime = 0x829A
114
+ FNumber = 0x829D
115
+ IPTCNAA = 0x83BB
116
+ ImageResources = 0x8649
117
+ ExifOffset = 0x8769
118
+ InterColorProfile = 0x8773
119
+ ExposureProgram = 0x8822
120
+ SpectralSensitivity = 0x8824
121
+ GPSInfo = 0x8825
122
+ ISOSpeedRatings = 0x8827
123
+ OECF = 0x8828
124
+ Interlace = 0x8829
125
+ TimeZoneOffset = 0x882A
126
+ SelfTimerMode = 0x882B
127
+ SensitivityType = 0x8830
128
+ StandardOutputSensitivity = 0x8831
129
+ RecommendedExposureIndex = 0x8832
130
+ ISOSpeed = 0x8833
131
+ ISOSpeedLatitudeyyy = 0x8834
132
+ ISOSpeedLatitudezzz = 0x8835
133
+ ExifVersion = 0x9000
134
+ DateTimeOriginal = 0x9003
135
+ DateTimeDigitized = 0x9004
136
+ OffsetTime = 0x9010
137
+ OffsetTimeOriginal = 0x9011
138
+ OffsetTimeDigitized = 0x9012
139
+ ComponentsConfiguration = 0x9101
140
+ CompressedBitsPerPixel = 0x9102
141
+ ShutterSpeedValue = 0x9201
142
+ ApertureValue = 0x9202
143
+ BrightnessValue = 0x9203
144
+ ExposureBiasValue = 0x9204
145
+ MaxApertureValue = 0x9205
146
+ SubjectDistance = 0x9206
147
+ MeteringMode = 0x9207
148
+ LightSource = 0x9208
149
+ Flash = 0x9209
150
+ FocalLength = 0x920A
151
+ Noise = 0x920D
152
+ ImageNumber = 0x9211
153
+ SecurityClassification = 0x9212
154
+ ImageHistory = 0x9213
155
+ TIFFEPStandardID = 0x9216
156
+ MakerNote = 0x927C
157
+ UserComment = 0x9286
158
+ SubsecTime = 0x9290
159
+ SubsecTimeOriginal = 0x9291
160
+ SubsecTimeDigitized = 0x9292
161
+ AmbientTemperature = 0x9400
162
+ Humidity = 0x9401
163
+ Pressure = 0x9402
164
+ WaterDepth = 0x9403
165
+ Acceleration = 0x9404
166
+ CameraElevationAngle = 0x9405
167
+ XPTitle = 0x9C9B
168
+ XPComment = 0x9C9C
169
+ XPAuthor = 0x9C9D
170
+ XPKeywords = 0x9C9E
171
+ XPSubject = 0x9C9F
172
+ FlashPixVersion = 0xA000
173
+ ColorSpace = 0xA001
174
+ ExifImageWidth = 0xA002
175
+ ExifImageHeight = 0xA003
176
+ RelatedSoundFile = 0xA004
177
+ ExifInteroperabilityOffset = 0xA005
178
+ FlashEnergy = 0xA20B
179
+ SpatialFrequencyResponse = 0xA20C
180
+ FocalPlaneXResolution = 0xA20E
181
+ FocalPlaneYResolution = 0xA20F
182
+ FocalPlaneResolutionUnit = 0xA210
183
+ SubjectLocation = 0xA214
184
+ ExposureIndex = 0xA215
185
+ SensingMethod = 0xA217
186
+ FileSource = 0xA300
187
+ SceneType = 0xA301
188
+ CFAPattern = 0xA302
189
+ CustomRendered = 0xA401
190
+ ExposureMode = 0xA402
191
+ WhiteBalance = 0xA403
192
+ DigitalZoomRatio = 0xA404
193
+ FocalLengthIn35mmFilm = 0xA405
194
+ SceneCaptureType = 0xA406
195
+ GainControl = 0xA407
196
+ Contrast = 0xA408
197
+ Saturation = 0xA409
198
+ Sharpness = 0xA40A
199
+ DeviceSettingDescription = 0xA40B
200
+ SubjectDistanceRange = 0xA40C
201
+ ImageUniqueID = 0xA420
202
+ CameraOwnerName = 0xA430
203
+ BodySerialNumber = 0xA431
204
+ LensSpecification = 0xA432
205
+ LensMake = 0xA433
206
+ LensModel = 0xA434
207
+ LensSerialNumber = 0xA435
208
+ CompositeImage = 0xA460
209
+ CompositeImageCount = 0xA461
210
+ CompositeImageExposureTimes = 0xA462
211
+ Gamma = 0xA500
212
+ PrintImageMatching = 0xC4A5
213
+ DNGVersion = 0xC612
214
+ DNGBackwardVersion = 0xC613
215
+ UniqueCameraModel = 0xC614
216
+ LocalizedCameraModel = 0xC615
217
+ CFAPlaneColor = 0xC616
218
+ CFALayout = 0xC617
219
+ LinearizationTable = 0xC618
220
+ BlackLevelRepeatDim = 0xC619
221
+ BlackLevel = 0xC61A
222
+ BlackLevelDeltaH = 0xC61B
223
+ BlackLevelDeltaV = 0xC61C
224
+ WhiteLevel = 0xC61D
225
+ DefaultScale = 0xC61E
226
+ DefaultCropOrigin = 0xC61F
227
+ DefaultCropSize = 0xC620
228
+ ColorMatrix1 = 0xC621
229
+ ColorMatrix2 = 0xC622
230
+ CameraCalibration1 = 0xC623
231
+ CameraCalibration2 = 0xC624
232
+ ReductionMatrix1 = 0xC625
233
+ ReductionMatrix2 = 0xC626
234
+ AnalogBalance = 0xC627
235
+ AsShotNeutral = 0xC628
236
+ AsShotWhiteXY = 0xC629
237
+ BaselineExposure = 0xC62A
238
+ BaselineNoise = 0xC62B
239
+ BaselineSharpness = 0xC62C
240
+ BayerGreenSplit = 0xC62D
241
+ LinearResponseLimit = 0xC62E
242
+ CameraSerialNumber = 0xC62F
243
+ LensInfo = 0xC630
244
+ ChromaBlurRadius = 0xC631
245
+ AntiAliasStrength = 0xC632
246
+ ShadowScale = 0xC633
247
+ DNGPrivateData = 0xC634
248
+ MakerNoteSafety = 0xC635
249
+ CalibrationIlluminant1 = 0xC65A
250
+ CalibrationIlluminant2 = 0xC65B
251
+ BestQualityScale = 0xC65C
252
+ RawDataUniqueID = 0xC65D
253
+ OriginalRawFileName = 0xC68B
254
+ OriginalRawFileData = 0xC68C
255
+ ActiveArea = 0xC68D
256
+ MaskedAreas = 0xC68E
257
+ AsShotICCProfile = 0xC68F
258
+ AsShotPreProfileMatrix = 0xC690
259
+ CurrentICCProfile = 0xC691
260
+ CurrentPreProfileMatrix = 0xC692
261
+ ColorimetricReference = 0xC6BF
262
+ CameraCalibrationSignature = 0xC6F3
263
+ ProfileCalibrationSignature = 0xC6F4
264
+ AsShotProfileName = 0xC6F6
265
+ NoiseReductionApplied = 0xC6F7
266
+ ProfileName = 0xC6F8
267
+ ProfileHueSatMapDims = 0xC6F9
268
+ ProfileHueSatMapData1 = 0xC6FA
269
+ ProfileHueSatMapData2 = 0xC6FB
270
+ ProfileToneCurve = 0xC6FC
271
+ ProfileEmbedPolicy = 0xC6FD
272
+ ProfileCopyright = 0xC6FE
273
+ ForwardMatrix1 = 0xC714
274
+ ForwardMatrix2 = 0xC715
275
+ PreviewApplicationName = 0xC716
276
+ PreviewApplicationVersion = 0xC717
277
+ PreviewSettingsName = 0xC718
278
+ PreviewSettingsDigest = 0xC719
279
+ PreviewColorSpace = 0xC71A
280
+ PreviewDateTime = 0xC71B
281
+ RawImageDigest = 0xC71C
282
+ OriginalRawFileDigest = 0xC71D
283
+ SubTileBlockSize = 0xC71E
284
+ RowInterleaveFactor = 0xC71F
285
+ ProfileLookTableDims = 0xC725
286
+ ProfileLookTableData = 0xC726
287
+ OpcodeList1 = 0xC740
288
+ OpcodeList2 = 0xC741
289
+ OpcodeList3 = 0xC74E
290
+ NoiseProfile = 0xC761
291
+
292
+
293
+ """Maps EXIF tags to tag names."""
294
+ TAGS = {
295
+ **{i.value: i.name for i in Base},
296
+ 0x920C: "SpatialFrequencyResponse",
297
+ 0x9214: "SubjectLocation",
298
+ 0x9215: "ExposureIndex",
299
+ 0x828E: "CFAPattern",
300
+ 0x920B: "FlashEnergy",
301
+ 0x9216: "TIFF/EPStandardID",
302
+ }
303
+
304
+
305
+ class GPS(IntEnum):
306
+ GPSVersionID = 0x00
307
+ GPSLatitudeRef = 0x01
308
+ GPSLatitude = 0x02
309
+ GPSLongitudeRef = 0x03
310
+ GPSLongitude = 0x04
311
+ GPSAltitudeRef = 0x05
312
+ GPSAltitude = 0x06
313
+ GPSTimeStamp = 0x07
314
+ GPSSatellites = 0x08
315
+ GPSStatus = 0x09
316
+ GPSMeasureMode = 0x0A
317
+ GPSDOP = 0x0B
318
+ GPSSpeedRef = 0x0C
319
+ GPSSpeed = 0x0D
320
+ GPSTrackRef = 0x0E
321
+ GPSTrack = 0x0F
322
+ GPSImgDirectionRef = 0x10
323
+ GPSImgDirection = 0x11
324
+ GPSMapDatum = 0x12
325
+ GPSDestLatitudeRef = 0x13
326
+ GPSDestLatitude = 0x14
327
+ GPSDestLongitudeRef = 0x15
328
+ GPSDestLongitude = 0x16
329
+ GPSDestBearingRef = 0x17
330
+ GPSDestBearing = 0x18
331
+ GPSDestDistanceRef = 0x19
332
+ GPSDestDistance = 0x1A
333
+ GPSProcessingMethod = 0x1B
334
+ GPSAreaInformation = 0x1C
335
+ GPSDateStamp = 0x1D
336
+ GPSDifferential = 0x1E
337
+ GPSHPositioningError = 0x1F
338
+
339
+
340
+ """Maps EXIF GPS tags to tag names."""
341
+ GPSTAGS = {i.value: i.name for i in GPS}
342
+
343
+
344
+ class Interop(IntEnum):
345
+ InteropIndex = 0x0001
346
+ InteropVersion = 0x0002
347
+ RelatedImageFileFormat = 0x1000
348
+ RelatedImageWidth = 0x1001
349
+ RelatedImageHeight = 0x1002
350
+
351
+
352
+ class IFD(IntEnum):
353
+ Exif = 0x8769
354
+ GPSInfo = 0x8825
355
+ MakerNote = 0x927C
356
+ Makernote = 0x927C # Deprecated
357
+ Interop = 0xA005
358
+ IFD1 = -1
359
+
360
+
361
+ class LightSource(IntEnum):
362
+ Unknown = 0x00
363
+ Daylight = 0x01
364
+ Fluorescent = 0x02
365
+ Tungsten = 0x03
366
+ Flash = 0x04
367
+ Fine = 0x09
368
+ Cloudy = 0x0A
369
+ Shade = 0x0B
370
+ DaylightFluorescent = 0x0C
371
+ DayWhiteFluorescent = 0x0D
372
+ CoolWhiteFluorescent = 0x0E
373
+ WhiteFluorescent = 0x0F
374
+ StandardLightA = 0x11
375
+ StandardLightB = 0x12
376
+ StandardLightC = 0x13
377
+ D55 = 0x14
378
+ D65 = 0x15
379
+ D75 = 0x16
380
+ D50 = 0x17
381
+ ISO = 0x18
382
+ Other = 0xFF
.venv/Lib/site-packages/PIL/FitsImagePlugin.py ADDED
@@ -0,0 +1,152 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #
2
+ # The Python Imaging Library
3
+ # $Id$
4
+ #
5
+ # FITS file handling
6
+ #
7
+ # Copyright (c) 1998-2003 by Fredrik Lundh
8
+ #
9
+ # See the README file for information on usage and redistribution.
10
+ #
11
+ from __future__ import annotations
12
+
13
+ import gzip
14
+ import math
15
+
16
+ from . import Image, ImageFile
17
+
18
+
19
+ def _accept(prefix: bytes) -> bool:
20
+ return prefix.startswith(b"SIMPLE")
21
+
22
+
23
+ class FitsImageFile(ImageFile.ImageFile):
24
+ format = "FITS"
25
+ format_description = "FITS"
26
+
27
+ def _open(self) -> None:
28
+ assert self.fp is not None
29
+
30
+ headers: dict[bytes, bytes] = {}
31
+ header_in_progress = False
32
+ decoder_name = ""
33
+ while True:
34
+ header = self.fp.read(80)
35
+ if not header:
36
+ msg = "Truncated FITS file"
37
+ raise OSError(msg)
38
+ keyword = header[:8].strip()
39
+ if keyword in (b"SIMPLE", b"XTENSION"):
40
+ header_in_progress = True
41
+ elif headers and not header_in_progress:
42
+ # This is now a data unit
43
+ break
44
+ elif keyword == b"END":
45
+ # Seek to the end of the header unit
46
+ self.fp.seek(math.ceil(self.fp.tell() / 2880) * 2880)
47
+ if not decoder_name:
48
+ decoder_name, offset, args = self._parse_headers(headers)
49
+
50
+ header_in_progress = False
51
+ continue
52
+
53
+ if decoder_name:
54
+ # Keep going to read past the headers
55
+ continue
56
+
57
+ value = header[8:].split(b"/")[0].strip()
58
+ if value.startswith(b"="):
59
+ value = value[1:].strip()
60
+ if not headers and (not _accept(keyword) or value != b"T"):
61
+ msg = "Not a FITS file"
62
+ raise SyntaxError(msg)
63
+ headers[keyword] = value
64
+
65
+ if not decoder_name:
66
+ msg = "No image data"
67
+ raise ValueError(msg)
68
+
69
+ offset += self.fp.tell() - 80
70
+ self.tile = [ImageFile._Tile(decoder_name, (0, 0) + self.size, offset, args)]
71
+
72
+ def _get_size(
73
+ self, headers: dict[bytes, bytes], prefix: bytes
74
+ ) -> tuple[int, int] | None:
75
+ naxis = int(headers[prefix + b"NAXIS"])
76
+ if naxis == 0:
77
+ return None
78
+
79
+ if naxis == 1:
80
+ return 1, int(headers[prefix + b"NAXIS1"])
81
+ else:
82
+ return int(headers[prefix + b"NAXIS1"]), int(headers[prefix + b"NAXIS2"])
83
+
84
+ def _parse_headers(
85
+ self, headers: dict[bytes, bytes]
86
+ ) -> tuple[str, int, tuple[str | int, ...]]:
87
+ prefix = b""
88
+ decoder_name = "raw"
89
+ offset = 0
90
+ if (
91
+ headers.get(b"XTENSION") == b"'BINTABLE'"
92
+ and headers.get(b"ZIMAGE") == b"T"
93
+ and headers[b"ZCMPTYPE"] == b"'GZIP_1 '"
94
+ ):
95
+ no_prefix_size = self._get_size(headers, prefix) or (0, 0)
96
+ number_of_bits = int(headers[b"BITPIX"])
97
+ offset = no_prefix_size[0] * no_prefix_size[1] * (number_of_bits // 8)
98
+
99
+ prefix = b"Z"
100
+ decoder_name = "fits_gzip"
101
+
102
+ size = self._get_size(headers, prefix)
103
+ if not size:
104
+ return "", 0, ()
105
+
106
+ self._size = size
107
+
108
+ number_of_bits = int(headers[prefix + b"BITPIX"])
109
+ if number_of_bits == 8:
110
+ self._mode = "L"
111
+ elif number_of_bits == 16:
112
+ self._mode = "I;16"
113
+ elif number_of_bits == 32:
114
+ self._mode = "I"
115
+ elif number_of_bits in (-32, -64):
116
+ self._mode = "F"
117
+
118
+ args: tuple[str | int, ...]
119
+ if decoder_name == "raw":
120
+ args = (self.mode, 0, -1)
121
+ else:
122
+ args = (number_of_bits,)
123
+ return decoder_name, offset, args
124
+
125
+
126
+ class FitsGzipDecoder(ImageFile.PyDecoder):
127
+ _pulls_fd = True
128
+
129
+ def decode(self, buffer: bytes | Image.SupportsArrayInterface) -> tuple[int, int]:
130
+ assert self.fd is not None
131
+ value = gzip.decompress(self.fd.read())
132
+
133
+ rows = []
134
+ offset = 0
135
+ number_of_bits = min(self.args[0] // 8, 4)
136
+ for y in range(self.state.ysize):
137
+ row = bytearray()
138
+ for x in range(self.state.xsize):
139
+ row += value[offset + (4 - number_of_bits) : offset + 4]
140
+ offset += 4
141
+ rows.append(row)
142
+ self.set_as_raw(bytes([pixel for row in rows[::-1] for pixel in row]))
143
+ return -1, 0
144
+
145
+
146
+ # --------------------------------------------------------------------
147
+ # Registry
148
+
149
+ Image.register_open(FitsImageFile.format, FitsImageFile, _accept)
150
+ Image.register_decoder("fits_gzip", FitsGzipDecoder)
151
+
152
+ Image.register_extensions(FitsImageFile.format, [".fit", ".fits"])
.venv/Lib/site-packages/PIL/FliImagePlugin.py ADDED
@@ -0,0 +1,184 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #
2
+ # The Python Imaging Library.
3
+ # $Id$
4
+ #
5
+ # FLI/FLC file handling.
6
+ #
7
+ # History:
8
+ # 95-09-01 fl Created
9
+ # 97-01-03 fl Fixed parser, setup decoder tile
10
+ # 98-07-15 fl Renamed offset attribute to avoid name clash
11
+ #
12
+ # Copyright (c) Secret Labs AB 1997-98.
13
+ # Copyright (c) Fredrik Lundh 1995-97.
14
+ #
15
+ # See the README file for information on usage and redistribution.
16
+ #
17
+ from __future__ import annotations
18
+
19
+ import os
20
+
21
+ from . import Image, ImageFile, ImagePalette
22
+ from ._binary import i16le as i16
23
+ from ._binary import i32le as i32
24
+ from ._binary import o8
25
+ from ._util import DeferredError
26
+
27
+ #
28
+ # decoder
29
+
30
+
31
+ def _accept(prefix: bytes) -> bool:
32
+ return (
33
+ len(prefix) >= 16
34
+ and i16(prefix, 4) in [0xAF11, 0xAF12]
35
+ and i16(prefix, 14) in [0, 3] # flags
36
+ )
37
+
38
+
39
+ ##
40
+ # Image plugin for the FLI/FLC animation format. Use the <b>seek</b>
41
+ # method to load individual frames.
42
+
43
+
44
+ class FliImageFile(ImageFile.ImageFile):
45
+ format = "FLI"
46
+ format_description = "Autodesk FLI/FLC Animation"
47
+ _close_exclusive_fp_after_loading = False
48
+
49
+ def _open(self) -> None:
50
+ # HEAD
51
+ assert self.fp is not None
52
+ s = self.fp.read(128)
53
+ if not (
54
+ _accept(s)
55
+ and s[20:22] == b"\x00" * 2
56
+ and s[42:80] == b"\x00" * 38
57
+ and s[88:] == b"\x00" * 40
58
+ ):
59
+ msg = "not an FLI/FLC file"
60
+ raise SyntaxError(msg)
61
+
62
+ # frames
63
+ self.n_frames = i16(s, 6)
64
+ self.is_animated = self.n_frames > 1
65
+
66
+ # image characteristics
67
+ self._mode = "P"
68
+ self._size = i16(s, 8), i16(s, 10)
69
+
70
+ # animation speed
71
+ duration = i32(s, 16)
72
+ magic = i16(s, 4)
73
+ if magic == 0xAF11:
74
+ duration = (duration * 1000) // 70
75
+ self.info["duration"] = duration
76
+
77
+ # look for palette
78
+ palette = [(a, a, a) for a in range(256)]
79
+
80
+ s = self.fp.read(16)
81
+
82
+ self.__offset = 128
83
+
84
+ if i16(s, 4) == 0xF100:
85
+ # prefix chunk; ignore it
86
+ self.fp.seek(self.__offset + i32(s))
87
+ s = self.fp.read(16)
88
+
89
+ if i16(s, 4) == 0xF1FA:
90
+ # look for palette chunk
91
+ number_of_subchunks = i16(s, 6)
92
+ chunk_size: int | None = None
93
+ for _ in range(number_of_subchunks):
94
+ if chunk_size is not None:
95
+ self.fp.seek(chunk_size - 6, os.SEEK_CUR)
96
+ s = self.fp.read(6)
97
+ chunk_type = i16(s, 4)
98
+ if chunk_type in (4, 11):
99
+ self._palette(palette, 2 if chunk_type == 11 else 0)
100
+ break
101
+ chunk_size = i32(s)
102
+ if not chunk_size:
103
+ break
104
+
105
+ self.palette = ImagePalette.raw(
106
+ "RGB", b"".join(o8(r) + o8(g) + o8(b) for (r, g, b) in palette)
107
+ )
108
+
109
+ # set things up to decode first frame
110
+ self.__frame = -1
111
+ self._fp = self.fp
112
+ self.__rewind = self.fp.tell()
113
+ self.seek(0)
114
+
115
+ def _palette(self, palette: list[tuple[int, int, int]], shift: int) -> None:
116
+ # load palette
117
+
118
+ i = 0
119
+ assert self.fp is not None
120
+ for e in range(i16(self.fp.read(2))):
121
+ s = self.fp.read(2)
122
+ i = i + s[0]
123
+ n = s[1]
124
+ if n == 0:
125
+ n = 256
126
+ s = self.fp.read(n * 3)
127
+ for n in range(0, len(s), 3):
128
+ r = s[n] << shift
129
+ g = s[n + 1] << shift
130
+ b = s[n + 2] << shift
131
+ palette[i] = (r, g, b)
132
+ i += 1
133
+
134
+ def seek(self, frame: int) -> None:
135
+ if not self._seek_check(frame):
136
+ return
137
+ if frame < self.__frame:
138
+ self._seek(0)
139
+
140
+ for f in range(self.__frame + 1, frame + 1):
141
+ self._seek(f)
142
+
143
+ def _seek(self, frame: int) -> None:
144
+ if isinstance(self._fp, DeferredError):
145
+ raise self._fp.ex
146
+ if frame == 0:
147
+ self.__frame = -1
148
+ self._fp.seek(self.__rewind)
149
+ self.__offset = 128
150
+ else:
151
+ # ensure that the previous frame was loaded
152
+ self.load()
153
+
154
+ if frame != self.__frame + 1:
155
+ msg = f"cannot seek to frame {frame}"
156
+ raise ValueError(msg)
157
+ self.__frame = frame
158
+
159
+ # move to next frame
160
+ self.fp = self._fp
161
+ self.fp.seek(self.__offset)
162
+
163
+ s = self.fp.read(4)
164
+ if not s:
165
+ msg = "missing frame size"
166
+ raise EOFError(msg)
167
+
168
+ framesize = i32(s)
169
+
170
+ self.decodermaxblock = framesize
171
+ self.tile = [ImageFile._Tile("fli", (0, 0) + self.size, self.__offset)]
172
+
173
+ self.__offset += framesize
174
+
175
+ def tell(self) -> int:
176
+ return self.__frame
177
+
178
+
179
+ #
180
+ # registry
181
+
182
+ Image.register_open(FliImageFile.format, FliImageFile, _accept)
183
+
184
+ Image.register_extensions(FliImageFile.format, [".fli", ".flc"])
.venv/Lib/site-packages/PIL/FontFile.py ADDED
@@ -0,0 +1,134 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #
2
+ # The Python Imaging Library
3
+ # $Id$
4
+ #
5
+ # base class for raster font file parsers
6
+ #
7
+ # history:
8
+ # 1997-06-05 fl created
9
+ # 1997-08-19 fl restrict image width
10
+ #
11
+ # Copyright (c) 1997-1998 by Secret Labs AB
12
+ # Copyright (c) 1997-1998 by Fredrik Lundh
13
+ #
14
+ # See the README file for information on usage and redistribution.
15
+ #
16
+ from __future__ import annotations
17
+
18
+ import os
19
+ from typing import BinaryIO
20
+
21
+ from . import Image, _binary
22
+
23
+ WIDTH = 800
24
+
25
+
26
+ def puti16(
27
+ fp: BinaryIO, values: tuple[int, int, int, int, int, int, int, int, int, int]
28
+ ) -> None:
29
+ """Write network order (big-endian) 16-bit sequence"""
30
+ for v in values:
31
+ if v < 0:
32
+ v += 65536
33
+ fp.write(_binary.o16be(v))
34
+
35
+
36
+ class FontFile:
37
+ """Base class for raster font file handlers."""
38
+
39
+ bitmap: Image.Image | None = None
40
+
41
+ def __init__(self) -> None:
42
+ self.info: dict[bytes, bytes | int] = {}
43
+ self.glyph: list[
44
+ tuple[
45
+ tuple[int, int],
46
+ tuple[int, int, int, int],
47
+ tuple[int, int, int, int],
48
+ Image.Image,
49
+ ]
50
+ | None
51
+ ] = [None] * 256
52
+
53
+ def __getitem__(self, ix: int) -> (
54
+ tuple[
55
+ tuple[int, int],
56
+ tuple[int, int, int, int],
57
+ tuple[int, int, int, int],
58
+ Image.Image,
59
+ ]
60
+ | None
61
+ ):
62
+ return self.glyph[ix]
63
+
64
+ def compile(self) -> None:
65
+ """Create metrics and bitmap"""
66
+
67
+ if self.bitmap:
68
+ return
69
+
70
+ # create bitmap large enough to hold all data
71
+ h = w = maxwidth = 0
72
+ lines = 1
73
+ for glyph in self.glyph:
74
+ if glyph:
75
+ d, dst, src, im = glyph
76
+ h = max(h, src[3] - src[1])
77
+ w = w + (src[2] - src[0])
78
+ if w > WIDTH:
79
+ lines += 1
80
+ w = src[2] - src[0]
81
+ maxwidth = max(maxwidth, w)
82
+
83
+ xsize = maxwidth
84
+ ysize = lines * h
85
+
86
+ if xsize == 0 and ysize == 0:
87
+ return
88
+
89
+ self.ysize = h
90
+
91
+ # paste glyphs into bitmap
92
+ self.bitmap = Image.new("1", (xsize, ysize))
93
+ self.metrics: list[
94
+ tuple[tuple[int, int], tuple[int, int, int, int], tuple[int, int, int, int]]
95
+ | None
96
+ ] = [None] * 256
97
+ x = y = 0
98
+ for i in range(256):
99
+ glyph = self[i]
100
+ if glyph:
101
+ d, dst, src, im = glyph
102
+ xx = src[2] - src[0]
103
+ x0, y0 = x, y
104
+ x = x + xx
105
+ if x > WIDTH:
106
+ x, y = 0, y + h
107
+ x0, y0 = x, y
108
+ x = xx
109
+ s = src[0] + x0, src[1] + y0, src[2] + x0, src[3] + y0
110
+ self.bitmap.paste(im.crop(src), s)
111
+ self.metrics[i] = d, dst, s
112
+
113
+ def save(self, filename: str) -> None:
114
+ """Save font"""
115
+
116
+ self.compile()
117
+
118
+ # font data
119
+ if not self.bitmap:
120
+ msg = "No bitmap created"
121
+ raise ValueError(msg)
122
+ self.bitmap.save(os.path.splitext(filename)[0] + ".pbm", "PNG")
123
+
124
+ # font metrics
125
+ with open(os.path.splitext(filename)[0] + ".pil", "wb") as fp:
126
+ fp.write(b"PILfont\n")
127
+ fp.write(f";;;;;;{self.ysize};\n".encode("ascii")) # HACK!!!
128
+ fp.write(b"DATA\n")
129
+ for id in range(256):
130
+ m = self.metrics[id]
131
+ if not m:
132
+ puti16(fp, (0,) * 10)
133
+ else:
134
+ puti16(fp, m[0] + m[1] + m[2])
.venv/Lib/site-packages/PIL/FpxImagePlugin.py ADDED
@@ -0,0 +1,257 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #
2
+ # THIS IS WORK IN PROGRESS
3
+ #
4
+ # The Python Imaging Library.
5
+ # $Id$
6
+ #
7
+ # FlashPix support for PIL
8
+ #
9
+ # History:
10
+ # 97-01-25 fl Created (reads uncompressed RGB images only)
11
+ #
12
+ # Copyright (c) Secret Labs AB 1997.
13
+ # Copyright (c) Fredrik Lundh 1997.
14
+ #
15
+ # See the README file for information on usage and redistribution.
16
+ #
17
+ from __future__ import annotations
18
+
19
+ import olefile
20
+
21
+ from . import Image, ImageFile
22
+ from ._binary import i32le as i32
23
+
24
+ # we map from colour field tuples to (mode, rawmode) descriptors
25
+ MODES = {
26
+ # opacity
27
+ (0x00007FFE,): ("A", "L"),
28
+ # monochrome
29
+ (0x00010000,): ("L", "L"),
30
+ (0x00018000, 0x00017FFE): ("RGBA", "LA"),
31
+ # photo YCC
32
+ (0x00020000, 0x00020001, 0x00020002): ("RGB", "YCC;P"),
33
+ (0x00028000, 0x00028001, 0x00028002, 0x00027FFE): ("RGBA", "YCCA;P"),
34
+ # standard RGB (NIFRGB)
35
+ (0x00030000, 0x00030001, 0x00030002): ("RGB", "RGB"),
36
+ (0x00038000, 0x00038001, 0x00038002, 0x00037FFE): ("RGBA", "RGBA"),
37
+ }
38
+
39
+
40
+ #
41
+ # --------------------------------------------------------------------
42
+
43
+
44
+ def _accept(prefix: bytes) -> bool:
45
+ return prefix.startswith(olefile.MAGIC)
46
+
47
+
48
+ ##
49
+ # Image plugin for the FlashPix images.
50
+
51
+
52
+ class FpxImageFile(ImageFile.ImageFile):
53
+ format = "FPX"
54
+ format_description = "FlashPix"
55
+
56
+ def _open(self) -> None:
57
+ #
58
+ # read the OLE directory and see if this is a likely
59
+ # to be a FlashPix file
60
+
61
+ try:
62
+ self.ole = olefile.OleFileIO(self.fp)
63
+ except OSError as e:
64
+ msg = "not an FPX file; invalid OLE file"
65
+ raise SyntaxError(msg) from e
66
+
67
+ root = self.ole.root
68
+ if not root or root.clsid != "56616700-C154-11CE-8553-00AA00A1F95B":
69
+ msg = "not an FPX file; bad root CLSID"
70
+ raise SyntaxError(msg)
71
+
72
+ self._open_index(1)
73
+
74
+ def _open_index(self, index: int = 1) -> None:
75
+ #
76
+ # get the Image Contents Property Set
77
+
78
+ prop = self.ole.getproperties(
79
+ [f"Data Object Store {index:06d}", "\005Image Contents"]
80
+ )
81
+
82
+ # size (highest resolution)
83
+
84
+ assert isinstance(prop[0x1000002], int)
85
+ assert isinstance(prop[0x1000003], int)
86
+ self._size = prop[0x1000002], prop[0x1000003]
87
+
88
+ size = max(self.size)
89
+ i = 1
90
+ while size > 64:
91
+ size = size // 2
92
+ i += 1
93
+ self.maxid = i - 1
94
+
95
+ # mode. instead of using a single field for this, flashpix
96
+ # requires you to specify the mode for each channel in each
97
+ # resolution subimage, and leaves it to the decoder to make
98
+ # sure that they all match. for now, we'll cheat and assume
99
+ # that this is always the case.
100
+
101
+ id = self.maxid << 16
102
+
103
+ s = prop[0x2000002 | id]
104
+
105
+ if not isinstance(s, bytes) or (bands := i32(s, 4)) > 4:
106
+ msg = "Invalid number of bands"
107
+ raise OSError(msg)
108
+
109
+ # note: for now, we ignore the "uncalibrated" flag
110
+ colors = tuple(i32(s, 8 + i * 4) & 0x7FFFFFFF for i in range(bands))
111
+
112
+ self._mode, self.rawmode = MODES[colors]
113
+
114
+ # load JPEG tables, if any
115
+ self.jpeg = {}
116
+ for i in range(256):
117
+ id = 0x3000001 | (i << 16)
118
+ if id in prop:
119
+ self.jpeg[i] = prop[id]
120
+
121
+ self._open_subimage(1, self.maxid)
122
+
123
+ def _open_subimage(self, index: int = 1, subimage: int = 0) -> None:
124
+ #
125
+ # setup tile descriptors for a given subimage
126
+
127
+ stream = [
128
+ f"Data Object Store {index:06d}",
129
+ f"Resolution {subimage:04d}",
130
+ "Subimage 0000 Header",
131
+ ]
132
+
133
+ fp = self.ole.openstream(stream)
134
+
135
+ # skip prefix
136
+ fp.read(28)
137
+
138
+ # header stream
139
+ s = fp.read(36)
140
+
141
+ size = i32(s, 4), i32(s, 8)
142
+ # tilecount = i32(s, 12)
143
+ tilesize = i32(s, 16), i32(s, 20)
144
+ # channels = i32(s, 24)
145
+ offset = i32(s, 28)
146
+ length = i32(s, 32)
147
+
148
+ if size != self.size:
149
+ msg = "subimage mismatch"
150
+ raise OSError(msg)
151
+
152
+ # get tile descriptors
153
+ fp.seek(28 + offset)
154
+ s = fp.read(i32(s, 12) * length)
155
+
156
+ x = y = 0
157
+ xsize, ysize = size
158
+ xtile, ytile = tilesize
159
+ self.tile = []
160
+
161
+ for i in range(0, len(s), length):
162
+ x1 = min(xsize, x + xtile)
163
+ y1 = min(ysize, y + ytile)
164
+
165
+ compression = i32(s, i + 8)
166
+
167
+ if compression == 0:
168
+ self.tile.append(
169
+ ImageFile._Tile(
170
+ "raw",
171
+ (x, y, x1, y1),
172
+ i32(s, i) + 28,
173
+ self.rawmode,
174
+ )
175
+ )
176
+
177
+ elif compression == 1:
178
+ # FIXME: the fill decoder is not implemented
179
+ self.tile.append(
180
+ ImageFile._Tile(
181
+ "fill",
182
+ (x, y, x1, y1),
183
+ i32(s, i) + 28,
184
+ (self.rawmode, s[12:16]),
185
+ )
186
+ )
187
+
188
+ elif compression == 2:
189
+ internal_color_conversion = s[14]
190
+ jpeg_tables = s[15]
191
+ rawmode = self.rawmode
192
+
193
+ if internal_color_conversion:
194
+ # The image is stored as usual (usually YCbCr).
195
+ if rawmode == "RGBA":
196
+ # For "RGBA", data is stored as YCbCrA based on
197
+ # negative RGB. The following trick works around
198
+ # this problem :
199
+ jpegmode, rawmode = "YCbCrK", "CMYK"
200
+ else:
201
+ jpegmode = None # let the decoder decide
202
+
203
+ else:
204
+ # The image is stored as defined by rawmode
205
+ jpegmode = rawmode
206
+
207
+ self.tile.append(
208
+ ImageFile._Tile(
209
+ "jpeg",
210
+ (x, y, x1, y1),
211
+ i32(s, i) + 28,
212
+ (rawmode, jpegmode),
213
+ )
214
+ )
215
+
216
+ # FIXME: jpeg tables are tile dependent; the prefix
217
+ # data must be placed in the tile descriptor itself!
218
+
219
+ if jpeg_tables:
220
+ self.tile_prefix = self.jpeg[jpeg_tables]
221
+
222
+ else:
223
+ msg = "unknown/invalid compression"
224
+ raise OSError(msg)
225
+
226
+ x = x + xtile
227
+ if x >= xsize:
228
+ x, y = 0, y + ytile
229
+ if y >= ysize:
230
+ break # isn't really required
231
+
232
+ self.stream = stream
233
+ self._fp = self.fp
234
+ self.fp = None
235
+
236
+ def load(self) -> Image.core.PixelAccess | None:
237
+ if not self.fp:
238
+ self.fp = self.ole.openstream(self.stream[:2] + ["Subimage 0000 Data"])
239
+
240
+ return ImageFile.ImageFile.load(self)
241
+
242
+ def close(self) -> None:
243
+ self.ole.close()
244
+ super().close()
245
+
246
+ def __exit__(self, *args: object) -> None:
247
+ self.ole.close()
248
+ super().__exit__()
249
+
250
+
251
+ #
252
+ # --------------------------------------------------------------------
253
+
254
+
255
+ Image.register_open(FpxImageFile.format, FpxImageFile, _accept)
256
+
257
+ Image.register_extension(FpxImageFile.format, ".fpx")
.venv/Lib/site-packages/PIL/FtexImagePlugin.py ADDED
@@ -0,0 +1,114 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ """
2
+ A Pillow loader for .ftc and .ftu files (FTEX)
3
+ Jerome Leclanche <jerome@leclan.ch>
4
+
5
+ The contents of this file are hereby released in the public domain (CC0)
6
+ Full text of the CC0 license:
7
+ https://creativecommons.org/publicdomain/zero/1.0/
8
+
9
+ Independence War 2: Edge Of Chaos - Texture File Format - 16 October 2001
10
+
11
+ The textures used for 3D objects in Independence War 2: Edge Of Chaos are in a
12
+ packed custom format called FTEX. This file format uses file extensions FTC
13
+ and FTU.
14
+ * FTC files are compressed textures (using standard texture compression).
15
+ * FTU files are not compressed.
16
+ Texture File Format
17
+ The FTC and FTU texture files both use the same format. This
18
+ has the following structure:
19
+ {header}
20
+ {format_directory}
21
+ {data}
22
+ Where:
23
+ {header} = {
24
+ u32:magic,
25
+ u32:version,
26
+ u32:width,
27
+ u32:height,
28
+ u32:mipmap_count,
29
+ u32:format_count
30
+ }
31
+
32
+ * The "magic" number is "FTEX".
33
+ * "width" and "height" are the dimensions of the texture.
34
+ * "mipmap_count" is the number of mipmaps in the texture.
35
+ * "format_count" is the number of texture formats (different versions of the
36
+ same texture) in this file.
37
+
38
+ {format_directory} = format_count * { u32:format, u32:where }
39
+
40
+ The format value is 0 for DXT1 compressed textures and 1 for 24-bit RGB
41
+ uncompressed textures.
42
+ The texture data for a format starts at the position "where" in the file.
43
+
44
+ Each set of texture data in the file has the following structure:
45
+ {data} = format_count * { u32:mipmap_size, mipmap_size * { u8 } }
46
+ * "mipmap_size" is the number of bytes in that mip level. For compressed
47
+ textures this is the size of the texture data compressed with DXT1. For 24 bit
48
+ uncompressed textures, this is 3 * width * height. Following this are the image
49
+ bytes for that mipmap level.
50
+
51
+ Note: All data is stored in little-Endian (Intel) byte order.
52
+ """
53
+
54
+ from __future__ import annotations
55
+
56
+ import struct
57
+ from enum import IntEnum
58
+ from io import BytesIO
59
+
60
+ from . import Image, ImageFile
61
+
62
+ MAGIC = b"FTEX"
63
+
64
+
65
+ class Format(IntEnum):
66
+ DXT1 = 0
67
+ UNCOMPRESSED = 1
68
+
69
+
70
+ class FtexImageFile(ImageFile.ImageFile):
71
+ format = "FTEX"
72
+ format_description = "Texture File Format (IW2:EOC)"
73
+
74
+ def _open(self) -> None:
75
+ if not _accept(self.fp.read(4)):
76
+ msg = "not an FTEX file"
77
+ raise SyntaxError(msg)
78
+ struct.unpack("<i", self.fp.read(4)) # version
79
+ self._size = struct.unpack("<2i", self.fp.read(8))
80
+ mipmap_count, format_count = struct.unpack("<2i", self.fp.read(8))
81
+
82
+ # Only support single-format files.
83
+ # I don't know of any multi-format file.
84
+ assert format_count == 1
85
+
86
+ format, where = struct.unpack("<2i", self.fp.read(8))
87
+ self.fp.seek(where)
88
+ (mipmap_size,) = struct.unpack("<i", self.fp.read(4))
89
+
90
+ data = self.fp.read(mipmap_size)
91
+
92
+ if format == Format.DXT1:
93
+ self._mode = "RGBA"
94
+ self.tile = [ImageFile._Tile("bcn", (0, 0) + self.size, 0, (1,))]
95
+ elif format == Format.UNCOMPRESSED:
96
+ self._mode = "RGB"
97
+ self.tile = [ImageFile._Tile("raw", (0, 0) + self.size, 0, "RGB")]
98
+ else:
99
+ msg = f"Invalid texture compression format: {repr(format)}"
100
+ raise ValueError(msg)
101
+
102
+ self.fp.close()
103
+ self.fp = BytesIO(data)
104
+
105
+ def load_seek(self, pos: int) -> None:
106
+ pass
107
+
108
+
109
+ def _accept(prefix: bytes) -> bool:
110
+ return prefix.startswith(MAGIC)
111
+
112
+
113
+ Image.register_open(FtexImageFile.format, FtexImageFile, _accept)
114
+ Image.register_extensions(FtexImageFile.format, [".ftc", ".ftu"])
.venv/Lib/site-packages/PIL/GbrImagePlugin.py ADDED
@@ -0,0 +1,101 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #
2
+ # The Python Imaging Library
3
+ #
4
+ # load a GIMP brush file
5
+ #
6
+ # History:
7
+ # 96-03-14 fl Created
8
+ # 16-01-08 es Version 2
9
+ #
10
+ # Copyright (c) Secret Labs AB 1997.
11
+ # Copyright (c) Fredrik Lundh 1996.
12
+ # Copyright (c) Eric Soroos 2016.
13
+ #
14
+ # See the README file for information on usage and redistribution.
15
+ #
16
+ #
17
+ # See https://github.com/GNOME/gimp/blob/mainline/devel-docs/gbr.txt for
18
+ # format documentation.
19
+ #
20
+ # This code Interprets version 1 and 2 .gbr files.
21
+ # Version 1 files are obsolete, and should not be used for new
22
+ # brushes.
23
+ # Version 2 files are saved by GIMP v2.8 (at least)
24
+ # Version 3 files have a format specifier of 18 for 16bit floats in
25
+ # the color depth field. This is currently unsupported by Pillow.
26
+ from __future__ import annotations
27
+
28
+ from . import Image, ImageFile
29
+ from ._binary import i32be as i32
30
+
31
+
32
+ def _accept(prefix: bytes) -> bool:
33
+ return len(prefix) >= 8 and i32(prefix, 0) >= 20 and i32(prefix, 4) in (1, 2)
34
+
35
+
36
+ ##
37
+ # Image plugin for the GIMP brush format.
38
+
39
+
40
+ class GbrImageFile(ImageFile.ImageFile):
41
+ format = "GBR"
42
+ format_description = "GIMP brush file"
43
+
44
+ def _open(self) -> None:
45
+ header_size = i32(self.fp.read(4))
46
+ if header_size < 20:
47
+ msg = "not a GIMP brush"
48
+ raise SyntaxError(msg)
49
+ version = i32(self.fp.read(4))
50
+ if version not in (1, 2):
51
+ msg = f"Unsupported GIMP brush version: {version}"
52
+ raise SyntaxError(msg)
53
+
54
+ width = i32(self.fp.read(4))
55
+ height = i32(self.fp.read(4))
56
+ color_depth = i32(self.fp.read(4))
57
+ if width == 0 or height == 0:
58
+ msg = "not a GIMP brush"
59
+ raise SyntaxError(msg)
60
+ if color_depth not in (1, 4):
61
+ msg = f"Unsupported GIMP brush color depth: {color_depth}"
62
+ raise SyntaxError(msg)
63
+
64
+ if version == 1:
65
+ comment_length = header_size - 20
66
+ else:
67
+ comment_length = header_size - 28
68
+ magic_number = self.fp.read(4)
69
+ if magic_number != b"GIMP":
70
+ msg = "not a GIMP brush, bad magic number"
71
+ raise SyntaxError(msg)
72
+ self.info["spacing"] = i32(self.fp.read(4))
73
+
74
+ self.info["comment"] = self.fp.read(comment_length)[:-1]
75
+
76
+ if color_depth == 1:
77
+ self._mode = "L"
78
+ else:
79
+ self._mode = "RGBA"
80
+
81
+ self._size = width, height
82
+
83
+ # Image might not be small
84
+ Image._decompression_bomb_check(self.size)
85
+
86
+ # Data is an uncompressed block of w * h * bytes/pixel
87
+ self._data_size = width * height * color_depth
88
+
89
+ def load(self) -> Image.core.PixelAccess | None:
90
+ if self._im is None:
91
+ self.im = Image.core.new(self.mode, self.size)
92
+ self.frombytes(self.fp.read(self._data_size))
93
+ return Image.Image.load(self)
94
+
95
+
96
+ #
97
+ # registry
98
+
99
+
100
+ Image.register_open(GbrImageFile.format, GbrImageFile, _accept)
101
+ Image.register_extension(GbrImageFile.format, ".gbr")
.venv/Lib/site-packages/PIL/GdImageFile.py ADDED
@@ -0,0 +1,102 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #
2
+ # The Python Imaging Library.
3
+ # $Id$
4
+ #
5
+ # GD file handling
6
+ #
7
+ # History:
8
+ # 1996-04-12 fl Created
9
+ #
10
+ # Copyright (c) 1997 by Secret Labs AB.
11
+ # Copyright (c) 1996 by Fredrik Lundh.
12
+ #
13
+ # See the README file for information on usage and redistribution.
14
+ #
15
+
16
+
17
+ """
18
+ .. note::
19
+ This format cannot be automatically recognized, so the
20
+ class is not registered for use with :py:func:`PIL.Image.open()`. To open a
21
+ gd file, use the :py:func:`PIL.GdImageFile.open()` function instead.
22
+
23
+ .. warning::
24
+ THE GD FORMAT IS NOT DESIGNED FOR DATA INTERCHANGE. This
25
+ implementation is provided for convenience and demonstrational
26
+ purposes only.
27
+ """
28
+ from __future__ import annotations
29
+
30
+ from typing import IO
31
+
32
+ from . import ImageFile, ImagePalette, UnidentifiedImageError
33
+ from ._binary import i16be as i16
34
+ from ._binary import i32be as i32
35
+ from ._typing import StrOrBytesPath
36
+
37
+
38
+ class GdImageFile(ImageFile.ImageFile):
39
+ """
40
+ Image plugin for the GD uncompressed format. Note that this format
41
+ is not supported by the standard :py:func:`PIL.Image.open()` function. To use
42
+ this plugin, you have to import the :py:mod:`PIL.GdImageFile` module and
43
+ use the :py:func:`PIL.GdImageFile.open()` function.
44
+ """
45
+
46
+ format = "GD"
47
+ format_description = "GD uncompressed images"
48
+
49
+ def _open(self) -> None:
50
+ # Header
51
+ assert self.fp is not None
52
+
53
+ s = self.fp.read(1037)
54
+
55
+ if i16(s) not in [65534, 65535]:
56
+ msg = "Not a valid GD 2.x .gd file"
57
+ raise SyntaxError(msg)
58
+
59
+ self._mode = "P"
60
+ self._size = i16(s, 2), i16(s, 4)
61
+
62
+ true_color = s[6]
63
+ true_color_offset = 2 if true_color else 0
64
+
65
+ # transparency index
66
+ tindex = i32(s, 7 + true_color_offset)
67
+ if tindex < 256:
68
+ self.info["transparency"] = tindex
69
+
70
+ self.palette = ImagePalette.raw(
71
+ "RGBX", s[7 + true_color_offset + 6 : 7 + true_color_offset + 6 + 256 * 4]
72
+ )
73
+
74
+ self.tile = [
75
+ ImageFile._Tile(
76
+ "raw",
77
+ (0, 0) + self.size,
78
+ 7 + true_color_offset + 6 + 256 * 4,
79
+ "L",
80
+ )
81
+ ]
82
+
83
+
84
+ def open(fp: StrOrBytesPath | IO[bytes], mode: str = "r") -> GdImageFile:
85
+ """
86
+ Load texture from a GD image file.
87
+
88
+ :param fp: GD file name, or an opened file handle.
89
+ :param mode: Optional mode. In this version, if the mode argument
90
+ is given, it must be "r".
91
+ :returns: An image instance.
92
+ :raises OSError: If the image could not be read.
93
+ """
94
+ if mode != "r":
95
+ msg = "bad mode"
96
+ raise ValueError(msg)
97
+
98
+ try:
99
+ return GdImageFile(fp)
100
+ except SyntaxError as e:
101
+ msg = "cannot identify this image file"
102
+ raise UnidentifiedImageError(msg) from e
.venv/Lib/site-packages/PIL/GifImagePlugin.py ADDED
@@ -0,0 +1,1215 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #
2
+ # The Python Imaging Library.
3
+ # $Id$
4
+ #
5
+ # GIF file handling
6
+ #
7
+ # History:
8
+ # 1995-09-01 fl Created
9
+ # 1996-12-14 fl Added interlace support
10
+ # 1996-12-30 fl Added animation support
11
+ # 1997-01-05 fl Added write support, fixed local colour map bug
12
+ # 1997-02-23 fl Make sure to load raster data in getdata()
13
+ # 1997-07-05 fl Support external decoder (0.4)
14
+ # 1998-07-09 fl Handle all modes when saving (0.5)
15
+ # 1998-07-15 fl Renamed offset attribute to avoid name clash
16
+ # 2001-04-16 fl Added rewind support (seek to frame 0) (0.6)
17
+ # 2001-04-17 fl Added palette optimization (0.7)
18
+ # 2002-06-06 fl Added transparency support for save (0.8)
19
+ # 2004-02-24 fl Disable interlacing for small images
20
+ #
21
+ # Copyright (c) 1997-2004 by Secret Labs AB
22
+ # Copyright (c) 1995-2004 by Fredrik Lundh
23
+ #
24
+ # See the README file for information on usage and redistribution.
25
+ #
26
+ from __future__ import annotations
27
+
28
+ import itertools
29
+ import math
30
+ import os
31
+ import subprocess
32
+ from enum import IntEnum
33
+ from functools import cached_property
34
+ from typing import Any, NamedTuple, cast
35
+
36
+ from . import (
37
+ Image,
38
+ ImageChops,
39
+ ImageFile,
40
+ ImageMath,
41
+ ImageOps,
42
+ ImagePalette,
43
+ ImageSequence,
44
+ )
45
+ from ._binary import i16le as i16
46
+ from ._binary import o8
47
+ from ._binary import o16le as o16
48
+ from ._util import DeferredError
49
+
50
+ TYPE_CHECKING = False
51
+ if TYPE_CHECKING:
52
+ from typing import IO, Literal
53
+
54
+ from . import _imaging
55
+ from ._typing import Buffer
56
+
57
+
58
+ class LoadingStrategy(IntEnum):
59
+ """.. versionadded:: 9.1.0"""
60
+
61
+ RGB_AFTER_FIRST = 0
62
+ RGB_AFTER_DIFFERENT_PALETTE_ONLY = 1
63
+ RGB_ALWAYS = 2
64
+
65
+
66
+ #: .. versionadded:: 9.1.0
67
+ LOADING_STRATEGY = LoadingStrategy.RGB_AFTER_FIRST
68
+
69
+ # --------------------------------------------------------------------
70
+ # Identify/read GIF files
71
+
72
+
73
+ def _accept(prefix: bytes) -> bool:
74
+ return prefix.startswith((b"GIF87a", b"GIF89a"))
75
+
76
+
77
+ ##
78
+ # Image plugin for GIF images. This plugin supports both GIF87 and
79
+ # GIF89 images.
80
+
81
+
82
+ class GifImageFile(ImageFile.ImageFile):
83
+ format = "GIF"
84
+ format_description = "Compuserve GIF"
85
+ _close_exclusive_fp_after_loading = False
86
+
87
+ global_palette = None
88
+
89
+ def data(self) -> bytes | None:
90
+ s = self.fp.read(1)
91
+ if s and s[0]:
92
+ return self.fp.read(s[0])
93
+ return None
94
+
95
+ def _is_palette_needed(self, p: bytes) -> bool:
96
+ for i in range(0, len(p), 3):
97
+ if not (i // 3 == p[i] == p[i + 1] == p[i + 2]):
98
+ return True
99
+ return False
100
+
101
+ def _open(self) -> None:
102
+ # Screen
103
+ s = self.fp.read(13)
104
+ if not _accept(s):
105
+ msg = "not a GIF file"
106
+ raise SyntaxError(msg)
107
+
108
+ self.info["version"] = s[:6]
109
+ self._size = i16(s, 6), i16(s, 8)
110
+ flags = s[10]
111
+ bits = (flags & 7) + 1
112
+
113
+ if flags & 128:
114
+ # get global palette
115
+ self.info["background"] = s[11]
116
+ # check if palette contains colour indices
117
+ p = self.fp.read(3 << bits)
118
+ if self._is_palette_needed(p):
119
+ p = ImagePalette.raw("RGB", p)
120
+ self.global_palette = self.palette = p
121
+
122
+ self._fp = self.fp # FIXME: hack
123
+ self.__rewind = self.fp.tell()
124
+ self._n_frames: int | None = None
125
+ self._seek(0) # get ready to read first frame
126
+
127
+ @property
128
+ def n_frames(self) -> int:
129
+ if self._n_frames is None:
130
+ current = self.tell()
131
+ try:
132
+ while True:
133
+ self._seek(self.tell() + 1, False)
134
+ except EOFError:
135
+ self._n_frames = self.tell() + 1
136
+ self.seek(current)
137
+ return self._n_frames
138
+
139
+ @cached_property
140
+ def is_animated(self) -> bool:
141
+ if self._n_frames is not None:
142
+ return self._n_frames != 1
143
+
144
+ current = self.tell()
145
+ if current:
146
+ return True
147
+
148
+ try:
149
+ self._seek(1, False)
150
+ is_animated = True
151
+ except EOFError:
152
+ is_animated = False
153
+
154
+ self.seek(current)
155
+ return is_animated
156
+
157
+ def seek(self, frame: int) -> None:
158
+ if not self._seek_check(frame):
159
+ return
160
+ if frame < self.__frame:
161
+ self._im = None
162
+ self._seek(0)
163
+
164
+ last_frame = self.__frame
165
+ for f in range(self.__frame + 1, frame + 1):
166
+ try:
167
+ self._seek(f)
168
+ except EOFError as e:
169
+ self.seek(last_frame)
170
+ msg = "no more images in GIF file"
171
+ raise EOFError(msg) from e
172
+
173
+ def _seek(self, frame: int, update_image: bool = True) -> None:
174
+ if isinstance(self._fp, DeferredError):
175
+ raise self._fp.ex
176
+ if frame == 0:
177
+ # rewind
178
+ self.__offset = 0
179
+ self.dispose: _imaging.ImagingCore | None = None
180
+ self.__frame = -1
181
+ self._fp.seek(self.__rewind)
182
+ self.disposal_method = 0
183
+ if "comment" in self.info:
184
+ del self.info["comment"]
185
+ else:
186
+ # ensure that the previous frame was loaded
187
+ if self.tile and update_image:
188
+ self.load()
189
+
190
+ if frame != self.__frame + 1:
191
+ msg = f"cannot seek to frame {frame}"
192
+ raise ValueError(msg)
193
+
194
+ self.fp = self._fp
195
+ if self.__offset:
196
+ # backup to last frame
197
+ self.fp.seek(self.__offset)
198
+ while self.data():
199
+ pass
200
+ self.__offset = 0
201
+
202
+ s = self.fp.read(1)
203
+ if not s or s == b";":
204
+ msg = "no more images in GIF file"
205
+ raise EOFError(msg)
206
+
207
+ palette: ImagePalette.ImagePalette | Literal[False] | None = None
208
+
209
+ info: dict[str, Any] = {}
210
+ frame_transparency = None
211
+ interlace = None
212
+ frame_dispose_extent = None
213
+ while True:
214
+ if not s:
215
+ s = self.fp.read(1)
216
+ if not s or s == b";":
217
+ break
218
+
219
+ elif s == b"!":
220
+ #
221
+ # extensions
222
+ #
223
+ s = self.fp.read(1)
224
+ block = self.data()
225
+ if s[0] == 249 and block is not None:
226
+ #
227
+ # graphic control extension
228
+ #
229
+ flags = block[0]
230
+ if flags & 1:
231
+ frame_transparency = block[3]
232
+ info["duration"] = i16(block, 1) * 10
233
+
234
+ # disposal method - find the value of bits 4 - 6
235
+ dispose_bits = 0b00011100 & flags
236
+ dispose_bits = dispose_bits >> 2
237
+ if dispose_bits:
238
+ # only set the dispose if it is not
239
+ # unspecified. I'm not sure if this is
240
+ # correct, but it seems to prevent the last
241
+ # frame from looking odd for some animations
242
+ self.disposal_method = dispose_bits
243
+ elif s[0] == 254:
244
+ #
245
+ # comment extension
246
+ #
247
+ comment = b""
248
+
249
+ # Read this comment block
250
+ while block:
251
+ comment += block
252
+ block = self.data()
253
+
254
+ if "comment" in info:
255
+ # If multiple comment blocks in frame, separate with \n
256
+ info["comment"] += b"\n" + comment
257
+ else:
258
+ info["comment"] = comment
259
+ s = None
260
+ continue
261
+ elif s[0] == 255 and frame == 0 and block is not None:
262
+ #
263
+ # application extension
264
+ #
265
+ info["extension"] = block, self.fp.tell()
266
+ if block.startswith(b"NETSCAPE2.0"):
267
+ block = self.data()
268
+ if block and len(block) >= 3 and block[0] == 1:
269
+ self.info["loop"] = i16(block, 1)
270
+ while self.data():
271
+ pass
272
+
273
+ elif s == b",":
274
+ #
275
+ # local image
276
+ #
277
+ s = self.fp.read(9)
278
+
279
+ # extent
280
+ x0, y0 = i16(s, 0), i16(s, 2)
281
+ x1, y1 = x0 + i16(s, 4), y0 + i16(s, 6)
282
+ if (x1 > self.size[0] or y1 > self.size[1]) and update_image:
283
+ self._size = max(x1, self.size[0]), max(y1, self.size[1])
284
+ Image._decompression_bomb_check(self._size)
285
+ frame_dispose_extent = x0, y0, x1, y1
286
+ flags = s[8]
287
+
288
+ interlace = (flags & 64) != 0
289
+
290
+ if flags & 128:
291
+ bits = (flags & 7) + 1
292
+ p = self.fp.read(3 << bits)
293
+ if self._is_palette_needed(p):
294
+ palette = ImagePalette.raw("RGB", p)
295
+ else:
296
+ palette = False
297
+
298
+ # image data
299
+ bits = self.fp.read(1)[0]
300
+ self.__offset = self.fp.tell()
301
+ break
302
+ s = None
303
+
304
+ if interlace is None:
305
+ msg = "image not found in GIF frame"
306
+ raise EOFError(msg)
307
+
308
+ self.__frame = frame
309
+ if not update_image:
310
+ return
311
+
312
+ self.tile = []
313
+
314
+ if self.dispose:
315
+ self.im.paste(self.dispose, self.dispose_extent)
316
+
317
+ self._frame_palette = palette if palette is not None else self.global_palette
318
+ self._frame_transparency = frame_transparency
319
+ if frame == 0:
320
+ if self._frame_palette:
321
+ if LOADING_STRATEGY == LoadingStrategy.RGB_ALWAYS:
322
+ self._mode = "RGBA" if frame_transparency is not None else "RGB"
323
+ else:
324
+ self._mode = "P"
325
+ else:
326
+ self._mode = "L"
327
+
328
+ if palette:
329
+ self.palette = palette
330
+ elif self.global_palette:
331
+ from copy import copy
332
+
333
+ self.palette = copy(self.global_palette)
334
+ else:
335
+ self.palette = None
336
+ else:
337
+ if self.mode == "P":
338
+ if (
339
+ LOADING_STRATEGY != LoadingStrategy.RGB_AFTER_DIFFERENT_PALETTE_ONLY
340
+ or palette
341
+ ):
342
+ if "transparency" in self.info:
343
+ self.im.putpalettealpha(self.info["transparency"], 0)
344
+ self.im = self.im.convert("RGBA", Image.Dither.FLOYDSTEINBERG)
345
+ self._mode = "RGBA"
346
+ del self.info["transparency"]
347
+ else:
348
+ self._mode = "RGB"
349
+ self.im = self.im.convert("RGB", Image.Dither.FLOYDSTEINBERG)
350
+
351
+ def _rgb(color: int) -> tuple[int, int, int]:
352
+ if self._frame_palette:
353
+ if color * 3 + 3 > len(self._frame_palette.palette):
354
+ color = 0
355
+ return cast(
356
+ tuple[int, int, int],
357
+ tuple(self._frame_palette.palette[color * 3 : color * 3 + 3]),
358
+ )
359
+ else:
360
+ return (color, color, color)
361
+
362
+ self.dispose = None
363
+ self.dispose_extent: tuple[int, int, int, int] | None = frame_dispose_extent
364
+ if self.dispose_extent and self.disposal_method >= 2:
365
+ try:
366
+ if self.disposal_method == 2:
367
+ # replace with background colour
368
+
369
+ # only dispose the extent in this frame
370
+ x0, y0, x1, y1 = self.dispose_extent
371
+ dispose_size = (x1 - x0, y1 - y0)
372
+
373
+ Image._decompression_bomb_check(dispose_size)
374
+
375
+ # by convention, attempt to use transparency first
376
+ dispose_mode = "P"
377
+ color = self.info.get("transparency", frame_transparency)
378
+ if color is not None:
379
+ if self.mode in ("RGB", "RGBA"):
380
+ dispose_mode = "RGBA"
381
+ color = _rgb(color) + (0,)
382
+ else:
383
+ color = self.info.get("background", 0)
384
+ if self.mode in ("RGB", "RGBA"):
385
+ dispose_mode = "RGB"
386
+ color = _rgb(color)
387
+ self.dispose = Image.core.fill(dispose_mode, dispose_size, color)
388
+ else:
389
+ # replace with previous contents
390
+ if self._im is not None:
391
+ # only dispose the extent in this frame
392
+ self.dispose = self._crop(self.im, self.dispose_extent)
393
+ elif frame_transparency is not None:
394
+ x0, y0, x1, y1 = self.dispose_extent
395
+ dispose_size = (x1 - x0, y1 - y0)
396
+
397
+ Image._decompression_bomb_check(dispose_size)
398
+ dispose_mode = "P"
399
+ color = frame_transparency
400
+ if self.mode in ("RGB", "RGBA"):
401
+ dispose_mode = "RGBA"
402
+ color = _rgb(frame_transparency) + (0,)
403
+ self.dispose = Image.core.fill(
404
+ dispose_mode, dispose_size, color
405
+ )
406
+ except AttributeError:
407
+ pass
408
+
409
+ if interlace is not None:
410
+ transparency = -1
411
+ if frame_transparency is not None:
412
+ if frame == 0:
413
+ if LOADING_STRATEGY != LoadingStrategy.RGB_ALWAYS:
414
+ self.info["transparency"] = frame_transparency
415
+ elif self.mode not in ("RGB", "RGBA"):
416
+ transparency = frame_transparency
417
+ self.tile = [
418
+ ImageFile._Tile(
419
+ "gif",
420
+ (x0, y0, x1, y1),
421
+ self.__offset,
422
+ (bits, interlace, transparency),
423
+ )
424
+ ]
425
+
426
+ if info.get("comment"):
427
+ self.info["comment"] = info["comment"]
428
+ for k in ["duration", "extension"]:
429
+ if k in info:
430
+ self.info[k] = info[k]
431
+ elif k in self.info:
432
+ del self.info[k]
433
+
434
+ def load_prepare(self) -> None:
435
+ temp_mode = "P" if self._frame_palette else "L"
436
+ self._prev_im = None
437
+ if self.__frame == 0:
438
+ if self._frame_transparency is not None:
439
+ self.im = Image.core.fill(
440
+ temp_mode, self.size, self._frame_transparency
441
+ )
442
+ elif self.mode in ("RGB", "RGBA"):
443
+ self._prev_im = self.im
444
+ if self._frame_palette:
445
+ self.im = Image.core.fill("P", self.size, self._frame_transparency or 0)
446
+ self.im.putpalette("RGB", *self._frame_palette.getdata())
447
+ else:
448
+ self._im = None
449
+ if not self._prev_im and self._im is not None and self.size != self.im.size:
450
+ expanded_im = Image.core.fill(self.im.mode, self.size)
451
+ if self._frame_palette:
452
+ expanded_im.putpalette("RGB", *self._frame_palette.getdata())
453
+ expanded_im.paste(self.im, (0, 0) + self.im.size)
454
+
455
+ self.im = expanded_im
456
+ self._mode = temp_mode
457
+ self._frame_palette = None
458
+
459
+ super().load_prepare()
460
+
461
+ def load_end(self) -> None:
462
+ if self.__frame == 0:
463
+ if self.mode == "P" and LOADING_STRATEGY == LoadingStrategy.RGB_ALWAYS:
464
+ if self._frame_transparency is not None:
465
+ self.im.putpalettealpha(self._frame_transparency, 0)
466
+ self._mode = "RGBA"
467
+ else:
468
+ self._mode = "RGB"
469
+ self.im = self.im.convert(self.mode, Image.Dither.FLOYDSTEINBERG)
470
+ return
471
+ if not self._prev_im:
472
+ return
473
+ if self.size != self._prev_im.size:
474
+ if self._frame_transparency is not None:
475
+ expanded_im = Image.core.fill("RGBA", self.size)
476
+ else:
477
+ expanded_im = Image.core.fill("P", self.size)
478
+ expanded_im.putpalette("RGB", "RGB", self.im.getpalette())
479
+ expanded_im = expanded_im.convert("RGB")
480
+ expanded_im.paste(self._prev_im, (0, 0) + self._prev_im.size)
481
+
482
+ self._prev_im = expanded_im
483
+ assert self._prev_im is not None
484
+ if self._frame_transparency is not None:
485
+ if self.mode == "L":
486
+ frame_im = self.im.convert_transparent("LA", self._frame_transparency)
487
+ else:
488
+ self.im.putpalettealpha(self._frame_transparency, 0)
489
+ frame_im = self.im.convert("RGBA")
490
+ else:
491
+ frame_im = self.im.convert("RGB")
492
+
493
+ assert self.dispose_extent is not None
494
+ frame_im = self._crop(frame_im, self.dispose_extent)
495
+
496
+ self.im = self._prev_im
497
+ self._mode = self.im.mode
498
+ if frame_im.mode in ("LA", "RGBA"):
499
+ self.im.paste(frame_im, self.dispose_extent, frame_im)
500
+ else:
501
+ self.im.paste(frame_im, self.dispose_extent)
502
+
503
+ def tell(self) -> int:
504
+ return self.__frame
505
+
506
+
507
+ # --------------------------------------------------------------------
508
+ # Write GIF files
509
+
510
+
511
+ RAWMODE = {"1": "L", "L": "L", "P": "P"}
512
+
513
+
514
+ def _normalize_mode(im: Image.Image) -> Image.Image:
515
+ """
516
+ Takes an image (or frame), returns an image in a mode that is appropriate
517
+ for saving in a Gif.
518
+
519
+ It may return the original image, or it may return an image converted to
520
+ palette or 'L' mode.
521
+
522
+ :param im: Image object
523
+ :returns: Image object
524
+ """
525
+ if im.mode in RAWMODE:
526
+ im.load()
527
+ return im
528
+ if Image.getmodebase(im.mode) == "RGB":
529
+ im = im.convert("P", palette=Image.Palette.ADAPTIVE)
530
+ assert im.palette is not None
531
+ if im.palette.mode == "RGBA":
532
+ for rgba in im.palette.colors:
533
+ if rgba[3] == 0:
534
+ im.info["transparency"] = im.palette.colors[rgba]
535
+ break
536
+ return im
537
+ return im.convert("L")
538
+
539
+
540
+ _Palette = bytes | bytearray | list[int] | ImagePalette.ImagePalette
541
+
542
+
543
+ def _normalize_palette(
544
+ im: Image.Image, palette: _Palette | None, info: dict[str, Any]
545
+ ) -> Image.Image:
546
+ """
547
+ Normalizes the palette for image.
548
+ - Sets the palette to the incoming palette, if provided.
549
+ - Ensures that there's a palette for L mode images
550
+ - Optimizes the palette if necessary/desired.
551
+
552
+ :param im: Image object
553
+ :param palette: bytes object containing the source palette, or ....
554
+ :param info: encoderinfo
555
+ :returns: Image object
556
+ """
557
+ source_palette = None
558
+ if palette:
559
+ # a bytes palette
560
+ if isinstance(palette, (bytes, bytearray, list)):
561
+ source_palette = bytearray(palette[:768])
562
+ if isinstance(palette, ImagePalette.ImagePalette):
563
+ source_palette = bytearray(palette.palette)
564
+
565
+ if im.mode == "P":
566
+ if not source_palette:
567
+ im_palette = im.getpalette(None)
568
+ assert im_palette is not None
569
+ source_palette = bytearray(im_palette)
570
+ else: # L-mode
571
+ if not source_palette:
572
+ source_palette = bytearray(i // 3 for i in range(768))
573
+ im.palette = ImagePalette.ImagePalette("RGB", palette=source_palette)
574
+ assert source_palette is not None
575
+
576
+ if palette:
577
+ used_palette_colors: list[int | None] = []
578
+ assert im.palette is not None
579
+ for i in range(0, len(source_palette), 3):
580
+ source_color = tuple(source_palette[i : i + 3])
581
+ index = im.palette.colors.get(source_color)
582
+ if index in used_palette_colors:
583
+ index = None
584
+ used_palette_colors.append(index)
585
+ for i, index in enumerate(used_palette_colors):
586
+ if index is None:
587
+ for j in range(len(used_palette_colors)):
588
+ if j not in used_palette_colors:
589
+ used_palette_colors[i] = j
590
+ break
591
+ dest_map: list[int] = []
592
+ for index in used_palette_colors:
593
+ assert index is not None
594
+ dest_map.append(index)
595
+ im = im.remap_palette(dest_map)
596
+ else:
597
+ optimized_palette_colors = _get_optimize(im, info)
598
+ if optimized_palette_colors is not None:
599
+ im = im.remap_palette(optimized_palette_colors, source_palette)
600
+ if "transparency" in info:
601
+ try:
602
+ info["transparency"] = optimized_palette_colors.index(
603
+ info["transparency"]
604
+ )
605
+ except ValueError:
606
+ del info["transparency"]
607
+ return im
608
+
609
+ assert im.palette is not None
610
+ im.palette.palette = source_palette
611
+ return im
612
+
613
+
614
+ def _write_single_frame(
615
+ im: Image.Image,
616
+ fp: IO[bytes],
617
+ palette: _Palette | None,
618
+ ) -> None:
619
+ im_out = _normalize_mode(im)
620
+ for k, v in im_out.info.items():
621
+ if isinstance(k, str):
622
+ im.encoderinfo.setdefault(k, v)
623
+ im_out = _normalize_palette(im_out, palette, im.encoderinfo)
624
+
625
+ for s in _get_global_header(im_out, im.encoderinfo):
626
+ fp.write(s)
627
+
628
+ # local image header
629
+ flags = 0
630
+ if get_interlace(im):
631
+ flags = flags | 64
632
+ _write_local_header(fp, im, (0, 0), flags)
633
+
634
+ im_out.encoderconfig = (8, get_interlace(im))
635
+ ImageFile._save(
636
+ im_out, fp, [ImageFile._Tile("gif", (0, 0) + im.size, 0, RAWMODE[im_out.mode])]
637
+ )
638
+
639
+ fp.write(b"\0") # end of image data
640
+
641
+
642
+ def _getbbox(
643
+ base_im: Image.Image, im_frame: Image.Image
644
+ ) -> tuple[Image.Image, tuple[int, int, int, int] | None]:
645
+ palette_bytes = [
646
+ bytes(im.palette.palette) if im.palette else b"" for im in (base_im, im_frame)
647
+ ]
648
+ if palette_bytes[0] != palette_bytes[1]:
649
+ im_frame = im_frame.convert("RGBA")
650
+ base_im = base_im.convert("RGBA")
651
+ delta = ImageChops.subtract_modulo(im_frame, base_im)
652
+ return delta, delta.getbbox(alpha_only=False)
653
+
654
+
655
+ class _Frame(NamedTuple):
656
+ im: Image.Image
657
+ bbox: tuple[int, int, int, int] | None
658
+ encoderinfo: dict[str, Any]
659
+
660
+
661
+ def _write_multiple_frames(
662
+ im: Image.Image, fp: IO[bytes], palette: _Palette | None
663
+ ) -> bool:
664
+ duration = im.encoderinfo.get("duration")
665
+ disposal = im.encoderinfo.get("disposal", im.info.get("disposal"))
666
+
667
+ im_frames: list[_Frame] = []
668
+ previous_im: Image.Image | None = None
669
+ frame_count = 0
670
+ background_im = None
671
+ for imSequence in itertools.chain([im], im.encoderinfo.get("append_images", [])):
672
+ for im_frame in ImageSequence.Iterator(imSequence):
673
+ # a copy is required here since seek can still mutate the image
674
+ im_frame = _normalize_mode(im_frame.copy())
675
+ if frame_count == 0:
676
+ for k, v in im_frame.info.items():
677
+ if k == "transparency":
678
+ continue
679
+ if isinstance(k, str):
680
+ im.encoderinfo.setdefault(k, v)
681
+
682
+ encoderinfo = im.encoderinfo.copy()
683
+ if "transparency" in im_frame.info:
684
+ encoderinfo.setdefault("transparency", im_frame.info["transparency"])
685
+ im_frame = _normalize_palette(im_frame, palette, encoderinfo)
686
+ if isinstance(duration, (list, tuple)):
687
+ encoderinfo["duration"] = duration[frame_count]
688
+ elif duration is None and "duration" in im_frame.info:
689
+ encoderinfo["duration"] = im_frame.info["duration"]
690
+ if isinstance(disposal, (list, tuple)):
691
+ encoderinfo["disposal"] = disposal[frame_count]
692
+ frame_count += 1
693
+
694
+ diff_frame = None
695
+ if im_frames and previous_im:
696
+ # delta frame
697
+ delta, bbox = _getbbox(previous_im, im_frame)
698
+ if not bbox:
699
+ # This frame is identical to the previous frame
700
+ if encoderinfo.get("duration"):
701
+ im_frames[-1].encoderinfo["duration"] += encoderinfo["duration"]
702
+ continue
703
+ if im_frames[-1].encoderinfo.get("disposal") == 2:
704
+ # To appear correctly in viewers using a convention,
705
+ # only consider transparency, and not background color
706
+ color = im.encoderinfo.get(
707
+ "transparency", im.info.get("transparency")
708
+ )
709
+ if color is not None:
710
+ if background_im is None:
711
+ background = _get_background(im_frame, color)
712
+ background_im = Image.new("P", im_frame.size, background)
713
+ first_palette = im_frames[0].im.palette
714
+ assert first_palette is not None
715
+ background_im.putpalette(first_palette, first_palette.mode)
716
+ bbox = _getbbox(background_im, im_frame)[1]
717
+ else:
718
+ bbox = (0, 0) + im_frame.size
719
+ elif encoderinfo.get("optimize") and im_frame.mode != "1":
720
+ if "transparency" not in encoderinfo:
721
+ assert im_frame.palette is not None
722
+ try:
723
+ encoderinfo["transparency"] = (
724
+ im_frame.palette._new_color_index(im_frame)
725
+ )
726
+ except ValueError:
727
+ pass
728
+ if "transparency" in encoderinfo:
729
+ # When the delta is zero, fill the image with transparency
730
+ diff_frame = im_frame.copy()
731
+ fill = Image.new("P", delta.size, encoderinfo["transparency"])
732
+ if delta.mode == "RGBA":
733
+ r, g, b, a = delta.split()
734
+ mask = ImageMath.lambda_eval(
735
+ lambda args: args["convert"](
736
+ args["max"](
737
+ args["max"](
738
+ args["max"](args["r"], args["g"]), args["b"]
739
+ ),
740
+ args["a"],
741
+ )
742
+ * 255,
743
+ "1",
744
+ ),
745
+ r=r,
746
+ g=g,
747
+ b=b,
748
+ a=a,
749
+ )
750
+ else:
751
+ if delta.mode == "P":
752
+ # Convert to L without considering palette
753
+ delta_l = Image.new("L", delta.size)
754
+ delta_l.putdata(delta.getdata())
755
+ delta = delta_l
756
+ mask = ImageMath.lambda_eval(
757
+ lambda args: args["convert"](args["im"] * 255, "1"),
758
+ im=delta,
759
+ )
760
+ diff_frame.paste(fill, mask=ImageOps.invert(mask))
761
+ else:
762
+ bbox = None
763
+ previous_im = im_frame
764
+ im_frames.append(_Frame(diff_frame or im_frame, bbox, encoderinfo))
765
+
766
+ if len(im_frames) == 1:
767
+ if "duration" in im.encoderinfo:
768
+ # Since multiple frames will not be written, use the combined duration
769
+ im.encoderinfo["duration"] = im_frames[0].encoderinfo["duration"]
770
+ return False
771
+
772
+ for frame_data in im_frames:
773
+ im_frame = frame_data.im
774
+ if not frame_data.bbox:
775
+ # global header
776
+ for s in _get_global_header(im_frame, frame_data.encoderinfo):
777
+ fp.write(s)
778
+ offset = (0, 0)
779
+ else:
780
+ # compress difference
781
+ if not palette:
782
+ frame_data.encoderinfo["include_color_table"] = True
783
+
784
+ if frame_data.bbox != (0, 0) + im_frame.size:
785
+ im_frame = im_frame.crop(frame_data.bbox)
786
+ offset = frame_data.bbox[:2]
787
+ _write_frame_data(fp, im_frame, offset, frame_data.encoderinfo)
788
+ return True
789
+
790
+
791
+ def _save_all(im: Image.Image, fp: IO[bytes], filename: str | bytes) -> None:
792
+ _save(im, fp, filename, save_all=True)
793
+
794
+
795
+ def _save(
796
+ im: Image.Image, fp: IO[bytes], filename: str | bytes, save_all: bool = False
797
+ ) -> None:
798
+ # header
799
+ if "palette" in im.encoderinfo or "palette" in im.info:
800
+ palette = im.encoderinfo.get("palette", im.info.get("palette"))
801
+ else:
802
+ palette = None
803
+ im.encoderinfo.setdefault("optimize", True)
804
+
805
+ if not save_all or not _write_multiple_frames(im, fp, palette):
806
+ _write_single_frame(im, fp, palette)
807
+
808
+ fp.write(b";") # end of file
809
+
810
+ if hasattr(fp, "flush"):
811
+ fp.flush()
812
+
813
+
814
+ def get_interlace(im: Image.Image) -> int:
815
+ interlace = im.encoderinfo.get("interlace", 1)
816
+
817
+ # workaround for @PIL153
818
+ if min(im.size) < 16:
819
+ interlace = 0
820
+
821
+ return interlace
822
+
823
+
824
+ def _write_local_header(
825
+ fp: IO[bytes], im: Image.Image, offset: tuple[int, int], flags: int
826
+ ) -> None:
827
+ try:
828
+ transparency = im.encoderinfo["transparency"]
829
+ except KeyError:
830
+ transparency = None
831
+
832
+ if "duration" in im.encoderinfo:
833
+ duration = int(im.encoderinfo["duration"] / 10)
834
+ else:
835
+ duration = 0
836
+
837
+ disposal = int(im.encoderinfo.get("disposal", 0))
838
+
839
+ if transparency is not None or duration != 0 or disposal:
840
+ packed_flag = 1 if transparency is not None else 0
841
+ packed_flag |= disposal << 2
842
+
843
+ fp.write(
844
+ b"!"
845
+ + o8(249) # extension intro
846
+ + o8(4) # length
847
+ + o8(packed_flag) # packed fields
848
+ + o16(duration) # duration
849
+ + o8(transparency or 0) # transparency index
850
+ + o8(0)
851
+ )
852
+
853
+ include_color_table = im.encoderinfo.get("include_color_table")
854
+ if include_color_table:
855
+ palette_bytes = _get_palette_bytes(im)
856
+ color_table_size = _get_color_table_size(palette_bytes)
857
+ if color_table_size:
858
+ flags = flags | 128 # local color table flag
859
+ flags = flags | color_table_size
860
+
861
+ fp.write(
862
+ b","
863
+ + o16(offset[0]) # offset
864
+ + o16(offset[1])
865
+ + o16(im.size[0]) # size
866
+ + o16(im.size[1])
867
+ + o8(flags) # flags
868
+ )
869
+ if include_color_table and color_table_size:
870
+ fp.write(_get_header_palette(palette_bytes))
871
+ fp.write(o8(8)) # bits
872
+
873
+
874
+ def _save_netpbm(im: Image.Image, fp: IO[bytes], filename: str | bytes) -> None:
875
+ # Unused by default.
876
+ # To use, uncomment the register_save call at the end of the file.
877
+ #
878
+ # If you need real GIF compression and/or RGB quantization, you
879
+ # can use the external NETPBM/PBMPLUS utilities. See comments
880
+ # below for information on how to enable this.
881
+ tempfile = im._dump()
882
+
883
+ try:
884
+ with open(filename, "wb") as f:
885
+ if im.mode != "RGB":
886
+ subprocess.check_call(
887
+ ["ppmtogif", tempfile], stdout=f, stderr=subprocess.DEVNULL
888
+ )
889
+ else:
890
+ # Pipe ppmquant output into ppmtogif
891
+ # "ppmquant 256 %s | ppmtogif > %s" % (tempfile, filename)
892
+ quant_cmd = ["ppmquant", "256", tempfile]
893
+ togif_cmd = ["ppmtogif"]
894
+ quant_proc = subprocess.Popen(
895
+ quant_cmd, stdout=subprocess.PIPE, stderr=subprocess.DEVNULL
896
+ )
897
+ togif_proc = subprocess.Popen(
898
+ togif_cmd,
899
+ stdin=quant_proc.stdout,
900
+ stdout=f,
901
+ stderr=subprocess.DEVNULL,
902
+ )
903
+
904
+ # Allow ppmquant to receive SIGPIPE if ppmtogif exits
905
+ assert quant_proc.stdout is not None
906
+ quant_proc.stdout.close()
907
+
908
+ retcode = quant_proc.wait()
909
+ if retcode:
910
+ raise subprocess.CalledProcessError(retcode, quant_cmd)
911
+
912
+ retcode = togif_proc.wait()
913
+ if retcode:
914
+ raise subprocess.CalledProcessError(retcode, togif_cmd)
915
+ finally:
916
+ try:
917
+ os.unlink(tempfile)
918
+ except OSError:
919
+ pass
920
+
921
+
922
+ # Force optimization so that we can test performance against
923
+ # cases where it took lots of memory and time previously.
924
+ _FORCE_OPTIMIZE = False
925
+
926
+
927
+ def _get_optimize(im: Image.Image, info: dict[str, Any]) -> list[int] | None:
928
+ """
929
+ Palette optimization is a potentially expensive operation.
930
+
931
+ This function determines if the palette should be optimized using
932
+ some heuristics, then returns the list of palette entries in use.
933
+
934
+ :param im: Image object
935
+ :param info: encoderinfo
936
+ :returns: list of indexes of palette entries in use, or None
937
+ """
938
+ if im.mode in ("P", "L") and info and info.get("optimize"):
939
+ # Potentially expensive operation.
940
+
941
+ # The palette saves 3 bytes per color not used, but palette
942
+ # lengths are restricted to 3*(2**N) bytes. Max saving would
943
+ # be 768 -> 6 bytes if we went all the way down to 2 colors.
944
+ # * If we're over 128 colors, we can't save any space.
945
+ # * If there aren't any holes, it's not worth collapsing.
946
+ # * If we have a 'large' image, the palette is in the noise.
947
+
948
+ # create the new palette if not every color is used
949
+ optimise = _FORCE_OPTIMIZE or im.mode == "L"
950
+ if optimise or im.width * im.height < 512 * 512:
951
+ # check which colors are used
952
+ used_palette_colors = []
953
+ for i, count in enumerate(im.histogram()):
954
+ if count:
955
+ used_palette_colors.append(i)
956
+
957
+ if optimise or max(used_palette_colors) >= len(used_palette_colors):
958
+ return used_palette_colors
959
+
960
+ assert im.palette is not None
961
+ num_palette_colors = len(im.palette.palette) // Image.getmodebands(
962
+ im.palette.mode
963
+ )
964
+ current_palette_size = 1 << (num_palette_colors - 1).bit_length()
965
+ if (
966
+ # check that the palette would become smaller when saved
967
+ len(used_palette_colors) <= current_palette_size // 2
968
+ # check that the palette is not already the smallest possible size
969
+ and current_palette_size > 2
970
+ ):
971
+ return used_palette_colors
972
+ return None
973
+
974
+
975
+ def _get_color_table_size(palette_bytes: bytes) -> int:
976
+ # calculate the palette size for the header
977
+ if not palette_bytes:
978
+ return 0
979
+ elif len(palette_bytes) < 9:
980
+ return 1
981
+ else:
982
+ return math.ceil(math.log(len(palette_bytes) // 3, 2)) - 1
983
+
984
+
985
+ def _get_header_palette(palette_bytes: bytes) -> bytes:
986
+ """
987
+ Returns the palette, null padded to the next power of 2 (*3) bytes
988
+ suitable for direct inclusion in the GIF header
989
+
990
+ :param palette_bytes: Unpadded palette bytes, in RGBRGB form
991
+ :returns: Null padded palette
992
+ """
993
+ color_table_size = _get_color_table_size(palette_bytes)
994
+
995
+ # add the missing amount of bytes
996
+ # the palette has to be 2<<n in size
997
+ actual_target_size_diff = (2 << color_table_size) - len(palette_bytes) // 3
998
+ if actual_target_size_diff > 0:
999
+ palette_bytes += o8(0) * 3 * actual_target_size_diff
1000
+ return palette_bytes
1001
+
1002
+
1003
+ def _get_palette_bytes(im: Image.Image) -> bytes:
1004
+ """
1005
+ Gets the palette for inclusion in the gif header
1006
+
1007
+ :param im: Image object
1008
+ :returns: Bytes, len<=768 suitable for inclusion in gif header
1009
+ """
1010
+ if not im.palette:
1011
+ return b""
1012
+
1013
+ palette = bytes(im.palette.palette)
1014
+ if im.palette.mode == "RGBA":
1015
+ palette = b"".join(palette[i * 4 : i * 4 + 3] for i in range(len(palette) // 3))
1016
+ return palette
1017
+
1018
+
1019
+ def _get_background(
1020
+ im: Image.Image,
1021
+ info_background: int | tuple[int, int, int] | tuple[int, int, int, int] | None,
1022
+ ) -> int:
1023
+ background = 0
1024
+ if info_background:
1025
+ if isinstance(info_background, tuple):
1026
+ # WebPImagePlugin stores an RGBA value in info["background"]
1027
+ # So it must be converted to the same format as GifImagePlugin's
1028
+ # info["background"] - a global color table index
1029
+ assert im.palette is not None
1030
+ try:
1031
+ background = im.palette.getcolor(info_background, im)
1032
+ except ValueError as e:
1033
+ if str(e) not in (
1034
+ # If all 256 colors are in use,
1035
+ # then there is no need for the background color
1036
+ "cannot allocate more than 256 colors",
1037
+ # Ignore non-opaque WebP background
1038
+ "cannot add non-opaque RGBA color to RGB palette",
1039
+ ):
1040
+ raise
1041
+ else:
1042
+ background = info_background
1043
+ return background
1044
+
1045
+
1046
+ def _get_global_header(im: Image.Image, info: dict[str, Any]) -> list[bytes]:
1047
+ """Return a list of strings representing a GIF header"""
1048
+
1049
+ # Header Block
1050
+ # https://www.matthewflickinger.com/lab/whatsinagif/bits_and_bytes.asp
1051
+
1052
+ version = b"87a"
1053
+ if im.info.get("version") == b"89a" or (
1054
+ info
1055
+ and (
1056
+ "transparency" in info
1057
+ or info.get("loop") is not None
1058
+ or info.get("duration")
1059
+ or info.get("comment")
1060
+ )
1061
+ ):
1062
+ version = b"89a"
1063
+
1064
+ background = _get_background(im, info.get("background"))
1065
+
1066
+ palette_bytes = _get_palette_bytes(im)
1067
+ color_table_size = _get_color_table_size(palette_bytes)
1068
+
1069
+ header = [
1070
+ b"GIF" # signature
1071
+ + version # version
1072
+ + o16(im.size[0]) # canvas width
1073
+ + o16(im.size[1]), # canvas height
1074
+ # Logical Screen Descriptor
1075
+ # size of global color table + global color table flag
1076
+ o8(color_table_size + 128), # packed fields
1077
+ # background + reserved/aspect
1078
+ o8(background) + o8(0),
1079
+ # Global Color Table
1080
+ _get_header_palette(palette_bytes),
1081
+ ]
1082
+ if info.get("loop") is not None:
1083
+ header.append(
1084
+ b"!"
1085
+ + o8(255) # extension intro
1086
+ + o8(11)
1087
+ + b"NETSCAPE2.0"
1088
+ + o8(3)
1089
+ + o8(1)
1090
+ + o16(info["loop"]) # number of loops
1091
+ + o8(0)
1092
+ )
1093
+ if info.get("comment"):
1094
+ comment_block = b"!" + o8(254) # extension intro
1095
+
1096
+ comment = info["comment"]
1097
+ if isinstance(comment, str):
1098
+ comment = comment.encode()
1099
+ for i in range(0, len(comment), 255):
1100
+ subblock = comment[i : i + 255]
1101
+ comment_block += o8(len(subblock)) + subblock
1102
+
1103
+ comment_block += o8(0)
1104
+ header.append(comment_block)
1105
+ return header
1106
+
1107
+
1108
+ def _write_frame_data(
1109
+ fp: IO[bytes],
1110
+ im_frame: Image.Image,
1111
+ offset: tuple[int, int],
1112
+ params: dict[str, Any],
1113
+ ) -> None:
1114
+ try:
1115
+ im_frame.encoderinfo = params
1116
+
1117
+ # local image header
1118
+ _write_local_header(fp, im_frame, offset, 0)
1119
+
1120
+ ImageFile._save(
1121
+ im_frame,
1122
+ fp,
1123
+ [ImageFile._Tile("gif", (0, 0) + im_frame.size, 0, RAWMODE[im_frame.mode])],
1124
+ )
1125
+
1126
+ fp.write(b"\0") # end of image data
1127
+ finally:
1128
+ del im_frame.encoderinfo
1129
+
1130
+
1131
+ # --------------------------------------------------------------------
1132
+ # Legacy GIF utilities
1133
+
1134
+
1135
+ def getheader(
1136
+ im: Image.Image, palette: _Palette | None = None, info: dict[str, Any] | None = None
1137
+ ) -> tuple[list[bytes], list[int] | None]:
1138
+ """
1139
+ Legacy Method to get Gif data from image.
1140
+
1141
+ Warning:: May modify image data.
1142
+
1143
+ :param im: Image object
1144
+ :param palette: bytes object containing the source palette, or ....
1145
+ :param info: encoderinfo
1146
+ :returns: tuple of(list of header items, optimized palette)
1147
+
1148
+ """
1149
+ if info is None:
1150
+ info = {}
1151
+
1152
+ used_palette_colors = _get_optimize(im, info)
1153
+
1154
+ if "background" not in info and "background" in im.info:
1155
+ info["background"] = im.info["background"]
1156
+
1157
+ im_mod = _normalize_palette(im, palette, info)
1158
+ im.palette = im_mod.palette
1159
+ im.im = im_mod.im
1160
+ header = _get_global_header(im, info)
1161
+
1162
+ return header, used_palette_colors
1163
+
1164
+
1165
+ def getdata(
1166
+ im: Image.Image, offset: tuple[int, int] = (0, 0), **params: Any
1167
+ ) -> list[bytes]:
1168
+ """
1169
+ Legacy Method
1170
+
1171
+ Return a list of strings representing this image.
1172
+ The first string is a local image header, the rest contains
1173
+ encoded image data.
1174
+
1175
+ To specify duration, add the time in milliseconds,
1176
+ e.g. ``getdata(im_frame, duration=1000)``
1177
+
1178
+ :param im: Image object
1179
+ :param offset: Tuple of (x, y) pixels. Defaults to (0, 0)
1180
+ :param \\**params: e.g. duration or other encoder info parameters
1181
+ :returns: List of bytes containing GIF encoded frame data
1182
+
1183
+ """
1184
+ from io import BytesIO
1185
+
1186
+ class Collector(BytesIO):
1187
+ data = []
1188
+
1189
+ def write(self, data: Buffer) -> int:
1190
+ self.data.append(data)
1191
+ return len(data)
1192
+
1193
+ im.load() # make sure raster data is available
1194
+
1195
+ fp = Collector()
1196
+
1197
+ _write_frame_data(fp, im, offset, params)
1198
+
1199
+ return fp.data
1200
+
1201
+
1202
+ # --------------------------------------------------------------------
1203
+ # Registry
1204
+
1205
+ Image.register_open(GifImageFile.format, GifImageFile, _accept)
1206
+ Image.register_save(GifImageFile.format, _save)
1207
+ Image.register_save_all(GifImageFile.format, _save_all)
1208
+ Image.register_extension(GifImageFile.format, ".gif")
1209
+ Image.register_mime(GifImageFile.format, "image/gif")
1210
+
1211
+ #
1212
+ # Uncomment the following line if you wish to use NETPBM/PBMPLUS
1213
+ # instead of the built-in "uncompressed" GIF encoder
1214
+
1215
+ # Image.register_save(GifImageFile.format, _save_netpbm)
.venv/Lib/site-packages/PIL/GimpGradientFile.py ADDED
@@ -0,0 +1,153 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #
2
+ # Python Imaging Library
3
+ # $Id$
4
+ #
5
+ # stuff to read (and render) GIMP gradient files
6
+ #
7
+ # History:
8
+ # 97-08-23 fl Created
9
+ #
10
+ # Copyright (c) Secret Labs AB 1997.
11
+ # Copyright (c) Fredrik Lundh 1997.
12
+ #
13
+ # See the README file for information on usage and redistribution.
14
+ #
15
+
16
+ """
17
+ Stuff to translate curve segments to palette values (derived from
18
+ the corresponding code in GIMP, written by Federico Mena Quintero.
19
+ See the GIMP distribution for more information.)
20
+ """
21
+ from __future__ import annotations
22
+
23
+ from math import log, pi, sin, sqrt
24
+
25
+ from ._binary import o8
26
+
27
+ TYPE_CHECKING = False
28
+ if TYPE_CHECKING:
29
+ from collections.abc import Callable
30
+ from typing import IO
31
+
32
+ EPSILON = 1e-10
33
+ """""" # Enable auto-doc for data member
34
+
35
+
36
+ def linear(middle: float, pos: float) -> float:
37
+ if pos <= middle:
38
+ if middle < EPSILON:
39
+ return 0.0
40
+ else:
41
+ return 0.5 * pos / middle
42
+ else:
43
+ pos = pos - middle
44
+ middle = 1.0 - middle
45
+ if middle < EPSILON:
46
+ return 1.0
47
+ else:
48
+ return 0.5 + 0.5 * pos / middle
49
+
50
+
51
+ def curved(middle: float, pos: float) -> float:
52
+ return pos ** (log(0.5) / log(max(middle, EPSILON)))
53
+
54
+
55
+ def sine(middle: float, pos: float) -> float:
56
+ return (sin((-pi / 2.0) + pi * linear(middle, pos)) + 1.0) / 2.0
57
+
58
+
59
+ def sphere_increasing(middle: float, pos: float) -> float:
60
+ return sqrt(1.0 - (linear(middle, pos) - 1.0) ** 2)
61
+
62
+
63
+ def sphere_decreasing(middle: float, pos: float) -> float:
64
+ return 1.0 - sqrt(1.0 - linear(middle, pos) ** 2)
65
+
66
+
67
+ SEGMENTS = [linear, curved, sine, sphere_increasing, sphere_decreasing]
68
+ """""" # Enable auto-doc for data member
69
+
70
+
71
+ class GradientFile:
72
+ gradient: (
73
+ list[
74
+ tuple[
75
+ float,
76
+ float,
77
+ float,
78
+ list[float],
79
+ list[float],
80
+ Callable[[float, float], float],
81
+ ]
82
+ ]
83
+ | None
84
+ ) = None
85
+
86
+ def getpalette(self, entries: int = 256) -> tuple[bytes, str]:
87
+ assert self.gradient is not None
88
+ palette = []
89
+
90
+ ix = 0
91
+ x0, x1, xm, rgb0, rgb1, segment = self.gradient[ix]
92
+
93
+ for i in range(entries):
94
+ x = i / (entries - 1)
95
+
96
+ while x1 < x:
97
+ ix += 1
98
+ x0, x1, xm, rgb0, rgb1, segment = self.gradient[ix]
99
+
100
+ w = x1 - x0
101
+
102
+ if w < EPSILON:
103
+ scale = segment(0.5, 0.5)
104
+ else:
105
+ scale = segment((xm - x0) / w, (x - x0) / w)
106
+
107
+ # expand to RGBA
108
+ r = o8(int(255 * ((rgb1[0] - rgb0[0]) * scale + rgb0[0]) + 0.5))
109
+ g = o8(int(255 * ((rgb1[1] - rgb0[1]) * scale + rgb0[1]) + 0.5))
110
+ b = o8(int(255 * ((rgb1[2] - rgb0[2]) * scale + rgb0[2]) + 0.5))
111
+ a = o8(int(255 * ((rgb1[3] - rgb0[3]) * scale + rgb0[3]) + 0.5))
112
+
113
+ # add to palette
114
+ palette.append(r + g + b + a)
115
+
116
+ return b"".join(palette), "RGBA"
117
+
118
+
119
+ class GimpGradientFile(GradientFile):
120
+ """File handler for GIMP's gradient format."""
121
+
122
+ def __init__(self, fp: IO[bytes]) -> None:
123
+ if not fp.readline().startswith(b"GIMP Gradient"):
124
+ msg = "not a GIMP gradient file"
125
+ raise SyntaxError(msg)
126
+
127
+ line = fp.readline()
128
+
129
+ # GIMP 1.2 gradient files don't contain a name, but GIMP 1.3 files do
130
+ if line.startswith(b"Name: "):
131
+ line = fp.readline().strip()
132
+
133
+ count = int(line)
134
+
135
+ self.gradient = []
136
+
137
+ for i in range(count):
138
+ s = fp.readline().split()
139
+ w = [float(x) for x in s[:11]]
140
+
141
+ x0, x1 = w[0], w[2]
142
+ xm = w[1]
143
+ rgb0 = w[3:7]
144
+ rgb1 = w[7:11]
145
+
146
+ segment = SEGMENTS[int(s[11])]
147
+ cspace = int(s[12])
148
+
149
+ if cspace != 0:
150
+ msg = "cannot handle HSV colour space"
151
+ raise OSError(msg)
152
+
153
+ self.gradient.append((x0, x1, xm, rgb0, rgb1, segment))
.venv/Lib/site-packages/PIL/GimpPaletteFile.py ADDED
@@ -0,0 +1,75 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #
2
+ # Python Imaging Library
3
+ # $Id$
4
+ #
5
+ # stuff to read GIMP palette files
6
+ #
7
+ # History:
8
+ # 1997-08-23 fl Created
9
+ # 2004-09-07 fl Support GIMP 2.0 palette files.
10
+ #
11
+ # Copyright (c) Secret Labs AB 1997-2004. All rights reserved.
12
+ # Copyright (c) Fredrik Lundh 1997-2004.
13
+ #
14
+ # See the README file for information on usage and redistribution.
15
+ #
16
+ from __future__ import annotations
17
+
18
+ import re
19
+ from io import BytesIO
20
+
21
+ TYPE_CHECKING = False
22
+ if TYPE_CHECKING:
23
+ from typing import IO
24
+
25
+
26
+ class GimpPaletteFile:
27
+ """File handler for GIMP's palette format."""
28
+
29
+ rawmode = "RGB"
30
+
31
+ def _read(self, fp: IO[bytes], limit: bool = True) -> None:
32
+ if not fp.readline().startswith(b"GIMP Palette"):
33
+ msg = "not a GIMP palette file"
34
+ raise SyntaxError(msg)
35
+
36
+ palette: list[int] = []
37
+ i = 0
38
+ while True:
39
+ if limit and i == 256 + 3:
40
+ break
41
+
42
+ i += 1
43
+ s = fp.readline()
44
+ if not s:
45
+ break
46
+
47
+ # skip fields and comment lines
48
+ if re.match(rb"\w+:|#", s):
49
+ continue
50
+ if limit and len(s) > 100:
51
+ msg = "bad palette file"
52
+ raise SyntaxError(msg)
53
+
54
+ v = s.split(maxsplit=3)
55
+ if len(v) < 3:
56
+ msg = "bad palette entry"
57
+ raise ValueError(msg)
58
+
59
+ palette += (int(v[i]) for i in range(3))
60
+ if limit and len(palette) == 768:
61
+ break
62
+
63
+ self.palette = bytes(palette)
64
+
65
+ def __init__(self, fp: IO[bytes]) -> None:
66
+ self._read(fp)
67
+
68
+ @classmethod
69
+ def frombytes(cls, data: bytes) -> GimpPaletteFile:
70
+ self = cls.__new__(cls)
71
+ self._read(BytesIO(data), False)
72
+ return self
73
+
74
+ def getpalette(self) -> tuple[bytes, str]:
75
+ return self.palette, self.rawmode
.venv/Lib/site-packages/PIL/GribStubImagePlugin.py ADDED
@@ -0,0 +1,75 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #
2
+ # The Python Imaging Library
3
+ # $Id$
4
+ #
5
+ # GRIB stub adapter
6
+ #
7
+ # Copyright (c) 1996-2003 by Fredrik Lundh
8
+ #
9
+ # See the README file for information on usage and redistribution.
10
+ #
11
+ from __future__ import annotations
12
+
13
+ import os
14
+ from typing import IO
15
+
16
+ from . import Image, ImageFile
17
+
18
+ _handler = None
19
+
20
+
21
+ def register_handler(handler: ImageFile.StubHandler | None) -> None:
22
+ """
23
+ Install application-specific GRIB image handler.
24
+
25
+ :param handler: Handler object.
26
+ """
27
+ global _handler
28
+ _handler = handler
29
+
30
+
31
+ # --------------------------------------------------------------------
32
+ # Image adapter
33
+
34
+
35
+ def _accept(prefix: bytes) -> bool:
36
+ return len(prefix) >= 8 and prefix.startswith(b"GRIB") and prefix[7] == 1
37
+
38
+
39
+ class GribStubImageFile(ImageFile.StubImageFile):
40
+ format = "GRIB"
41
+ format_description = "GRIB"
42
+
43
+ def _open(self) -> None:
44
+ if not _accept(self.fp.read(8)):
45
+ msg = "Not a GRIB file"
46
+ raise SyntaxError(msg)
47
+
48
+ self.fp.seek(-8, os.SEEK_CUR)
49
+
50
+ # make something up
51
+ self._mode = "F"
52
+ self._size = 1, 1
53
+
54
+ loader = self._load()
55
+ if loader:
56
+ loader.open(self)
57
+
58
+ def _load(self) -> ImageFile.StubHandler | None:
59
+ return _handler
60
+
61
+
62
+ def _save(im: Image.Image, fp: IO[bytes], filename: str | bytes) -> None:
63
+ if _handler is None or not hasattr(_handler, "save"):
64
+ msg = "GRIB save handler not installed"
65
+ raise OSError(msg)
66
+ _handler.save(im, fp, filename)
67
+
68
+
69
+ # --------------------------------------------------------------------
70
+ # Registry
71
+
72
+ Image.register_open(GribStubImageFile.format, GribStubImageFile, _accept)
73
+ Image.register_save(GribStubImageFile.format, _save)
74
+
75
+ Image.register_extension(GribStubImageFile.format, ".grib")
.venv/Lib/site-packages/PIL/Hdf5StubImagePlugin.py ADDED
@@ -0,0 +1,75 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #
2
+ # The Python Imaging Library
3
+ # $Id$
4
+ #
5
+ # HDF5 stub adapter
6
+ #
7
+ # Copyright (c) 2000-2003 by Fredrik Lundh
8
+ #
9
+ # See the README file for information on usage and redistribution.
10
+ #
11
+ from __future__ import annotations
12
+
13
+ import os
14
+ from typing import IO
15
+
16
+ from . import Image, ImageFile
17
+
18
+ _handler = None
19
+
20
+
21
+ def register_handler(handler: ImageFile.StubHandler | None) -> None:
22
+ """
23
+ Install application-specific HDF5 image handler.
24
+
25
+ :param handler: Handler object.
26
+ """
27
+ global _handler
28
+ _handler = handler
29
+
30
+
31
+ # --------------------------------------------------------------------
32
+ # Image adapter
33
+
34
+
35
+ def _accept(prefix: bytes) -> bool:
36
+ return prefix.startswith(b"\x89HDF\r\n\x1a\n")
37
+
38
+
39
+ class HDF5StubImageFile(ImageFile.StubImageFile):
40
+ format = "HDF5"
41
+ format_description = "HDF5"
42
+
43
+ def _open(self) -> None:
44
+ if not _accept(self.fp.read(8)):
45
+ msg = "Not an HDF file"
46
+ raise SyntaxError(msg)
47
+
48
+ self.fp.seek(-8, os.SEEK_CUR)
49
+
50
+ # make something up
51
+ self._mode = "F"
52
+ self._size = 1, 1
53
+
54
+ loader = self._load()
55
+ if loader:
56
+ loader.open(self)
57
+
58
+ def _load(self) -> ImageFile.StubHandler | None:
59
+ return _handler
60
+
61
+
62
+ def _save(im: Image.Image, fp: IO[bytes], filename: str | bytes) -> None:
63
+ if _handler is None or not hasattr(_handler, "save"):
64
+ msg = "HDF5 save handler not installed"
65
+ raise OSError(msg)
66
+ _handler.save(im, fp, filename)
67
+
68
+
69
+ # --------------------------------------------------------------------
70
+ # Registry
71
+
72
+ Image.register_open(HDF5StubImageFile.format, HDF5StubImageFile, _accept)
73
+ Image.register_save(HDF5StubImageFile.format, _save)
74
+
75
+ Image.register_extensions(HDF5StubImageFile.format, [".h5", ".hdf"])
.venv/Lib/site-packages/PIL/IcnsImagePlugin.py ADDED
@@ -0,0 +1,401 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #
2
+ # The Python Imaging Library.
3
+ # $Id$
4
+ #
5
+ # macOS icns file decoder, based on icns.py by Bob Ippolito.
6
+ #
7
+ # history:
8
+ # 2004-10-09 fl Turned into a PIL plugin; removed 2.3 dependencies.
9
+ # 2020-04-04 Allow saving on all operating systems.
10
+ #
11
+ # Copyright (c) 2004 by Bob Ippolito.
12
+ # Copyright (c) 2004 by Secret Labs.
13
+ # Copyright (c) 2004 by Fredrik Lundh.
14
+ # Copyright (c) 2014 by Alastair Houghton.
15
+ # Copyright (c) 2020 by Pan Jing.
16
+ #
17
+ # See the README file for information on usage and redistribution.
18
+ #
19
+ from __future__ import annotations
20
+
21
+ import io
22
+ import os
23
+ import struct
24
+ import sys
25
+ from typing import IO
26
+
27
+ from . import Image, ImageFile, PngImagePlugin, features
28
+
29
+ enable_jpeg2k = features.check_codec("jpg_2000")
30
+ if enable_jpeg2k:
31
+ from . import Jpeg2KImagePlugin
32
+
33
+ MAGIC = b"icns"
34
+ HEADERSIZE = 8
35
+
36
+
37
+ def nextheader(fobj: IO[bytes]) -> tuple[bytes, int]:
38
+ return struct.unpack(">4sI", fobj.read(HEADERSIZE))
39
+
40
+
41
+ def read_32t(
42
+ fobj: IO[bytes], start_length: tuple[int, int], size: tuple[int, int, int]
43
+ ) -> dict[str, Image.Image]:
44
+ # The 128x128 icon seems to have an extra header for some reason.
45
+ (start, length) = start_length
46
+ fobj.seek(start)
47
+ sig = fobj.read(4)
48
+ if sig != b"\x00\x00\x00\x00":
49
+ msg = "Unknown signature, expecting 0x00000000"
50
+ raise SyntaxError(msg)
51
+ return read_32(fobj, (start + 4, length - 4), size)
52
+
53
+
54
+ def read_32(
55
+ fobj: IO[bytes], start_length: tuple[int, int], size: tuple[int, int, int]
56
+ ) -> dict[str, Image.Image]:
57
+ """
58
+ Read a 32bit RGB icon resource. Seems to be either uncompressed or
59
+ an RLE packbits-like scheme.
60
+ """
61
+ (start, length) = start_length
62
+ fobj.seek(start)
63
+ pixel_size = (size[0] * size[2], size[1] * size[2])
64
+ sizesq = pixel_size[0] * pixel_size[1]
65
+ if length == sizesq * 3:
66
+ # uncompressed ("RGBRGBGB")
67
+ indata = fobj.read(length)
68
+ im = Image.frombuffer("RGB", pixel_size, indata, "raw", "RGB", 0, 1)
69
+ else:
70
+ # decode image
71
+ im = Image.new("RGB", pixel_size, None)
72
+ for band_ix in range(3):
73
+ data = []
74
+ bytesleft = sizesq
75
+ while bytesleft > 0:
76
+ byte = fobj.read(1)
77
+ if not byte:
78
+ break
79
+ byte_int = byte[0]
80
+ if byte_int & 0x80:
81
+ blocksize = byte_int - 125
82
+ byte = fobj.read(1)
83
+ for i in range(blocksize):
84
+ data.append(byte)
85
+ else:
86
+ blocksize = byte_int + 1
87
+ data.append(fobj.read(blocksize))
88
+ bytesleft -= blocksize
89
+ if bytesleft <= 0:
90
+ break
91
+ if bytesleft != 0:
92
+ msg = f"Error reading channel [{repr(bytesleft)} left]"
93
+ raise SyntaxError(msg)
94
+ band = Image.frombuffer("L", pixel_size, b"".join(data), "raw", "L", 0, 1)
95
+ im.im.putband(band.im, band_ix)
96
+ return {"RGB": im}
97
+
98
+
99
+ def read_mk(
100
+ fobj: IO[bytes], start_length: tuple[int, int], size: tuple[int, int, int]
101
+ ) -> dict[str, Image.Image]:
102
+ # Alpha masks seem to be uncompressed
103
+ start = start_length[0]
104
+ fobj.seek(start)
105
+ pixel_size = (size[0] * size[2], size[1] * size[2])
106
+ sizesq = pixel_size[0] * pixel_size[1]
107
+ band = Image.frombuffer("L", pixel_size, fobj.read(sizesq), "raw", "L", 0, 1)
108
+ return {"A": band}
109
+
110
+
111
+ def read_png_or_jpeg2000(
112
+ fobj: IO[bytes], start_length: tuple[int, int], size: tuple[int, int, int]
113
+ ) -> dict[str, Image.Image]:
114
+ (start, length) = start_length
115
+ fobj.seek(start)
116
+ sig = fobj.read(12)
117
+
118
+ im: Image.Image
119
+ if sig.startswith(b"\x89PNG\x0d\x0a\x1a\x0a"):
120
+ fobj.seek(start)
121
+ im = PngImagePlugin.PngImageFile(fobj)
122
+ Image._decompression_bomb_check(im.size)
123
+ return {"RGBA": im}
124
+ elif (
125
+ sig.startswith((b"\xff\x4f\xff\x51", b"\x0d\x0a\x87\x0a"))
126
+ or sig == b"\x00\x00\x00\x0cjP \x0d\x0a\x87\x0a"
127
+ ):
128
+ if not enable_jpeg2k:
129
+ msg = (
130
+ "Unsupported icon subimage format (rebuild PIL "
131
+ "with JPEG 2000 support to fix this)"
132
+ )
133
+ raise ValueError(msg)
134
+ # j2k, jpc or j2c
135
+ fobj.seek(start)
136
+ jp2kstream = fobj.read(length)
137
+ f = io.BytesIO(jp2kstream)
138
+ im = Jpeg2KImagePlugin.Jpeg2KImageFile(f)
139
+ Image._decompression_bomb_check(im.size)
140
+ if im.mode != "RGBA":
141
+ im = im.convert("RGBA")
142
+ return {"RGBA": im}
143
+ else:
144
+ msg = "Unsupported icon subimage format"
145
+ raise ValueError(msg)
146
+
147
+
148
+ class IcnsFile:
149
+ SIZES = {
150
+ (512, 512, 2): [(b"ic10", read_png_or_jpeg2000)],
151
+ (512, 512, 1): [(b"ic09", read_png_or_jpeg2000)],
152
+ (256, 256, 2): [(b"ic14", read_png_or_jpeg2000)],
153
+ (256, 256, 1): [(b"ic08", read_png_or_jpeg2000)],
154
+ (128, 128, 2): [(b"ic13", read_png_or_jpeg2000)],
155
+ (128, 128, 1): [
156
+ (b"ic07", read_png_or_jpeg2000),
157
+ (b"it32", read_32t),
158
+ (b"t8mk", read_mk),
159
+ ],
160
+ (64, 64, 1): [(b"icp6", read_png_or_jpeg2000)],
161
+ (32, 32, 2): [(b"ic12", read_png_or_jpeg2000)],
162
+ (48, 48, 1): [(b"ih32", read_32), (b"h8mk", read_mk)],
163
+ (32, 32, 1): [
164
+ (b"icp5", read_png_or_jpeg2000),
165
+ (b"il32", read_32),
166
+ (b"l8mk", read_mk),
167
+ ],
168
+ (16, 16, 2): [(b"ic11", read_png_or_jpeg2000)],
169
+ (16, 16, 1): [
170
+ (b"icp4", read_png_or_jpeg2000),
171
+ (b"is32", read_32),
172
+ (b"s8mk", read_mk),
173
+ ],
174
+ }
175
+
176
+ def __init__(self, fobj: IO[bytes]) -> None:
177
+ """
178
+ fobj is a file-like object as an icns resource
179
+ """
180
+ # signature : (start, length)
181
+ self.dct = {}
182
+ self.fobj = fobj
183
+ sig, filesize = nextheader(fobj)
184
+ if not _accept(sig):
185
+ msg = "not an icns file"
186
+ raise SyntaxError(msg)
187
+ i = HEADERSIZE
188
+ while i < filesize:
189
+ sig, blocksize = nextheader(fobj)
190
+ if blocksize <= 0:
191
+ msg = "invalid block header"
192
+ raise SyntaxError(msg)
193
+ i += HEADERSIZE
194
+ blocksize -= HEADERSIZE
195
+ self.dct[sig] = (i, blocksize)
196
+ fobj.seek(blocksize, io.SEEK_CUR)
197
+ i += blocksize
198
+
199
+ def itersizes(self) -> list[tuple[int, int, int]]:
200
+ sizes = []
201
+ for size, fmts in self.SIZES.items():
202
+ for fmt, reader in fmts:
203
+ if fmt in self.dct:
204
+ sizes.append(size)
205
+ break
206
+ return sizes
207
+
208
+ def bestsize(self) -> tuple[int, int, int]:
209
+ sizes = self.itersizes()
210
+ if not sizes:
211
+ msg = "No 32bit icon resources found"
212
+ raise SyntaxError(msg)
213
+ return max(sizes)
214
+
215
+ def dataforsize(self, size: tuple[int, int, int]) -> dict[str, Image.Image]:
216
+ """
217
+ Get an icon resource as {channel: array}. Note that
218
+ the arrays are bottom-up like windows bitmaps and will likely
219
+ need to be flipped or transposed in some way.
220
+ """
221
+ dct = {}
222
+ for code, reader in self.SIZES[size]:
223
+ desc = self.dct.get(code)
224
+ if desc is not None:
225
+ dct.update(reader(self.fobj, desc, size))
226
+ return dct
227
+
228
+ def getimage(
229
+ self, size: tuple[int, int] | tuple[int, int, int] | None = None
230
+ ) -> Image.Image:
231
+ if size is None:
232
+ size = self.bestsize()
233
+ elif len(size) == 2:
234
+ size = (size[0], size[1], 1)
235
+ channels = self.dataforsize(size)
236
+
237
+ im = channels.get("RGBA")
238
+ if im:
239
+ return im
240
+
241
+ im = channels["RGB"].copy()
242
+ try:
243
+ im.putalpha(channels["A"])
244
+ except KeyError:
245
+ pass
246
+ return im
247
+
248
+
249
+ ##
250
+ # Image plugin for Mac OS icons.
251
+
252
+
253
+ class IcnsImageFile(ImageFile.ImageFile):
254
+ """
255
+ PIL image support for Mac OS .icns files.
256
+ Chooses the best resolution, but will possibly load
257
+ a different size image if you mutate the size attribute
258
+ before calling 'load'.
259
+
260
+ The info dictionary has a key 'sizes' that is a list
261
+ of sizes that the icns file has.
262
+ """
263
+
264
+ format = "ICNS"
265
+ format_description = "Mac OS icns resource"
266
+
267
+ def _open(self) -> None:
268
+ self.icns = IcnsFile(self.fp)
269
+ self._mode = "RGBA"
270
+ self.info["sizes"] = self.icns.itersizes()
271
+ self.best_size = self.icns.bestsize()
272
+ self.size = (
273
+ self.best_size[0] * self.best_size[2],
274
+ self.best_size[1] * self.best_size[2],
275
+ )
276
+
277
+ @property
278
+ def size(self) -> tuple[int, int]:
279
+ return self._size
280
+
281
+ @size.setter
282
+ def size(self, value: tuple[int, int]) -> None:
283
+ # Check that a matching size exists,
284
+ # or that there is a scale that would create a size that matches
285
+ for size in self.info["sizes"]:
286
+ simple_size = size[0] * size[2], size[1] * size[2]
287
+ scale = simple_size[0] // value[0]
288
+ if simple_size[1] / value[1] == scale:
289
+ self._size = value
290
+ return
291
+ msg = "This is not one of the allowed sizes of this image"
292
+ raise ValueError(msg)
293
+
294
+ def load(self, scale: int | None = None) -> Image.core.PixelAccess | None:
295
+ if scale is not None:
296
+ width, height = self.size[:2]
297
+ self.size = width * scale, height * scale
298
+ self.best_size = width, height, scale
299
+
300
+ px = Image.Image.load(self)
301
+ if self._im is not None and self.im.size == self.size:
302
+ # Already loaded
303
+ return px
304
+ self.load_prepare()
305
+ # This is likely NOT the best way to do it, but whatever.
306
+ im = self.icns.getimage(self.best_size)
307
+
308
+ # If this is a PNG or JPEG 2000, it won't be loaded yet
309
+ px = im.load()
310
+
311
+ self.im = im.im
312
+ self._mode = im.mode
313
+ self.size = im.size
314
+
315
+ return px
316
+
317
+
318
+ def _save(im: Image.Image, fp: IO[bytes], filename: str | bytes) -> None:
319
+ """
320
+ Saves the image as a series of PNG files,
321
+ that are then combined into a .icns file.
322
+ """
323
+ if hasattr(fp, "flush"):
324
+ fp.flush()
325
+
326
+ sizes = {
327
+ b"ic07": 128,
328
+ b"ic08": 256,
329
+ b"ic09": 512,
330
+ b"ic10": 1024,
331
+ b"ic11": 32,
332
+ b"ic12": 64,
333
+ b"ic13": 256,
334
+ b"ic14": 512,
335
+ }
336
+ provided_images = {im.width: im for im in im.encoderinfo.get("append_images", [])}
337
+ size_streams = {}
338
+ for size in set(sizes.values()):
339
+ image = (
340
+ provided_images[size]
341
+ if size in provided_images
342
+ else im.resize((size, size))
343
+ )
344
+
345
+ temp = io.BytesIO()
346
+ image.save(temp, "png")
347
+ size_streams[size] = temp.getvalue()
348
+
349
+ entries = []
350
+ for type, size in sizes.items():
351
+ stream = size_streams[size]
352
+ entries.append((type, HEADERSIZE + len(stream), stream))
353
+
354
+ # Header
355
+ fp.write(MAGIC)
356
+ file_length = HEADERSIZE # Header
357
+ file_length += HEADERSIZE + 8 * len(entries) # TOC
358
+ file_length += sum(entry[1] for entry in entries)
359
+ fp.write(struct.pack(">i", file_length))
360
+
361
+ # TOC
362
+ fp.write(b"TOC ")
363
+ fp.write(struct.pack(">i", HEADERSIZE + len(entries) * HEADERSIZE))
364
+ for entry in entries:
365
+ fp.write(entry[0])
366
+ fp.write(struct.pack(">i", entry[1]))
367
+
368
+ # Data
369
+ for entry in entries:
370
+ fp.write(entry[0])
371
+ fp.write(struct.pack(">i", entry[1]))
372
+ fp.write(entry[2])
373
+
374
+ if hasattr(fp, "flush"):
375
+ fp.flush()
376
+
377
+
378
+ def _accept(prefix: bytes) -> bool:
379
+ return prefix.startswith(MAGIC)
380
+
381
+
382
+ Image.register_open(IcnsImageFile.format, IcnsImageFile, _accept)
383
+ Image.register_extension(IcnsImageFile.format, ".icns")
384
+
385
+ Image.register_save(IcnsImageFile.format, _save)
386
+ Image.register_mime(IcnsImageFile.format, "image/icns")
387
+
388
+ if __name__ == "__main__":
389
+ if len(sys.argv) < 2:
390
+ print("Syntax: python3 IcnsImagePlugin.py [file]")
391
+ sys.exit()
392
+
393
+ with open(sys.argv[1], "rb") as fp:
394
+ imf = IcnsImageFile(fp)
395
+ for size in imf.info["sizes"]:
396
+ width, height, scale = imf.size = size
397
+ imf.save(f"out-{width}-{height}-{scale}.png")
398
+ with Image.open(sys.argv[1]) as im:
399
+ im.save("out.png")
400
+ if sys.platform == "windows":
401
+ os.startfile("out.png")
.venv/Lib/site-packages/PIL/IcoImagePlugin.py ADDED
@@ -0,0 +1,381 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #
2
+ # The Python Imaging Library.
3
+ # $Id$
4
+ #
5
+ # Windows Icon support for PIL
6
+ #
7
+ # History:
8
+ # 96-05-27 fl Created
9
+ #
10
+ # Copyright (c) Secret Labs AB 1997.
11
+ # Copyright (c) Fredrik Lundh 1996.
12
+ #
13
+ # See the README file for information on usage and redistribution.
14
+ #
15
+
16
+ # This plugin is a refactored version of Win32IconImagePlugin by Bryan Davis
17
+ # <casadebender@gmail.com>.
18
+ # https://code.google.com/archive/p/casadebender/wikis/Win32IconImagePlugin.wiki
19
+ #
20
+ # Icon format references:
21
+ # * https://en.wikipedia.org/wiki/ICO_(file_format)
22
+ # * https://msdn.microsoft.com/en-us/library/ms997538.aspx
23
+ from __future__ import annotations
24
+
25
+ import warnings
26
+ from io import BytesIO
27
+ from math import ceil, log
28
+ from typing import IO, NamedTuple
29
+
30
+ from . import BmpImagePlugin, Image, ImageFile, PngImagePlugin
31
+ from ._binary import i16le as i16
32
+ from ._binary import i32le as i32
33
+ from ._binary import o8
34
+ from ._binary import o16le as o16
35
+ from ._binary import o32le as o32
36
+
37
+ #
38
+ # --------------------------------------------------------------------
39
+
40
+ _MAGIC = b"\0\0\1\0"
41
+
42
+
43
+ def _save(im: Image.Image, fp: IO[bytes], filename: str | bytes) -> None:
44
+ fp.write(_MAGIC) # (2+2)
45
+ bmp = im.encoderinfo.get("bitmap_format") == "bmp"
46
+ sizes = im.encoderinfo.get(
47
+ "sizes",
48
+ [(16, 16), (24, 24), (32, 32), (48, 48), (64, 64), (128, 128), (256, 256)],
49
+ )
50
+ frames = []
51
+ provided_ims = [im] + im.encoderinfo.get("append_images", [])
52
+ width, height = im.size
53
+ for size in sorted(set(sizes)):
54
+ if size[0] > width or size[1] > height or size[0] > 256 or size[1] > 256:
55
+ continue
56
+
57
+ for provided_im in provided_ims:
58
+ if provided_im.size != size:
59
+ continue
60
+ frames.append(provided_im)
61
+ if bmp:
62
+ bits = BmpImagePlugin.SAVE[provided_im.mode][1]
63
+ bits_used = [bits]
64
+ for other_im in provided_ims:
65
+ if other_im.size != size:
66
+ continue
67
+ bits = BmpImagePlugin.SAVE[other_im.mode][1]
68
+ if bits not in bits_used:
69
+ # Another image has been supplied for this size
70
+ # with a different bit depth
71
+ frames.append(other_im)
72
+ bits_used.append(bits)
73
+ break
74
+ else:
75
+ # TODO: invent a more convenient method for proportional scalings
76
+ frame = provided_im.copy()
77
+ frame.thumbnail(size, Image.Resampling.LANCZOS, reducing_gap=None)
78
+ frames.append(frame)
79
+ fp.write(o16(len(frames))) # idCount(2)
80
+ offset = fp.tell() + len(frames) * 16
81
+ for frame in frames:
82
+ width, height = frame.size
83
+ # 0 means 256
84
+ fp.write(o8(width if width < 256 else 0)) # bWidth(1)
85
+ fp.write(o8(height if height < 256 else 0)) # bHeight(1)
86
+
87
+ bits, colors = BmpImagePlugin.SAVE[frame.mode][1:] if bmp else (32, 0)
88
+ fp.write(o8(colors)) # bColorCount(1)
89
+ fp.write(b"\0") # bReserved(1)
90
+ fp.write(b"\0\0") # wPlanes(2)
91
+ fp.write(o16(bits)) # wBitCount(2)
92
+
93
+ image_io = BytesIO()
94
+ if bmp:
95
+ frame.save(image_io, "dib")
96
+
97
+ if bits != 32:
98
+ and_mask = Image.new("1", size)
99
+ ImageFile._save(
100
+ and_mask,
101
+ image_io,
102
+ [ImageFile._Tile("raw", (0, 0) + size, 0, ("1", 0, -1))],
103
+ )
104
+ else:
105
+ frame.save(image_io, "png")
106
+ image_io.seek(0)
107
+ image_bytes = image_io.read()
108
+ if bmp:
109
+ image_bytes = image_bytes[:8] + o32(height * 2) + image_bytes[12:]
110
+ bytes_len = len(image_bytes)
111
+ fp.write(o32(bytes_len)) # dwBytesInRes(4)
112
+ fp.write(o32(offset)) # dwImageOffset(4)
113
+ current = fp.tell()
114
+ fp.seek(offset)
115
+ fp.write(image_bytes)
116
+ offset = offset + bytes_len
117
+ fp.seek(current)
118
+
119
+
120
+ def _accept(prefix: bytes) -> bool:
121
+ return prefix.startswith(_MAGIC)
122
+
123
+
124
+ class IconHeader(NamedTuple):
125
+ width: int
126
+ height: int
127
+ nb_color: int
128
+ reserved: int
129
+ planes: int
130
+ bpp: int
131
+ size: int
132
+ offset: int
133
+ dim: tuple[int, int]
134
+ square: int
135
+ color_depth: int
136
+
137
+
138
+ class IcoFile:
139
+ def __init__(self, buf: IO[bytes]) -> None:
140
+ """
141
+ Parse image from file-like object containing ico file data
142
+ """
143
+
144
+ # check magic
145
+ s = buf.read(6)
146
+ if not _accept(s):
147
+ msg = "not an ICO file"
148
+ raise SyntaxError(msg)
149
+
150
+ self.buf = buf
151
+ self.entry = []
152
+
153
+ # Number of items in file
154
+ self.nb_items = i16(s, 4)
155
+
156
+ # Get headers for each item
157
+ for i in range(self.nb_items):
158
+ s = buf.read(16)
159
+
160
+ # See Wikipedia
161
+ width = s[0] or 256
162
+ height = s[1] or 256
163
+
164
+ # No. of colors in image (0 if >=8bpp)
165
+ nb_color = s[2]
166
+ bpp = i16(s, 6)
167
+ icon_header = IconHeader(
168
+ width=width,
169
+ height=height,
170
+ nb_color=nb_color,
171
+ reserved=s[3],
172
+ planes=i16(s, 4),
173
+ bpp=i16(s, 6),
174
+ size=i32(s, 8),
175
+ offset=i32(s, 12),
176
+ dim=(width, height),
177
+ square=width * height,
178
+ # See Wikipedia notes about color depth.
179
+ # We need this just to differ images with equal sizes
180
+ color_depth=bpp or (nb_color != 0 and ceil(log(nb_color, 2))) or 256,
181
+ )
182
+
183
+ self.entry.append(icon_header)
184
+
185
+ self.entry = sorted(self.entry, key=lambda x: x.color_depth)
186
+ # ICO images are usually squares
187
+ self.entry = sorted(self.entry, key=lambda x: x.square, reverse=True)
188
+
189
+ def sizes(self) -> set[tuple[int, int]]:
190
+ """
191
+ Get a set of all available icon sizes and color depths.
192
+ """
193
+ return {(h.width, h.height) for h in self.entry}
194
+
195
+ def getentryindex(self, size: tuple[int, int], bpp: int | bool = False) -> int:
196
+ for i, h in enumerate(self.entry):
197
+ if size == h.dim and (bpp is False or bpp == h.color_depth):
198
+ return i
199
+ return 0
200
+
201
+ def getimage(self, size: tuple[int, int], bpp: int | bool = False) -> Image.Image:
202
+ """
203
+ Get an image from the icon
204
+ """
205
+ return self.frame(self.getentryindex(size, bpp))
206
+
207
+ def frame(self, idx: int) -> Image.Image:
208
+ """
209
+ Get an image from frame idx
210
+ """
211
+
212
+ header = self.entry[idx]
213
+
214
+ self.buf.seek(header.offset)
215
+ data = self.buf.read(8)
216
+ self.buf.seek(header.offset)
217
+
218
+ im: Image.Image
219
+ if data[:8] == PngImagePlugin._MAGIC:
220
+ # png frame
221
+ im = PngImagePlugin.PngImageFile(self.buf)
222
+ Image._decompression_bomb_check(im.size)
223
+ else:
224
+ # XOR + AND mask bmp frame
225
+ im = BmpImagePlugin.DibImageFile(self.buf)
226
+ Image._decompression_bomb_check(im.size)
227
+
228
+ # change tile dimension to only encompass XOR image
229
+ im._size = (im.size[0], int(im.size[1] / 2))
230
+ d, e, o, a = im.tile[0]
231
+ im.tile[0] = ImageFile._Tile(d, (0, 0) + im.size, o, a)
232
+
233
+ # figure out where AND mask image starts
234
+ if header.bpp == 32:
235
+ # 32-bit color depth icon image allows semitransparent areas
236
+ # PIL's DIB format ignores transparency bits, recover them.
237
+ # The DIB is packed in BGRX byte order where X is the alpha
238
+ # channel.
239
+
240
+ # Back up to start of bmp data
241
+ self.buf.seek(o)
242
+ # extract every 4th byte (eg. 3,7,11,15,...)
243
+ alpha_bytes = self.buf.read(im.size[0] * im.size[1] * 4)[3::4]
244
+
245
+ # convert to an 8bpp grayscale image
246
+ try:
247
+ mask = Image.frombuffer(
248
+ "L", # 8bpp
249
+ im.size, # (w, h)
250
+ alpha_bytes, # source chars
251
+ "raw", # raw decoder
252
+ ("L", 0, -1), # 8bpp inverted, unpadded, reversed
253
+ )
254
+ except ValueError:
255
+ if ImageFile.LOAD_TRUNCATED_IMAGES:
256
+ mask = None
257
+ else:
258
+ raise
259
+ else:
260
+ # get AND image from end of bitmap
261
+ w = im.size[0]
262
+ if (w % 32) > 0:
263
+ # bitmap row data is aligned to word boundaries
264
+ w += 32 - (im.size[0] % 32)
265
+
266
+ # the total mask data is
267
+ # padded row size * height / bits per char
268
+
269
+ total_bytes = int((w * im.size[1]) / 8)
270
+ and_mask_offset = header.offset + header.size - total_bytes
271
+
272
+ self.buf.seek(and_mask_offset)
273
+ mask_data = self.buf.read(total_bytes)
274
+
275
+ # convert raw data to image
276
+ try:
277
+ mask = Image.frombuffer(
278
+ "1", # 1 bpp
279
+ im.size, # (w, h)
280
+ mask_data, # source chars
281
+ "raw", # raw decoder
282
+ ("1;I", int(w / 8), -1), # 1bpp inverted, padded, reversed
283
+ )
284
+ except ValueError:
285
+ if ImageFile.LOAD_TRUNCATED_IMAGES:
286
+ mask = None
287
+ else:
288
+ raise
289
+
290
+ # now we have two images, im is XOR image and mask is AND image
291
+
292
+ # apply mask image as alpha channel
293
+ if mask:
294
+ im = im.convert("RGBA")
295
+ im.putalpha(mask)
296
+
297
+ return im
298
+
299
+
300
+ ##
301
+ # Image plugin for Windows Icon files.
302
+
303
+
304
+ class IcoImageFile(ImageFile.ImageFile):
305
+ """
306
+ PIL read-only image support for Microsoft Windows .ico files.
307
+
308
+ By default the largest resolution image in the file will be loaded. This
309
+ can be changed by altering the 'size' attribute before calling 'load'.
310
+
311
+ The info dictionary has a key 'sizes' that is a list of the sizes available
312
+ in the icon file.
313
+
314
+ Handles classic, XP and Vista icon formats.
315
+
316
+ When saving, PNG compression is used. Support for this was only added in
317
+ Windows Vista. If you are unable to view the icon in Windows, convert the
318
+ image to "RGBA" mode before saving.
319
+
320
+ This plugin is a refactored version of Win32IconImagePlugin by Bryan Davis
321
+ <casadebender@gmail.com>.
322
+ https://code.google.com/archive/p/casadebender/wikis/Win32IconImagePlugin.wiki
323
+ """
324
+
325
+ format = "ICO"
326
+ format_description = "Windows Icon"
327
+
328
+ def _open(self) -> None:
329
+ self.ico = IcoFile(self.fp)
330
+ self.info["sizes"] = self.ico.sizes()
331
+ self.size = self.ico.entry[0].dim
332
+ self.load()
333
+
334
+ @property
335
+ def size(self) -> tuple[int, int]:
336
+ return self._size
337
+
338
+ @size.setter
339
+ def size(self, value: tuple[int, int]) -> None:
340
+ if value not in self.info["sizes"]:
341
+ msg = "This is not one of the allowed sizes of this image"
342
+ raise ValueError(msg)
343
+ self._size = value
344
+
345
+ def load(self) -> Image.core.PixelAccess | None:
346
+ if self._im is not None and self.im.size == self.size:
347
+ # Already loaded
348
+ return Image.Image.load(self)
349
+ im = self.ico.getimage(self.size)
350
+ # if tile is PNG, it won't really be loaded yet
351
+ im.load()
352
+ self.im = im.im
353
+ self._mode = im.mode
354
+ if im.palette:
355
+ self.palette = im.palette
356
+ if im.size != self.size:
357
+ warnings.warn("Image was not the expected size")
358
+
359
+ index = self.ico.getentryindex(self.size)
360
+ sizes = list(self.info["sizes"])
361
+ sizes[index] = im.size
362
+ self.info["sizes"] = set(sizes)
363
+
364
+ self.size = im.size
365
+ return Image.Image.load(self)
366
+
367
+ def load_seek(self, pos: int) -> None:
368
+ # Flag the ImageFile.Parser so that it
369
+ # just does all the decode at the end.
370
+ pass
371
+
372
+
373
+ #
374
+ # --------------------------------------------------------------------
375
+
376
+
377
+ Image.register_open(IcoImageFile.format, IcoImageFile, _accept)
378
+ Image.register_save(IcoImageFile.format, _save)
379
+ Image.register_extension(IcoImageFile.format, ".ico")
380
+
381
+ Image.register_mime(IcoImageFile.format, "image/x-icon")
.venv/Lib/site-packages/PIL/ImImagePlugin.py ADDED
@@ -0,0 +1,389 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #
2
+ # The Python Imaging Library.
3
+ # $Id$
4
+ #
5
+ # IFUNC IM file handling for PIL
6
+ #
7
+ # history:
8
+ # 1995-09-01 fl Created.
9
+ # 1997-01-03 fl Save palette images
10
+ # 1997-01-08 fl Added sequence support
11
+ # 1997-01-23 fl Added P and RGB save support
12
+ # 1997-05-31 fl Read floating point images
13
+ # 1997-06-22 fl Save floating point images
14
+ # 1997-08-27 fl Read and save 1-bit images
15
+ # 1998-06-25 fl Added support for RGB+LUT images
16
+ # 1998-07-02 fl Added support for YCC images
17
+ # 1998-07-15 fl Renamed offset attribute to avoid name clash
18
+ # 1998-12-29 fl Added I;16 support
19
+ # 2001-02-17 fl Use 're' instead of 'regex' (Python 2.1) (0.7)
20
+ # 2003-09-26 fl Added LA/PA support
21
+ #
22
+ # Copyright (c) 1997-2003 by Secret Labs AB.
23
+ # Copyright (c) 1995-2001 by Fredrik Lundh.
24
+ #
25
+ # See the README file for information on usage and redistribution.
26
+ #
27
+ from __future__ import annotations
28
+
29
+ import os
30
+ import re
31
+ from typing import IO, Any
32
+
33
+ from . import Image, ImageFile, ImagePalette
34
+ from ._util import DeferredError
35
+
36
+ # --------------------------------------------------------------------
37
+ # Standard tags
38
+
39
+ COMMENT = "Comment"
40
+ DATE = "Date"
41
+ EQUIPMENT = "Digitalization equipment"
42
+ FRAMES = "File size (no of images)"
43
+ LUT = "Lut"
44
+ NAME = "Name"
45
+ SCALE = "Scale (x,y)"
46
+ SIZE = "Image size (x*y)"
47
+ MODE = "Image type"
48
+
49
+ TAGS = {
50
+ COMMENT: 0,
51
+ DATE: 0,
52
+ EQUIPMENT: 0,
53
+ FRAMES: 0,
54
+ LUT: 0,
55
+ NAME: 0,
56
+ SCALE: 0,
57
+ SIZE: 0,
58
+ MODE: 0,
59
+ }
60
+
61
+ OPEN = {
62
+ # ifunc93/p3cfunc formats
63
+ "0 1 image": ("1", "1"),
64
+ "L 1 image": ("1", "1"),
65
+ "Greyscale image": ("L", "L"),
66
+ "Grayscale image": ("L", "L"),
67
+ "RGB image": ("RGB", "RGB;L"),
68
+ "RLB image": ("RGB", "RLB"),
69
+ "RYB image": ("RGB", "RLB"),
70
+ "B1 image": ("1", "1"),
71
+ "B2 image": ("P", "P;2"),
72
+ "B4 image": ("P", "P;4"),
73
+ "X 24 image": ("RGB", "RGB"),
74
+ "L 32 S image": ("I", "I;32"),
75
+ "L 32 F image": ("F", "F;32"),
76
+ # old p3cfunc formats
77
+ "RGB3 image": ("RGB", "RGB;T"),
78
+ "RYB3 image": ("RGB", "RYB;T"),
79
+ # extensions
80
+ "LA image": ("LA", "LA;L"),
81
+ "PA image": ("LA", "PA;L"),
82
+ "RGBA image": ("RGBA", "RGBA;L"),
83
+ "RGBX image": ("RGB", "RGBX;L"),
84
+ "CMYK image": ("CMYK", "CMYK;L"),
85
+ "YCC image": ("YCbCr", "YCbCr;L"),
86
+ }
87
+
88
+ # ifunc95 extensions
89
+ for i in ["8", "8S", "16", "16S", "32", "32F"]:
90
+ OPEN[f"L {i} image"] = ("F", f"F;{i}")
91
+ OPEN[f"L*{i} image"] = ("F", f"F;{i}")
92
+ for i in ["16", "16L", "16B"]:
93
+ OPEN[f"L {i} image"] = (f"I;{i}", f"I;{i}")
94
+ OPEN[f"L*{i} image"] = (f"I;{i}", f"I;{i}")
95
+ for i in ["32S"]:
96
+ OPEN[f"L {i} image"] = ("I", f"I;{i}")
97
+ OPEN[f"L*{i} image"] = ("I", f"I;{i}")
98
+ for j in range(2, 33):
99
+ OPEN[f"L*{j} image"] = ("F", f"F;{j}")
100
+
101
+
102
+ # --------------------------------------------------------------------
103
+ # Read IM directory
104
+
105
+ split = re.compile(rb"^([A-Za-z][^:]*):[ \t]*(.*)[ \t]*$")
106
+
107
+
108
+ def number(s: Any) -> float:
109
+ try:
110
+ return int(s)
111
+ except ValueError:
112
+ return float(s)
113
+
114
+
115
+ ##
116
+ # Image plugin for the IFUNC IM file format.
117
+
118
+
119
+ class ImImageFile(ImageFile.ImageFile):
120
+ format = "IM"
121
+ format_description = "IFUNC Image Memory"
122
+ _close_exclusive_fp_after_loading = False
123
+
124
+ def _open(self) -> None:
125
+ # Quick rejection: if there's not an LF among the first
126
+ # 100 bytes, this is (probably) not a text header.
127
+
128
+ if b"\n" not in self.fp.read(100):
129
+ msg = "not an IM file"
130
+ raise SyntaxError(msg)
131
+ self.fp.seek(0)
132
+
133
+ n = 0
134
+
135
+ # Default values
136
+ self.info[MODE] = "L"
137
+ self.info[SIZE] = (512, 512)
138
+ self.info[FRAMES] = 1
139
+
140
+ self.rawmode = "L"
141
+
142
+ while True:
143
+ s = self.fp.read(1)
144
+
145
+ # Some versions of IFUNC uses \n\r instead of \r\n...
146
+ if s == b"\r":
147
+ continue
148
+
149
+ if not s or s == b"\0" or s == b"\x1a":
150
+ break
151
+
152
+ # FIXME: this may read whole file if not a text file
153
+ s = s + self.fp.readline()
154
+
155
+ if len(s) > 100:
156
+ msg = "not an IM file"
157
+ raise SyntaxError(msg)
158
+
159
+ if s.endswith(b"\r\n"):
160
+ s = s[:-2]
161
+ elif s.endswith(b"\n"):
162
+ s = s[:-1]
163
+
164
+ try:
165
+ m = split.match(s)
166
+ except re.error as e:
167
+ msg = "not an IM file"
168
+ raise SyntaxError(msg) from e
169
+
170
+ if m:
171
+ k, v = m.group(1, 2)
172
+
173
+ # Don't know if this is the correct encoding,
174
+ # but a decent guess (I guess)
175
+ k = k.decode("latin-1", "replace")
176
+ v = v.decode("latin-1", "replace")
177
+
178
+ # Convert value as appropriate
179
+ if k in [FRAMES, SCALE, SIZE]:
180
+ v = v.replace("*", ",")
181
+ v = tuple(map(number, v.split(",")))
182
+ if len(v) == 1:
183
+ v = v[0]
184
+ elif k == MODE and v in OPEN:
185
+ v, self.rawmode = OPEN[v]
186
+
187
+ # Add to dictionary. Note that COMMENT tags are
188
+ # combined into a list of strings.
189
+ if k == COMMENT:
190
+ if k in self.info:
191
+ self.info[k].append(v)
192
+ else:
193
+ self.info[k] = [v]
194
+ else:
195
+ self.info[k] = v
196
+
197
+ if k in TAGS:
198
+ n += 1
199
+
200
+ else:
201
+ msg = f"Syntax error in IM header: {s.decode('ascii', 'replace')}"
202
+ raise SyntaxError(msg)
203
+
204
+ if not n:
205
+ msg = "Not an IM file"
206
+ raise SyntaxError(msg)
207
+
208
+ # Basic attributes
209
+ self._size = self.info[SIZE]
210
+ self._mode = self.info[MODE]
211
+
212
+ # Skip forward to start of image data
213
+ while s and not s.startswith(b"\x1a"):
214
+ s = self.fp.read(1)
215
+ if not s:
216
+ msg = "File truncated"
217
+ raise SyntaxError(msg)
218
+
219
+ if LUT in self.info:
220
+ # convert lookup table to palette or lut attribute
221
+ palette = self.fp.read(768)
222
+ greyscale = 1 # greyscale palette
223
+ linear = 1 # linear greyscale palette
224
+ for i in range(256):
225
+ if palette[i] == palette[i + 256] == palette[i + 512]:
226
+ if palette[i] != i:
227
+ linear = 0
228
+ else:
229
+ greyscale = 0
230
+ if self.mode in ["L", "LA", "P", "PA"]:
231
+ if greyscale:
232
+ if not linear:
233
+ self.lut = list(palette[:256])
234
+ else:
235
+ if self.mode in ["L", "P"]:
236
+ self._mode = self.rawmode = "P"
237
+ elif self.mode in ["LA", "PA"]:
238
+ self._mode = "PA"
239
+ self.rawmode = "PA;L"
240
+ self.palette = ImagePalette.raw("RGB;L", palette)
241
+ elif self.mode == "RGB":
242
+ if not greyscale or not linear:
243
+ self.lut = list(palette)
244
+
245
+ self.frame = 0
246
+
247
+ self.__offset = offs = self.fp.tell()
248
+
249
+ self._fp = self.fp # FIXME: hack
250
+
251
+ if self.rawmode.startswith("F;"):
252
+ # ifunc95 formats
253
+ try:
254
+ # use bit decoder (if necessary)
255
+ bits = int(self.rawmode[2:])
256
+ if bits not in [8, 16, 32]:
257
+ self.tile = [
258
+ ImageFile._Tile(
259
+ "bit", (0, 0) + self.size, offs, (bits, 8, 3, 0, -1)
260
+ )
261
+ ]
262
+ return
263
+ except ValueError:
264
+ pass
265
+
266
+ if self.rawmode in ["RGB;T", "RYB;T"]:
267
+ # Old LabEye/3PC files. Would be very surprised if anyone
268
+ # ever stumbled upon such a file ;-)
269
+ size = self.size[0] * self.size[1]
270
+ self.tile = [
271
+ ImageFile._Tile("raw", (0, 0) + self.size, offs, ("G", 0, -1)),
272
+ ImageFile._Tile("raw", (0, 0) + self.size, offs + size, ("R", 0, -1)),
273
+ ImageFile._Tile(
274
+ "raw", (0, 0) + self.size, offs + 2 * size, ("B", 0, -1)
275
+ ),
276
+ ]
277
+ else:
278
+ # LabEye/IFUNC files
279
+ self.tile = [
280
+ ImageFile._Tile("raw", (0, 0) + self.size, offs, (self.rawmode, 0, -1))
281
+ ]
282
+
283
+ @property
284
+ def n_frames(self) -> int:
285
+ return self.info[FRAMES]
286
+
287
+ @property
288
+ def is_animated(self) -> bool:
289
+ return self.info[FRAMES] > 1
290
+
291
+ def seek(self, frame: int) -> None:
292
+ if not self._seek_check(frame):
293
+ return
294
+ if isinstance(self._fp, DeferredError):
295
+ raise self._fp.ex
296
+
297
+ self.frame = frame
298
+
299
+ if self.mode == "1":
300
+ bits = 1
301
+ else:
302
+ bits = 8 * len(self.mode)
303
+
304
+ size = ((self.size[0] * bits + 7) // 8) * self.size[1]
305
+ offs = self.__offset + frame * size
306
+
307
+ self.fp = self._fp
308
+
309
+ self.tile = [
310
+ ImageFile._Tile("raw", (0, 0) + self.size, offs, (self.rawmode, 0, -1))
311
+ ]
312
+
313
+ def tell(self) -> int:
314
+ return self.frame
315
+
316
+
317
+ #
318
+ # --------------------------------------------------------------------
319
+ # Save IM files
320
+
321
+
322
+ SAVE = {
323
+ # mode: (im type, raw mode)
324
+ "1": ("0 1", "1"),
325
+ "L": ("Greyscale", "L"),
326
+ "LA": ("LA", "LA;L"),
327
+ "P": ("Greyscale", "P"),
328
+ "PA": ("LA", "PA;L"),
329
+ "I": ("L 32S", "I;32S"),
330
+ "I;16": ("L 16", "I;16"),
331
+ "I;16L": ("L 16L", "I;16L"),
332
+ "I;16B": ("L 16B", "I;16B"),
333
+ "F": ("L 32F", "F;32F"),
334
+ "RGB": ("RGB", "RGB;L"),
335
+ "RGBA": ("RGBA", "RGBA;L"),
336
+ "RGBX": ("RGBX", "RGBX;L"),
337
+ "CMYK": ("CMYK", "CMYK;L"),
338
+ "YCbCr": ("YCC", "YCbCr;L"),
339
+ }
340
+
341
+
342
+ def _save(im: Image.Image, fp: IO[bytes], filename: str | bytes) -> None:
343
+ try:
344
+ image_type, rawmode = SAVE[im.mode]
345
+ except KeyError as e:
346
+ msg = f"Cannot save {im.mode} images as IM"
347
+ raise ValueError(msg) from e
348
+
349
+ frames = im.encoderinfo.get("frames", 1)
350
+
351
+ fp.write(f"Image type: {image_type} image\r\n".encode("ascii"))
352
+ if filename:
353
+ # Each line must be 100 characters or less,
354
+ # or: SyntaxError("not an IM file")
355
+ # 8 characters are used for "Name: " and "\r\n"
356
+ # Keep just the filename, ditch the potentially overlong path
357
+ if isinstance(filename, bytes):
358
+ filename = filename.decode("ascii")
359
+ name, ext = os.path.splitext(os.path.basename(filename))
360
+ name = "".join([name[: 92 - len(ext)], ext])
361
+
362
+ fp.write(f"Name: {name}\r\n".encode("ascii"))
363
+ fp.write(f"Image size (x*y): {im.size[0]}*{im.size[1]}\r\n".encode("ascii"))
364
+ fp.write(f"File size (no of images): {frames}\r\n".encode("ascii"))
365
+ if im.mode in ["P", "PA"]:
366
+ fp.write(b"Lut: 1\r\n")
367
+ fp.write(b"\000" * (511 - fp.tell()) + b"\032")
368
+ if im.mode in ["P", "PA"]:
369
+ im_palette = im.im.getpalette("RGB", "RGB;L")
370
+ colors = len(im_palette) // 3
371
+ palette = b""
372
+ for i in range(3):
373
+ palette += im_palette[colors * i : colors * (i + 1)]
374
+ palette += b"\x00" * (256 - colors)
375
+ fp.write(palette) # 768 bytes
376
+ ImageFile._save(
377
+ im, fp, [ImageFile._Tile("raw", (0, 0) + im.size, 0, (rawmode, 0, -1))]
378
+ )
379
+
380
+
381
+ #
382
+ # --------------------------------------------------------------------
383
+ # Registry
384
+
385
+
386
+ Image.register_open(ImImageFile.format, ImImageFile)
387
+ Image.register_save(ImImageFile.format, _save)
388
+
389
+ Image.register_extension(ImImageFile.format, ".im")
.venv/Lib/site-packages/PIL/Image.py ADDED
The diff for this file is too large to render. See raw diff
 
.venv/Lib/site-packages/PIL/ImageChops.py ADDED
@@ -0,0 +1,311 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #
2
+ # The Python Imaging Library.
3
+ # $Id$
4
+ #
5
+ # standard channel operations
6
+ #
7
+ # History:
8
+ # 1996-03-24 fl Created
9
+ # 1996-08-13 fl Added logical operations (for "1" images)
10
+ # 2000-10-12 fl Added offset method (from Image.py)
11
+ #
12
+ # Copyright (c) 1997-2000 by Secret Labs AB
13
+ # Copyright (c) 1996-2000 by Fredrik Lundh
14
+ #
15
+ # See the README file for information on usage and redistribution.
16
+ #
17
+
18
+ from __future__ import annotations
19
+
20
+ from . import Image
21
+
22
+
23
+ def constant(image: Image.Image, value: int) -> Image.Image:
24
+ """Fill a channel with a given gray level.
25
+
26
+ :rtype: :py:class:`~PIL.Image.Image`
27
+ """
28
+
29
+ return Image.new("L", image.size, value)
30
+
31
+
32
+ def duplicate(image: Image.Image) -> Image.Image:
33
+ """Copy a channel. Alias for :py:meth:`PIL.Image.Image.copy`.
34
+
35
+ :rtype: :py:class:`~PIL.Image.Image`
36
+ """
37
+
38
+ return image.copy()
39
+
40
+
41
+ def invert(image: Image.Image) -> Image.Image:
42
+ """
43
+ Invert an image (channel). ::
44
+
45
+ out = MAX - image
46
+
47
+ :rtype: :py:class:`~PIL.Image.Image`
48
+ """
49
+
50
+ image.load()
51
+ return image._new(image.im.chop_invert())
52
+
53
+
54
+ def lighter(image1: Image.Image, image2: Image.Image) -> Image.Image:
55
+ """
56
+ Compares the two images, pixel by pixel, and returns a new image containing
57
+ the lighter values. ::
58
+
59
+ out = max(image1, image2)
60
+
61
+ :rtype: :py:class:`~PIL.Image.Image`
62
+ """
63
+
64
+ image1.load()
65
+ image2.load()
66
+ return image1._new(image1.im.chop_lighter(image2.im))
67
+
68
+
69
+ def darker(image1: Image.Image, image2: Image.Image) -> Image.Image:
70
+ """
71
+ Compares the two images, pixel by pixel, and returns a new image containing
72
+ the darker values. ::
73
+
74
+ out = min(image1, image2)
75
+
76
+ :rtype: :py:class:`~PIL.Image.Image`
77
+ """
78
+
79
+ image1.load()
80
+ image2.load()
81
+ return image1._new(image1.im.chop_darker(image2.im))
82
+
83
+
84
+ def difference(image1: Image.Image, image2: Image.Image) -> Image.Image:
85
+ """
86
+ Returns the absolute value of the pixel-by-pixel difference between the two
87
+ images. ::
88
+
89
+ out = abs(image1 - image2)
90
+
91
+ :rtype: :py:class:`~PIL.Image.Image`
92
+ """
93
+
94
+ image1.load()
95
+ image2.load()
96
+ return image1._new(image1.im.chop_difference(image2.im))
97
+
98
+
99
+ def multiply(image1: Image.Image, image2: Image.Image) -> Image.Image:
100
+ """
101
+ Superimposes two images on top of each other.
102
+
103
+ If you multiply an image with a solid black image, the result is black. If
104
+ you multiply with a solid white image, the image is unaffected. ::
105
+
106
+ out = image1 * image2 / MAX
107
+
108
+ :rtype: :py:class:`~PIL.Image.Image`
109
+ """
110
+
111
+ image1.load()
112
+ image2.load()
113
+ return image1._new(image1.im.chop_multiply(image2.im))
114
+
115
+
116
+ def screen(image1: Image.Image, image2: Image.Image) -> Image.Image:
117
+ """
118
+ Superimposes two inverted images on top of each other. ::
119
+
120
+ out = MAX - ((MAX - image1) * (MAX - image2) / MAX)
121
+
122
+ :rtype: :py:class:`~PIL.Image.Image`
123
+ """
124
+
125
+ image1.load()
126
+ image2.load()
127
+ return image1._new(image1.im.chop_screen(image2.im))
128
+
129
+
130
+ def soft_light(image1: Image.Image, image2: Image.Image) -> Image.Image:
131
+ """
132
+ Superimposes two images on top of each other using the Soft Light algorithm
133
+
134
+ :rtype: :py:class:`~PIL.Image.Image`
135
+ """
136
+
137
+ image1.load()
138
+ image2.load()
139
+ return image1._new(image1.im.chop_soft_light(image2.im))
140
+
141
+
142
+ def hard_light(image1: Image.Image, image2: Image.Image) -> Image.Image:
143
+ """
144
+ Superimposes two images on top of each other using the Hard Light algorithm
145
+
146
+ :rtype: :py:class:`~PIL.Image.Image`
147
+ """
148
+
149
+ image1.load()
150
+ image2.load()
151
+ return image1._new(image1.im.chop_hard_light(image2.im))
152
+
153
+
154
+ def overlay(image1: Image.Image, image2: Image.Image) -> Image.Image:
155
+ """
156
+ Superimposes two images on top of each other using the Overlay algorithm
157
+
158
+ :rtype: :py:class:`~PIL.Image.Image`
159
+ """
160
+
161
+ image1.load()
162
+ image2.load()
163
+ return image1._new(image1.im.chop_overlay(image2.im))
164
+
165
+
166
+ def add(
167
+ image1: Image.Image, image2: Image.Image, scale: float = 1.0, offset: float = 0
168
+ ) -> Image.Image:
169
+ """
170
+ Adds two images, dividing the result by scale and adding the
171
+ offset. If omitted, scale defaults to 1.0, and offset to 0.0. ::
172
+
173
+ out = ((image1 + image2) / scale + offset)
174
+
175
+ :rtype: :py:class:`~PIL.Image.Image`
176
+ """
177
+
178
+ image1.load()
179
+ image2.load()
180
+ return image1._new(image1.im.chop_add(image2.im, scale, offset))
181
+
182
+
183
+ def subtract(
184
+ image1: Image.Image, image2: Image.Image, scale: float = 1.0, offset: float = 0
185
+ ) -> Image.Image:
186
+ """
187
+ Subtracts two images, dividing the result by scale and adding the offset.
188
+ If omitted, scale defaults to 1.0, and offset to 0.0. ::
189
+
190
+ out = ((image1 - image2) / scale + offset)
191
+
192
+ :rtype: :py:class:`~PIL.Image.Image`
193
+ """
194
+
195
+ image1.load()
196
+ image2.load()
197
+ return image1._new(image1.im.chop_subtract(image2.im, scale, offset))
198
+
199
+
200
+ def add_modulo(image1: Image.Image, image2: Image.Image) -> Image.Image:
201
+ """Add two images, without clipping the result. ::
202
+
203
+ out = ((image1 + image2) % MAX)
204
+
205
+ :rtype: :py:class:`~PIL.Image.Image`
206
+ """
207
+
208
+ image1.load()
209
+ image2.load()
210
+ return image1._new(image1.im.chop_add_modulo(image2.im))
211
+
212
+
213
+ def subtract_modulo(image1: Image.Image, image2: Image.Image) -> Image.Image:
214
+ """Subtract two images, without clipping the result. ::
215
+
216
+ out = ((image1 - image2) % MAX)
217
+
218
+ :rtype: :py:class:`~PIL.Image.Image`
219
+ """
220
+
221
+ image1.load()
222
+ image2.load()
223
+ return image1._new(image1.im.chop_subtract_modulo(image2.im))
224
+
225
+
226
+ def logical_and(image1: Image.Image, image2: Image.Image) -> Image.Image:
227
+ """Logical AND between two images.
228
+
229
+ Both of the images must have mode "1". If you would like to perform a
230
+ logical AND on an image with a mode other than "1", try
231
+ :py:meth:`~PIL.ImageChops.multiply` instead, using a black-and-white mask
232
+ as the second image. ::
233
+
234
+ out = ((image1 and image2) % MAX)
235
+
236
+ :rtype: :py:class:`~PIL.Image.Image`
237
+ """
238
+
239
+ image1.load()
240
+ image2.load()
241
+ return image1._new(image1.im.chop_and(image2.im))
242
+
243
+
244
+ def logical_or(image1: Image.Image, image2: Image.Image) -> Image.Image:
245
+ """Logical OR between two images.
246
+
247
+ Both of the images must have mode "1". ::
248
+
249
+ out = ((image1 or image2) % MAX)
250
+
251
+ :rtype: :py:class:`~PIL.Image.Image`
252
+ """
253
+
254
+ image1.load()
255
+ image2.load()
256
+ return image1._new(image1.im.chop_or(image2.im))
257
+
258
+
259
+ def logical_xor(image1: Image.Image, image2: Image.Image) -> Image.Image:
260
+ """Logical XOR between two images.
261
+
262
+ Both of the images must have mode "1". ::
263
+
264
+ out = ((bool(image1) != bool(image2)) % MAX)
265
+
266
+ :rtype: :py:class:`~PIL.Image.Image`
267
+ """
268
+
269
+ image1.load()
270
+ image2.load()
271
+ return image1._new(image1.im.chop_xor(image2.im))
272
+
273
+
274
+ def blend(image1: Image.Image, image2: Image.Image, alpha: float) -> Image.Image:
275
+ """Blend images using constant transparency weight. Alias for
276
+ :py:func:`PIL.Image.blend`.
277
+
278
+ :rtype: :py:class:`~PIL.Image.Image`
279
+ """
280
+
281
+ return Image.blend(image1, image2, alpha)
282
+
283
+
284
+ def composite(
285
+ image1: Image.Image, image2: Image.Image, mask: Image.Image
286
+ ) -> Image.Image:
287
+ """Create composite using transparency mask. Alias for
288
+ :py:func:`PIL.Image.composite`.
289
+
290
+ :rtype: :py:class:`~PIL.Image.Image`
291
+ """
292
+
293
+ return Image.composite(image1, image2, mask)
294
+
295
+
296
+ def offset(image: Image.Image, xoffset: int, yoffset: int | None = None) -> Image.Image:
297
+ """Returns a copy of the image where data has been offset by the given
298
+ distances. Data wraps around the edges. If ``yoffset`` is omitted, it
299
+ is assumed to be equal to ``xoffset``.
300
+
301
+ :param image: Input image.
302
+ :param xoffset: The horizontal distance.
303
+ :param yoffset: The vertical distance. If omitted, both
304
+ distances are set to the same value.
305
+ :rtype: :py:class:`~PIL.Image.Image`
306
+ """
307
+
308
+ if yoffset is None:
309
+ yoffset = xoffset
310
+ image.load()
311
+ return image._new(image.im.offset(xoffset, yoffset))
.venv/Lib/site-packages/PIL/ImageCms.py ADDED
@@ -0,0 +1,1076 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # The Python Imaging Library.
2
+ # $Id$
3
+
4
+ # Optional color management support, based on Kevin Cazabon's PyCMS
5
+ # library.
6
+
7
+ # Originally released under LGPL. Graciously donated to PIL in
8
+ # March 2009, for distribution under the standard PIL license
9
+
10
+ # History:
11
+
12
+ # 2009-03-08 fl Added to PIL.
13
+
14
+ # Copyright (C) 2002-2003 Kevin Cazabon
15
+ # Copyright (c) 2009 by Fredrik Lundh
16
+ # Copyright (c) 2013 by Eric Soroos
17
+
18
+ # See the README file for information on usage and redistribution. See
19
+ # below for the original description.
20
+ from __future__ import annotations
21
+
22
+ import operator
23
+ import sys
24
+ from enum import IntEnum, IntFlag
25
+ from functools import reduce
26
+ from typing import Any, Literal, SupportsFloat, SupportsInt, Union
27
+
28
+ from . import Image
29
+ from ._deprecate import deprecate
30
+ from ._typing import SupportsRead
31
+
32
+ try:
33
+ from . import _imagingcms as core
34
+
35
+ _CmsProfileCompatible = Union[
36
+ str, SupportsRead[bytes], core.CmsProfile, "ImageCmsProfile"
37
+ ]
38
+ except ImportError as ex:
39
+ # Allow error import for doc purposes, but error out when accessing
40
+ # anything in core.
41
+ from ._util import DeferredError
42
+
43
+ core = DeferredError.new(ex)
44
+
45
+ _DESCRIPTION = """
46
+ pyCMS
47
+
48
+ a Python / PIL interface to the littleCMS ICC Color Management System
49
+ Copyright (C) 2002-2003 Kevin Cazabon
50
+ kevin@cazabon.com
51
+ https://www.cazabon.com
52
+
53
+ pyCMS home page: https://www.cazabon.com/pyCMS
54
+ littleCMS home page: https://www.littlecms.com
55
+ (littleCMS is Copyright (C) 1998-2001 Marti Maria)
56
+
57
+ Originally released under LGPL. Graciously donated to PIL in
58
+ March 2009, for distribution under the standard PIL license
59
+
60
+ The pyCMS.py module provides a "clean" interface between Python/PIL and
61
+ pyCMSdll, taking care of some of the more complex handling of the direct
62
+ pyCMSdll functions, as well as error-checking and making sure that all
63
+ relevant data is kept together.
64
+
65
+ While it is possible to call pyCMSdll functions directly, it's not highly
66
+ recommended.
67
+
68
+ Version History:
69
+
70
+ 1.0.0 pil Oct 2013 Port to LCMS 2.
71
+
72
+ 0.1.0 pil mod March 10, 2009
73
+
74
+ Renamed display profile to proof profile. The proof
75
+ profile is the profile of the device that is being
76
+ simulated, not the profile of the device which is
77
+ actually used to display/print the final simulation
78
+ (that'd be the output profile) - also see LCMSAPI.txt
79
+ input colorspace -> using 'renderingIntent' -> proof
80
+ colorspace -> using 'proofRenderingIntent' -> output
81
+ colorspace
82
+
83
+ Added LCMS FLAGS support.
84
+ Added FLAGS["SOFTPROOFING"] as default flag for
85
+ buildProofTransform (otherwise the proof profile/intent
86
+ would be ignored).
87
+
88
+ 0.1.0 pil March 2009 - added to PIL, as PIL.ImageCms
89
+
90
+ 0.0.2 alpha Jan 6, 2002
91
+
92
+ Added try/except statements around type() checks of
93
+ potential CObjects... Python won't let you use type()
94
+ on them, and raises a TypeError (stupid, if you ask
95
+ me!)
96
+
97
+ Added buildProofTransformFromOpenProfiles() function.
98
+ Additional fixes in DLL, see DLL code for details.
99
+
100
+ 0.0.1 alpha first public release, Dec. 26, 2002
101
+
102
+ Known to-do list with current version (of Python interface, not pyCMSdll):
103
+
104
+ none
105
+
106
+ """
107
+
108
+ _VERSION = "1.0.0 pil"
109
+
110
+
111
+ # --------------------------------------------------------------------.
112
+
113
+
114
+ #
115
+ # intent/direction values
116
+
117
+
118
+ class Intent(IntEnum):
119
+ PERCEPTUAL = 0
120
+ RELATIVE_COLORIMETRIC = 1
121
+ SATURATION = 2
122
+ ABSOLUTE_COLORIMETRIC = 3
123
+
124
+
125
+ class Direction(IntEnum):
126
+ INPUT = 0
127
+ OUTPUT = 1
128
+ PROOF = 2
129
+
130
+
131
+ #
132
+ # flags
133
+
134
+
135
+ class Flags(IntFlag):
136
+ """Flags and documentation are taken from ``lcms2.h``."""
137
+
138
+ NONE = 0
139
+ NOCACHE = 0x0040
140
+ """Inhibit 1-pixel cache"""
141
+ NOOPTIMIZE = 0x0100
142
+ """Inhibit optimizations"""
143
+ NULLTRANSFORM = 0x0200
144
+ """Don't transform anyway"""
145
+ GAMUTCHECK = 0x1000
146
+ """Out of Gamut alarm"""
147
+ SOFTPROOFING = 0x4000
148
+ """Do softproofing"""
149
+ BLACKPOINTCOMPENSATION = 0x2000
150
+ NOWHITEONWHITEFIXUP = 0x0004
151
+ """Don't fix scum dot"""
152
+ HIGHRESPRECALC = 0x0400
153
+ """Use more memory to give better accuracy"""
154
+ LOWRESPRECALC = 0x0800
155
+ """Use less memory to minimize resources"""
156
+ # this should be 8BITS_DEVICELINK, but that is not a valid name in Python:
157
+ USE_8BITS_DEVICELINK = 0x0008
158
+ """Create 8 bits devicelinks"""
159
+ GUESSDEVICECLASS = 0x0020
160
+ """Guess device class (for ``transform2devicelink``)"""
161
+ KEEP_SEQUENCE = 0x0080
162
+ """Keep profile sequence for devicelink creation"""
163
+ FORCE_CLUT = 0x0002
164
+ """Force CLUT optimization"""
165
+ CLUT_POST_LINEARIZATION = 0x0001
166
+ """create postlinearization tables if possible"""
167
+ CLUT_PRE_LINEARIZATION = 0x0010
168
+ """create prelinearization tables if possible"""
169
+ NONEGATIVES = 0x8000
170
+ """Prevent negative numbers in floating point transforms"""
171
+ COPY_ALPHA = 0x04000000
172
+ """Alpha channels are copied on ``cmsDoTransform()``"""
173
+ NODEFAULTRESOURCEDEF = 0x01000000
174
+
175
+ _GRIDPOINTS_1 = 1 << 16
176
+ _GRIDPOINTS_2 = 2 << 16
177
+ _GRIDPOINTS_4 = 4 << 16
178
+ _GRIDPOINTS_8 = 8 << 16
179
+ _GRIDPOINTS_16 = 16 << 16
180
+ _GRIDPOINTS_32 = 32 << 16
181
+ _GRIDPOINTS_64 = 64 << 16
182
+ _GRIDPOINTS_128 = 128 << 16
183
+
184
+ @staticmethod
185
+ def GRIDPOINTS(n: int) -> Flags:
186
+ """
187
+ Fine-tune control over number of gridpoints
188
+
189
+ :param n: :py:class:`int` in range ``0 <= n <= 255``
190
+ """
191
+ return Flags.NONE | ((n & 0xFF) << 16)
192
+
193
+
194
+ _MAX_FLAG = reduce(operator.or_, Flags)
195
+
196
+
197
+ _FLAGS = {
198
+ "MATRIXINPUT": 1,
199
+ "MATRIXOUTPUT": 2,
200
+ "MATRIXONLY": (1 | 2),
201
+ "NOWHITEONWHITEFIXUP": 4, # Don't hot fix scum dot
202
+ # Don't create prelinearization tables on precalculated transforms
203
+ # (internal use):
204
+ "NOPRELINEARIZATION": 16,
205
+ "GUESSDEVICECLASS": 32, # Guess device class (for transform2devicelink)
206
+ "NOTCACHE": 64, # Inhibit 1-pixel cache
207
+ "NOTPRECALC": 256,
208
+ "NULLTRANSFORM": 512, # Don't transform anyway
209
+ "HIGHRESPRECALC": 1024, # Use more memory to give better accuracy
210
+ "LOWRESPRECALC": 2048, # Use less memory to minimize resources
211
+ "WHITEBLACKCOMPENSATION": 8192,
212
+ "BLACKPOINTCOMPENSATION": 8192,
213
+ "GAMUTCHECK": 4096, # Out of Gamut alarm
214
+ "SOFTPROOFING": 16384, # Do softproofing
215
+ "PRESERVEBLACK": 32768, # Black preservation
216
+ "NODEFAULTRESOURCEDEF": 16777216, # CRD special
217
+ "GRIDPOINTS": lambda n: (n & 0xFF) << 16, # Gridpoints
218
+ }
219
+
220
+
221
+ # --------------------------------------------------------------------.
222
+ # Experimental PIL-level API
223
+ # --------------------------------------------------------------------.
224
+
225
+ ##
226
+ # Profile.
227
+
228
+
229
+ class ImageCmsProfile:
230
+ def __init__(self, profile: str | SupportsRead[bytes] | core.CmsProfile) -> None:
231
+ """
232
+ :param profile: Either a string representing a filename,
233
+ a file like object containing a profile or a
234
+ low-level profile object
235
+
236
+ """
237
+ self.filename: str | None = None
238
+
239
+ if isinstance(profile, str):
240
+ if sys.platform == "win32":
241
+ profile_bytes_path = profile.encode()
242
+ try:
243
+ profile_bytes_path.decode("ascii")
244
+ except UnicodeDecodeError:
245
+ with open(profile, "rb") as f:
246
+ self.profile = core.profile_frombytes(f.read())
247
+ return
248
+ self.filename = profile
249
+ self.profile = core.profile_open(profile)
250
+ elif hasattr(profile, "read"):
251
+ self.profile = core.profile_frombytes(profile.read())
252
+ elif isinstance(profile, core.CmsProfile):
253
+ self.profile = profile
254
+ else:
255
+ msg = "Invalid type for Profile" # type: ignore[unreachable]
256
+ raise TypeError(msg)
257
+
258
+ def __getattr__(self, name: str) -> Any:
259
+ if name in ("product_name", "product_info"):
260
+ deprecate(f"ImageCms.ImageCmsProfile.{name}", 13)
261
+ return None
262
+ msg = f"'{self.__class__.__name__}' object has no attribute '{name}'"
263
+ raise AttributeError(msg)
264
+
265
+ def tobytes(self) -> bytes:
266
+ """
267
+ Returns the profile in a format suitable for embedding in
268
+ saved images.
269
+
270
+ :returns: a bytes object containing the ICC profile.
271
+ """
272
+
273
+ return core.profile_tobytes(self.profile)
274
+
275
+
276
+ class ImageCmsTransform(Image.ImagePointHandler):
277
+ """
278
+ Transform. This can be used with the procedural API, or with the standard
279
+ :py:func:`~PIL.Image.Image.point` method.
280
+
281
+ Will return the output profile in the ``output.info['icc_profile']``.
282
+ """
283
+
284
+ def __init__(
285
+ self,
286
+ input: ImageCmsProfile,
287
+ output: ImageCmsProfile,
288
+ input_mode: str,
289
+ output_mode: str,
290
+ intent: Intent = Intent.PERCEPTUAL,
291
+ proof: ImageCmsProfile | None = None,
292
+ proof_intent: Intent = Intent.ABSOLUTE_COLORIMETRIC,
293
+ flags: Flags = Flags.NONE,
294
+ ):
295
+ if proof is None:
296
+ self.transform = core.buildTransform(
297
+ input.profile, output.profile, input_mode, output_mode, intent, flags
298
+ )
299
+ else:
300
+ self.transform = core.buildProofTransform(
301
+ input.profile,
302
+ output.profile,
303
+ proof.profile,
304
+ input_mode,
305
+ output_mode,
306
+ intent,
307
+ proof_intent,
308
+ flags,
309
+ )
310
+ # Note: inputMode and outputMode are for pyCMS compatibility only
311
+ self.input_mode = self.inputMode = input_mode
312
+ self.output_mode = self.outputMode = output_mode
313
+
314
+ self.output_profile = output
315
+
316
+ def point(self, im: Image.Image) -> Image.Image:
317
+ return self.apply(im)
318
+
319
+ def apply(self, im: Image.Image, imOut: Image.Image | None = None) -> Image.Image:
320
+ if imOut is None:
321
+ imOut = Image.new(self.output_mode, im.size, None)
322
+ self.transform.apply(im.getim(), imOut.getim())
323
+ imOut.info["icc_profile"] = self.output_profile.tobytes()
324
+ return imOut
325
+
326
+ def apply_in_place(self, im: Image.Image) -> Image.Image:
327
+ if im.mode != self.output_mode:
328
+ msg = "mode mismatch"
329
+ raise ValueError(msg) # wrong output mode
330
+ self.transform.apply(im.getim(), im.getim())
331
+ im.info["icc_profile"] = self.output_profile.tobytes()
332
+ return im
333
+
334
+
335
+ def get_display_profile(handle: SupportsInt | None = None) -> ImageCmsProfile | None:
336
+ """
337
+ (experimental) Fetches the profile for the current display device.
338
+
339
+ :returns: ``None`` if the profile is not known.
340
+ """
341
+
342
+ if sys.platform != "win32":
343
+ return None
344
+
345
+ from . import ImageWin # type: ignore[unused-ignore, unreachable]
346
+
347
+ if isinstance(handle, ImageWin.HDC):
348
+ profile = core.get_display_profile_win32(int(handle), 1)
349
+ else:
350
+ profile = core.get_display_profile_win32(int(handle or 0))
351
+ if profile is None:
352
+ return None
353
+ return ImageCmsProfile(profile)
354
+
355
+
356
+ # --------------------------------------------------------------------.
357
+ # pyCMS compatible layer
358
+ # --------------------------------------------------------------------.
359
+
360
+
361
+ class PyCMSError(Exception):
362
+ """(pyCMS) Exception class.
363
+ This is used for all errors in the pyCMS API."""
364
+
365
+ pass
366
+
367
+
368
+ def profileToProfile(
369
+ im: Image.Image,
370
+ inputProfile: _CmsProfileCompatible,
371
+ outputProfile: _CmsProfileCompatible,
372
+ renderingIntent: Intent = Intent.PERCEPTUAL,
373
+ outputMode: str | None = None,
374
+ inPlace: bool = False,
375
+ flags: Flags = Flags.NONE,
376
+ ) -> Image.Image | None:
377
+ """
378
+ (pyCMS) Applies an ICC transformation to a given image, mapping from
379
+ ``inputProfile`` to ``outputProfile``.
380
+
381
+ If the input or output profiles specified are not valid filenames, a
382
+ :exc:`PyCMSError` will be raised. If ``inPlace`` is ``True`` and
383
+ ``outputMode != im.mode``, a :exc:`PyCMSError` will be raised.
384
+ If an error occurs during application of the profiles,
385
+ a :exc:`PyCMSError` will be raised.
386
+ If ``outputMode`` is not a mode supported by the ``outputProfile`` (or by pyCMS),
387
+ a :exc:`PyCMSError` will be raised.
388
+
389
+ This function applies an ICC transformation to im from ``inputProfile``'s
390
+ color space to ``outputProfile``'s color space using the specified rendering
391
+ intent to decide how to handle out-of-gamut colors.
392
+
393
+ ``outputMode`` can be used to specify that a color mode conversion is to
394
+ be done using these profiles, but the specified profiles must be able
395
+ to handle that mode. I.e., if converting im from RGB to CMYK using
396
+ profiles, the input profile must handle RGB data, and the output
397
+ profile must handle CMYK data.
398
+
399
+ :param im: An open :py:class:`~PIL.Image.Image` object (i.e. Image.new(...)
400
+ or Image.open(...), etc.)
401
+ :param inputProfile: String, as a valid filename path to the ICC input
402
+ profile you wish to use for this image, or a profile object
403
+ :param outputProfile: String, as a valid filename path to the ICC output
404
+ profile you wish to use for this image, or a profile object
405
+ :param renderingIntent: Integer (0-3) specifying the rendering intent you
406
+ wish to use for the transform
407
+
408
+ ImageCms.Intent.PERCEPTUAL = 0 (DEFAULT)
409
+ ImageCms.Intent.RELATIVE_COLORIMETRIC = 1
410
+ ImageCms.Intent.SATURATION = 2
411
+ ImageCms.Intent.ABSOLUTE_COLORIMETRIC = 3
412
+
413
+ see the pyCMS documentation for details on rendering intents and what
414
+ they do.
415
+ :param outputMode: A valid PIL mode for the output image (i.e. "RGB",
416
+ "CMYK", etc.). Note: if rendering the image "inPlace", outputMode
417
+ MUST be the same mode as the input, or omitted completely. If
418
+ omitted, the outputMode will be the same as the mode of the input
419
+ image (im.mode)
420
+ :param inPlace: Boolean. If ``True``, the original image is modified in-place,
421
+ and ``None`` is returned. If ``False`` (default), a new
422
+ :py:class:`~PIL.Image.Image` object is returned with the transform applied.
423
+ :param flags: Integer (0-...) specifying additional flags
424
+ :returns: Either None or a new :py:class:`~PIL.Image.Image` object, depending on
425
+ the value of ``inPlace``
426
+ :exception PyCMSError:
427
+ """
428
+
429
+ if outputMode is None:
430
+ outputMode = im.mode
431
+
432
+ if not isinstance(renderingIntent, int) or not (0 <= renderingIntent <= 3):
433
+ msg = "renderingIntent must be an integer between 0 and 3"
434
+ raise PyCMSError(msg)
435
+
436
+ if not isinstance(flags, int) or not (0 <= flags <= _MAX_FLAG):
437
+ msg = f"flags must be an integer between 0 and {_MAX_FLAG}"
438
+ raise PyCMSError(msg)
439
+
440
+ try:
441
+ if not isinstance(inputProfile, ImageCmsProfile):
442
+ inputProfile = ImageCmsProfile(inputProfile)
443
+ if not isinstance(outputProfile, ImageCmsProfile):
444
+ outputProfile = ImageCmsProfile(outputProfile)
445
+ transform = ImageCmsTransform(
446
+ inputProfile,
447
+ outputProfile,
448
+ im.mode,
449
+ outputMode,
450
+ renderingIntent,
451
+ flags=flags,
452
+ )
453
+ if inPlace:
454
+ transform.apply_in_place(im)
455
+ imOut = None
456
+ else:
457
+ imOut = transform.apply(im)
458
+ except (OSError, TypeError, ValueError) as v:
459
+ raise PyCMSError(v) from v
460
+
461
+ return imOut
462
+
463
+
464
+ def getOpenProfile(
465
+ profileFilename: str | SupportsRead[bytes] | core.CmsProfile,
466
+ ) -> ImageCmsProfile:
467
+ """
468
+ (pyCMS) Opens an ICC profile file.
469
+
470
+ The PyCMSProfile object can be passed back into pyCMS for use in creating
471
+ transforms and such (as in ImageCms.buildTransformFromOpenProfiles()).
472
+
473
+ If ``profileFilename`` is not a valid filename for an ICC profile,
474
+ a :exc:`PyCMSError` will be raised.
475
+
476
+ :param profileFilename: String, as a valid filename path to the ICC profile
477
+ you wish to open, or a file-like object.
478
+ :returns: A CmsProfile class object.
479
+ :exception PyCMSError:
480
+ """
481
+
482
+ try:
483
+ return ImageCmsProfile(profileFilename)
484
+ except (OSError, TypeError, ValueError) as v:
485
+ raise PyCMSError(v) from v
486
+
487
+
488
+ def buildTransform(
489
+ inputProfile: _CmsProfileCompatible,
490
+ outputProfile: _CmsProfileCompatible,
491
+ inMode: str,
492
+ outMode: str,
493
+ renderingIntent: Intent = Intent.PERCEPTUAL,
494
+ flags: Flags = Flags.NONE,
495
+ ) -> ImageCmsTransform:
496
+ """
497
+ (pyCMS) Builds an ICC transform mapping from the ``inputProfile`` to the
498
+ ``outputProfile``. Use applyTransform to apply the transform to a given
499
+ image.
500
+
501
+ If the input or output profiles specified are not valid filenames, a
502
+ :exc:`PyCMSError` will be raised. If an error occurs during creation
503
+ of the transform, a :exc:`PyCMSError` will be raised.
504
+
505
+ If ``inMode`` or ``outMode`` are not a mode supported by the ``outputProfile``
506
+ (or by pyCMS), a :exc:`PyCMSError` will be raised.
507
+
508
+ This function builds and returns an ICC transform from the ``inputProfile``
509
+ to the ``outputProfile`` using the ``renderingIntent`` to determine what to do
510
+ with out-of-gamut colors. It will ONLY work for converting images that
511
+ are in ``inMode`` to images that are in ``outMode`` color format (PIL mode,
512
+ i.e. "RGB", "RGBA", "CMYK", etc.).
513
+
514
+ Building the transform is a fair part of the overhead in
515
+ ImageCms.profileToProfile(), so if you're planning on converting multiple
516
+ images using the same input/output settings, this can save you time.
517
+ Once you have a transform object, it can be used with
518
+ ImageCms.applyProfile() to convert images without the need to re-compute
519
+ the lookup table for the transform.
520
+
521
+ The reason pyCMS returns a class object rather than a handle directly
522
+ to the transform is that it needs to keep track of the PIL input/output
523
+ modes that the transform is meant for. These attributes are stored in
524
+ the ``inMode`` and ``outMode`` attributes of the object (which can be
525
+ manually overridden if you really want to, but I don't know of any
526
+ time that would be of use, or would even work).
527
+
528
+ :param inputProfile: String, as a valid filename path to the ICC input
529
+ profile you wish to use for this transform, or a profile object
530
+ :param outputProfile: String, as a valid filename path to the ICC output
531
+ profile you wish to use for this transform, or a profile object
532
+ :param inMode: String, as a valid PIL mode that the appropriate profile
533
+ also supports (i.e. "RGB", "RGBA", "CMYK", etc.)
534
+ :param outMode: String, as a valid PIL mode that the appropriate profile
535
+ also supports (i.e. "RGB", "RGBA", "CMYK", etc.)
536
+ :param renderingIntent: Integer (0-3) specifying the rendering intent you
537
+ wish to use for the transform
538
+
539
+ ImageCms.Intent.PERCEPTUAL = 0 (DEFAULT)
540
+ ImageCms.Intent.RELATIVE_COLORIMETRIC = 1
541
+ ImageCms.Intent.SATURATION = 2
542
+ ImageCms.Intent.ABSOLUTE_COLORIMETRIC = 3
543
+
544
+ see the pyCMS documentation for details on rendering intents and what
545
+ they do.
546
+ :param flags: Integer (0-...) specifying additional flags
547
+ :returns: A CmsTransform class object.
548
+ :exception PyCMSError:
549
+ """
550
+
551
+ if not isinstance(renderingIntent, int) or not (0 <= renderingIntent <= 3):
552
+ msg = "renderingIntent must be an integer between 0 and 3"
553
+ raise PyCMSError(msg)
554
+
555
+ if not isinstance(flags, int) or not (0 <= flags <= _MAX_FLAG):
556
+ msg = f"flags must be an integer between 0 and {_MAX_FLAG}"
557
+ raise PyCMSError(msg)
558
+
559
+ try:
560
+ if not isinstance(inputProfile, ImageCmsProfile):
561
+ inputProfile = ImageCmsProfile(inputProfile)
562
+ if not isinstance(outputProfile, ImageCmsProfile):
563
+ outputProfile = ImageCmsProfile(outputProfile)
564
+ return ImageCmsTransform(
565
+ inputProfile, outputProfile, inMode, outMode, renderingIntent, flags=flags
566
+ )
567
+ except (OSError, TypeError, ValueError) as v:
568
+ raise PyCMSError(v) from v
569
+
570
+
571
+ def buildProofTransform(
572
+ inputProfile: _CmsProfileCompatible,
573
+ outputProfile: _CmsProfileCompatible,
574
+ proofProfile: _CmsProfileCompatible,
575
+ inMode: str,
576
+ outMode: str,
577
+ renderingIntent: Intent = Intent.PERCEPTUAL,
578
+ proofRenderingIntent: Intent = Intent.ABSOLUTE_COLORIMETRIC,
579
+ flags: Flags = Flags.SOFTPROOFING,
580
+ ) -> ImageCmsTransform:
581
+ """
582
+ (pyCMS) Builds an ICC transform mapping from the ``inputProfile`` to the
583
+ ``outputProfile``, but tries to simulate the result that would be
584
+ obtained on the ``proofProfile`` device.
585
+
586
+ If the input, output, or proof profiles specified are not valid
587
+ filenames, a :exc:`PyCMSError` will be raised.
588
+
589
+ If an error occurs during creation of the transform,
590
+ a :exc:`PyCMSError` will be raised.
591
+
592
+ If ``inMode`` or ``outMode`` are not a mode supported by the ``outputProfile``
593
+ (or by pyCMS), a :exc:`PyCMSError` will be raised.
594
+
595
+ This function builds and returns an ICC transform from the ``inputProfile``
596
+ to the ``outputProfile``, but tries to simulate the result that would be
597
+ obtained on the ``proofProfile`` device using ``renderingIntent`` and
598
+ ``proofRenderingIntent`` to determine what to do with out-of-gamut
599
+ colors. This is known as "soft-proofing". It will ONLY work for
600
+ converting images that are in ``inMode`` to images that are in outMode
601
+ color format (PIL mode, i.e. "RGB", "RGBA", "CMYK", etc.).
602
+
603
+ Usage of the resulting transform object is exactly the same as with
604
+ ImageCms.buildTransform().
605
+
606
+ Proof profiling is generally used when using an output device to get a
607
+ good idea of what the final printed/displayed image would look like on
608
+ the ``proofProfile`` device when it's quicker and easier to use the
609
+ output device for judging color. Generally, this means that the
610
+ output device is a monitor, or a dye-sub printer (etc.), and the simulated
611
+ device is something more expensive, complicated, or time consuming
612
+ (making it difficult to make a real print for color judgement purposes).
613
+
614
+ Soft-proofing basically functions by adjusting the colors on the
615
+ output device to match the colors of the device being simulated. However,
616
+ when the simulated device has a much wider gamut than the output
617
+ device, you may obtain marginal results.
618
+
619
+ :param inputProfile: String, as a valid filename path to the ICC input
620
+ profile you wish to use for this transform, or a profile object
621
+ :param outputProfile: String, as a valid filename path to the ICC output
622
+ (monitor, usually) profile you wish to use for this transform, or a
623
+ profile object
624
+ :param proofProfile: String, as a valid filename path to the ICC proof
625
+ profile you wish to use for this transform, or a profile object
626
+ :param inMode: String, as a valid PIL mode that the appropriate profile
627
+ also supports (i.e. "RGB", "RGBA", "CMYK", etc.)
628
+ :param outMode: String, as a valid PIL mode that the appropriate profile
629
+ also supports (i.e. "RGB", "RGBA", "CMYK", etc.)
630
+ :param renderingIntent: Integer (0-3) specifying the rendering intent you
631
+ wish to use for the input->proof (simulated) transform
632
+
633
+ ImageCms.Intent.PERCEPTUAL = 0 (DEFAULT)
634
+ ImageCms.Intent.RELATIVE_COLORIMETRIC = 1
635
+ ImageCms.Intent.SATURATION = 2
636
+ ImageCms.Intent.ABSOLUTE_COLORIMETRIC = 3
637
+
638
+ see the pyCMS documentation for details on rendering intents and what
639
+ they do.
640
+ :param proofRenderingIntent: Integer (0-3) specifying the rendering intent
641
+ you wish to use for proof->output transform
642
+
643
+ ImageCms.Intent.PERCEPTUAL = 0 (DEFAULT)
644
+ ImageCms.Intent.RELATIVE_COLORIMETRIC = 1
645
+ ImageCms.Intent.SATURATION = 2
646
+ ImageCms.Intent.ABSOLUTE_COLORIMETRIC = 3
647
+
648
+ see the pyCMS documentation for details on rendering intents and what
649
+ they do.
650
+ :param flags: Integer (0-...) specifying additional flags
651
+ :returns: A CmsTransform class object.
652
+ :exception PyCMSError:
653
+ """
654
+
655
+ if not isinstance(renderingIntent, int) or not (0 <= renderingIntent <= 3):
656
+ msg = "renderingIntent must be an integer between 0 and 3"
657
+ raise PyCMSError(msg)
658
+
659
+ if not isinstance(flags, int) or not (0 <= flags <= _MAX_FLAG):
660
+ msg = f"flags must be an integer between 0 and {_MAX_FLAG}"
661
+ raise PyCMSError(msg)
662
+
663
+ try:
664
+ if not isinstance(inputProfile, ImageCmsProfile):
665
+ inputProfile = ImageCmsProfile(inputProfile)
666
+ if not isinstance(outputProfile, ImageCmsProfile):
667
+ outputProfile = ImageCmsProfile(outputProfile)
668
+ if not isinstance(proofProfile, ImageCmsProfile):
669
+ proofProfile = ImageCmsProfile(proofProfile)
670
+ return ImageCmsTransform(
671
+ inputProfile,
672
+ outputProfile,
673
+ inMode,
674
+ outMode,
675
+ renderingIntent,
676
+ proofProfile,
677
+ proofRenderingIntent,
678
+ flags,
679
+ )
680
+ except (OSError, TypeError, ValueError) as v:
681
+ raise PyCMSError(v) from v
682
+
683
+
684
+ buildTransformFromOpenProfiles = buildTransform
685
+ buildProofTransformFromOpenProfiles = buildProofTransform
686
+
687
+
688
+ def applyTransform(
689
+ im: Image.Image, transform: ImageCmsTransform, inPlace: bool = False
690
+ ) -> Image.Image | None:
691
+ """
692
+ (pyCMS) Applies a transform to a given image.
693
+
694
+ If ``im.mode != transform.input_mode``, a :exc:`PyCMSError` is raised.
695
+
696
+ If ``inPlace`` is ``True`` and ``transform.input_mode != transform.output_mode``, a
697
+ :exc:`PyCMSError` is raised.
698
+
699
+ If ``im.mode``, ``transform.input_mode`` or ``transform.output_mode`` is not
700
+ supported by pyCMSdll or the profiles you used for the transform, a
701
+ :exc:`PyCMSError` is raised.
702
+
703
+ If an error occurs while the transform is being applied,
704
+ a :exc:`PyCMSError` is raised.
705
+
706
+ This function applies a pre-calculated transform (from
707
+ ImageCms.buildTransform() or ImageCms.buildTransformFromOpenProfiles())
708
+ to an image. The transform can be used for multiple images, saving
709
+ considerable calculation time if doing the same conversion multiple times.
710
+
711
+ If you want to modify im in-place instead of receiving a new image as
712
+ the return value, set ``inPlace`` to ``True``. This can only be done if
713
+ ``transform.input_mode`` and ``transform.output_mode`` are the same, because we
714
+ can't change the mode in-place (the buffer sizes for some modes are
715
+ different). The default behavior is to return a new :py:class:`~PIL.Image.Image`
716
+ object of the same dimensions in mode ``transform.output_mode``.
717
+
718
+ :param im: An :py:class:`~PIL.Image.Image` object, and ``im.mode`` must be the same
719
+ as the ``input_mode`` supported by the transform.
720
+ :param transform: A valid CmsTransform class object
721
+ :param inPlace: Bool. If ``True``, ``im`` is modified in place and ``None`` is
722
+ returned, if ``False``, a new :py:class:`~PIL.Image.Image` object with the
723
+ transform applied is returned (and ``im`` is not changed). The default is
724
+ ``False``.
725
+ :returns: Either ``None``, or a new :py:class:`~PIL.Image.Image` object,
726
+ depending on the value of ``inPlace``. The profile will be returned in
727
+ the image's ``info['icc_profile']``.
728
+ :exception PyCMSError:
729
+ """
730
+
731
+ try:
732
+ if inPlace:
733
+ transform.apply_in_place(im)
734
+ imOut = None
735
+ else:
736
+ imOut = transform.apply(im)
737
+ except (TypeError, ValueError) as v:
738
+ raise PyCMSError(v) from v
739
+
740
+ return imOut
741
+
742
+
743
+ def createProfile(
744
+ colorSpace: Literal["LAB", "XYZ", "sRGB"], colorTemp: SupportsFloat = 0
745
+ ) -> core.CmsProfile:
746
+ """
747
+ (pyCMS) Creates a profile.
748
+
749
+ If colorSpace not in ``["LAB", "XYZ", "sRGB"]``,
750
+ a :exc:`PyCMSError` is raised.
751
+
752
+ If using LAB and ``colorTemp`` is not a positive integer,
753
+ a :exc:`PyCMSError` is raised.
754
+
755
+ If an error occurs while creating the profile,
756
+ a :exc:`PyCMSError` is raised.
757
+
758
+ Use this function to create common profiles on-the-fly instead of
759
+ having to supply a profile on disk and knowing the path to it. It
760
+ returns a normal CmsProfile object that can be passed to
761
+ ImageCms.buildTransformFromOpenProfiles() to create a transform to apply
762
+ to images.
763
+
764
+ :param colorSpace: String, the color space of the profile you wish to
765
+ create.
766
+ Currently only "LAB", "XYZ", and "sRGB" are supported.
767
+ :param colorTemp: Positive number for the white point for the profile, in
768
+ degrees Kelvin (i.e. 5000, 6500, 9600, etc.). The default is for D50
769
+ illuminant if omitted (5000k). colorTemp is ONLY applied to LAB
770
+ profiles, and is ignored for XYZ and sRGB.
771
+ :returns: A CmsProfile class object
772
+ :exception PyCMSError:
773
+ """
774
+
775
+ if colorSpace not in ["LAB", "XYZ", "sRGB"]:
776
+ msg = (
777
+ f"Color space not supported for on-the-fly profile creation ({colorSpace})"
778
+ )
779
+ raise PyCMSError(msg)
780
+
781
+ if colorSpace == "LAB":
782
+ try:
783
+ colorTemp = float(colorTemp)
784
+ except (TypeError, ValueError) as e:
785
+ msg = f'Color temperature must be numeric, "{colorTemp}" not valid'
786
+ raise PyCMSError(msg) from e
787
+
788
+ try:
789
+ return core.createProfile(colorSpace, colorTemp)
790
+ except (TypeError, ValueError) as v:
791
+ raise PyCMSError(v) from v
792
+
793
+
794
+ def getProfileName(profile: _CmsProfileCompatible) -> str:
795
+ """
796
+
797
+ (pyCMS) Gets the internal product name for the given profile.
798
+
799
+ If ``profile`` isn't a valid CmsProfile object or filename to a profile,
800
+ a :exc:`PyCMSError` is raised If an error occurs while trying
801
+ to obtain the name tag, a :exc:`PyCMSError` is raised.
802
+
803
+ Use this function to obtain the INTERNAL name of the profile (stored
804
+ in an ICC tag in the profile itself), usually the one used when the
805
+ profile was originally created. Sometimes this tag also contains
806
+ additional information supplied by the creator.
807
+
808
+ :param profile: EITHER a valid CmsProfile object, OR a string of the
809
+ filename of an ICC profile.
810
+ :returns: A string containing the internal name of the profile as stored
811
+ in an ICC tag.
812
+ :exception PyCMSError:
813
+ """
814
+
815
+ try:
816
+ # add an extra newline to preserve pyCMS compatibility
817
+ if not isinstance(profile, ImageCmsProfile):
818
+ profile = ImageCmsProfile(profile)
819
+ # do it in python, not c.
820
+ # // name was "%s - %s" (model, manufacturer) || Description ,
821
+ # // but if the Model and Manufacturer were the same or the model
822
+ # // was long, Just the model, in 1.x
823
+ model = profile.profile.model
824
+ manufacturer = profile.profile.manufacturer
825
+
826
+ if not (model or manufacturer):
827
+ return (profile.profile.profile_description or "") + "\n"
828
+ if not manufacturer or (model and len(model) > 30):
829
+ return f"{model}\n"
830
+ return f"{model} - {manufacturer}\n"
831
+
832
+ except (AttributeError, OSError, TypeError, ValueError) as v:
833
+ raise PyCMSError(v) from v
834
+
835
+
836
+ def getProfileInfo(profile: _CmsProfileCompatible) -> str:
837
+ """
838
+ (pyCMS) Gets the internal product information for the given profile.
839
+
840
+ If ``profile`` isn't a valid CmsProfile object or filename to a profile,
841
+ a :exc:`PyCMSError` is raised.
842
+
843
+ If an error occurs while trying to obtain the info tag,
844
+ a :exc:`PyCMSError` is raised.
845
+
846
+ Use this function to obtain the information stored in the profile's
847
+ info tag. This often contains details about the profile, and how it
848
+ was created, as supplied by the creator.
849
+
850
+ :param profile: EITHER a valid CmsProfile object, OR a string of the
851
+ filename of an ICC profile.
852
+ :returns: A string containing the internal profile information stored in
853
+ an ICC tag.
854
+ :exception PyCMSError:
855
+ """
856
+
857
+ try:
858
+ if not isinstance(profile, ImageCmsProfile):
859
+ profile = ImageCmsProfile(profile)
860
+ # add an extra newline to preserve pyCMS compatibility
861
+ # Python, not C. the white point bits weren't working well,
862
+ # so skipping.
863
+ # info was description \r\n\r\n copyright \r\n\r\n K007 tag \r\n\r\n whitepoint
864
+ description = profile.profile.profile_description
865
+ cpright = profile.profile.copyright
866
+ elements = [element for element in (description, cpright) if element]
867
+ return "\r\n\r\n".join(elements) + "\r\n\r\n"
868
+
869
+ except (AttributeError, OSError, TypeError, ValueError) as v:
870
+ raise PyCMSError(v) from v
871
+
872
+
873
+ def getProfileCopyright(profile: _CmsProfileCompatible) -> str:
874
+ """
875
+ (pyCMS) Gets the copyright for the given profile.
876
+
877
+ If ``profile`` isn't a valid CmsProfile object or filename to a profile, a
878
+ :exc:`PyCMSError` is raised.
879
+
880
+ If an error occurs while trying to obtain the copyright tag,
881
+ a :exc:`PyCMSError` is raised.
882
+
883
+ Use this function to obtain the information stored in the profile's
884
+ copyright tag.
885
+
886
+ :param profile: EITHER a valid CmsProfile object, OR a string of the
887
+ filename of an ICC profile.
888
+ :returns: A string containing the internal profile information stored in
889
+ an ICC tag.
890
+ :exception PyCMSError:
891
+ """
892
+ try:
893
+ # add an extra newline to preserve pyCMS compatibility
894
+ if not isinstance(profile, ImageCmsProfile):
895
+ profile = ImageCmsProfile(profile)
896
+ return (profile.profile.copyright or "") + "\n"
897
+ except (AttributeError, OSError, TypeError, ValueError) as v:
898
+ raise PyCMSError(v) from v
899
+
900
+
901
+ def getProfileManufacturer(profile: _CmsProfileCompatible) -> str:
902
+ """
903
+ (pyCMS) Gets the manufacturer for the given profile.
904
+
905
+ If ``profile`` isn't a valid CmsProfile object or filename to a profile, a
906
+ :exc:`PyCMSError` is raised.
907
+
908
+ If an error occurs while trying to obtain the manufacturer tag, a
909
+ :exc:`PyCMSError` is raised.
910
+
911
+ Use this function to obtain the information stored in the profile's
912
+ manufacturer tag.
913
+
914
+ :param profile: EITHER a valid CmsProfile object, OR a string of the
915
+ filename of an ICC profile.
916
+ :returns: A string containing the internal profile information stored in
917
+ an ICC tag.
918
+ :exception PyCMSError:
919
+ """
920
+ try:
921
+ # add an extra newline to preserve pyCMS compatibility
922
+ if not isinstance(profile, ImageCmsProfile):
923
+ profile = ImageCmsProfile(profile)
924
+ return (profile.profile.manufacturer or "") + "\n"
925
+ except (AttributeError, OSError, TypeError, ValueError) as v:
926
+ raise PyCMSError(v) from v
927
+
928
+
929
+ def getProfileModel(profile: _CmsProfileCompatible) -> str:
930
+ """
931
+ (pyCMS) Gets the model for the given profile.
932
+
933
+ If ``profile`` isn't a valid CmsProfile object or filename to a profile, a
934
+ :exc:`PyCMSError` is raised.
935
+
936
+ If an error occurs while trying to obtain the model tag,
937
+ a :exc:`PyCMSError` is raised.
938
+
939
+ Use this function to obtain the information stored in the profile's
940
+ model tag.
941
+
942
+ :param profile: EITHER a valid CmsProfile object, OR a string of the
943
+ filename of an ICC profile.
944
+ :returns: A string containing the internal profile information stored in
945
+ an ICC tag.
946
+ :exception PyCMSError:
947
+ """
948
+
949
+ try:
950
+ # add an extra newline to preserve pyCMS compatibility
951
+ if not isinstance(profile, ImageCmsProfile):
952
+ profile = ImageCmsProfile(profile)
953
+ return (profile.profile.model or "") + "\n"
954
+ except (AttributeError, OSError, TypeError, ValueError) as v:
955
+ raise PyCMSError(v) from v
956
+
957
+
958
+ def getProfileDescription(profile: _CmsProfileCompatible) -> str:
959
+ """
960
+ (pyCMS) Gets the description for the given profile.
961
+
962
+ If ``profile`` isn't a valid CmsProfile object or filename to a profile, a
963
+ :exc:`PyCMSError` is raised.
964
+
965
+ If an error occurs while trying to obtain the description tag,
966
+ a :exc:`PyCMSError` is raised.
967
+
968
+ Use this function to obtain the information stored in the profile's
969
+ description tag.
970
+
971
+ :param profile: EITHER a valid CmsProfile object, OR a string of the
972
+ filename of an ICC profile.
973
+ :returns: A string containing the internal profile information stored in an
974
+ ICC tag.
975
+ :exception PyCMSError:
976
+ """
977
+
978
+ try:
979
+ # add an extra newline to preserve pyCMS compatibility
980
+ if not isinstance(profile, ImageCmsProfile):
981
+ profile = ImageCmsProfile(profile)
982
+ return (profile.profile.profile_description or "") + "\n"
983
+ except (AttributeError, OSError, TypeError, ValueError) as v:
984
+ raise PyCMSError(v) from v
985
+
986
+
987
+ def getDefaultIntent(profile: _CmsProfileCompatible) -> int:
988
+ """
989
+ (pyCMS) Gets the default intent name for the given profile.
990
+
991
+ If ``profile`` isn't a valid CmsProfile object or filename to a profile, a
992
+ :exc:`PyCMSError` is raised.
993
+
994
+ If an error occurs while trying to obtain the default intent, a
995
+ :exc:`PyCMSError` is raised.
996
+
997
+ Use this function to determine the default (and usually best optimized)
998
+ rendering intent for this profile. Most profiles support multiple
999
+ rendering intents, but are intended mostly for one type of conversion.
1000
+ If you wish to use a different intent than returned, use
1001
+ ImageCms.isIntentSupported() to verify it will work first.
1002
+
1003
+ :param profile: EITHER a valid CmsProfile object, OR a string of the
1004
+ filename of an ICC profile.
1005
+ :returns: Integer 0-3 specifying the default rendering intent for this
1006
+ profile.
1007
+
1008
+ ImageCms.Intent.PERCEPTUAL = 0 (DEFAULT)
1009
+ ImageCms.Intent.RELATIVE_COLORIMETRIC = 1
1010
+ ImageCms.Intent.SATURATION = 2
1011
+ ImageCms.Intent.ABSOLUTE_COLORIMETRIC = 3
1012
+
1013
+ see the pyCMS documentation for details on rendering intents and what
1014
+ they do.
1015
+ :exception PyCMSError:
1016
+ """
1017
+
1018
+ try:
1019
+ if not isinstance(profile, ImageCmsProfile):
1020
+ profile = ImageCmsProfile(profile)
1021
+ return profile.profile.rendering_intent
1022
+ except (AttributeError, OSError, TypeError, ValueError) as v:
1023
+ raise PyCMSError(v) from v
1024
+
1025
+
1026
+ def isIntentSupported(
1027
+ profile: _CmsProfileCompatible, intent: Intent, direction: Direction
1028
+ ) -> Literal[-1, 1]:
1029
+ """
1030
+ (pyCMS) Checks if a given intent is supported.
1031
+
1032
+ Use this function to verify that you can use your desired
1033
+ ``intent`` with ``profile``, and that ``profile`` can be used for the
1034
+ input/output/proof profile as you desire.
1035
+
1036
+ Some profiles are created specifically for one "direction", can cannot
1037
+ be used for others. Some profiles can only be used for certain
1038
+ rendering intents, so it's best to either verify this before trying
1039
+ to create a transform with them (using this function), or catch the
1040
+ potential :exc:`PyCMSError` that will occur if they don't
1041
+ support the modes you select.
1042
+
1043
+ :param profile: EITHER a valid CmsProfile object, OR a string of the
1044
+ filename of an ICC profile.
1045
+ :param intent: Integer (0-3) specifying the rendering intent you wish to
1046
+ use with this profile
1047
+
1048
+ ImageCms.Intent.PERCEPTUAL = 0 (DEFAULT)
1049
+ ImageCms.Intent.RELATIVE_COLORIMETRIC = 1
1050
+ ImageCms.Intent.SATURATION = 2
1051
+ ImageCms.Intent.ABSOLUTE_COLORIMETRIC = 3
1052
+
1053
+ see the pyCMS documentation for details on rendering intents and what
1054
+ they do.
1055
+ :param direction: Integer specifying if the profile is to be used for
1056
+ input, output, or proof
1057
+
1058
+ INPUT = 0 (or use ImageCms.Direction.INPUT)
1059
+ OUTPUT = 1 (or use ImageCms.Direction.OUTPUT)
1060
+ PROOF = 2 (or use ImageCms.Direction.PROOF)
1061
+
1062
+ :returns: 1 if the intent/direction are supported, -1 if they are not.
1063
+ :exception PyCMSError:
1064
+ """
1065
+
1066
+ try:
1067
+ if not isinstance(profile, ImageCmsProfile):
1068
+ profile = ImageCmsProfile(profile)
1069
+ # FIXME: I get different results for the same data w. different
1070
+ # compilers. Bug in LittleCMS or in the binding?
1071
+ if profile.profile.is_intent_supported(intent, direction):
1072
+ return 1
1073
+ else:
1074
+ return -1
1075
+ except (AttributeError, OSError, TypeError, ValueError) as v:
1076
+ raise PyCMSError(v) from v
.venv/Lib/site-packages/PIL/ImageColor.py ADDED
@@ -0,0 +1,320 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #
2
+ # The Python Imaging Library
3
+ # $Id$
4
+ #
5
+ # map CSS3-style colour description strings to RGB
6
+ #
7
+ # History:
8
+ # 2002-10-24 fl Added support for CSS-style color strings
9
+ # 2002-12-15 fl Added RGBA support
10
+ # 2004-03-27 fl Fixed remaining int() problems for Python 1.5.2
11
+ # 2004-07-19 fl Fixed gray/grey spelling issues
12
+ # 2009-03-05 fl Fixed rounding error in grayscale calculation
13
+ #
14
+ # Copyright (c) 2002-2004 by Secret Labs AB
15
+ # Copyright (c) 2002-2004 by Fredrik Lundh
16
+ #
17
+ # See the README file for information on usage and redistribution.
18
+ #
19
+ from __future__ import annotations
20
+
21
+ import re
22
+ from functools import lru_cache
23
+
24
+ from . import Image
25
+
26
+
27
+ @lru_cache
28
+ def getrgb(color: str) -> tuple[int, int, int] | tuple[int, int, int, int]:
29
+ """
30
+ Convert a color string to an RGB or RGBA tuple. If the string cannot be
31
+ parsed, this function raises a :py:exc:`ValueError` exception.
32
+
33
+ .. versionadded:: 1.1.4
34
+
35
+ :param color: A color string
36
+ :return: ``(red, green, blue[, alpha])``
37
+ """
38
+ if len(color) > 100:
39
+ msg = "color specifier is too long"
40
+ raise ValueError(msg)
41
+ color = color.lower()
42
+
43
+ rgb = colormap.get(color, None)
44
+ if rgb:
45
+ if isinstance(rgb, tuple):
46
+ return rgb
47
+ rgb_tuple = getrgb(rgb)
48
+ assert len(rgb_tuple) == 3
49
+ colormap[color] = rgb_tuple
50
+ return rgb_tuple
51
+
52
+ # check for known string formats
53
+ if re.match("#[a-f0-9]{3}$", color):
54
+ return int(color[1] * 2, 16), int(color[2] * 2, 16), int(color[3] * 2, 16)
55
+
56
+ if re.match("#[a-f0-9]{4}$", color):
57
+ return (
58
+ int(color[1] * 2, 16),
59
+ int(color[2] * 2, 16),
60
+ int(color[3] * 2, 16),
61
+ int(color[4] * 2, 16),
62
+ )
63
+
64
+ if re.match("#[a-f0-9]{6}$", color):
65
+ return int(color[1:3], 16), int(color[3:5], 16), int(color[5:7], 16)
66
+
67
+ if re.match("#[a-f0-9]{8}$", color):
68
+ return (
69
+ int(color[1:3], 16),
70
+ int(color[3:5], 16),
71
+ int(color[5:7], 16),
72
+ int(color[7:9], 16),
73
+ )
74
+
75
+ m = re.match(r"rgb\(\s*(\d+)\s*,\s*(\d+)\s*,\s*(\d+)\s*\)$", color)
76
+ if m:
77
+ return int(m.group(1)), int(m.group(2)), int(m.group(3))
78
+
79
+ m = re.match(r"rgb\(\s*(\d+)%\s*,\s*(\d+)%\s*,\s*(\d+)%\s*\)$", color)
80
+ if m:
81
+ return (
82
+ int((int(m.group(1)) * 255) / 100.0 + 0.5),
83
+ int((int(m.group(2)) * 255) / 100.0 + 0.5),
84
+ int((int(m.group(3)) * 255) / 100.0 + 0.5),
85
+ )
86
+
87
+ m = re.match(
88
+ r"hsl\(\s*(\d+\.?\d*)\s*,\s*(\d+\.?\d*)%\s*,\s*(\d+\.?\d*)%\s*\)$", color
89
+ )
90
+ if m:
91
+ from colorsys import hls_to_rgb
92
+
93
+ rgb_floats = hls_to_rgb(
94
+ float(m.group(1)) / 360.0,
95
+ float(m.group(3)) / 100.0,
96
+ float(m.group(2)) / 100.0,
97
+ )
98
+ return (
99
+ int(rgb_floats[0] * 255 + 0.5),
100
+ int(rgb_floats[1] * 255 + 0.5),
101
+ int(rgb_floats[2] * 255 + 0.5),
102
+ )
103
+
104
+ m = re.match(
105
+ r"hs[bv]\(\s*(\d+\.?\d*)\s*,\s*(\d+\.?\d*)%\s*,\s*(\d+\.?\d*)%\s*\)$", color
106
+ )
107
+ if m:
108
+ from colorsys import hsv_to_rgb
109
+
110
+ rgb_floats = hsv_to_rgb(
111
+ float(m.group(1)) / 360.0,
112
+ float(m.group(2)) / 100.0,
113
+ float(m.group(3)) / 100.0,
114
+ )
115
+ return (
116
+ int(rgb_floats[0] * 255 + 0.5),
117
+ int(rgb_floats[1] * 255 + 0.5),
118
+ int(rgb_floats[2] * 255 + 0.5),
119
+ )
120
+
121
+ m = re.match(r"rgba\(\s*(\d+)\s*,\s*(\d+)\s*,\s*(\d+)\s*,\s*(\d+)\s*\)$", color)
122
+ if m:
123
+ return int(m.group(1)), int(m.group(2)), int(m.group(3)), int(m.group(4))
124
+ msg = f"unknown color specifier: {repr(color)}"
125
+ raise ValueError(msg)
126
+
127
+
128
+ @lru_cache
129
+ def getcolor(color: str, mode: str) -> int | tuple[int, ...]:
130
+ """
131
+ Same as :py:func:`~PIL.ImageColor.getrgb` for most modes. However, if
132
+ ``mode`` is HSV, converts the RGB value to a HSV value, or if ``mode`` is
133
+ not color or a palette image, converts the RGB value to a grayscale value.
134
+ If the string cannot be parsed, this function raises a :py:exc:`ValueError`
135
+ exception.
136
+
137
+ .. versionadded:: 1.1.4
138
+
139
+ :param color: A color string
140
+ :param mode: Convert result to this mode
141
+ :return: ``graylevel, (graylevel, alpha) or (red, green, blue[, alpha])``
142
+ """
143
+ # same as getrgb, but converts the result to the given mode
144
+ rgb, alpha = getrgb(color), 255
145
+ if len(rgb) == 4:
146
+ alpha = rgb[3]
147
+ rgb = rgb[:3]
148
+
149
+ if mode == "HSV":
150
+ from colorsys import rgb_to_hsv
151
+
152
+ r, g, b = rgb
153
+ h, s, v = rgb_to_hsv(r / 255, g / 255, b / 255)
154
+ return int(h * 255), int(s * 255), int(v * 255)
155
+ elif Image.getmodebase(mode) == "L":
156
+ r, g, b = rgb
157
+ # ITU-R Recommendation 601-2 for nonlinear RGB
158
+ # scaled to 24 bits to match the convert's implementation.
159
+ graylevel = (r * 19595 + g * 38470 + b * 7471 + 0x8000) >> 16
160
+ if mode[-1] == "A":
161
+ return graylevel, alpha
162
+ return graylevel
163
+ elif mode[-1] == "A":
164
+ return rgb + (alpha,)
165
+ return rgb
166
+
167
+
168
+ colormap: dict[str, str | tuple[int, int, int]] = {
169
+ # X11 colour table from https://drafts.csswg.org/css-color-4/, with
170
+ # gray/grey spelling issues fixed. This is a superset of HTML 4.0
171
+ # colour names used in CSS 1.
172
+ "aliceblue": "#f0f8ff",
173
+ "antiquewhite": "#faebd7",
174
+ "aqua": "#00ffff",
175
+ "aquamarine": "#7fffd4",
176
+ "azure": "#f0ffff",
177
+ "beige": "#f5f5dc",
178
+ "bisque": "#ffe4c4",
179
+ "black": "#000000",
180
+ "blanchedalmond": "#ffebcd",
181
+ "blue": "#0000ff",
182
+ "blueviolet": "#8a2be2",
183
+ "brown": "#a52a2a",
184
+ "burlywood": "#deb887",
185
+ "cadetblue": "#5f9ea0",
186
+ "chartreuse": "#7fff00",
187
+ "chocolate": "#d2691e",
188
+ "coral": "#ff7f50",
189
+ "cornflowerblue": "#6495ed",
190
+ "cornsilk": "#fff8dc",
191
+ "crimson": "#dc143c",
192
+ "cyan": "#00ffff",
193
+ "darkblue": "#00008b",
194
+ "darkcyan": "#008b8b",
195
+ "darkgoldenrod": "#b8860b",
196
+ "darkgray": "#a9a9a9",
197
+ "darkgrey": "#a9a9a9",
198
+ "darkgreen": "#006400",
199
+ "darkkhaki": "#bdb76b",
200
+ "darkmagenta": "#8b008b",
201
+ "darkolivegreen": "#556b2f",
202
+ "darkorange": "#ff8c00",
203
+ "darkorchid": "#9932cc",
204
+ "darkred": "#8b0000",
205
+ "darksalmon": "#e9967a",
206
+ "darkseagreen": "#8fbc8f",
207
+ "darkslateblue": "#483d8b",
208
+ "darkslategray": "#2f4f4f",
209
+ "darkslategrey": "#2f4f4f",
210
+ "darkturquoise": "#00ced1",
211
+ "darkviolet": "#9400d3",
212
+ "deeppink": "#ff1493",
213
+ "deepskyblue": "#00bfff",
214
+ "dimgray": "#696969",
215
+ "dimgrey": "#696969",
216
+ "dodgerblue": "#1e90ff",
217
+ "firebrick": "#b22222",
218
+ "floralwhite": "#fffaf0",
219
+ "forestgreen": "#228b22",
220
+ "fuchsia": "#ff00ff",
221
+ "gainsboro": "#dcdcdc",
222
+ "ghostwhite": "#f8f8ff",
223
+ "gold": "#ffd700",
224
+ "goldenrod": "#daa520",
225
+ "gray": "#808080",
226
+ "grey": "#808080",
227
+ "green": "#008000",
228
+ "greenyellow": "#adff2f",
229
+ "honeydew": "#f0fff0",
230
+ "hotpink": "#ff69b4",
231
+ "indianred": "#cd5c5c",
232
+ "indigo": "#4b0082",
233
+ "ivory": "#fffff0",
234
+ "khaki": "#f0e68c",
235
+ "lavender": "#e6e6fa",
236
+ "lavenderblush": "#fff0f5",
237
+ "lawngreen": "#7cfc00",
238
+ "lemonchiffon": "#fffacd",
239
+ "lightblue": "#add8e6",
240
+ "lightcoral": "#f08080",
241
+ "lightcyan": "#e0ffff",
242
+ "lightgoldenrodyellow": "#fafad2",
243
+ "lightgreen": "#90ee90",
244
+ "lightgray": "#d3d3d3",
245
+ "lightgrey": "#d3d3d3",
246
+ "lightpink": "#ffb6c1",
247
+ "lightsalmon": "#ffa07a",
248
+ "lightseagreen": "#20b2aa",
249
+ "lightskyblue": "#87cefa",
250
+ "lightslategray": "#778899",
251
+ "lightslategrey": "#778899",
252
+ "lightsteelblue": "#b0c4de",
253
+ "lightyellow": "#ffffe0",
254
+ "lime": "#00ff00",
255
+ "limegreen": "#32cd32",
256
+ "linen": "#faf0e6",
257
+ "magenta": "#ff00ff",
258
+ "maroon": "#800000",
259
+ "mediumaquamarine": "#66cdaa",
260
+ "mediumblue": "#0000cd",
261
+ "mediumorchid": "#ba55d3",
262
+ "mediumpurple": "#9370db",
263
+ "mediumseagreen": "#3cb371",
264
+ "mediumslateblue": "#7b68ee",
265
+ "mediumspringgreen": "#00fa9a",
266
+ "mediumturquoise": "#48d1cc",
267
+ "mediumvioletred": "#c71585",
268
+ "midnightblue": "#191970",
269
+ "mintcream": "#f5fffa",
270
+ "mistyrose": "#ffe4e1",
271
+ "moccasin": "#ffe4b5",
272
+ "navajowhite": "#ffdead",
273
+ "navy": "#000080",
274
+ "oldlace": "#fdf5e6",
275
+ "olive": "#808000",
276
+ "olivedrab": "#6b8e23",
277
+ "orange": "#ffa500",
278
+ "orangered": "#ff4500",
279
+ "orchid": "#da70d6",
280
+ "palegoldenrod": "#eee8aa",
281
+ "palegreen": "#98fb98",
282
+ "paleturquoise": "#afeeee",
283
+ "palevioletred": "#db7093",
284
+ "papayawhip": "#ffefd5",
285
+ "peachpuff": "#ffdab9",
286
+ "peru": "#cd853f",
287
+ "pink": "#ffc0cb",
288
+ "plum": "#dda0dd",
289
+ "powderblue": "#b0e0e6",
290
+ "purple": "#800080",
291
+ "rebeccapurple": "#663399",
292
+ "red": "#ff0000",
293
+ "rosybrown": "#bc8f8f",
294
+ "royalblue": "#4169e1",
295
+ "saddlebrown": "#8b4513",
296
+ "salmon": "#fa8072",
297
+ "sandybrown": "#f4a460",
298
+ "seagreen": "#2e8b57",
299
+ "seashell": "#fff5ee",
300
+ "sienna": "#a0522d",
301
+ "silver": "#c0c0c0",
302
+ "skyblue": "#87ceeb",
303
+ "slateblue": "#6a5acd",
304
+ "slategray": "#708090",
305
+ "slategrey": "#708090",
306
+ "snow": "#fffafa",
307
+ "springgreen": "#00ff7f",
308
+ "steelblue": "#4682b4",
309
+ "tan": "#d2b48c",
310
+ "teal": "#008080",
311
+ "thistle": "#d8bfd8",
312
+ "tomato": "#ff6347",
313
+ "turquoise": "#40e0d0",
314
+ "violet": "#ee82ee",
315
+ "wheat": "#f5deb3",
316
+ "white": "#ffffff",
317
+ "whitesmoke": "#f5f5f5",
318
+ "yellow": "#ffff00",
319
+ "yellowgreen": "#9acd32",
320
+ }
.venv/Lib/site-packages/PIL/ImageDraw.py ADDED
@@ -0,0 +1,1036 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #
2
+ # The Python Imaging Library
3
+ # $Id$
4
+ #
5
+ # drawing interface operations
6
+ #
7
+ # History:
8
+ # 1996-04-13 fl Created (experimental)
9
+ # 1996-08-07 fl Filled polygons, ellipses.
10
+ # 1996-08-13 fl Added text support
11
+ # 1998-06-28 fl Handle I and F images
12
+ # 1998-12-29 fl Added arc; use arc primitive to draw ellipses
13
+ # 1999-01-10 fl Added shape stuff (experimental)
14
+ # 1999-02-06 fl Added bitmap support
15
+ # 1999-02-11 fl Changed all primitives to take options
16
+ # 1999-02-20 fl Fixed backwards compatibility
17
+ # 2000-10-12 fl Copy on write, when necessary
18
+ # 2001-02-18 fl Use default ink for bitmap/text also in fill mode
19
+ # 2002-10-24 fl Added support for CSS-style color strings
20
+ # 2002-12-10 fl Added experimental support for RGBA-on-RGB drawing
21
+ # 2002-12-11 fl Refactored low-level drawing API (work in progress)
22
+ # 2004-08-26 fl Made Draw() a factory function, added getdraw() support
23
+ # 2004-09-04 fl Added width support to line primitive
24
+ # 2004-09-10 fl Added font mode handling
25
+ # 2006-06-19 fl Added font bearing support (getmask2)
26
+ #
27
+ # Copyright (c) 1997-2006 by Secret Labs AB
28
+ # Copyright (c) 1996-2006 by Fredrik Lundh
29
+ #
30
+ # See the README file for information on usage and redistribution.
31
+ #
32
+ from __future__ import annotations
33
+
34
+ import math
35
+ import struct
36
+ from collections.abc import Sequence
37
+ from typing import cast
38
+
39
+ from . import Image, ImageColor, ImageText
40
+
41
+ TYPE_CHECKING = False
42
+ if TYPE_CHECKING:
43
+ from collections.abc import Callable
44
+ from types import ModuleType
45
+ from typing import Any, AnyStr
46
+
47
+ from . import ImageDraw2, ImageFont
48
+ from ._typing import Coords, _Ink
49
+
50
+ # experimental access to the outline API
51
+ Outline: Callable[[], Image.core._Outline] = Image.core.outline
52
+
53
+ """
54
+ A simple 2D drawing interface for PIL images.
55
+ <p>
56
+ Application code should use the <b>Draw</b> factory, instead of
57
+ directly.
58
+ """
59
+
60
+
61
+ class ImageDraw:
62
+ font: (
63
+ ImageFont.ImageFont | ImageFont.FreeTypeFont | ImageFont.TransposedFont | None
64
+ ) = None
65
+
66
+ def __init__(self, im: Image.Image, mode: str | None = None) -> None:
67
+ """
68
+ Create a drawing instance.
69
+
70
+ :param im: The image to draw in.
71
+ :param mode: Optional mode to use for color values. For RGB
72
+ images, this argument can be RGB or RGBA (to blend the
73
+ drawing into the image). For all other modes, this argument
74
+ must be the same as the image mode. If omitted, the mode
75
+ defaults to the mode of the image.
76
+ """
77
+ im._ensure_mutable()
78
+ blend = 0
79
+ if mode is None:
80
+ mode = im.mode
81
+ if mode != im.mode:
82
+ if mode == "RGBA" and im.mode == "RGB":
83
+ blend = 1
84
+ else:
85
+ msg = "mode mismatch"
86
+ raise ValueError(msg)
87
+ if mode == "P":
88
+ self.palette = im.palette
89
+ else:
90
+ self.palette = None
91
+ self._image = im
92
+ self.im = im.im
93
+ self.draw = Image.core.draw(self.im, blend)
94
+ self.mode = mode
95
+ if mode in ("I", "F"):
96
+ self.ink = self.draw.draw_ink(1)
97
+ else:
98
+ self.ink = self.draw.draw_ink(-1)
99
+ if mode in ("1", "P", "I", "F"):
100
+ # FIXME: fix Fill2 to properly support matte for I+F images
101
+ self.fontmode = "1"
102
+ else:
103
+ self.fontmode = "L" # aliasing is okay for other modes
104
+ self.fill = False
105
+
106
+ def getfont(
107
+ self,
108
+ ) -> ImageFont.ImageFont | ImageFont.FreeTypeFont | ImageFont.TransposedFont:
109
+ """
110
+ Get the current default font.
111
+
112
+ To set the default font for this ImageDraw instance::
113
+
114
+ from PIL import ImageDraw, ImageFont
115
+ draw.font = ImageFont.truetype("Tests/fonts/FreeMono.ttf")
116
+
117
+ To set the default font for all future ImageDraw instances::
118
+
119
+ from PIL import ImageDraw, ImageFont
120
+ ImageDraw.ImageDraw.font = ImageFont.truetype("Tests/fonts/FreeMono.ttf")
121
+
122
+ If the current default font is ``None``,
123
+ it is initialized with ``ImageFont.load_default()``.
124
+
125
+ :returns: An image font."""
126
+ if not self.font:
127
+ # FIXME: should add a font repository
128
+ from . import ImageFont
129
+
130
+ self.font = ImageFont.load_default()
131
+ return self.font
132
+
133
+ def _getfont(
134
+ self, font_size: float | None
135
+ ) -> ImageFont.ImageFont | ImageFont.FreeTypeFont | ImageFont.TransposedFont:
136
+ if font_size is not None:
137
+ from . import ImageFont
138
+
139
+ return ImageFont.load_default(font_size)
140
+ else:
141
+ return self.getfont()
142
+
143
+ def _getink(
144
+ self, ink: _Ink | None, fill: _Ink | None = None
145
+ ) -> tuple[int | None, int | None]:
146
+ result_ink = None
147
+ result_fill = None
148
+ if ink is None and fill is None:
149
+ if self.fill:
150
+ result_fill = self.ink
151
+ else:
152
+ result_ink = self.ink
153
+ else:
154
+ if ink is not None:
155
+ if isinstance(ink, str):
156
+ ink = ImageColor.getcolor(ink, self.mode)
157
+ if self.palette and isinstance(ink, tuple):
158
+ ink = self.palette.getcolor(ink, self._image)
159
+ result_ink = self.draw.draw_ink(ink)
160
+ if fill is not None:
161
+ if isinstance(fill, str):
162
+ fill = ImageColor.getcolor(fill, self.mode)
163
+ if self.palette and isinstance(fill, tuple):
164
+ fill = self.palette.getcolor(fill, self._image)
165
+ result_fill = self.draw.draw_ink(fill)
166
+ return result_ink, result_fill
167
+
168
+ def arc(
169
+ self,
170
+ xy: Coords,
171
+ start: float,
172
+ end: float,
173
+ fill: _Ink | None = None,
174
+ width: int = 1,
175
+ ) -> None:
176
+ """Draw an arc."""
177
+ ink, fill = self._getink(fill)
178
+ if ink is not None:
179
+ self.draw.draw_arc(xy, start, end, ink, width)
180
+
181
+ def bitmap(
182
+ self, xy: Sequence[int], bitmap: Image.Image, fill: _Ink | None = None
183
+ ) -> None:
184
+ """Draw a bitmap."""
185
+ bitmap.load()
186
+ ink, fill = self._getink(fill)
187
+ if ink is None:
188
+ ink = fill
189
+ if ink is not None:
190
+ self.draw.draw_bitmap(xy, bitmap.im, ink)
191
+
192
+ def chord(
193
+ self,
194
+ xy: Coords,
195
+ start: float,
196
+ end: float,
197
+ fill: _Ink | None = None,
198
+ outline: _Ink | None = None,
199
+ width: int = 1,
200
+ ) -> None:
201
+ """Draw a chord."""
202
+ ink, fill_ink = self._getink(outline, fill)
203
+ if fill_ink is not None:
204
+ self.draw.draw_chord(xy, start, end, fill_ink, 1)
205
+ if ink is not None and ink != fill_ink and width != 0:
206
+ self.draw.draw_chord(xy, start, end, ink, 0, width)
207
+
208
+ def ellipse(
209
+ self,
210
+ xy: Coords,
211
+ fill: _Ink | None = None,
212
+ outline: _Ink | None = None,
213
+ width: int = 1,
214
+ ) -> None:
215
+ """Draw an ellipse."""
216
+ ink, fill_ink = self._getink(outline, fill)
217
+ if fill_ink is not None:
218
+ self.draw.draw_ellipse(xy, fill_ink, 1)
219
+ if ink is not None and ink != fill_ink and width != 0:
220
+ self.draw.draw_ellipse(xy, ink, 0, width)
221
+
222
+ def circle(
223
+ self,
224
+ xy: Sequence[float],
225
+ radius: float,
226
+ fill: _Ink | None = None,
227
+ outline: _Ink | None = None,
228
+ width: int = 1,
229
+ ) -> None:
230
+ """Draw a circle given center coordinates and a radius."""
231
+ ellipse_xy = (xy[0] - radius, xy[1] - radius, xy[0] + radius, xy[1] + radius)
232
+ self.ellipse(ellipse_xy, fill, outline, width)
233
+
234
+ def line(
235
+ self,
236
+ xy: Coords,
237
+ fill: _Ink | None = None,
238
+ width: int = 0,
239
+ joint: str | None = None,
240
+ ) -> None:
241
+ """Draw a line, or a connected sequence of line segments."""
242
+ ink = self._getink(fill)[0]
243
+ if ink is not None:
244
+ self.draw.draw_lines(xy, ink, width)
245
+ if joint == "curve" and width > 4:
246
+ points: Sequence[Sequence[float]]
247
+ if isinstance(xy[0], (list, tuple)):
248
+ points = cast(Sequence[Sequence[float]], xy)
249
+ else:
250
+ points = [
251
+ cast(Sequence[float], tuple(xy[i : i + 2]))
252
+ for i in range(0, len(xy), 2)
253
+ ]
254
+ for i in range(1, len(points) - 1):
255
+ point = points[i]
256
+ angles = [
257
+ math.degrees(math.atan2(end[0] - start[0], start[1] - end[1]))
258
+ % 360
259
+ for start, end in (
260
+ (points[i - 1], point),
261
+ (point, points[i + 1]),
262
+ )
263
+ ]
264
+ if angles[0] == angles[1]:
265
+ # This is a straight line, so no joint is required
266
+ continue
267
+
268
+ def coord_at_angle(
269
+ coord: Sequence[float], angle: float
270
+ ) -> tuple[float, ...]:
271
+ x, y = coord
272
+ angle -= 90
273
+ distance = width / 2 - 1
274
+ return tuple(
275
+ p + (math.floor(p_d) if p_d > 0 else math.ceil(p_d))
276
+ for p, p_d in (
277
+ (x, distance * math.cos(math.radians(angle))),
278
+ (y, distance * math.sin(math.radians(angle))),
279
+ )
280
+ )
281
+
282
+ flipped = (
283
+ angles[1] > angles[0] and angles[1] - 180 > angles[0]
284
+ ) or (angles[1] < angles[0] and angles[1] + 180 > angles[0])
285
+ coords = [
286
+ (point[0] - width / 2 + 1, point[1] - width / 2 + 1),
287
+ (point[0] + width / 2 - 1, point[1] + width / 2 - 1),
288
+ ]
289
+ if flipped:
290
+ start, end = (angles[1] + 90, angles[0] + 90)
291
+ else:
292
+ start, end = (angles[0] - 90, angles[1] - 90)
293
+ self.pieslice(coords, start - 90, end - 90, fill)
294
+
295
+ if width > 8:
296
+ # Cover potential gaps between the line and the joint
297
+ if flipped:
298
+ gap_coords = [
299
+ coord_at_angle(point, angles[0] + 90),
300
+ point,
301
+ coord_at_angle(point, angles[1] + 90),
302
+ ]
303
+ else:
304
+ gap_coords = [
305
+ coord_at_angle(point, angles[0] - 90),
306
+ point,
307
+ coord_at_angle(point, angles[1] - 90),
308
+ ]
309
+ self.line(gap_coords, fill, width=3)
310
+
311
+ def shape(
312
+ self,
313
+ shape: Image.core._Outline,
314
+ fill: _Ink | None = None,
315
+ outline: _Ink | None = None,
316
+ ) -> None:
317
+ """(Experimental) Draw a shape."""
318
+ shape.close()
319
+ ink, fill_ink = self._getink(outline, fill)
320
+ if fill_ink is not None:
321
+ self.draw.draw_outline(shape, fill_ink, 1)
322
+ if ink is not None and ink != fill_ink:
323
+ self.draw.draw_outline(shape, ink, 0)
324
+
325
+ def pieslice(
326
+ self,
327
+ xy: Coords,
328
+ start: float,
329
+ end: float,
330
+ fill: _Ink | None = None,
331
+ outline: _Ink | None = None,
332
+ width: int = 1,
333
+ ) -> None:
334
+ """Draw a pieslice."""
335
+ ink, fill_ink = self._getink(outline, fill)
336
+ if fill_ink is not None:
337
+ self.draw.draw_pieslice(xy, start, end, fill_ink, 1)
338
+ if ink is not None and ink != fill_ink and width != 0:
339
+ self.draw.draw_pieslice(xy, start, end, ink, 0, width)
340
+
341
+ def point(self, xy: Coords, fill: _Ink | None = None) -> None:
342
+ """Draw one or more individual pixels."""
343
+ ink, fill = self._getink(fill)
344
+ if ink is not None:
345
+ self.draw.draw_points(xy, ink)
346
+
347
+ def polygon(
348
+ self,
349
+ xy: Coords,
350
+ fill: _Ink | None = None,
351
+ outline: _Ink | None = None,
352
+ width: int = 1,
353
+ ) -> None:
354
+ """Draw a polygon."""
355
+ ink, fill_ink = self._getink(outline, fill)
356
+ if fill_ink is not None:
357
+ self.draw.draw_polygon(xy, fill_ink, 1)
358
+ if ink is not None and ink != fill_ink and width != 0:
359
+ if width == 1:
360
+ self.draw.draw_polygon(xy, ink, 0, width)
361
+ elif self.im is not None:
362
+ # To avoid expanding the polygon outwards,
363
+ # use the fill as a mask
364
+ mask = Image.new("1", self.im.size)
365
+ mask_ink = self._getink(1)[0]
366
+ draw = Draw(mask)
367
+ draw.draw.draw_polygon(xy, mask_ink, 1)
368
+
369
+ self.draw.draw_polygon(xy, ink, 0, width * 2 - 1, mask.im)
370
+
371
+ def regular_polygon(
372
+ self,
373
+ bounding_circle: Sequence[Sequence[float] | float],
374
+ n_sides: int,
375
+ rotation: float = 0,
376
+ fill: _Ink | None = None,
377
+ outline: _Ink | None = None,
378
+ width: int = 1,
379
+ ) -> None:
380
+ """Draw a regular polygon."""
381
+ xy = _compute_regular_polygon_vertices(bounding_circle, n_sides, rotation)
382
+ self.polygon(xy, fill, outline, width)
383
+
384
+ def rectangle(
385
+ self,
386
+ xy: Coords,
387
+ fill: _Ink | None = None,
388
+ outline: _Ink | None = None,
389
+ width: int = 1,
390
+ ) -> None:
391
+ """Draw a rectangle."""
392
+ ink, fill_ink = self._getink(outline, fill)
393
+ if fill_ink is not None:
394
+ self.draw.draw_rectangle(xy, fill_ink, 1)
395
+ if ink is not None and ink != fill_ink and width != 0:
396
+ self.draw.draw_rectangle(xy, ink, 0, width)
397
+
398
+ def rounded_rectangle(
399
+ self,
400
+ xy: Coords,
401
+ radius: float = 0,
402
+ fill: _Ink | None = None,
403
+ outline: _Ink | None = None,
404
+ width: int = 1,
405
+ *,
406
+ corners: tuple[bool, bool, bool, bool] | None = None,
407
+ ) -> None:
408
+ """Draw a rounded rectangle."""
409
+ if isinstance(xy[0], (list, tuple)):
410
+ (x0, y0), (x1, y1) = cast(Sequence[Sequence[float]], xy)
411
+ else:
412
+ x0, y0, x1, y1 = cast(Sequence[float], xy)
413
+ if x1 < x0:
414
+ msg = "x1 must be greater than or equal to x0"
415
+ raise ValueError(msg)
416
+ if y1 < y0:
417
+ msg = "y1 must be greater than or equal to y0"
418
+ raise ValueError(msg)
419
+ if corners is None:
420
+ corners = (True, True, True, True)
421
+
422
+ d = radius * 2
423
+
424
+ x0 = round(x0)
425
+ y0 = round(y0)
426
+ x1 = round(x1)
427
+ y1 = round(y1)
428
+ full_x, full_y = False, False
429
+ if all(corners):
430
+ full_x = d >= x1 - x0 - 1
431
+ if full_x:
432
+ # The two left and two right corners are joined
433
+ d = x1 - x0
434
+ full_y = d >= y1 - y0 - 1
435
+ if full_y:
436
+ # The two top and two bottom corners are joined
437
+ d = y1 - y0
438
+ if full_x and full_y:
439
+ # If all corners are joined, that is a circle
440
+ return self.ellipse(xy, fill, outline, width)
441
+
442
+ if d == 0 or not any(corners):
443
+ # If the corners have no curve,
444
+ # or there are no corners,
445
+ # that is a rectangle
446
+ return self.rectangle(xy, fill, outline, width)
447
+
448
+ r = int(d // 2)
449
+ ink, fill_ink = self._getink(outline, fill)
450
+
451
+ def draw_corners(pieslice: bool) -> None:
452
+ parts: tuple[tuple[tuple[float, float, float, float], int, int], ...]
453
+ if full_x:
454
+ # Draw top and bottom halves
455
+ parts = (
456
+ ((x0, y0, x0 + d, y0 + d), 180, 360),
457
+ ((x0, y1 - d, x0 + d, y1), 0, 180),
458
+ )
459
+ elif full_y:
460
+ # Draw left and right halves
461
+ parts = (
462
+ ((x0, y0, x0 + d, y0 + d), 90, 270),
463
+ ((x1 - d, y0, x1, y0 + d), 270, 90),
464
+ )
465
+ else:
466
+ # Draw four separate corners
467
+ parts = tuple(
468
+ part
469
+ for i, part in enumerate(
470
+ (
471
+ ((x0, y0, x0 + d, y0 + d), 180, 270),
472
+ ((x1 - d, y0, x1, y0 + d), 270, 360),
473
+ ((x1 - d, y1 - d, x1, y1), 0, 90),
474
+ ((x0, y1 - d, x0 + d, y1), 90, 180),
475
+ )
476
+ )
477
+ if corners[i]
478
+ )
479
+ for part in parts:
480
+ if pieslice:
481
+ self.draw.draw_pieslice(*(part + (fill_ink, 1)))
482
+ else:
483
+ self.draw.draw_arc(*(part + (ink, width)))
484
+
485
+ if fill_ink is not None:
486
+ draw_corners(True)
487
+
488
+ if full_x:
489
+ self.draw.draw_rectangle((x0, y0 + r + 1, x1, y1 - r - 1), fill_ink, 1)
490
+ elif x1 - r - 1 > x0 + r + 1:
491
+ self.draw.draw_rectangle((x0 + r + 1, y0, x1 - r - 1, y1), fill_ink, 1)
492
+ if not full_x and not full_y:
493
+ left = [x0, y0, x0 + r, y1]
494
+ if corners[0]:
495
+ left[1] += r + 1
496
+ if corners[3]:
497
+ left[3] -= r + 1
498
+ self.draw.draw_rectangle(left, fill_ink, 1)
499
+
500
+ right = [x1 - r, y0, x1, y1]
501
+ if corners[1]:
502
+ right[1] += r + 1
503
+ if corners[2]:
504
+ right[3] -= r + 1
505
+ self.draw.draw_rectangle(right, fill_ink, 1)
506
+ if ink is not None and ink != fill_ink and width != 0:
507
+ draw_corners(False)
508
+
509
+ if not full_x:
510
+ top = [x0, y0, x1, y0 + width - 1]
511
+ if corners[0]:
512
+ top[0] += r + 1
513
+ if corners[1]:
514
+ top[2] -= r + 1
515
+ self.draw.draw_rectangle(top, ink, 1)
516
+
517
+ bottom = [x0, y1 - width + 1, x1, y1]
518
+ if corners[3]:
519
+ bottom[0] += r + 1
520
+ if corners[2]:
521
+ bottom[2] -= r + 1
522
+ self.draw.draw_rectangle(bottom, ink, 1)
523
+ if not full_y:
524
+ left = [x0, y0, x0 + width - 1, y1]
525
+ if corners[0]:
526
+ left[1] += r + 1
527
+ if corners[3]:
528
+ left[3] -= r + 1
529
+ self.draw.draw_rectangle(left, ink, 1)
530
+
531
+ right = [x1 - width + 1, y0, x1, y1]
532
+ if corners[1]:
533
+ right[1] += r + 1
534
+ if corners[2]:
535
+ right[3] -= r + 1
536
+ self.draw.draw_rectangle(right, ink, 1)
537
+
538
+ def text(
539
+ self,
540
+ xy: tuple[float, float],
541
+ text: AnyStr | ImageText.Text,
542
+ fill: _Ink | None = None,
543
+ font: (
544
+ ImageFont.ImageFont
545
+ | ImageFont.FreeTypeFont
546
+ | ImageFont.TransposedFont
547
+ | None
548
+ ) = None,
549
+ anchor: str | None = None,
550
+ spacing: float = 4,
551
+ align: str = "left",
552
+ direction: str | None = None,
553
+ features: list[str] | None = None,
554
+ language: str | None = None,
555
+ stroke_width: float = 0,
556
+ stroke_fill: _Ink | None = None,
557
+ embedded_color: bool = False,
558
+ *args: Any,
559
+ **kwargs: Any,
560
+ ) -> None:
561
+ """Draw text."""
562
+ if isinstance(text, ImageText.Text):
563
+ image_text = text
564
+ else:
565
+ if font is None:
566
+ font = self._getfont(kwargs.get("font_size"))
567
+ image_text = ImageText.Text(
568
+ text, font, self.mode, spacing, direction, features, language
569
+ )
570
+ if embedded_color:
571
+ image_text.embed_color()
572
+ if stroke_width:
573
+ image_text.stroke(stroke_width, stroke_fill)
574
+
575
+ def getink(fill: _Ink | None) -> int:
576
+ ink, fill_ink = self._getink(fill)
577
+ if ink is None:
578
+ assert fill_ink is not None
579
+ return fill_ink
580
+ return ink
581
+
582
+ ink = getink(fill)
583
+ if ink is None:
584
+ return
585
+
586
+ stroke_ink = None
587
+ if image_text.stroke_width:
588
+ stroke_ink = (
589
+ getink(image_text.stroke_fill)
590
+ if image_text.stroke_fill is not None
591
+ else ink
592
+ )
593
+
594
+ for xy, anchor, line in image_text._split(xy, anchor, align):
595
+
596
+ def draw_text(ink: int, stroke_width: float = 0) -> None:
597
+ mode = self.fontmode
598
+ if stroke_width == 0 and embedded_color:
599
+ mode = "RGBA"
600
+ coord = []
601
+ for i in range(2):
602
+ coord.append(int(xy[i]))
603
+ start = (math.modf(xy[0])[0], math.modf(xy[1])[0])
604
+ try:
605
+ mask, offset = image_text.font.getmask2( # type: ignore[union-attr,misc]
606
+ line,
607
+ mode,
608
+ direction=direction,
609
+ features=features,
610
+ language=language,
611
+ stroke_width=stroke_width,
612
+ stroke_filled=True,
613
+ anchor=anchor,
614
+ ink=ink,
615
+ start=start,
616
+ *args,
617
+ **kwargs,
618
+ )
619
+ coord = [coord[0] + offset[0], coord[1] + offset[1]]
620
+ except AttributeError:
621
+ try:
622
+ mask = image_text.font.getmask( # type: ignore[misc]
623
+ line,
624
+ mode,
625
+ direction,
626
+ features,
627
+ language,
628
+ stroke_width,
629
+ anchor,
630
+ ink,
631
+ start=start,
632
+ *args,
633
+ **kwargs,
634
+ )
635
+ except TypeError:
636
+ mask = image_text.font.getmask(line)
637
+ if mode == "RGBA":
638
+ # image_text.font.getmask2(mode="RGBA")
639
+ # returns color in RGB bands and mask in A
640
+ # extract mask and set text alpha
641
+ color, mask = mask, mask.getband(3)
642
+ ink_alpha = struct.pack("i", ink)[3]
643
+ color.fillband(3, ink_alpha)
644
+ x, y = coord
645
+ if self.im is not None:
646
+ self.im.paste(
647
+ color, (x, y, x + mask.size[0], y + mask.size[1]), mask
648
+ )
649
+ else:
650
+ self.draw.draw_bitmap(coord, mask, ink)
651
+
652
+ if stroke_ink is not None:
653
+ # Draw stroked text
654
+ draw_text(stroke_ink, image_text.stroke_width)
655
+
656
+ # Draw normal text
657
+ if ink != stroke_ink:
658
+ draw_text(ink)
659
+ else:
660
+ # Only draw normal text
661
+ draw_text(ink)
662
+
663
+ def multiline_text(
664
+ self,
665
+ xy: tuple[float, float],
666
+ text: AnyStr,
667
+ fill: _Ink | None = None,
668
+ font: (
669
+ ImageFont.ImageFont
670
+ | ImageFont.FreeTypeFont
671
+ | ImageFont.TransposedFont
672
+ | None
673
+ ) = None,
674
+ anchor: str | None = None,
675
+ spacing: float = 4,
676
+ align: str = "left",
677
+ direction: str | None = None,
678
+ features: list[str] | None = None,
679
+ language: str | None = None,
680
+ stroke_width: float = 0,
681
+ stroke_fill: _Ink | None = None,
682
+ embedded_color: bool = False,
683
+ *,
684
+ font_size: float | None = None,
685
+ ) -> None:
686
+ return self.text(
687
+ xy,
688
+ text,
689
+ fill,
690
+ font,
691
+ anchor,
692
+ spacing,
693
+ align,
694
+ direction,
695
+ features,
696
+ language,
697
+ stroke_width,
698
+ stroke_fill,
699
+ embedded_color,
700
+ font_size=font_size,
701
+ )
702
+
703
+ def textlength(
704
+ self,
705
+ text: AnyStr,
706
+ font: (
707
+ ImageFont.ImageFont
708
+ | ImageFont.FreeTypeFont
709
+ | ImageFont.TransposedFont
710
+ | None
711
+ ) = None,
712
+ direction: str | None = None,
713
+ features: list[str] | None = None,
714
+ language: str | None = None,
715
+ embedded_color: bool = False,
716
+ *,
717
+ font_size: float | None = None,
718
+ ) -> float:
719
+ """Get the length of a given string, in pixels with 1/64 precision."""
720
+ if font is None:
721
+ font = self._getfont(font_size)
722
+ image_text = ImageText.Text(
723
+ text,
724
+ font,
725
+ self.mode,
726
+ direction=direction,
727
+ features=features,
728
+ language=language,
729
+ )
730
+ if embedded_color:
731
+ image_text.embed_color()
732
+ return image_text.get_length()
733
+
734
+ def textbbox(
735
+ self,
736
+ xy: tuple[float, float],
737
+ text: AnyStr,
738
+ font: (
739
+ ImageFont.ImageFont
740
+ | ImageFont.FreeTypeFont
741
+ | ImageFont.TransposedFont
742
+ | None
743
+ ) = None,
744
+ anchor: str | None = None,
745
+ spacing: float = 4,
746
+ align: str = "left",
747
+ direction: str | None = None,
748
+ features: list[str] | None = None,
749
+ language: str | None = None,
750
+ stroke_width: float = 0,
751
+ embedded_color: bool = False,
752
+ *,
753
+ font_size: float | None = None,
754
+ ) -> tuple[float, float, float, float]:
755
+ """Get the bounding box of a given string, in pixels."""
756
+ if font is None:
757
+ font = self._getfont(font_size)
758
+ image_text = ImageText.Text(
759
+ text, font, self.mode, spacing, direction, features, language
760
+ )
761
+ if embedded_color:
762
+ image_text.embed_color()
763
+ if stroke_width:
764
+ image_text.stroke(stroke_width)
765
+ return image_text.get_bbox(xy, anchor, align)
766
+
767
+ def multiline_textbbox(
768
+ self,
769
+ xy: tuple[float, float],
770
+ text: AnyStr,
771
+ font: (
772
+ ImageFont.ImageFont
773
+ | ImageFont.FreeTypeFont
774
+ | ImageFont.TransposedFont
775
+ | None
776
+ ) = None,
777
+ anchor: str | None = None,
778
+ spacing: float = 4,
779
+ align: str = "left",
780
+ direction: str | None = None,
781
+ features: list[str] | None = None,
782
+ language: str | None = None,
783
+ stroke_width: float = 0,
784
+ embedded_color: bool = False,
785
+ *,
786
+ font_size: float | None = None,
787
+ ) -> tuple[float, float, float, float]:
788
+ return self.textbbox(
789
+ xy,
790
+ text,
791
+ font,
792
+ anchor,
793
+ spacing,
794
+ align,
795
+ direction,
796
+ features,
797
+ language,
798
+ stroke_width,
799
+ embedded_color,
800
+ font_size=font_size,
801
+ )
802
+
803
+
804
+ def Draw(im: Image.Image, mode: str | None = None) -> ImageDraw:
805
+ """
806
+ A simple 2D drawing interface for PIL images.
807
+
808
+ :param im: The image to draw in.
809
+ :param mode: Optional mode to use for color values. For RGB
810
+ images, this argument can be RGB or RGBA (to blend the
811
+ drawing into the image). For all other modes, this argument
812
+ must be the same as the image mode. If omitted, the mode
813
+ defaults to the mode of the image.
814
+ """
815
+ try:
816
+ return getattr(im, "getdraw")(mode)
817
+ except AttributeError:
818
+ return ImageDraw(im, mode)
819
+
820
+
821
+ def getdraw(im: Image.Image | None = None) -> tuple[ImageDraw2.Draw | None, ModuleType]:
822
+ """
823
+ :param im: The image to draw in.
824
+ :returns: A (drawing context, drawing resource factory) tuple.
825
+ """
826
+ from . import ImageDraw2
827
+
828
+ draw = ImageDraw2.Draw(im) if im is not None else None
829
+ return draw, ImageDraw2
830
+
831
+
832
+ def floodfill(
833
+ image: Image.Image,
834
+ xy: tuple[int, int],
835
+ value: float | tuple[int, ...],
836
+ border: float | tuple[int, ...] | None = None,
837
+ thresh: float = 0,
838
+ ) -> None:
839
+ """
840
+ .. warning:: This method is experimental.
841
+
842
+ Fills a bounded region with a given color.
843
+
844
+ :param image: Target image.
845
+ :param xy: Seed position (a 2-item coordinate tuple). See
846
+ :ref:`coordinate-system`.
847
+ :param value: Fill color.
848
+ :param border: Optional border value. If given, the region consists of
849
+ pixels with a color different from the border color. If not given,
850
+ the region consists of pixels having the same color as the seed
851
+ pixel.
852
+ :param thresh: Optional threshold value which specifies a maximum
853
+ tolerable difference of a pixel value from the 'background' in
854
+ order for it to be replaced. Useful for filling regions of
855
+ non-homogeneous, but similar, colors.
856
+ """
857
+ # based on an implementation by Eric S. Raymond
858
+ # amended by yo1995 @20180806
859
+ pixel = image.load()
860
+ assert pixel is not None
861
+ x, y = xy
862
+ try:
863
+ background = pixel[x, y]
864
+ if _color_diff(value, background) <= thresh:
865
+ return # seed point already has fill color
866
+ pixel[x, y] = value
867
+ except (ValueError, IndexError):
868
+ return # seed point outside image
869
+ edge = {(x, y)}
870
+ # use a set to keep record of current and previous edge pixels
871
+ # to reduce memory consumption
872
+ full_edge = set()
873
+ while edge:
874
+ new_edge = set()
875
+ for x, y in edge: # 4 adjacent method
876
+ for s, t in ((x + 1, y), (x - 1, y), (x, y + 1), (x, y - 1)):
877
+ # If already processed, or if a coordinate is negative, skip
878
+ if (s, t) in full_edge or s < 0 or t < 0:
879
+ continue
880
+ try:
881
+ p = pixel[s, t]
882
+ except (ValueError, IndexError):
883
+ pass
884
+ else:
885
+ full_edge.add((s, t))
886
+ if border is None:
887
+ fill = _color_diff(p, background) <= thresh
888
+ else:
889
+ fill = p not in (value, border)
890
+ if fill:
891
+ pixel[s, t] = value
892
+ new_edge.add((s, t))
893
+ full_edge = edge # discard pixels processed
894
+ edge = new_edge
895
+
896
+
897
+ def _compute_regular_polygon_vertices(
898
+ bounding_circle: Sequence[Sequence[float] | float], n_sides: int, rotation: float
899
+ ) -> list[tuple[float, float]]:
900
+ """
901
+ Generate a list of vertices for a 2D regular polygon.
902
+
903
+ :param bounding_circle: The bounding circle is a sequence defined
904
+ by a point and radius. The polygon is inscribed in this circle.
905
+ (e.g. ``bounding_circle=(x, y, r)`` or ``((x, y), r)``)
906
+ :param n_sides: Number of sides
907
+ (e.g. ``n_sides=3`` for a triangle, ``6`` for a hexagon)
908
+ :param rotation: Apply an arbitrary rotation to the polygon
909
+ (e.g. ``rotation=90``, applies a 90 degree rotation)
910
+ :return: List of regular polygon vertices
911
+ (e.g. ``[(25, 50), (50, 50), (50, 25), (25, 25)]``)
912
+
913
+ How are the vertices computed?
914
+ 1. Compute the following variables
915
+ - theta: Angle between the apothem & the nearest polygon vertex
916
+ - side_length: Length of each polygon edge
917
+ - centroid: Center of bounding circle (1st, 2nd elements of bounding_circle)
918
+ - polygon_radius: Polygon radius (last element of bounding_circle)
919
+ - angles: Location of each polygon vertex in polar grid
920
+ (e.g. A square with 0 degree rotation => [225.0, 315.0, 45.0, 135.0])
921
+
922
+ 2. For each angle in angles, get the polygon vertex at that angle
923
+ The vertex is computed using the equation below.
924
+ X= xcos(φ) + ysin(φ)
925
+ Y= −xsin(φ) + ycos(φ)
926
+
927
+ Note:
928
+ φ = angle in degrees
929
+ x = 0
930
+ y = polygon_radius
931
+
932
+ The formula above assumes rotation around the origin.
933
+ In our case, we are rotating around the centroid.
934
+ To account for this, we use the formula below
935
+ X = xcos(φ) + ysin(φ) + centroid_x
936
+ Y = −xsin(φ) + ycos(φ) + centroid_y
937
+ """
938
+ # 1. Error Handling
939
+ # 1.1 Check `n_sides` has an appropriate value
940
+ if not isinstance(n_sides, int):
941
+ msg = "n_sides should be an int" # type: ignore[unreachable]
942
+ raise TypeError(msg)
943
+ if n_sides < 3:
944
+ msg = "n_sides should be an int > 2"
945
+ raise ValueError(msg)
946
+
947
+ # 1.2 Check `bounding_circle` has an appropriate value
948
+ if not isinstance(bounding_circle, (list, tuple)):
949
+ msg = "bounding_circle should be a sequence"
950
+ raise TypeError(msg)
951
+
952
+ if len(bounding_circle) == 3:
953
+ if not all(isinstance(i, (int, float)) for i in bounding_circle):
954
+ msg = "bounding_circle should only contain numeric data"
955
+ raise ValueError(msg)
956
+
957
+ *centroid, polygon_radius = cast(list[float], list(bounding_circle))
958
+ elif len(bounding_circle) == 2 and isinstance(bounding_circle[0], (list, tuple)):
959
+ if not all(
960
+ isinstance(i, (int, float)) for i in bounding_circle[0]
961
+ ) or not isinstance(bounding_circle[1], (int, float)):
962
+ msg = "bounding_circle should only contain numeric data"
963
+ raise ValueError(msg)
964
+
965
+ if len(bounding_circle[0]) != 2:
966
+ msg = "bounding_circle centre should contain 2D coordinates (e.g. (x, y))"
967
+ raise ValueError(msg)
968
+
969
+ centroid = cast(list[float], list(bounding_circle[0]))
970
+ polygon_radius = cast(float, bounding_circle[1])
971
+ else:
972
+ msg = (
973
+ "bounding_circle should contain 2D coordinates "
974
+ "and a radius (e.g. (x, y, r) or ((x, y), r) )"
975
+ )
976
+ raise ValueError(msg)
977
+
978
+ if polygon_radius <= 0:
979
+ msg = "bounding_circle radius should be > 0"
980
+ raise ValueError(msg)
981
+
982
+ # 1.3 Check `rotation` has an appropriate value
983
+ if not isinstance(rotation, (int, float)):
984
+ msg = "rotation should be an int or float" # type: ignore[unreachable]
985
+ raise ValueError(msg)
986
+
987
+ # 2. Define Helper Functions
988
+ def _apply_rotation(point: list[float], degrees: float) -> tuple[float, float]:
989
+ return (
990
+ round(
991
+ point[0] * math.cos(math.radians(360 - degrees))
992
+ - point[1] * math.sin(math.radians(360 - degrees))
993
+ + centroid[0],
994
+ 2,
995
+ ),
996
+ round(
997
+ point[1] * math.cos(math.radians(360 - degrees))
998
+ + point[0] * math.sin(math.radians(360 - degrees))
999
+ + centroid[1],
1000
+ 2,
1001
+ ),
1002
+ )
1003
+
1004
+ def _compute_polygon_vertex(angle: float) -> tuple[float, float]:
1005
+ start_point = [polygon_radius, 0]
1006
+ return _apply_rotation(start_point, angle)
1007
+
1008
+ def _get_angles(n_sides: int, rotation: float) -> list[float]:
1009
+ angles = []
1010
+ degrees = 360 / n_sides
1011
+ # Start with the bottom left polygon vertex
1012
+ current_angle = (270 - 0.5 * degrees) + rotation
1013
+ for _ in range(n_sides):
1014
+ angles.append(current_angle)
1015
+ current_angle += degrees
1016
+ if current_angle > 360:
1017
+ current_angle -= 360
1018
+ return angles
1019
+
1020
+ # 3. Variable Declarations
1021
+ angles = _get_angles(n_sides, rotation)
1022
+
1023
+ # 4. Compute Vertices
1024
+ return [_compute_polygon_vertex(angle) for angle in angles]
1025
+
1026
+
1027
+ def _color_diff(
1028
+ color1: float | tuple[int, ...], color2: float | tuple[int, ...]
1029
+ ) -> float:
1030
+ """
1031
+ Uses 1-norm distance to calculate difference between two values.
1032
+ """
1033
+ first = color1 if isinstance(color1, tuple) else (color1,)
1034
+ second = color2 if isinstance(color2, tuple) else (color2,)
1035
+
1036
+ return sum(abs(first[i] - second[i]) for i in range(len(second)))
.venv/Lib/site-packages/PIL/ImageDraw2.py ADDED
@@ -0,0 +1,243 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #
2
+ # The Python Imaging Library
3
+ # $Id$
4
+ #
5
+ # WCK-style drawing interface operations
6
+ #
7
+ # History:
8
+ # 2003-12-07 fl created
9
+ # 2005-05-15 fl updated; added to PIL as ImageDraw2
10
+ # 2005-05-15 fl added text support
11
+ # 2005-05-20 fl added arc/chord/pieslice support
12
+ #
13
+ # Copyright (c) 2003-2005 by Secret Labs AB
14
+ # Copyright (c) 2003-2005 by Fredrik Lundh
15
+ #
16
+ # See the README file for information on usage and redistribution.
17
+ #
18
+
19
+
20
+ """
21
+ (Experimental) WCK-style drawing interface operations
22
+
23
+ .. seealso:: :py:mod:`PIL.ImageDraw`
24
+ """
25
+ from __future__ import annotations
26
+
27
+ from typing import Any, AnyStr, BinaryIO
28
+
29
+ from . import Image, ImageColor, ImageDraw, ImageFont, ImagePath
30
+ from ._typing import Coords, StrOrBytesPath
31
+
32
+
33
+ class Pen:
34
+ """Stores an outline color and width."""
35
+
36
+ def __init__(self, color: str, width: int = 1, opacity: int = 255) -> None:
37
+ self.color = ImageColor.getrgb(color)
38
+ self.width = width
39
+
40
+
41
+ class Brush:
42
+ """Stores a fill color"""
43
+
44
+ def __init__(self, color: str, opacity: int = 255) -> None:
45
+ self.color = ImageColor.getrgb(color)
46
+
47
+
48
+ class Font:
49
+ """Stores a TrueType font and color"""
50
+
51
+ def __init__(
52
+ self, color: str, file: StrOrBytesPath | BinaryIO, size: float = 12
53
+ ) -> None:
54
+ # FIXME: add support for bitmap fonts
55
+ self.color = ImageColor.getrgb(color)
56
+ self.font = ImageFont.truetype(file, size)
57
+
58
+
59
+ class Draw:
60
+ """
61
+ (Experimental) WCK-style drawing interface
62
+ """
63
+
64
+ def __init__(
65
+ self,
66
+ image: Image.Image | str,
67
+ size: tuple[int, int] | list[int] | None = None,
68
+ color: float | tuple[float, ...] | str | None = None,
69
+ ) -> None:
70
+ if isinstance(image, str):
71
+ if size is None:
72
+ msg = "If image argument is mode string, size must be a list or tuple"
73
+ raise ValueError(msg)
74
+ image = Image.new(image, size, color)
75
+ self.draw = ImageDraw.Draw(image)
76
+ self.image = image
77
+ self.transform: tuple[float, float, float, float, float, float] | None = None
78
+
79
+ def flush(self) -> Image.Image:
80
+ return self.image
81
+
82
+ def render(
83
+ self,
84
+ op: str,
85
+ xy: Coords,
86
+ pen: Pen | Brush | None,
87
+ brush: Brush | Pen | None = None,
88
+ **kwargs: Any,
89
+ ) -> None:
90
+ # handle color arguments
91
+ outline = fill = None
92
+ width = 1
93
+ if isinstance(pen, Pen):
94
+ outline = pen.color
95
+ width = pen.width
96
+ elif isinstance(brush, Pen):
97
+ outline = brush.color
98
+ width = brush.width
99
+ if isinstance(brush, Brush):
100
+ fill = brush.color
101
+ elif isinstance(pen, Brush):
102
+ fill = pen.color
103
+ # handle transformation
104
+ if self.transform:
105
+ path = ImagePath.Path(xy)
106
+ path.transform(self.transform)
107
+ xy = path
108
+ # render the item
109
+ if op in ("arc", "line"):
110
+ kwargs.setdefault("fill", outline)
111
+ else:
112
+ kwargs.setdefault("fill", fill)
113
+ kwargs.setdefault("outline", outline)
114
+ if op == "line":
115
+ kwargs.setdefault("width", width)
116
+ getattr(self.draw, op)(xy, **kwargs)
117
+
118
+ def settransform(self, offset: tuple[float, float]) -> None:
119
+ """Sets a transformation offset."""
120
+ (xoffset, yoffset) = offset
121
+ self.transform = (1, 0, xoffset, 0, 1, yoffset)
122
+
123
+ def arc(
124
+ self,
125
+ xy: Coords,
126
+ pen: Pen | Brush | None,
127
+ start: float,
128
+ end: float,
129
+ *options: Any,
130
+ ) -> None:
131
+ """
132
+ Draws an arc (a portion of a circle outline) between the start and end
133
+ angles, inside the given bounding box.
134
+
135
+ .. seealso:: :py:meth:`PIL.ImageDraw.ImageDraw.arc`
136
+ """
137
+ self.render("arc", xy, pen, *options, start=start, end=end)
138
+
139
+ def chord(
140
+ self,
141
+ xy: Coords,
142
+ pen: Pen | Brush | None,
143
+ start: float,
144
+ end: float,
145
+ *options: Any,
146
+ ) -> None:
147
+ """
148
+ Same as :py:meth:`~PIL.ImageDraw2.Draw.arc`, but connects the end points
149
+ with a straight line.
150
+
151
+ .. seealso:: :py:meth:`PIL.ImageDraw.ImageDraw.chord`
152
+ """
153
+ self.render("chord", xy, pen, *options, start=start, end=end)
154
+
155
+ def ellipse(self, xy: Coords, pen: Pen | Brush | None, *options: Any) -> None:
156
+ """
157
+ Draws an ellipse inside the given bounding box.
158
+
159
+ .. seealso:: :py:meth:`PIL.ImageDraw.ImageDraw.ellipse`
160
+ """
161
+ self.render("ellipse", xy, pen, *options)
162
+
163
+ def line(self, xy: Coords, pen: Pen | Brush | None, *options: Any) -> None:
164
+ """
165
+ Draws a line between the coordinates in the ``xy`` list.
166
+
167
+ .. seealso:: :py:meth:`PIL.ImageDraw.ImageDraw.line`
168
+ """
169
+ self.render("line", xy, pen, *options)
170
+
171
+ def pieslice(
172
+ self,
173
+ xy: Coords,
174
+ pen: Pen | Brush | None,
175
+ start: float,
176
+ end: float,
177
+ *options: Any,
178
+ ) -> None:
179
+ """
180
+ Same as arc, but also draws straight lines between the end points and the
181
+ center of the bounding box.
182
+
183
+ .. seealso:: :py:meth:`PIL.ImageDraw.ImageDraw.pieslice`
184
+ """
185
+ self.render("pieslice", xy, pen, *options, start=start, end=end)
186
+
187
+ def polygon(self, xy: Coords, pen: Pen | Brush | None, *options: Any) -> None:
188
+ """
189
+ Draws a polygon.
190
+
191
+ The polygon outline consists of straight lines between the given
192
+ coordinates, plus a straight line between the last and the first
193
+ coordinate.
194
+
195
+
196
+ .. seealso:: :py:meth:`PIL.ImageDraw.ImageDraw.polygon`
197
+ """
198
+ self.render("polygon", xy, pen, *options)
199
+
200
+ def rectangle(self, xy: Coords, pen: Pen | Brush | None, *options: Any) -> None:
201
+ """
202
+ Draws a rectangle.
203
+
204
+ .. seealso:: :py:meth:`PIL.ImageDraw.ImageDraw.rectangle`
205
+ """
206
+ self.render("rectangle", xy, pen, *options)
207
+
208
+ def text(self, xy: tuple[float, float], text: AnyStr, font: Font) -> None:
209
+ """
210
+ Draws the string at the given position.
211
+
212
+ .. seealso:: :py:meth:`PIL.ImageDraw.ImageDraw.text`
213
+ """
214
+ if self.transform:
215
+ path = ImagePath.Path(xy)
216
+ path.transform(self.transform)
217
+ xy = path
218
+ self.draw.text(xy, text, font=font.font, fill=font.color)
219
+
220
+ def textbbox(
221
+ self, xy: tuple[float, float], text: AnyStr, font: Font
222
+ ) -> tuple[float, float, float, float]:
223
+ """
224
+ Returns bounding box (in pixels) of given text.
225
+
226
+ :return: ``(left, top, right, bottom)`` bounding box
227
+
228
+ .. seealso:: :py:meth:`PIL.ImageDraw.ImageDraw.textbbox`
229
+ """
230
+ if self.transform:
231
+ path = ImagePath.Path(xy)
232
+ path.transform(self.transform)
233
+ xy = path
234
+ return self.draw.textbbox(xy, text, font=font.font)
235
+
236
+ def textlength(self, text: AnyStr, font: Font) -> float:
237
+ """
238
+ Returns length (in pixels) of given text.
239
+ This is the amount by which following text should be offset.
240
+
241
+ .. seealso:: :py:meth:`PIL.ImageDraw.ImageDraw.textlength`
242
+ """
243
+ return self.draw.textlength(text, font=font.font)
.venv/Lib/site-packages/PIL/ImageEnhance.py ADDED
@@ -0,0 +1,113 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #
2
+ # The Python Imaging Library.
3
+ # $Id$
4
+ #
5
+ # image enhancement classes
6
+ #
7
+ # For a background, see "Image Processing By Interpolation and
8
+ # Extrapolation", Paul Haeberli and Douglas Voorhies. Available
9
+ # at http://www.graficaobscura.com/interp/index.html
10
+ #
11
+ # History:
12
+ # 1996-03-23 fl Created
13
+ # 2009-06-16 fl Fixed mean calculation
14
+ #
15
+ # Copyright (c) Secret Labs AB 1997.
16
+ # Copyright (c) Fredrik Lundh 1996.
17
+ #
18
+ # See the README file for information on usage and redistribution.
19
+ #
20
+ from __future__ import annotations
21
+
22
+ from . import Image, ImageFilter, ImageStat
23
+
24
+
25
+ class _Enhance:
26
+ image: Image.Image
27
+ degenerate: Image.Image
28
+
29
+ def enhance(self, factor: float) -> Image.Image:
30
+ """
31
+ Returns an enhanced image.
32
+
33
+ :param factor: A floating point value controlling the enhancement.
34
+ Factor 1.0 always returns a copy of the original image,
35
+ lower factors mean less color (brightness, contrast,
36
+ etc), and higher values more. There are no restrictions
37
+ on this value.
38
+ :rtype: :py:class:`~PIL.Image.Image`
39
+ """
40
+ return Image.blend(self.degenerate, self.image, factor)
41
+
42
+
43
+ class Color(_Enhance):
44
+ """Adjust image color balance.
45
+
46
+ This class can be used to adjust the colour balance of an image, in
47
+ a manner similar to the controls on a colour TV set. An enhancement
48
+ factor of 0.0 gives a black and white image. A factor of 1.0 gives
49
+ the original image.
50
+ """
51
+
52
+ def __init__(self, image: Image.Image) -> None:
53
+ self.image = image
54
+ self.intermediate_mode = "L"
55
+ if "A" in image.getbands():
56
+ self.intermediate_mode = "LA"
57
+
58
+ if self.intermediate_mode != image.mode:
59
+ image = image.convert(self.intermediate_mode).convert(image.mode)
60
+ self.degenerate = image
61
+
62
+
63
+ class Contrast(_Enhance):
64
+ """Adjust image contrast.
65
+
66
+ This class can be used to control the contrast of an image, similar
67
+ to the contrast control on a TV set. An enhancement factor of 0.0
68
+ gives a solid gray image. A factor of 1.0 gives the original image.
69
+ """
70
+
71
+ def __init__(self, image: Image.Image) -> None:
72
+ self.image = image
73
+ if image.mode != "L":
74
+ image = image.convert("L")
75
+ mean = int(ImageStat.Stat(image).mean[0] + 0.5)
76
+ self.degenerate = Image.new("L", image.size, mean)
77
+ if self.degenerate.mode != self.image.mode:
78
+ self.degenerate = self.degenerate.convert(self.image.mode)
79
+
80
+ if "A" in self.image.getbands():
81
+ self.degenerate.putalpha(self.image.getchannel("A"))
82
+
83
+
84
+ class Brightness(_Enhance):
85
+ """Adjust image brightness.
86
+
87
+ This class can be used to control the brightness of an image. An
88
+ enhancement factor of 0.0 gives a black image. A factor of 1.0 gives the
89
+ original image.
90
+ """
91
+
92
+ def __init__(self, image: Image.Image) -> None:
93
+ self.image = image
94
+ self.degenerate = Image.new(image.mode, image.size, 0)
95
+
96
+ if "A" in image.getbands():
97
+ self.degenerate.putalpha(image.getchannel("A"))
98
+
99
+
100
+ class Sharpness(_Enhance):
101
+ """Adjust image sharpness.
102
+
103
+ This class can be used to adjust the sharpness of an image. An
104
+ enhancement factor of 0.0 gives a blurred image, a factor of 1.0 gives the
105
+ original image, and a factor of 2.0 gives a sharpened image.
106
+ """
107
+
108
+ def __init__(self, image: Image.Image) -> None:
109
+ self.image = image
110
+ self.degenerate = image.filter(ImageFilter.SMOOTH)
111
+
112
+ if "A" in image.getbands():
113
+ self.degenerate.putalpha(image.getchannel("A"))
.venv/Lib/site-packages/PIL/ImageFile.py ADDED
@@ -0,0 +1,926 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #
2
+ # The Python Imaging Library.
3
+ # $Id$
4
+ #
5
+ # base class for image file handlers
6
+ #
7
+ # history:
8
+ # 1995-09-09 fl Created
9
+ # 1996-03-11 fl Fixed load mechanism.
10
+ # 1996-04-15 fl Added pcx/xbm decoders.
11
+ # 1996-04-30 fl Added encoders.
12
+ # 1996-12-14 fl Added load helpers
13
+ # 1997-01-11 fl Use encode_to_file where possible
14
+ # 1997-08-27 fl Flush output in _save
15
+ # 1998-03-05 fl Use memory mapping for some modes
16
+ # 1999-02-04 fl Use memory mapping also for "I;16" and "I;16B"
17
+ # 1999-05-31 fl Added image parser
18
+ # 2000-10-12 fl Set readonly flag on memory-mapped images
19
+ # 2002-03-20 fl Use better messages for common decoder errors
20
+ # 2003-04-21 fl Fall back on mmap/map_buffer if map is not available
21
+ # 2003-10-30 fl Added StubImageFile class
22
+ # 2004-02-25 fl Made incremental parser more robust
23
+ #
24
+ # Copyright (c) 1997-2004 by Secret Labs AB
25
+ # Copyright (c) 1995-2004 by Fredrik Lundh
26
+ #
27
+ # See the README file for information on usage and redistribution.
28
+ #
29
+ from __future__ import annotations
30
+
31
+ import abc
32
+ import io
33
+ import itertools
34
+ import logging
35
+ import os
36
+ import struct
37
+ from typing import IO, Any, NamedTuple, cast
38
+
39
+ from . import ExifTags, Image
40
+ from ._util import DeferredError, is_path
41
+
42
+ TYPE_CHECKING = False
43
+ if TYPE_CHECKING:
44
+ from ._typing import StrOrBytesPath
45
+
46
+ logger = logging.getLogger(__name__)
47
+
48
+ MAXBLOCK = 65536
49
+ """
50
+ By default, Pillow processes image data in blocks. This helps to prevent excessive use
51
+ of resources. Codecs may disable this behaviour with ``_pulls_fd`` or ``_pushes_fd``.
52
+
53
+ When reading an image, this is the number of bytes to read at once.
54
+
55
+ When writing an image, this is the number of bytes to write at once.
56
+ If the image width times 4 is greater, then that will be used instead.
57
+ Plugins may also set a greater number.
58
+
59
+ User code may set this to another number.
60
+ """
61
+
62
+ SAFEBLOCK = 1024 * 1024
63
+
64
+ LOAD_TRUNCATED_IMAGES = False
65
+ """Whether or not to load truncated image files. User code may change this."""
66
+
67
+ ERRORS = {
68
+ -1: "image buffer overrun error",
69
+ -2: "decoding error",
70
+ -3: "unknown error",
71
+ -8: "bad configuration",
72
+ -9: "out of memory error",
73
+ }
74
+ """
75
+ Dict of known error codes returned from :meth:`.PyDecoder.decode`,
76
+ :meth:`.PyEncoder.encode` :meth:`.PyEncoder.encode_to_pyfd` and
77
+ :meth:`.PyEncoder.encode_to_file`.
78
+ """
79
+
80
+
81
+ #
82
+ # --------------------------------------------------------------------
83
+ # Helpers
84
+
85
+
86
+ def _get_oserror(error: int, *, encoder: bool) -> OSError:
87
+ try:
88
+ msg = Image.core.getcodecstatus(error)
89
+ except AttributeError:
90
+ msg = ERRORS.get(error)
91
+ if not msg:
92
+ msg = f"{'encoder' if encoder else 'decoder'} error {error}"
93
+ msg += f" when {'writing' if encoder else 'reading'} image file"
94
+ return OSError(msg)
95
+
96
+
97
+ def _tilesort(t: _Tile) -> int:
98
+ # sort on offset
99
+ return t[2]
100
+
101
+
102
+ class _Tile(NamedTuple):
103
+ codec_name: str
104
+ extents: tuple[int, int, int, int] | None
105
+ offset: int = 0
106
+ args: tuple[Any, ...] | str | None = None
107
+
108
+
109
+ #
110
+ # --------------------------------------------------------------------
111
+ # ImageFile base class
112
+
113
+
114
+ class ImageFile(Image.Image):
115
+ """Base class for image file format handlers."""
116
+
117
+ def __init__(
118
+ self, fp: StrOrBytesPath | IO[bytes], filename: str | bytes | None = None
119
+ ) -> None:
120
+ super().__init__()
121
+
122
+ self._min_frame = 0
123
+
124
+ self.custom_mimetype: str | None = None
125
+
126
+ self.tile: list[_Tile] = []
127
+ """ A list of tile descriptors """
128
+
129
+ self.readonly = 1 # until we know better
130
+
131
+ self.decoderconfig: tuple[Any, ...] = ()
132
+ self.decodermaxblock = MAXBLOCK
133
+
134
+ if is_path(fp):
135
+ # filename
136
+ self.fp = open(fp, "rb")
137
+ self.filename = os.fspath(fp)
138
+ self._exclusive_fp = True
139
+ else:
140
+ # stream
141
+ self.fp = cast(IO[bytes], fp)
142
+ self.filename = filename if filename is not None else ""
143
+ # can be overridden
144
+ self._exclusive_fp = False
145
+
146
+ try:
147
+ try:
148
+ self._open()
149
+ except (
150
+ IndexError, # end of data
151
+ TypeError, # end of data (ord)
152
+ KeyError, # unsupported mode
153
+ EOFError, # got header but not the first frame
154
+ struct.error,
155
+ ) as v:
156
+ raise SyntaxError(v) from v
157
+
158
+ if not self.mode or self.size[0] <= 0 or self.size[1] <= 0:
159
+ msg = "not identified by this driver"
160
+ raise SyntaxError(msg)
161
+ except BaseException:
162
+ # close the file only if we have opened it this constructor
163
+ if self._exclusive_fp:
164
+ self.fp.close()
165
+ raise
166
+
167
+ def _open(self) -> None:
168
+ pass
169
+
170
+ def _close_fp(self):
171
+ if getattr(self, "_fp", False) and not isinstance(self._fp, DeferredError):
172
+ if self._fp != self.fp:
173
+ self._fp.close()
174
+ self._fp = DeferredError(ValueError("Operation on closed image"))
175
+ if self.fp:
176
+ self.fp.close()
177
+
178
+ def close(self) -> None:
179
+ """
180
+ Closes the file pointer, if possible.
181
+
182
+ This operation will destroy the image core and release its memory.
183
+ The image data will be unusable afterward.
184
+
185
+ This function is required to close images that have multiple frames or
186
+ have not had their file read and closed by the
187
+ :py:meth:`~PIL.Image.Image.load` method. See :ref:`file-handling` for
188
+ more information.
189
+ """
190
+ try:
191
+ self._close_fp()
192
+ self.fp = None
193
+ except Exception as msg:
194
+ logger.debug("Error closing: %s", msg)
195
+
196
+ super().close()
197
+
198
+ def get_child_images(self) -> list[ImageFile]:
199
+ child_images = []
200
+ exif = self.getexif()
201
+ ifds = []
202
+ if ExifTags.Base.SubIFDs in exif:
203
+ subifd_offsets = exif[ExifTags.Base.SubIFDs]
204
+ if subifd_offsets:
205
+ if not isinstance(subifd_offsets, tuple):
206
+ subifd_offsets = (subifd_offsets,)
207
+ for subifd_offset in subifd_offsets:
208
+ ifds.append((exif._get_ifd_dict(subifd_offset), subifd_offset))
209
+ ifd1 = exif.get_ifd(ExifTags.IFD.IFD1)
210
+ if ifd1 and ifd1.get(ExifTags.Base.JpegIFOffset):
211
+ assert exif._info is not None
212
+ ifds.append((ifd1, exif._info.next))
213
+
214
+ offset = None
215
+ for ifd, ifd_offset in ifds:
216
+ assert self.fp is not None
217
+ current_offset = self.fp.tell()
218
+ if offset is None:
219
+ offset = current_offset
220
+
221
+ fp = self.fp
222
+ if ifd is not None:
223
+ thumbnail_offset = ifd.get(ExifTags.Base.JpegIFOffset)
224
+ if thumbnail_offset is not None:
225
+ thumbnail_offset += getattr(self, "_exif_offset", 0)
226
+ self.fp.seek(thumbnail_offset)
227
+
228
+ length = ifd.get(ExifTags.Base.JpegIFByteCount)
229
+ assert isinstance(length, int)
230
+ data = self.fp.read(length)
231
+ fp = io.BytesIO(data)
232
+
233
+ with Image.open(fp) as im:
234
+ from . import TiffImagePlugin
235
+
236
+ if thumbnail_offset is None and isinstance(
237
+ im, TiffImagePlugin.TiffImageFile
238
+ ):
239
+ im._frame_pos = [ifd_offset]
240
+ im._seek(0)
241
+ im.load()
242
+ child_images.append(im)
243
+
244
+ if offset is not None:
245
+ assert self.fp is not None
246
+ self.fp.seek(offset)
247
+ return child_images
248
+
249
+ def get_format_mimetype(self) -> str | None:
250
+ if self.custom_mimetype:
251
+ return self.custom_mimetype
252
+ if self.format is not None:
253
+ return Image.MIME.get(self.format.upper())
254
+ return None
255
+
256
+ def __getstate__(self) -> list[Any]:
257
+ return super().__getstate__() + [self.filename]
258
+
259
+ def __setstate__(self, state: list[Any]) -> None:
260
+ self.tile = []
261
+ if len(state) > 5:
262
+ self.filename = state[5]
263
+ super().__setstate__(state)
264
+
265
+ def verify(self) -> None:
266
+ """Check file integrity"""
267
+
268
+ # raise exception if something's wrong. must be called
269
+ # directly after open, and closes file when finished.
270
+ if self._exclusive_fp:
271
+ self.fp.close()
272
+ self.fp = None
273
+
274
+ def load(self) -> Image.core.PixelAccess | None:
275
+ """Load image data based on tile list"""
276
+
277
+ if not self.tile and self._im is None:
278
+ msg = "cannot load this image"
279
+ raise OSError(msg)
280
+
281
+ pixel = Image.Image.load(self)
282
+ if not self.tile:
283
+ return pixel
284
+
285
+ self.map: mmap.mmap | None = None
286
+ use_mmap = self.filename and len(self.tile) == 1
287
+
288
+ readonly = 0
289
+
290
+ # look for read/seek overrides
291
+ if hasattr(self, "load_read"):
292
+ read = self.load_read
293
+ # don't use mmap if there are custom read/seek functions
294
+ use_mmap = False
295
+ else:
296
+ read = self.fp.read
297
+
298
+ if hasattr(self, "load_seek"):
299
+ seek = self.load_seek
300
+ use_mmap = False
301
+ else:
302
+ seek = self.fp.seek
303
+
304
+ if use_mmap:
305
+ # try memory mapping
306
+ decoder_name, extents, offset, args = self.tile[0]
307
+ if isinstance(args, str):
308
+ args = (args, 0, 1)
309
+ if (
310
+ decoder_name == "raw"
311
+ and isinstance(args, tuple)
312
+ and len(args) >= 3
313
+ and args[0] == self.mode
314
+ and args[0] in Image._MAPMODES
315
+ ):
316
+ if offset < 0:
317
+ msg = "Tile offset cannot be negative"
318
+ raise ValueError(msg)
319
+ try:
320
+ # use mmap, if possible
321
+ import mmap
322
+
323
+ with open(self.filename) as fp:
324
+ self.map = mmap.mmap(fp.fileno(), 0, access=mmap.ACCESS_READ)
325
+ if offset + self.size[1] * args[1] > self.map.size():
326
+ msg = "buffer is not large enough"
327
+ raise OSError(msg)
328
+ self.im = Image.core.map_buffer(
329
+ self.map, self.size, decoder_name, offset, args
330
+ )
331
+ readonly = 1
332
+ # After trashing self.im,
333
+ # we might need to reload the palette data.
334
+ if self.palette:
335
+ self.palette.dirty = 1
336
+ except (AttributeError, OSError, ImportError):
337
+ self.map = None
338
+
339
+ self.load_prepare()
340
+ err_code = -3 # initialize to unknown error
341
+ if not self.map:
342
+ # sort tiles in file order
343
+ self.tile.sort(key=_tilesort)
344
+
345
+ # FIXME: This is a hack to handle TIFF's JpegTables tag.
346
+ prefix = getattr(self, "tile_prefix", b"")
347
+
348
+ # Remove consecutive duplicates that only differ by their offset
349
+ self.tile = [
350
+ list(tiles)[-1]
351
+ for _, tiles in itertools.groupby(
352
+ self.tile, lambda tile: (tile[0], tile[1], tile[3])
353
+ )
354
+ ]
355
+ for i, (decoder_name, extents, offset, args) in enumerate(self.tile):
356
+ seek(offset)
357
+ decoder = Image._getdecoder(
358
+ self.mode, decoder_name, args, self.decoderconfig
359
+ )
360
+ try:
361
+ decoder.setimage(self.im, extents)
362
+ if decoder.pulls_fd:
363
+ decoder.setfd(self.fp)
364
+ err_code = decoder.decode(b"")[1]
365
+ else:
366
+ b = prefix
367
+ while True:
368
+ read_bytes = self.decodermaxblock
369
+ if i + 1 < len(self.tile):
370
+ next_offset = self.tile[i + 1].offset
371
+ if next_offset > offset:
372
+ read_bytes = next_offset - offset
373
+ try:
374
+ s = read(read_bytes)
375
+ except (IndexError, struct.error) as e:
376
+ # truncated png/gif
377
+ if LOAD_TRUNCATED_IMAGES:
378
+ break
379
+ else:
380
+ msg = "image file is truncated"
381
+ raise OSError(msg) from e
382
+
383
+ if not s: # truncated jpeg
384
+ if LOAD_TRUNCATED_IMAGES:
385
+ break
386
+ else:
387
+ msg = (
388
+ "image file is truncated "
389
+ f"({len(b)} bytes not processed)"
390
+ )
391
+ raise OSError(msg)
392
+
393
+ b = b + s
394
+ n, err_code = decoder.decode(b)
395
+ if n < 0:
396
+ break
397
+ b = b[n:]
398
+ finally:
399
+ # Need to cleanup here to prevent leaks
400
+ decoder.cleanup()
401
+
402
+ self.tile = []
403
+ self.readonly = readonly
404
+
405
+ self.load_end()
406
+
407
+ if self._exclusive_fp and self._close_exclusive_fp_after_loading:
408
+ self.fp.close()
409
+ self.fp = None
410
+
411
+ if not self.map and not LOAD_TRUNCATED_IMAGES and err_code < 0:
412
+ # still raised if decoder fails to return anything
413
+ raise _get_oserror(err_code, encoder=False)
414
+
415
+ return Image.Image.load(self)
416
+
417
+ def load_prepare(self) -> None:
418
+ # create image memory if necessary
419
+ if self._im is None:
420
+ self.im = Image.core.new(self.mode, self.size)
421
+ # create palette (optional)
422
+ if self.mode == "P":
423
+ Image.Image.load(self)
424
+
425
+ def load_end(self) -> None:
426
+ # may be overridden
427
+ pass
428
+
429
+ # may be defined for contained formats
430
+ # def load_seek(self, pos: int) -> None:
431
+ # pass
432
+
433
+ # may be defined for blocked formats (e.g. PNG)
434
+ # def load_read(self, read_bytes: int) -> bytes:
435
+ # pass
436
+
437
+ def _seek_check(self, frame: int) -> bool:
438
+ if (
439
+ frame < self._min_frame
440
+ # Only check upper limit on frames if additional seek operations
441
+ # are not required to do so
442
+ or (
443
+ not (hasattr(self, "_n_frames") and self._n_frames is None)
444
+ and frame >= getattr(self, "n_frames") + self._min_frame
445
+ )
446
+ ):
447
+ msg = "attempt to seek outside sequence"
448
+ raise EOFError(msg)
449
+
450
+ return self.tell() != frame
451
+
452
+
453
+ class StubHandler(abc.ABC):
454
+ def open(self, im: StubImageFile) -> None:
455
+ pass
456
+
457
+ @abc.abstractmethod
458
+ def load(self, im: StubImageFile) -> Image.Image:
459
+ pass
460
+
461
+
462
+ class StubImageFile(ImageFile, metaclass=abc.ABCMeta):
463
+ """
464
+ Base class for stub image loaders.
465
+
466
+ A stub loader is an image loader that can identify files of a
467
+ certain format, but relies on external code to load the file.
468
+ """
469
+
470
+ @abc.abstractmethod
471
+ def _open(self) -> None:
472
+ pass
473
+
474
+ def load(self) -> Image.core.PixelAccess | None:
475
+ loader = self._load()
476
+ if loader is None:
477
+ msg = f"cannot find loader for this {self.format} file"
478
+ raise OSError(msg)
479
+ image = loader.load(self)
480
+ assert image is not None
481
+ # become the other object (!)
482
+ self.__class__ = image.__class__ # type: ignore[assignment]
483
+ self.__dict__ = image.__dict__
484
+ return image.load()
485
+
486
+ @abc.abstractmethod
487
+ def _load(self) -> StubHandler | None:
488
+ """(Hook) Find actual image loader."""
489
+ pass
490
+
491
+
492
+ class Parser:
493
+ """
494
+ Incremental image parser. This class implements the standard
495
+ feed/close consumer interface.
496
+ """
497
+
498
+ incremental = None
499
+ image: Image.Image | None = None
500
+ data: bytes | None = None
501
+ decoder: Image.core.ImagingDecoder | PyDecoder | None = None
502
+ offset = 0
503
+ finished = 0
504
+
505
+ def reset(self) -> None:
506
+ """
507
+ (Consumer) Reset the parser. Note that you can only call this
508
+ method immediately after you've created a parser; parser
509
+ instances cannot be reused.
510
+ """
511
+ assert self.data is None, "cannot reuse parsers"
512
+
513
+ def feed(self, data: bytes) -> None:
514
+ """
515
+ (Consumer) Feed data to the parser.
516
+
517
+ :param data: A string buffer.
518
+ :exception OSError: If the parser failed to parse the image file.
519
+ """
520
+ # collect data
521
+
522
+ if self.finished:
523
+ return
524
+
525
+ if self.data is None:
526
+ self.data = data
527
+ else:
528
+ self.data = self.data + data
529
+
530
+ # parse what we have
531
+ if self.decoder:
532
+ if self.offset > 0:
533
+ # skip header
534
+ skip = min(len(self.data), self.offset)
535
+ self.data = self.data[skip:]
536
+ self.offset = self.offset - skip
537
+ if self.offset > 0 or not self.data:
538
+ return
539
+
540
+ n, e = self.decoder.decode(self.data)
541
+
542
+ if n < 0:
543
+ # end of stream
544
+ self.data = None
545
+ self.finished = 1
546
+ if e < 0:
547
+ # decoding error
548
+ self.image = None
549
+ raise _get_oserror(e, encoder=False)
550
+ else:
551
+ # end of image
552
+ return
553
+ self.data = self.data[n:]
554
+
555
+ elif self.image:
556
+ # if we end up here with no decoder, this file cannot
557
+ # be incrementally parsed. wait until we've gotten all
558
+ # available data
559
+ pass
560
+
561
+ else:
562
+ # attempt to open this file
563
+ try:
564
+ with io.BytesIO(self.data) as fp:
565
+ im = Image.open(fp)
566
+ except OSError:
567
+ pass # not enough data
568
+ else:
569
+ flag = hasattr(im, "load_seek") or hasattr(im, "load_read")
570
+ if flag or len(im.tile) != 1:
571
+ # custom load code, or multiple tiles
572
+ self.decode = None
573
+ else:
574
+ # initialize decoder
575
+ im.load_prepare()
576
+ d, e, o, a = im.tile[0]
577
+ im.tile = []
578
+ self.decoder = Image._getdecoder(im.mode, d, a, im.decoderconfig)
579
+ self.decoder.setimage(im.im, e)
580
+
581
+ # calculate decoder offset
582
+ self.offset = o
583
+ if self.offset <= len(self.data):
584
+ self.data = self.data[self.offset :]
585
+ self.offset = 0
586
+
587
+ self.image = im
588
+
589
+ def __enter__(self) -> Parser:
590
+ return self
591
+
592
+ def __exit__(self, *args: object) -> None:
593
+ self.close()
594
+
595
+ def close(self) -> Image.Image:
596
+ """
597
+ (Consumer) Close the stream.
598
+
599
+ :returns: An image object.
600
+ :exception OSError: If the parser failed to parse the image file either
601
+ because it cannot be identified or cannot be
602
+ decoded.
603
+ """
604
+ # finish decoding
605
+ if self.decoder:
606
+ # get rid of what's left in the buffers
607
+ self.feed(b"")
608
+ self.data = self.decoder = None
609
+ if not self.finished:
610
+ msg = "image was incomplete"
611
+ raise OSError(msg)
612
+ if not self.image:
613
+ msg = "cannot parse this image"
614
+ raise OSError(msg)
615
+ if self.data:
616
+ # incremental parsing not possible; reopen the file
617
+ # not that we have all data
618
+ with io.BytesIO(self.data) as fp:
619
+ try:
620
+ self.image = Image.open(fp)
621
+ finally:
622
+ self.image.load()
623
+ return self.image
624
+
625
+
626
+ # --------------------------------------------------------------------
627
+
628
+
629
+ def _save(im: Image.Image, fp: IO[bytes], tile: list[_Tile], bufsize: int = 0) -> None:
630
+ """Helper to save image based on tile list
631
+
632
+ :param im: Image object.
633
+ :param fp: File object.
634
+ :param tile: Tile list.
635
+ :param bufsize: Optional buffer size
636
+ """
637
+
638
+ im.load()
639
+ if not hasattr(im, "encoderconfig"):
640
+ im.encoderconfig = ()
641
+ tile.sort(key=_tilesort)
642
+ # FIXME: make MAXBLOCK a configuration parameter
643
+ # It would be great if we could have the encoder specify what it needs
644
+ # But, it would need at least the image size in most cases. RawEncode is
645
+ # a tricky case.
646
+ bufsize = max(MAXBLOCK, bufsize, im.size[0] * 4) # see RawEncode.c
647
+ try:
648
+ fh = fp.fileno()
649
+ fp.flush()
650
+ _encode_tile(im, fp, tile, bufsize, fh)
651
+ except (AttributeError, io.UnsupportedOperation) as exc:
652
+ _encode_tile(im, fp, tile, bufsize, None, exc)
653
+ if hasattr(fp, "flush"):
654
+ fp.flush()
655
+
656
+
657
+ def _encode_tile(
658
+ im: Image.Image,
659
+ fp: IO[bytes],
660
+ tile: list[_Tile],
661
+ bufsize: int,
662
+ fh: int | None,
663
+ exc: BaseException | None = None,
664
+ ) -> None:
665
+ for encoder_name, extents, offset, args in tile:
666
+ if offset > 0:
667
+ fp.seek(offset)
668
+ encoder = Image._getencoder(im.mode, encoder_name, args, im.encoderconfig)
669
+ try:
670
+ encoder.setimage(im.im, extents)
671
+ if encoder.pushes_fd:
672
+ encoder.setfd(fp)
673
+ errcode = encoder.encode_to_pyfd()[1]
674
+ else:
675
+ if exc:
676
+ # compress to Python file-compatible object
677
+ while True:
678
+ errcode, data = encoder.encode(bufsize)[1:]
679
+ fp.write(data)
680
+ if errcode:
681
+ break
682
+ else:
683
+ # slight speedup: compress to real file object
684
+ assert fh is not None
685
+ errcode = encoder.encode_to_file(fh, bufsize)
686
+ if errcode < 0:
687
+ raise _get_oserror(errcode, encoder=True) from exc
688
+ finally:
689
+ encoder.cleanup()
690
+
691
+
692
+ def _safe_read(fp: IO[bytes], size: int) -> bytes:
693
+ """
694
+ Reads large blocks in a safe way. Unlike fp.read(n), this function
695
+ doesn't trust the user. If the requested size is larger than
696
+ SAFEBLOCK, the file is read block by block.
697
+
698
+ :param fp: File handle. Must implement a <b>read</b> method.
699
+ :param size: Number of bytes to read.
700
+ :returns: A string containing <i>size</i> bytes of data.
701
+
702
+ Raises an OSError if the file is truncated and the read cannot be completed
703
+
704
+ """
705
+ if size <= 0:
706
+ return b""
707
+ if size <= SAFEBLOCK:
708
+ data = fp.read(size)
709
+ if len(data) < size:
710
+ msg = "Truncated File Read"
711
+ raise OSError(msg)
712
+ return data
713
+ blocks: list[bytes] = []
714
+ remaining_size = size
715
+ while remaining_size > 0:
716
+ block = fp.read(min(remaining_size, SAFEBLOCK))
717
+ if not block:
718
+ break
719
+ blocks.append(block)
720
+ remaining_size -= len(block)
721
+ if sum(len(block) for block in blocks) < size:
722
+ msg = "Truncated File Read"
723
+ raise OSError(msg)
724
+ return b"".join(blocks)
725
+
726
+
727
+ class PyCodecState:
728
+ def __init__(self) -> None:
729
+ self.xsize = 0
730
+ self.ysize = 0
731
+ self.xoff = 0
732
+ self.yoff = 0
733
+
734
+ def extents(self) -> tuple[int, int, int, int]:
735
+ return self.xoff, self.yoff, self.xoff + self.xsize, self.yoff + self.ysize
736
+
737
+
738
+ class PyCodec:
739
+ fd: IO[bytes] | None
740
+
741
+ def __init__(self, mode: str, *args: Any) -> None:
742
+ self.im: Image.core.ImagingCore | None = None
743
+ self.state = PyCodecState()
744
+ self.fd = None
745
+ self.mode = mode
746
+ self.init(args)
747
+
748
+ def init(self, args: tuple[Any, ...]) -> None:
749
+ """
750
+ Override to perform codec specific initialization
751
+
752
+ :param args: Tuple of arg items from the tile entry
753
+ :returns: None
754
+ """
755
+ self.args = args
756
+
757
+ def cleanup(self) -> None:
758
+ """
759
+ Override to perform codec specific cleanup
760
+
761
+ :returns: None
762
+ """
763
+ pass
764
+
765
+ def setfd(self, fd: IO[bytes]) -> None:
766
+ """
767
+ Called from ImageFile to set the Python file-like object
768
+
769
+ :param fd: A Python file-like object
770
+ :returns: None
771
+ """
772
+ self.fd = fd
773
+
774
+ def setimage(
775
+ self,
776
+ im: Image.core.ImagingCore,
777
+ extents: tuple[int, int, int, int] | None = None,
778
+ ) -> None:
779
+ """
780
+ Called from ImageFile to set the core output image for the codec
781
+
782
+ :param im: A core image object
783
+ :param extents: a 4 tuple of (x0, y0, x1, y1) defining the rectangle
784
+ for this tile
785
+ :returns: None
786
+ """
787
+
788
+ # following c code
789
+ self.im = im
790
+
791
+ if extents:
792
+ (x0, y0, x1, y1) = extents
793
+ else:
794
+ (x0, y0, x1, y1) = (0, 0, 0, 0)
795
+
796
+ if x0 == 0 and x1 == 0:
797
+ self.state.xsize, self.state.ysize = self.im.size
798
+ else:
799
+ self.state.xoff = x0
800
+ self.state.yoff = y0
801
+ self.state.xsize = x1 - x0
802
+ self.state.ysize = y1 - y0
803
+
804
+ if self.state.xsize <= 0 or self.state.ysize <= 0:
805
+ msg = "Size cannot be negative"
806
+ raise ValueError(msg)
807
+
808
+ if (
809
+ self.state.xsize + self.state.xoff > self.im.size[0]
810
+ or self.state.ysize + self.state.yoff > self.im.size[1]
811
+ ):
812
+ msg = "Tile cannot extend outside image"
813
+ raise ValueError(msg)
814
+
815
+
816
+ class PyDecoder(PyCodec):
817
+ """
818
+ Python implementation of a format decoder. Override this class and
819
+ add the decoding logic in the :meth:`decode` method.
820
+
821
+ See :ref:`Writing Your Own File Codec in Python<file-codecs-py>`
822
+ """
823
+
824
+ _pulls_fd = False
825
+
826
+ @property
827
+ def pulls_fd(self) -> bool:
828
+ return self._pulls_fd
829
+
830
+ def decode(self, buffer: bytes | Image.SupportsArrayInterface) -> tuple[int, int]:
831
+ """
832
+ Override to perform the decoding process.
833
+
834
+ :param buffer: A bytes object with the data to be decoded.
835
+ :returns: A tuple of ``(bytes consumed, errcode)``.
836
+ If finished with decoding return -1 for the bytes consumed.
837
+ Err codes are from :data:`.ImageFile.ERRORS`.
838
+ """
839
+ msg = "unavailable in base decoder"
840
+ raise NotImplementedError(msg)
841
+
842
+ def set_as_raw(
843
+ self, data: bytes, rawmode: str | None = None, extra: tuple[Any, ...] = ()
844
+ ) -> None:
845
+ """
846
+ Convenience method to set the internal image from a stream of raw data
847
+
848
+ :param data: Bytes to be set
849
+ :param rawmode: The rawmode to be used for the decoder.
850
+ If not specified, it will default to the mode of the image
851
+ :param extra: Extra arguments for the decoder.
852
+ :returns: None
853
+ """
854
+
855
+ if not rawmode:
856
+ rawmode = self.mode
857
+ d = Image._getdecoder(self.mode, "raw", rawmode, extra)
858
+ assert self.im is not None
859
+ d.setimage(self.im, self.state.extents())
860
+ s = d.decode(data)
861
+
862
+ if s[0] >= 0:
863
+ msg = "not enough image data"
864
+ raise ValueError(msg)
865
+ if s[1] != 0:
866
+ msg = "cannot decode image data"
867
+ raise ValueError(msg)
868
+
869
+
870
+ class PyEncoder(PyCodec):
871
+ """
872
+ Python implementation of a format encoder. Override this class and
873
+ add the decoding logic in the :meth:`encode` method.
874
+
875
+ See :ref:`Writing Your Own File Codec in Python<file-codecs-py>`
876
+ """
877
+
878
+ _pushes_fd = False
879
+
880
+ @property
881
+ def pushes_fd(self) -> bool:
882
+ return self._pushes_fd
883
+
884
+ def encode(self, bufsize: int) -> tuple[int, int, bytes]:
885
+ """
886
+ Override to perform the encoding process.
887
+
888
+ :param bufsize: Buffer size.
889
+ :returns: A tuple of ``(bytes encoded, errcode, bytes)``.
890
+ If finished with encoding return 1 for the error code.
891
+ Err codes are from :data:`.ImageFile.ERRORS`.
892
+ """
893
+ msg = "unavailable in base encoder"
894
+ raise NotImplementedError(msg)
895
+
896
+ def encode_to_pyfd(self) -> tuple[int, int]:
897
+ """
898
+ If ``pushes_fd`` is ``True``, then this method will be used,
899
+ and ``encode()`` will only be called once.
900
+
901
+ :returns: A tuple of ``(bytes consumed, errcode)``.
902
+ Err codes are from :data:`.ImageFile.ERRORS`.
903
+ """
904
+ if not self.pushes_fd:
905
+ return 0, -8 # bad configuration
906
+ bytes_consumed, errcode, data = self.encode(0)
907
+ if data:
908
+ assert self.fd is not None
909
+ self.fd.write(data)
910
+ return bytes_consumed, errcode
911
+
912
+ def encode_to_file(self, fh: int, bufsize: int) -> int:
913
+ """
914
+ :param fh: File handle.
915
+ :param bufsize: Buffer size.
916
+
917
+ :returns: If finished successfully, return 0.
918
+ Otherwise, return an error code. Err codes are from
919
+ :data:`.ImageFile.ERRORS`.
920
+ """
921
+ errcode = 0
922
+ while errcode == 0:
923
+ status, errcode, buf = self.encode(bufsize)
924
+ if status > 0:
925
+ os.write(fh, buf[status:])
926
+ return errcode
.venv/Lib/site-packages/PIL/ImageFilter.py ADDED
@@ -0,0 +1,607 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #
2
+ # The Python Imaging Library.
3
+ # $Id$
4
+ #
5
+ # standard filters
6
+ #
7
+ # History:
8
+ # 1995-11-27 fl Created
9
+ # 2002-06-08 fl Added rank and mode filters
10
+ # 2003-09-15 fl Fixed rank calculation in rank filter; added expand call
11
+ #
12
+ # Copyright (c) 1997-2003 by Secret Labs AB.
13
+ # Copyright (c) 1995-2002 by Fredrik Lundh.
14
+ #
15
+ # See the README file for information on usage and redistribution.
16
+ #
17
+ from __future__ import annotations
18
+
19
+ import abc
20
+ import functools
21
+ from collections.abc import Sequence
22
+ from typing import cast
23
+
24
+ TYPE_CHECKING = False
25
+ if TYPE_CHECKING:
26
+ from collections.abc import Callable
27
+ from types import ModuleType
28
+ from typing import Any
29
+
30
+ from . import _imaging
31
+ from ._typing import NumpyArray
32
+
33
+
34
+ class Filter(abc.ABC):
35
+ @abc.abstractmethod
36
+ def filter(self, image: _imaging.ImagingCore) -> _imaging.ImagingCore:
37
+ pass
38
+
39
+
40
+ class MultibandFilter(Filter):
41
+ pass
42
+
43
+
44
+ class BuiltinFilter(MultibandFilter):
45
+ filterargs: tuple[Any, ...]
46
+
47
+ def filter(self, image: _imaging.ImagingCore) -> _imaging.ImagingCore:
48
+ if image.mode == "P":
49
+ msg = "cannot filter palette images"
50
+ raise ValueError(msg)
51
+ return image.filter(*self.filterargs)
52
+
53
+
54
+ class Kernel(BuiltinFilter):
55
+ """
56
+ Create a convolution kernel. This only supports 3x3 and 5x5 integer and floating
57
+ point kernels.
58
+
59
+ Kernels can only be applied to "L" and "RGB" images.
60
+
61
+ :param size: Kernel size, given as (width, height). This must be (3,3) or (5,5).
62
+ :param kernel: A sequence containing kernel weights. The kernel will be flipped
63
+ vertically before being applied to the image.
64
+ :param scale: Scale factor. If given, the result for each pixel is divided by this
65
+ value. The default is the sum of the kernel weights.
66
+ :param offset: Offset. If given, this value is added to the result, after it has
67
+ been divided by the scale factor.
68
+ """
69
+
70
+ name = "Kernel"
71
+
72
+ def __init__(
73
+ self,
74
+ size: tuple[int, int],
75
+ kernel: Sequence[float],
76
+ scale: float | None = None,
77
+ offset: float = 0,
78
+ ) -> None:
79
+ if scale is None:
80
+ # default scale is sum of kernel
81
+ scale = functools.reduce(lambda a, b: a + b, kernel)
82
+ if size[0] * size[1] != len(kernel):
83
+ msg = "not enough coefficients in kernel"
84
+ raise ValueError(msg)
85
+ self.filterargs = size, scale, offset, kernel
86
+
87
+
88
+ class RankFilter(Filter):
89
+ """
90
+ Create a rank filter. The rank filter sorts all pixels in
91
+ a window of the given size, and returns the ``rank``'th value.
92
+
93
+ :param size: The kernel size, in pixels.
94
+ :param rank: What pixel value to pick. Use 0 for a min filter,
95
+ ``size * size / 2`` for a median filter, ``size * size - 1``
96
+ for a max filter, etc.
97
+ """
98
+
99
+ name = "Rank"
100
+
101
+ def __init__(self, size: int, rank: int) -> None:
102
+ self.size = size
103
+ self.rank = rank
104
+
105
+ def filter(self, image: _imaging.ImagingCore) -> _imaging.ImagingCore:
106
+ if image.mode == "P":
107
+ msg = "cannot filter palette images"
108
+ raise ValueError(msg)
109
+ image = image.expand(self.size // 2, self.size // 2)
110
+ return image.rankfilter(self.size, self.rank)
111
+
112
+
113
+ class MedianFilter(RankFilter):
114
+ """
115
+ Create a median filter. Picks the median pixel value in a window with the
116
+ given size.
117
+
118
+ :param size: The kernel size, in pixels.
119
+ """
120
+
121
+ name = "Median"
122
+
123
+ def __init__(self, size: int = 3) -> None:
124
+ self.size = size
125
+ self.rank = size * size // 2
126
+
127
+
128
+ class MinFilter(RankFilter):
129
+ """
130
+ Create a min filter. Picks the lowest pixel value in a window with the
131
+ given size.
132
+
133
+ :param size: The kernel size, in pixels.
134
+ """
135
+
136
+ name = "Min"
137
+
138
+ def __init__(self, size: int = 3) -> None:
139
+ self.size = size
140
+ self.rank = 0
141
+
142
+
143
+ class MaxFilter(RankFilter):
144
+ """
145
+ Create a max filter. Picks the largest pixel value in a window with the
146
+ given size.
147
+
148
+ :param size: The kernel size, in pixels.
149
+ """
150
+
151
+ name = "Max"
152
+
153
+ def __init__(self, size: int = 3) -> None:
154
+ self.size = size
155
+ self.rank = size * size - 1
156
+
157
+
158
+ class ModeFilter(Filter):
159
+ """
160
+ Create a mode filter. Picks the most frequent pixel value in a box with the
161
+ given size. Pixel values that occur only once or twice are ignored; if no
162
+ pixel value occurs more than twice, the original pixel value is preserved.
163
+
164
+ :param size: The kernel size, in pixels.
165
+ """
166
+
167
+ name = "Mode"
168
+
169
+ def __init__(self, size: int = 3) -> None:
170
+ self.size = size
171
+
172
+ def filter(self, image: _imaging.ImagingCore) -> _imaging.ImagingCore:
173
+ return image.modefilter(self.size)
174
+
175
+
176
+ class GaussianBlur(MultibandFilter):
177
+ """Blurs the image with a sequence of extended box filters, which
178
+ approximates a Gaussian kernel. For details on accuracy see
179
+ <https://www.mia.uni-saarland.de/Publications/gwosdek-ssvm11.pdf>
180
+
181
+ :param radius: Standard deviation of the Gaussian kernel. Either a sequence of two
182
+ numbers for x and y, or a single number for both.
183
+ """
184
+
185
+ name = "GaussianBlur"
186
+
187
+ def __init__(self, radius: float | Sequence[float] = 2) -> None:
188
+ self.radius = radius
189
+
190
+ def filter(self, image: _imaging.ImagingCore) -> _imaging.ImagingCore:
191
+ xy = self.radius
192
+ if isinstance(xy, (int, float)):
193
+ xy = (xy, xy)
194
+ if xy == (0, 0):
195
+ return image.copy()
196
+ return image.gaussian_blur(xy)
197
+
198
+
199
+ class BoxBlur(MultibandFilter):
200
+ """Blurs the image by setting each pixel to the average value of the pixels
201
+ in a square box extending radius pixels in each direction.
202
+ Supports float radius of arbitrary size. Uses an optimized implementation
203
+ which runs in linear time relative to the size of the image
204
+ for any radius value.
205
+
206
+ :param radius: Size of the box in a direction. Either a sequence of two numbers for
207
+ x and y, or a single number for both.
208
+
209
+ Radius 0 does not blur, returns an identical image.
210
+ Radius 1 takes 1 pixel in each direction, i.e. 9 pixels in total.
211
+ """
212
+
213
+ name = "BoxBlur"
214
+
215
+ def __init__(self, radius: float | Sequence[float]) -> None:
216
+ xy = radius if isinstance(radius, (tuple, list)) else (radius, radius)
217
+ if xy[0] < 0 or xy[1] < 0:
218
+ msg = "radius must be >= 0"
219
+ raise ValueError(msg)
220
+ self.radius = radius
221
+
222
+ def filter(self, image: _imaging.ImagingCore) -> _imaging.ImagingCore:
223
+ xy = self.radius
224
+ if isinstance(xy, (int, float)):
225
+ xy = (xy, xy)
226
+ if xy == (0, 0):
227
+ return image.copy()
228
+ return image.box_blur(xy)
229
+
230
+
231
+ class UnsharpMask(MultibandFilter):
232
+ """Unsharp mask filter.
233
+
234
+ See Wikipedia's entry on `digital unsharp masking`_ for an explanation of
235
+ the parameters.
236
+
237
+ :param radius: Blur Radius
238
+ :param percent: Unsharp strength, in percent
239
+ :param threshold: Threshold controls the minimum brightness change that
240
+ will be sharpened
241
+
242
+ .. _digital unsharp masking: https://en.wikipedia.org/wiki/Unsharp_masking#Digital_unsharp_masking
243
+
244
+ """
245
+
246
+ name = "UnsharpMask"
247
+
248
+ def __init__(
249
+ self, radius: float = 2, percent: int = 150, threshold: int = 3
250
+ ) -> None:
251
+ self.radius = radius
252
+ self.percent = percent
253
+ self.threshold = threshold
254
+
255
+ def filter(self, image: _imaging.ImagingCore) -> _imaging.ImagingCore:
256
+ return image.unsharp_mask(self.radius, self.percent, self.threshold)
257
+
258
+
259
+ class BLUR(BuiltinFilter):
260
+ name = "Blur"
261
+ # fmt: off
262
+ filterargs = (5, 5), 16, 0, (
263
+ 1, 1, 1, 1, 1,
264
+ 1, 0, 0, 0, 1,
265
+ 1, 0, 0, 0, 1,
266
+ 1, 0, 0, 0, 1,
267
+ 1, 1, 1, 1, 1,
268
+ )
269
+ # fmt: on
270
+
271
+
272
+ class CONTOUR(BuiltinFilter):
273
+ name = "Contour"
274
+ # fmt: off
275
+ filterargs = (3, 3), 1, 255, (
276
+ -1, -1, -1,
277
+ -1, 8, -1,
278
+ -1, -1, -1,
279
+ )
280
+ # fmt: on
281
+
282
+
283
+ class DETAIL(BuiltinFilter):
284
+ name = "Detail"
285
+ # fmt: off
286
+ filterargs = (3, 3), 6, 0, (
287
+ 0, -1, 0,
288
+ -1, 10, -1,
289
+ 0, -1, 0,
290
+ )
291
+ # fmt: on
292
+
293
+
294
+ class EDGE_ENHANCE(BuiltinFilter):
295
+ name = "Edge-enhance"
296
+ # fmt: off
297
+ filterargs = (3, 3), 2, 0, (
298
+ -1, -1, -1,
299
+ -1, 10, -1,
300
+ -1, -1, -1,
301
+ )
302
+ # fmt: on
303
+
304
+
305
+ class EDGE_ENHANCE_MORE(BuiltinFilter):
306
+ name = "Edge-enhance More"
307
+ # fmt: off
308
+ filterargs = (3, 3), 1, 0, (
309
+ -1, -1, -1,
310
+ -1, 9, -1,
311
+ -1, -1, -1,
312
+ )
313
+ # fmt: on
314
+
315
+
316
+ class EMBOSS(BuiltinFilter):
317
+ name = "Emboss"
318
+ # fmt: off
319
+ filterargs = (3, 3), 1, 128, (
320
+ -1, 0, 0,
321
+ 0, 1, 0,
322
+ 0, 0, 0,
323
+ )
324
+ # fmt: on
325
+
326
+
327
+ class FIND_EDGES(BuiltinFilter):
328
+ name = "Find Edges"
329
+ # fmt: off
330
+ filterargs = (3, 3), 1, 0, (
331
+ -1, -1, -1,
332
+ -1, 8, -1,
333
+ -1, -1, -1,
334
+ )
335
+ # fmt: on
336
+
337
+
338
+ class SHARPEN(BuiltinFilter):
339
+ name = "Sharpen"
340
+ # fmt: off
341
+ filterargs = (3, 3), 16, 0, (
342
+ -2, -2, -2,
343
+ -2, 32, -2,
344
+ -2, -2, -2,
345
+ )
346
+ # fmt: on
347
+
348
+
349
+ class SMOOTH(BuiltinFilter):
350
+ name = "Smooth"
351
+ # fmt: off
352
+ filterargs = (3, 3), 13, 0, (
353
+ 1, 1, 1,
354
+ 1, 5, 1,
355
+ 1, 1, 1,
356
+ )
357
+ # fmt: on
358
+
359
+
360
+ class SMOOTH_MORE(BuiltinFilter):
361
+ name = "Smooth More"
362
+ # fmt: off
363
+ filterargs = (5, 5), 100, 0, (
364
+ 1, 1, 1, 1, 1,
365
+ 1, 5, 5, 5, 1,
366
+ 1, 5, 44, 5, 1,
367
+ 1, 5, 5, 5, 1,
368
+ 1, 1, 1, 1, 1,
369
+ )
370
+ # fmt: on
371
+
372
+
373
+ class Color3DLUT(MultibandFilter):
374
+ """Three-dimensional color lookup table.
375
+
376
+ Transforms 3-channel pixels using the values of the channels as coordinates
377
+ in the 3D lookup table and interpolating the nearest elements.
378
+
379
+ This method allows you to apply almost any color transformation
380
+ in constant time by using pre-calculated decimated tables.
381
+
382
+ .. versionadded:: 5.2.0
383
+
384
+ :param size: Size of the table. One int or tuple of (int, int, int).
385
+ Minimal size in any dimension is 2, maximum is 65.
386
+ :param table: Flat lookup table. A list of ``channels * size**3``
387
+ float elements or a list of ``size**3`` channels-sized
388
+ tuples with floats. Channels are changed first,
389
+ then first dimension, then second, then third.
390
+ Value 0.0 corresponds lowest value of output, 1.0 highest.
391
+ :param channels: Number of channels in the table. Could be 3 or 4.
392
+ Default is 3.
393
+ :param target_mode: A mode for the result image. Should have not less
394
+ than ``channels`` channels. Default is ``None``,
395
+ which means that mode wouldn't be changed.
396
+ """
397
+
398
+ name = "Color 3D LUT"
399
+
400
+ def __init__(
401
+ self,
402
+ size: int | tuple[int, int, int],
403
+ table: Sequence[float] | Sequence[Sequence[int]] | NumpyArray,
404
+ channels: int = 3,
405
+ target_mode: str | None = None,
406
+ **kwargs: bool,
407
+ ) -> None:
408
+ if channels not in (3, 4):
409
+ msg = "Only 3 or 4 output channels are supported"
410
+ raise ValueError(msg)
411
+ self.size = size = self._check_size(size)
412
+ self.channels = channels
413
+ self.mode = target_mode
414
+
415
+ # Hidden flag `_copy_table=False` could be used to avoid extra copying
416
+ # of the table if the table is specially made for the constructor.
417
+ copy_table = kwargs.get("_copy_table", True)
418
+ items = size[0] * size[1] * size[2]
419
+ wrong_size = False
420
+
421
+ numpy: ModuleType | None = None
422
+ if hasattr(table, "shape"):
423
+ try:
424
+ import numpy
425
+ except ImportError:
426
+ pass
427
+
428
+ if numpy and isinstance(table, numpy.ndarray):
429
+ numpy_table: NumpyArray = table
430
+ if copy_table:
431
+ numpy_table = numpy_table.copy()
432
+
433
+ if numpy_table.shape in [
434
+ (items * channels,),
435
+ (items, channels),
436
+ (size[2], size[1], size[0], channels),
437
+ ]:
438
+ table = numpy_table.reshape(items * channels)
439
+ else:
440
+ wrong_size = True
441
+
442
+ else:
443
+ if copy_table:
444
+ table = list(table)
445
+
446
+ # Convert to a flat list
447
+ if table and isinstance(table[0], (list, tuple)):
448
+ raw_table = cast(Sequence[Sequence[int]], table)
449
+ flat_table: list[int] = []
450
+ for pixel in raw_table:
451
+ if len(pixel) != channels:
452
+ msg = (
453
+ "The elements of the table should "
454
+ f"have a length of {channels}."
455
+ )
456
+ raise ValueError(msg)
457
+ flat_table.extend(pixel)
458
+ table = flat_table
459
+
460
+ if wrong_size or len(table) != items * channels:
461
+ msg = (
462
+ "The table should have either channels * size**3 float items "
463
+ "or size**3 items of channels-sized tuples with floats. "
464
+ f"Table should be: {channels}x{size[0]}x{size[1]}x{size[2]}. "
465
+ f"Actual length: {len(table)}"
466
+ )
467
+ raise ValueError(msg)
468
+ self.table = table
469
+
470
+ @staticmethod
471
+ def _check_size(size: Any) -> tuple[int, int, int]:
472
+ try:
473
+ _, _, _ = size
474
+ except ValueError as e:
475
+ msg = "Size should be either an integer or a tuple of three integers."
476
+ raise ValueError(msg) from e
477
+ except TypeError:
478
+ size = (size, size, size)
479
+ size = tuple(int(x) for x in size)
480
+ for size_1d in size:
481
+ if not 2 <= size_1d <= 65:
482
+ msg = "Size should be in [2, 65] range."
483
+ raise ValueError(msg)
484
+ return size
485
+
486
+ @classmethod
487
+ def generate(
488
+ cls,
489
+ size: int | tuple[int, int, int],
490
+ callback: Callable[[float, float, float], tuple[float, ...]],
491
+ channels: int = 3,
492
+ target_mode: str | None = None,
493
+ ) -> Color3DLUT:
494
+ """Generates new LUT using provided callback.
495
+
496
+ :param size: Size of the table. Passed to the constructor.
497
+ :param callback: Function with three parameters which correspond
498
+ three color channels. Will be called ``size**3``
499
+ times with values from 0.0 to 1.0 and should return
500
+ a tuple with ``channels`` elements.
501
+ :param channels: The number of channels which should return callback.
502
+ :param target_mode: Passed to the constructor of the resulting
503
+ lookup table.
504
+ """
505
+ size_1d, size_2d, size_3d = cls._check_size(size)
506
+ if channels not in (3, 4):
507
+ msg = "Only 3 or 4 output channels are supported"
508
+ raise ValueError(msg)
509
+
510
+ table: list[float] = [0] * (size_1d * size_2d * size_3d * channels)
511
+ idx_out = 0
512
+ for b in range(size_3d):
513
+ for g in range(size_2d):
514
+ for r in range(size_1d):
515
+ table[idx_out : idx_out + channels] = callback(
516
+ r / (size_1d - 1), g / (size_2d - 1), b / (size_3d - 1)
517
+ )
518
+ idx_out += channels
519
+
520
+ return cls(
521
+ (size_1d, size_2d, size_3d),
522
+ table,
523
+ channels=channels,
524
+ target_mode=target_mode,
525
+ _copy_table=False,
526
+ )
527
+
528
+ def transform(
529
+ self,
530
+ callback: Callable[..., tuple[float, ...]],
531
+ with_normals: bool = False,
532
+ channels: int | None = None,
533
+ target_mode: str | None = None,
534
+ ) -> Color3DLUT:
535
+ """Transforms the table values using provided callback and returns
536
+ a new LUT with altered values.
537
+
538
+ :param callback: A function which takes old lookup table values
539
+ and returns a new set of values. The number
540
+ of arguments which function should take is
541
+ ``self.channels`` or ``3 + self.channels``
542
+ if ``with_normals`` flag is set.
543
+ Should return a tuple of ``self.channels`` or
544
+ ``channels`` elements if it is set.
545
+ :param with_normals: If true, ``callback`` will be called with
546
+ coordinates in the color cube as the first
547
+ three arguments. Otherwise, ``callback``
548
+ will be called only with actual color values.
549
+ :param channels: The number of channels in the resulting lookup table.
550
+ :param target_mode: Passed to the constructor of the resulting
551
+ lookup table.
552
+ """
553
+ if channels not in (None, 3, 4):
554
+ msg = "Only 3 or 4 output channels are supported"
555
+ raise ValueError(msg)
556
+ ch_in = self.channels
557
+ ch_out = channels or ch_in
558
+ size_1d, size_2d, size_3d = self.size
559
+
560
+ table: list[float] = [0] * (size_1d * size_2d * size_3d * ch_out)
561
+ idx_in = 0
562
+ idx_out = 0
563
+ for b in range(size_3d):
564
+ for g in range(size_2d):
565
+ for r in range(size_1d):
566
+ values = self.table[idx_in : idx_in + ch_in]
567
+ if with_normals:
568
+ values = callback(
569
+ r / (size_1d - 1),
570
+ g / (size_2d - 1),
571
+ b / (size_3d - 1),
572
+ *values,
573
+ )
574
+ else:
575
+ values = callback(*values)
576
+ table[idx_out : idx_out + ch_out] = values
577
+ idx_in += ch_in
578
+ idx_out += ch_out
579
+
580
+ return type(self)(
581
+ self.size,
582
+ table,
583
+ channels=ch_out,
584
+ target_mode=target_mode or self.mode,
585
+ _copy_table=False,
586
+ )
587
+
588
+ def __repr__(self) -> str:
589
+ r = [
590
+ f"{self.__class__.__name__} from {self.table.__class__.__name__}",
591
+ "size={:d}x{:d}x{:d}".format(*self.size),
592
+ f"channels={self.channels:d}",
593
+ ]
594
+ if self.mode:
595
+ r.append(f"target_mode={self.mode}")
596
+ return "<{}>".format(" ".join(r))
597
+
598
+ def filter(self, image: _imaging.ImagingCore) -> _imaging.ImagingCore:
599
+ from . import Image
600
+
601
+ return image.color_lut_3d(
602
+ self.mode or image.mode,
603
+ Image.Resampling.BILINEAR,
604
+ self.channels,
605
+ self.size,
606
+ self.table,
607
+ )
.venv/Lib/site-packages/PIL/ImageFont.py ADDED
@@ -0,0 +1,1312 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #
2
+ # The Python Imaging Library.
3
+ # $Id$
4
+ #
5
+ # PIL raster font management
6
+ #
7
+ # History:
8
+ # 1996-08-07 fl created (experimental)
9
+ # 1997-08-25 fl minor adjustments to handle fonts from pilfont 0.3
10
+ # 1999-02-06 fl rewrote most font management stuff in C
11
+ # 1999-03-17 fl take pth files into account in load_path (from Richard Jones)
12
+ # 2001-02-17 fl added freetype support
13
+ # 2001-05-09 fl added TransposedFont wrapper class
14
+ # 2002-03-04 fl make sure we have a "L" or "1" font
15
+ # 2002-12-04 fl skip non-directory entries in the system path
16
+ # 2003-04-29 fl add embedded default font
17
+ # 2003-09-27 fl added support for truetype charmap encodings
18
+ #
19
+ # Todo:
20
+ # Adapt to PILFONT2 format (16-bit fonts, compressed, single file)
21
+ #
22
+ # Copyright (c) 1997-2003 by Secret Labs AB
23
+ # Copyright (c) 1996-2003 by Fredrik Lundh
24
+ #
25
+ # See the README file for information on usage and redistribution.
26
+ #
27
+
28
+ from __future__ import annotations
29
+
30
+ import base64
31
+ import os
32
+ import sys
33
+ import warnings
34
+ from enum import IntEnum
35
+ from io import BytesIO
36
+ from types import ModuleType
37
+ from typing import IO, Any, BinaryIO, TypedDict, cast
38
+
39
+ from . import Image
40
+ from ._typing import StrOrBytesPath
41
+ from ._util import DeferredError, is_path
42
+
43
+ TYPE_CHECKING = False
44
+ if TYPE_CHECKING:
45
+ from . import ImageFile
46
+ from ._imaging import ImagingFont
47
+ from ._imagingft import Font
48
+
49
+
50
+ class Axis(TypedDict):
51
+ minimum: int | None
52
+ default: int | None
53
+ maximum: int | None
54
+ name: bytes | None
55
+
56
+
57
+ class Layout(IntEnum):
58
+ BASIC = 0
59
+ RAQM = 1
60
+
61
+
62
+ MAX_STRING_LENGTH = 1_000_000
63
+
64
+
65
+ core: ModuleType | DeferredError
66
+ try:
67
+ from . import _imagingft as core
68
+ except ImportError as ex:
69
+ core = DeferredError.new(ex)
70
+
71
+
72
+ def _string_length_check(text: str | bytes | bytearray) -> None:
73
+ if MAX_STRING_LENGTH is not None and len(text) > MAX_STRING_LENGTH:
74
+ msg = "too many characters in string"
75
+ raise ValueError(msg)
76
+
77
+
78
+ # FIXME: add support for pilfont2 format (see FontFile.py)
79
+
80
+ # --------------------------------------------------------------------
81
+ # Font metrics format:
82
+ # "PILfont" LF
83
+ # fontdescriptor LF
84
+ # (optional) key=value... LF
85
+ # "DATA" LF
86
+ # binary data: 256*10*2 bytes (dx, dy, dstbox, srcbox)
87
+ #
88
+ # To place a character, cut out srcbox and paste at dstbox,
89
+ # relative to the character position. Then move the character
90
+ # position according to dx, dy.
91
+ # --------------------------------------------------------------------
92
+
93
+
94
+ class ImageFont:
95
+ """PIL font wrapper"""
96
+
97
+ font: ImagingFont
98
+
99
+ def _load_pilfont(self, filename: str) -> None:
100
+ with open(filename, "rb") as fp:
101
+ image: ImageFile.ImageFile | None = None
102
+ root = os.path.splitext(filename)[0]
103
+
104
+ for ext in (".png", ".gif", ".pbm"):
105
+ if image:
106
+ image.close()
107
+ try:
108
+ fullname = root + ext
109
+ image = Image.open(fullname)
110
+ except Exception:
111
+ pass
112
+ else:
113
+ if image and image.mode in ("1", "L"):
114
+ break
115
+ else:
116
+ if image:
117
+ image.close()
118
+
119
+ msg = f"cannot find glyph data file {root}.{{gif|pbm|png}}"
120
+ raise OSError(msg)
121
+
122
+ self.file = fullname
123
+
124
+ self._load_pilfont_data(fp, image)
125
+ image.close()
126
+
127
+ def _load_pilfont_data(self, file: IO[bytes], image: Image.Image) -> None:
128
+ # check image
129
+ if image.mode not in ("1", "L"):
130
+ msg = "invalid font image mode"
131
+ raise TypeError(msg)
132
+
133
+ # read PILfont header
134
+ if file.read(8) != b"PILfont\n":
135
+ msg = "Not a PILfont file"
136
+ raise SyntaxError(msg)
137
+ file.readline()
138
+ self.info = [] # FIXME: should be a dictionary
139
+ while True:
140
+ s = file.readline()
141
+ if not s or s == b"DATA\n":
142
+ break
143
+ self.info.append(s)
144
+
145
+ # read PILfont metrics
146
+ data = file.read(256 * 20)
147
+
148
+ image.load()
149
+
150
+ self.font = Image.core.font(image.im, data)
151
+
152
+ def getmask(
153
+ self, text: str | bytes, mode: str = "", *args: Any, **kwargs: Any
154
+ ) -> Image.core.ImagingCore:
155
+ """
156
+ Create a bitmap for the text.
157
+
158
+ If the font uses antialiasing, the bitmap should have mode ``L`` and use a
159
+ maximum value of 255. Otherwise, it should have mode ``1``.
160
+
161
+ :param text: Text to render.
162
+ :param mode: Used by some graphics drivers to indicate what mode the
163
+ driver prefers; if empty, the renderer may return either
164
+ mode. Note that the mode is always a string, to simplify
165
+ C-level implementations.
166
+
167
+ .. versionadded:: 1.1.5
168
+
169
+ :return: An internal PIL storage memory instance as defined by the
170
+ :py:mod:`PIL.Image.core` interface module.
171
+ """
172
+ _string_length_check(text)
173
+ Image._decompression_bomb_check(self.font.getsize(text))
174
+ return self.font.getmask(text, mode)
175
+
176
+ def getbbox(
177
+ self, text: str | bytes | bytearray, *args: Any, **kwargs: Any
178
+ ) -> tuple[int, int, int, int]:
179
+ """
180
+ Returns bounding box (in pixels) of given text.
181
+
182
+ .. versionadded:: 9.2.0
183
+
184
+ :param text: Text to render.
185
+
186
+ :return: ``(left, top, right, bottom)`` bounding box
187
+ """
188
+ _string_length_check(text)
189
+ width, height = self.font.getsize(text)
190
+ return 0, 0, width, height
191
+
192
+ def getlength(
193
+ self, text: str | bytes | bytearray, *args: Any, **kwargs: Any
194
+ ) -> int:
195
+ """
196
+ Returns length (in pixels) of given text.
197
+ This is the amount by which following text should be offset.
198
+
199
+ .. versionadded:: 9.2.0
200
+ """
201
+ _string_length_check(text)
202
+ width, height = self.font.getsize(text)
203
+ return width
204
+
205
+
206
+ ##
207
+ # Wrapper for FreeType fonts. Application code should use the
208
+ # <b>truetype</b> factory function to create font objects.
209
+
210
+
211
+ class FreeTypeFont:
212
+ """FreeType font wrapper (requires _imagingft service)"""
213
+
214
+ font: Font
215
+ font_bytes: bytes
216
+
217
+ def __init__(
218
+ self,
219
+ font: StrOrBytesPath | BinaryIO,
220
+ size: float = 10,
221
+ index: int = 0,
222
+ encoding: str = "",
223
+ layout_engine: Layout | None = None,
224
+ ) -> None:
225
+ # FIXME: use service provider instead
226
+
227
+ if isinstance(core, DeferredError):
228
+ raise core.ex
229
+
230
+ if size <= 0:
231
+ msg = f"font size must be greater than 0, not {size}"
232
+ raise ValueError(msg)
233
+
234
+ self.path = font
235
+ self.size = size
236
+ self.index = index
237
+ self.encoding = encoding
238
+
239
+ if layout_engine not in (Layout.BASIC, Layout.RAQM):
240
+ layout_engine = Layout.BASIC
241
+ if core.HAVE_RAQM:
242
+ layout_engine = Layout.RAQM
243
+ elif layout_engine == Layout.RAQM and not core.HAVE_RAQM:
244
+ warnings.warn(
245
+ "Raqm layout was requested, but Raqm is not available. "
246
+ "Falling back to basic layout."
247
+ )
248
+ layout_engine = Layout.BASIC
249
+
250
+ self.layout_engine = layout_engine
251
+
252
+ def load_from_bytes(f: IO[bytes]) -> None:
253
+ self.font_bytes = f.read()
254
+ self.font = core.getfont(
255
+ "", size, index, encoding, self.font_bytes, layout_engine
256
+ )
257
+
258
+ if is_path(font):
259
+ font = os.fspath(font)
260
+ if sys.platform == "win32":
261
+ font_bytes_path = font if isinstance(font, bytes) else font.encode()
262
+ try:
263
+ font_bytes_path.decode("ascii")
264
+ except UnicodeDecodeError:
265
+ # FreeType cannot load fonts with non-ASCII characters on Windows
266
+ # So load it into memory first
267
+ with open(font, "rb") as f:
268
+ load_from_bytes(f)
269
+ return
270
+ self.font = core.getfont(
271
+ font, size, index, encoding, layout_engine=layout_engine
272
+ )
273
+ else:
274
+ load_from_bytes(cast(IO[bytes], font))
275
+
276
+ def __getstate__(self) -> list[Any]:
277
+ return [self.path, self.size, self.index, self.encoding, self.layout_engine]
278
+
279
+ def __setstate__(self, state: list[Any]) -> None:
280
+ path, size, index, encoding, layout_engine = state
281
+ FreeTypeFont.__init__(self, path, size, index, encoding, layout_engine)
282
+
283
+ def getname(self) -> tuple[str | None, str | None]:
284
+ """
285
+ :return: A tuple of the font family (e.g. Helvetica) and the font style
286
+ (e.g. Bold)
287
+ """
288
+ return self.font.family, self.font.style
289
+
290
+ def getmetrics(self) -> tuple[int, int]:
291
+ """
292
+ :return: A tuple of the font ascent (the distance from the baseline to
293
+ the highest outline point) and descent (the distance from the
294
+ baseline to the lowest outline point, a negative value)
295
+ """
296
+ return self.font.ascent, self.font.descent
297
+
298
+ def getlength(
299
+ self,
300
+ text: str | bytes,
301
+ mode: str = "",
302
+ direction: str | None = None,
303
+ features: list[str] | None = None,
304
+ language: str | None = None,
305
+ ) -> float:
306
+ """
307
+ Returns length (in pixels with 1/64 precision) of given text when rendered
308
+ in font with provided direction, features, and language.
309
+
310
+ This is the amount by which following text should be offset.
311
+ Text bounding box may extend past the length in some fonts,
312
+ e.g. when using italics or accents.
313
+
314
+ The result is returned as a float; it is a whole number if using basic layout.
315
+
316
+ Note that the sum of two lengths may not equal the length of a concatenated
317
+ string due to kerning. If you need to adjust for kerning, include the following
318
+ character and subtract its length.
319
+
320
+ For example, instead of ::
321
+
322
+ hello = font.getlength("Hello")
323
+ world = font.getlength("World")
324
+ hello_world = hello + world # not adjusted for kerning
325
+ assert hello_world == font.getlength("HelloWorld") # may fail
326
+
327
+ use ::
328
+
329
+ hello = font.getlength("HelloW") - font.getlength("W") # adjusted for kerning
330
+ world = font.getlength("World")
331
+ hello_world = hello + world # adjusted for kerning
332
+ assert hello_world == font.getlength("HelloWorld") # True
333
+
334
+ or disable kerning with (requires libraqm) ::
335
+
336
+ hello = draw.textlength("Hello", font, features=["-kern"])
337
+ world = draw.textlength("World", font, features=["-kern"])
338
+ hello_world = hello + world # kerning is disabled, no need to adjust
339
+ assert hello_world == draw.textlength("HelloWorld", font, features=["-kern"])
340
+
341
+ .. versionadded:: 8.0.0
342
+
343
+ :param text: Text to measure.
344
+ :param mode: Used by some graphics drivers to indicate what mode the
345
+ driver prefers; if empty, the renderer may return either
346
+ mode. Note that the mode is always a string, to simplify
347
+ C-level implementations.
348
+
349
+ :param direction: Direction of the text. It can be 'rtl' (right to
350
+ left), 'ltr' (left to right) or 'ttb' (top to bottom).
351
+ Requires libraqm.
352
+
353
+ :param features: A list of OpenType font features to be used during text
354
+ layout. This is usually used to turn on optional
355
+ font features that are not enabled by default,
356
+ for example 'dlig' or 'ss01', but can be also
357
+ used to turn off default font features for
358
+ example '-liga' to disable ligatures or '-kern'
359
+ to disable kerning. To get all supported
360
+ features, see
361
+ https://learn.microsoft.com/en-us/typography/opentype/spec/featurelist
362
+ Requires libraqm.
363
+
364
+ :param language: Language of the text. Different languages may use
365
+ different glyph shapes or ligatures. This parameter tells
366
+ the font which language the text is in, and to apply the
367
+ correct substitutions as appropriate, if available.
368
+ It should be a `BCP 47 language code
369
+ <https://www.w3.org/International/articles/language-tags/>`_
370
+ Requires libraqm.
371
+
372
+ :return: Either width for horizontal text, or height for vertical text.
373
+ """
374
+ _string_length_check(text)
375
+ return self.font.getlength(text, mode, direction, features, language) / 64
376
+
377
+ def getbbox(
378
+ self,
379
+ text: str | bytes,
380
+ mode: str = "",
381
+ direction: str | None = None,
382
+ features: list[str] | None = None,
383
+ language: str | None = None,
384
+ stroke_width: float = 0,
385
+ anchor: str | None = None,
386
+ ) -> tuple[float, float, float, float]:
387
+ """
388
+ Returns bounding box (in pixels) of given text relative to given anchor
389
+ when rendered in font with provided direction, features, and language.
390
+
391
+ Use :py:meth:`getlength()` to get the offset of following text with
392
+ 1/64 pixel precision. The bounding box includes extra margins for
393
+ some fonts, e.g. italics or accents.
394
+
395
+ .. versionadded:: 8.0.0
396
+
397
+ :param text: Text to render.
398
+ :param mode: Used by some graphics drivers to indicate what mode the
399
+ driver prefers; if empty, the renderer may return either
400
+ mode. Note that the mode is always a string, to simplify
401
+ C-level implementations.
402
+
403
+ :param direction: Direction of the text. It can be 'rtl' (right to
404
+ left), 'ltr' (left to right) or 'ttb' (top to bottom).
405
+ Requires libraqm.
406
+
407
+ :param features: A list of OpenType font features to be used during text
408
+ layout. This is usually used to turn on optional
409
+ font features that are not enabled by default,
410
+ for example 'dlig' or 'ss01', but can be also
411
+ used to turn off default font features for
412
+ example '-liga' to disable ligatures or '-kern'
413
+ to disable kerning. To get all supported
414
+ features, see
415
+ https://learn.microsoft.com/en-us/typography/opentype/spec/featurelist
416
+ Requires libraqm.
417
+
418
+ :param language: Language of the text. Different languages may use
419
+ different glyph shapes or ligatures. This parameter tells
420
+ the font which language the text is in, and to apply the
421
+ correct substitutions as appropriate, if available.
422
+ It should be a `BCP 47 language code
423
+ <https://www.w3.org/International/articles/language-tags/>`_
424
+ Requires libraqm.
425
+
426
+ :param stroke_width: The width of the text stroke.
427
+
428
+ :param anchor: The text anchor alignment. Determines the relative location of
429
+ the anchor to the text. The default alignment is top left,
430
+ specifically ``la`` for horizontal text and ``lt`` for
431
+ vertical text. See :ref:`text-anchors` for details.
432
+
433
+ :return: ``(left, top, right, bottom)`` bounding box
434
+ """
435
+ _string_length_check(text)
436
+ size, offset = self.font.getsize(
437
+ text, mode, direction, features, language, anchor
438
+ )
439
+ left, top = offset[0] - stroke_width, offset[1] - stroke_width
440
+ width, height = size[0] + 2 * stroke_width, size[1] + 2 * stroke_width
441
+ return left, top, left + width, top + height
442
+
443
+ def getmask(
444
+ self,
445
+ text: str | bytes,
446
+ mode: str = "",
447
+ direction: str | None = None,
448
+ features: list[str] | None = None,
449
+ language: str | None = None,
450
+ stroke_width: float = 0,
451
+ anchor: str | None = None,
452
+ ink: int = 0,
453
+ start: tuple[float, float] | None = None,
454
+ ) -> Image.core.ImagingCore:
455
+ """
456
+ Create a bitmap for the text.
457
+
458
+ If the font uses antialiasing, the bitmap should have mode ``L`` and use a
459
+ maximum value of 255. If the font has embedded color data, the bitmap
460
+ should have mode ``RGBA``. Otherwise, it should have mode ``1``.
461
+
462
+ :param text: Text to render.
463
+ :param mode: Used by some graphics drivers to indicate what mode the
464
+ driver prefers; if empty, the renderer may return either
465
+ mode. Note that the mode is always a string, to simplify
466
+ C-level implementations.
467
+
468
+ .. versionadded:: 1.1.5
469
+
470
+ :param direction: Direction of the text. It can be 'rtl' (right to
471
+ left), 'ltr' (left to right) or 'ttb' (top to bottom).
472
+ Requires libraqm.
473
+
474
+ .. versionadded:: 4.2.0
475
+
476
+ :param features: A list of OpenType font features to be used during text
477
+ layout. This is usually used to turn on optional
478
+ font features that are not enabled by default,
479
+ for example 'dlig' or 'ss01', but can be also
480
+ used to turn off default font features for
481
+ example '-liga' to disable ligatures or '-kern'
482
+ to disable kerning. To get all supported
483
+ features, see
484
+ https://learn.microsoft.com/en-us/typography/opentype/spec/featurelist
485
+ Requires libraqm.
486
+
487
+ .. versionadded:: 4.2.0
488
+
489
+ :param language: Language of the text. Different languages may use
490
+ different glyph shapes or ligatures. This parameter tells
491
+ the font which language the text is in, and to apply the
492
+ correct substitutions as appropriate, if available.
493
+ It should be a `BCP 47 language code
494
+ <https://www.w3.org/International/articles/language-tags/>`_
495
+ Requires libraqm.
496
+
497
+ .. versionadded:: 6.0.0
498
+
499
+ :param stroke_width: The width of the text stroke.
500
+
501
+ .. versionadded:: 6.2.0
502
+
503
+ :param anchor: The text anchor alignment. Determines the relative location of
504
+ the anchor to the text. The default alignment is top left,
505
+ specifically ``la`` for horizontal text and ``lt`` for
506
+ vertical text. See :ref:`text-anchors` for details.
507
+
508
+ .. versionadded:: 8.0.0
509
+
510
+ :param ink: Foreground ink for rendering in RGBA mode.
511
+
512
+ .. versionadded:: 8.0.0
513
+
514
+ :param start: Tuple of horizontal and vertical offset, as text may render
515
+ differently when starting at fractional coordinates.
516
+
517
+ .. versionadded:: 9.4.0
518
+
519
+ :return: An internal PIL storage memory instance as defined by the
520
+ :py:mod:`PIL.Image.core` interface module.
521
+ """
522
+ return self.getmask2(
523
+ text,
524
+ mode,
525
+ direction=direction,
526
+ features=features,
527
+ language=language,
528
+ stroke_width=stroke_width,
529
+ anchor=anchor,
530
+ ink=ink,
531
+ start=start,
532
+ )[0]
533
+
534
+ def getmask2(
535
+ self,
536
+ text: str | bytes,
537
+ mode: str = "",
538
+ direction: str | None = None,
539
+ features: list[str] | None = None,
540
+ language: str | None = None,
541
+ stroke_width: float = 0,
542
+ anchor: str | None = None,
543
+ ink: int = 0,
544
+ start: tuple[float, float] | None = None,
545
+ *args: Any,
546
+ **kwargs: Any,
547
+ ) -> tuple[Image.core.ImagingCore, tuple[int, int]]:
548
+ """
549
+ Create a bitmap for the text.
550
+
551
+ If the font uses antialiasing, the bitmap should have mode ``L`` and use a
552
+ maximum value of 255. If the font has embedded color data, the bitmap
553
+ should have mode ``RGBA``. Otherwise, it should have mode ``1``.
554
+
555
+ :param text: Text to render.
556
+ :param mode: Used by some graphics drivers to indicate what mode the
557
+ driver prefers; if empty, the renderer may return either
558
+ mode. Note that the mode is always a string, to simplify
559
+ C-level implementations.
560
+
561
+ .. versionadded:: 1.1.5
562
+
563
+ :param direction: Direction of the text. It can be 'rtl' (right to
564
+ left), 'ltr' (left to right) or 'ttb' (top to bottom).
565
+ Requires libraqm.
566
+
567
+ .. versionadded:: 4.2.0
568
+
569
+ :param features: A list of OpenType font features to be used during text
570
+ layout. This is usually used to turn on optional
571
+ font features that are not enabled by default,
572
+ for example 'dlig' or 'ss01', but can be also
573
+ used to turn off default font features for
574
+ example '-liga' to disable ligatures or '-kern'
575
+ to disable kerning. To get all supported
576
+ features, see
577
+ https://learn.microsoft.com/en-us/typography/opentype/spec/featurelist
578
+ Requires libraqm.
579
+
580
+ .. versionadded:: 4.2.0
581
+
582
+ :param language: Language of the text. Different languages may use
583
+ different glyph shapes or ligatures. This parameter tells
584
+ the font which language the text is in, and to apply the
585
+ correct substitutions as appropriate, if available.
586
+ It should be a `BCP 47 language code
587
+ <https://www.w3.org/International/articles/language-tags/>`_
588
+ Requires libraqm.
589
+
590
+ .. versionadded:: 6.0.0
591
+
592
+ :param stroke_width: The width of the text stroke.
593
+
594
+ .. versionadded:: 6.2.0
595
+
596
+ :param anchor: The text anchor alignment. Determines the relative location of
597
+ the anchor to the text. The default alignment is top left,
598
+ specifically ``la`` for horizontal text and ``lt`` for
599
+ vertical text. See :ref:`text-anchors` for details.
600
+
601
+ .. versionadded:: 8.0.0
602
+
603
+ :param ink: Foreground ink for rendering in RGBA mode.
604
+
605
+ .. versionadded:: 8.0.0
606
+
607
+ :param start: Tuple of horizontal and vertical offset, as text may render
608
+ differently when starting at fractional coordinates.
609
+
610
+ .. versionadded:: 9.4.0
611
+
612
+ :return: A tuple of an internal PIL storage memory instance as defined by the
613
+ :py:mod:`PIL.Image.core` interface module, and the text offset, the
614
+ gap between the starting coordinate and the first marking
615
+ """
616
+ _string_length_check(text)
617
+ if start is None:
618
+ start = (0, 0)
619
+
620
+ def fill(width: int, height: int) -> Image.core.ImagingCore:
621
+ size = (width, height)
622
+ Image._decompression_bomb_check(size)
623
+ return Image.core.fill("RGBA" if mode == "RGBA" else "L", size)
624
+
625
+ return self.font.render(
626
+ text,
627
+ fill,
628
+ mode,
629
+ direction,
630
+ features,
631
+ language,
632
+ stroke_width,
633
+ kwargs.get("stroke_filled", False),
634
+ anchor,
635
+ ink,
636
+ start,
637
+ )
638
+
639
+ def font_variant(
640
+ self,
641
+ font: StrOrBytesPath | BinaryIO | None = None,
642
+ size: float | None = None,
643
+ index: int | None = None,
644
+ encoding: str | None = None,
645
+ layout_engine: Layout | None = None,
646
+ ) -> FreeTypeFont:
647
+ """
648
+ Create a copy of this FreeTypeFont object,
649
+ using any specified arguments to override the settings.
650
+
651
+ Parameters are identical to the parameters used to initialize this
652
+ object.
653
+
654
+ :return: A FreeTypeFont object.
655
+ """
656
+ if font is None:
657
+ try:
658
+ font = BytesIO(self.font_bytes)
659
+ except AttributeError:
660
+ font = self.path
661
+ return FreeTypeFont(
662
+ font=font,
663
+ size=self.size if size is None else size,
664
+ index=self.index if index is None else index,
665
+ encoding=self.encoding if encoding is None else encoding,
666
+ layout_engine=layout_engine or self.layout_engine,
667
+ )
668
+
669
+ def get_variation_names(self) -> list[bytes]:
670
+ """
671
+ :returns: A list of the named styles in a variation font.
672
+ :exception OSError: If the font is not a variation font.
673
+ """
674
+ names = self.font.getvarnames()
675
+ return [name.replace(b"\x00", b"") for name in names]
676
+
677
+ def set_variation_by_name(self, name: str | bytes) -> None:
678
+ """
679
+ :param name: The name of the style.
680
+ :exception OSError: If the font is not a variation font.
681
+ """
682
+ names = self.get_variation_names()
683
+ if not isinstance(name, bytes):
684
+ name = name.encode()
685
+ index = names.index(name) + 1
686
+
687
+ if index == getattr(self, "_last_variation_index", None):
688
+ # When the same name is set twice in a row,
689
+ # there is an 'unknown freetype error'
690
+ # https://savannah.nongnu.org/bugs/?56186
691
+ return
692
+ self._last_variation_index = index
693
+
694
+ self.font.setvarname(index)
695
+
696
+ def get_variation_axes(self) -> list[Axis]:
697
+ """
698
+ :returns: A list of the axes in a variation font.
699
+ :exception OSError: If the font is not a variation font.
700
+ """
701
+ axes = self.font.getvaraxes()
702
+ for axis in axes:
703
+ if axis["name"]:
704
+ axis["name"] = axis["name"].replace(b"\x00", b"")
705
+ return axes
706
+
707
+ def set_variation_by_axes(self, axes: list[float]) -> None:
708
+ """
709
+ :param axes: A list of values for each axis.
710
+ :exception OSError: If the font is not a variation font.
711
+ """
712
+ self.font.setvaraxes(axes)
713
+
714
+
715
+ class TransposedFont:
716
+ """Wrapper for writing rotated or mirrored text"""
717
+
718
+ def __init__(
719
+ self, font: ImageFont | FreeTypeFont, orientation: Image.Transpose | None = None
720
+ ):
721
+ """
722
+ Wrapper that creates a transposed font from any existing font
723
+ object.
724
+
725
+ :param font: A font object.
726
+ :param orientation: An optional orientation. If given, this should
727
+ be one of Image.Transpose.FLIP_LEFT_RIGHT, Image.Transpose.FLIP_TOP_BOTTOM,
728
+ Image.Transpose.ROTATE_90, Image.Transpose.ROTATE_180, or
729
+ Image.Transpose.ROTATE_270.
730
+ """
731
+ self.font = font
732
+ self.orientation = orientation # any 'transpose' argument, or None
733
+
734
+ def getmask(
735
+ self, text: str | bytes, mode: str = "", *args: Any, **kwargs: Any
736
+ ) -> Image.core.ImagingCore:
737
+ im = self.font.getmask(text, mode, *args, **kwargs)
738
+ if self.orientation is not None:
739
+ return im.transpose(self.orientation)
740
+ return im
741
+
742
+ def getbbox(
743
+ self, text: str | bytes, *args: Any, **kwargs: Any
744
+ ) -> tuple[int, int, float, float]:
745
+ # TransposedFont doesn't support getmask2, move top-left point to (0, 0)
746
+ # this has no effect on ImageFont and simulates anchor="lt" for FreeTypeFont
747
+ left, top, right, bottom = self.font.getbbox(text, *args, **kwargs)
748
+ width = right - left
749
+ height = bottom - top
750
+ if self.orientation in (Image.Transpose.ROTATE_90, Image.Transpose.ROTATE_270):
751
+ return 0, 0, height, width
752
+ return 0, 0, width, height
753
+
754
+ def getlength(self, text: str | bytes, *args: Any, **kwargs: Any) -> float:
755
+ if self.orientation in (Image.Transpose.ROTATE_90, Image.Transpose.ROTATE_270):
756
+ msg = "text length is undefined for text rotated by 90 or 270 degrees"
757
+ raise ValueError(msg)
758
+ return self.font.getlength(text, *args, **kwargs)
759
+
760
+
761
+ def load(filename: str) -> ImageFont:
762
+ """
763
+ Load a font file. This function loads a font object from the given
764
+ bitmap font file, and returns the corresponding font object. For loading TrueType
765
+ or OpenType fonts instead, see :py:func:`~PIL.ImageFont.truetype`.
766
+
767
+ :param filename: Name of font file.
768
+ :return: A font object.
769
+ :exception OSError: If the file could not be read.
770
+ """
771
+ f = ImageFont()
772
+ f._load_pilfont(filename)
773
+ return f
774
+
775
+
776
+ def truetype(
777
+ font: StrOrBytesPath | BinaryIO,
778
+ size: float = 10,
779
+ index: int = 0,
780
+ encoding: str = "",
781
+ layout_engine: Layout | None = None,
782
+ ) -> FreeTypeFont:
783
+ """
784
+ Load a TrueType or OpenType font from a file or file-like object,
785
+ and create a font object. This function loads a font object from the given
786
+ file or file-like object, and creates a font object for a font of the given
787
+ size. For loading bitmap fonts instead, see :py:func:`~PIL.ImageFont.load`
788
+ and :py:func:`~PIL.ImageFont.load_path`.
789
+
790
+ Pillow uses FreeType to open font files. On Windows, be aware that FreeType
791
+ will keep the file open as long as the FreeTypeFont object exists. Windows
792
+ limits the number of files that can be open in C at once to 512, so if many
793
+ fonts are opened simultaneously and that limit is approached, an
794
+ ``OSError`` may be thrown, reporting that FreeType "cannot open resource".
795
+ A workaround would be to copy the file(s) into memory, and open that instead.
796
+
797
+ This function requires the _imagingft service.
798
+
799
+ :param font: A filename or file-like object containing a TrueType font.
800
+ If the file is not found in this filename, the loader may also
801
+ search in other directories, such as:
802
+
803
+ * The :file:`fonts/` directory on Windows,
804
+ * :file:`/Library/Fonts/`, :file:`/System/Library/Fonts/`
805
+ and :file:`~/Library/Fonts/` on macOS.
806
+ * :file:`~/.local/share/fonts`, :file:`/usr/local/share/fonts`,
807
+ and :file:`/usr/share/fonts` on Linux; or those specified by
808
+ the ``XDG_DATA_HOME`` and ``XDG_DATA_DIRS`` environment variables
809
+ for user-installed and system-wide fonts, respectively.
810
+
811
+ :param size: The requested size, in pixels.
812
+ :param index: Which font face to load (default is first available face).
813
+ :param encoding: Which font encoding to use (default is Unicode). Possible
814
+ encodings include (see the FreeType documentation for more
815
+ information):
816
+
817
+ * "unic" (Unicode)
818
+ * "symb" (Microsoft Symbol)
819
+ * "ADOB" (Adobe Standard)
820
+ * "ADBE" (Adobe Expert)
821
+ * "ADBC" (Adobe Custom)
822
+ * "armn" (Apple Roman)
823
+ * "sjis" (Shift JIS)
824
+ * "gb " (PRC)
825
+ * "big5"
826
+ * "wans" (Extended Wansung)
827
+ * "joha" (Johab)
828
+ * "lat1" (Latin-1)
829
+
830
+ This specifies the character set to use. It does not alter the
831
+ encoding of any text provided in subsequent operations.
832
+ :param layout_engine: Which layout engine to use, if available:
833
+ :attr:`.ImageFont.Layout.BASIC` or :attr:`.ImageFont.Layout.RAQM`.
834
+ If it is available, Raqm layout will be used by default.
835
+ Otherwise, basic layout will be used.
836
+
837
+ Raqm layout is recommended for all non-English text. If Raqm layout
838
+ is not required, basic layout will have better performance.
839
+
840
+ You can check support for Raqm layout using
841
+ :py:func:`PIL.features.check_feature` with ``feature="raqm"``.
842
+
843
+ .. versionadded:: 4.2.0
844
+ :return: A font object.
845
+ :exception OSError: If the file could not be read.
846
+ :exception ValueError: If the font size is not greater than zero.
847
+ """
848
+
849
+ def freetype(font: StrOrBytesPath | BinaryIO) -> FreeTypeFont:
850
+ return FreeTypeFont(font, size, index, encoding, layout_engine)
851
+
852
+ try:
853
+ return freetype(font)
854
+ except OSError:
855
+ if not is_path(font):
856
+ raise
857
+ ttf_filename = os.path.basename(font)
858
+
859
+ dirs = []
860
+ if sys.platform == "win32":
861
+ # check the windows font repository
862
+ # NOTE: must use uppercase WINDIR, to work around bugs in
863
+ # 1.5.2's os.environ.get()
864
+ windir = os.environ.get("WINDIR")
865
+ if windir:
866
+ dirs.append(os.path.join(windir, "fonts"))
867
+ elif sys.platform in ("linux", "linux2"):
868
+ data_home = os.environ.get("XDG_DATA_HOME")
869
+ if not data_home:
870
+ # The freedesktop spec defines the following default directory for
871
+ # when XDG_DATA_HOME is unset or empty. This user-level directory
872
+ # takes precedence over system-level directories.
873
+ data_home = os.path.expanduser("~/.local/share")
874
+ xdg_dirs = [data_home]
875
+
876
+ data_dirs = os.environ.get("XDG_DATA_DIRS")
877
+ if not data_dirs:
878
+ # Similarly, defaults are defined for the system-level directories
879
+ data_dirs = "/usr/local/share:/usr/share"
880
+ xdg_dirs += data_dirs.split(":")
881
+
882
+ dirs += [os.path.join(xdg_dir, "fonts") for xdg_dir in xdg_dirs]
883
+ elif sys.platform == "darwin":
884
+ dirs += [
885
+ "/Library/Fonts",
886
+ "/System/Library/Fonts",
887
+ os.path.expanduser("~/Library/Fonts"),
888
+ ]
889
+
890
+ ext = os.path.splitext(ttf_filename)[1]
891
+ first_font_with_a_different_extension = None
892
+ for directory in dirs:
893
+ for walkroot, walkdir, walkfilenames in os.walk(directory):
894
+ for walkfilename in walkfilenames:
895
+ if ext and walkfilename == ttf_filename:
896
+ return freetype(os.path.join(walkroot, walkfilename))
897
+ elif not ext and os.path.splitext(walkfilename)[0] == ttf_filename:
898
+ fontpath = os.path.join(walkroot, walkfilename)
899
+ if os.path.splitext(fontpath)[1] == ".ttf":
900
+ return freetype(fontpath)
901
+ if not ext and first_font_with_a_different_extension is None:
902
+ first_font_with_a_different_extension = fontpath
903
+ if first_font_with_a_different_extension:
904
+ return freetype(first_font_with_a_different_extension)
905
+ raise
906
+
907
+
908
+ def load_path(filename: str | bytes) -> ImageFont:
909
+ """
910
+ Load font file. Same as :py:func:`~PIL.ImageFont.load`, but searches for a
911
+ bitmap font along the Python path.
912
+
913
+ :param filename: Name of font file.
914
+ :return: A font object.
915
+ :exception OSError: If the file could not be read.
916
+ """
917
+ if not isinstance(filename, str):
918
+ filename = filename.decode("utf-8")
919
+ for directory in sys.path:
920
+ try:
921
+ return load(os.path.join(directory, filename))
922
+ except OSError:
923
+ pass
924
+ msg = f'cannot find font file "{filename}" in sys.path'
925
+ if os.path.exists(filename):
926
+ msg += f', did you mean ImageFont.load("{filename}") instead?'
927
+
928
+ raise OSError(msg)
929
+
930
+
931
+ def load_default_imagefont() -> ImageFont:
932
+ f = ImageFont()
933
+ f._load_pilfont_data(
934
+ # courB08
935
+ BytesIO(
936
+ base64.b64decode(
937
+ b"""
938
+ UElMZm9udAo7Ozs7OzsxMDsKREFUQQoAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
939
+ AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
940
+ AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
941
+ AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
942
+ AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
943
+ AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
944
+ AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
945
+ AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
946
+ AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
947
+ AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
948
+ AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
949
+ AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAYAAAAA//8AAQAAAAAAAAABAAEA
950
+ BgAAAAH/+gADAAAAAQAAAAMABgAGAAAAAf/6AAT//QADAAAABgADAAYAAAAA//kABQABAAYAAAAL
951
+ AAgABgAAAAD/+AAFAAEACwAAABAACQAGAAAAAP/5AAUAAAAQAAAAFQAHAAYAAP////oABQAAABUA
952
+ AAAbAAYABgAAAAH/+QAE//wAGwAAAB4AAwAGAAAAAf/5AAQAAQAeAAAAIQAIAAYAAAAB//kABAAB
953
+ ACEAAAAkAAgABgAAAAD/+QAE//0AJAAAACgABAAGAAAAAP/6AAX//wAoAAAALQAFAAYAAAAB//8A
954
+ BAACAC0AAAAwAAMABgAAAAD//AAF//0AMAAAADUAAQAGAAAAAf//AAMAAAA1AAAANwABAAYAAAAB
955
+ //kABQABADcAAAA7AAgABgAAAAD/+QAFAAAAOwAAAEAABwAGAAAAAP/5AAYAAABAAAAARgAHAAYA
956
+ AAAA//kABQAAAEYAAABLAAcABgAAAAD/+QAFAAAASwAAAFAABwAGAAAAAP/5AAYAAABQAAAAVgAH
957
+ AAYAAAAA//kABQAAAFYAAABbAAcABgAAAAD/+QAFAAAAWwAAAGAABwAGAAAAAP/5AAUAAABgAAAA
958
+ ZQAHAAYAAAAA//kABQAAAGUAAABqAAcABgAAAAD/+QAFAAAAagAAAG8ABwAGAAAAAf/8AAMAAABv
959
+ AAAAcQAEAAYAAAAA//wAAwACAHEAAAB0AAYABgAAAAD/+gAE//8AdAAAAHgABQAGAAAAAP/7AAT/
960
+ /gB4AAAAfAADAAYAAAAB//oABf//AHwAAACAAAUABgAAAAD/+gAFAAAAgAAAAIUABgAGAAAAAP/5
961
+ AAYAAQCFAAAAiwAIAAYAAP////oABgAAAIsAAACSAAYABgAA////+gAFAAAAkgAAAJgABgAGAAAA
962
+ AP/6AAUAAACYAAAAnQAGAAYAAP////oABQAAAJ0AAACjAAYABgAA////+gAFAAAAowAAAKkABgAG
963
+ AAD////6AAUAAACpAAAArwAGAAYAAAAA//oABQAAAK8AAAC0AAYABgAA////+gAGAAAAtAAAALsA
964
+ BgAGAAAAAP/6AAQAAAC7AAAAvwAGAAYAAP////oABQAAAL8AAADFAAYABgAA////+gAGAAAAxQAA
965
+ AMwABgAGAAD////6AAUAAADMAAAA0gAGAAYAAP////oABQAAANIAAADYAAYABgAA////+gAGAAAA
966
+ 2AAAAN8ABgAGAAAAAP/6AAUAAADfAAAA5AAGAAYAAP////oABQAAAOQAAADqAAYABgAAAAD/+gAF
967
+ AAEA6gAAAO8ABwAGAAD////6AAYAAADvAAAA9gAGAAYAAAAA//oABQAAAPYAAAD7AAYABgAA////
968
+ +gAFAAAA+wAAAQEABgAGAAD////6AAYAAAEBAAABCAAGAAYAAP////oABgAAAQgAAAEPAAYABgAA
969
+ ////+gAGAAABDwAAARYABgAGAAAAAP/6AAYAAAEWAAABHAAGAAYAAP////oABgAAARwAAAEjAAYA
970
+ BgAAAAD/+gAFAAABIwAAASgABgAGAAAAAf/5AAQAAQEoAAABKwAIAAYAAAAA//kABAABASsAAAEv
971
+ AAgABgAAAAH/+QAEAAEBLwAAATIACAAGAAAAAP/5AAX//AEyAAABNwADAAYAAAAAAAEABgACATcA
972
+ AAE9AAEABgAAAAH/+QAE//wBPQAAAUAAAwAGAAAAAP/7AAYAAAFAAAABRgAFAAYAAP////kABQAA
973
+ AUYAAAFMAAcABgAAAAD/+wAFAAABTAAAAVEABQAGAAAAAP/5AAYAAAFRAAABVwAHAAYAAAAA//sA
974
+ BQAAAVcAAAFcAAUABgAAAAD/+QAFAAABXAAAAWEABwAGAAAAAP/7AAYAAgFhAAABZwAHAAYAAP//
975
+ //kABQAAAWcAAAFtAAcABgAAAAD/+QAGAAABbQAAAXMABwAGAAAAAP/5AAQAAgFzAAABdwAJAAYA
976
+ AP////kABgAAAXcAAAF+AAcABgAAAAD/+QAGAAABfgAAAYQABwAGAAD////7AAUAAAGEAAABigAF
977
+ AAYAAP////sABQAAAYoAAAGQAAUABgAAAAD/+wAFAAABkAAAAZUABQAGAAD////7AAUAAgGVAAAB
978
+ mwAHAAYAAAAA//sABgACAZsAAAGhAAcABgAAAAD/+wAGAAABoQAAAacABQAGAAAAAP/7AAYAAAGn
979
+ AAABrQAFAAYAAAAA//kABgAAAa0AAAGzAAcABgAA////+wAGAAABswAAAboABQAGAAD////7AAUA
980
+ AAG6AAABwAAFAAYAAP////sABgAAAcAAAAHHAAUABgAAAAD/+wAGAAABxwAAAc0ABQAGAAD////7
981
+ AAYAAgHNAAAB1AAHAAYAAAAA//sABQAAAdQAAAHZAAUABgAAAAH/+QAFAAEB2QAAAd0ACAAGAAAA
982
+ Av/6AAMAAQHdAAAB3gAHAAYAAAAA//kABAABAd4AAAHiAAgABgAAAAD/+wAF//0B4gAAAecAAgAA
983
+ AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
984
+ AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
985
+ AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
986
+ AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
987
+ AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
988
+ AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
989
+ AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
990
+ AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
991
+ AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
992
+ AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
993
+ AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
994
+ AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAYAAAAB
995
+ //sAAwACAecAAAHpAAcABgAAAAD/+QAFAAEB6QAAAe4ACAAGAAAAAP/5AAYAAAHuAAAB9AAHAAYA
996
+ AAAA//oABf//AfQAAAH5AAUABgAAAAD/+QAGAAAB+QAAAf8ABwAGAAAAAv/5AAMAAgH/AAACAAAJ
997
+ AAYAAAAA//kABQABAgAAAAIFAAgABgAAAAH/+gAE//sCBQAAAggAAQAGAAAAAP/5AAYAAAIIAAAC
998
+ DgAHAAYAAAAB//kABf/+Ag4AAAISAAUABgAA////+wAGAAACEgAAAhkABQAGAAAAAP/7AAX//gIZ
999
+ AAACHgADAAYAAAAA//wABf/9Ah4AAAIjAAEABgAAAAD/+QAHAAACIwAAAioABwAGAAAAAP/6AAT/
1000
+ +wIqAAACLgABAAYAAAAA//kABP/8Ai4AAAIyAAMABgAAAAD/+gAFAAACMgAAAjcABgAGAAAAAf/5
1001
+ AAT//QI3AAACOgAEAAYAAAAB//kABP/9AjoAAAI9AAQABgAAAAL/+QAE//sCPQAAAj8AAgAGAAD/
1002
+ ///7AAYAAgI/AAACRgAHAAYAAAAA//kABgABAkYAAAJMAAgABgAAAAH//AAD//0CTAAAAk4AAQAG
1003
+ AAAAAf//AAQAAgJOAAACUQADAAYAAAAB//kABP/9AlEAAAJUAAQABgAAAAH/+QAF//4CVAAAAlgA
1004
+ BQAGAAD////7AAYAAAJYAAACXwAFAAYAAP////kABgAAAl8AAAJmAAcABgAA////+QAGAAACZgAA
1005
+ Am0ABwAGAAD////5AAYAAAJtAAACdAAHAAYAAAAA//sABQACAnQAAAJ5AAcABgAA////9wAGAAAC
1006
+ eQAAAoAACQAGAAD////3AAYAAAKAAAAChwAJAAYAAP////cABgAAAocAAAKOAAkABgAA////9wAG
1007
+ AAACjgAAApUACQAGAAD////4AAYAAAKVAAACnAAIAAYAAP////cABgAAApwAAAKjAAkABgAA////
1008
+ +gAGAAACowAAAqoABgAGAAAAAP/6AAUAAgKqAAACrwAIAAYAAP////cABQAAAq8AAAK1AAkABgAA
1009
+ ////9wAFAAACtQAAArsACQAGAAD////3AAUAAAK7AAACwQAJAAYAAP////gABQAAAsEAAALHAAgA
1010
+ BgAAAAD/9wAEAAACxwAAAssACQAGAAAAAP/3AAQAAALLAAACzwAJAAYAAAAA//cABAAAAs8AAALT
1011
+ AAkABgAAAAD/+AAEAAAC0wAAAtcACAAGAAD////6AAUAAALXAAAC3QAGAAYAAP////cABgAAAt0A
1012
+ AALkAAkABgAAAAD/9wAFAAAC5AAAAukACQAGAAAAAP/3AAUAAALpAAAC7gAJAAYAAAAA//cABQAA
1013
+ Au4AAALzAAkABgAAAAD/9wAFAAAC8wAAAvgACQAGAAAAAP/4AAUAAAL4AAAC/QAIAAYAAAAA//oA
1014
+ Bf//Av0AAAMCAAUABgAA////+gAGAAADAgAAAwkABgAGAAD////3AAYAAAMJAAADEAAJAAYAAP//
1015
+ //cABgAAAxAAAAMXAAkABgAA////9wAGAAADFwAAAx4ACQAGAAD////4AAYAAAAAAAoABwASAAYA
1016
+ AP////cABgAAAAcACgAOABMABgAA////+gAFAAAADgAKABQAEAAGAAD////6AAYAAAAUAAoAGwAQ
1017
+ AAYAAAAA//gABgAAABsACgAhABIABgAAAAD/+AAGAAAAIQAKACcAEgAGAAAAAP/4AAYAAAAnAAoA
1018
+ LQASAAYAAAAA//gABgAAAC0ACgAzABIABgAAAAD/+QAGAAAAMwAKADkAEQAGAAAAAP/3AAYAAAA5
1019
+ AAoAPwATAAYAAP////sABQAAAD8ACgBFAA8ABgAAAAD/+wAFAAIARQAKAEoAEQAGAAAAAP/4AAUA
1020
+ AABKAAoATwASAAYAAAAA//gABQAAAE8ACgBUABIABgAAAAD/+AAFAAAAVAAKAFkAEgAGAAAAAP/5
1021
+ AAUAAABZAAoAXgARAAYAAAAA//gABgAAAF4ACgBkABIABgAAAAD/+AAGAAAAZAAKAGoAEgAGAAAA
1022
+ AP/4AAYAAABqAAoAcAASAAYAAAAA//kABgAAAHAACgB2ABEABgAAAAD/+AAFAAAAdgAKAHsAEgAG
1023
+ AAD////4AAYAAAB7AAoAggASAAYAAAAA//gABQAAAIIACgCHABIABgAAAAD/+AAFAAAAhwAKAIwA
1024
+ EgAGAAAAAP/4AAUAAACMAAoAkQASAAYAAAAA//gABQAAAJEACgCWABIABgAAAAD/+QAFAAAAlgAK
1025
+ AJsAEQAGAAAAAP/6AAX//wCbAAoAoAAPAAYAAAAA//oABQABAKAACgClABEABgAA////+AAGAAAA
1026
+ pQAKAKwAEgAGAAD////4AAYAAACsAAoAswASAAYAAP////gABgAAALMACgC6ABIABgAA////+QAG
1027
+ AAAAugAKAMEAEQAGAAD////4AAYAAgDBAAoAyAAUAAYAAP////kABQACAMgACgDOABMABgAA////
1028
+ +QAGAAIAzgAKANUAEw==
1029
+ """
1030
+ )
1031
+ ),
1032
+ Image.open(
1033
+ BytesIO(
1034
+ base64.b64decode(
1035
+ b"""
1036
+ iVBORw0KGgoAAAANSUhEUgAAAx4AAAAUAQAAAAArMtZoAAAEwElEQVR4nABlAJr/AHVE4czCI/4u
1037
+ Mc4b7vuds/xzjz5/3/7u/n9vMe7vnfH/9++vPn/xyf5zhxzjt8GHw8+2d83u8x27199/nxuQ6Od9
1038
+ M43/5z2I+9n9ZtmDBwMQECDRQw/eQIQohJXxpBCNVE6QCCAAAAD//wBlAJr/AgALyj1t/wINwq0g
1039
+ LeNZUworuN1cjTPIzrTX6ofHWeo3v336qPzfEwRmBnHTtf95/fglZK5N0PDgfRTslpGBvz7LFc4F
1040
+ IUXBWQGjQ5MGCx34EDFPwXiY4YbYxavpnhHFrk14CDAAAAD//wBlAJr/AgKqRooH2gAgPeggvUAA
1041
+ Bu2WfgPoAwzRAABAAAAAAACQgLz/3Uv4Gv+gX7BJgDeeGP6AAAD1NMDzKHD7ANWr3loYbxsAD791
1042
+ NAADfcoIDyP44K/jv4Y63/Z+t98Ovt+ub4T48LAAAAD//wBlAJr/AuplMlADJAAAAGuAphWpqhMx
1043
+ in0A/fRvAYBABPgBwBUgABBQ/sYAyv9g0bCHgOLoGAAAAAAAREAAwI7nr0ArYpow7aX8//9LaP/9
1044
+ SjdavWA8ePHeBIKB//81/83ndznOaXx379wAAAD//wBlAJr/AqDxW+D3AABAAbUh/QMnbQag/gAY
1045
+ AYDAAACgtgD/gOqAAAB5IA/8AAAk+n9w0AAA8AAAmFRJuPo27ciC0cD5oeW4E7KA/wD3ECMAn2tt
1046
+ y8PgwH8AfAxFzC0JzeAMtratAsC/ffwAAAD//wBlAJr/BGKAyCAA4AAAAvgeYTAwHd1kmQF5chkG
1047
+ ABoMIHcL5xVpTfQbUqzlAAAErwAQBgAAEOClA5D9il08AEh/tUzdCBsXkbgACED+woQg8Si9VeqY
1048
+ lODCn7lmF6NhnAEYgAAA/NMIAAAAAAD//2JgjLZgVGBg5Pv/Tvpc8hwGBjYGJADjHDrAwPzAjv/H
1049
+ /Wf3PzCwtzcwHmBgYGcwbZz8wHaCAQMDOwMDQ8MCBgYOC3W7mp+f0w+wHOYxO3OG+e376hsMZjk3
1050
+ AAAAAP//YmCMY2A4wMAIN5e5gQETPD6AZisDAwMDgzSDAAPjByiHcQMDAwMDg1nOze1lByRu5/47
1051
+ c4859311AYNZzg0AAAAA//9iYGDBYihOIIMuwIjGL39/fwffA8b//xv/P2BPtzzHwCBjUQAAAAD/
1052
+ /yLFBrIBAAAA//9i1HhcwdhizX7u8NZNzyLbvT97bfrMf/QHI8evOwcSqGUJAAAA//9iYBB81iSw
1053
+ pEE170Qrg5MIYydHqwdDQRMrAwcVrQAAAAD//2J4x7j9AAMDn8Q/BgYLBoaiAwwMjPdvMDBYM1Tv
1054
+ oJodAAAAAP//Yqo/83+dxePWlxl3npsel9lvLfPcqlE9725C+acfVLMEAAAA//9i+s9gwCoaaGMR
1055
+ evta/58PTEWzr21hufPjA8N+qlnBwAAAAAD//2JiWLci5v1+HmFXDqcnULE/MxgYGBj+f6CaJQAA
1056
+ AAD//2Ji2FrkY3iYpYC5qDeGgeEMAwPDvwQBBoYvcTwOVLMEAAAA//9isDBgkP///0EOg9z35v//
1057
+ Gc/eeW7BwPj5+QGZhANUswMAAAD//2JgqGBgYGBgqEMXlvhMPUsAAAAA//8iYDd1AAAAAP//AwDR
1058
+ w7IkEbzhVQAAAABJRU5ErkJggg==
1059
+ """
1060
+ )
1061
+ )
1062
+ ),
1063
+ )
1064
+ return f
1065
+
1066
+
1067
+ def load_default(size: float | None = None) -> FreeTypeFont | ImageFont:
1068
+ """If FreeType support is available, load a version of Aileron Regular,
1069
+ https://dotcolon.net/fonts/aileron, with a more limited character set.
1070
+
1071
+ Otherwise, load a "better than nothing" font.
1072
+
1073
+ .. versionadded:: 1.1.4
1074
+
1075
+ :param size: The font size of Aileron Regular.
1076
+
1077
+ .. versionadded:: 10.1.0
1078
+
1079
+ :return: A font object.
1080
+ """
1081
+ if isinstance(core, ModuleType) or size is not None:
1082
+ return truetype(
1083
+ BytesIO(
1084
+ base64.b64decode(
1085
+ b"""
1086
+ AAEAAAAPAIAAAwBwRkZUTYwDlUAAADFoAAAAHEdERUYAqADnAAAo8AAAACRHUE9ThhmITwAAKfgAA
1087
+ AduR1NVQnHxefoAACkUAAAA4k9TLzJovoHLAAABeAAAAGBjbWFw5lFQMQAAA6gAAAGqZ2FzcP//AA
1088
+ MAACjoAAAACGdseWYmRXoPAAAGQAAAHfhoZWFkE18ayQAAAPwAAAA2aGhlYQboArEAAAE0AAAAJGh
1089
+ tdHjjERZ8AAAB2AAAAdBsb2NhuOexrgAABVQAAADqbWF4cAC7AEYAAAFYAAAAIG5hbWUr+h5lAAAk
1090
+ OAAAA6Jwb3N0D3oPTQAAJ9wAAAEKAAEAAAABGhxJDqIhXw889QALA+gAAAAA0Bqf2QAAAADhCh2h/
1091
+ 2r/LgOxAyAAAAAIAAIAAAAAAAAAAQAAA8r/GgAAA7j/av9qA7EAAQAAAAAAAAAAAAAAAAAAAHQAAQ
1092
+ AAAHQAQwAFAAAAAAACAAAAAQABAAAAQAAAAAAAAAADAfoBkAAFAAgCigJYAAAASwKKAlgAAAFeADI
1093
+ BPgAAAAAFAAAAAAAAAAAAAAcAAAAAAAAAAAAAAABVS1dOAEAAIPsCAwL/GgDIA8oA5iAAAJMAAAAA
1094
+ AhICsgAAACAAAwH0AAAAAAAAAU0AAADYAAAA8gA5AVMAVgJEAEYCRAA1AuQAKQKOAEAAsAArATsAZ
1095
+ AE7AB4CMABVAkQAUADc/+EBEgAgANwAJQEv//sCRAApAkQAggJEADwCRAAtAkQAIQJEADkCRAArAk
1096
+ QAMgJEACwCRAAxANwAJQDc/+ECRABnAkQAUAJEAEQB8wAjA1QANgJ/AB0CcwBkArsALwLFAGQCSwB
1097
+ kAjcAZALGAC8C2gBkAQgAZAIgADcCYQBkAj8AZANiAGQCzgBkAuEALwJWAGQC3QAvAmsAZAJJADQC
1098
+ ZAAiAqoAXgJuACADuAAaAnEAGQJFABMCTwAuATMAYgEv//sBJwAiAkQAUAH0ADIBLAApAhMAJAJjA
1099
+ EoCEQAeAmcAHgIlAB4BIgAVAmcAHgJRAEoA7gA+AOn/8wIKAEoA9wBGA1cASgJRAEoCSgAeAmMASg
1100
+ JnAB4BSgBKAcsAGAE5ABQCUABCAgIAAQMRAAEB4v/6AgEAAQHOABQBLwBAAPoAYAEvACECRABNA0Y
1101
+ AJAItAHgBKgAcAkQAUAEsAHQAygAgAi0AOQD3ADYA9wAWAaEANgGhABYCbAAlAYMAeAGDADkA6/9q
1102
+ AhsAFAIKABUB/QAVAAAAAwAAAAMAAAAcAAEAAAAAAKQAAwABAAAAHAAEAIgAAAAeABAAAwAOAH4Aq
1103
+ QCrALEAtAC3ALsgGSAdICYgOiBEISL7Av//AAAAIACpAKsAsAC0ALcAuyAYIBwgJiA5IEQhIvsB//
1104
+ //4/+5/7j/tP+y/7D/reBR4E/gR+A14CzfTwVxAAEAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
1105
+ AAAAAAAEGAAABAAAAAAAAAAECAAAAAgAAAAAAAAAAAAAAAAAAAAEAAAMEBQYHCAkKCwwNDg8QERIT
1106
+ FBUWFxgZGhscHR4fICEiIyQlJicoKSorLC0uLzAxMjM0NTY3ODk6Ozw9Pj9AQUJDREVGR0hJSktMT
1107
+ U5PUFFSU1RVVldYWVpbXF1eX2BhAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAGQAAA
1108
+ AAAAAAYnFmAAAAAABlAAAAAAAAAAAAAAAAAAAAAAAAAAAAY2htAAAAAAAAAABrbGlqAAAAAHAAbm9
1109
+ ycwBnAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAmACYAJgAmAD4AUgCCAMoBCgFO
1110
+ AVwBcgGIAaYBvAHKAdYB6AH2AgwCIAJKAogCpgLWAw4DIgNkA5wDugPUA+gD/AQQBEYEogS8BPoFJ
1111
+ gVSBWoFgAWwBcoF1gX6BhQGJAZMBmgGiga0BuIHGgdUB2YHkAeiB8AH3AfyCAoIHAgqCDoITghcCG
1112
+ oIogjSCPoJKglYCXwJwgnqCgIKKApACl4Klgq8CtwLDAs8C1YLjAuyC9oL7gwMDCYMSAxgDKAMrAz
1113
+ qDQoNTA1mDYQNoA2uDcAN2g3oDfYODA4iDkoOXA5sDnoOnA7EDvwAAAAFAAAAAAH0ArwAAwAGAAkA
1114
+ DAAPAAAxESERAxMhExcRASELARETAfT6qv6syKr+jgFUqsiqArz9RAGLAP/+1P8B/v3VAP8BLP4CA
1115
+ P8AAgA5//IAuQKyAAMACwAANyMDMwIyFhQGIiY0oE4MZk84JCQ4JLQB/v3AJDgkJDgAAgBWAeUBPA
1116
+ LfAAMABwAAEyMnMxcjJzOmRgpagkYKWgHl+vr6AAAAAAIARgAAAf4CsgAbAB8AAAEHMxUjByM3Iwc
1117
+ jNyM1MzcjNTM3MwczNzMHMxUrAQczAZgdZXEvOi9bLzovWmYdZXEvOi9bLzovWp9bHlsBn4w429vb
1118
+ 2ziMONvb29s4jAAAAAMANf+mAg4DDAAfACYALAAAJRQGBxUjNS4BJzMeARcRLgE0Njc1MxUeARcjJ
1119
+ icVHgEBFBYXNQ4BExU+ATU0Ag5xWDpgcgRcBz41Xl9oVTpVYwpcC1ttXP6cLTQuM5szOrVRZwlOTQ
1120
+ ZqVzZECAEAGlukZAlOTQdrUG8O7iNlAQgxNhDlCDj+8/YGOjReAAAAAAUAKf/yArsCvAAHAAsAFQA
1121
+ dACcAABIyFhQGIiY0EyMBMwQiBhUUFjI2NTQSMhYUBiImNDYiBhUUFjI2NTR5iFBQiFCVVwHAV/5c
1122
+ OiMjOiPmiFBQiFCxOiMjOiMCvFaSVlaS/ZoCsjIzMC80NC8w/uNWklZWkhozMC80NC8wAAAAAgBA/
1123
+ /ICbgLAACIALgAAARUjEQYjIiY1NDY3LgE1NDYzMhcVJiMiBhUUFhcWOwE1MxUFFBYzMjc1IyIHDg
1124
+ ECbmBcYYOOVkg7R4hsQjY4Q0RNRD4SLDxW/pJUXzksPCkUUk0BgUb+zBVUZ0BkDw5RO1huCkULQzp
1125
+ COAMBcHDHRz0J/AIHRQAAAAEAKwHlAIUC3wADAAATIycze0YKWgHl+gAAAAABAGT/sAEXAwwACQAA
1126
+ EzMGEBcjLgE0Nt06dXU6OUBAAwzG/jDGVePs4wAAAAEAHv+wANEDDAAJAAATMx4BFAYHIzYQHjo5Q
1127
+ EA5OnUDDFXj7ONVxgHQAAAAAQBVAFIB2wHbAA4AAAE3FwcXBycHJzcnNxcnMwEtmxOfcTJjYzJxnx
1128
+ ObCj4BKD07KYolmZkliik7PbMAAQBQAFUB9AIlAAsAAAEjFSM1IzUzNTMVMwH0tTq1tTq1AR/Kyjj
1129
+ OzgAAAAAB/+H/iACMAGQABAAANwcjNzOMWlFOXVrS3AAAAQAgAP8A8gE3AAMAABMjNTPy0tIA/zgA
1130
+ AQAl//IApQByAAcAADYyFhQGIiY0STgkJDgkciQ4JCQ4AAAAAf/7/+IBNALQAAMAABcjEzM5Pvs+H
1131
+ gLuAAAAAAIAKf/yAhsCwAADAAcAABIgECA2IBAgKQHy/g5gATL+zgLA/TJEAkYAAAAAAQCCAAABlg
1132
+ KyAAgAAAERIxEHNTc2MwGWVr6SIygCsv1OAldxW1sWAAEAPAAAAg4CwAAZAAA3IRUhNRM+ATU0JiM
1133
+ iDwEjNz4BMzIWFRQGB7kBUv4x+kI2QTt+EAFWAQp8aGVtSl5GRjEA/0RVLzlLmAoKa3FsUkNxXQAA
1134
+ AAEALf/yAhYCwAAqAAABHgEVFAYjIi8BMxceATMyNjU0KwE1MzI2NTQmIyIGDwEjNz4BMzIWFRQGA
1135
+ YxBSZJo2RUBVgEHV0JBUaQREUBUQzc5TQcBVgEKfGhfcEMBbxJbQl1x0AoKRkZHPn9GSD80QUVCCg
1136
+ pfbGBPOlgAAAACACEAAAIkArIACgAPAAAlIxUjNSE1ATMRMyMRBg8BAiRXVv6qAVZWV60dHLCurq4
1137
+ rAdn+QgFLMibzAAABADn/8gIZArIAHQAAATIWFRQGIyIvATMXFjMyNjU0JiMiByMTIRUhBzc2ATNv
1138
+ d5Fl1RQBVgIad0VSTkVhL1IwAYj+vh8rMAHHgGdtgcUKCoFXTU5bYgGRRvAuHQAAAAACACv/8gITA
1139
+ sAAFwAjAAABMhYVFAYjIhE0NjMyFh8BIycmIyIDNzYTMjY1NCYjIgYVFBYBLmp7imr0l3RZdAgBXA
1140
+ IYZ5wKJzU6QVNJSz5SUAHSgWltiQFGxcNlVQoKdv7sPiz+ZF1LTmJbU0lhAAAAAQAyAAACGgKyAAY
1141
+ AAAEVASMBITUCGv6oXAFL/oECsij9dgJsRgAAAAMALP/xAhgCwAAWACAALAAAAR4BFRQGIyImNTQ2
1142
+ Ny4BNTQ2MhYVFAYmIgYVFBYyNjU0AzI2NTQmIyIGFRQWAZQ5S5BmbIpPOjA7ecp5P2F8Q0J8RIVJS
1143
+ 0pLTEtOAW0TXTxpZ2ZqPF0SE1A3VWVlVTdQ/UU0N0RENzT9/ko+Ok1NOj1LAAIAMf/yAhkCwAAXAC
1144
+ MAAAEyERQGIyImLwEzFxYzMhMHBiMiJjU0NhMyNjU0JiMiBhUUFgEl9Jd0WXQIAVwCGGecCic1SWp
1145
+ 7imo+UlBAQVNJAsD+usXDZVUKCnYBFD4sgWltif5kW1NJYV1LTmIAAAACACX/8gClAiAABwAPAAAS
1146
+ MhYUBiImNBIyFhQGIiY0STgkJDgkJDgkJDgkAiAkOCQkOP52JDgkJDgAAAAC/+H/iAClAiAABwAMA
1147
+ AASMhYUBiImNBMHIzczSTgkJDgkaFpSTl4CICQ4JCQ4/mba5gAAAQBnAB4B+AH0AAYAAAENARUlNS
1148
+ UB+P6qAVb+bwGRAbCmpkbJRMkAAAIAUAC7AfQBuwADAAcAAAEhNSERITUhAfT+XAGk/lwBpAGDOP8
1149
+ AOAABAEQAHgHVAfQABgAAARUFNS0BNQHV/m8BVv6qAStEyUSmpkYAAAAAAgAj//IB1ALAABgAIAAA
1150
+ ATIWFRQHDgEHIz4BNz4BNTQmIyIGByM+ARIyFhQGIiY0AQRibmktIAJWBSEqNig+NTlHBFoDezQ4J
1151
+ CQ4JALAZ1BjaS03JS1DMD5LLDQ/SUVgcv2yJDgkJDgAAAAAAgA2/5gDFgKYADYAQgAAAQMGFRQzMj
1152
+ Y1NCYjIg4CFRQWMzI2NxcGIyImNTQ+AjMyFhUUBiMiJwcGIyImNTQ2MzIfATcHNzYmIyIGFRQzMjY
1153
+ Cej8EJjJJlnBAfGQ+oHtAhjUYg5OPx0h2k06Os3xRWQsVLjY5VHtdPBwJETcJDyUoOkZEJz8B0f74
1154
+ EQ8kZl6EkTFZjVOLlyknMVm1pmCiaTq4lX6CSCknTVRmmR8wPdYnQzxuSWVGAAIAHQAAAncCsgAHA
1155
+ AoAACUjByMTMxMjATMDAcj+UVz4dO5d/sjPZPT0ArL9TgE6ATQAAAADAGQAAAJMArIAEAAbACcAAA
1156
+ EeARUUBgcGKwERMzIXFhUUJRUzMjc2NTQnJiMTPgE1NCcmKwEVMzIBvkdHZkwiNt7LOSGq/oeFHBt
1157
+ hahIlSTM+cB8Yj5UWAW8QT0VYYgwFArIEF5Fv1eMED2NfDAL93AU+N24PBP0AAAAAAQAv//ICjwLA
1158
+ ABsAAAEyFh8BIycmIyIGFRQWMzI/ATMHDgEjIiY1NDYBdX+PCwFWAiKiaHx5ZaIiAlYBCpWBk6a0A
1159
+ sCAagoKpqN/gaOmCgplhcicn8sAAAIAZAAAAp8CsgAMABkAAAEeARUUBgcGKwERMzITPgE1NCYnJi
1160
+ sBETMyAY59lJp8IzXN0jUVWmdjWRs5d3I4Aq4QqJWUug8EArL9mQ+PeHGHDgX92gAAAAABAGQAAAI
1161
+ vArIACwAAJRUhESEVIRUhFSEVAi/+NQHB/pUBTf6zRkYCskbwRvAAAAABAGQAAAIlArIACQAAExUh
1162
+ FSERIxEhFboBQ/69VgHBAmzwRv7KArJGAAAAAAEAL//yAo8CwAAfAAABMxEjNQcGIyImNTQ2MzIWH
1163
+ wEjJyYjIgYVFBYzMjY1IwGP90wfPnWTprSSf48LAVYCIqJofHllVG+hAU3+s3hARsicn8uAagoKpq
1164
+ N/gaN1XAAAAAEAZAAAAowCsgALAAABESMRIREjETMRIRECjFb+hFZWAXwCsv1OAS7+0gKy/sQBPAA
1165
+ AAAABAGQAAAC6ArIAAwAAMyMRM7pWVgKyAAABADf/8gHoArIAEwAAAREUBw4BIyImLwEzFxYzMjc2
1166
+ NREB6AIFcGpgbQIBVgIHfXQKAQKy/lYxIltob2EpKYyEFD0BpwAAAAABAGQAAAJ0ArIACwAACQEjA
1167
+ wcVIxEzEQEzATsBJ3ntQlZWAVVlAWH+nwEnR+ACsv6RAW8AAQBkAAACLwKyAAUAACUVIREzEQIv/j
1168
+ VWRkYCsv2UAAABAGQAAAMUArIAFAAAAREjETQ3BgcDIwMmJxYVESMRMxsBAxRWAiMxemx8NxsCVo7
1169
+ MywKy/U4BY7ZLco7+nAFmoFxLtP6dArL9lwJpAAAAAAEAZAAAAoACsgANAAAhIwEWFREjETMBJjUR
1170
+ MwKAhP67A1aEAUUDVAJeeov+pwKy/aJ5jAFZAAAAAgAv//ICuwLAAAkAEwAAEiAWFRQGICY1NBIyN
1171
+ jU0JiIGFRTbATSsrP7MrNrYenrYegLAxaKhxsahov47nIeIm5uIhwACAGQAAAJHArIADgAYAAABHg
1172
+ EVFAYHBisBESMRMzITNjQnJisBETMyAZRUX2VOHzuAVtY7GlxcGDWIiDUCrgtnVlVpCgT+5gKy/rU
1173
+ V1BUF/vgAAAACAC//zAK9AsAAEgAcAAAlFhcHJiMiBwYjIiY1NDYgFhUUJRQWMjY1NCYiBgI9PUMx
1174
+ UDcfKh8omqysATSs/dR62Hp62HpICTg7NgkHxqGixcWitbWHnJyHiJubAAIAZAAAAlgCsgAXACMAA
1175
+ CUWFyMmJyYnJisBESMRMzIXHgEVFAYHFiUzMjc+ATU0JyYrAQIqDCJfGQwNWhAhglbiOx9QXEY1Tv
1176
+ 6bhDATMj1lGSyMtYgtOXR0BwH+1wKyBApbU0BSESRAAgVAOGoQBAABADT/8gIoAsAAJQAAATIWFyM
1177
+ uASMiBhUUFhceARUUBiMiJiczHgEzMjY1NCYnLgE1NDYBOmd2ClwGS0E6SUNRdW+HZnKKC1wPWkQ9
1178
+ Uk1cZGuEAsBwXUJHNjQ3OhIbZVZZbm5kREo+NT5DFRdYUFdrAAAAAAEAIgAAAmQCsgAHAAABIxEjE
1179
+ SM1IQJk9lb2AkICbP2UAmxGAAEAXv/yAmQCsgAXAAABERQHDgEiJicmNREzERQXHgEyNjc2NRECZA
1180
+ IIgfCBCAJWAgZYmlgGAgKy/k0qFFxzc1wUKgGz/lUrEkRQUEQSKwGrAAAAAAEAIAAAAnoCsgAGAAA
1181
+ hIwMzGwEzAYJ07l3N1FwCsv2PAnEAAAEAGgAAA7ECsgAMAAABAyMLASMDMxsBMxsBA7HAcZyicrZi
1182
+ kaB0nJkCsv1OAlP9rQKy/ZsCW/2kAmYAAAEAGQAAAm8CsgALAAAhCwEjEwMzGwEzAxMCCsrEY/bkY
1183
+ re+Y/D6AST+3AFcAVb+5gEa/q3+oQAAAQATAAACUQKyAAgAAAERIxEDMxsBMwFdVvRjwLphARD+8A
1184
+ EQAaL+sQFPAAABAC4AAAI5ArIACQAAJRUhNQEhNSEVAQI5/fUBof57Aen+YUZGQgIqRkX92QAAAAA
1185
+ BAGL/sAEFAwwABwAAARUjETMVIxEBBWlpowMMOP0UOANcAAAB//v/4gE0AtAAAwAABSMDMwE0Pvs+
1186
+ HgLuAAAAAQAi/7AAxQMMAAcAABcjNTMRIzUzxaNpaaNQOALsOAABAFAA1wH0AmgABgAAJQsBIxMzE
1187
+ wGwjY1GsESw1wFZ/qcBkf5vAAAAAQAy/6oBwv/iAAMAAAUhNSEBwv5wAZBWOAAAAAEAKQJEALYCsg
1188
+ ADAAATIycztjhVUAJEbgAAAAACACT/8gHQAiAAHQAlAAAhJwcGIyImNTQ2OwE1NCcmIyIHIz4BMzI
1189
+ XFh0BFBcnMjY9ASYVFAF6CR0wVUtgkJoiAgdgaQlaBm1Zrg4DCuQ9R+5MOSFQR1tbDiwUUXBUXowf
1190
+ J8c9SjRORzYSgVwAAAAAAgBK//ICRQLfABEAHgAAATIWFRQGIyImLwEVIxEzETc2EzI2NTQmIyIGH
1191
+ QEUFgFUcYCVbiNJEyNWVigySElcU01JXmECIJd4i5QTEDRJAt/+3jkq/hRuZV55ZWsdX14AAQAe//
1192
+ IB9wIgABgAAAEyFhcjJiMiBhUUFjMyNjczDgEjIiY1NDYBF152DFocbEJXU0A1Rw1aE3pbaoKQAiB
1193
+ oWH5qZm1tPDlaXYuLgZcAAAACAB7/8gIZAt8AEQAeAAABESM1BwYjIiY1NDYzMhYfAREDMjY9ATQm
1194
+ IyIGFRQWAhlWKDJacYCVbiNJEyOnSV5hQUlcUwLf/SFVOSqXeIuUExA0ARb9VWVrHV9ebmVeeQACA
1195
+ B7/8gH9AiAAFQAbAAABFAchHgEzMjY3Mw4BIyImNTQ2MzIWJyIGByEmAf0C/oAGUkA1SwlaD4FXbI
1196
+ WObmt45UBVBwEqDQEYFhNjWD84W16Oh3+akU9aU60AAAEAFQAAARoC8gAWAAATBh0BMxUjESMRIzU
1197
+ zNTQ3PgEzMhcVJqcDbW1WOTkDB0k8Hx5oAngVITRC/jQBzEIsJRs5PwVHEwAAAAIAHv8uAhkCIAAi
1198
+ AC8AAAERFAcOASMiLwEzFx4BMzI2NzY9AQcGIyImNTQ2MzIWHwE1AzI2PQE0JiMiBhUUFgIZAQSEd
1199
+ NwRAVcBBU5DTlUDASgyWnGAlW4jSRMjp0leYUFJXFMCEv5wSh1zeq8KCTI8VU0ZIQk5Kpd4i5QTED
1200
+ RJ/iJlax1fXm5lXnkAAQBKAAACCgLkABcAAAEWFREjETQnLgEHDgEdASMRMxE3NjMyFgIIAlYCBDs
1201
+ 6RVRWViE5UVViAYUbQP7WASQxGzI7AQJyf+kC5P7TPSxUAAACAD4AAACsAsAABwALAAASMhYUBiIm
1202
+ NBMjETNeLiAgLiBiVlYCwCAuICAu/WACEgAC//P/LgCnAsAABwAVAAASMhYUBiImNBcRFAcGIyInN
1203
+ RY3NjURWS4gIC4gYgMLcRwNSgYCAsAgLiAgLo79wCUbZAJGBzMOHgJEAAAAAQBKAAACCALfAAsAAC
1204
+ EnBxUjETMREzMHEwGTwTJWVvdu9/rgN6kC3/4oAQv6/ugAAQBG//wA3gLfAA8AABMRFBceATcVBiM
1205
+ iJicmNRGcAQIcIxkkKi4CAQLf/bkhERoSBD4EJC8SNAJKAAAAAQBKAAADEAIgACQAAAEWFREjETQn
1206
+ JiMiFREjETQnJiMiFREjETMVNzYzMhYXNzYzMhYDCwVWBAxedFYEDF50VlYiJko7ThAvJkpEVAGfI
1207
+ jn+vAEcQyRZ1v76ARxDJFnW/voCEk08HzYtRB9HAAAAAAEASgAAAgoCIAAWAAABFhURIxE0JyYjIg
1208
+ YdASMRMxU3NjMyFgIIAlYCCXBEVVZWITlRVWIBhRtA/tYBJDEbbHR/6QISWz0sVAAAAAACAB7/8gI
1209
+ sAiAABwARAAASIBYUBiAmNBIyNjU0JiIGFRSlAQCHh/8Ah7ieWlqeWgIgn/Cfn/D+s3ZfYHV1YF8A
1210
+ AgBK/zwCRQIgABEAHgAAATIWFRQGIyImLwERIxEzFTc2EzI2NTQmIyIGHQEUFgFUcYCVbiNJEyNWV
1211
+ igySElcU01JXmECIJd4i5QTEDT+8wLWVTkq/hRuZV55ZWsdX14AAgAe/zwCGQIgABEAHgAAAREjEQ
1212
+ cGIyImNTQ2MzIWHwE1AzI2PQE0JiMiBhUUFgIZVigyWnGAlW4jSRMjp0leYUFJXFMCEv0qARk5Kpd
1213
+ 4i5QTEDRJ/iJlax1fXm5lXnkAAQBKAAABPgIeAA0AAAEyFxUmBhURIxEzFTc2ARoWDkdXVlYwIwIe
1214
+ B0EFVlf+0gISU0cYAAEAGP/yAa0CIAAjAAATMhYXIyYjIgYVFBYXHgEVFAYjIiYnMxYzMjY1NCYnL
1215
+ gE1NDbkV2MJWhNdKy04PF1XbVhWbgxaE2ktOjlEUllkAiBaS2MrJCUoEBlPQkhOVFZoKCUmLhIWSE
1216
+ BIUwAAAAEAFP/4ARQCiQAXAAATERQXHgE3FQYjIiYnJjURIzUzNTMVMxWxAQMmMx8qMjMEAUdHVmM
1217
+ BzP7PGw4mFgY/BSwxDjQBNUJ7e0IAAAABAEL/8gICAhIAFwAAAREjNQcGIyImJyY1ETMRFBceATMy
1218
+ Nj0BAgJWITlRT2EKBVYEBkA1RFECEv3uWj4qTToiOQE+/tIlJC43c4DpAAAAAAEAAQAAAfwCEgAGA
1219
+ AABAyMDMxsBAfzJaclfop8CEv3uAhL+LQHTAAABAAEAAAMLAhIADAAAAQMjCwEjAzMbATMbAQMLqW
1220
+ Z2dmapY3t0a3Z7AhL97gG+/kICEv5AAcD+QwG9AAAB//oAAAHWAhIACwAAARMjJwcjEwMzFzczARq
1221
+ 8ZIuKY763ZoWFYwEO/vLV1QEMAQbNzQAAAQAB/y4B+wISABEAAAEDDgEjIic1FjMyNj8BAzMbAQH7
1222
+ 2iFZQB8NDRIpNhQH02GenQIS/cFVUAJGASozEwIt/i4B0gABABQAAAGxAg4ACQAAJRUhNQEhNSEVA
1223
+ QGx/mMBNP7iAYL+zkREQgGIREX+ewAAAAABAED/sAEOAwwALAAAASMiBhUUFxYVFAYHHgEVFAcGFR
1224
+ QWOwEVIyImNTQ3NjU0JzU2NTQnJjU0NjsBAQ4MKiMLDS4pKS4NCyMqDAtERAwLUlILDERECwLUGBk
1225
+ WTlsgKzUFBTcrIFtOFhkYOC87GFVMIkUIOAhFIkxVGDsvAAAAAAEAYP84AJoDIAADAAAXIxEzmjo6
1226
+ yAPoAAEAIf+wAO8DDAAsAAATFQYVFBcWFRQGKwE1MzI2NTQnJjU0NjcuATU0NzY1NCYrATUzMhYVF
1227
+ AcGFRTvUgsMREQLDCojCw0uKSkuDQsjKgwLREQMCwF6OAhFIkxVGDsvOBgZFk5bICs1BQU3KyBbTh
1228
+ YZGDgvOxhVTCJFAAABAE0A3wH2AWQAEwAAATMUIyImJyYjIhUjNDMyFhcWMzIBvjhuGywtQR0xOG4
1229
+ bLC1BHTEBZIURGCNMhREYIwAAAwAk/94DIgLoAAcAEQApAAAAIBYQBiAmECQgBhUUFiA2NTQlMhYX
1230
+ IyYjIgYUFjMyNjczDgEjIiY1NDYBAQFE3d3+vN0CB/7wubkBELn+xVBnD1wSWDo+QTcqOQZcEmZWX
1231
+ HN2Aujg/rbg4AFKpr+Mjb6+jYxbWEldV5ZZNShLVn5na34AAgB4AFIB9AGeAAUACwAAAQcXIyc3Mw
1232
+ cXIyc3AUqJiUmJifOJiUmJiQGepqampqampqYAAAIAHAHSAQ4CwAAHAA8AABIyFhQGIiY0NiIGFBY
1233
+ yNjRgakREakSTNCEhNCECwEJqQkJqCiM4IyM4AAAAAAIAUAAAAfQCCwALAA8AAAEzFSMVIzUjNTM1
1234
+ MxMhNSEBP7W1OrW1OrX+XAGkAVs4tLQ4sP31OAAAAQB0AkQBAQKyAAMAABMjNzOsOD1QAkRuAAAAA
1235
+ AEAIADsAKoBdgAHAAASMhYUBiImNEg6KCg6KAF2KDooKDoAAAIAOQBSAbUBngAFAAsAACUHIzcnMw
1236
+ UHIzcnMwELiUmJiUkBM4lJiYlJ+KampqampqYAAAABADYB5QDhAt8ABAAAEzczByM2Xk1OXQHv8Po
1237
+ AAQAWAeUAwQLfAAQAABMHIzczwV5NTl0C1fD6AAIANgHlAYsC3wAEAAkAABM3MwcjPwEzByM2Xk1O
1238
+ XapeTU5dAe/w+grw+gAAAgAWAeUBawLfAAQACQAAEwcjNzMXByM3M8FeTU5dql5NTl0C1fD6CvD6A
1239
+ AADACX/8gI1AHIABwAPABcAADYyFhQGIiY0NjIWFAYiJjQ2MhYUBiImNEk4JCQ4JOw4JCQ4JOw4JC
1240
+ Q4JHIkOCQkOCQkOCQkOCQkOCQkOAAAAAEAeABSAUoBngAFAAABBxcjJzcBSomJSYmJAZ6mpqamAAA
1241
+ AAAEAOQBSAQsBngAFAAAlByM3JzMBC4lJiYlJ+KampgAAAf9qAAABgQKyAAMAACsBATM/VwHAVwKy
1242
+ AAAAAAIAFAHIAdwClAAHABQAABMVIxUjNSM1BRUjNwcjJxcjNTMXN9pKMkoByDICKzQqATJLKysCl
1243
+ CmjoykBy46KiY3Lm5sAAQAVAAABvALyABgAAAERIxEjESMRIzUzNTQ3NjMyFxUmBgcGHQEBvFbCVj
1244
+ k5AxHHHx5iVgcDAg798gHM/jQBzEIOJRuWBUcIJDAVIRYAAAABABX//AHkAvIAJQAAJR4BNxUGIyI
1245
+ mJyY1ESYjIgcGHQEzFSMRIxEjNTM1NDc2MzIXERQBowIcIxkkKi4CAR4nXgwDbW1WLy8DEbNdOmYa
1246
+ EQQ/BCQvEjQCFQZWFSEWQv40AcxCDiUblhP9uSEAAAAAAAAWAQ4AAQAAAAAAAAATACgAAQAAAAAAA
1247
+ QAHAEwAAQAAAAAAAgAHAGQAAQAAAAAAAwAaAKIAAQAAAAAABAAHAM0AAQAAAAAABQA8AU8AAQAAAA
1248
+ AABgAPAawAAQAAAAAACAALAdQAAQAAAAAACQALAfgAAQAAAAAACwAXAjQAAQAAAAAADAAXAnwAAwA
1249
+ BBAkAAAAmAAAAAwABBAkAAQAOADwAAwABBAkAAgAOAFQAAwABBAkAAwA0AGwAAwABBAkABAAOAL0A
1250
+ AwABBAkABQB4ANUAAwABBAkABgAeAYwAAwABBAkACAAWAbwAAwABBAkACQAWAeAAAwABBAkACwAuA
1251
+ gQAAwABBAkADAAuAkwATgBvACAAUgBpAGcAaAB0AHMAIABSAGUAcwBlAHIAdgBlAGQALgAATm8gUm
1252
+ lnaHRzIFJlc2VydmVkLgAAQQBpAGwAZQByAG8AbgAAQWlsZXJvbgAAUgBlAGcAdQBsAGEAcgAAUmV
1253
+ ndWxhcgAAMQAuADEAMAAyADsAVQBLAFcATgA7AEEAaQBsAGUAcgBvAG4ALQBSAGUAZwB1AGwAYQBy
1254
+ AAAxLjEwMjtVS1dOO0FpbGVyb24tUmVndWxhcgAAQQBpAGwAZQByAG8AbgAAQWlsZXJvbgAAVgBlA
1255
+ HIAcwBpAG8AbgAgADEALgAxADAAMgA7AFAAUwAgADAAMAAxAC4AMQAwADIAOwBoAG8AdABjAG8Abg
1256
+ B2ACAAMQAuADAALgA3ADAAOwBtAGEAawBlAG8AdABmAC4AbABpAGIAMgAuADUALgA1ADgAMwAyADk
1257
+ AAFZlcnNpb24gMS4xMDI7UFMgMDAxLjEwMjtob3Rjb252IDEuMC43MDttYWtlb3RmLmxpYjIuNS41
1258
+ ODMyOQAAQQBpAGwAZQByAG8AbgAtAFIAZQBnAHUAbABhAHIAAEFpbGVyb24tUmVndWxhcgAAUwBvA
1259
+ HIAYQAgAFMAYQBnAGEAbgBvAABTb3JhIFNhZ2FubwAAUwBvAHIAYQAgAFMAYQBnAGEAbgBvAABTb3
1260
+ JhIFNhZ2FubwAAaAB0AHQAcAA6AC8ALwB3AHcAdwAuAGQAbwB0AGMAbwBsAG8AbgAuAG4AZQB0AAB
1261
+ odHRwOi8vd3d3LmRvdGNvbG9uLm5ldAAAaAB0AHQAcAA6AC8ALwB3AHcAdwAuAGQAbwB0AGMAbwBs
1262
+ AG8AbgAuAG4AZQB0AABodHRwOi8vd3d3LmRvdGNvbG9uLm5ldAAAAAACAAAAAAAA/4MAMgAAAAAAA
1263
+ AAAAAAAAAAAAAAAAAAAAHQAAAABAAIAAwAEAAUABgAHAAgACQAKAAsADAANAA4ADwAQABEAEgATAB
1264
+ QAFQAWABcAGAAZABoAGwAcAB0AHgAfACAAIQAiACMAJAAlACYAJwAoACkAKgArACwALQAuAC8AMAA
1265
+ xADIAMwA0ADUANgA3ADgAOQA6ADsAPAA9AD4APwBAAEEAQgBDAEQARQBGAEcASABJAEoASwBMAE0A
1266
+ TgBPAFAAUQBSAFMAVABVAFYAVwBYAFkAWgBbAFwAXQBeAF8AYABhAIsAqQCDAJMAjQDDAKoAtgC3A
1267
+ LQAtQCrAL4AvwC8AIwAwADBAAAAAAAB//8AAgABAAAADAAAABwAAAACAAIAAwBxAAEAcgBzAAIABA
1268
+ AAAAIAAAABAAAACgBMAGYAAkRGTFQADmxhdG4AGgAEAAAAAP//AAEAAAAWAANDQVQgAB5NT0wgABZ
1269
+ ST00gABYAAP//AAEAAAAA//8AAgAAAAEAAmxpZ2EADmxvY2wAFAAAAAEAAQAAAAEAAAACAAYAEAAG
1270
+ AAAAAgASADQABAAAAAEATAADAAAAAgAQABYAAQAcAAAAAQABAE8AAQABAGcAAQABAE8AAwAAAAIAE
1271
+ AAWAAEAHAAAAAEAAQAvAAEAAQBnAAEAAQAvAAEAGgABAAgAAgAGAAwAcwACAE8AcgACAEwAAQABAE
1272
+ kAAAABAAAACgBGAGAAAkRGTFQADmxhdG4AHAAEAAAAAP//AAIAAAABABYAA0NBVCAAFk1PTCAAFlJ
1273
+ PTSAAFgAA//8AAgAAAAEAAmNwc3AADmtlcm4AFAAAAAEAAAAAAAEAAQACAAYADgABAAAAAQASAAIA
1274
+ AAACAB4ANgABAAoABQAFAAoAAgABACQAPQAAAAEAEgAEAAAAAQAMAAEAOP/nAAEAAQAkAAIGigAEA
1275
+ AAFJAXKABoAGQAA//gAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
1276
+ AAAAD/sv+4/+z/7v/MAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
1277
+ AAAAAAAD/xAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA/9T/6AAAAAD/8QAA
1278
+ ABD/vQAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAD/7gAAAAAAAAAAAAAAAAAA//MAA
1279
+ AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAABIAAAAAAAAAAP/5AAAAAAAAAA
1280
+ AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAP/gAAD/4AAAAAAAAAAAAAAAAAA
1281
+ AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA//L/9AAAAAAAAAAAAAAAAAAAAAAA
1282
+ AAAAAAAAAAAA/+gAAAAAAAkAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
1283
+ AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAP/zAAAAAA
1284
+ AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAP/mAAAAAAAAAAAAAAAAAAD
1285
+ /4gAA//AAAAAA//YAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAD/+AAAAAAAAP/OAAAAAAAAAAAAAAAA
1286
+ AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAD/zv/qAAAAAP/0AAAACAAAAAAAAAAAAAAAAAAAAAAAA
1287
+ AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAP/ZAAD/egAA/1kAAAAA/5D/rgAAAAAAAAAAAA
1288
+ AAAAAAAAAAAAAAAAD/9AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
1289
+ AAAAAAAAAAAAAAAAAAAD/8AAA/7b/8P+wAAD/8P/E/98AAAAA/8P/+P/0//oAAAAAAAAAAAAA//gA
1290
+ AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA/+AAAAAAAAAAAAAAA
1291
+ AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAD/w//C/9MAAP/SAAD/9wAAAAAAAA
1292
+ AAAAAAAAAAAAAAAAAAAAAAAAAAAAD/yAAA/+kAAAAA//QAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
1293
+ AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAD/9wAAAAD//QAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
1294
+ AAAAAAAAAAAAAAAAAAAAAP/2AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
1295
+ AAAAAAAAP/cAAAAAAAAAAAAAAAA/7YAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA
1296
+ AAAAAAAAAAAAAAAAAAAAAAAAAAAP/8AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAD/6AAAAAAAAAA
1297
+ AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAQAkAFAAEAAAAAQACwAAABcA
1298
+ BgAAAAAAAAAIAA4AAAAAAAsAEgAAAAAAAAATABkAAwANAAAAAQAJAAAAAAAAAAAAAAAAAAAAGAAAA
1299
+ AAABwAAAAAAAAAAAAAAFQAFAAAAAAAYABgAAAAUAAAACgAAAAwAAgAPABEAFgAAAAAAAAAAAAAAAA
1300
+ AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAFAAEAEQBdAAYAAAAAAAAAAAAAAAAAAAAAAAA
1301
+ AAAAAAAAAAAAAAAAAAAAAAAAAAQAAAAcAAAAAAAAABwAAAAAACAAAAAAAAAAAAAcAAAAHAAAAEwAJ
1302
+ ABUADgAPAAAACwAQAAAAAAAAAAAAAAAAAAUAGAACAAIAAgAAAAIAGAAXAAAAGAAAABYAFgACABYAA
1303
+ gAWAAAAEQADAAoAFAAMAA0ABAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAASAAAAEgAGAAEAHgAkAC
1304
+ YAJwApACoALQAuAC8AMgAzADcAOAA5ADoAPAA9AEUASABOAE8AUgBTAFUAVwBZAFoAWwBcAF0AcwA
1305
+ AAAAAAQAAAADa3tfFAAAAANAan9kAAAAA4QodoQ==
1306
+ """
1307
+ )
1308
+ ),
1309
+ 10 if size is None else size,
1310
+ layout_engine=Layout.BASIC,
1311
+ )
1312
+ return load_default_imagefont()
.venv/Lib/site-packages/PIL/ImageGrab.py ADDED
@@ -0,0 +1,196 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #
2
+ # The Python Imaging Library
3
+ # $Id$
4
+ #
5
+ # screen grabber
6
+ #
7
+ # History:
8
+ # 2001-04-26 fl created
9
+ # 2001-09-17 fl use builtin driver, if present
10
+ # 2002-11-19 fl added grabclipboard support
11
+ #
12
+ # Copyright (c) 2001-2002 by Secret Labs AB
13
+ # Copyright (c) 2001-2002 by Fredrik Lundh
14
+ #
15
+ # See the README file for information on usage and redistribution.
16
+ #
17
+ from __future__ import annotations
18
+
19
+ import io
20
+ import os
21
+ import shutil
22
+ import subprocess
23
+ import sys
24
+ import tempfile
25
+
26
+ from . import Image
27
+
28
+ TYPE_CHECKING = False
29
+ if TYPE_CHECKING:
30
+ from . import ImageWin
31
+
32
+
33
+ def grab(
34
+ bbox: tuple[int, int, int, int] | None = None,
35
+ include_layered_windows: bool = False,
36
+ all_screens: bool = False,
37
+ xdisplay: str | None = None,
38
+ window: int | ImageWin.HWND | None = None,
39
+ ) -> Image.Image:
40
+ im: Image.Image
41
+ if xdisplay is None:
42
+ if sys.platform == "darwin":
43
+ fh, filepath = tempfile.mkstemp(".png")
44
+ os.close(fh)
45
+ args = ["screencapture"]
46
+ if bbox:
47
+ left, top, right, bottom = bbox
48
+ args += ["-R", f"{left},{top},{right-left},{bottom-top}"]
49
+ subprocess.call(args + ["-x", filepath])
50
+ im = Image.open(filepath)
51
+ im.load()
52
+ os.unlink(filepath)
53
+ if bbox:
54
+ im_resized = im.resize((right - left, bottom - top))
55
+ im.close()
56
+ return im_resized
57
+ return im
58
+ elif sys.platform == "win32":
59
+ if window is not None:
60
+ all_screens = -1
61
+ offset, size, data = Image.core.grabscreen_win32(
62
+ include_layered_windows,
63
+ all_screens,
64
+ int(window) if window is not None else 0,
65
+ )
66
+ im = Image.frombytes(
67
+ "RGB",
68
+ size,
69
+ data,
70
+ # RGB, 32-bit line padding, origin lower left corner
71
+ "raw",
72
+ "BGR",
73
+ (size[0] * 3 + 3) & -4,
74
+ -1,
75
+ )
76
+ if bbox:
77
+ x0, y0 = offset
78
+ left, top, right, bottom = bbox
79
+ im = im.crop((left - x0, top - y0, right - x0, bottom - y0))
80
+ return im
81
+ # Cast to Optional[str] needed for Windows and macOS.
82
+ display_name: str | None = xdisplay
83
+ try:
84
+ if not Image.core.HAVE_XCB:
85
+ msg = "Pillow was built without XCB support"
86
+ raise OSError(msg)
87
+ size, data = Image.core.grabscreen_x11(display_name)
88
+ except OSError:
89
+ if display_name is None and sys.platform not in ("darwin", "win32"):
90
+ if shutil.which("gnome-screenshot"):
91
+ args = ["gnome-screenshot", "-f"]
92
+ elif shutil.which("grim"):
93
+ args = ["grim"]
94
+ elif shutil.which("spectacle"):
95
+ args = ["spectacle", "-n", "-b", "-f", "-o"]
96
+ else:
97
+ raise
98
+ fh, filepath = tempfile.mkstemp(".png")
99
+ os.close(fh)
100
+ subprocess.call(args + [filepath])
101
+ im = Image.open(filepath)
102
+ im.load()
103
+ os.unlink(filepath)
104
+ if bbox:
105
+ im_cropped = im.crop(bbox)
106
+ im.close()
107
+ return im_cropped
108
+ return im
109
+ else:
110
+ raise
111
+ else:
112
+ im = Image.frombytes("RGB", size, data, "raw", "BGRX", size[0] * 4, 1)
113
+ if bbox:
114
+ im = im.crop(bbox)
115
+ return im
116
+
117
+
118
+ def grabclipboard() -> Image.Image | list[str] | None:
119
+ if sys.platform == "darwin":
120
+ p = subprocess.run(
121
+ ["osascript", "-e", "get the clipboard as «class PNGf»"],
122
+ capture_output=True,
123
+ )
124
+ if p.returncode != 0:
125
+ return None
126
+
127
+ import binascii
128
+
129
+ data = io.BytesIO(binascii.unhexlify(p.stdout[11:-3]))
130
+ return Image.open(data)
131
+ elif sys.platform == "win32":
132
+ fmt, data = Image.core.grabclipboard_win32()
133
+ if fmt == "file": # CF_HDROP
134
+ import struct
135
+
136
+ o = struct.unpack_from("I", data)[0]
137
+ if data[16] == 0:
138
+ files = data[o:].decode("mbcs").split("\0")
139
+ else:
140
+ files = data[o:].decode("utf-16le").split("\0")
141
+ return files[: files.index("")]
142
+ if isinstance(data, bytes):
143
+ data = io.BytesIO(data)
144
+ if fmt == "png":
145
+ from . import PngImagePlugin
146
+
147
+ return PngImagePlugin.PngImageFile(data)
148
+ elif fmt == "DIB":
149
+ from . import BmpImagePlugin
150
+
151
+ return BmpImagePlugin.DibImageFile(data)
152
+ return None
153
+ else:
154
+ if os.getenv("WAYLAND_DISPLAY"):
155
+ session_type = "wayland"
156
+ elif os.getenv("DISPLAY"):
157
+ session_type = "x11"
158
+ else: # Session type check failed
159
+ session_type = None
160
+
161
+ if shutil.which("wl-paste") and session_type in ("wayland", None):
162
+ args = ["wl-paste", "-t", "image"]
163
+ elif shutil.which("xclip") and session_type in ("x11", None):
164
+ args = ["xclip", "-selection", "clipboard", "-t", "image/png", "-o"]
165
+ else:
166
+ msg = "wl-paste or xclip is required for ImageGrab.grabclipboard() on Linux"
167
+ raise NotImplementedError(msg)
168
+
169
+ p = subprocess.run(args, capture_output=True)
170
+ if p.returncode != 0:
171
+ err = p.stderr
172
+ for silent_error in [
173
+ # wl-paste, when the clipboard is empty
174
+ b"Nothing is copied",
175
+ # Ubuntu/Debian wl-paste, when the clipboard is empty
176
+ b"No selection",
177
+ # Ubuntu/Debian wl-paste, when an image isn't available
178
+ b"No suitable type of content copied",
179
+ # wl-paste or Ubuntu/Debian xclip, when an image isn't available
180
+ b" not available",
181
+ # xclip, when an image isn't available
182
+ b"cannot convert ",
183
+ # xclip, when the clipboard isn't initialized
184
+ b"xclip: Error: There is no owner for the ",
185
+ ]:
186
+ if silent_error in err:
187
+ return None
188
+ msg = f"{args[0]} error"
189
+ if err:
190
+ msg += f": {err.strip().decode()}"
191
+ raise ChildProcessError(msg)
192
+
193
+ data = io.BytesIO(p.stdout)
194
+ im = Image.open(data)
195
+ im.load()
196
+ return im
.venv/Lib/site-packages/PIL/ImageMath.py ADDED
@@ -0,0 +1,314 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #
2
+ # The Python Imaging Library
3
+ # $Id$
4
+ #
5
+ # a simple math add-on for the Python Imaging Library
6
+ #
7
+ # History:
8
+ # 1999-02-15 fl Original PIL Plus release
9
+ # 2005-05-05 fl Simplified and cleaned up for PIL 1.1.6
10
+ # 2005-09-12 fl Fixed int() and float() for Python 2.4.1
11
+ #
12
+ # Copyright (c) 1999-2005 by Secret Labs AB
13
+ # Copyright (c) 2005 by Fredrik Lundh
14
+ #
15
+ # See the README file for information on usage and redistribution.
16
+ #
17
+ from __future__ import annotations
18
+
19
+ import builtins
20
+
21
+ from . import Image, _imagingmath
22
+
23
+ TYPE_CHECKING = False
24
+ if TYPE_CHECKING:
25
+ from collections.abc import Callable
26
+ from types import CodeType
27
+ from typing import Any
28
+
29
+
30
+ class _Operand:
31
+ """Wraps an image operand, providing standard operators"""
32
+
33
+ def __init__(self, im: Image.Image):
34
+ self.im = im
35
+
36
+ def __fixup(self, im1: _Operand | float) -> Image.Image:
37
+ # convert image to suitable mode
38
+ if isinstance(im1, _Operand):
39
+ # argument was an image.
40
+ if im1.im.mode in ("1", "L"):
41
+ return im1.im.convert("I")
42
+ elif im1.im.mode in ("I", "F"):
43
+ return im1.im
44
+ else:
45
+ msg = f"unsupported mode: {im1.im.mode}"
46
+ raise ValueError(msg)
47
+ else:
48
+ # argument was a constant
49
+ if isinstance(im1, (int, float)) and self.im.mode in ("1", "L", "I"):
50
+ return Image.new("I", self.im.size, im1)
51
+ else:
52
+ return Image.new("F", self.im.size, im1)
53
+
54
+ def apply(
55
+ self,
56
+ op: str,
57
+ im1: _Operand | float,
58
+ im2: _Operand | float | None = None,
59
+ mode: str | None = None,
60
+ ) -> _Operand:
61
+ im_1 = self.__fixup(im1)
62
+ if im2 is None:
63
+ # unary operation
64
+ out = Image.new(mode or im_1.mode, im_1.size, None)
65
+ try:
66
+ op = getattr(_imagingmath, f"{op}_{im_1.mode}")
67
+ except AttributeError as e:
68
+ msg = f"bad operand type for '{op}'"
69
+ raise TypeError(msg) from e
70
+ _imagingmath.unop(op, out.getim(), im_1.getim())
71
+ else:
72
+ # binary operation
73
+ im_2 = self.__fixup(im2)
74
+ if im_1.mode != im_2.mode:
75
+ # convert both arguments to floating point
76
+ if im_1.mode != "F":
77
+ im_1 = im_1.convert("F")
78
+ if im_2.mode != "F":
79
+ im_2 = im_2.convert("F")
80
+ if im_1.size != im_2.size:
81
+ # crop both arguments to a common size
82
+ size = (
83
+ min(im_1.size[0], im_2.size[0]),
84
+ min(im_1.size[1], im_2.size[1]),
85
+ )
86
+ if im_1.size != size:
87
+ im_1 = im_1.crop((0, 0) + size)
88
+ if im_2.size != size:
89
+ im_2 = im_2.crop((0, 0) + size)
90
+ out = Image.new(mode or im_1.mode, im_1.size, None)
91
+ try:
92
+ op = getattr(_imagingmath, f"{op}_{im_1.mode}")
93
+ except AttributeError as e:
94
+ msg = f"bad operand type for '{op}'"
95
+ raise TypeError(msg) from e
96
+ _imagingmath.binop(op, out.getim(), im_1.getim(), im_2.getim())
97
+ return _Operand(out)
98
+
99
+ # unary operators
100
+ def __bool__(self) -> bool:
101
+ # an image is "true" if it contains at least one non-zero pixel
102
+ return self.im.getbbox() is not None
103
+
104
+ def __abs__(self) -> _Operand:
105
+ return self.apply("abs", self)
106
+
107
+ def __pos__(self) -> _Operand:
108
+ return self
109
+
110
+ def __neg__(self) -> _Operand:
111
+ return self.apply("neg", self)
112
+
113
+ # binary operators
114
+ def __add__(self, other: _Operand | float) -> _Operand:
115
+ return self.apply("add", self, other)
116
+
117
+ def __radd__(self, other: _Operand | float) -> _Operand:
118
+ return self.apply("add", other, self)
119
+
120
+ def __sub__(self, other: _Operand | float) -> _Operand:
121
+ return self.apply("sub", self, other)
122
+
123
+ def __rsub__(self, other: _Operand | float) -> _Operand:
124
+ return self.apply("sub", other, self)
125
+
126
+ def __mul__(self, other: _Operand | float) -> _Operand:
127
+ return self.apply("mul", self, other)
128
+
129
+ def __rmul__(self, other: _Operand | float) -> _Operand:
130
+ return self.apply("mul", other, self)
131
+
132
+ def __truediv__(self, other: _Operand | float) -> _Operand:
133
+ return self.apply("div", self, other)
134
+
135
+ def __rtruediv__(self, other: _Operand | float) -> _Operand:
136
+ return self.apply("div", other, self)
137
+
138
+ def __mod__(self, other: _Operand | float) -> _Operand:
139
+ return self.apply("mod", self, other)
140
+
141
+ def __rmod__(self, other: _Operand | float) -> _Operand:
142
+ return self.apply("mod", other, self)
143
+
144
+ def __pow__(self, other: _Operand | float) -> _Operand:
145
+ return self.apply("pow", self, other)
146
+
147
+ def __rpow__(self, other: _Operand | float) -> _Operand:
148
+ return self.apply("pow", other, self)
149
+
150
+ # bitwise
151
+ def __invert__(self) -> _Operand:
152
+ return self.apply("invert", self)
153
+
154
+ def __and__(self, other: _Operand | float) -> _Operand:
155
+ return self.apply("and", self, other)
156
+
157
+ def __rand__(self, other: _Operand | float) -> _Operand:
158
+ return self.apply("and", other, self)
159
+
160
+ def __or__(self, other: _Operand | float) -> _Operand:
161
+ return self.apply("or", self, other)
162
+
163
+ def __ror__(self, other: _Operand | float) -> _Operand:
164
+ return self.apply("or", other, self)
165
+
166
+ def __xor__(self, other: _Operand | float) -> _Operand:
167
+ return self.apply("xor", self, other)
168
+
169
+ def __rxor__(self, other: _Operand | float) -> _Operand:
170
+ return self.apply("xor", other, self)
171
+
172
+ def __lshift__(self, other: _Operand | float) -> _Operand:
173
+ return self.apply("lshift", self, other)
174
+
175
+ def __rshift__(self, other: _Operand | float) -> _Operand:
176
+ return self.apply("rshift", self, other)
177
+
178
+ # logical
179
+ def __eq__(self, other: _Operand | float) -> _Operand: # type: ignore[override]
180
+ return self.apply("eq", self, other)
181
+
182
+ def __ne__(self, other: _Operand | float) -> _Operand: # type: ignore[override]
183
+ return self.apply("ne", self, other)
184
+
185
+ def __lt__(self, other: _Operand | float) -> _Operand:
186
+ return self.apply("lt", self, other)
187
+
188
+ def __le__(self, other: _Operand | float) -> _Operand:
189
+ return self.apply("le", self, other)
190
+
191
+ def __gt__(self, other: _Operand | float) -> _Operand:
192
+ return self.apply("gt", self, other)
193
+
194
+ def __ge__(self, other: _Operand | float) -> _Operand:
195
+ return self.apply("ge", self, other)
196
+
197
+
198
+ # conversions
199
+ def imagemath_int(self: _Operand) -> _Operand:
200
+ return _Operand(self.im.convert("I"))
201
+
202
+
203
+ def imagemath_float(self: _Operand) -> _Operand:
204
+ return _Operand(self.im.convert("F"))
205
+
206
+
207
+ # logical
208
+ def imagemath_equal(self: _Operand, other: _Operand | float | None) -> _Operand:
209
+ return self.apply("eq", self, other, mode="I")
210
+
211
+
212
+ def imagemath_notequal(self: _Operand, other: _Operand | float | None) -> _Operand:
213
+ return self.apply("ne", self, other, mode="I")
214
+
215
+
216
+ def imagemath_min(self: _Operand, other: _Operand | float | None) -> _Operand:
217
+ return self.apply("min", self, other)
218
+
219
+
220
+ def imagemath_max(self: _Operand, other: _Operand | float | None) -> _Operand:
221
+ return self.apply("max", self, other)
222
+
223
+
224
+ def imagemath_convert(self: _Operand, mode: str) -> _Operand:
225
+ return _Operand(self.im.convert(mode))
226
+
227
+
228
+ ops = {
229
+ "int": imagemath_int,
230
+ "float": imagemath_float,
231
+ "equal": imagemath_equal,
232
+ "notequal": imagemath_notequal,
233
+ "min": imagemath_min,
234
+ "max": imagemath_max,
235
+ "convert": imagemath_convert,
236
+ }
237
+
238
+
239
+ def lambda_eval(expression: Callable[[dict[str, Any]], Any], **kw: Any) -> Any:
240
+ """
241
+ Returns the result of an image function.
242
+
243
+ :py:mod:`~PIL.ImageMath` only supports single-layer images. To process multi-band
244
+ images, use the :py:meth:`~PIL.Image.Image.split` method or
245
+ :py:func:`~PIL.Image.merge` function.
246
+
247
+ :param expression: A function that receives a dictionary.
248
+ :param **kw: Values to add to the function's dictionary.
249
+ :return: The expression result. This is usually an image object, but can
250
+ also be an integer, a floating point value, or a pixel tuple,
251
+ depending on the expression.
252
+ """
253
+
254
+ args: dict[str, Any] = ops.copy()
255
+ args.update(kw)
256
+ for k, v in args.items():
257
+ if isinstance(v, Image.Image):
258
+ args[k] = _Operand(v)
259
+
260
+ out = expression(args)
261
+ try:
262
+ return out.im
263
+ except AttributeError:
264
+ return out
265
+
266
+
267
+ def unsafe_eval(expression: str, **kw: Any) -> Any:
268
+ """
269
+ Evaluates an image expression. This uses Python's ``eval()`` function to process
270
+ the expression string, and carries the security risks of doing so. It is not
271
+ recommended to process expressions without considering this.
272
+ :py:meth:`~lambda_eval` is a more secure alternative.
273
+
274
+ :py:mod:`~PIL.ImageMath` only supports single-layer images. To process multi-band
275
+ images, use the :py:meth:`~PIL.Image.Image.split` method or
276
+ :py:func:`~PIL.Image.merge` function.
277
+
278
+ :param expression: A string containing a Python-style expression.
279
+ :param **kw: Values to add to the evaluation context.
280
+ :return: The evaluated expression. This is usually an image object, but can
281
+ also be an integer, a floating point value, or a pixel tuple,
282
+ depending on the expression.
283
+ """
284
+
285
+ # build execution namespace
286
+ args: dict[str, Any] = ops.copy()
287
+ for k in kw:
288
+ if "__" in k or hasattr(builtins, k):
289
+ msg = f"'{k}' not allowed"
290
+ raise ValueError(msg)
291
+
292
+ args.update(kw)
293
+ for k, v in args.items():
294
+ if isinstance(v, Image.Image):
295
+ args[k] = _Operand(v)
296
+
297
+ compiled_code = compile(expression, "<string>", "eval")
298
+
299
+ def scan(code: CodeType) -> None:
300
+ for const in code.co_consts:
301
+ if type(const) is type(compiled_code):
302
+ scan(const)
303
+
304
+ for name in code.co_names:
305
+ if name not in args and name != "abs":
306
+ msg = f"'{name}' not allowed"
307
+ raise ValueError(msg)
308
+
309
+ scan(compiled_code)
310
+ out = builtins.eval(expression, {"__builtins": {"abs": abs}}, args)
311
+ try:
312
+ return out.im
313
+ except AttributeError:
314
+ return out
.venv/Lib/site-packages/PIL/ImageMode.py ADDED
@@ -0,0 +1,85 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #
2
+ # The Python Imaging Library.
3
+ # $Id$
4
+ #
5
+ # standard mode descriptors
6
+ #
7
+ # History:
8
+ # 2006-03-20 fl Added
9
+ #
10
+ # Copyright (c) 2006 by Secret Labs AB.
11
+ # Copyright (c) 2006 by Fredrik Lundh.
12
+ #
13
+ # See the README file for information on usage and redistribution.
14
+ #
15
+ from __future__ import annotations
16
+
17
+ import sys
18
+ from functools import lru_cache
19
+ from typing import NamedTuple
20
+
21
+
22
+ class ModeDescriptor(NamedTuple):
23
+ """Wrapper for mode strings."""
24
+
25
+ mode: str
26
+ bands: tuple[str, ...]
27
+ basemode: str
28
+ basetype: str
29
+ typestr: str
30
+
31
+ def __str__(self) -> str:
32
+ return self.mode
33
+
34
+
35
+ @lru_cache
36
+ def getmode(mode: str) -> ModeDescriptor:
37
+ """Gets a mode descriptor for the given mode."""
38
+ endian = "<" if sys.byteorder == "little" else ">"
39
+
40
+ modes = {
41
+ # core modes
42
+ # Bits need to be extended to bytes
43
+ "1": ("L", "L", ("1",), "|b1"),
44
+ "L": ("L", "L", ("L",), "|u1"),
45
+ "I": ("L", "I", ("I",), f"{endian}i4"),
46
+ "F": ("L", "F", ("F",), f"{endian}f4"),
47
+ "P": ("P", "L", ("P",), "|u1"),
48
+ "RGB": ("RGB", "L", ("R", "G", "B"), "|u1"),
49
+ "RGBX": ("RGB", "L", ("R", "G", "B", "X"), "|u1"),
50
+ "RGBA": ("RGB", "L", ("R", "G", "B", "A"), "|u1"),
51
+ "CMYK": ("RGB", "L", ("C", "M", "Y", "K"), "|u1"),
52
+ "YCbCr": ("RGB", "L", ("Y", "Cb", "Cr"), "|u1"),
53
+ # UNDONE - unsigned |u1i1i1
54
+ "LAB": ("RGB", "L", ("L", "A", "B"), "|u1"),
55
+ "HSV": ("RGB", "L", ("H", "S", "V"), "|u1"),
56
+ # extra experimental modes
57
+ "RGBa": ("RGB", "L", ("R", "G", "B", "a"), "|u1"),
58
+ "LA": ("L", "L", ("L", "A"), "|u1"),
59
+ "La": ("L", "L", ("L", "a"), "|u1"),
60
+ "PA": ("RGB", "L", ("P", "A"), "|u1"),
61
+ }
62
+ if mode in modes:
63
+ base_mode, base_type, bands, type_str = modes[mode]
64
+ return ModeDescriptor(mode, bands, base_mode, base_type, type_str)
65
+
66
+ mapping_modes = {
67
+ # I;16 == I;16L, and I;32 == I;32L
68
+ "I;16": "<u2",
69
+ "I;16S": "<i2",
70
+ "I;16L": "<u2",
71
+ "I;16LS": "<i2",
72
+ "I;16B": ">u2",
73
+ "I;16BS": ">i2",
74
+ "I;16N": f"{endian}u2",
75
+ "I;16NS": f"{endian}i2",
76
+ "I;32": "<u4",
77
+ "I;32B": ">u4",
78
+ "I;32L": "<u4",
79
+ "I;32S": "<i4",
80
+ "I;32BS": ">i4",
81
+ "I;32LS": "<i4",
82
+ }
83
+
84
+ type_str = mapping_modes[mode]
85
+ return ModeDescriptor(mode, ("I",), "L", "L", type_str)
.venv/Lib/site-packages/PIL/ImageMorph.py ADDED
@@ -0,0 +1,265 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # A binary morphology add-on for the Python Imaging Library
2
+ #
3
+ # History:
4
+ # 2014-06-04 Initial version.
5
+ #
6
+ # Copyright (c) 2014 Dov Grobgeld <dov.grobgeld@gmail.com>
7
+ from __future__ import annotations
8
+
9
+ import re
10
+
11
+ from . import Image, _imagingmorph
12
+
13
+ LUT_SIZE = 1 << 9
14
+
15
+ # fmt: off
16
+ ROTATION_MATRIX = [
17
+ 6, 3, 0,
18
+ 7, 4, 1,
19
+ 8, 5, 2,
20
+ ]
21
+ MIRROR_MATRIX = [
22
+ 2, 1, 0,
23
+ 5, 4, 3,
24
+ 8, 7, 6,
25
+ ]
26
+ # fmt: on
27
+
28
+
29
+ class LutBuilder:
30
+ """A class for building a MorphLut from a descriptive language
31
+
32
+ The input patterns is a list of a strings sequences like these::
33
+
34
+ 4:(...
35
+ .1.
36
+ 111)->1
37
+
38
+ (whitespaces including linebreaks are ignored). The option 4
39
+ describes a series of symmetry operations (in this case a
40
+ 4-rotation), the pattern is described by:
41
+
42
+ - . or X - Ignore
43
+ - 1 - Pixel is on
44
+ - 0 - Pixel is off
45
+
46
+ The result of the operation is described after "->" string.
47
+
48
+ The default is to return the current pixel value, which is
49
+ returned if no other match is found.
50
+
51
+ Operations:
52
+
53
+ - 4 - 4 way rotation
54
+ - N - Negate
55
+ - 1 - Dummy op for no other operation (an op must always be given)
56
+ - M - Mirroring
57
+
58
+ Example::
59
+
60
+ lb = LutBuilder(patterns = ["4:(... .1. 111)->1"])
61
+ lut = lb.build_lut()
62
+
63
+ """
64
+
65
+ def __init__(
66
+ self, patterns: list[str] | None = None, op_name: str | None = None
67
+ ) -> None:
68
+ if patterns is not None:
69
+ self.patterns = patterns
70
+ else:
71
+ self.patterns = []
72
+ self.lut: bytearray | None = None
73
+ if op_name is not None:
74
+ known_patterns = {
75
+ "corner": ["1:(... ... ...)->0", "4:(00. 01. ...)->1"],
76
+ "dilation4": ["4:(... .0. .1.)->1"],
77
+ "dilation8": ["4:(... .0. .1.)->1", "4:(... .0. ..1)->1"],
78
+ "erosion4": ["4:(... .1. .0.)->0"],
79
+ "erosion8": ["4:(... .1. .0.)->0", "4:(... .1. ..0)->0"],
80
+ "edge": [
81
+ "1:(... ... ...)->0",
82
+ "4:(.0. .1. ...)->1",
83
+ "4:(01. .1. ...)->1",
84
+ ],
85
+ }
86
+ if op_name not in known_patterns:
87
+ msg = f"Unknown pattern {op_name}!"
88
+ raise Exception(msg)
89
+
90
+ self.patterns = known_patterns[op_name]
91
+
92
+ def add_patterns(self, patterns: list[str]) -> None:
93
+ self.patterns += patterns
94
+
95
+ def build_default_lut(self) -> None:
96
+ symbols = [0, 1]
97
+ m = 1 << 4 # pos of current pixel
98
+ self.lut = bytearray(symbols[(i & m) > 0] for i in range(LUT_SIZE))
99
+
100
+ def get_lut(self) -> bytearray | None:
101
+ return self.lut
102
+
103
+ def _string_permute(self, pattern: str, permutation: list[int]) -> str:
104
+ """string_permute takes a pattern and a permutation and returns the
105
+ string permuted according to the permutation list.
106
+ """
107
+ assert len(permutation) == 9
108
+ return "".join(pattern[p] for p in permutation)
109
+
110
+ def _pattern_permute(
111
+ self, basic_pattern: str, options: str, basic_result: int
112
+ ) -> list[tuple[str, int]]:
113
+ """pattern_permute takes a basic pattern and its result and clones
114
+ the pattern according to the modifications described in the $options
115
+ parameter. It returns a list of all cloned patterns."""
116
+ patterns = [(basic_pattern, basic_result)]
117
+
118
+ # rotations
119
+ if "4" in options:
120
+ res = patterns[-1][1]
121
+ for i in range(4):
122
+ patterns.append(
123
+ (self._string_permute(patterns[-1][0], ROTATION_MATRIX), res)
124
+ )
125
+ # mirror
126
+ if "M" in options:
127
+ n = len(patterns)
128
+ for pattern, res in patterns[:n]:
129
+ patterns.append((self._string_permute(pattern, MIRROR_MATRIX), res))
130
+
131
+ # negate
132
+ if "N" in options:
133
+ n = len(patterns)
134
+ for pattern, res in patterns[:n]:
135
+ # Swap 0 and 1
136
+ pattern = pattern.replace("0", "Z").replace("1", "0").replace("Z", "1")
137
+ res = 1 - int(res)
138
+ patterns.append((pattern, res))
139
+
140
+ return patterns
141
+
142
+ def build_lut(self) -> bytearray:
143
+ """Compile all patterns into a morphology lut.
144
+
145
+ TBD :Build based on (file) morphlut:modify_lut
146
+ """
147
+ self.build_default_lut()
148
+ assert self.lut is not None
149
+ patterns = []
150
+
151
+ # Parse and create symmetries of the patterns strings
152
+ for p in self.patterns:
153
+ m = re.search(r"(\w):?\s*\((.+?)\)\s*->\s*(\d)", p.replace("\n", ""))
154
+ if not m:
155
+ msg = 'Syntax error in pattern "' + p + '"'
156
+ raise Exception(msg)
157
+ options = m.group(1)
158
+ pattern = m.group(2)
159
+ result = int(m.group(3))
160
+
161
+ # Get rid of spaces
162
+ pattern = pattern.replace(" ", "").replace("\n", "")
163
+
164
+ patterns += self._pattern_permute(pattern, options, result)
165
+
166
+ # compile the patterns into regular expressions for speed
167
+ compiled_patterns = []
168
+ for pattern in patterns:
169
+ p = pattern[0].replace(".", "X").replace("X", "[01]")
170
+ compiled_patterns.append((re.compile(p), pattern[1]))
171
+
172
+ # Step through table and find patterns that match.
173
+ # Note that all the patterns are searched. The last one
174
+ # caught overrides
175
+ for i in range(LUT_SIZE):
176
+ # Build the bit pattern
177
+ bitpattern = bin(i)[2:]
178
+ bitpattern = ("0" * (9 - len(bitpattern)) + bitpattern)[::-1]
179
+
180
+ for pattern, r in compiled_patterns:
181
+ if pattern.match(bitpattern):
182
+ self.lut[i] = [0, 1][r]
183
+
184
+ return self.lut
185
+
186
+
187
+ class MorphOp:
188
+ """A class for binary morphological operators"""
189
+
190
+ def __init__(
191
+ self,
192
+ lut: bytearray | None = None,
193
+ op_name: str | None = None,
194
+ patterns: list[str] | None = None,
195
+ ) -> None:
196
+ """Create a binary morphological operator"""
197
+ self.lut = lut
198
+ if op_name is not None:
199
+ self.lut = LutBuilder(op_name=op_name).build_lut()
200
+ elif patterns is not None:
201
+ self.lut = LutBuilder(patterns=patterns).build_lut()
202
+
203
+ def apply(self, image: Image.Image) -> tuple[int, Image.Image]:
204
+ """Run a single morphological operation on an image
205
+
206
+ Returns a tuple of the number of changed pixels and the
207
+ morphed image"""
208
+ if self.lut is None:
209
+ msg = "No operator loaded"
210
+ raise Exception(msg)
211
+
212
+ if image.mode != "L":
213
+ msg = "Image mode must be L"
214
+ raise ValueError(msg)
215
+ outimage = Image.new(image.mode, image.size, None)
216
+ count = _imagingmorph.apply(bytes(self.lut), image.getim(), outimage.getim())
217
+ return count, outimage
218
+
219
+ def match(self, image: Image.Image) -> list[tuple[int, int]]:
220
+ """Get a list of coordinates matching the morphological operation on
221
+ an image.
222
+
223
+ Returns a list of tuples of (x,y) coordinates
224
+ of all matching pixels. See :ref:`coordinate-system`."""
225
+ if self.lut is None:
226
+ msg = "No operator loaded"
227
+ raise Exception(msg)
228
+
229
+ if image.mode != "L":
230
+ msg = "Image mode must be L"
231
+ raise ValueError(msg)
232
+ return _imagingmorph.match(bytes(self.lut), image.getim())
233
+
234
+ def get_on_pixels(self, image: Image.Image) -> list[tuple[int, int]]:
235
+ """Get a list of all turned on pixels in a binary image
236
+
237
+ Returns a list of tuples of (x,y) coordinates
238
+ of all matching pixels. See :ref:`coordinate-system`."""
239
+
240
+ if image.mode != "L":
241
+ msg = "Image mode must be L"
242
+ raise ValueError(msg)
243
+ return _imagingmorph.get_on_pixels(image.getim())
244
+
245
+ def load_lut(self, filename: str) -> None:
246
+ """Load an operator from an mrl file"""
247
+ with open(filename, "rb") as f:
248
+ self.lut = bytearray(f.read())
249
+
250
+ if len(self.lut) != LUT_SIZE:
251
+ self.lut = None
252
+ msg = "Wrong size operator file!"
253
+ raise Exception(msg)
254
+
255
+ def save_lut(self, filename: str) -> None:
256
+ """Save an operator to an mrl file"""
257
+ if self.lut is None:
258
+ msg = "No operator loaded"
259
+ raise Exception(msg)
260
+ with open(filename, "wb") as f:
261
+ f.write(self.lut)
262
+
263
+ def set_lut(self, lut: bytearray | None) -> None:
264
+ """Set the lut from an external source"""
265
+ self.lut = lut