tcsenpai/pensieve
1
0
Fork 0
mirror of https://github.com/tcsenpai/pensieve.git synced 2025-06-09 20:47:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: support png info

Browse Source
This commit is contained in:
arkohut 2024-08-13 12:39:31 +08:00
parent c95b1fcfee
commit 615a938e5c
2 changed files with 55 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
screen-recorder/read_metadata.py
View File

@ -1,23 +1,34 @@
import piexif import piexif
import json import json
import argparse import argparse
from PIL import Image from PIL import Image, PngImagePlugin
def read_metadata(image_path): def read_metadata(image_path):
try: try:
img = Image.open(image_path) img = Image.open(image_path)
exif_data = img.info.get('exif') exif_data = img.info.get('exif')
if not exif_data: png_info = img.info if isinstance(img, PngImagePlugin.PngImageFile) else None
print("No EXIF metadata found.")
if not exif_data and not png_info:
print("No EXIF or PNG metadata found.")
return return
if exif_data:
exif_dict = piexif.load(exif_data) exif_dict = piexif.load(exif_data)
metadata_json = exif_dict["0th"].get(piexif.ImageIFD.ImageDescription) metadata_json = exif_dict["0th"].get(piexif.ImageIFD.ImageDescription)
if metadata_json: if metadata_json:
metadata = json.loads(metadata_json.decode()) metadata = json.loads(metadata_json.decode())
print("Metadata:", json.dumps(metadata, indent=4)) print("EXIF Metadata:", json.dumps(metadata, indent=4))
else: else:
print("No metadata found in the ImageDescription field.") print("No metadata found in the ImageDescription field of EXIF.")
if png_info:
metadata_json = png_info.get("Description")
if metadata_json:
metadata = json.loads(metadata_json)
print("PNG Metadata:", json.dumps(metadata, indent=4))
else:
print("No metadata found in the Description field of PNG.")
except Exception as e: except Exception as e:
print(f"An error occurred: {str(e)}") print(f"An error occurred: {str(e)}")

33
screen-recorder/video_generator.py
View File

@ -5,6 +5,7 @@ import glob
import subprocess import subprocess
from PIL import Image from PIL import Image
import piexif import piexif
from PIL.PngImagePlugin import PngInfo
from multiprocessing import Pool, Manager from multiprocessing import Pool, Manager
from tqdm import tqdm from tqdm import tqdm
@ -19,25 +20,39 @@ def compress_and_save_image(image_path, order):
# Open the image # Open the image
img = Image.open(image_path) img = Image.open(image_path)
# Add order to the image metadata if image_path.endswith(('.jpg', '.jpeg', '.tiff')):
# Add order to the image metadata for JPEG/TIFF
exif_dict = piexif.load(image_path) exif_dict = piexif.load(image_path)
# Get existing ImageDescription if it exists
existing_description = exif_dict["0th"].get(piexif.ImageIFD.ImageDescription, b'{}') existing_description = exif_dict["0th"].get(piexif.ImageIFD.ImageDescription, b'{}')
try: try:
existing_data = json.loads(existing_description.decode('utf-8')) existing_data = json.loads(existing_description.decode('utf-8'))
except json.JSONDecodeError: except json.JSONDecodeError:
existing_data = {} existing_data = {}
# Add or update the "sequence" key
existing_data["sequence"] = order existing_data["sequence"] = order
existing_data["is_thumbnail"] = True
# Convert back to JSON string and encode
updated_description = json.dumps(existing_data).encode('utf-8') updated_description = json.dumps(existing_data).encode('utf-8')
exif_dict["0th"][piexif.ImageIFD.ImageDescription] = updated_description exif_dict["0th"][piexif.ImageIFD.ImageDescription] = updated_description
exif_bytes = piexif.dump(exif_dict)
elif image_path.endswith('.png'):
# Add order to the image metadata for PNG
metadata = PngInfo()
existing_description = img.info.get("Description", '{}')
try:
existing_data = json.loads(existing_description)
except json.JSONDecodeError:
existing_data = {}
existing_data["sequence"] = order
existing_data["is_thumbnail"] = True
updated_description = json.dumps(existing_data)
metadata.add_text("Description", updated_description)
else:
print(f"Skipping unsupported file format: {image_path}")
return
# Compress the image # Compress the image
img = img.convert("RGB") img = img.convert("RGB")
if image_path.endswith('.png'): if image_path.endswith('.png'):
img.save(image_path, "PNG", optimize=True) img.save(image_path, "PNG", optimize=True, pnginfo=metadata)
else: else:
img.save(image_path, "JPEG", quality=30) # Lower quality for higher compression img.save(image_path, "JPEG", quality=30) # Lower quality for higher compression
@ -45,12 +60,12 @@ def compress_and_save_image(image_path, order):
max_size = (960, 960) # Define the maximum size for the thumbnail max_size = (960, 960) # Define the maximum size for the thumbnail
img.thumbnail(max_size) img.thumbnail(max_size)
if image_path.endswith('.png'): if image_path.endswith('.png'):
img.save(image_path, "PNG", optimize=True) img.save(image_path, "PNG", optimize=True, pnginfo=metadata)
else: else:
img.save(image_path, "JPEG", quality=30) # Lower quality for higher compression img.save(image_path, "JPEG", quality=30) # Lower quality for higher compression
# Insert updated EXIF data if image_path.endswith(('.jpg', '.jpeg', '.tiff')):
exif_bytes = piexif.dump(exif_dict) # Insert updated EXIF data for JPEG/TIFF
piexif.insert(exif_bytes, image_path) piexif.insert(exif_bytes, image_path)
return image_path return image_path