saikanov commited on
Commit
267040a
·
verified ·
1 Parent(s): 94acfcb

Upload folder using huggingface_hub

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 +25 -0
  2. .github/workflows/update_space.yml +28 -0
  3. .gradio/certificate.pem +31 -0
  4. README.md +2 -8
  5. app.py +212 -0
  6. vectordb-chroma/gformdb/355c5a5d-b607-4435-a4cf-2358574ab34b/data_level0.bin +3 -0
  7. vectordb-chroma/gformdb/355c5a5d-b607-4435-a4cf-2358574ab34b/header.bin +3 -0
  8. vectordb-chroma/gformdb/355c5a5d-b607-4435-a4cf-2358574ab34b/length.bin +3 -0
  9. vectordb-chroma/gformdb/355c5a5d-b607-4435-a4cf-2358574ab34b/link_lists.bin +0 -0
  10. vectordb-chroma/gformdb/chroma.sqlite3 +3 -0
  11. venv/Include/site/python3.10/greenlet/greenlet.h +164 -0
  12. venv/Lib/site-packages/Deprecated-1.2.14.dist-info/INSTALLER +1 -0
  13. venv/Lib/site-packages/Deprecated-1.2.14.dist-info/LICENSE.rst +21 -0
  14. venv/Lib/site-packages/Deprecated-1.2.14.dist-info/METADATA +181 -0
  15. venv/Lib/site-packages/Deprecated-1.2.14.dist-info/RECORD +12 -0
  16. venv/Lib/site-packages/Deprecated-1.2.14.dist-info/WHEEL +6 -0
  17. venv/Lib/site-packages/Deprecated-1.2.14.dist-info/top_level.txt +1 -0
  18. venv/Lib/site-packages/MarkupSafe-2.1.5.dist-info/INSTALLER +1 -0
  19. venv/Lib/site-packages/MarkupSafe-2.1.5.dist-info/LICENSE.rst +28 -0
  20. venv/Lib/site-packages/MarkupSafe-2.1.5.dist-info/METADATA +93 -0
  21. venv/Lib/site-packages/MarkupSafe-2.1.5.dist-info/RECORD +14 -0
  22. venv/Lib/site-packages/MarkupSafe-2.1.5.dist-info/WHEEL +5 -0
  23. venv/Lib/site-packages/MarkupSafe-2.1.5.dist-info/top_level.txt +1 -0
  24. venv/Lib/site-packages/PIL/BdfFontFile.py +133 -0
  25. venv/Lib/site-packages/PIL/BlpImagePlugin.py +493 -0
  26. venv/Lib/site-packages/PIL/BmpImagePlugin.py +511 -0
  27. venv/Lib/site-packages/PIL/BufrStubImagePlugin.py +76 -0
  28. venv/Lib/site-packages/PIL/ContainerIO.py +173 -0
  29. venv/Lib/site-packages/PIL/CurImagePlugin.py +75 -0
  30. venv/Lib/site-packages/PIL/DcxImagePlugin.py +80 -0
  31. venv/Lib/site-packages/PIL/DdsImagePlugin.py +575 -0
  32. venv/Lib/site-packages/PIL/EpsImagePlugin.py +474 -0
  33. venv/Lib/site-packages/PIL/ExifTags.py +381 -0
  34. venv/Lib/site-packages/PIL/FitsImagePlugin.py +152 -0
  35. venv/Lib/site-packages/PIL/FliImagePlugin.py +175 -0
  36. venv/Lib/site-packages/PIL/FontFile.py +134 -0
  37. venv/Lib/site-packages/PIL/FpxImagePlugin.py +257 -0
  38. venv/Lib/site-packages/PIL/FtexImagePlugin.py +115 -0
  39. venv/Lib/site-packages/PIL/GbrImagePlugin.py +103 -0
  40. venv/Lib/site-packages/PIL/GdImageFile.py +102 -0
  41. venv/Lib/site-packages/PIL/GifImagePlugin.py +1197 -0
  42. venv/Lib/site-packages/PIL/GimpGradientFile.py +149 -0
  43. venv/Lib/site-packages/PIL/GimpPaletteFile.py +58 -0
  44. venv/Lib/site-packages/PIL/GribStubImagePlugin.py +76 -0
  45. venv/Lib/site-packages/PIL/Hdf5StubImagePlugin.py +76 -0
  46. venv/Lib/site-packages/PIL/IcnsImagePlugin.py +412 -0
  47. venv/Lib/site-packages/PIL/IcoImagePlugin.py +381 -0
  48. venv/Lib/site-packages/PIL/ImImagePlugin.py +386 -0
  49. venv/Lib/site-packages/PIL/Image.py +0 -0
  50. venv/Lib/site-packages/PIL/ImageChops.py +311 -0
.gitattributes CHANGED
@@ -33,3 +33,28 @@ 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
+ vectordb-chroma/gformdb/chroma.sqlite3 filter=lfs diff=lfs merge=lfs -text
37
+ venv/Lib/site-packages/gradio/_frontend_code/lite/dist/assets/Canvas3D-B-WeHuGX.js.map filter=lfs diff=lfs merge=lfs -text
38
+ venv/Lib/site-packages/gradio/frpc_windows_amd64_v0.3 filter=lfs diff=lfs merge=lfs -text
39
+ venv/Lib/site-packages/gradio/templates/frontend/assets/Canvas3D-dmZ3UW-M.js.map filter=lfs diff=lfs merge=lfs -text
40
+ venv/Lib/site-packages/gradio/templates/node/build/client/_app/immutable/chunks/Canvas3D.ezSBJWjS.js.br filter=lfs diff=lfs merge=lfs -text
41
+ venv/Lib/site-packages/gradio/templates/node/build/server/chunks/Canvas3D-DQQiLfH-.js.map filter=lfs diff=lfs merge=lfs -text
42
+ venv/Lib/site-packages/gradio/templates/node/build/server/chunks/PlotlyPlot-P3y0I-J2.js.map filter=lfs diff=lfs merge=lfs -text
43
+ venv/Lib/site-packages/grpc/_cython/cygrpc.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
44
+ venv/Lib/site-packages/kubernetes/client/api/__pycache__/core_v1_api.cpython-310.pyc filter=lfs diff=lfs merge=lfs -text
45
+ venv/Lib/site-packages/numpy/core/_multiarray_umath.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
46
+ venv/Lib/site-packages/numpy/core/_simd.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
47
+ venv/Lib/site-packages/numpy.libs/libopenblas64__v0.3.23-293-gc2f4bdbb-gcc_10_3_0-2bde3a66a51006b2b53eb373ff767a3f.dll filter=lfs diff=lfs merge=lfs -text
48
+ venv/Lib/site-packages/onnxruntime/capi/onnxruntime.dll filter=lfs diff=lfs merge=lfs -text
49
+ venv/Lib/site-packages/onnxruntime/capi/onnxruntime_pybind11_state.pyd filter=lfs diff=lfs merge=lfs -text
50
+ venv/Lib/site-packages/pandas/_libs/algos.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
51
+ venv/Lib/site-packages/pandas/_libs/groupby.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
52
+ venv/Lib/site-packages/pandas/_libs/hashtable.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
53
+ venv/Lib/site-packages/pandas/_libs/interval.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
54
+ venv/Lib/site-packages/pandas/_libs/join.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
55
+ venv/Lib/site-packages/PIL/_imaging.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
56
+ venv/Lib/site-packages/PIL/_imagingft.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
57
+ venv/Lib/site-packages/pydantic_core/_pydantic_core.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
58
+ venv/Lib/site-packages/tiktoken/_tiktoken.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
59
+ venv/Lib/site-packages/tokenizers/tokenizers.cp310-win_amd64.pyd filter=lfs diff=lfs merge=lfs -text
60
+ venv/Scripts/ruff.exe filter=lfs diff=lfs merge=lfs -text
.github/workflows/update_space.yml ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ name: Run Python script
2
+
3
+ on:
4
+ push:
5
+ branches:
6
+ - main
7
+
8
+ jobs:
9
+ build:
10
+ runs-on: ubuntu-latest
11
+
12
+ steps:
13
+ - name: Checkout
14
+ uses: actions/checkout@v2
15
+
16
+ - name: Set up Python
17
+ uses: actions/setup-python@v2
18
+ with:
19
+ python-version: '3.9'
20
+
21
+ - name: Install Gradio
22
+ run: python -m pip install gradio
23
+
24
+ - name: Log in to Hugging Face
25
+ run: python -c 'import huggingface_hub; huggingface_hub.login(token="${{ secrets.hf_token }}")'
26
+
27
+ - name: Deploy to Spaces
28
+ run: gradio deploy
.gradio/certificate.pem ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ -----BEGIN CERTIFICATE-----
2
+ MIIFazCCA1OgAwIBAgIRAIIQz7DSQONZRGPgu2OCiwAwDQYJKoZIhvcNAQELBQAw
3
+ TzELMAkGA1UEBhMCVVMxKTAnBgNVBAoTIEludGVybmV0IFNlY3VyaXR5IFJlc2Vh
4
+ cmNoIEdyb3VwMRUwEwYDVQQDEwxJU1JHIFJvb3QgWDEwHhcNMTUwNjA0MTEwNDM4
5
+ WhcNMzUwNjA0MTEwNDM4WjBPMQswCQYDVQQGEwJVUzEpMCcGA1UEChMgSW50ZXJu
6
+ ZXQgU2VjdXJpdHkgUmVzZWFyY2ggR3JvdXAxFTATBgNVBAMTDElTUkcgUm9vdCBY
7
+ MTCCAiIwDQYJKoZIhvcNAQEBBQADggIPADCCAgoCggIBAK3oJHP0FDfzm54rVygc
8
+ h77ct984kIxuPOZXoHj3dcKi/vVqbvYATyjb3miGbESTtrFj/RQSa78f0uoxmyF+
9
+ 0TM8ukj13Xnfs7j/EvEhmkvBioZxaUpmZmyPfjxwv60pIgbz5MDmgK7iS4+3mX6U
10
+ A5/TR5d8mUgjU+g4rk8Kb4Mu0UlXjIB0ttov0DiNewNwIRt18jA8+o+u3dpjq+sW
11
+ T8KOEUt+zwvo/7V3LvSye0rgTBIlDHCNAymg4VMk7BPZ7hm/ELNKjD+Jo2FR3qyH
12
+ B5T0Y3HsLuJvW5iB4YlcNHlsdu87kGJ55tukmi8mxdAQ4Q7e2RCOFvu396j3x+UC
13
+ B5iPNgiV5+I3lg02dZ77DnKxHZu8A/lJBdiB3QW0KtZB6awBdpUKD9jf1b0SHzUv
14
+ KBds0pjBqAlkd25HN7rOrFleaJ1/ctaJxQZBKT5ZPt0m9STJEadao0xAH0ahmbWn
15
+ OlFuhjuefXKnEgV4We0+UXgVCwOPjdAvBbI+e0ocS3MFEvzG6uBQE3xDk3SzynTn
16
+ jh8BCNAw1FtxNrQHusEwMFxIt4I7mKZ9YIqioymCzLq9gwQbooMDQaHWBfEbwrbw
17
+ qHyGO0aoSCqI3Haadr8faqU9GY/rOPNk3sgrDQoo//fb4hVC1CLQJ13hef4Y53CI
18
+ rU7m2Ys6xt0nUW7/vGT1M0NPAgMBAAGjQjBAMA4GA1UdDwEB/wQEAwIBBjAPBgNV
19
+ HRMBAf8EBTADAQH/MB0GA1UdDgQWBBR5tFnme7bl5AFzgAiIyBpY9umbbjANBgkq
20
+ hkiG9w0BAQsFAAOCAgEAVR9YqbyyqFDQDLHYGmkgJykIrGF1XIpu+ILlaS/V9lZL
21
+ ubhzEFnTIZd+50xx+7LSYK05qAvqFyFWhfFQDlnrzuBZ6brJFe+GnY+EgPbk6ZGQ
22
+ 3BebYhtF8GaV0nxvwuo77x/Py9auJ/GpsMiu/X1+mvoiBOv/2X/qkSsisRcOj/KK
23
+ NFtY2PwByVS5uCbMiogziUwthDyC3+6WVwW6LLv3xLfHTjuCvjHIInNzktHCgKQ5
24
+ ORAzI4JMPJ+GslWYHb4phowim57iaztXOoJwTdwJx4nLCgdNbOhdjsnvzqvHu7Ur
25
+ TkXWStAmzOVyyghqpZXjFaH3pO3JLF+l+/+sKAIuvtd7u+Nxe5AW0wdeRlN8NwdC
26
+ jNPElpzVmbUq4JUagEiuTDkHzsxHpFKVK7q4+63SM1N95R1NbdWhscdCb+ZAJzVc
27
+ oyi3B43njTOQ5yOf+1CceWxG1bQVs5ZufpsMljq4Ui0/1lvh+wjChP4kqKOJ2qxq
28
+ 4RgqsahDYVvTH9w7jXbyLeiNdd8XM2w9U/t7y0Ff/9yi0GE44Za4rF2LN9d11TPA
29
+ mRGunUHBcnWEvgJBQl9nJEiU0Zsnvgc/ubhPgXRR4Xq37Z0j4r7g1SgEEzwxA57d
30
+ emyPxgcYxn/eR44/KJ4EBs+lVDR3veyJm+kXQ99b21/+jh5Xos1AnX5iItreGCc=
31
+ -----END CERTIFICATE-----
README.md CHANGED
@@ -1,12 +1,6 @@
1
  ---
2
- title: 'G-Form Quiz Answering '
3
- emoji: 🐢
4
- colorFrom: blue
5
- colorTo: indigo
6
  sdk: gradio
7
  sdk_version: 5.4.0
8
- app_file: app.py
9
- pinned: false
10
  ---
11
-
12
- Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
 
1
  ---
2
+ title: G-Form_Quiz_Answering_
3
+ app_file: app.py
 
 
4
  sdk: gradio
5
  sdk_version: 5.4.0
 
 
6
  ---
 
 
app.py ADDED
@@ -0,0 +1,212 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ import pandas as pd
3
+ from langchain.memory import VectorStoreRetrieverMemory
4
+ from langchain_community.vectorstores import Chroma
5
+ from langchain_openai import OpenAIEmbeddings, ChatOpenAI
6
+ from langchain_community.document_loaders import PyPDFLoader
7
+ from langchain_core.prompts import PromptTemplate
8
+ import chromadb
9
+ import os
10
+ from uuid import uuid4
11
+
12
+ # Initialize persistent client
13
+ CHROMA_PATH = "./vectordb-chroma/gformdb"
14
+ os.makedirs(CHROMA_PATH, exist_ok=True)
15
+ default_user_aspek = "Jika Jawaban salah berikan nilai 0, jika jawaban benar namun tidak tepat berikan nilai 50, jika jawaban benar dan lengkap serta penjelasan baik, beri nilai 100"
16
+
17
+ def process_files(csv_file, openai_key, user_aspek=default_user_aspek, pdf_file=False, model_name="gpt-4o-mini"):
18
+ try:
19
+ context = ""
20
+
21
+ def context_search(query="a",source_name=False, k=5):
22
+
23
+ if source_name==True:
24
+ results = vectorstore.similarity_search(
25
+ query=query,
26
+ k=k,
27
+ filter={"source": source_name}
28
+ )
29
+ else:
30
+ results = vectorstore.similarity_search(
31
+ query=query,
32
+ k=k,
33
+ )
34
+ for res in results:
35
+ page_content = res.page_content
36
+ metadata = res.metadata
37
+ return page_content, metadata
38
+
39
+ def format_qa_efficient(data_path):
40
+ data = pd.read_csv(data_path)
41
+ data.head()
42
+ qa_efficient = ""
43
+ indexs = 0
44
+ Question = ""
45
+ for column in data.columns[3:]: # Lewati kolom Timestamp, Email Address, dan Nama
46
+ qa_efficient += f"Nomor-{indexs+1}.{column}:\n"
47
+ indexs += 1
48
+ Question += column
49
+ for index, row in data.iterrows():
50
+ name = row['Nama']
51
+ email = row["Email_Address"]
52
+ answer = row[column]
53
+ qa_efficient += f"- {name}|{email}: {answer}\n"
54
+
55
+ qa_efficient += "\n" # Baris kosong antara setiap pertanyaan
56
+
57
+ return qa_efficient, Question
58
+
59
+ # Set OpenAI API
60
+ os.environ["OPENAI_API_KEY"] = openai_key
61
+
62
+ # Process CSV (Gradio provides the file path directly)
63
+
64
+ QA, Question = format_qa_efficient(csv_file)
65
+
66
+ if pdf_file:
67
+ # Initialize Chroma client
68
+ persistent_client = chromadb.PersistentClient(path=CHROMA_PATH)
69
+ collection = persistent_client.get_or_create_collection("RAG")
70
+
71
+ vectorstore = Chroma(
72
+ client=persistent_client,
73
+ collection_name=collection.name,
74
+ embedding_function=OpenAIEmbeddings()
75
+ )
76
+
77
+ res, metadata = context_search(pdf_file,k=1)
78
+
79
+ if metadata == pdf_file.name:
80
+ # Process PDF (Gradio provides the file path directly)
81
+ pages = PyPDFLoader(pdf_file.name).load_and_split()
82
+ uuids = [str(uuid4()) for _ in range(len(pages))]
83
+ vectorstore.add_documents(documents=pages, ids=uuids)
84
+ context, metadata = context_search(Question, pdf_file.name)
85
+
86
+ elif pdf_file == False:
87
+ context = "Tidak ada context tambahan yang diberikan, tolong gunakan pengetahuan anda untuk menjawab pertanyaan"
88
+
89
+ # Evaluation template
90
+ TEMPLATE = """
91
+ Kamu adalah AI evaluator pendidikan. Nilai jawaban siswa berikut menggunakan panduan ini:
92
+
93
+ Buat output dengan format(format ini untuk satu murid, jadi selesaikan dulu 1 murid untuk semua nomor, baru ke murid berikutnya):
94
+ HASIL EVALUASI
95
+ =============
96
+ Nama dan Email Murid:
97
+ [Nomor Soal].
98
+ Nilai: [0-100]
99
+ Alasan: [alasan singkat penilaian]
100
+ Saran Perbaikan: [saran]
101
+
102
+ [buat seperti di atas untuk setiap jawaban]
103
+
104
+ ================
105
+ Rata-rata Nilai: [nilai]
106
+
107
+ Rekomendasi Umum:
108
+ [rekomendasi]
109
+
110
+
111
+
112
+ MATERI REFERENSI:
113
+ {context}
114
+
115
+ SOAL & JAWABAN:
116
+ {QA}
117
+
118
+ Aspek Penilaian:
119
+ {Aspek}
120
+
121
+
122
+ """
123
+
124
+ PROMPT = PromptTemplate(
125
+ input_variables=["QA", "context", "Aspek"],
126
+ template=TEMPLATE
127
+ )
128
+
129
+ # Initialize ChatGPT
130
+ chat = ChatOpenAI(model=model_name)
131
+ chain = PROMPT | chat
132
+
133
+ # Generate evaluation
134
+ Aspek_penilaian = user_aspek
135
+
136
+ response = chain.invoke({
137
+ "Aspek": Aspek_penilaian,
138
+ "QA": QA,
139
+ "context": context
140
+ })
141
+
142
+ return response.content
143
+
144
+ except Exception as e:
145
+ return f"Error occurred: {str(e)}\nType: {type(e)}"
146
+
147
+ # Create Gradio interface
148
+ def create_interface():
149
+ with gr.Blocks(title="Quiz Evaluator", theme=gr.themes.Soft()) as app:
150
+ with gr.Column(scale=1):
151
+ gr.Markdown(
152
+ """
153
+ # 📝 Quiz Response Evaluator
154
+ Upload your quiz responses (CSV) and reference material (PDF) to get AI-powered evaluation.
155
+ """
156
+ )
157
+
158
+ with gr.Row():
159
+ with gr.Column(scale=1):
160
+ csv_input = gr.File(
161
+ label="Quiz Responses (CSV)",
162
+ file_types=[".csv"]
163
+ )
164
+ with gr.Column(scale=1):
165
+ pdf_input = gr.File(
166
+ label="Reference Material (PDF)",
167
+ file_types=[".pdf"]
168
+ )
169
+
170
+ user_aspek = gr.Textbox(
171
+ label="Aspek",
172
+ placeholder="Jika Jawaban salah berikan nilai 0, jika jawaban benar namun tidak tepat berikan nilai 50, jika jawaban benar dan lengkap serta penjelasan baik, beri nilai 100",
173
+ show_copy_button=True
174
+ )
175
+
176
+ model_name = gr.Textbox(
177
+ label="openAI model",
178
+ placeholder="input your openAI model"
179
+ )
180
+
181
+ api_key = gr.Textbox(
182
+ label="OpenAI API Key",
183
+ placeholder="Enter your OpenAI API key",
184
+ type="password"
185
+ )
186
+
187
+ submit_btn = gr.Button(
188
+ "Evaluate Responses",
189
+ variant="primary",
190
+ size="lg"
191
+ )
192
+
193
+ # Using Textbox instead of Markdown for better formatting
194
+ output = gr.Textbox(
195
+ label="Evaluation Results",
196
+ lines=20,
197
+ max_lines=30,
198
+ show_copy_button=True
199
+ )
200
+
201
+ submit_btn.click(
202
+ fn=process_files,
203
+ inputs=[csv_input,api_key,user_aspek,pdf_input,model_name],
204
+ outputs=output
205
+ )
206
+
207
+ return app
208
+
209
+ # Launch the interface
210
+ if __name__ == "__main__":
211
+ app = create_interface()
212
+ app.launch(share=True)
vectordb-chroma/gformdb/355c5a5d-b607-4435-a4cf-2358574ab34b/data_level0.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f18abd8c514282db82706e52b0a33ed659cd534e925a6f149deb7af9ce34bd8e
3
+ size 6284000
vectordb-chroma/gformdb/355c5a5d-b607-4435-a4cf-2358574ab34b/header.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:effaa959ce2b30070fdafc2fe82096fc46e4ee7561b75920dd3ce43d09679b21
3
+ size 100
vectordb-chroma/gformdb/355c5a5d-b607-4435-a4cf-2358574ab34b/length.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f4c51adf24dfe0cd5ee49b97bab1f5cb06d90edc7fb475d318f9642f8234b2a2
3
+ size 4000
vectordb-chroma/gformdb/355c5a5d-b607-4435-a4cf-2358574ab34b/link_lists.bin ADDED
File without changes
vectordb-chroma/gformdb/chroma.sqlite3 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d97c47728c6565f950ffdd45bc35d7546bdf58ada308a18d8beb6eab53ff2c81
3
+ size 1064960
venv/Include/site/python3.10/greenlet/greenlet.h ADDED
@@ -0,0 +1,164 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ /* -*- indent-tabs-mode: nil; tab-width: 4; -*- */
2
+
3
+ /* Greenlet object interface */
4
+
5
+ #ifndef Py_GREENLETOBJECT_H
6
+ #define Py_GREENLETOBJECT_H
7
+
8
+
9
+ #include <Python.h>
10
+
11
+ #ifdef __cplusplus
12
+ extern "C" {
13
+ #endif
14
+
15
+ /* This is deprecated and undocumented. It does not change. */
16
+ #define GREENLET_VERSION "1.0.0"
17
+
18
+ #ifndef GREENLET_MODULE
19
+ #define implementation_ptr_t void*
20
+ #endif
21
+
22
+ typedef struct _greenlet {
23
+ PyObject_HEAD
24
+ PyObject* weakreflist;
25
+ PyObject* dict;
26
+ implementation_ptr_t pimpl;
27
+ } PyGreenlet;
28
+
29
+ #define PyGreenlet_Check(op) (op && PyObject_TypeCheck(op, &PyGreenlet_Type))
30
+
31
+
32
+ /* C API functions */
33
+
34
+ /* Total number of symbols that are exported */
35
+ #define PyGreenlet_API_pointers 12
36
+
37
+ #define PyGreenlet_Type_NUM 0
38
+ #define PyExc_GreenletError_NUM 1
39
+ #define PyExc_GreenletExit_NUM 2
40
+
41
+ #define PyGreenlet_New_NUM 3
42
+ #define PyGreenlet_GetCurrent_NUM 4
43
+ #define PyGreenlet_Throw_NUM 5
44
+ #define PyGreenlet_Switch_NUM 6
45
+ #define PyGreenlet_SetParent_NUM 7
46
+
47
+ #define PyGreenlet_MAIN_NUM 8
48
+ #define PyGreenlet_STARTED_NUM 9
49
+ #define PyGreenlet_ACTIVE_NUM 10
50
+ #define PyGreenlet_GET_PARENT_NUM 11
51
+
52
+ #ifndef GREENLET_MODULE
53
+ /* This section is used by modules that uses the greenlet C API */
54
+ static void** _PyGreenlet_API = NULL;
55
+
56
+ # define PyGreenlet_Type \
57
+ (*(PyTypeObject*)_PyGreenlet_API[PyGreenlet_Type_NUM])
58
+
59
+ # define PyExc_GreenletError \
60
+ ((PyObject*)_PyGreenlet_API[PyExc_GreenletError_NUM])
61
+
62
+ # define PyExc_GreenletExit \
63
+ ((PyObject*)_PyGreenlet_API[PyExc_GreenletExit_NUM])
64
+
65
+ /*
66
+ * PyGreenlet_New(PyObject *args)
67
+ *
68
+ * greenlet.greenlet(run, parent=None)
69
+ */
70
+ # define PyGreenlet_New \
71
+ (*(PyGreenlet * (*)(PyObject * run, PyGreenlet * parent)) \
72
+ _PyGreenlet_API[PyGreenlet_New_NUM])
73
+
74
+ /*
75
+ * PyGreenlet_GetCurrent(void)
76
+ *
77
+ * greenlet.getcurrent()
78
+ */
79
+ # define PyGreenlet_GetCurrent \
80
+ (*(PyGreenlet * (*)(void)) _PyGreenlet_API[PyGreenlet_GetCurrent_NUM])
81
+
82
+ /*
83
+ * PyGreenlet_Throw(
84
+ * PyGreenlet *greenlet,
85
+ * PyObject *typ,
86
+ * PyObject *val,
87
+ * PyObject *tb)
88
+ *
89
+ * g.throw(...)
90
+ */
91
+ # define PyGreenlet_Throw \
92
+ (*(PyObject * (*)(PyGreenlet * self, \
93
+ PyObject * typ, \
94
+ PyObject * val, \
95
+ PyObject * tb)) \
96
+ _PyGreenlet_API[PyGreenlet_Throw_NUM])
97
+
98
+ /*
99
+ * PyGreenlet_Switch(PyGreenlet *greenlet, PyObject *args)
100
+ *
101
+ * g.switch(*args, **kwargs)
102
+ */
103
+ # define PyGreenlet_Switch \
104
+ (*(PyObject * \
105
+ (*)(PyGreenlet * greenlet, PyObject * args, PyObject * kwargs)) \
106
+ _PyGreenlet_API[PyGreenlet_Switch_NUM])
107
+
108
+ /*
109
+ * PyGreenlet_SetParent(PyObject *greenlet, PyObject *new_parent)
110
+ *
111
+ * g.parent = new_parent
112
+ */
113
+ # define PyGreenlet_SetParent \
114
+ (*(int (*)(PyGreenlet * greenlet, PyGreenlet * nparent)) \
115
+ _PyGreenlet_API[PyGreenlet_SetParent_NUM])
116
+
117
+ /*
118
+ * PyGreenlet_GetParent(PyObject* greenlet)
119
+ *
120
+ * return greenlet.parent;
121
+ *
122
+ * This could return NULL even if there is no exception active.
123
+ * If it does not return NULL, you are responsible for decrementing the
124
+ * reference count.
125
+ */
126
+ # define PyGreenlet_GetParent \
127
+ (*(PyGreenlet* (*)(PyGreenlet*)) \
128
+ _PyGreenlet_API[PyGreenlet_GET_PARENT_NUM])
129
+
130
+ /*
131
+ * deprecated, undocumented alias.
132
+ */
133
+ # define PyGreenlet_GET_PARENT PyGreenlet_GetParent
134
+
135
+ # define PyGreenlet_MAIN \
136
+ (*(int (*)(PyGreenlet*)) \
137
+ _PyGreenlet_API[PyGreenlet_MAIN_NUM])
138
+
139
+ # define PyGreenlet_STARTED \
140
+ (*(int (*)(PyGreenlet*)) \
141
+ _PyGreenlet_API[PyGreenlet_STARTED_NUM])
142
+
143
+ # define PyGreenlet_ACTIVE \
144
+ (*(int (*)(PyGreenlet*)) \
145
+ _PyGreenlet_API[PyGreenlet_ACTIVE_NUM])
146
+
147
+
148
+
149
+
150
+ /* Macro that imports greenlet and initializes C API */
151
+ /* NOTE: This has actually moved to ``greenlet._greenlet._C_API``, but we
152
+ keep the older definition to be sure older code that might have a copy of
153
+ the header still works. */
154
+ # define PyGreenlet_Import() \
155
+ { \
156
+ _PyGreenlet_API = (void**)PyCapsule_Import("greenlet._C_API", 0); \
157
+ }
158
+
159
+ #endif /* GREENLET_MODULE */
160
+
161
+ #ifdef __cplusplus
162
+ }
163
+ #endif
164
+ #endif /* !Py_GREENLETOBJECT_H */
venv/Lib/site-packages/Deprecated-1.2.14.dist-info/INSTALLER ADDED
@@ -0,0 +1 @@
 
 
1
+ pip
venv/Lib/site-packages/Deprecated-1.2.14.dist-info/LICENSE.rst ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ The MIT License (MIT)
2
+
3
+ Copyright (c) 2017 Laurent LAPORTE
4
+
5
+ Permission is hereby granted, free of charge, to any person obtaining a copy
6
+ of this software and associated documentation files (the "Software"), to deal
7
+ in the Software without restriction, including without limitation the rights
8
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
9
+ copies of the Software, and to permit persons to whom the Software is
10
+ furnished to do so, subject to the following conditions:
11
+
12
+ The above copyright notice and this permission notice shall be included in all
13
+ copies or substantial portions of the Software.
14
+
15
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
18
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
20
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
21
+ SOFTWARE.
venv/Lib/site-packages/Deprecated-1.2.14.dist-info/METADATA ADDED
@@ -0,0 +1,181 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ Metadata-Version: 2.1
2
+ Name: Deprecated
3
+ Version: 1.2.14
4
+ Summary: Python @deprecated decorator to deprecate old python classes, functions or methods.
5
+ Home-page: https://github.com/tantale/deprecated
6
+ Author: Laurent LAPORTE
7
+ Author-email: tantale.solutions@gmail.com
8
+ License: MIT
9
+ Project-URL: Documentation, https://deprecated.readthedocs.io/en/latest/
10
+ Project-URL: Source, https://github.com/tantale/deprecated
11
+ Project-URL: Bug Tracker, https://github.com/tantale/deprecated/issues
12
+ Keywords: deprecate,deprecated,deprecation,warning,warn,decorator
13
+ Platform: any
14
+ Classifier: Development Status :: 5 - Production/Stable
15
+ Classifier: Environment :: Web Environment
16
+ Classifier: Intended Audience :: Developers
17
+ Classifier: License :: OSI Approved :: MIT License
18
+ Classifier: Operating System :: OS Independent
19
+ Classifier: Programming Language :: Python
20
+ Classifier: Programming Language :: Python :: 2
21
+ Classifier: Programming Language :: Python :: 2.7
22
+ Classifier: Programming Language :: Python :: 3
23
+ Classifier: Programming Language :: Python :: 3.4
24
+ Classifier: Programming Language :: Python :: 3.5
25
+ Classifier: Programming Language :: Python :: 3.6
26
+ Classifier: Programming Language :: Python :: 3.7
27
+ Classifier: Programming Language :: Python :: 3.8
28
+ Classifier: Programming Language :: Python :: 3.9
29
+ Classifier: Programming Language :: Python :: 3.10
30
+ Classifier: Programming Language :: Python :: 3.11
31
+ Classifier: Programming Language :: Python :: 3.12
32
+ Classifier: Topic :: Software Development :: Libraries :: Python Modules
33
+ Requires-Python: >=2.7, !=3.0.*, !=3.1.*, !=3.2.*, !=3.3.*
34
+ Description-Content-Type: text/x-rst
35
+ License-File: LICENSE.rst
36
+ Requires-Dist: wrapt (<2,>=1.10)
37
+ Provides-Extra: dev
38
+ Requires-Dist: tox ; extra == 'dev'
39
+ Requires-Dist: PyTest ; extra == 'dev'
40
+ Requires-Dist: PyTest-Cov ; extra == 'dev'
41
+ Requires-Dist: bump2version (<1) ; extra == 'dev'
42
+ Requires-Dist: sphinx (<2) ; extra == 'dev'
43
+
44
+
45
+ Deprecated Library
46
+ ------------------
47
+
48
+ Deprecated is Easy to Use
49
+ `````````````````````````
50
+
51
+ If you need to mark a function or a method as deprecated,
52
+ you can use the ``@deprecated`` decorator:
53
+
54
+ Save in a hello.py:
55
+
56
+ .. code:: python
57
+
58
+ from deprecated import deprecated
59
+
60
+
61
+ @deprecated(version='1.2.1', reason="You should use another function")
62
+ def some_old_function(x, y):
63
+ return x + y
64
+
65
+
66
+ class SomeClass(object):
67
+ @deprecated(version='1.3.0', reason="This method is deprecated")
68
+ def some_old_method(self, x, y):
69
+ return x + y
70
+
71
+
72
+ some_old_function(12, 34)
73
+ obj = SomeClass()
74
+ obj.some_old_method(5, 8)
75
+
76
+
77
+ And Easy to Setup
78
+ `````````````````
79
+
80
+ And run it:
81
+
82
+ .. code:: bash
83
+
84
+ $ pip install Deprecated
85
+ $ python hello.py
86
+ hello.py:15: DeprecationWarning: Call to deprecated function (or staticmethod) some_old_function.
87
+ (You should use another function) -- Deprecated since version 1.2.0.
88
+ some_old_function(12, 34)
89
+ hello.py:17: DeprecationWarning: Call to deprecated method some_old_method.
90
+ (This method is deprecated) -- Deprecated since version 1.3.0.
91
+ obj.some_old_method(5, 8)
92
+
93
+
94
+ You can document your code
95
+ ``````````````````````````
96
+
97
+ Have you ever wonder how to document that some functions, classes, methods, etc. are deprecated?
98
+ This is now possible with the integrated Sphinx directives:
99
+
100
+ For instance, in hello_sphinx.py:
101
+
102
+ .. code:: python
103
+
104
+ from deprecated.sphinx import deprecated
105
+ from deprecated.sphinx import versionadded
106
+ from deprecated.sphinx import versionchanged
107
+
108
+
109
+ @versionadded(version='1.0', reason="This function is new")
110
+ def function_one():
111
+ '''This is the function one'''
112
+
113
+
114
+ @versionchanged(version='1.0', reason="This function is modified")
115
+ def function_two():
116
+ '''This is the function two'''
117
+
118
+
119
+ @deprecated(version='1.0', reason="This function will be removed soon")
120
+ def function_three():
121
+ '''This is the function three'''
122
+
123
+
124
+ function_one()
125
+ function_two()
126
+ function_three() # warns
127
+
128
+ help(function_one)
129
+ help(function_two)
130
+ help(function_three)
131
+
132
+
133
+ The result it immediate
134
+ ```````````````````````
135
+
136
+ Run it:
137
+
138
+ .. code:: bash
139
+
140
+ $ python hello_sphinx.py
141
+
142
+ hello_sphinx.py:23: DeprecationWarning: Call to deprecated function (or staticmethod) function_three.
143
+ (This function will be removed soon) -- Deprecated since version 1.0.
144
+ function_three() # warns
145
+
146
+ Help on function function_one in module __main__:
147
+
148
+ function_one()
149
+ This is the function one
150
+
151
+ .. versionadded:: 1.0
152
+ This function is new
153
+
154
+ Help on function function_two in module __main__:
155
+
156
+ function_two()
157
+ This is the function two
158
+
159
+ .. versionchanged:: 1.0
160
+ This function is modified
161
+
162
+ Help on function function_three in module __main__:
163
+
164
+ function_three()
165
+ This is the function three
166
+
167
+ .. deprecated:: 1.0
168
+ This function will be removed soon
169
+
170
+
171
+ Links
172
+ `````
173
+
174
+ * `Python package index (PyPi) <https://pypi.python.org/pypi/deprecated>`_
175
+ * `GitHub website <https://github.com/tantale/deprecated>`_
176
+ * `Read The Docs <https://readthedocs.org/projects/deprecated>`_
177
+ * `EBook on Lulu.com <http://www.lulu.com/commerce/index.php?fBuyContent=21305117>`_
178
+ * `StackOverFlow Q&A <https://stackoverflow.com/a/40301488/1513933>`_
179
+ * `Development version
180
+ <https://github.com/tantale/deprecated/zipball/master#egg=Deprecated-dev>`_
181
+
venv/Lib/site-packages/Deprecated-1.2.14.dist-info/RECORD ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ Deprecated-1.2.14.dist-info/INSTALLER,sha256=zuuue4knoyJ-UwPPXg8fezS7VCrXJQrAP7zeNuwvFQg,4
2
+ Deprecated-1.2.14.dist-info/LICENSE.rst,sha256=HoPt0VvkGbXVveNy4yXlJ_9PmRX1SOfHUxS0H2aZ6Dw,1081
3
+ Deprecated-1.2.14.dist-info/METADATA,sha256=xQYvk5nwOfnkxxRD-VHkpE-sMu0IBHRZ8ayspypfkTs,5354
4
+ Deprecated-1.2.14.dist-info/RECORD,,
5
+ Deprecated-1.2.14.dist-info/WHEEL,sha256=a-zpFRIJzOq5QfuhBzbhiA1eHTzNCJn8OdRvhdNX0Rk,110
6
+ Deprecated-1.2.14.dist-info/top_level.txt,sha256=nHbOYawKPQQE5lQl-toUB1JBRJjUyn_m_Mb8RVJ0RjA,11
7
+ deprecated/__init__.py,sha256=ZphiULqDVrESSB0mLV2WA88JyhQxZSK44zuDGbV5k-g,349
8
+ deprecated/__pycache__/__init__.cpython-310.pyc,,
9
+ deprecated/__pycache__/classic.cpython-310.pyc,,
10
+ deprecated/__pycache__/sphinx.cpython-310.pyc,,
11
+ deprecated/classic.py,sha256=QugmUi7IhBvp2nDvMtyWqFDPRR43-9nfSZG1ZJSDpFM,9880
12
+ deprecated/sphinx.py,sha256=NqQ0oKGcVn6yUe23iGbCieCgvWbEDQSPt9QelbXJnDU,10258
venv/Lib/site-packages/Deprecated-1.2.14.dist-info/WHEEL ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ Wheel-Version: 1.0
2
+ Generator: bdist_wheel (0.40.0)
3
+ Root-Is-Purelib: true
4
+ Tag: py2-none-any
5
+ Tag: py3-none-any
6
+
venv/Lib/site-packages/Deprecated-1.2.14.dist-info/top_level.txt ADDED
@@ -0,0 +1 @@
 
 
1
+ deprecated
venv/Lib/site-packages/MarkupSafe-2.1.5.dist-info/INSTALLER ADDED
@@ -0,0 +1 @@
 
 
1
+ pip
venv/Lib/site-packages/MarkupSafe-2.1.5.dist-info/LICENSE.rst ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ Copyright 2010 Pallets
2
+
3
+ Redistribution and use in source and binary forms, with or without
4
+ modification, are permitted provided that the following conditions are
5
+ met:
6
+
7
+ 1. Redistributions of source code must retain the above copyright
8
+ notice, this list of conditions and the following disclaimer.
9
+
10
+ 2. Redistributions in binary form must reproduce the above copyright
11
+ notice, this list of conditions and the following disclaimer in the
12
+ documentation and/or other materials provided with the distribution.
13
+
14
+ 3. Neither the name of the copyright holder nor the names of its
15
+ contributors may be used to endorse or promote products derived from
16
+ this software without specific prior written permission.
17
+
18
+ THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
19
+ "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
20
+ LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A
21
+ PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
22
+ HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
23
+ SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED
24
+ TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
25
+ PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
26
+ LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
27
+ NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
28
+ SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
venv/Lib/site-packages/MarkupSafe-2.1.5.dist-info/METADATA ADDED
@@ -0,0 +1,93 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ Metadata-Version: 2.1
2
+ Name: MarkupSafe
3
+ Version: 2.1.5
4
+ Summary: Safely add untrusted strings to HTML/XML markup.
5
+ Home-page: https://palletsprojects.com/p/markupsafe/
6
+ Maintainer: Pallets
7
+ Maintainer-email: contact@palletsprojects.com
8
+ License: BSD-3-Clause
9
+ Project-URL: Donate, https://palletsprojects.com/donate
10
+ Project-URL: Documentation, https://markupsafe.palletsprojects.com/
11
+ Project-URL: Changes, https://markupsafe.palletsprojects.com/changes/
12
+ Project-URL: Source Code, https://github.com/pallets/markupsafe/
13
+ Project-URL: Issue Tracker, https://github.com/pallets/markupsafe/issues/
14
+ Project-URL: Chat, https://discord.gg/pallets
15
+ Classifier: Development Status :: 5 - Production/Stable
16
+ Classifier: Environment :: Web Environment
17
+ Classifier: Intended Audience :: Developers
18
+ Classifier: License :: OSI Approved :: BSD License
19
+ Classifier: Operating System :: OS Independent
20
+ Classifier: Programming Language :: Python
21
+ Classifier: Topic :: Internet :: WWW/HTTP :: Dynamic Content
22
+ Classifier: Topic :: Text Processing :: Markup :: HTML
23
+ Requires-Python: >=3.7
24
+ Description-Content-Type: text/x-rst
25
+ License-File: LICENSE.rst
26
+
27
+ MarkupSafe
28
+ ==========
29
+
30
+ MarkupSafe implements a text object that escapes characters so it is
31
+ safe to use in HTML and XML. Characters that have special meanings are
32
+ replaced so that they display as the actual characters. This mitigates
33
+ injection attacks, meaning untrusted user input can safely be displayed
34
+ on a page.
35
+
36
+
37
+ Installing
38
+ ----------
39
+
40
+ Install and update using `pip`_:
41
+
42
+ .. code-block:: text
43
+
44
+ pip install -U MarkupSafe
45
+
46
+ .. _pip: https://pip.pypa.io/en/stable/getting-started/
47
+
48
+
49
+ Examples
50
+ --------
51
+
52
+ .. code-block:: pycon
53
+
54
+ >>> from markupsafe import Markup, escape
55
+
56
+ >>> # escape replaces special characters and wraps in Markup
57
+ >>> escape("<script>alert(document.cookie);</script>")
58
+ Markup('&lt;script&gt;alert(document.cookie);&lt;/script&gt;')
59
+
60
+ >>> # wrap in Markup to mark text "safe" and prevent escaping
61
+ >>> Markup("<strong>Hello</strong>")
62
+ Markup('<strong>hello</strong>')
63
+
64
+ >>> escape(Markup("<strong>Hello</strong>"))
65
+ Markup('<strong>hello</strong>')
66
+
67
+ >>> # Markup is a str subclass
68
+ >>> # methods and operators escape their arguments
69
+ >>> template = Markup("Hello <em>{name}</em>")
70
+ >>> template.format(name='"World"')
71
+ Markup('Hello <em>&#34;World&#34;</em>')
72
+
73
+
74
+ Donate
75
+ ------
76
+
77
+ The Pallets organization develops and supports MarkupSafe and other
78
+ popular packages. In order to grow the community of contributors and
79
+ users, and allow the maintainers to devote more time to the projects,
80
+ `please donate today`_.
81
+
82
+ .. _please donate today: https://palletsprojects.com/donate
83
+
84
+
85
+ Links
86
+ -----
87
+
88
+ - Documentation: https://markupsafe.palletsprojects.com/
89
+ - Changes: https://markupsafe.palletsprojects.com/changes/
90
+ - PyPI Releases: https://pypi.org/project/MarkupSafe/
91
+ - Source Code: https://github.com/pallets/markupsafe/
92
+ - Issue Tracker: https://github.com/pallets/markupsafe/issues/
93
+ - Chat: https://discord.gg/pallets
venv/Lib/site-packages/MarkupSafe-2.1.5.dist-info/RECORD ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ MarkupSafe-2.1.5.dist-info/INSTALLER,sha256=zuuue4knoyJ-UwPPXg8fezS7VCrXJQrAP7zeNuwvFQg,4
2
+ MarkupSafe-2.1.5.dist-info/LICENSE.rst,sha256=RjHsDbX9kKVH4zaBcmTGeYIUM4FG-KyUtKV_lu6MnsQ,1503
3
+ MarkupSafe-2.1.5.dist-info/METADATA,sha256=icNlaniV7YIQZ1BScCVqNaRtm7MAgfw8d3OBmoSVyAY,3096
4
+ MarkupSafe-2.1.5.dist-info/RECORD,,
5
+ MarkupSafe-2.1.5.dist-info/WHEEL,sha256=5JPYeYl5ZdvdSkrGS4u21mmpPzpFx42qrXOSIgWf4pg,102
6
+ MarkupSafe-2.1.5.dist-info/top_level.txt,sha256=qy0Plje5IJuvsCBjejJyhDCjEAdcDLK_2agVcex8Z6U,11
7
+ markupsafe/__init__.py,sha256=m1ysNeqf55zbEoJtaovca40ivrkEFolPlw5bGoC5Gi4,11290
8
+ markupsafe/__pycache__/__init__.cpython-310.pyc,,
9
+ markupsafe/__pycache__/_native.cpython-310.pyc,,
10
+ markupsafe/_native.py,sha256=_Q7UsXCOvgdonCgqG3l5asANI6eo50EKnDM-mlwEC5M,1776
11
+ markupsafe/_speedups.c,sha256=n3jzzaJwXcoN8nTFyA53f3vSqsWK2vujI-v6QYifjhQ,7403
12
+ markupsafe/_speedups.cp310-win_amd64.pyd,sha256=f7I42eA3ZdDubzr--gvmEu2tzH32l9kfbhyWzWLfI7o,15872
13
+ markupsafe/_speedups.pyi,sha256=f5QtwIOP0eLrxh2v5p6SmaYmlcHIGIfmz0DovaqL0OU,238
14
+ markupsafe/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
venv/Lib/site-packages/MarkupSafe-2.1.5.dist-info/WHEEL ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ Wheel-Version: 1.0
2
+ Generator: bdist_wheel (0.42.0)
3
+ Root-Is-Purelib: false
4
+ Tag: cp310-cp310-win_amd64
5
+
venv/Lib/site-packages/MarkupSafe-2.1.5.dist-info/top_level.txt ADDED
@@ -0,0 +1 @@
 
 
1
+ markupsafe
venv/Lib/site-packages/PIL/BdfFontFile.py ADDED
@@ -0,0 +1,133 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ bdf_slant = {
30
+ "R": "Roman",
31
+ "I": "Italic",
32
+ "O": "Oblique",
33
+ "RI": "Reverse Italic",
34
+ "RO": "Reverse Oblique",
35
+ "OT": "Other",
36
+ }
37
+
38
+ bdf_spacing = {"P": "Proportional", "M": "Monospaced", "C": "Cell"}
39
+
40
+
41
+ def bdf_char(
42
+ f: BinaryIO,
43
+ ) -> (
44
+ tuple[
45
+ str,
46
+ int,
47
+ tuple[tuple[int, int], tuple[int, int, int, int], tuple[int, int, int, int]],
48
+ Image.Image,
49
+ ]
50
+ | None
51
+ ):
52
+ # skip to STARTCHAR
53
+ while True:
54
+ s = f.readline()
55
+ if not s:
56
+ return None
57
+ if s[:9] == b"STARTCHAR":
58
+ break
59
+ id = s[9:].strip().decode("ascii")
60
+
61
+ # load symbol properties
62
+ props = {}
63
+ while True:
64
+ s = f.readline()
65
+ if not s or s[:6] == b"BITMAP":
66
+ break
67
+ i = s.find(b" ")
68
+ props[s[:i].decode("ascii")] = s[i + 1 : -1].decode("ascii")
69
+
70
+ # load bitmap
71
+ bitmap = bytearray()
72
+ while True:
73
+ s = f.readline()
74
+ if not s or s[:7] == b"ENDCHAR":
75
+ break
76
+ bitmap += s[:-1]
77
+
78
+ # The word BBX
79
+ # followed by the width in x (BBw), height in y (BBh),
80
+ # and x and y displacement (BBxoff0, BByoff0)
81
+ # of the lower left corner from the origin of the character.
82
+ width, height, x_disp, y_disp = (int(p) for p in props["BBX"].split())
83
+
84
+ # The word DWIDTH
85
+ # followed by the width in x and y of the character in device pixels.
86
+ dwx, dwy = (int(p) for p in props["DWIDTH"].split())
87
+
88
+ bbox = (
89
+ (dwx, dwy),
90
+ (x_disp, -y_disp - height, width + x_disp, -y_disp),
91
+ (0, 0, width, height),
92
+ )
93
+
94
+ try:
95
+ im = Image.frombytes("1", (width, height), bitmap, "hex", "1")
96
+ except ValueError:
97
+ # deal with zero-width characters
98
+ im = Image.new("1", (width, height))
99
+
100
+ return id, int(props["ENCODING"]), bbox, im
101
+
102
+
103
+ class BdfFontFile(FontFile.FontFile):
104
+ """Font file plugin for the X11 BDF format."""
105
+
106
+ def __init__(self, fp: BinaryIO) -> None:
107
+ super().__init__()
108
+
109
+ s = fp.readline()
110
+ if s[:13] != b"STARTFONT 2.1":
111
+ msg = "not a valid BDF file"
112
+ raise SyntaxError(msg)
113
+
114
+ props = {}
115
+ comments = []
116
+
117
+ while True:
118
+ s = fp.readline()
119
+ if not s or s[:13] == b"ENDPROPERTIES":
120
+ break
121
+ i = s.find(b" ")
122
+ props[s[:i].decode("ascii")] = s[i + 1 : -1].decode("ascii")
123
+ if s[:i] in [b"COMMENT", b"COPYRIGHT"]:
124
+ if s.find(b"LogicalFontDescription") < 0:
125
+ comments.append(s[i + 1 : -1].decode("ascii"))
126
+
127
+ while True:
128
+ c = bdf_char(fp)
129
+ if not c:
130
+ break
131
+ id, ch, (xy, dst, src), im = c
132
+ if 0 <= ch < len(self.glyph):
133
+ self.glyph[ch] = xy, dst, src, im
venv/Lib/site-packages/PIL/BlpImagePlugin.py ADDED
@@ -0,0 +1,493 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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[:4] in (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
+
263
+ self.fp.seek(5, os.SEEK_CUR)
264
+ (self._blp_alpha_depth,) = struct.unpack("<b", self.fp.read(1))
265
+
266
+ self.fp.seek(2, os.SEEK_CUR)
267
+ self._size = struct.unpack("<II", self.fp.read(8))
268
+
269
+ if self.magic in (b"BLP1", b"BLP2"):
270
+ decoder = self.magic.decode()
271
+ else:
272
+ msg = f"Bad BLP magic {repr(self.magic)}"
273
+ raise BLPFormatError(msg)
274
+
275
+ self._mode = "RGBA" if self._blp_alpha_depth else "RGB"
276
+ self.tile = [ImageFile._Tile(decoder, (0, 0) + self.size, 0, (self.mode, 0, 1))]
277
+
278
+
279
+ class _BLPBaseDecoder(ImageFile.PyDecoder):
280
+ _pulls_fd = True
281
+
282
+ def decode(self, buffer: bytes | Image.SupportsArrayInterface) -> tuple[int, int]:
283
+ try:
284
+ self._read_blp_header()
285
+ self._load()
286
+ except struct.error as e:
287
+ msg = "Truncated BLP file"
288
+ raise OSError(msg) from e
289
+ return -1, 0
290
+
291
+ @abc.abstractmethod
292
+ def _load(self) -> None:
293
+ pass
294
+
295
+ def _read_blp_header(self) -> None:
296
+ assert self.fd is not None
297
+ self.fd.seek(4)
298
+ (self._blp_compression,) = struct.unpack("<i", self._safe_read(4))
299
+
300
+ (self._blp_encoding,) = struct.unpack("<b", self._safe_read(1))
301
+ (self._blp_alpha_depth,) = struct.unpack("<b", self._safe_read(1))
302
+ (self._blp_alpha_encoding,) = struct.unpack("<b", self._safe_read(1))
303
+ self.fd.seek(1, os.SEEK_CUR) # mips
304
+
305
+ self.size = struct.unpack("<II", self._safe_read(8))
306
+
307
+ if isinstance(self, BLP1Decoder):
308
+ # Only present for BLP1
309
+ (self._blp_encoding,) = struct.unpack("<i", self._safe_read(4))
310
+ self.fd.seek(4, os.SEEK_CUR) # subtype
311
+
312
+ self._blp_offsets = struct.unpack("<16I", self._safe_read(16 * 4))
313
+ self._blp_lengths = struct.unpack("<16I", self._safe_read(16 * 4))
314
+
315
+ def _safe_read(self, length: int) -> bytes:
316
+ assert self.fd is not None
317
+ return ImageFile._safe_read(self.fd, length)
318
+
319
+ def _read_palette(self) -> list[tuple[int, int, int, int]]:
320
+ ret = []
321
+ for i in range(256):
322
+ try:
323
+ b, g, r, a = struct.unpack("<4B", self._safe_read(4))
324
+ except struct.error:
325
+ break
326
+ ret.append((b, g, r, a))
327
+ return ret
328
+
329
+ def _read_bgra(self, palette: list[tuple[int, int, int, int]]) -> bytearray:
330
+ data = bytearray()
331
+ _data = BytesIO(self._safe_read(self._blp_lengths[0]))
332
+ while True:
333
+ try:
334
+ (offset,) = struct.unpack("<B", _data.read(1))
335
+ except struct.error:
336
+ break
337
+ b, g, r, a = palette[offset]
338
+ d: tuple[int, ...] = (r, g, b)
339
+ if self._blp_alpha_depth:
340
+ d += (a,)
341
+ data.extend(d)
342
+ return data
343
+
344
+
345
+ class BLP1Decoder(_BLPBaseDecoder):
346
+ def _load(self) -> None:
347
+ if self._blp_compression == Format.JPEG:
348
+ self._decode_jpeg_stream()
349
+
350
+ elif self._blp_compression == 1:
351
+ if self._blp_encoding in (4, 5):
352
+ palette = self._read_palette()
353
+ data = self._read_bgra(palette)
354
+ self.set_as_raw(data)
355
+ else:
356
+ msg = f"Unsupported BLP encoding {repr(self._blp_encoding)}"
357
+ raise BLPFormatError(msg)
358
+ else:
359
+ msg = f"Unsupported BLP compression {repr(self._blp_encoding)}"
360
+ raise BLPFormatError(msg)
361
+
362
+ def _decode_jpeg_stream(self) -> None:
363
+ from .JpegImagePlugin import JpegImageFile
364
+
365
+ (jpeg_header_size,) = struct.unpack("<I", self._safe_read(4))
366
+ jpeg_header = self._safe_read(jpeg_header_size)
367
+ assert self.fd is not None
368
+ self._safe_read(self._blp_offsets[0] - self.fd.tell()) # What IS this?
369
+ data = self._safe_read(self._blp_lengths[0])
370
+ data = jpeg_header + data
371
+ image = JpegImageFile(BytesIO(data))
372
+ Image._decompression_bomb_check(image.size)
373
+ if image.mode == "CMYK":
374
+ decoder_name, extents, offset, args = image.tile[0]
375
+ assert isinstance(args, tuple)
376
+ image.tile = [
377
+ ImageFile._Tile(decoder_name, extents, offset, (args[0], "CMYK"))
378
+ ]
379
+ r, g, b = image.convert("RGB").split()
380
+ reversed_image = Image.merge("RGB", (b, g, r))
381
+ self.set_as_raw(reversed_image.tobytes())
382
+
383
+
384
+ class BLP2Decoder(_BLPBaseDecoder):
385
+ def _load(self) -> None:
386
+ palette = self._read_palette()
387
+
388
+ assert self.fd is not None
389
+ self.fd.seek(self._blp_offsets[0])
390
+
391
+ if self._blp_compression == 1:
392
+ # Uncompressed or DirectX compression
393
+
394
+ if self._blp_encoding == Encoding.UNCOMPRESSED:
395
+ data = self._read_bgra(palette)
396
+
397
+ elif self._blp_encoding == Encoding.DXT:
398
+ data = bytearray()
399
+ if self._blp_alpha_encoding == AlphaEncoding.DXT1:
400
+ linesize = (self.size[0] + 3) // 4 * 8
401
+ for yb in range((self.size[1] + 3) // 4):
402
+ for d in decode_dxt1(
403
+ self._safe_read(linesize), alpha=bool(self._blp_alpha_depth)
404
+ ):
405
+ data += d
406
+
407
+ elif self._blp_alpha_encoding == AlphaEncoding.DXT3:
408
+ linesize = (self.size[0] + 3) // 4 * 16
409
+ for yb in range((self.size[1] + 3) // 4):
410
+ for d in decode_dxt3(self._safe_read(linesize)):
411
+ data += d
412
+
413
+ elif self._blp_alpha_encoding == AlphaEncoding.DXT5:
414
+ linesize = (self.size[0] + 3) // 4 * 16
415
+ for yb in range((self.size[1] + 3) // 4):
416
+ for d in decode_dxt5(self._safe_read(linesize)):
417
+ data += d
418
+ else:
419
+ msg = f"Unsupported alpha encoding {repr(self._blp_alpha_encoding)}"
420
+ raise BLPFormatError(msg)
421
+ else:
422
+ msg = f"Unknown BLP encoding {repr(self._blp_encoding)}"
423
+ raise BLPFormatError(msg)
424
+
425
+ else:
426
+ msg = f"Unknown BLP compression {repr(self._blp_compression)}"
427
+ raise BLPFormatError(msg)
428
+
429
+ self.set_as_raw(data)
430
+
431
+
432
+ class BLPEncoder(ImageFile.PyEncoder):
433
+ _pushes_fd = True
434
+
435
+ def _write_palette(self) -> bytes:
436
+ data = b""
437
+ assert self.im is not None
438
+ palette = self.im.getpalette("RGBA", "RGBA")
439
+ for i in range(len(palette) // 4):
440
+ r, g, b, a = palette[i * 4 : (i + 1) * 4]
441
+ data += struct.pack("<4B", b, g, r, a)
442
+ while len(data) < 256 * 4:
443
+ data += b"\x00" * 4
444
+ return data
445
+
446
+ def encode(self, bufsize: int) -> tuple[int, int, bytes]:
447
+ palette_data = self._write_palette()
448
+
449
+ offset = 20 + 16 * 4 * 2 + len(palette_data)
450
+ data = struct.pack("<16I", offset, *((0,) * 15))
451
+
452
+ assert self.im is not None
453
+ w, h = self.im.size
454
+ data += struct.pack("<16I", w * h, *((0,) * 15))
455
+
456
+ data += palette_data
457
+
458
+ for y in range(h):
459
+ for x in range(w):
460
+ data += struct.pack("<B", self.im.getpixel((x, y)))
461
+
462
+ return len(data), 0, data
463
+
464
+
465
+ def _save(im: Image.Image, fp: IO[bytes], filename: str | bytes) -> None:
466
+ if im.mode != "P":
467
+ msg = "Unsupported BLP image mode"
468
+ raise ValueError(msg)
469
+
470
+ magic = b"BLP1" if im.encoderinfo.get("blp_version") == "BLP1" else b"BLP2"
471
+ fp.write(magic)
472
+
473
+ assert im.palette is not None
474
+ fp.write(struct.pack("<i", 1)) # Uncompressed or DirectX compression
475
+ fp.write(struct.pack("<b", Encoding.UNCOMPRESSED))
476
+ fp.write(struct.pack("<b", 1 if im.palette.mode == "RGBA" else 0))
477
+ fp.write(struct.pack("<b", 0)) # alpha encoding
478
+ fp.write(struct.pack("<b", 0)) # mips
479
+ fp.write(struct.pack("<II", *im.size))
480
+ if magic == b"BLP1":
481
+ fp.write(struct.pack("<i", 5))
482
+ fp.write(struct.pack("<i", 0))
483
+
484
+ ImageFile._save(im, fp, [ImageFile._Tile("BLP", (0, 0) + im.size, 0, im.mode)])
485
+
486
+
487
+ Image.register_open(BlpImageFile.format, BlpImageFile, _accept)
488
+ Image.register_extension(BlpImageFile.format, ".blp")
489
+ Image.register_decoder("BLP1", BLP1Decoder)
490
+ Image.register_decoder("BLP2", BLP2Decoder)
491
+
492
+ Image.register_save(BlpImageFile.format, _save)
493
+ Image.register_encoder("BLP", BLPEncoder)
venv/Lib/site-packages/PIL/BmpImagePlugin.py ADDED
@@ -0,0 +1,511 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+
52
+ def _accept(prefix: bytes) -> bool:
53
+ return prefix[:2] == b"BM"
54
+
55
+
56
+ def _dib_accept(prefix: bytes) -> bool:
57
+ return i32(prefix) in [12, 40, 52, 56, 64, 108, 124]
58
+
59
+
60
+ # =============================================================================
61
+ # Image plugin for the Windows BMP format.
62
+ # =============================================================================
63
+ class BmpImageFile(ImageFile.ImageFile):
64
+ """Image plugin for the Windows Bitmap format (BMP)"""
65
+
66
+ # ------------------------------------------------------------- Description
67
+ format_description = "Windows Bitmap"
68
+ format = "BMP"
69
+
70
+ # -------------------------------------------------- BMP Compression values
71
+ COMPRESSIONS = {"RAW": 0, "RLE8": 1, "RLE4": 2, "BITFIELDS": 3, "JPEG": 4, "PNG": 5}
72
+ for k, v in COMPRESSIONS.items():
73
+ vars()[k] = v
74
+
75
+ def _bitmap(self, header: int = 0, offset: int = 0) -> None:
76
+ """Read relevant info about the BMP"""
77
+ read, seek = self.fp.read, self.fp.seek
78
+ if header:
79
+ seek(header)
80
+ # read bmp header size @offset 14 (this is part of the header size)
81
+ file_info: dict[str, bool | int | tuple[int, ...]] = {
82
+ "header_size": i32(read(4)),
83
+ "direction": -1,
84
+ }
85
+
86
+ # -------------------- If requested, read header at a specific position
87
+ # read the rest of the bmp header, without its size
88
+ assert isinstance(file_info["header_size"], int)
89
+ header_data = ImageFile._safe_read(self.fp, file_info["header_size"] - 4)
90
+
91
+ # ------------------------------- Windows Bitmap v2, IBM OS/2 Bitmap v1
92
+ # ----- This format has different offsets because of width/height types
93
+ # 12: BITMAPCOREHEADER/OS21XBITMAPHEADER
94
+ if file_info["header_size"] == 12:
95
+ file_info["width"] = i16(header_data, 0)
96
+ file_info["height"] = i16(header_data, 2)
97
+ file_info["planes"] = i16(header_data, 4)
98
+ file_info["bits"] = i16(header_data, 6)
99
+ file_info["compression"] = self.COMPRESSIONS["RAW"]
100
+ file_info["palette_padding"] = 3
101
+
102
+ # --------------------------------------------- Windows Bitmap v3 to v5
103
+ # 40: BITMAPINFOHEADER
104
+ # 52: BITMAPV2HEADER
105
+ # 56: BITMAPV3HEADER
106
+ # 64: BITMAPCOREHEADER2/OS22XBITMAPHEADER
107
+ # 108: BITMAPV4HEADER
108
+ # 124: BITMAPV5HEADER
109
+ elif file_info["header_size"] in (40, 52, 56, 64, 108, 124):
110
+ file_info["y_flip"] = header_data[7] == 0xFF
111
+ file_info["direction"] = 1 if file_info["y_flip"] else -1
112
+ file_info["width"] = i32(header_data, 0)
113
+ file_info["height"] = (
114
+ i32(header_data, 4)
115
+ if not file_info["y_flip"]
116
+ else 2**32 - i32(header_data, 4)
117
+ )
118
+ file_info["planes"] = i16(header_data, 8)
119
+ file_info["bits"] = i16(header_data, 10)
120
+ file_info["compression"] = i32(header_data, 12)
121
+ # byte size of pixel data
122
+ file_info["data_size"] = i32(header_data, 16)
123
+ file_info["pixels_per_meter"] = (
124
+ i32(header_data, 20),
125
+ i32(header_data, 24),
126
+ )
127
+ file_info["colors"] = i32(header_data, 28)
128
+ file_info["palette_padding"] = 4
129
+ assert isinstance(file_info["pixels_per_meter"], tuple)
130
+ self.info["dpi"] = tuple(x / 39.3701 for x in file_info["pixels_per_meter"])
131
+ if file_info["compression"] == self.COMPRESSIONS["BITFIELDS"]:
132
+ masks = ["r_mask", "g_mask", "b_mask"]
133
+ if len(header_data) >= 48:
134
+ if len(header_data) >= 52:
135
+ masks.append("a_mask")
136
+ else:
137
+ file_info["a_mask"] = 0x0
138
+ for idx, mask in enumerate(masks):
139
+ file_info[mask] = i32(header_data, 36 + idx * 4)
140
+ else:
141
+ # 40 byte headers only have the three components in the
142
+ # bitfields masks, ref:
143
+ # https://msdn.microsoft.com/en-us/library/windows/desktop/dd183376(v=vs.85).aspx
144
+ # See also
145
+ # https://github.com/python-pillow/Pillow/issues/1293
146
+ # There is a 4th component in the RGBQuad, in the alpha
147
+ # location, but it is listed as a reserved component,
148
+ # and it is not generally an alpha channel
149
+ file_info["a_mask"] = 0x0
150
+ for mask in masks:
151
+ file_info[mask] = i32(read(4))
152
+ assert isinstance(file_info["r_mask"], int)
153
+ assert isinstance(file_info["g_mask"], int)
154
+ assert isinstance(file_info["b_mask"], int)
155
+ assert isinstance(file_info["a_mask"], int)
156
+ file_info["rgb_mask"] = (
157
+ file_info["r_mask"],
158
+ file_info["g_mask"],
159
+ file_info["b_mask"],
160
+ )
161
+ file_info["rgba_mask"] = (
162
+ file_info["r_mask"],
163
+ file_info["g_mask"],
164
+ file_info["b_mask"],
165
+ file_info["a_mask"],
166
+ )
167
+ else:
168
+ msg = f"Unsupported BMP header type ({file_info['header_size']})"
169
+ raise OSError(msg)
170
+
171
+ # ------------------ Special case : header is reported 40, which
172
+ # ---------------------- is shorter than real size for bpp >= 16
173
+ assert isinstance(file_info["width"], int)
174
+ assert isinstance(file_info["height"], int)
175
+ self._size = file_info["width"], file_info["height"]
176
+
177
+ # ------- If color count was not found in the header, compute from bits
178
+ assert isinstance(file_info["bits"], int)
179
+ file_info["colors"] = (
180
+ file_info["colors"]
181
+ if file_info.get("colors", 0)
182
+ else (1 << file_info["bits"])
183
+ )
184
+ assert isinstance(file_info["colors"], int)
185
+ if offset == 14 + file_info["header_size"] and file_info["bits"] <= 8:
186
+ offset += 4 * file_info["colors"]
187
+
188
+ # ---------------------- Check bit depth for unusual unsupported values
189
+ self._mode, raw_mode = BIT2MODE.get(file_info["bits"], ("", ""))
190
+ if not self.mode:
191
+ msg = f"Unsupported BMP pixel depth ({file_info['bits']})"
192
+ raise OSError(msg)
193
+
194
+ # ---------------- Process BMP with Bitfields compression (not palette)
195
+ decoder_name = "raw"
196
+ if file_info["compression"] == self.COMPRESSIONS["BITFIELDS"]:
197
+ SUPPORTED: dict[int, list[tuple[int, ...]]] = {
198
+ 32: [
199
+ (0xFF0000, 0xFF00, 0xFF, 0x0),
200
+ (0xFF000000, 0xFF0000, 0xFF00, 0x0),
201
+ (0xFF000000, 0xFF00, 0xFF, 0x0),
202
+ (0xFF000000, 0xFF0000, 0xFF00, 0xFF),
203
+ (0xFF, 0xFF00, 0xFF0000, 0xFF000000),
204
+ (0xFF0000, 0xFF00, 0xFF, 0xFF000000),
205
+ (0xFF000000, 0xFF00, 0xFF, 0xFF0000),
206
+ (0x0, 0x0, 0x0, 0x0),
207
+ ],
208
+ 24: [(0xFF0000, 0xFF00, 0xFF)],
209
+ 16: [(0xF800, 0x7E0, 0x1F), (0x7C00, 0x3E0, 0x1F)],
210
+ }
211
+ MASK_MODES = {
212
+ (32, (0xFF0000, 0xFF00, 0xFF, 0x0)): "BGRX",
213
+ (32, (0xFF000000, 0xFF0000, 0xFF00, 0x0)): "XBGR",
214
+ (32, (0xFF000000, 0xFF00, 0xFF, 0x0)): "BGXR",
215
+ (32, (0xFF000000, 0xFF0000, 0xFF00, 0xFF)): "ABGR",
216
+ (32, (0xFF, 0xFF00, 0xFF0000, 0xFF000000)): "RGBA",
217
+ (32, (0xFF0000, 0xFF00, 0xFF, 0xFF000000)): "BGRA",
218
+ (32, (0xFF000000, 0xFF00, 0xFF, 0xFF0000)): "BGAR",
219
+ (32, (0x0, 0x0, 0x0, 0x0)): "BGRA",
220
+ (24, (0xFF0000, 0xFF00, 0xFF)): "BGR",
221
+ (16, (0xF800, 0x7E0, 0x1F)): "BGR;16",
222
+ (16, (0x7C00, 0x3E0, 0x1F)): "BGR;15",
223
+ }
224
+ if file_info["bits"] in SUPPORTED:
225
+ if (
226
+ file_info["bits"] == 32
227
+ and file_info["rgba_mask"] in SUPPORTED[file_info["bits"]]
228
+ ):
229
+ assert isinstance(file_info["rgba_mask"], tuple)
230
+ raw_mode = MASK_MODES[(file_info["bits"], file_info["rgba_mask"])]
231
+ self._mode = "RGBA" if "A" in raw_mode else self.mode
232
+ elif (
233
+ file_info["bits"] in (24, 16)
234
+ and file_info["rgb_mask"] in SUPPORTED[file_info["bits"]]
235
+ ):
236
+ assert isinstance(file_info["rgb_mask"], tuple)
237
+ raw_mode = MASK_MODES[(file_info["bits"], file_info["rgb_mask"])]
238
+ else:
239
+ msg = "Unsupported BMP bitfields layout"
240
+ raise OSError(msg)
241
+ else:
242
+ msg = "Unsupported BMP bitfields layout"
243
+ raise OSError(msg)
244
+ elif file_info["compression"] == self.COMPRESSIONS["RAW"]:
245
+ if file_info["bits"] == 32 and header == 22: # 32-bit .cur offset
246
+ raw_mode, self._mode = "BGRA", "RGBA"
247
+ elif file_info["compression"] in (
248
+ self.COMPRESSIONS["RLE8"],
249
+ self.COMPRESSIONS["RLE4"],
250
+ ):
251
+ decoder_name = "bmp_rle"
252
+ else:
253
+ msg = f"Unsupported BMP compression ({file_info['compression']})"
254
+ raise OSError(msg)
255
+
256
+ # --------------- Once the header is processed, process the palette/LUT
257
+ if self.mode == "P": # Paletted for 1, 4 and 8 bit images
258
+ # ---------------------------------------------------- 1-bit images
259
+ if not (0 < file_info["colors"] <= 65536):
260
+ msg = f"Unsupported BMP Palette size ({file_info['colors']})"
261
+ raise OSError(msg)
262
+ else:
263
+ assert isinstance(file_info["palette_padding"], int)
264
+ padding = file_info["palette_padding"]
265
+ palette = read(padding * file_info["colors"])
266
+ grayscale = True
267
+ indices = (
268
+ (0, 255)
269
+ if file_info["colors"] == 2
270
+ else list(range(file_info["colors"]))
271
+ )
272
+
273
+ # ----------------- Check if grayscale and ignore palette if so
274
+ for ind, val in enumerate(indices):
275
+ rgb = palette[ind * padding : ind * padding + 3]
276
+ if rgb != o8(val) * 3:
277
+ grayscale = False
278
+
279
+ # ------- If all colors are gray, white or black, ditch palette
280
+ if grayscale:
281
+ self._mode = "1" if file_info["colors"] == 2 else "L"
282
+ raw_mode = self.mode
283
+ else:
284
+ self._mode = "P"
285
+ self.palette = ImagePalette.raw(
286
+ "BGRX" if padding == 4 else "BGR", palette
287
+ )
288
+
289
+ # ---------------------------- Finally set the tile data for the plugin
290
+ self.info["compression"] = file_info["compression"]
291
+ args: list[Any] = [raw_mode]
292
+ if decoder_name == "bmp_rle":
293
+ args.append(file_info["compression"] == self.COMPRESSIONS["RLE4"])
294
+ else:
295
+ assert isinstance(file_info["width"], int)
296
+ args.append(((file_info["width"] * file_info["bits"] + 31) >> 3) & (~3))
297
+ args.append(file_info["direction"])
298
+ self.tile = [
299
+ ImageFile._Tile(
300
+ decoder_name,
301
+ (0, 0, file_info["width"], file_info["height"]),
302
+ offset or self.fp.tell(),
303
+ tuple(args),
304
+ )
305
+ ]
306
+
307
+ def _open(self) -> None:
308
+ """Open file, check magic number and read header"""
309
+ # read 14 bytes: magic number, filesize, reserved, header final offset
310
+ head_data = self.fp.read(14)
311
+ # choke if the file does not have the required magic bytes
312
+ if not _accept(head_data):
313
+ msg = "Not a BMP file"
314
+ raise SyntaxError(msg)
315
+ # read the start position of the BMP image data (u32)
316
+ offset = i32(head_data, 10)
317
+ # load bitmap information (offset=raster info)
318
+ self._bitmap(offset=offset)
319
+
320
+
321
+ class BmpRleDecoder(ImageFile.PyDecoder):
322
+ _pulls_fd = True
323
+
324
+ def decode(self, buffer: bytes | Image.SupportsArrayInterface) -> tuple[int, int]:
325
+ assert self.fd is not None
326
+ rle4 = self.args[1]
327
+ data = bytearray()
328
+ x = 0
329
+ dest_length = self.state.xsize * self.state.ysize
330
+ while len(data) < dest_length:
331
+ pixels = self.fd.read(1)
332
+ byte = self.fd.read(1)
333
+ if not pixels or not byte:
334
+ break
335
+ num_pixels = pixels[0]
336
+ if num_pixels:
337
+ # encoded mode
338
+ if x + num_pixels > self.state.xsize:
339
+ # Too much data for row
340
+ num_pixels = max(0, self.state.xsize - x)
341
+ if rle4:
342
+ first_pixel = o8(byte[0] >> 4)
343
+ second_pixel = o8(byte[0] & 0x0F)
344
+ for index in range(num_pixels):
345
+ if index % 2 == 0:
346
+ data += first_pixel
347
+ else:
348
+ data += second_pixel
349
+ else:
350
+ data += byte * num_pixels
351
+ x += num_pixels
352
+ else:
353
+ if byte[0] == 0:
354
+ # end of line
355
+ while len(data) % self.state.xsize != 0:
356
+ data += b"\x00"
357
+ x = 0
358
+ elif byte[0] == 1:
359
+ # end of bitmap
360
+ break
361
+ elif byte[0] == 2:
362
+ # delta
363
+ bytes_read = self.fd.read(2)
364
+ if len(bytes_read) < 2:
365
+ break
366
+ right, up = self.fd.read(2)
367
+ data += b"\x00" * (right + up * self.state.xsize)
368
+ x = len(data) % self.state.xsize
369
+ else:
370
+ # absolute mode
371
+ if rle4:
372
+ # 2 pixels per byte
373
+ byte_count = byte[0] // 2
374
+ bytes_read = self.fd.read(byte_count)
375
+ for byte_read in bytes_read:
376
+ data += o8(byte_read >> 4)
377
+ data += o8(byte_read & 0x0F)
378
+ else:
379
+ byte_count = byte[0]
380
+ bytes_read = self.fd.read(byte_count)
381
+ data += bytes_read
382
+ if len(bytes_read) < byte_count:
383
+ break
384
+ x += byte[0]
385
+
386
+ # align to 16-bit word boundary
387
+ if self.fd.tell() % 2 != 0:
388
+ self.fd.seek(1, os.SEEK_CUR)
389
+ rawmode = "L" if self.mode == "L" else "P"
390
+ self.set_as_raw(bytes(data), rawmode, (0, self.args[-1]))
391
+ return -1, 0
392
+
393
+
394
+ # =============================================================================
395
+ # Image plugin for the DIB format (BMP alias)
396
+ # =============================================================================
397
+ class DibImageFile(BmpImageFile):
398
+ format = "DIB"
399
+ format_description = "Windows Bitmap"
400
+
401
+ def _open(self) -> None:
402
+ self._bitmap()
403
+
404
+
405
+ #
406
+ # --------------------------------------------------------------------
407
+ # Write BMP file
408
+
409
+
410
+ SAVE = {
411
+ "1": ("1", 1, 2),
412
+ "L": ("L", 8, 256),
413
+ "P": ("P", 8, 256),
414
+ "RGB": ("BGR", 24, 0),
415
+ "RGBA": ("BGRA", 32, 0),
416
+ }
417
+
418
+
419
+ def _dib_save(im: Image.Image, fp: IO[bytes], filename: str | bytes) -> None:
420
+ _save(im, fp, filename, False)
421
+
422
+
423
+ def _save(
424
+ im: Image.Image, fp: IO[bytes], filename: str | bytes, bitmap_header: bool = True
425
+ ) -> None:
426
+ try:
427
+ rawmode, bits, colors = SAVE[im.mode]
428
+ except KeyError as e:
429
+ msg = f"cannot write mode {im.mode} as BMP"
430
+ raise OSError(msg) from e
431
+
432
+ info = im.encoderinfo
433
+
434
+ dpi = info.get("dpi", (96, 96))
435
+
436
+ # 1 meter == 39.3701 inches
437
+ ppm = tuple(int(x * 39.3701 + 0.5) for x in dpi)
438
+
439
+ stride = ((im.size[0] * bits + 7) // 8 + 3) & (~3)
440
+ header = 40 # or 64 for OS/2 version 2
441
+ image = stride * im.size[1]
442
+
443
+ if im.mode == "1":
444
+ palette = b"".join(o8(i) * 4 for i in (0, 255))
445
+ elif im.mode == "L":
446
+ palette = b"".join(o8(i) * 4 for i in range(256))
447
+ elif im.mode == "P":
448
+ palette = im.im.getpalette("RGB", "BGRX")
449
+ colors = len(palette) // 4
450
+ else:
451
+ palette = None
452
+
453
+ # bitmap header
454
+ if bitmap_header:
455
+ offset = 14 + header + colors * 4
456
+ file_size = offset + image
457
+ if file_size > 2**32 - 1:
458
+ msg = "File size is too large for the BMP format"
459
+ raise ValueError(msg)
460
+ fp.write(
461
+ b"BM" # file type (magic)
462
+ + o32(file_size) # file size
463
+ + o32(0) # reserved
464
+ + o32(offset) # image data offset
465
+ )
466
+
467
+ # bitmap info header
468
+ fp.write(
469
+ o32(header) # info header size
470
+ + o32(im.size[0]) # width
471
+ + o32(im.size[1]) # height
472
+ + o16(1) # planes
473
+ + o16(bits) # depth
474
+ + o32(0) # compression (0=uncompressed)
475
+ + o32(image) # size of bitmap
476
+ + o32(ppm[0]) # resolution
477
+ + o32(ppm[1]) # resolution
478
+ + o32(colors) # colors used
479
+ + o32(colors) # colors important
480
+ )
481
+
482
+ fp.write(b"\0" * (header - 40)) # padding (for OS/2 format)
483
+
484
+ if palette:
485
+ fp.write(palette)
486
+
487
+ ImageFile._save(
488
+ im, fp, [ImageFile._Tile("raw", (0, 0) + im.size, 0, (rawmode, stride, -1))]
489
+ )
490
+
491
+
492
+ #
493
+ # --------------------------------------------------------------------
494
+ # Registry
495
+
496
+
497
+ Image.register_open(BmpImageFile.format, BmpImageFile, _accept)
498
+ Image.register_save(BmpImageFile.format, _save)
499
+
500
+ Image.register_extension(BmpImageFile.format, ".bmp")
501
+
502
+ Image.register_mime(BmpImageFile.format, "image/bmp")
503
+
504
+ Image.register_decoder("bmp_rle", BmpRleDecoder)
505
+
506
+ Image.register_open(DibImageFile.format, DibImageFile, _dib_accept)
507
+ Image.register_save(DibImageFile.format, _dib_save)
508
+
509
+ Image.register_extension(DibImageFile.format, ".dib")
510
+
511
+ Image.register_mime(DibImageFile.format, "image/bmp")
venv/Lib/site-packages/PIL/BufrStubImagePlugin.py ADDED
@@ -0,0 +1,76 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ from typing import IO
14
+
15
+ from . import Image, ImageFile
16
+
17
+ _handler = None
18
+
19
+
20
+ def register_handler(handler: ImageFile.StubHandler | None) -> None:
21
+ """
22
+ Install application-specific BUFR image handler.
23
+
24
+ :param handler: Handler object.
25
+ """
26
+ global _handler
27
+ _handler = handler
28
+
29
+
30
+ # --------------------------------------------------------------------
31
+ # Image adapter
32
+
33
+
34
+ def _accept(prefix: bytes) -> bool:
35
+ return prefix[:4] == b"BUFR" or prefix[:4] == b"ZCZC"
36
+
37
+
38
+ class BufrStubImageFile(ImageFile.StubImageFile):
39
+ format = "BUFR"
40
+ format_description = "BUFR"
41
+
42
+ def _open(self) -> None:
43
+ offset = self.fp.tell()
44
+
45
+ if not _accept(self.fp.read(4)):
46
+ msg = "Not a BUFR file"
47
+ raise SyntaxError(msg)
48
+
49
+ self.fp.seek(offset)
50
+
51
+ # make something up
52
+ self._mode = "F"
53
+ self._size = 1, 1
54
+
55
+ loader = self._load()
56
+ if loader:
57
+ loader.open(self)
58
+
59
+ def _load(self) -> ImageFile.StubHandler | None:
60
+ return _handler
61
+
62
+
63
+ def _save(im: Image.Image, fp: IO[bytes], filename: str | bytes) -> None:
64
+ if _handler is None or not hasattr(_handler, "save"):
65
+ msg = "BUFR save handler not installed"
66
+ raise OSError(msg)
67
+ _handler.save(im, fp, filename)
68
+
69
+
70
+ # --------------------------------------------------------------------
71
+ # Registry
72
+
73
+ Image.register_open(BufrStubImageFile.format, BufrStubImageFile, _accept)
74
+ Image.register_save(BufrStubImageFile.format, _save)
75
+
76
+ 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, ImageFile
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[:4] == 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
+ offset = self.fp.tell()
42
+
43
+ # check magic
44
+ s = self.fp.read(6)
45
+ if not _accept(s):
46
+ msg = "not a CUR file"
47
+ raise SyntaxError(msg)
48
+
49
+ # pick the largest cursor in the file
50
+ m = b""
51
+ for i in range(i16(s, 4)):
52
+ s = self.fp.read(16)
53
+ if not m:
54
+ m = s
55
+ elif s[0] > m[0] and s[1] > m[1]:
56
+ m = s
57
+ if not m:
58
+ msg = "No cursors were found"
59
+ raise TypeError(msg)
60
+
61
+ # load as bitmap
62
+ self._bitmap(i32(m, 12) + offset)
63
+
64
+ # patch up the bitmap height
65
+ self._size = self.size[0], self.size[1] // 2
66
+ d, e, o, a = self.tile[0]
67
+ self.tile[0] = ImageFile._Tile(d, (0, 0) + self.size, o, a)
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,80 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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 .PcxImagePlugin import PcxImageFile
28
+
29
+ MAGIC = 0x3ADE68B1 # QUIZ: what's this value, then?
30
+
31
+
32
+ def _accept(prefix: bytes) -> bool:
33
+ return len(prefix) >= 4 and i32(prefix) == MAGIC
34
+
35
+
36
+ ##
37
+ # Image plugin for the Intel DCX format.
38
+
39
+
40
+ class DcxImageFile(PcxImageFile):
41
+ format = "DCX"
42
+ format_description = "Intel DCX"
43
+ _close_exclusive_fp_after_loading = False
44
+
45
+ def _open(self) -> None:
46
+ # Header
47
+ s = self.fp.read(4)
48
+ if not _accept(s):
49
+ msg = "not a DCX file"
50
+ raise SyntaxError(msg)
51
+
52
+ # Component directory
53
+ self._offset = []
54
+ for i in range(1024):
55
+ offset = i32(self.fp.read(4))
56
+ if not offset:
57
+ break
58
+ self._offset.append(offset)
59
+
60
+ self._fp = self.fp
61
+ self.frame = -1
62
+ self.n_frames = len(self._offset)
63
+ self.is_animated = self.n_frames > 1
64
+ self.seek(0)
65
+
66
+ def seek(self, frame: int) -> None:
67
+ if not self._seek_check(frame):
68
+ return
69
+ self.frame = frame
70
+ self.fp = self._fp
71
+ self.fp.seek(self._offset[frame])
72
+ PcxImageFile._open(self)
73
+
74
+ def tell(self) -> int:
75
+ return self.frame
76
+
77
+
78
+ Image.register_open(DcxImageFile.format, DcxImageFile, _accept)
79
+
80
+ Image.register_extension(DcxImageFile.format, ".dcx")
venv/Lib/site-packages/PIL/DdsImagePlugin.py ADDED
@@ -0,0 +1,575 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ """
2
+ A Pillow loader 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.BC4_TYPELESS, DXGI_FORMAT.BC4_UNORM):
423
+ self._mode = "L"
424
+ self.pixel_format = "BC4"
425
+ n = 4
426
+ elif dxgi_format in (DXGI_FORMAT.BC5_TYPELESS, DXGI_FORMAT.BC5_UNORM):
427
+ self._mode = "RGB"
428
+ self.pixel_format = "BC5"
429
+ n = 5
430
+ elif dxgi_format == DXGI_FORMAT.BC5_SNORM:
431
+ self._mode = "RGB"
432
+ self.pixel_format = "BC5S"
433
+ n = 5
434
+ elif dxgi_format == DXGI_FORMAT.BC6H_UF16:
435
+ self._mode = "RGB"
436
+ self.pixel_format = "BC6H"
437
+ n = 6
438
+ elif dxgi_format == DXGI_FORMAT.BC6H_SF16:
439
+ self._mode = "RGB"
440
+ self.pixel_format = "BC6HS"
441
+ n = 6
442
+ elif dxgi_format in (
443
+ DXGI_FORMAT.BC7_TYPELESS,
444
+ DXGI_FORMAT.BC7_UNORM,
445
+ DXGI_FORMAT.BC7_UNORM_SRGB,
446
+ ):
447
+ self._mode = "RGBA"
448
+ self.pixel_format = "BC7"
449
+ n = 7
450
+ if dxgi_format == DXGI_FORMAT.BC7_UNORM_SRGB:
451
+ self.info["gamma"] = 1 / 2.2
452
+ elif dxgi_format in (
453
+ DXGI_FORMAT.R8G8B8A8_TYPELESS,
454
+ DXGI_FORMAT.R8G8B8A8_UNORM,
455
+ DXGI_FORMAT.R8G8B8A8_UNORM_SRGB,
456
+ ):
457
+ self._mode = "RGBA"
458
+ if dxgi_format == DXGI_FORMAT.R8G8B8A8_UNORM_SRGB:
459
+ self.info["gamma"] = 1 / 2.2
460
+ else:
461
+ msg = f"Unimplemented DXGI format {dxgi_format}"
462
+ raise NotImplementedError(msg)
463
+ else:
464
+ msg = f"Unimplemented pixel format {repr(fourcc)}"
465
+ raise NotImplementedError(msg)
466
+ else:
467
+ msg = f"Unknown pixel format flags {pfflags}"
468
+ raise NotImplementedError(msg)
469
+
470
+ if n:
471
+ self.tile = [
472
+ ImageFile._Tile("bcn", extents, offset, (n, self.pixel_format))
473
+ ]
474
+ else:
475
+ self.tile = [ImageFile._Tile("raw", extents, 0, rawmode or self.mode)]
476
+
477
+ def load_seek(self, pos: int) -> None:
478
+ pass
479
+
480
+
481
+ class DdsRgbDecoder(ImageFile.PyDecoder):
482
+ _pulls_fd = True
483
+
484
+ def decode(self, buffer: bytes | Image.SupportsArrayInterface) -> tuple[int, int]:
485
+ assert self.fd is not None
486
+ bitcount, masks = self.args
487
+
488
+ # Some masks will be padded with zeros, e.g. R 0b11 G 0b1100
489
+ # Calculate how many zeros each mask is padded with
490
+ mask_offsets = []
491
+ # And the maximum value of each channel without the padding
492
+ mask_totals = []
493
+ for mask in masks:
494
+ offset = 0
495
+ if mask != 0:
496
+ while mask >> (offset + 1) << (offset + 1) == mask:
497
+ offset += 1
498
+ mask_offsets.append(offset)
499
+ mask_totals.append(mask >> offset)
500
+
501
+ data = bytearray()
502
+ bytecount = bitcount // 8
503
+ dest_length = self.state.xsize * self.state.ysize * len(masks)
504
+ while len(data) < dest_length:
505
+ value = int.from_bytes(self.fd.read(bytecount), "little")
506
+ for i, mask in enumerate(masks):
507
+ masked_value = value & mask
508
+ # Remove the zero padding, and scale it to 8 bits
509
+ data += o8(
510
+ int(((masked_value >> mask_offsets[i]) / mask_totals[i]) * 255)
511
+ )
512
+ self.set_as_raw(data)
513
+ return -1, 0
514
+
515
+
516
+ def _save(im: Image.Image, fp: IO[bytes], filename: str | bytes) -> None:
517
+ if im.mode not in ("RGB", "RGBA", "L", "LA"):
518
+ msg = f"cannot write mode {im.mode} as DDS"
519
+ raise OSError(msg)
520
+
521
+ alpha = im.mode[-1] == "A"
522
+ if im.mode[0] == "L":
523
+ pixel_flags = DDPF.LUMINANCE
524
+ rawmode = im.mode
525
+ if alpha:
526
+ rgba_mask = [0x000000FF, 0x000000FF, 0x000000FF]
527
+ else:
528
+ rgba_mask = [0xFF000000, 0xFF000000, 0xFF000000]
529
+ else:
530
+ pixel_flags = DDPF.RGB
531
+ rawmode = im.mode[::-1]
532
+ rgba_mask = [0x00FF0000, 0x0000FF00, 0x000000FF]
533
+
534
+ if alpha:
535
+ r, g, b, a = im.split()
536
+ im = Image.merge("RGBA", (a, r, g, b))
537
+ if alpha:
538
+ pixel_flags |= DDPF.ALPHAPIXELS
539
+ rgba_mask.append(0xFF000000 if alpha else 0)
540
+
541
+ flags = DDSD.CAPS | DDSD.HEIGHT | DDSD.WIDTH | DDSD.PITCH | DDSD.PIXELFORMAT
542
+ bitcount = len(im.getbands()) * 8
543
+ pitch = (im.width * bitcount + 7) // 8
544
+
545
+ fp.write(
546
+ o32(DDS_MAGIC)
547
+ + struct.pack(
548
+ "<7I",
549
+ 124, # header size
550
+ flags, # flags
551
+ im.height,
552
+ im.width,
553
+ pitch,
554
+ 0, # depth
555
+ 0, # mipmaps
556
+ )
557
+ + struct.pack("11I", *((0,) * 11)) # reserved
558
+ # pfsize, pfflags, fourcc, bitcount
559
+ + struct.pack("<4I", 32, pixel_flags, 0, bitcount)
560
+ + struct.pack("<4I", *rgba_mask) # dwRGBABitMask
561
+ + struct.pack("<5I", DDSCAPS.TEXTURE, 0, 0, 0, 0)
562
+ )
563
+ ImageFile._save(
564
+ im, fp, [ImageFile._Tile("raw", (0, 0) + im.size, 0, (rawmode, 0, 1))]
565
+ )
566
+
567
+
568
+ def _accept(prefix: bytes) -> bool:
569
+ return prefix[:4] == b"DDS "
570
+
571
+
572
+ Image.register_open(DdsImageFile.format, DdsImageFile, _accept)
573
+ Image.register_decoder("dds_rgb", DdsRgbDecoder)
574
+ Image.register_save(DdsImageFile.format, _save)
575
+ Image.register_extension(DdsImageFile.format, ".dds")
venv/Lib/site-packages/PIL/EpsImagePlugin.py ADDED
@@ -0,0 +1,474 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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[:4] == b"%!PS" or (len(prefix) >= 4 and i32(prefix) == 0xC6D3D0C5)
174
+
175
+
176
+ ##
177
+ # Image plugin for Encapsulated PostScript. This plugin supports only
178
+ # a few variants of this format.
179
+
180
+
181
+ class EpsImageFile(ImageFile.ImageFile):
182
+ """EPS File Parser for the Python Imaging Library"""
183
+
184
+ format = "EPS"
185
+ format_description = "Encapsulated Postscript"
186
+
187
+ mode_map = {1: "L", 2: "LAB", 3: "RGB", 4: "CMYK"}
188
+
189
+ def _open(self) -> None:
190
+ (length, offset) = self._find_offset(self.fp)
191
+
192
+ # go to offset - start of "%!PS"
193
+ self.fp.seek(offset)
194
+
195
+ self._mode = "RGB"
196
+
197
+ # When reading header comments, the first comment is used.
198
+ # When reading trailer comments, the last comment is used.
199
+ bounding_box: list[int] | None = None
200
+ imagedata_size: tuple[int, int] | None = None
201
+
202
+ byte_arr = bytearray(255)
203
+ bytes_mv = memoryview(byte_arr)
204
+ bytes_read = 0
205
+ reading_header_comments = True
206
+ reading_trailer_comments = False
207
+ trailer_reached = False
208
+
209
+ def check_required_header_comments() -> None:
210
+ """
211
+ The EPS specification requires that some headers exist.
212
+ This should be checked when the header comments formally end,
213
+ when image data starts, or when the file ends, whichever comes first.
214
+ """
215
+ if "PS-Adobe" not in self.info:
216
+ msg = 'EPS header missing "%!PS-Adobe" comment'
217
+ raise SyntaxError(msg)
218
+ if "BoundingBox" not in self.info:
219
+ msg = 'EPS header missing "%%BoundingBox" comment'
220
+ raise SyntaxError(msg)
221
+
222
+ def read_comment(s: str) -> bool:
223
+ nonlocal bounding_box, reading_trailer_comments
224
+ try:
225
+ m = split.match(s)
226
+ except re.error as e:
227
+ msg = "not an EPS file"
228
+ raise SyntaxError(msg) from e
229
+
230
+ if not m:
231
+ return False
232
+
233
+ k, v = m.group(1, 2)
234
+ self.info[k] = v
235
+ if k == "BoundingBox":
236
+ if v == "(atend)":
237
+ reading_trailer_comments = True
238
+ elif not bounding_box or (trailer_reached and reading_trailer_comments):
239
+ try:
240
+ # Note: The DSC spec says that BoundingBox
241
+ # fields should be integers, but some drivers
242
+ # put floating point values there anyway.
243
+ bounding_box = [int(float(i)) for i in v.split()]
244
+ except Exception:
245
+ pass
246
+ return True
247
+
248
+ while True:
249
+ byte = self.fp.read(1)
250
+ if byte == b"":
251
+ # if we didn't read a byte we must be at the end of the file
252
+ if bytes_read == 0:
253
+ if reading_header_comments:
254
+ check_required_header_comments()
255
+ break
256
+ elif byte in b"\r\n":
257
+ # if we read a line ending character, ignore it and parse what
258
+ # we have already read. if we haven't read any other characters,
259
+ # continue reading
260
+ if bytes_read == 0:
261
+ continue
262
+ else:
263
+ # ASCII/hexadecimal lines in an EPS file must not exceed
264
+ # 255 characters, not including line ending characters
265
+ if bytes_read >= 255:
266
+ # only enforce this for lines starting with a "%",
267
+ # otherwise assume it's binary data
268
+ if byte_arr[0] == ord("%"):
269
+ msg = "not an EPS file"
270
+ raise SyntaxError(msg)
271
+ else:
272
+ if reading_header_comments:
273
+ check_required_header_comments()
274
+ reading_header_comments = False
275
+ # reset bytes_read so we can keep reading
276
+ # data until the end of the line
277
+ bytes_read = 0
278
+ byte_arr[bytes_read] = byte[0]
279
+ bytes_read += 1
280
+ continue
281
+
282
+ if reading_header_comments:
283
+ # Load EPS header
284
+
285
+ # if this line doesn't start with a "%",
286
+ # or does start with "%%EndComments",
287
+ # then we've reached the end of the header/comments
288
+ if byte_arr[0] != ord("%") or bytes_mv[:13] == b"%%EndComments":
289
+ check_required_header_comments()
290
+ reading_header_comments = False
291
+ continue
292
+
293
+ s = str(bytes_mv[:bytes_read], "latin-1")
294
+ if not read_comment(s):
295
+ m = field.match(s)
296
+ if m:
297
+ k = m.group(1)
298
+ if k[:8] == "PS-Adobe":
299
+ self.info["PS-Adobe"] = k[9:]
300
+ else:
301
+ self.info[k] = ""
302
+ elif s[0] == "%":
303
+ # handle non-DSC PostScript comments that some
304
+ # tools mistakenly put in the Comments section
305
+ pass
306
+ else:
307
+ msg = "bad EPS header"
308
+ raise OSError(msg)
309
+ elif bytes_mv[:11] == b"%ImageData:":
310
+ # Check for an "ImageData" descriptor
311
+ # https://www.adobe.com/devnet-apps/photoshop/fileformatashtml/#50577413_pgfId-1035096
312
+
313
+ # If we've already read an "ImageData" descriptor,
314
+ # don't read another one.
315
+ if imagedata_size:
316
+ bytes_read = 0
317
+ continue
318
+
319
+ # Values:
320
+ # columns
321
+ # rows
322
+ # bit depth (1 or 8)
323
+ # mode (1: L, 2: LAB, 3: RGB, 4: CMYK)
324
+ # number of padding channels
325
+ # block size (number of bytes per row per channel)
326
+ # binary/ascii (1: binary, 2: ascii)
327
+ # data start identifier (the image data follows after a single line
328
+ # consisting only of this quoted value)
329
+ image_data_values = byte_arr[11:bytes_read].split(None, 7)
330
+ columns, rows, bit_depth, mode_id = (
331
+ int(value) for value in image_data_values[:4]
332
+ )
333
+
334
+ if bit_depth == 1:
335
+ self._mode = "1"
336
+ elif bit_depth == 8:
337
+ try:
338
+ self._mode = self.mode_map[mode_id]
339
+ except ValueError:
340
+ break
341
+ else:
342
+ break
343
+
344
+ # Parse the columns and rows after checking the bit depth and mode
345
+ # in case the bit depth and/or mode are invalid.
346
+ imagedata_size = columns, rows
347
+ elif bytes_mv[:5] == b"%%EOF":
348
+ break
349
+ elif trailer_reached and reading_trailer_comments:
350
+ # Load EPS trailer
351
+ s = str(bytes_mv[:bytes_read], "latin-1")
352
+ read_comment(s)
353
+ elif bytes_mv[:9] == b"%%Trailer":
354
+ trailer_reached = True
355
+ bytes_read = 0
356
+
357
+ # A "BoundingBox" is always required,
358
+ # even if an "ImageData" descriptor size exists.
359
+ if not bounding_box:
360
+ msg = "cannot determine EPS bounding box"
361
+ raise OSError(msg)
362
+
363
+ # An "ImageData" size takes precedence over the "BoundingBox".
364
+ self._size = imagedata_size or (
365
+ bounding_box[2] - bounding_box[0],
366
+ bounding_box[3] - bounding_box[1],
367
+ )
368
+
369
+ self.tile = [
370
+ ImageFile._Tile("eps", (0, 0) + self.size, offset, (length, bounding_box))
371
+ ]
372
+
373
+ def _find_offset(self, fp: IO[bytes]) -> tuple[int, int]:
374
+ s = fp.read(4)
375
+
376
+ if s == b"%!PS":
377
+ # for HEAD without binary preview
378
+ fp.seek(0, io.SEEK_END)
379
+ length = fp.tell()
380
+ offset = 0
381
+ elif i32(s) == 0xC6D3D0C5:
382
+ # FIX for: Some EPS file not handled correctly / issue #302
383
+ # EPS can contain binary data
384
+ # or start directly with latin coding
385
+ # more info see:
386
+ # https://web.archive.org/web/20160528181353/http://partners.adobe.com/public/developer/en/ps/5002.EPSF_Spec.pdf
387
+ s = fp.read(8)
388
+ offset = i32(s)
389
+ length = i32(s, 4)
390
+ else:
391
+ msg = "not an EPS file"
392
+ raise SyntaxError(msg)
393
+
394
+ return length, offset
395
+
396
+ def load(
397
+ self, scale: int = 1, transparency: bool = False
398
+ ) -> Image.core.PixelAccess | None:
399
+ # Load EPS via Ghostscript
400
+ if self.tile:
401
+ self.im = Ghostscript(self.tile, self.size, self.fp, scale, transparency)
402
+ self._mode = self.im.mode
403
+ self._size = self.im.size
404
+ self.tile = []
405
+ return Image.Image.load(self)
406
+
407
+ def load_seek(self, pos: int) -> None:
408
+ # we can't incrementally load, so force ImageFile.parser to
409
+ # use our custom load method by defining this method.
410
+ pass
411
+
412
+
413
+ # --------------------------------------------------------------------
414
+
415
+
416
+ def _save(im: Image.Image, fp: IO[bytes], filename: str | bytes, eps: int = 1) -> None:
417
+ """EPS Writer for the Python Imaging Library."""
418
+
419
+ # make sure image data is available
420
+ im.load()
421
+
422
+ # determine PostScript image mode
423
+ if im.mode == "L":
424
+ operator = (8, 1, b"image")
425
+ elif im.mode == "RGB":
426
+ operator = (8, 3, b"false 3 colorimage")
427
+ elif im.mode == "CMYK":
428
+ operator = (8, 4, b"false 4 colorimage")
429
+ else:
430
+ msg = "image mode is not supported"
431
+ raise ValueError(msg)
432
+
433
+ if eps:
434
+ # write EPS header
435
+ fp.write(b"%!PS-Adobe-3.0 EPSF-3.0\n")
436
+ fp.write(b"%%Creator: PIL 0.1 EpsEncode\n")
437
+ # fp.write("%%CreationDate: %s"...)
438
+ fp.write(b"%%%%BoundingBox: 0 0 %d %d\n" % im.size)
439
+ fp.write(b"%%Pages: 1\n")
440
+ fp.write(b"%%EndComments\n")
441
+ fp.write(b"%%Page: 1 1\n")
442
+ fp.write(b"%%ImageData: %d %d " % im.size)
443
+ fp.write(b'%d %d 0 1 1 "%s"\n' % operator)
444
+
445
+ # image header
446
+ fp.write(b"gsave\n")
447
+ fp.write(b"10 dict begin\n")
448
+ fp.write(b"/buf %d string def\n" % (im.size[0] * operator[1]))
449
+ fp.write(b"%d %d scale\n" % im.size)
450
+ fp.write(b"%d %d 8\n" % im.size) # <= bits
451
+ fp.write(b"[%d 0 0 -%d 0 %d]\n" % (im.size[0], im.size[1], im.size[1]))
452
+ fp.write(b"{ currentfile buf readhexstring pop } bind\n")
453
+ fp.write(operator[2] + b"\n")
454
+ if hasattr(fp, "flush"):
455
+ fp.flush()
456
+
457
+ ImageFile._save(im, fp, [ImageFile._Tile("eps", (0, 0) + im.size, 0, None)])
458
+
459
+ fp.write(b"\n%%%%EndBinary\n")
460
+ fp.write(b"grestore end\n")
461
+ if hasattr(fp, "flush"):
462
+ fp.flush()
463
+
464
+
465
+ # --------------------------------------------------------------------
466
+
467
+
468
+ Image.register_open(EpsImageFile.format, EpsImageFile, _accept)
469
+
470
+ Image.register_save(EpsImageFile.format, _save)
471
+
472
+ Image.register_extensions(EpsImageFile.format, [".ps", ".eps"])
473
+
474
+ Image.register_mime(EpsImageFile.format, "application/postscript")
venv/Lib/site-packages/PIL/ExifTags.py ADDED
@@ -0,0 +1,381 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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 = 0
307
+ GPSLatitudeRef = 1
308
+ GPSLatitude = 2
309
+ GPSLongitudeRef = 3
310
+ GPSLongitude = 4
311
+ GPSAltitudeRef = 5
312
+ GPSAltitude = 6
313
+ GPSTimeStamp = 7
314
+ GPSSatellites = 8
315
+ GPSStatus = 9
316
+ GPSMeasureMode = 10
317
+ GPSDOP = 11
318
+ GPSSpeedRef = 12
319
+ GPSSpeed = 13
320
+ GPSTrackRef = 14
321
+ GPSTrack = 15
322
+ GPSImgDirectionRef = 16
323
+ GPSImgDirection = 17
324
+ GPSMapDatum = 18
325
+ GPSDestLatitudeRef = 19
326
+ GPSDestLatitude = 20
327
+ GPSDestLongitudeRef = 21
328
+ GPSDestLongitude = 22
329
+ GPSDestBearingRef = 23
330
+ GPSDestBearing = 24
331
+ GPSDestDistanceRef = 25
332
+ GPSDestDistance = 26
333
+ GPSProcessingMethod = 27
334
+ GPSAreaInformation = 28
335
+ GPSDateStamp = 29
336
+ GPSDifferential = 30
337
+ GPSHPositioningError = 31
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 = 1
346
+ InteropVersion = 2
347
+ RelatedImageFileFormat = 4096
348
+ RelatedImageWidth = 4097
349
+ RelatedImageHeight = 4098
350
+
351
+
352
+ class IFD(IntEnum):
353
+ Exif = 34665
354
+ GPSInfo = 34853
355
+ Makernote = 37500
356
+ Interop = 40965
357
+ IFD1 = -1
358
+
359
+
360
+ class LightSource(IntEnum):
361
+ Unknown = 0
362
+ Daylight = 1
363
+ Fluorescent = 2
364
+ Tungsten = 3
365
+ Flash = 4
366
+ Fine = 9
367
+ Cloudy = 10
368
+ Shade = 11
369
+ DaylightFluorescent = 12
370
+ DayWhiteFluorescent = 13
371
+ CoolWhiteFluorescent = 14
372
+ WhiteFluorescent = 15
373
+ StandardLightA = 17
374
+ StandardLightB = 18
375
+ StandardLightC = 19
376
+ D55 = 20
377
+ D65 = 21
378
+ D75 = 22
379
+ D50 = 23
380
+ ISO = 24
381
+ Other = 255
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[:6] == 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,175 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+
26
+ #
27
+ # decoder
28
+
29
+
30
+ def _accept(prefix: bytes) -> bool:
31
+ return (
32
+ len(prefix) >= 6
33
+ and i16(prefix, 4) in [0xAF11, 0xAF12]
34
+ and i16(prefix, 14) in [0, 3] # flags
35
+ )
36
+
37
+
38
+ ##
39
+ # Image plugin for the FLI/FLC animation format. Use the <b>seek</b>
40
+ # method to load individual frames.
41
+
42
+
43
+ class FliImageFile(ImageFile.ImageFile):
44
+ format = "FLI"
45
+ format_description = "Autodesk FLI/FLC Animation"
46
+ _close_exclusive_fp_after_loading = False
47
+
48
+ def _open(self) -> None:
49
+ # HEAD
50
+ s = self.fp.read(128)
51
+ if not (_accept(s) and s[20:22] == b"\x00\x00"):
52
+ msg = "not an FLI/FLC file"
53
+ raise SyntaxError(msg)
54
+
55
+ # frames
56
+ self.n_frames = i16(s, 6)
57
+ self.is_animated = self.n_frames > 1
58
+
59
+ # image characteristics
60
+ self._mode = "P"
61
+ self._size = i16(s, 8), i16(s, 10)
62
+
63
+ # animation speed
64
+ duration = i32(s, 16)
65
+ magic = i16(s, 4)
66
+ if magic == 0xAF11:
67
+ duration = (duration * 1000) // 70
68
+ self.info["duration"] = duration
69
+
70
+ # look for palette
71
+ palette = [(a, a, a) for a in range(256)]
72
+
73
+ s = self.fp.read(16)
74
+
75
+ self.__offset = 128
76
+
77
+ if i16(s, 4) == 0xF100:
78
+ # prefix chunk; ignore it
79
+ self.__offset = self.__offset + i32(s)
80
+ self.fp.seek(self.__offset)
81
+ s = self.fp.read(16)
82
+
83
+ if i16(s, 4) == 0xF1FA:
84
+ # look for palette chunk
85
+ number_of_subchunks = i16(s, 6)
86
+ chunk_size: int | None = None
87
+ for _ in range(number_of_subchunks):
88
+ if chunk_size is not None:
89
+ self.fp.seek(chunk_size - 6, os.SEEK_CUR)
90
+ s = self.fp.read(6)
91
+ chunk_type = i16(s, 4)
92
+ if chunk_type in (4, 11):
93
+ self._palette(palette, 2 if chunk_type == 11 else 0)
94
+ break
95
+ chunk_size = i32(s)
96
+ if not chunk_size:
97
+ break
98
+
99
+ self.palette = ImagePalette.raw(
100
+ "RGB", b"".join(o8(r) + o8(g) + o8(b) for (r, g, b) in palette)
101
+ )
102
+
103
+ # set things up to decode first frame
104
+ self.__frame = -1
105
+ self._fp = self.fp
106
+ self.__rewind = self.fp.tell()
107
+ self.seek(0)
108
+
109
+ def _palette(self, palette: list[tuple[int, int, int]], shift: int) -> None:
110
+ # load palette
111
+
112
+ i = 0
113
+ for e in range(i16(self.fp.read(2))):
114
+ s = self.fp.read(2)
115
+ i = i + s[0]
116
+ n = s[1]
117
+ if n == 0:
118
+ n = 256
119
+ s = self.fp.read(n * 3)
120
+ for n in range(0, len(s), 3):
121
+ r = s[n] << shift
122
+ g = s[n + 1] << shift
123
+ b = s[n + 2] << shift
124
+ palette[i] = (r, g, b)
125
+ i += 1
126
+
127
+ def seek(self, frame: int) -> None:
128
+ if not self._seek_check(frame):
129
+ return
130
+ if frame < self.__frame:
131
+ self._seek(0)
132
+
133
+ for f in range(self.__frame + 1, frame + 1):
134
+ self._seek(f)
135
+
136
+ def _seek(self, frame: int) -> None:
137
+ if frame == 0:
138
+ self.__frame = -1
139
+ self._fp.seek(self.__rewind)
140
+ self.__offset = 128
141
+ else:
142
+ # ensure that the previous frame was loaded
143
+ self.load()
144
+
145
+ if frame != self.__frame + 1:
146
+ msg = f"cannot seek to frame {frame}"
147
+ raise ValueError(msg)
148
+ self.__frame = frame
149
+
150
+ # move to next frame
151
+ self.fp = self._fp
152
+ self.fp.seek(self.__offset)
153
+
154
+ s = self.fp.read(4)
155
+ if not s:
156
+ msg = "missing frame size"
157
+ raise EOFError(msg)
158
+
159
+ framesize = i32(s)
160
+
161
+ self.decodermaxblock = framesize
162
+ self.tile = [ImageFile._Tile("fli", (0, 0) + self.size, self.__offset, None)]
163
+
164
+ self.__offset += framesize
165
+
166
+ def tell(self) -> int:
167
+ return self.__frame
168
+
169
+
170
+ #
171
+ # registry
172
+
173
+ Image.register_open(FliImageFile.format, FliImageFile, _accept)
174
+
175
+ 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[:8] == 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,115 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ self._mode = "RGB"
83
+
84
+ # Only support single-format files.
85
+ # I don't know of any multi-format file.
86
+ assert format_count == 1
87
+
88
+ format, where = struct.unpack("<2i", self.fp.read(8))
89
+ self.fp.seek(where)
90
+ (mipmap_size,) = struct.unpack("<i", self.fp.read(4))
91
+
92
+ data = self.fp.read(mipmap_size)
93
+
94
+ if format == Format.DXT1:
95
+ self._mode = "RGBA"
96
+ self.tile = [ImageFile._Tile("bcn", (0, 0) + self.size, 0, (1,))]
97
+ elif format == Format.UNCOMPRESSED:
98
+ self.tile = [ImageFile._Tile("raw", (0, 0) + self.size, 0, ("RGB", 0, 1))]
99
+ else:
100
+ msg = f"Invalid texture compression format: {repr(format)}"
101
+ raise ValueError(msg)
102
+
103
+ self.fp.close()
104
+ self.fp = BytesIO(data)
105
+
106
+ def load_seek(self, pos: int) -> None:
107
+ pass
108
+
109
+
110
+ def _accept(prefix: bytes) -> bool:
111
+ return prefix[:4] == MAGIC
112
+
113
+
114
+ Image.register_open(FtexImageFile.format, FtexImageFile, _accept)
115
+ Image.register_extensions(FtexImageFile.format, [".ftc", ".ftu"])
venv/Lib/site-packages/PIL/GbrImagePlugin.py ADDED
@@ -0,0 +1,103 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ 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
+ self.info["comment"] = comment
84
+
85
+ # Image might not be small
86
+ Image._decompression_bomb_check(self.size)
87
+
88
+ # Data is an uncompressed block of w * h * bytes/pixel
89
+ self._data_size = width * height * color_depth
90
+
91
+ def load(self) -> Image.core.PixelAccess | None:
92
+ if self._im is None:
93
+ self.im = Image.core.new(self.mode, self.size)
94
+ self.frombytes(self.fp.read(self._data_size))
95
+ return Image.Image.load(self)
96
+
97
+
98
+ #
99
+ # registry
100
+
101
+
102
+ Image.register_open(GbrImageFile.format, GbrImageFile, _accept)
103
+ 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 = "L" # FIXME: "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
+ "XBGR", s[7 + true_color_offset + 4 : 7 + true_color_offset + 4 + 256 * 4]
72
+ )
73
+
74
+ self.tile = [
75
+ ImageFile._Tile(
76
+ "raw",
77
+ (0, 0) + self.size,
78
+ 7 + true_color_offset + 4 + 256 * 4,
79
+ ("L", 0, 1),
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,1197 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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 IO, TYPE_CHECKING, Any, Literal, NamedTuple, Union
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
+
49
+ if TYPE_CHECKING:
50
+ from . import _imaging
51
+ from ._typing import Buffer
52
+
53
+
54
+ class LoadingStrategy(IntEnum):
55
+ """.. versionadded:: 9.1.0"""
56
+
57
+ RGB_AFTER_FIRST = 0
58
+ RGB_AFTER_DIFFERENT_PALETTE_ONLY = 1
59
+ RGB_ALWAYS = 2
60
+
61
+
62
+ #: .. versionadded:: 9.1.0
63
+ LOADING_STRATEGY = LoadingStrategy.RGB_AFTER_FIRST
64
+
65
+ # --------------------------------------------------------------------
66
+ # Identify/read GIF files
67
+
68
+
69
+ def _accept(prefix: bytes) -> bool:
70
+ return prefix[:6] in [b"GIF87a", b"GIF89a"]
71
+
72
+
73
+ ##
74
+ # Image plugin for GIF images. This plugin supports both GIF87 and
75
+ # GIF89 images.
76
+
77
+
78
+ class GifImageFile(ImageFile.ImageFile):
79
+ format = "GIF"
80
+ format_description = "Compuserve GIF"
81
+ _close_exclusive_fp_after_loading = False
82
+
83
+ global_palette = None
84
+
85
+ def data(self) -> bytes | None:
86
+ s = self.fp.read(1)
87
+ if s and s[0]:
88
+ return self.fp.read(s[0])
89
+ return None
90
+
91
+ def _is_palette_needed(self, p: bytes) -> bool:
92
+ for i in range(0, len(p), 3):
93
+ if not (i // 3 == p[i] == p[i + 1] == p[i + 2]):
94
+ return True
95
+ return False
96
+
97
+ def _open(self) -> None:
98
+ # Screen
99
+ s = self.fp.read(13)
100
+ if not _accept(s):
101
+ msg = "not a GIF file"
102
+ raise SyntaxError(msg)
103
+
104
+ self.info["version"] = s[:6]
105
+ self._size = i16(s, 6), i16(s, 8)
106
+ self.tile = []
107
+ flags = s[10]
108
+ bits = (flags & 7) + 1
109
+
110
+ if flags & 128:
111
+ # get global palette
112
+ self.info["background"] = s[11]
113
+ # check if palette contains colour indices
114
+ p = self.fp.read(3 << bits)
115
+ if self._is_palette_needed(p):
116
+ p = ImagePalette.raw("RGB", p)
117
+ self.global_palette = self.palette = p
118
+
119
+ self._fp = self.fp # FIXME: hack
120
+ self.__rewind = self.fp.tell()
121
+ self._n_frames: int | None = None
122
+ self._seek(0) # get ready to read first frame
123
+
124
+ @property
125
+ def n_frames(self) -> int:
126
+ if self._n_frames is None:
127
+ current = self.tell()
128
+ try:
129
+ while True:
130
+ self._seek(self.tell() + 1, False)
131
+ except EOFError:
132
+ self._n_frames = self.tell() + 1
133
+ self.seek(current)
134
+ return self._n_frames
135
+
136
+ @cached_property
137
+ def is_animated(self) -> bool:
138
+ if self._n_frames is not None:
139
+ return self._n_frames != 1
140
+
141
+ current = self.tell()
142
+ if current:
143
+ return True
144
+
145
+ try:
146
+ self._seek(1, False)
147
+ is_animated = True
148
+ except EOFError:
149
+ is_animated = False
150
+
151
+ self.seek(current)
152
+ return is_animated
153
+
154
+ def seek(self, frame: int) -> None:
155
+ if not self._seek_check(frame):
156
+ return
157
+ if frame < self.__frame:
158
+ self._im = None
159
+ self._seek(0)
160
+
161
+ last_frame = self.__frame
162
+ for f in range(self.__frame + 1, frame + 1):
163
+ try:
164
+ self._seek(f)
165
+ except EOFError as e:
166
+ self.seek(last_frame)
167
+ msg = "no more images in GIF file"
168
+ raise EOFError(msg) from e
169
+
170
+ def _seek(self, frame: int, update_image: bool = True) -> None:
171
+ if frame == 0:
172
+ # rewind
173
+ self.__offset = 0
174
+ self.dispose: _imaging.ImagingCore | None = None
175
+ self.__frame = -1
176
+ self._fp.seek(self.__rewind)
177
+ self.disposal_method = 0
178
+ if "comment" in self.info:
179
+ del self.info["comment"]
180
+ else:
181
+ # ensure that the previous frame was loaded
182
+ if self.tile and update_image:
183
+ self.load()
184
+
185
+ if frame != self.__frame + 1:
186
+ msg = f"cannot seek to frame {frame}"
187
+ raise ValueError(msg)
188
+
189
+ self.fp = self._fp
190
+ if self.__offset:
191
+ # backup to last frame
192
+ self.fp.seek(self.__offset)
193
+ while self.data():
194
+ pass
195
+ self.__offset = 0
196
+
197
+ s = self.fp.read(1)
198
+ if not s or s == b";":
199
+ msg = "no more images in GIF file"
200
+ raise EOFError(msg)
201
+
202
+ palette: ImagePalette.ImagePalette | Literal[False] | None = None
203
+
204
+ info: dict[str, Any] = {}
205
+ frame_transparency = None
206
+ interlace = None
207
+ frame_dispose_extent = None
208
+ while True:
209
+ if not s:
210
+ s = self.fp.read(1)
211
+ if not s or s == b";":
212
+ break
213
+
214
+ elif s == b"!":
215
+ #
216
+ # extensions
217
+ #
218
+ s = self.fp.read(1)
219
+ block = self.data()
220
+ if s[0] == 249 and block is not None:
221
+ #
222
+ # graphic control extension
223
+ #
224
+ flags = block[0]
225
+ if flags & 1:
226
+ frame_transparency = block[3]
227
+ info["duration"] = i16(block, 1) * 10
228
+
229
+ # disposal method - find the value of bits 4 - 6
230
+ dispose_bits = 0b00011100 & flags
231
+ dispose_bits = dispose_bits >> 2
232
+ if dispose_bits:
233
+ # only set the dispose if it is not
234
+ # unspecified. I'm not sure if this is
235
+ # correct, but it seems to prevent the last
236
+ # frame from looking odd for some animations
237
+ self.disposal_method = dispose_bits
238
+ elif s[0] == 254:
239
+ #
240
+ # comment extension
241
+ #
242
+ comment = b""
243
+
244
+ # Read this comment block
245
+ while block:
246
+ comment += block
247
+ block = self.data()
248
+
249
+ if "comment" in info:
250
+ # If multiple comment blocks in frame, separate with \n
251
+ info["comment"] += b"\n" + comment
252
+ else:
253
+ info["comment"] = comment
254
+ s = None
255
+ continue
256
+ elif s[0] == 255 and frame == 0 and block is not None:
257
+ #
258
+ # application extension
259
+ #
260
+ info["extension"] = block, self.fp.tell()
261
+ if block[:11] == b"NETSCAPE2.0":
262
+ block = self.data()
263
+ if block and len(block) >= 3 and block[0] == 1:
264
+ self.info["loop"] = i16(block, 1)
265
+ while self.data():
266
+ pass
267
+
268
+ elif s == b",":
269
+ #
270
+ # local image
271
+ #
272
+ s = self.fp.read(9)
273
+
274
+ # extent
275
+ x0, y0 = i16(s, 0), i16(s, 2)
276
+ x1, y1 = x0 + i16(s, 4), y0 + i16(s, 6)
277
+ if (x1 > self.size[0] or y1 > self.size[1]) and update_image:
278
+ self._size = max(x1, self.size[0]), max(y1, self.size[1])
279
+ Image._decompression_bomb_check(self._size)
280
+ frame_dispose_extent = x0, y0, x1, y1
281
+ flags = s[8]
282
+
283
+ interlace = (flags & 64) != 0
284
+
285
+ if flags & 128:
286
+ bits = (flags & 7) + 1
287
+ p = self.fp.read(3 << bits)
288
+ if self._is_palette_needed(p):
289
+ palette = ImagePalette.raw("RGB", p)
290
+ else:
291
+ palette = False
292
+
293
+ # image data
294
+ bits = self.fp.read(1)[0]
295
+ self.__offset = self.fp.tell()
296
+ break
297
+ s = None
298
+
299
+ if interlace is None:
300
+ msg = "image not found in GIF frame"
301
+ raise EOFError(msg)
302
+
303
+ self.__frame = frame
304
+ if not update_image:
305
+ return
306
+
307
+ self.tile = []
308
+
309
+ if self.dispose:
310
+ self.im.paste(self.dispose, self.dispose_extent)
311
+
312
+ self._frame_palette = palette if palette is not None else self.global_palette
313
+ self._frame_transparency = frame_transparency
314
+ if frame == 0:
315
+ if self._frame_palette:
316
+ if LOADING_STRATEGY == LoadingStrategy.RGB_ALWAYS:
317
+ self._mode = "RGBA" if frame_transparency is not None else "RGB"
318
+ else:
319
+ self._mode = "P"
320
+ else:
321
+ self._mode = "L"
322
+
323
+ if palette:
324
+ self.palette = palette
325
+ elif self.global_palette:
326
+ from copy import copy
327
+
328
+ self.palette = copy(self.global_palette)
329
+ else:
330
+ self.palette = None
331
+ else:
332
+ if self.mode == "P":
333
+ if (
334
+ LOADING_STRATEGY != LoadingStrategy.RGB_AFTER_DIFFERENT_PALETTE_ONLY
335
+ or palette
336
+ ):
337
+ if "transparency" in self.info:
338
+ self.im.putpalettealpha(self.info["transparency"], 0)
339
+ self.im = self.im.convert("RGBA", Image.Dither.FLOYDSTEINBERG)
340
+ self._mode = "RGBA"
341
+ del self.info["transparency"]
342
+ else:
343
+ self._mode = "RGB"
344
+ self.im = self.im.convert("RGB", Image.Dither.FLOYDSTEINBERG)
345
+
346
+ def _rgb(color: int) -> tuple[int, int, int]:
347
+ if self._frame_palette:
348
+ if color * 3 + 3 > len(self._frame_palette.palette):
349
+ color = 0
350
+ return tuple(self._frame_palette.palette[color * 3 : color * 3 + 3])
351
+ else:
352
+ return (color, color, color)
353
+
354
+ self.dispose = None
355
+ self.dispose_extent = frame_dispose_extent
356
+ if self.dispose_extent and self.disposal_method >= 2:
357
+ try:
358
+ if self.disposal_method == 2:
359
+ # replace with background colour
360
+
361
+ # only dispose the extent in this frame
362
+ x0, y0, x1, y1 = self.dispose_extent
363
+ dispose_size = (x1 - x0, y1 - y0)
364
+
365
+ Image._decompression_bomb_check(dispose_size)
366
+
367
+ # by convention, attempt to use transparency first
368
+ dispose_mode = "P"
369
+ color = self.info.get("transparency", frame_transparency)
370
+ if color is not None:
371
+ if self.mode in ("RGB", "RGBA"):
372
+ dispose_mode = "RGBA"
373
+ color = _rgb(color) + (0,)
374
+ else:
375
+ color = self.info.get("background", 0)
376
+ if self.mode in ("RGB", "RGBA"):
377
+ dispose_mode = "RGB"
378
+ color = _rgb(color)
379
+ self.dispose = Image.core.fill(dispose_mode, dispose_size, color)
380
+ else:
381
+ # replace with previous contents
382
+ if self._im is not None:
383
+ # only dispose the extent in this frame
384
+ self.dispose = self._crop(self.im, self.dispose_extent)
385
+ elif frame_transparency is not None:
386
+ x0, y0, x1, y1 = self.dispose_extent
387
+ dispose_size = (x1 - x0, y1 - y0)
388
+
389
+ Image._decompression_bomb_check(dispose_size)
390
+ dispose_mode = "P"
391
+ color = frame_transparency
392
+ if self.mode in ("RGB", "RGBA"):
393
+ dispose_mode = "RGBA"
394
+ color = _rgb(frame_transparency) + (0,)
395
+ self.dispose = Image.core.fill(
396
+ dispose_mode, dispose_size, color
397
+ )
398
+ except AttributeError:
399
+ pass
400
+
401
+ if interlace is not None:
402
+ transparency = -1
403
+ if frame_transparency is not None:
404
+ if frame == 0:
405
+ if LOADING_STRATEGY != LoadingStrategy.RGB_ALWAYS:
406
+ self.info["transparency"] = frame_transparency
407
+ elif self.mode not in ("RGB", "RGBA"):
408
+ transparency = frame_transparency
409
+ self.tile = [
410
+ ImageFile._Tile(
411
+ "gif",
412
+ (x0, y0, x1, y1),
413
+ self.__offset,
414
+ (bits, interlace, transparency),
415
+ )
416
+ ]
417
+
418
+ if info.get("comment"):
419
+ self.info["comment"] = info["comment"]
420
+ for k in ["duration", "extension"]:
421
+ if k in info:
422
+ self.info[k] = info[k]
423
+ elif k in self.info:
424
+ del self.info[k]
425
+
426
+ def load_prepare(self) -> None:
427
+ temp_mode = "P" if self._frame_palette else "L"
428
+ self._prev_im = None
429
+ if self.__frame == 0:
430
+ if self._frame_transparency is not None:
431
+ self.im = Image.core.fill(
432
+ temp_mode, self.size, self._frame_transparency
433
+ )
434
+ elif self.mode in ("RGB", "RGBA"):
435
+ self._prev_im = self.im
436
+ if self._frame_palette:
437
+ self.im = Image.core.fill("P", self.size, self._frame_transparency or 0)
438
+ self.im.putpalette("RGB", *self._frame_palette.getdata())
439
+ else:
440
+ self._im = None
441
+ if not self._prev_im and self._im is not None and self.size != self.im.size:
442
+ expanded_im = Image.core.fill(self.im.mode, self.size)
443
+ if self._frame_palette:
444
+ expanded_im.putpalette("RGB", *self._frame_palette.getdata())
445
+ expanded_im.paste(self.im, (0, 0) + self.im.size)
446
+
447
+ self.im = expanded_im
448
+ self._mode = temp_mode
449
+ self._frame_palette = None
450
+
451
+ super().load_prepare()
452
+
453
+ def load_end(self) -> None:
454
+ if self.__frame == 0:
455
+ if self.mode == "P" and LOADING_STRATEGY == LoadingStrategy.RGB_ALWAYS:
456
+ if self._frame_transparency is not None:
457
+ self.im.putpalettealpha(self._frame_transparency, 0)
458
+ self._mode = "RGBA"
459
+ else:
460
+ self._mode = "RGB"
461
+ self.im = self.im.convert(self.mode, Image.Dither.FLOYDSTEINBERG)
462
+ return
463
+ if not self._prev_im:
464
+ return
465
+ if self.size != self._prev_im.size:
466
+ if self._frame_transparency is not None:
467
+ expanded_im = Image.core.fill("RGBA", self.size)
468
+ else:
469
+ expanded_im = Image.core.fill("P", self.size)
470
+ expanded_im.putpalette("RGB", "RGB", self.im.getpalette())
471
+ expanded_im = expanded_im.convert("RGB")
472
+ expanded_im.paste(self._prev_im, (0, 0) + self._prev_im.size)
473
+
474
+ self._prev_im = expanded_im
475
+ assert self._prev_im is not None
476
+ if self._frame_transparency is not None:
477
+ self.im.putpalettealpha(self._frame_transparency, 0)
478
+ frame_im = self.im.convert("RGBA")
479
+ else:
480
+ frame_im = self.im.convert("RGB")
481
+
482
+ assert self.dispose_extent is not None
483
+ frame_im = self._crop(frame_im, self.dispose_extent)
484
+
485
+ self.im = self._prev_im
486
+ self._mode = self.im.mode
487
+ if frame_im.mode == "RGBA":
488
+ self.im.paste(frame_im, self.dispose_extent, frame_im)
489
+ else:
490
+ self.im.paste(frame_im, self.dispose_extent)
491
+
492
+ def tell(self) -> int:
493
+ return self.__frame
494
+
495
+
496
+ # --------------------------------------------------------------------
497
+ # Write GIF files
498
+
499
+
500
+ RAWMODE = {"1": "L", "L": "L", "P": "P"}
501
+
502
+
503
+ def _normalize_mode(im: Image.Image) -> Image.Image:
504
+ """
505
+ Takes an image (or frame), returns an image in a mode that is appropriate
506
+ for saving in a Gif.
507
+
508
+ It may return the original image, or it may return an image converted to
509
+ palette or 'L' mode.
510
+
511
+ :param im: Image object
512
+ :returns: Image object
513
+ """
514
+ if im.mode in RAWMODE:
515
+ im.load()
516
+ return im
517
+ if Image.getmodebase(im.mode) == "RGB":
518
+ im = im.convert("P", palette=Image.Palette.ADAPTIVE)
519
+ assert im.palette is not None
520
+ if im.palette.mode == "RGBA":
521
+ for rgba in im.palette.colors:
522
+ if rgba[3] == 0:
523
+ im.info["transparency"] = im.palette.colors[rgba]
524
+ break
525
+ return im
526
+ return im.convert("L")
527
+
528
+
529
+ _Palette = Union[bytes, bytearray, list[int], ImagePalette.ImagePalette]
530
+
531
+
532
+ def _normalize_palette(
533
+ im: Image.Image, palette: _Palette | None, info: dict[str, Any]
534
+ ) -> Image.Image:
535
+ """
536
+ Normalizes the palette for image.
537
+ - Sets the palette to the incoming palette, if provided.
538
+ - Ensures that there's a palette for L mode images
539
+ - Optimizes the palette if necessary/desired.
540
+
541
+ :param im: Image object
542
+ :param palette: bytes object containing the source palette, or ....
543
+ :param info: encoderinfo
544
+ :returns: Image object
545
+ """
546
+ source_palette = None
547
+ if palette:
548
+ # a bytes palette
549
+ if isinstance(palette, (bytes, bytearray, list)):
550
+ source_palette = bytearray(palette[:768])
551
+ if isinstance(palette, ImagePalette.ImagePalette):
552
+ source_palette = bytearray(palette.palette)
553
+
554
+ if im.mode == "P":
555
+ if not source_palette:
556
+ im_palette = im.getpalette(None)
557
+ assert im_palette is not None
558
+ source_palette = bytearray(im_palette)
559
+ else: # L-mode
560
+ if not source_palette:
561
+ source_palette = bytearray(i // 3 for i in range(768))
562
+ im.palette = ImagePalette.ImagePalette("RGB", palette=source_palette)
563
+ assert source_palette is not None
564
+
565
+ if palette:
566
+ used_palette_colors: list[int | None] = []
567
+ assert im.palette is not None
568
+ for i in range(0, len(source_palette), 3):
569
+ source_color = tuple(source_palette[i : i + 3])
570
+ index = im.palette.colors.get(source_color)
571
+ if index in used_palette_colors:
572
+ index = None
573
+ used_palette_colors.append(index)
574
+ for i, index in enumerate(used_palette_colors):
575
+ if index is None:
576
+ for j in range(len(used_palette_colors)):
577
+ if j not in used_palette_colors:
578
+ used_palette_colors[i] = j
579
+ break
580
+ dest_map: list[int] = []
581
+ for index in used_palette_colors:
582
+ assert index is not None
583
+ dest_map.append(index)
584
+ im = im.remap_palette(dest_map)
585
+ else:
586
+ optimized_palette_colors = _get_optimize(im, info)
587
+ if optimized_palette_colors is not None:
588
+ im = im.remap_palette(optimized_palette_colors, source_palette)
589
+ if "transparency" in info:
590
+ try:
591
+ info["transparency"] = optimized_palette_colors.index(
592
+ info["transparency"]
593
+ )
594
+ except ValueError:
595
+ del info["transparency"]
596
+ return im
597
+
598
+ assert im.palette is not None
599
+ im.palette.palette = source_palette
600
+ return im
601
+
602
+
603
+ def _write_single_frame(
604
+ im: Image.Image,
605
+ fp: IO[bytes],
606
+ palette: _Palette | None,
607
+ ) -> None:
608
+ im_out = _normalize_mode(im)
609
+ for k, v in im_out.info.items():
610
+ if isinstance(k, str):
611
+ im.encoderinfo.setdefault(k, v)
612
+ im_out = _normalize_palette(im_out, palette, im.encoderinfo)
613
+
614
+ for s in _get_global_header(im_out, im.encoderinfo):
615
+ fp.write(s)
616
+
617
+ # local image header
618
+ flags = 0
619
+ if get_interlace(im):
620
+ flags = flags | 64
621
+ _write_local_header(fp, im, (0, 0), flags)
622
+
623
+ im_out.encoderconfig = (8, get_interlace(im))
624
+ ImageFile._save(
625
+ im_out, fp, [ImageFile._Tile("gif", (0, 0) + im.size, 0, RAWMODE[im_out.mode])]
626
+ )
627
+
628
+ fp.write(b"\0") # end of image data
629
+
630
+
631
+ def _getbbox(
632
+ base_im: Image.Image, im_frame: Image.Image
633
+ ) -> tuple[Image.Image, tuple[int, int, int, int] | None]:
634
+ palette_bytes = [
635
+ bytes(im.palette.palette) if im.palette else b"" for im in (base_im, im_frame)
636
+ ]
637
+ if palette_bytes[0] != palette_bytes[1]:
638
+ im_frame = im_frame.convert("RGBA")
639
+ base_im = base_im.convert("RGBA")
640
+ delta = ImageChops.subtract_modulo(im_frame, base_im)
641
+ return delta, delta.getbbox(alpha_only=False)
642
+
643
+
644
+ class _Frame(NamedTuple):
645
+ im: Image.Image
646
+ bbox: tuple[int, int, int, int] | None
647
+ encoderinfo: dict[str, Any]
648
+
649
+
650
+ def _write_multiple_frames(
651
+ im: Image.Image, fp: IO[bytes], palette: _Palette | None
652
+ ) -> bool:
653
+ duration = im.encoderinfo.get("duration")
654
+ disposal = im.encoderinfo.get("disposal", im.info.get("disposal"))
655
+
656
+ im_frames: list[_Frame] = []
657
+ previous_im: Image.Image | None = None
658
+ frame_count = 0
659
+ background_im = None
660
+ for imSequence in itertools.chain([im], im.encoderinfo.get("append_images", [])):
661
+ for im_frame in ImageSequence.Iterator(imSequence):
662
+ # a copy is required here since seek can still mutate the image
663
+ im_frame = _normalize_mode(im_frame.copy())
664
+ if frame_count == 0:
665
+ for k, v in im_frame.info.items():
666
+ if k == "transparency":
667
+ continue
668
+ if isinstance(k, str):
669
+ im.encoderinfo.setdefault(k, v)
670
+
671
+ encoderinfo = im.encoderinfo.copy()
672
+ if "transparency" in im_frame.info:
673
+ encoderinfo.setdefault("transparency", im_frame.info["transparency"])
674
+ im_frame = _normalize_palette(im_frame, palette, encoderinfo)
675
+ if isinstance(duration, (list, tuple)):
676
+ encoderinfo["duration"] = duration[frame_count]
677
+ elif duration is None and "duration" in im_frame.info:
678
+ encoderinfo["duration"] = im_frame.info["duration"]
679
+ if isinstance(disposal, (list, tuple)):
680
+ encoderinfo["disposal"] = disposal[frame_count]
681
+ frame_count += 1
682
+
683
+ diff_frame = None
684
+ if im_frames and previous_im:
685
+ # delta frame
686
+ delta, bbox = _getbbox(previous_im, im_frame)
687
+ if not bbox:
688
+ # This frame is identical to the previous frame
689
+ if encoderinfo.get("duration"):
690
+ im_frames[-1].encoderinfo["duration"] += encoderinfo["duration"]
691
+ continue
692
+ if im_frames[-1].encoderinfo.get("disposal") == 2:
693
+ if background_im is None:
694
+ color = im.encoderinfo.get(
695
+ "transparency", im.info.get("transparency", (0, 0, 0))
696
+ )
697
+ background = _get_background(im_frame, color)
698
+ background_im = Image.new("P", im_frame.size, background)
699
+ assert im_frames[0].im.palette is not None
700
+ background_im.putpalette(im_frames[0].im.palette)
701
+ bbox = _getbbox(background_im, im_frame)[1]
702
+ elif encoderinfo.get("optimize") and im_frame.mode != "1":
703
+ if "transparency" not in encoderinfo:
704
+ assert im_frame.palette is not None
705
+ try:
706
+ encoderinfo["transparency"] = (
707
+ im_frame.palette._new_color_index(im_frame)
708
+ )
709
+ except ValueError:
710
+ pass
711
+ if "transparency" in encoderinfo:
712
+ # When the delta is zero, fill the image with transparency
713
+ diff_frame = im_frame.copy()
714
+ fill = Image.new("P", delta.size, encoderinfo["transparency"])
715
+ if delta.mode == "RGBA":
716
+ r, g, b, a = delta.split()
717
+ mask = ImageMath.lambda_eval(
718
+ lambda args: args["convert"](
719
+ args["max"](
720
+ args["max"](
721
+ args["max"](args["r"], args["g"]), args["b"]
722
+ ),
723
+ args["a"],
724
+ )
725
+ * 255,
726
+ "1",
727
+ ),
728
+ r=r,
729
+ g=g,
730
+ b=b,
731
+ a=a,
732
+ )
733
+ else:
734
+ if delta.mode == "P":
735
+ # Convert to L without considering palette
736
+ delta_l = Image.new("L", delta.size)
737
+ delta_l.putdata(delta.getdata())
738
+ delta = delta_l
739
+ mask = ImageMath.lambda_eval(
740
+ lambda args: args["convert"](args["im"] * 255, "1"),
741
+ im=delta,
742
+ )
743
+ diff_frame.paste(fill, mask=ImageOps.invert(mask))
744
+ else:
745
+ bbox = None
746
+ previous_im = im_frame
747
+ im_frames.append(_Frame(diff_frame or im_frame, bbox, encoderinfo))
748
+
749
+ if len(im_frames) == 1:
750
+ if "duration" in im.encoderinfo:
751
+ # Since multiple frames will not be written, use the combined duration
752
+ im.encoderinfo["duration"] = im_frames[0].encoderinfo["duration"]
753
+ return False
754
+
755
+ for frame_data in im_frames:
756
+ im_frame = frame_data.im
757
+ if not frame_data.bbox:
758
+ # global header
759
+ for s in _get_global_header(im_frame, frame_data.encoderinfo):
760
+ fp.write(s)
761
+ offset = (0, 0)
762
+ else:
763
+ # compress difference
764
+ if not palette:
765
+ frame_data.encoderinfo["include_color_table"] = True
766
+
767
+ im_frame = im_frame.crop(frame_data.bbox)
768
+ offset = frame_data.bbox[:2]
769
+ _write_frame_data(fp, im_frame, offset, frame_data.encoderinfo)
770
+ return True
771
+
772
+
773
+ def _save_all(im: Image.Image, fp: IO[bytes], filename: str | bytes) -> None:
774
+ _save(im, fp, filename, save_all=True)
775
+
776
+
777
+ def _save(
778
+ im: Image.Image, fp: IO[bytes], filename: str | bytes, save_all: bool = False
779
+ ) -> None:
780
+ # header
781
+ if "palette" in im.encoderinfo or "palette" in im.info:
782
+ palette = im.encoderinfo.get("palette", im.info.get("palette"))
783
+ else:
784
+ palette = None
785
+ im.encoderinfo.setdefault("optimize", True)
786
+
787
+ if not save_all or not _write_multiple_frames(im, fp, palette):
788
+ _write_single_frame(im, fp, palette)
789
+
790
+ fp.write(b";") # end of file
791
+
792
+ if hasattr(fp, "flush"):
793
+ fp.flush()
794
+
795
+
796
+ def get_interlace(im: Image.Image) -> int:
797
+ interlace = im.encoderinfo.get("interlace", 1)
798
+
799
+ # workaround for @PIL153
800
+ if min(im.size) < 16:
801
+ interlace = 0
802
+
803
+ return interlace
804
+
805
+
806
+ def _write_local_header(
807
+ fp: IO[bytes], im: Image.Image, offset: tuple[int, int], flags: int
808
+ ) -> None:
809
+ try:
810
+ transparency = im.encoderinfo["transparency"]
811
+ except KeyError:
812
+ transparency = None
813
+
814
+ if "duration" in im.encoderinfo:
815
+ duration = int(im.encoderinfo["duration"] / 10)
816
+ else:
817
+ duration = 0
818
+
819
+ disposal = int(im.encoderinfo.get("disposal", 0))
820
+
821
+ if transparency is not None or duration != 0 or disposal:
822
+ packed_flag = 1 if transparency is not None else 0
823
+ packed_flag |= disposal << 2
824
+
825
+ fp.write(
826
+ b"!"
827
+ + o8(249) # extension intro
828
+ + o8(4) # length
829
+ + o8(packed_flag) # packed fields
830
+ + o16(duration) # duration
831
+ + o8(transparency or 0) # transparency index
832
+ + o8(0)
833
+ )
834
+
835
+ include_color_table = im.encoderinfo.get("include_color_table")
836
+ if include_color_table:
837
+ palette_bytes = _get_palette_bytes(im)
838
+ color_table_size = _get_color_table_size(palette_bytes)
839
+ if color_table_size:
840
+ flags = flags | 128 # local color table flag
841
+ flags = flags | color_table_size
842
+
843
+ fp.write(
844
+ b","
845
+ + o16(offset[0]) # offset
846
+ + o16(offset[1])
847
+ + o16(im.size[0]) # size
848
+ + o16(im.size[1])
849
+ + o8(flags) # flags
850
+ )
851
+ if include_color_table and color_table_size:
852
+ fp.write(_get_header_palette(palette_bytes))
853
+ fp.write(o8(8)) # bits
854
+
855
+
856
+ def _save_netpbm(im: Image.Image, fp: IO[bytes], filename: str | bytes) -> None:
857
+ # Unused by default.
858
+ # To use, uncomment the register_save call at the end of the file.
859
+ #
860
+ # If you need real GIF compression and/or RGB quantization, you
861
+ # can use the external NETPBM/PBMPLUS utilities. See comments
862
+ # below for information on how to enable this.
863
+ tempfile = im._dump()
864
+
865
+ try:
866
+ with open(filename, "wb") as f:
867
+ if im.mode != "RGB":
868
+ subprocess.check_call(
869
+ ["ppmtogif", tempfile], stdout=f, stderr=subprocess.DEVNULL
870
+ )
871
+ else:
872
+ # Pipe ppmquant output into ppmtogif
873
+ # "ppmquant 256 %s | ppmtogif > %s" % (tempfile, filename)
874
+ quant_cmd = ["ppmquant", "256", tempfile]
875
+ togif_cmd = ["ppmtogif"]
876
+ quant_proc = subprocess.Popen(
877
+ quant_cmd, stdout=subprocess.PIPE, stderr=subprocess.DEVNULL
878
+ )
879
+ togif_proc = subprocess.Popen(
880
+ togif_cmd,
881
+ stdin=quant_proc.stdout,
882
+ stdout=f,
883
+ stderr=subprocess.DEVNULL,
884
+ )
885
+
886
+ # Allow ppmquant to receive SIGPIPE if ppmtogif exits
887
+ assert quant_proc.stdout is not None
888
+ quant_proc.stdout.close()
889
+
890
+ retcode = quant_proc.wait()
891
+ if retcode:
892
+ raise subprocess.CalledProcessError(retcode, quant_cmd)
893
+
894
+ retcode = togif_proc.wait()
895
+ if retcode:
896
+ raise subprocess.CalledProcessError(retcode, togif_cmd)
897
+ finally:
898
+ try:
899
+ os.unlink(tempfile)
900
+ except OSError:
901
+ pass
902
+
903
+
904
+ # Force optimization so that we can test performance against
905
+ # cases where it took lots of memory and time previously.
906
+ _FORCE_OPTIMIZE = False
907
+
908
+
909
+ def _get_optimize(im: Image.Image, info: dict[str, Any]) -> list[int] | None:
910
+ """
911
+ Palette optimization is a potentially expensive operation.
912
+
913
+ This function determines if the palette should be optimized using
914
+ some heuristics, then returns the list of palette entries in use.
915
+
916
+ :param im: Image object
917
+ :param info: encoderinfo
918
+ :returns: list of indexes of palette entries in use, or None
919
+ """
920
+ if im.mode in ("P", "L") and info and info.get("optimize"):
921
+ # Potentially expensive operation.
922
+
923
+ # The palette saves 3 bytes per color not used, but palette
924
+ # lengths are restricted to 3*(2**N) bytes. Max saving would
925
+ # be 768 -> 6 bytes if we went all the way down to 2 colors.
926
+ # * If we're over 128 colors, we can't save any space.
927
+ # * If there aren't any holes, it's not worth collapsing.
928
+ # * If we have a 'large' image, the palette is in the noise.
929
+
930
+ # create the new palette if not every color is used
931
+ optimise = _FORCE_OPTIMIZE or im.mode == "L"
932
+ if optimise or im.width * im.height < 512 * 512:
933
+ # check which colors are used
934
+ used_palette_colors = []
935
+ for i, count in enumerate(im.histogram()):
936
+ if count:
937
+ used_palette_colors.append(i)
938
+
939
+ if optimise or max(used_palette_colors) >= len(used_palette_colors):
940
+ return used_palette_colors
941
+
942
+ assert im.palette is not None
943
+ num_palette_colors = len(im.palette.palette) // Image.getmodebands(
944
+ im.palette.mode
945
+ )
946
+ current_palette_size = 1 << (num_palette_colors - 1).bit_length()
947
+ if (
948
+ # check that the palette would become smaller when saved
949
+ len(used_palette_colors) <= current_palette_size // 2
950
+ # check that the palette is not already the smallest possible size
951
+ and current_palette_size > 2
952
+ ):
953
+ return used_palette_colors
954
+ return None
955
+
956
+
957
+ def _get_color_table_size(palette_bytes: bytes) -> int:
958
+ # calculate the palette size for the header
959
+ if not palette_bytes:
960
+ return 0
961
+ elif len(palette_bytes) < 9:
962
+ return 1
963
+ else:
964
+ return math.ceil(math.log(len(palette_bytes) // 3, 2)) - 1
965
+
966
+
967
+ def _get_header_palette(palette_bytes: bytes) -> bytes:
968
+ """
969
+ Returns the palette, null padded to the next power of 2 (*3) bytes
970
+ suitable for direct inclusion in the GIF header
971
+
972
+ :param palette_bytes: Unpadded palette bytes, in RGBRGB form
973
+ :returns: Null padded palette
974
+ """
975
+ color_table_size = _get_color_table_size(palette_bytes)
976
+
977
+ # add the missing amount of bytes
978
+ # the palette has to be 2<<n in size
979
+ actual_target_size_diff = (2 << color_table_size) - len(palette_bytes) // 3
980
+ if actual_target_size_diff > 0:
981
+ palette_bytes += o8(0) * 3 * actual_target_size_diff
982
+ return palette_bytes
983
+
984
+
985
+ def _get_palette_bytes(im: Image.Image) -> bytes:
986
+ """
987
+ Gets the palette for inclusion in the gif header
988
+
989
+ :param im: Image object
990
+ :returns: Bytes, len<=768 suitable for inclusion in gif header
991
+ """
992
+ if not im.palette:
993
+ return b""
994
+
995
+ palette = bytes(im.palette.palette)
996
+ if im.palette.mode == "RGBA":
997
+ palette = b"".join(palette[i * 4 : i * 4 + 3] for i in range(len(palette) // 3))
998
+ return palette
999
+
1000
+
1001
+ def _get_background(
1002
+ im: Image.Image,
1003
+ info_background: int | tuple[int, int, int] | tuple[int, int, int, int] | None,
1004
+ ) -> int:
1005
+ background = 0
1006
+ if info_background:
1007
+ if isinstance(info_background, tuple):
1008
+ # WebPImagePlugin stores an RGBA value in info["background"]
1009
+ # So it must be converted to the same format as GifImagePlugin's
1010
+ # info["background"] - a global color table index
1011
+ assert im.palette is not None
1012
+ try:
1013
+ background = im.palette.getcolor(info_background, im)
1014
+ except ValueError as e:
1015
+ if str(e) not in (
1016
+ # If all 256 colors are in use,
1017
+ # then there is no need for the background color
1018
+ "cannot allocate more than 256 colors",
1019
+ # Ignore non-opaque WebP background
1020
+ "cannot add non-opaque RGBA color to RGB palette",
1021
+ ):
1022
+ raise
1023
+ else:
1024
+ background = info_background
1025
+ return background
1026
+
1027
+
1028
+ def _get_global_header(im: Image.Image, info: dict[str, Any]) -> list[bytes]:
1029
+ """Return a list of strings representing a GIF header"""
1030
+
1031
+ # Header Block
1032
+ # https://www.matthewflickinger.com/lab/whatsinagif/bits_and_bytes.asp
1033
+
1034
+ version = b"87a"
1035
+ if im.info.get("version") == b"89a" or (
1036
+ info
1037
+ and (
1038
+ "transparency" in info
1039
+ or info.get("loop") is not None
1040
+ or info.get("duration")
1041
+ or info.get("comment")
1042
+ )
1043
+ ):
1044
+ version = b"89a"
1045
+
1046
+ background = _get_background(im, info.get("background"))
1047
+
1048
+ palette_bytes = _get_palette_bytes(im)
1049
+ color_table_size = _get_color_table_size(palette_bytes)
1050
+
1051
+ header = [
1052
+ b"GIF" # signature
1053
+ + version # version
1054
+ + o16(im.size[0]) # canvas width
1055
+ + o16(im.size[1]), # canvas height
1056
+ # Logical Screen Descriptor
1057
+ # size of global color table + global color table flag
1058
+ o8(color_table_size + 128), # packed fields
1059
+ # background + reserved/aspect
1060
+ o8(background) + o8(0),
1061
+ # Global Color Table
1062
+ _get_header_palette(palette_bytes),
1063
+ ]
1064
+ if info.get("loop") is not None:
1065
+ header.append(
1066
+ b"!"
1067
+ + o8(255) # extension intro
1068
+ + o8(11)
1069
+ + b"NETSCAPE2.0"
1070
+ + o8(3)
1071
+ + o8(1)
1072
+ + o16(info["loop"]) # number of loops
1073
+ + o8(0)
1074
+ )
1075
+ if info.get("comment"):
1076
+ comment_block = b"!" + o8(254) # extension intro
1077
+
1078
+ comment = info["comment"]
1079
+ if isinstance(comment, str):
1080
+ comment = comment.encode()
1081
+ for i in range(0, len(comment), 255):
1082
+ subblock = comment[i : i + 255]
1083
+ comment_block += o8(len(subblock)) + subblock
1084
+
1085
+ comment_block += o8(0)
1086
+ header.append(comment_block)
1087
+ return header
1088
+
1089
+
1090
+ def _write_frame_data(
1091
+ fp: IO[bytes],
1092
+ im_frame: Image.Image,
1093
+ offset: tuple[int, int],
1094
+ params: dict[str, Any],
1095
+ ) -> None:
1096
+ try:
1097
+ im_frame.encoderinfo = params
1098
+
1099
+ # local image header
1100
+ _write_local_header(fp, im_frame, offset, 0)
1101
+
1102
+ ImageFile._save(
1103
+ im_frame,
1104
+ fp,
1105
+ [ImageFile._Tile("gif", (0, 0) + im_frame.size, 0, RAWMODE[im_frame.mode])],
1106
+ )
1107
+
1108
+ fp.write(b"\0") # end of image data
1109
+ finally:
1110
+ del im_frame.encoderinfo
1111
+
1112
+
1113
+ # --------------------------------------------------------------------
1114
+ # Legacy GIF utilities
1115
+
1116
+
1117
+ def getheader(
1118
+ im: Image.Image, palette: _Palette | None = None, info: dict[str, Any] | None = None
1119
+ ) -> tuple[list[bytes], list[int] | None]:
1120
+ """
1121
+ Legacy Method to get Gif data from image.
1122
+
1123
+ Warning:: May modify image data.
1124
+
1125
+ :param im: Image object
1126
+ :param palette: bytes object containing the source palette, or ....
1127
+ :param info: encoderinfo
1128
+ :returns: tuple of(list of header items, optimized palette)
1129
+
1130
+ """
1131
+ if info is None:
1132
+ info = {}
1133
+
1134
+ used_palette_colors = _get_optimize(im, info)
1135
+
1136
+ if "background" not in info and "background" in im.info:
1137
+ info["background"] = im.info["background"]
1138
+
1139
+ im_mod = _normalize_palette(im, palette, info)
1140
+ im.palette = im_mod.palette
1141
+ im.im = im_mod.im
1142
+ header = _get_global_header(im, info)
1143
+
1144
+ return header, used_palette_colors
1145
+
1146
+
1147
+ def getdata(
1148
+ im: Image.Image, offset: tuple[int, int] = (0, 0), **params: Any
1149
+ ) -> list[bytes]:
1150
+ """
1151
+ Legacy Method
1152
+
1153
+ Return a list of strings representing this image.
1154
+ The first string is a local image header, the rest contains
1155
+ encoded image data.
1156
+
1157
+ To specify duration, add the time in milliseconds,
1158
+ e.g. ``getdata(im_frame, duration=1000)``
1159
+
1160
+ :param im: Image object
1161
+ :param offset: Tuple of (x, y) pixels. Defaults to (0, 0)
1162
+ :param \\**params: e.g. duration or other encoder info parameters
1163
+ :returns: List of bytes containing GIF encoded frame data
1164
+
1165
+ """
1166
+ from io import BytesIO
1167
+
1168
+ class Collector(BytesIO):
1169
+ data = []
1170
+
1171
+ def write(self, data: Buffer) -> int:
1172
+ self.data.append(data)
1173
+ return len(data)
1174
+
1175
+ im.load() # make sure raster data is available
1176
+
1177
+ fp = Collector()
1178
+
1179
+ _write_frame_data(fp, im, offset, params)
1180
+
1181
+ return fp.data
1182
+
1183
+
1184
+ # --------------------------------------------------------------------
1185
+ # Registry
1186
+
1187
+ Image.register_open(GifImageFile.format, GifImageFile, _accept)
1188
+ Image.register_save(GifImageFile.format, _save)
1189
+ Image.register_save_all(GifImageFile.format, _save_all)
1190
+ Image.register_extension(GifImageFile.format, ".gif")
1191
+ Image.register_mime(GifImageFile.format, "image/gif")
1192
+
1193
+ #
1194
+ # Uncomment the following line if you wish to use NETPBM/PBMPLUS
1195
+ # instead of the built-in "uncompressed" GIF encoder
1196
+
1197
+ # Image.register_save(GifImageFile.format, _save_netpbm)
venv/Lib/site-packages/PIL/GimpGradientFile.py ADDED
@@ -0,0 +1,149 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ from typing import IO, Callable
25
+
26
+ from ._binary import o8
27
+
28
+ EPSILON = 1e-10
29
+ """""" # Enable auto-doc for data member
30
+
31
+
32
+ def linear(middle: float, pos: float) -> float:
33
+ if pos <= middle:
34
+ if middle < EPSILON:
35
+ return 0.0
36
+ else:
37
+ return 0.5 * pos / middle
38
+ else:
39
+ pos = pos - middle
40
+ middle = 1.0 - middle
41
+ if middle < EPSILON:
42
+ return 1.0
43
+ else:
44
+ return 0.5 + 0.5 * pos / middle
45
+
46
+
47
+ def curved(middle: float, pos: float) -> float:
48
+ return pos ** (log(0.5) / log(max(middle, EPSILON)))
49
+
50
+
51
+ def sine(middle: float, pos: float) -> float:
52
+ return (sin((-pi / 2.0) + pi * linear(middle, pos)) + 1.0) / 2.0
53
+
54
+
55
+ def sphere_increasing(middle: float, pos: float) -> float:
56
+ return sqrt(1.0 - (linear(middle, pos) - 1.0) ** 2)
57
+
58
+
59
+ def sphere_decreasing(middle: float, pos: float) -> float:
60
+ return 1.0 - sqrt(1.0 - linear(middle, pos) ** 2)
61
+
62
+
63
+ SEGMENTS = [linear, curved, sine, sphere_increasing, sphere_decreasing]
64
+ """""" # Enable auto-doc for data member
65
+
66
+
67
+ class GradientFile:
68
+ gradient: (
69
+ list[
70
+ tuple[
71
+ float,
72
+ float,
73
+ float,
74
+ list[float],
75
+ list[float],
76
+ Callable[[float, float], float],
77
+ ]
78
+ ]
79
+ | None
80
+ ) = None
81
+
82
+ def getpalette(self, entries: int = 256) -> tuple[bytes, str]:
83
+ assert self.gradient is not None
84
+ palette = []
85
+
86
+ ix = 0
87
+ x0, x1, xm, rgb0, rgb1, segment = self.gradient[ix]
88
+
89
+ for i in range(entries):
90
+ x = i / (entries - 1)
91
+
92
+ while x1 < x:
93
+ ix += 1
94
+ x0, x1, xm, rgb0, rgb1, segment = self.gradient[ix]
95
+
96
+ w = x1 - x0
97
+
98
+ if w < EPSILON:
99
+ scale = segment(0.5, 0.5)
100
+ else:
101
+ scale = segment((xm - x0) / w, (x - x0) / w)
102
+
103
+ # expand to RGBA
104
+ r = o8(int(255 * ((rgb1[0] - rgb0[0]) * scale + rgb0[0]) + 0.5))
105
+ g = o8(int(255 * ((rgb1[1] - rgb0[1]) * scale + rgb0[1]) + 0.5))
106
+ b = o8(int(255 * ((rgb1[2] - rgb0[2]) * scale + rgb0[2]) + 0.5))
107
+ a = o8(int(255 * ((rgb1[3] - rgb0[3]) * scale + rgb0[3]) + 0.5))
108
+
109
+ # add to palette
110
+ palette.append(r + g + b + a)
111
+
112
+ return b"".join(palette), "RGBA"
113
+
114
+
115
+ class GimpGradientFile(GradientFile):
116
+ """File handler for GIMP's gradient format."""
117
+
118
+ def __init__(self, fp: IO[bytes]) -> None:
119
+ if fp.readline()[:13] != b"GIMP Gradient":
120
+ msg = "not a GIMP gradient file"
121
+ raise SyntaxError(msg)
122
+
123
+ line = fp.readline()
124
+
125
+ # GIMP 1.2 gradient files don't contain a name, but GIMP 1.3 files do
126
+ if line.startswith(b"Name: "):
127
+ line = fp.readline().strip()
128
+
129
+ count = int(line)
130
+
131
+ self.gradient = []
132
+
133
+ for i in range(count):
134
+ s = fp.readline().split()
135
+ w = [float(x) for x in s[:11]]
136
+
137
+ x0, x1 = w[0], w[2]
138
+ xm = w[1]
139
+ rgb0 = w[3:7]
140
+ rgb1 = w[7:11]
141
+
142
+ segment = SEGMENTS[int(s[11])]
143
+ cspace = int(s[12])
144
+
145
+ if cspace != 0:
146
+ msg = "cannot handle HSV colour space"
147
+ raise OSError(msg)
148
+
149
+ self.gradient.append((x0, x1, xm, rgb0, rgb1, segment))
venv/Lib/site-packages/PIL/GimpPaletteFile.py ADDED
@@ -0,0 +1,58 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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 typing import IO
20
+
21
+ from ._binary import o8
22
+
23
+
24
+ class GimpPaletteFile:
25
+ """File handler for GIMP's palette format."""
26
+
27
+ rawmode = "RGB"
28
+
29
+ def __init__(self, fp: IO[bytes]) -> None:
30
+ palette = [o8(i) * 3 for i in range(256)]
31
+
32
+ if fp.readline()[:12] != b"GIMP Palette":
33
+ msg = "not a GIMP palette file"
34
+ raise SyntaxError(msg)
35
+
36
+ for i in range(256):
37
+ s = fp.readline()
38
+ if not s:
39
+ break
40
+
41
+ # skip fields and comment lines
42
+ if re.match(rb"\w+:|#", s):
43
+ continue
44
+ if len(s) > 100:
45
+ msg = "bad palette file"
46
+ raise SyntaxError(msg)
47
+
48
+ v = tuple(map(int, s.split()[:3]))
49
+ if len(v) != 3:
50
+ msg = "bad palette entry"
51
+ raise ValueError(msg)
52
+
53
+ palette[i] = o8(v[0]) + o8(v[1]) + o8(v[2])
54
+
55
+ self.palette = b"".join(palette)
56
+
57
+ def getpalette(self) -> tuple[bytes, str]:
58
+ return self.palette, self.rawmode
venv/Lib/site-packages/PIL/GribStubImagePlugin.py ADDED
@@ -0,0 +1,76 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ from typing import IO
14
+
15
+ from . import Image, ImageFile
16
+
17
+ _handler = None
18
+
19
+
20
+ def register_handler(handler: ImageFile.StubHandler | None) -> None:
21
+ """
22
+ Install application-specific GRIB image handler.
23
+
24
+ :param handler: Handler object.
25
+ """
26
+ global _handler
27
+ _handler = handler
28
+
29
+
30
+ # --------------------------------------------------------------------
31
+ # Image adapter
32
+
33
+
34
+ def _accept(prefix: bytes) -> bool:
35
+ return prefix[:4] == b"GRIB" and prefix[7] == 1
36
+
37
+
38
+ class GribStubImageFile(ImageFile.StubImageFile):
39
+ format = "GRIB"
40
+ format_description = "GRIB"
41
+
42
+ def _open(self) -> None:
43
+ offset = self.fp.tell()
44
+
45
+ if not _accept(self.fp.read(8)):
46
+ msg = "Not a GRIB file"
47
+ raise SyntaxError(msg)
48
+
49
+ self.fp.seek(offset)
50
+
51
+ # make something up
52
+ self._mode = "F"
53
+ self._size = 1, 1
54
+
55
+ loader = self._load()
56
+ if loader:
57
+ loader.open(self)
58
+
59
+ def _load(self) -> ImageFile.StubHandler | None:
60
+ return _handler
61
+
62
+
63
+ def _save(im: Image.Image, fp: IO[bytes], filename: str | bytes) -> None:
64
+ if _handler is None or not hasattr(_handler, "save"):
65
+ msg = "GRIB save handler not installed"
66
+ raise OSError(msg)
67
+ _handler.save(im, fp, filename)
68
+
69
+
70
+ # --------------------------------------------------------------------
71
+ # Registry
72
+
73
+ Image.register_open(GribStubImageFile.format, GribStubImageFile, _accept)
74
+ Image.register_save(GribStubImageFile.format, _save)
75
+
76
+ Image.register_extension(GribStubImageFile.format, ".grib")
venv/Lib/site-packages/PIL/Hdf5StubImagePlugin.py ADDED
@@ -0,0 +1,76 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ from typing import IO
14
+
15
+ from . import Image, ImageFile
16
+
17
+ _handler = None
18
+
19
+
20
+ def register_handler(handler: ImageFile.StubHandler | None) -> None:
21
+ """
22
+ Install application-specific HDF5 image handler.
23
+
24
+ :param handler: Handler object.
25
+ """
26
+ global _handler
27
+ _handler = handler
28
+
29
+
30
+ # --------------------------------------------------------------------
31
+ # Image adapter
32
+
33
+
34
+ def _accept(prefix: bytes) -> bool:
35
+ return prefix[:8] == b"\x89HDF\r\n\x1a\n"
36
+
37
+
38
+ class HDF5StubImageFile(ImageFile.StubImageFile):
39
+ format = "HDF5"
40
+ format_description = "HDF5"
41
+
42
+ def _open(self) -> None:
43
+ offset = self.fp.tell()
44
+
45
+ if not _accept(self.fp.read(8)):
46
+ msg = "Not an HDF file"
47
+ raise SyntaxError(msg)
48
+
49
+ self.fp.seek(offset)
50
+
51
+ # make something up
52
+ self._mode = "F"
53
+ self._size = 1, 1
54
+
55
+ loader = self._load()
56
+ if loader:
57
+ loader.open(self)
58
+
59
+ def _load(self) -> ImageFile.StubHandler | None:
60
+ return _handler
61
+
62
+
63
+ def _save(im: Image.Image, fp: IO[bytes], filename: str | bytes) -> None:
64
+ if _handler is None or not hasattr(_handler, "save"):
65
+ msg = "HDF5 save handler not installed"
66
+ raise OSError(msg)
67
+ _handler.save(im, fp, filename)
68
+
69
+
70
+ # --------------------------------------------------------------------
71
+ # Registry
72
+
73
+ Image.register_open(HDF5StubImageFile.format, HDF5StubImageFile, _accept)
74
+ Image.register_save(HDF5StubImageFile.format, _save)
75
+
76
+ Image.register_extensions(HDF5StubImageFile.format, [".h5", ".hdf"])
venv/Lib/site-packages/PIL/IcnsImagePlugin.py ADDED
@@ -0,0 +1,412 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ from ._deprecate import deprecate
29
+
30
+ enable_jpeg2k = features.check_codec("jpg_2000")
31
+ if enable_jpeg2k:
32
+ from . import Jpeg2KImagePlugin
33
+
34
+ MAGIC = b"icns"
35
+ HEADERSIZE = 8
36
+
37
+
38
+ def nextheader(fobj: IO[bytes]) -> tuple[bytes, int]:
39
+ return struct.unpack(">4sI", fobj.read(HEADERSIZE))
40
+
41
+
42
+ def read_32t(
43
+ fobj: IO[bytes], start_length: tuple[int, int], size: tuple[int, int, int]
44
+ ) -> dict[str, Image.Image]:
45
+ # The 128x128 icon seems to have an extra header for some reason.
46
+ (start, length) = start_length
47
+ fobj.seek(start)
48
+ sig = fobj.read(4)
49
+ if sig != b"\x00\x00\x00\x00":
50
+ msg = "Unknown signature, expecting 0x00000000"
51
+ raise SyntaxError(msg)
52
+ return read_32(fobj, (start + 4, length - 4), size)
53
+
54
+
55
+ def read_32(
56
+ fobj: IO[bytes], start_length: tuple[int, int], size: tuple[int, int, int]
57
+ ) -> dict[str, Image.Image]:
58
+ """
59
+ Read a 32bit RGB icon resource. Seems to be either uncompressed or
60
+ an RLE packbits-like scheme.
61
+ """
62
+ (start, length) = start_length
63
+ fobj.seek(start)
64
+ pixel_size = (size[0] * size[2], size[1] * size[2])
65
+ sizesq = pixel_size[0] * pixel_size[1]
66
+ if length == sizesq * 3:
67
+ # uncompressed ("RGBRGBGB")
68
+ indata = fobj.read(length)
69
+ im = Image.frombuffer("RGB", pixel_size, indata, "raw", "RGB", 0, 1)
70
+ else:
71
+ # decode image
72
+ im = Image.new("RGB", pixel_size, None)
73
+ for band_ix in range(3):
74
+ data = []
75
+ bytesleft = sizesq
76
+ while bytesleft > 0:
77
+ byte = fobj.read(1)
78
+ if not byte:
79
+ break
80
+ byte_int = byte[0]
81
+ if byte_int & 0x80:
82
+ blocksize = byte_int - 125
83
+ byte = fobj.read(1)
84
+ for i in range(blocksize):
85
+ data.append(byte)
86
+ else:
87
+ blocksize = byte_int + 1
88
+ data.append(fobj.read(blocksize))
89
+ bytesleft -= blocksize
90
+ if bytesleft <= 0:
91
+ break
92
+ if bytesleft != 0:
93
+ msg = f"Error reading channel [{repr(bytesleft)} left]"
94
+ raise SyntaxError(msg)
95
+ band = Image.frombuffer("L", pixel_size, b"".join(data), "raw", "L", 0, 1)
96
+ im.im.putband(band.im, band_ix)
97
+ return {"RGB": im}
98
+
99
+
100
+ def read_mk(
101
+ fobj: IO[bytes], start_length: tuple[int, int], size: tuple[int, int, int]
102
+ ) -> dict[str, Image.Image]:
103
+ # Alpha masks seem to be uncompressed
104
+ start = start_length[0]
105
+ fobj.seek(start)
106
+ pixel_size = (size[0] * size[2], size[1] * size[2])
107
+ sizesq = pixel_size[0] * pixel_size[1]
108
+ band = Image.frombuffer("L", pixel_size, fobj.read(sizesq), "raw", "L", 0, 1)
109
+ return {"A": band}
110
+
111
+
112
+ def read_png_or_jpeg2000(
113
+ fobj: IO[bytes], start_length: tuple[int, int], size: tuple[int, int, int]
114
+ ) -> dict[str, Image.Image]:
115
+ (start, length) = start_length
116
+ fobj.seek(start)
117
+ sig = fobj.read(12)
118
+
119
+ im: Image.Image
120
+ if sig[:8] == b"\x89PNG\x0d\x0a\x1a\x0a":
121
+ fobj.seek(start)
122
+ im = PngImagePlugin.PngImageFile(fobj)
123
+ Image._decompression_bomb_check(im.size)
124
+ return {"RGBA": im}
125
+ elif (
126
+ sig[:4] == b"\xff\x4f\xff\x51"
127
+ or sig[:4] == b"\x0d\x0a\x87\x0a"
128
+ or sig == b"\x00\x00\x00\x0cjP \x0d\x0a\x87\x0a"
129
+ ):
130
+ if not enable_jpeg2k:
131
+ msg = (
132
+ "Unsupported icon subimage format (rebuild PIL "
133
+ "with JPEG 2000 support to fix this)"
134
+ )
135
+ raise ValueError(msg)
136
+ # j2k, jpc or j2c
137
+ fobj.seek(start)
138
+ jp2kstream = fobj.read(length)
139
+ f = io.BytesIO(jp2kstream)
140
+ im = Jpeg2KImagePlugin.Jpeg2KImageFile(f)
141
+ Image._decompression_bomb_check(im.size)
142
+ if im.mode != "RGBA":
143
+ im = im.convert("RGBA")
144
+ return {"RGBA": im}
145
+ else:
146
+ msg = "Unsupported icon subimage format"
147
+ raise ValueError(msg)
148
+
149
+
150
+ class IcnsFile:
151
+ SIZES = {
152
+ (512, 512, 2): [(b"ic10", read_png_or_jpeg2000)],
153
+ (512, 512, 1): [(b"ic09", read_png_or_jpeg2000)],
154
+ (256, 256, 2): [(b"ic14", read_png_or_jpeg2000)],
155
+ (256, 256, 1): [(b"ic08", read_png_or_jpeg2000)],
156
+ (128, 128, 2): [(b"ic13", read_png_or_jpeg2000)],
157
+ (128, 128, 1): [
158
+ (b"ic07", read_png_or_jpeg2000),
159
+ (b"it32", read_32t),
160
+ (b"t8mk", read_mk),
161
+ ],
162
+ (64, 64, 1): [(b"icp6", read_png_or_jpeg2000)],
163
+ (32, 32, 2): [(b"ic12", read_png_or_jpeg2000)],
164
+ (48, 48, 1): [(b"ih32", read_32), (b"h8mk", read_mk)],
165
+ (32, 32, 1): [
166
+ (b"icp5", read_png_or_jpeg2000),
167
+ (b"il32", read_32),
168
+ (b"l8mk", read_mk),
169
+ ],
170
+ (16, 16, 2): [(b"ic11", read_png_or_jpeg2000)],
171
+ (16, 16, 1): [
172
+ (b"icp4", read_png_or_jpeg2000),
173
+ (b"is32", read_32),
174
+ (b"s8mk", read_mk),
175
+ ],
176
+ }
177
+
178
+ def __init__(self, fobj: IO[bytes]) -> None:
179
+ """
180
+ fobj is a file-like object as an icns resource
181
+ """
182
+ # signature : (start, length)
183
+ self.dct = {}
184
+ self.fobj = fobj
185
+ sig, filesize = nextheader(fobj)
186
+ if not _accept(sig):
187
+ msg = "not an icns file"
188
+ raise SyntaxError(msg)
189
+ i = HEADERSIZE
190
+ while i < filesize:
191
+ sig, blocksize = nextheader(fobj)
192
+ if blocksize <= 0:
193
+ msg = "invalid block header"
194
+ raise SyntaxError(msg)
195
+ i += HEADERSIZE
196
+ blocksize -= HEADERSIZE
197
+ self.dct[sig] = (i, blocksize)
198
+ fobj.seek(blocksize, io.SEEK_CUR)
199
+ i += blocksize
200
+
201
+ def itersizes(self) -> list[tuple[int, int, int]]:
202
+ sizes = []
203
+ for size, fmts in self.SIZES.items():
204
+ for fmt, reader in fmts:
205
+ if fmt in self.dct:
206
+ sizes.append(size)
207
+ break
208
+ return sizes
209
+
210
+ def bestsize(self) -> tuple[int, int, int]:
211
+ sizes = self.itersizes()
212
+ if not sizes:
213
+ msg = "No 32bit icon resources found"
214
+ raise SyntaxError(msg)
215
+ return max(sizes)
216
+
217
+ def dataforsize(self, size: tuple[int, int, int]) -> dict[str, Image.Image]:
218
+ """
219
+ Get an icon resource as {channel: array}. Note that
220
+ the arrays are bottom-up like windows bitmaps and will likely
221
+ need to be flipped or transposed in some way.
222
+ """
223
+ dct = {}
224
+ for code, reader in self.SIZES[size]:
225
+ desc = self.dct.get(code)
226
+ if desc is not None:
227
+ dct.update(reader(self.fobj, desc, size))
228
+ return dct
229
+
230
+ def getimage(
231
+ self, size: tuple[int, int] | tuple[int, int, int] | None = None
232
+ ) -> Image.Image:
233
+ if size is None:
234
+ size = self.bestsize()
235
+ elif len(size) == 2:
236
+ size = (size[0], size[1], 1)
237
+ channels = self.dataforsize(size)
238
+
239
+ im = channels.get("RGBA")
240
+ if im:
241
+ return im
242
+
243
+ im = channels["RGB"].copy()
244
+ try:
245
+ im.putalpha(channels["A"])
246
+ except KeyError:
247
+ pass
248
+ return im
249
+
250
+
251
+ ##
252
+ # Image plugin for Mac OS icons.
253
+
254
+
255
+ class IcnsImageFile(ImageFile.ImageFile):
256
+ """
257
+ PIL image support for Mac OS .icns files.
258
+ Chooses the best resolution, but will possibly load
259
+ a different size image if you mutate the size attribute
260
+ before calling 'load'.
261
+
262
+ The info dictionary has a key 'sizes' that is a list
263
+ of sizes that the icns file has.
264
+ """
265
+
266
+ format = "ICNS"
267
+ format_description = "Mac OS icns resource"
268
+
269
+ def _open(self) -> None:
270
+ self.icns = IcnsFile(self.fp)
271
+ self._mode = "RGBA"
272
+ self.info["sizes"] = self.icns.itersizes()
273
+ self.best_size = self.icns.bestsize()
274
+ self.size = (
275
+ self.best_size[0] * self.best_size[2],
276
+ self.best_size[1] * self.best_size[2],
277
+ )
278
+
279
+ @property # type: ignore[override]
280
+ def size(self) -> tuple[int, int] | tuple[int, int, int]:
281
+ return self._size
282
+
283
+ @size.setter
284
+ def size(self, value: tuple[int, int] | tuple[int, int, int]) -> None:
285
+ if len(value) == 3:
286
+ deprecate("Setting size to (width, height, scale)", 12, "load(scale)")
287
+ if value in self.info["sizes"]:
288
+ self._size = value # type: ignore[assignment]
289
+ return
290
+ else:
291
+ # Check that a matching size exists,
292
+ # or that there is a scale that would create a size that matches
293
+ for size in self.info["sizes"]:
294
+ simple_size = size[0] * size[2], size[1] * size[2]
295
+ scale = simple_size[0] // value[0]
296
+ if simple_size[1] / value[1] == scale:
297
+ self._size = value
298
+ return
299
+ msg = "This is not one of the allowed sizes of this image"
300
+ raise ValueError(msg)
301
+
302
+ def load(self, scale: int | None = None) -> Image.core.PixelAccess | None:
303
+ if scale is not None or len(self.size) == 3:
304
+ if scale is None and len(self.size) == 3:
305
+ scale = self.size[2]
306
+ assert scale is not None
307
+ width, height = self.size[:2]
308
+ self.size = width * scale, height * scale
309
+ self.best_size = width, height, scale
310
+
311
+ px = Image.Image.load(self)
312
+ if self._im is not None and self.im.size == self.size:
313
+ # Already loaded
314
+ return px
315
+ self.load_prepare()
316
+ # This is likely NOT the best way to do it, but whatever.
317
+ im = self.icns.getimage(self.best_size)
318
+
319
+ # If this is a PNG or JPEG 2000, it won't be loaded yet
320
+ px = im.load()
321
+
322
+ self.im = im.im
323
+ self._mode = im.mode
324
+ self.size = im.size
325
+
326
+ return px
327
+
328
+
329
+ def _save(im: Image.Image, fp: IO[bytes], filename: str | bytes) -> None:
330
+ """
331
+ Saves the image as a series of PNG files,
332
+ that are then combined into a .icns file.
333
+ """
334
+ if hasattr(fp, "flush"):
335
+ fp.flush()
336
+
337
+ sizes = {
338
+ b"ic07": 128,
339
+ b"ic08": 256,
340
+ b"ic09": 512,
341
+ b"ic10": 1024,
342
+ b"ic11": 32,
343
+ b"ic12": 64,
344
+ b"ic13": 256,
345
+ b"ic14": 512,
346
+ }
347
+ provided_images = {im.width: im for im in im.encoderinfo.get("append_images", [])}
348
+ size_streams = {}
349
+ for size in set(sizes.values()):
350
+ image = (
351
+ provided_images[size]
352
+ if size in provided_images
353
+ else im.resize((size, size))
354
+ )
355
+
356
+ temp = io.BytesIO()
357
+ image.save(temp, "png")
358
+ size_streams[size] = temp.getvalue()
359
+
360
+ entries = []
361
+ for type, size in sizes.items():
362
+ stream = size_streams[size]
363
+ entries.append((type, HEADERSIZE + len(stream), stream))
364
+
365
+ # Header
366
+ fp.write(MAGIC)
367
+ file_length = HEADERSIZE # Header
368
+ file_length += HEADERSIZE + 8 * len(entries) # TOC
369
+ file_length += sum(entry[1] for entry in entries)
370
+ fp.write(struct.pack(">i", file_length))
371
+
372
+ # TOC
373
+ fp.write(b"TOC ")
374
+ fp.write(struct.pack(">i", HEADERSIZE + len(entries) * HEADERSIZE))
375
+ for entry in entries:
376
+ fp.write(entry[0])
377
+ fp.write(struct.pack(">i", entry[1]))
378
+
379
+ # Data
380
+ for entry in entries:
381
+ fp.write(entry[0])
382
+ fp.write(struct.pack(">i", entry[1]))
383
+ fp.write(entry[2])
384
+
385
+ if hasattr(fp, "flush"):
386
+ fp.flush()
387
+
388
+
389
+ def _accept(prefix: bytes) -> bool:
390
+ return prefix[:4] == MAGIC
391
+
392
+
393
+ Image.register_open(IcnsImageFile.format, IcnsImageFile, _accept)
394
+ Image.register_extension(IcnsImageFile.format, ".icns")
395
+
396
+ Image.register_save(IcnsImageFile.format, _save)
397
+ Image.register_mime(IcnsImageFile.format, "image/icns")
398
+
399
+ if __name__ == "__main__":
400
+ if len(sys.argv) < 2:
401
+ print("Syntax: python3 IcnsImagePlugin.py [file]")
402
+ sys.exit()
403
+
404
+ with open(sys.argv[1], "rb") as fp:
405
+ imf = IcnsImageFile(fp)
406
+ for size in imf.info["sizes"]:
407
+ width, height, scale = imf.size = size
408
+ imf.save(f"out-{width}-{height}-{scale}.png")
409
+ with Image.open(sys.argv[1]) as im:
410
+ im.save("out.png")
411
+ if sys.platform == "windows":
412
+ 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[:4] == _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 None
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,386 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+
35
+ # --------------------------------------------------------------------
36
+ # Standard tags
37
+
38
+ COMMENT = "Comment"
39
+ DATE = "Date"
40
+ EQUIPMENT = "Digitalization equipment"
41
+ FRAMES = "File size (no of images)"
42
+ LUT = "Lut"
43
+ NAME = "Name"
44
+ SCALE = "Scale (x,y)"
45
+ SIZE = "Image size (x*y)"
46
+ MODE = "Image type"
47
+
48
+ TAGS = {
49
+ COMMENT: 0,
50
+ DATE: 0,
51
+ EQUIPMENT: 0,
52
+ FRAMES: 0,
53
+ LUT: 0,
54
+ NAME: 0,
55
+ SCALE: 0,
56
+ SIZE: 0,
57
+ MODE: 0,
58
+ }
59
+
60
+ OPEN = {
61
+ # ifunc93/p3cfunc formats
62
+ "0 1 image": ("1", "1"),
63
+ "L 1 image": ("1", "1"),
64
+ "Greyscale image": ("L", "L"),
65
+ "Grayscale image": ("L", "L"),
66
+ "RGB image": ("RGB", "RGB;L"),
67
+ "RLB image": ("RGB", "RLB"),
68
+ "RYB image": ("RGB", "RLB"),
69
+ "B1 image": ("1", "1"),
70
+ "B2 image": ("P", "P;2"),
71
+ "B4 image": ("P", "P;4"),
72
+ "X 24 image": ("RGB", "RGB"),
73
+ "L 32 S image": ("I", "I;32"),
74
+ "L 32 F image": ("F", "F;32"),
75
+ # old p3cfunc formats
76
+ "RGB3 image": ("RGB", "RGB;T"),
77
+ "RYB3 image": ("RGB", "RYB;T"),
78
+ # extensions
79
+ "LA image": ("LA", "LA;L"),
80
+ "PA image": ("LA", "PA;L"),
81
+ "RGBA image": ("RGBA", "RGBA;L"),
82
+ "RGBX image": ("RGB", "RGBX;L"),
83
+ "CMYK image": ("CMYK", "CMYK;L"),
84
+ "YCC image": ("YCbCr", "YCbCr;L"),
85
+ }
86
+
87
+ # ifunc95 extensions
88
+ for i in ["8", "8S", "16", "16S", "32", "32F"]:
89
+ OPEN[f"L {i} image"] = ("F", f"F;{i}")
90
+ OPEN[f"L*{i} image"] = ("F", f"F;{i}")
91
+ for i in ["16", "16L", "16B"]:
92
+ OPEN[f"L {i} image"] = (f"I;{i}", f"I;{i}")
93
+ OPEN[f"L*{i} image"] = (f"I;{i}", f"I;{i}")
94
+ for i in ["32S"]:
95
+ OPEN[f"L {i} image"] = ("I", f"I;{i}")
96
+ OPEN[f"L*{i} image"] = ("I", f"I;{i}")
97
+ for j in range(2, 33):
98
+ OPEN[f"L*{j} image"] = ("F", f"F;{j}")
99
+
100
+
101
+ # --------------------------------------------------------------------
102
+ # Read IM directory
103
+
104
+ split = re.compile(rb"^([A-Za-z][^:]*):[ \t]*(.*)[ \t]*$")
105
+
106
+
107
+ def number(s: Any) -> float:
108
+ try:
109
+ return int(s)
110
+ except ValueError:
111
+ return float(s)
112
+
113
+
114
+ ##
115
+ # Image plugin for the IFUNC IM file format.
116
+
117
+
118
+ class ImImageFile(ImageFile.ImageFile):
119
+ format = "IM"
120
+ format_description = "IFUNC Image Memory"
121
+ _close_exclusive_fp_after_loading = False
122
+
123
+ def _open(self) -> None:
124
+ # Quick rejection: if there's not an LF among the first
125
+ # 100 bytes, this is (probably) not a text header.
126
+
127
+ if b"\n" not in self.fp.read(100):
128
+ msg = "not an IM file"
129
+ raise SyntaxError(msg)
130
+ self.fp.seek(0)
131
+
132
+ n = 0
133
+
134
+ # Default values
135
+ self.info[MODE] = "L"
136
+ self.info[SIZE] = (512, 512)
137
+ self.info[FRAMES] = 1
138
+
139
+ self.rawmode = "L"
140
+
141
+ while True:
142
+ s = self.fp.read(1)
143
+
144
+ # Some versions of IFUNC uses \n\r instead of \r\n...
145
+ if s == b"\r":
146
+ continue
147
+
148
+ if not s or s == b"\0" or s == b"\x1A":
149
+ break
150
+
151
+ # FIXME: this may read whole file if not a text file
152
+ s = s + self.fp.readline()
153
+
154
+ if len(s) > 100:
155
+ msg = "not an IM file"
156
+ raise SyntaxError(msg)
157
+
158
+ if s[-2:] == b"\r\n":
159
+ s = s[:-2]
160
+ elif s[-1:] == b"\n":
161
+ s = s[:-1]
162
+
163
+ try:
164
+ m = split.match(s)
165
+ except re.error as e:
166
+ msg = "not an IM file"
167
+ raise SyntaxError(msg) from e
168
+
169
+ if m:
170
+ k, v = m.group(1, 2)
171
+
172
+ # Don't know if this is the correct encoding,
173
+ # but a decent guess (I guess)
174
+ k = k.decode("latin-1", "replace")
175
+ v = v.decode("latin-1", "replace")
176
+
177
+ # Convert value as appropriate
178
+ if k in [FRAMES, SCALE, SIZE]:
179
+ v = v.replace("*", ",")
180
+ v = tuple(map(number, v.split(",")))
181
+ if len(v) == 1:
182
+ v = v[0]
183
+ elif k == MODE and v in OPEN:
184
+ v, self.rawmode = OPEN[v]
185
+
186
+ # Add to dictionary. Note that COMMENT tags are
187
+ # combined into a list of strings.
188
+ if k == COMMENT:
189
+ if k in self.info:
190
+ self.info[k].append(v)
191
+ else:
192
+ self.info[k] = [v]
193
+ else:
194
+ self.info[k] = v
195
+
196
+ if k in TAGS:
197
+ n += 1
198
+
199
+ else:
200
+ msg = f"Syntax error in IM header: {s.decode('ascii', 'replace')}"
201
+ raise SyntaxError(msg)
202
+
203
+ if not n:
204
+ msg = "Not an IM file"
205
+ raise SyntaxError(msg)
206
+
207
+ # Basic attributes
208
+ self._size = self.info[SIZE]
209
+ self._mode = self.info[MODE]
210
+
211
+ # Skip forward to start of image data
212
+ while s and s[:1] != b"\x1A":
213
+ s = self.fp.read(1)
214
+ if not s:
215
+ msg = "File truncated"
216
+ raise SyntaxError(msg)
217
+
218
+ if LUT in self.info:
219
+ # convert lookup table to palette or lut attribute
220
+ palette = self.fp.read(768)
221
+ greyscale = 1 # greyscale palette
222
+ linear = 1 # linear greyscale palette
223
+ for i in range(256):
224
+ if palette[i] == palette[i + 256] == palette[i + 512]:
225
+ if palette[i] != i:
226
+ linear = 0
227
+ else:
228
+ greyscale = 0
229
+ if self.mode in ["L", "LA", "P", "PA"]:
230
+ if greyscale:
231
+ if not linear:
232
+ self.lut = list(palette[:256])
233
+ else:
234
+ if self.mode in ["L", "P"]:
235
+ self._mode = self.rawmode = "P"
236
+ elif self.mode in ["LA", "PA"]:
237
+ self._mode = "PA"
238
+ self.rawmode = "PA;L"
239
+ self.palette = ImagePalette.raw("RGB;L", palette)
240
+ elif self.mode == "RGB":
241
+ if not greyscale or not linear:
242
+ self.lut = list(palette)
243
+
244
+ self.frame = 0
245
+
246
+ self.__offset = offs = self.fp.tell()
247
+
248
+ self._fp = self.fp # FIXME: hack
249
+
250
+ if self.rawmode[:2] == "F;":
251
+ # ifunc95 formats
252
+ try:
253
+ # use bit decoder (if necessary)
254
+ bits = int(self.rawmode[2:])
255
+ if bits not in [8, 16, 32]:
256
+ self.tile = [
257
+ ImageFile._Tile(
258
+ "bit", (0, 0) + self.size, offs, (bits, 8, 3, 0, -1)
259
+ )
260
+ ]
261
+ return
262
+ except ValueError:
263
+ pass
264
+
265
+ if self.rawmode in ["RGB;T", "RYB;T"]:
266
+ # Old LabEye/3PC files. Would be very surprised if anyone
267
+ # ever stumbled upon such a file ;-)
268
+ size = self.size[0] * self.size[1]
269
+ self.tile = [
270
+ ImageFile._Tile("raw", (0, 0) + self.size, offs, ("G", 0, -1)),
271
+ ImageFile._Tile("raw", (0, 0) + self.size, offs + size, ("R", 0, -1)),
272
+ ImageFile._Tile(
273
+ "raw", (0, 0) + self.size, offs + 2 * size, ("B", 0, -1)
274
+ ),
275
+ ]
276
+ else:
277
+ # LabEye/IFUNC files
278
+ self.tile = [
279
+ ImageFile._Tile("raw", (0, 0) + self.size, offs, (self.rawmode, 0, -1))
280
+ ]
281
+
282
+ @property
283
+ def n_frames(self) -> int:
284
+ return self.info[FRAMES]
285
+
286
+ @property
287
+ def is_animated(self) -> bool:
288
+ return self.info[FRAMES] > 1
289
+
290
+ def seek(self, frame: int) -> None:
291
+ if not self._seek_check(frame):
292
+ return
293
+
294
+ self.frame = frame
295
+
296
+ if self.mode == "1":
297
+ bits = 1
298
+ else:
299
+ bits = 8 * len(self.mode)
300
+
301
+ size = ((self.size[0] * bits + 7) // 8) * self.size[1]
302
+ offs = self.__offset + frame * size
303
+
304
+ self.fp = self._fp
305
+
306
+ self.tile = [
307
+ ImageFile._Tile("raw", (0, 0) + self.size, offs, (self.rawmode, 0, -1))
308
+ ]
309
+
310
+ def tell(self) -> int:
311
+ return self.frame
312
+
313
+
314
+ #
315
+ # --------------------------------------------------------------------
316
+ # Save IM files
317
+
318
+
319
+ SAVE = {
320
+ # mode: (im type, raw mode)
321
+ "1": ("0 1", "1"),
322
+ "L": ("Greyscale", "L"),
323
+ "LA": ("LA", "LA;L"),
324
+ "P": ("Greyscale", "P"),
325
+ "PA": ("LA", "PA;L"),
326
+ "I": ("L 32S", "I;32S"),
327
+ "I;16": ("L 16", "I;16"),
328
+ "I;16L": ("L 16L", "I;16L"),
329
+ "I;16B": ("L 16B", "I;16B"),
330
+ "F": ("L 32F", "F;32F"),
331
+ "RGB": ("RGB", "RGB;L"),
332
+ "RGBA": ("RGBA", "RGBA;L"),
333
+ "RGBX": ("RGBX", "RGBX;L"),
334
+ "CMYK": ("CMYK", "CMYK;L"),
335
+ "YCbCr": ("YCC", "YCbCr;L"),
336
+ }
337
+
338
+
339
+ def _save(im: Image.Image, fp: IO[bytes], filename: str | bytes) -> None:
340
+ try:
341
+ image_type, rawmode = SAVE[im.mode]
342
+ except KeyError as e:
343
+ msg = f"Cannot save {im.mode} images as IM"
344
+ raise ValueError(msg) from e
345
+
346
+ frames = im.encoderinfo.get("frames", 1)
347
+
348
+ fp.write(f"Image type: {image_type} image\r\n".encode("ascii"))
349
+ if filename:
350
+ # Each line must be 100 characters or less,
351
+ # or: SyntaxError("not an IM file")
352
+ # 8 characters are used for "Name: " and "\r\n"
353
+ # Keep just the filename, ditch the potentially overlong path
354
+ if isinstance(filename, bytes):
355
+ filename = filename.decode("ascii")
356
+ name, ext = os.path.splitext(os.path.basename(filename))
357
+ name = "".join([name[: 92 - len(ext)], ext])
358
+
359
+ fp.write(f"Name: {name}\r\n".encode("ascii"))
360
+ fp.write(("Image size (x*y): %d*%d\r\n" % im.size).encode("ascii"))
361
+ fp.write(f"File size (no of images): {frames}\r\n".encode("ascii"))
362
+ if im.mode in ["P", "PA"]:
363
+ fp.write(b"Lut: 1\r\n")
364
+ fp.write(b"\000" * (511 - fp.tell()) + b"\032")
365
+ if im.mode in ["P", "PA"]:
366
+ im_palette = im.im.getpalette("RGB", "RGB;L")
367
+ colors = len(im_palette) // 3
368
+ palette = b""
369
+ for i in range(3):
370
+ palette += im_palette[colors * i : colors * (i + 1)]
371
+ palette += b"\x00" * (256 - colors)
372
+ fp.write(palette) # 768 bytes
373
+ ImageFile._save(
374
+ im, fp, [ImageFile._Tile("raw", (0, 0) + im.size, 0, (rawmode, 0, -1))]
375
+ )
376
+
377
+
378
+ #
379
+ # --------------------------------------------------------------------
380
+ # Registry
381
+
382
+
383
+ Image.register_open(ImImageFile.format, ImImageFile)
384
+ Image.register_save(ImImageFile.format, _save)
385
+
386
+ 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))