pub mod host;
pub mod manager;
pub mod output;

pub use manager::*;
use rayon::prelude::*;

type MaskMessage = (u64, u32, Vec<u64>);
type RowMessage = (u64, Vec<Vec<u32>>);

#[derive(Debug)]
pub enum Message {
    CheckRequest(CheckRequest),
    HostMessage(MaskMessage),
    OutputMessage(RowMessage),
    ResultMessage(ResultMessage),
    RowResult(RowResult),
    Terminate,
    CpuDone,
    GpuDone,
}

#[derive(Debug)]
pub struct ResultMessage {
    data: Vec<u64>,
    offset: usize,
    size: usize,
    wg_size: usize,
    id: u64,
}

impl ResultMessage {
    fn new(data: Vec<u64>, offset: usize, size: usize, wg_size: usize, id: u64) -> Self {
        Self {
            data,
            offset,
            size,
            wg_size,
            id,
        }
    }
    fn valid_walls(&self) -> Vec<Vec<u32>> {
        let mut result = vec![Vec::new(); self.wg_size];
        let tup: Vec<(usize, u32)> = self
            .data
            .par_iter()
            .enumerate()
            .filter_map(|(j, r)| {
                if *r != 0 {
                    Some(
                        (0..64)
                            .filter_map(|b| {
                                if r & (1 << b) != 0 {
                                    let permutation = j / self.size + self.offset;
                                    let instruction = (j % self.size) * 64 + b;
                                    Some((instruction, permutation as u32))
                                } else {
                                    None
                                }
                            })
                            .collect(),
                    )
                } else {
                    None
                }
            })
            .flat_map(|a: Vec<_>| a.to_vec())
            .collect();
        for t in tup {
            result[t.0 as usize].push(t.1);
        }
        result
    }
}
use std::sync::mpsc::{RecvError, SendError};

pub struct GpuError {
    message: String,
}

impl<T> From<SendError<T>> for GpuError {
    fn from(error: SendError<T>) -> Self {
        Self {
            message: format!("Gpu error occured: {}", error),
        }
    }
}
impl From<RecvError> for GpuError {
    fn from(error: RecvError) -> Self {
        Self {
            message: format!("Gpu error occured: {}", error),
        }
    }
}
impl From<String> for GpuError {
    fn from(error: String) -> Self {
        Self {
            message: format!("Gpu error occured: {}", error),
        }
    }
}
impl From<ocl::Error> for GpuError {
    fn from(error: ocl::Error) -> Self {
        Self {
            message: format!("Gpu error occured: {}", error),
        }
    }
}
impl std::fmt::Display for GpuError {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        write!(f, "{}", self.message)
    }
}

#[derive(Debug)]
pub struct CheckRequest {
    rows: Vec<u32>,
    bitmask: u64,
    queue: u32,
}

impl CheckRequest {
    pub fn new(rows: Vec<u32>, bitmask: u64, queue: u32) -> Self {
        Self {
            rows,
            bitmask,
            queue,
        }
    }
}

#[derive(Clone, PartialEq, Eq, Hash, Debug)]
pub struct RowResult {
    rows: Vec<u32>,
}

impl RowResult {
    fn new(mut rows: Vec<u32>) -> Self {
        //rows.push(0);
        rows.sort();
        Self { rows }
    }
    fn output(&self) {
        /*let s = self
            .rows
            .iter()
            .fold(String::new(), |s, d| s + &d.to_string() + " ");
        println!("{}", s);*/
        let i = self.rows[0];
        for r in self.rows.iter() {
            println!("{}\t{}", i, r);
        }
    }
}