Compare commits

..

3 Commits

Author SHA1 Message Date
4b17c94da4 update gitignore 2025-07-03 05:55:24 +02:00
cbfff3f6f7 basic x bullshittery 2025-07-03 05:55:19 +02:00
f7d8292029 Download + basic stupid pixel RGB value extractor (fps, resolution) 2025-07-03 05:31:21 +02:00
5 changed files with 267 additions and 0 deletions

2
.gitignore vendored Normal file
View File

@ -0,0 +1,2 @@
apples/
a.out

View File

@ -0,0 +1,146 @@
import yt_dlp
import os
import cv2
from time import sleep
from typing import List
from tqdm import tqdm
BAD_APPLE = {
'960': {
'url': 'https://www.youtube.com/watch?v=9lNZ_Rnr7Jc',
'format': 'bestvideo[ext=mp4][width<=960]+bestaudio[ext=m4a]/best[ext=mp4]',
"ratio": (4, 3)
},
'1080': {
'url': 'https://www.youtube.com/watch?v=0n3jW9vlH70',
'format': 'bestvideo[ext=mp4][width<=1080]+bestaudio[ext=m4a]/best[ext=mp4]',
"ratio":(16, 9)
},
'1440': {
'url': 'https://www.youtube.com/watch?v=0n3jW9vlH70',
'format': 'bestvideo[ext=mp4][width<=1440]+bestaudio[ext=m4a]/best[ext=mp4]',
"ratio": (16, 9)
},
'2160': {
'url': 'https://www.youtube.com/watch?v=0n3jW9vlH70',
'format': 'bestvideo[ext=mp4][width<=2160]+bestaudio[ext=m4a]/best[ext=mp4]',
"ratio": (16, 9)
},
'audio': {
'url': 'https://www.youtube.com/watch?v=9lNZ_Rnr7Jc',
'format': 'bestaudio[ext=m4a]/best',
"ratio": None
},
}
class BadApple:
def __init__(self, download:bool=True, keys: List[str] = None) -> None:
self.keys = keys if keys is not None else []
if download and self.keys:
self.download_videos(self.keys)
def header(self, video: str, fps: int) -> str:
cap = cv2.VideoCapture(video)
video_fps = cap.get(cv2.CAP_PROP_FPS)
total_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
cap.release()
header_lines = [
f"fps: {fps}",
f"frames: {total_frames}",
f"resolution: {width}x{height}"
]
return "\n".join(header_lines)
def basic(self, video: str, output_file: str, fps: int = 15, width: int = None, height: int = None) -> None:
vid = f"{os.path.splitext(video)[0]}.mp4"
cap = cv2.VideoCapture(vid)
video_fps = cap.get(cv2.CAP_PROP_FPS)
orig_width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
orig_height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
total_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
if width is None and height is None:
target_width, target_height = orig_width, orig_height
elif width is not None and height is not None:
# Check aspect ratio
if orig_width * height != orig_height * width:
raise ValueError("Violated aspect ratio")
target_width, target_height = width, height
elif width is not None:
target_width = width
target_height = int(orig_height * (width / orig_width))
else:
target_height = height
target_width = int(orig_width * (height / orig_height))
if target_width > orig_width or target_height > orig_height:
raise ValueError("Lower your resolution")
frame_interval = int(round(video_fps / fps)) if video_fps > 0 else 1
frame_count = 0
with open(output_file, "w") as f, tqdm(total=total_frames, desc="Extracting frames") as pbar:
# Write header
f.write(self.header(vid, fps) + f"\ntarget_resolution: {target_width}x{target_height}\n")
while cap.isOpened():
ret, frame = cap.read()
if not ret:
break
if frame_count % frame_interval == 0:
if (target_width, target_height) != (orig_width, orig_height):
frame = cv2.resize(frame, (target_width, target_height), interpolation=cv2.INTER_AREA)
rgb_frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
rows = []
for row in rgb_frame:
row_str = ";".join([f"{r},{g},{b}" for r, g, b in row])
rows.append(row_str)
f.write("|".join(rows) + "\n")
frame_count += 1
pbar.update(1)
cap.release()
# print filesize
file_size_in_gb = os.path.getsize(output_file) / (2**30)
print(f"Extracted {frame_count} frames to {output_file} ({file_size:.2f} gb)")
def download_videos(self, keys: List[str]) -> None:
os.makedirs("apples", exist_ok=True)
for key in keys:
info = BAD_APPLE[key]
if key == "audio":
out_path = f"apples/audio.m4a"
else:
out_path = f"apples/{key}.mp4"
ydl_opts = {
'format': info['format'],
'outtmpl': out_path,
'quiet': True,
'merge_output_format': 'mp4' if key != "audio" else 'm4a'
}
with yt_dlp.YoutubeDL(ydl_opts) as ydl:
ydl.download([info['url']])
new_path = f"apples/{key}_new.mp4"
print(f"Downlaoded! Processing {key} video...")
if key != "audio" and info["ratio"] != (4, 3):
# crop to 4:3
crop_filter = "crop=ih*4/3:ih"
os.system(
f'ffmpeg -y -loglevel error -i "{out_path}" -vf "{crop_filter}" -c:v h264 -pix_fmt yuv420p -c:a copy "{new_path}"'
)
# print aspect ratio, encoding info
os.replace(new_path, out_path)
elif info["ratio"] == (4, 3):
# no cropping needed, only h264 encoding
os.system(
f'ffmpeg -y -loglevel error -i "{out_path}" -c:v h264 -pix_fmt yuv420p -c:a copy "{new_path}"'
)
os.replace(new_path, out_path)
os.system(f'ffprobe -v error -select_streams v:0 -show_entries stream=width,height,codec_name,bit_rate -of default=noprint_wrappers=1 "{out_path}"')
if __name__ == "__main__":
# ba = BadApple(download=False)
ba = BadApple(download=False, keys=['960'])
ba.basic("apples/960.mp4", "apples/960_pixels_320x240.txt", width=320)

7
requirements.txt Normal file
View File

@ -0,0 +1,7 @@
ffmpeg==1.4
ffmpeg-python==0.2.0
future==1.0.0
numpy==2.3.1
opencv-python==4.11.0.86
tqdm==4.67.1
yt-dlp==2025.6.30

18
x11/Makefile Normal file
View File

@ -0,0 +1,18 @@
CC = gcc
CFLAGS = -Wall -g
LDFLAGS = -lX11
TARGET = a.out
SRC = *.c
all: $(TARGET)
$(TARGET): $(SRC)
$(CC) $(CFLAGS) -o $(TARGET) $(SRC) $(LDFLAGS)
clean:
rm -f $(TARGET)
run: $(TARGET)
$(MAKE) clean
$(MAKE) all
./$(TARGET)

94
x11/xHandler.c Normal file
View File

@ -0,0 +1,94 @@
#include <X11/Xlib.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <X11/keysym.h>
#include <stdbool.h>
int WIDTH = 1920;
int HEIGHT = 1080;
void hex_color(XColor *color, Display *display, Colormap cm, const char *hexcode) {
XParseColor(display, cm, hexcode, color);
XAllocColor(display, cm, color);
}
void fill_color(Display *display, Window window, GC gc, Colormap cm, const char *hexcode) {
XColor color;
hex_color(&color, display, cm, hexcode);
XSetForeground(display, gc, color.pixel);
XFillRectangle(display, window, gc, 0, 0, WIDTH, HEIGHT);
}
void fill_black(Display *display, Window window, GC gc, Colormap cm) {
fill_color(display, window, gc, cm, "#000000");
}
void fill_white(Display *display, Window window, GC gc, Colormap cm) {
fill_color(display, window, gc, cm, "#FFFFFF");
}
int main(void) {
Display *display;
Window window;
int screen;
XEvent event;
display = XOpenDisplay(NULL);
if (display == NULL) {
fprintf(stderr, "Cannot open display\n");
exit(1);
}
screen = DefaultScreen(display);
window = XCreateSimpleWindow(display, RootWindow(display, screen), 100, 100,
WIDTH, HEIGHT, 1, BlackPixel(display, screen),
WhitePixel(display, screen));
XSelectInput(display, window, ExposureMask | KeyPressMask);
XMapWindow(display, window);
GC gc = XCreateGC(display, window, 0, NULL);
Colormap cm = DefaultColormap(display, screen);
XColor black;
hex_color(&black, display, cm, "#000000");
XColor white;
hex_color(&white, display, cm, "#FFFFFF");
while (1) {
XNextEvent(display, &event);
if (event.type == Expose) {
// loop forever (until q)
static int is_black = false;
while (1) {
// set color to black
XSetForeground(display, gc, black.pixel);
for (int i = 0; i < 50; ++i) {
int x = (WIDTH / 2) + (i % 10) * 5;
int y = (HEIGHT / 2) + (i / 10) * 5;
XDrawRectangle(display, window, gc, x, y, 1, 1);
}
XFlush(display);
sleep(2);
while (XPending(display)) {
XNextEvent(display, &event);
if (event.type == KeyPress) {
KeySym key = XLookupKeysym(&event.xkey, 0);
if (key == XK_q || key == XK_Q) {
printf("Stopping...\n");
exit(0);
}
}
}
}
}
}
XFreeGC(display, gc);
XDestroyWindow(display, window);
XCloseDisplay(display);
return 0;
}