import time
from PIL import Image, ImageDraw, ImageFont
import smbus as smbus
import os
import threading
import textwrap

class OLEDDisplay:
    def __init__(self, i2c_bus=1, i2c_address=0x3C):
        self.i2c_bus = i2c_bus
        self.i2c_address = i2c_address
        self.bus = smbus.SMBus(self.i2c_bus)
        
        self.width = 128
        self.height = 64
        self.pages = self.height // 8
        
        self.init_display()
        
    def init_display(self):
        self.command_list = [
            0xAE,  # Display off
            0xD5, 0x80,  # Set display clock divide ratio/oscillator frequency
            0xA8, 0x3F,  # Set multiplex ratio(1 to 64)
            0xD3, 0x00,  # Set display offset. 00 = no offset
            0x40,  # Set start line address
            0x8D, 0x14,  # Charge Pump Setting
            0x20, 0x00,  # Memory Mode
            0xA1,  # Set Segment Re-map. A0=address mapped; A1=address 127 mapped.
            0xC8,  # Set COM Output Scan Direction
            0xDA, 0x12,  # Set COM Pins hardware configuration
            0x81, 0xCF,  # Set contrast control register
            0xD9, 0xF1,  # Set pre-charge period
            0xDB, 0x40,  # Set VCOMH Deselect Level
            0xA4,  # Entire Display ON; A4=Follow RAM content; A5=Ignore RAM content
            0xA6,  # Set Normal Display. A6=Normal; A7=Inverse
            0x2E,  # Deactivate scroll
            0xAF   # Display ON
        ]
        
        for cmd in self.command_list:
            self.write_command(cmd)
    
    

    def write_command(self, cmd):
        self.bus.write_byte_data(self.i2c_address, 0x00, cmd)
    
    def write_data(self, data):
        for i in range(0, len(data), 32):
            self.bus.write_i2c_block_data(self.i2c_address, 0x40, data[i:i+32])
    
    def display_image(self, image_path):
        if not os.path.exists(image_path):
            print(f"Error: File '{image_path}' not found.")
            return
        
        image = Image.open(image_path).convert('1').resize((self.width, self.height), Image.LANCZOS)
        pixel_data = list(image.getdata())
        pages = [0x00] * (self.width * self.pages)
        
        for i in range(self.height):
            for j in range(self.width):
                if pixel_data[i * self.width + j]:
                    pages[j + (i // 8) * self.width] |= (1 << (i % 8))
        
        self.write_command(0x22)
        self.write_command(0x00)
        self.write_command(0x07)
        
        self.write_command(0x21)
        self.write_command(0x00)
        self.write_command(0x7F)
        
        for page in range(self.pages):
            self.write_command(0xB0 + page)
            self.write_command(0x00)
            self.write_command(0x10)
            self.write_data(pages[page * self.width:(page + 1) * self.width])
    
    def write_text(self, text):
        # Create an image with the same dimensions as the display
        image = Image.new('1', (self.width, self.height), 0)
        draw = ImageDraw.Draw(image)

        # Use a truetype font
        font_path = "/usr/share/fonts/truetype/dejavu/DejaVuSans-Bold.ttf"
        font_size = 10
        font = ImageFont.truetype(font_path, font_size)

        # Text wrapping
        max_width = self.width
        lines = textwrap.wrap(text, width=int(max_width / (font_size * 0.6)))

        # Draw the text
        y = 0
        #line_height = font.getbbox('A')[1]  # Get the height of a line
        for line in lines:
            draw.text((1, y), line, font=font, fill=255)
            y += (font_size+1)  # Move to the next line with proper spacing

        # Convert the image to pixel data
        pixel_data = list(image.getdata())
        pages = [0x00] * (self.width * self.pages)

        for i in range(self.height):
            for j in range(self.width):
                if pixel_data[i * self.width + j]:
                    pages[j + (i // 8) * self.width] |= (1 << (i % 8))

        self.write_command(0x22)
        self.write_command(0x00)
        self.write_command(0x07)

        self.write_command(0x21)
        self.write_command(0x00)
        self.write_command(0x7F)

        for page in range(self.pages):
            self.write_command(0xB0 + page)
            self.write_command(0x00)
            self.write_command(0x10)
            self.write_data(pages[page * self.width:(page + 1) * self.width])

    def display_text(self, text):
        with self.lock:
            self.write_text(text)

    def clear_display(self):
        self.write_command(0x22)
        self.write_command(0x00)
        self.write_command(0x07)
        
        self.write_command(0x21)
        self.write_command(0x00)
        self.write_command(0x7F)
        
        pages = [0x00] * (self.width * self.pages)
        for page in range(self.pages):
            self.write_command(0xB0 + page)
            self.write_command(0x00)
            self.write_command(0x10)
            self.write_data(pages[page * self.width:(page + 1) * self.width])
  
    def display_image_thread(self, image_path, duration):
        def display_image_thread_worker():
            self.display_image(image_path)
            time.sleep(duration)
            self.clear_display()
        
        with self.lock:
            thread = threading.Thread(target=display_image_thread_worker)
            thread.start()

    def close(self):
        self.bus.close()

# Example usage
if __name__ == "__main__":
    oled = OLEDDisplay(i2c_bus=1, i2c_address=0x3C)
    try:
        oled.clear_display()
        oled.display_image('/home/user/2023-2024-projectone-mct-Maciej-Mitura/backend/Geert.bmp')
        time.sleep(2)  # Display the image for 10 seconds
    finally:
        oled.clear_display()
        oled.close()