1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
|
from OpenGL.GL import *
from OpenGL.GLUT import *
from OpenGL.GLU import *
from PIL import Image, ExifTags
class ImageTexture:
def __init__(self, image_source, exif=None):
self.id = None
img = Image.open(image_source)
self.exif = exif or self.get_exif(img)
img = self.handle_orientation(img, self.exif)
img = img.convert("RGBA") # Ensure the image is in RGBA mode
self.width = img.width
self.height = img.height
self._img_data = img.tobytes("raw", "RGBA", 0, -1) # Convert image data to bytes
self.initialized = True
def gl_init(self):
if self.id:
return
#glPixelStorei(GL_UNPACK_ALIGNMENT, 1)
self.id = glGenTextures(1)
glBindTexture(GL_TEXTURE_2D, self.id)
glTexImage2D(GL_TEXTURE_2D, 0, GL_RGBA, self.width, self.height, 0, GL_RGBA, GL_UNSIGNED_BYTE, self._img_data)
#glTexImage2D(GL_TEXTURE_2D, 0, GL_RGB, img.width, img.height, 0, GL_RGB, GL_UNSIGNED_BYTE, img_data)
glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_LINEAR)
glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_LINEAR)
del self._img_data # No longer needed, clear mem
#print(f"Loaded texture {self.id}")
def free(self):
if self.id is None:
del self._img_data
else:
glBindTexture(GL_TEXTURE_2D, 0) # Unbind
glDeleteTextures([self.id])
@staticmethod
def get_exif(img):
return { ExifTags.TAGS[k]: v for k, v in img.getexif().items() if k in ExifTags.TAGS }
@staticmethod
def handle_orientation(img, exif):
orientation = exif.get("Orientation", 1) if exif is not None else 1
if orientation == 3: return img.rotate(180, expand=True)
if orientation == 6: return img.rotate(270, expand=True)
if orientation == 8: return img.rotate( 90, expand=True)
return img
class ImageTextureImmichAsset(ImageTexture):
def __init__(self, asset, asset_index):
self.initialized = False
self.asset_index = asset_index
self.asset_key = asset["id"]
self.exif = asset.get("exif", None)
# So that free can work
self.id = None
self._img_data = None
def initialize(self, image_source):
super().__init__(image_source, self.exif)
|
