akern-gkgoat-fork/ableos/src/kmain.rs

#![allow(clippy::empty_loop)]

// use std::println;

use alloc::format;

use shadeable::pixel_format::{from_vga_16, into_vga_16, new_rgba64};
use vga::{colors::Color16, writers::GraphicsWriter};

use crate::{
    graphics::VgaBuffer,
    relib::network::socket::{SimpleSock, SocketReturns},
    vga_e::{self, num_to_vga16, VGAE},
};

use log::*;

use {
    crate::{
        arch::{drivers::graphics::GraphicsBuffer, init, sloop},
        driver_traits::graphics::Graphics,
        experiments::{
            info::master,
            systeminfo::{KERNEL_VERSION, RELEASE_TYPE},
        },

        graphics::ScreenBuffer,
        // log::LOG_STATE,
        relib::math::rand::RAND_HANDLE,
        relib::network::socket::Socket,
        scheduler::{test_fn, Thread, ThreadList},
    },
    alloc::{
        string::{String, ToString},
        vec,
    },
    lazy_static::lazy_static,
};

#[no_mangle]
#[allow(unconditional_recursion)]
pub extern "C" fn stack_overflow() -> u8 {
    stack_overflow();
    // meme number
    69 // NOTE: Any specific reason for this number aside from memes?
}

lazy_static! {
    pub static ref KEY_BUFFER: spin::Mutex<String> = spin::Mutex::new("".to_string());
    pub static ref THREAD_LIST: spin::Mutex<ThreadList> = spin::Mutex::new(vec![]);
    pub static ref TICK: spin::Mutex<u64> = spin::Mutex::new(0);
}

#[no_mangle]
pub fn kernel_main() -> ! {
    init::init();

    log::set_max_level(LevelFilter::Trace);
    // crate::wasm::evaluate();

    let mut abcde = ScreenBuffer::new(640, 480);
    // abcde.clear();
    // abcde.copy_to_buffer();

    VGAE.lock().clear_screen(Color16::Black);

    trace!("length of screen buffer {}", abcde.buff.len());
    trace!("Screen size {:?}", abcde.size);

    for y in 0..480 {
        for x in 0..640 {
            let segment_x = x * 4 / 640;
            let segment_y = y * 4 / 480;
            let segment = segment_x + segment_y * 4;

            abcde.set_pixel(x, y, from_vga_16(num_to_vga16(segment as u8)));
        }
    }

    // abcde.copy_to_buffer();

    {
        let mut xyz = SimpleSock::new();

        xyz.peek();

        xyz.write(vec![0, 1, 2, 3]);

        let x = "simple 🧦".to_string().into_bytes();
        xyz.write(x);

        info!("{:?}", &xyz.read(4).unwrap());

        match &xyz.peek() {
            SocketReturns::ReadOk(strr) => {
                let out = String::from_utf8_lossy(strr);

                info!("{}", out);
            }
            SocketReturns::ReadIndexOutOfBounds => todo!(),
            SocketReturns::WriteOk => todo!(),
        }
    }

    let mut sock_print_id = SimpleSock::new();

    {
        sock_print_id.write(format!("a原b画cフdァeイfル 集").into());
        // sock_print_id.write(format!("simple sockets are based").into());
        // sock_print_id.write(format!("1....2....3....4....5....6....7....8....9").into());
    }
    // crate::relib::image::mono_bitmap::bruh();

    if true {
        let mut prev = None;
        for current in (*String::from_utf8_lossy(&sock_print_id.peek().unwrap())).chars() {
            vga_e::draw_char(prev, current, 0);

            prev = Some(current);
        }
    }

    if false {
        draw_filled_circle(400, 350, 99, Color16::Black);
        // let _ = shadeable::evaluate_shader(1);
    }

    // abcde.clear();
    // abcde.copy_to_buffer();

    // stack_overflow();

    sloop()
}

/// called by arch specific timers to tick up all kernel related functions
pub fn tick() {
    let mut data = TICK.lock();
    *data += 1;
    RAND_HANDLE.lock().seed_entropy_timer(*data);

    // println!("{}", *data);
    crate::kernel_state::KERNEL_STATE.lock().update_state();
}

pub fn key_entropy(key: u8) {
    RAND_HANDLE.lock().seed_entropy_keyboard(key);
}

pub fn test_threads() {
    let mut a_thread = Thread::new();

    a_thread.new_task(test_fn);
    a_thread.new_task(test_fn);
    THREAD_LIST.lock().push(a_thread);
}

fn draw_filled_circle(cx: i32, cy: i32, radius: usize, color: Color16) {
    let vga = VGAE.lock();

    let r = radius as i32 * 2;
    for y in 0..640 {
        for x in 0..480 {
            let dx = cx - x as i32 * 2 - 1;
            let dy = cy - y as i32 * 2 - 1;

            if dx * dx + dy * dy <= r * r {
                vga.set_pixel(x, y, color);
            };
        }
    }
}

pub fn log_version_data() {
    info!("{} v{}", RELEASE_TYPE, KERNEL_VERSION);
    info!(
        "Brand String: {:?}",
        master().unwrap().brand_string().unwrap()
    );
}