ldmc/src/main.rs

use {
    ariadne::{Color, Label, Report, ReportKind, Source},
    chumsky::{
        prelude::*, text::*
    },
    laddertypes::{
        dict::TypeDict, parser::ParseLadderType, subtype_unify, unparser::UnparseLadderType, BimapTypeDict, Morphism, MorphismType
    },
    std::sync::{Arc, RwLock}
};

/* for a given ladder type `t`, get the corresponding C type
 */
pub fn get_c_repr_type(dict: &mut impl TypeDict, t: laddertypes::TypeTerm, skip_pointer: bool) -> Option<String> {
    let lnf = t.normalize().decurry().get_lnf_vec();

    match lnf.last() {
        Some(t) => {
            if t == &dict.parse("Byte").expect("parse")
            || t == &dict.parse("x86.UInt8").expect("parse")
            || t == &dict.parse("<StaticLength 8 Bit>").expect("parse")
            {
                Some("uint8_t".into())
            } else if t == &dict.parse("x86.UInt16").expect("parse") {
                Some("uint16_t".into())
            } else if t == &dict.parse("x86.UInt32").expect("parse") {
                Some("uint32_t".into())
            } else if t == &dict.parse("x86.UInt64").expect("parse") {
                Some("uint64_t".into())
            } else {
                match t {
                    laddertypes::TypeTerm::App(args) => {
                        if args[0] == laddertypes::TypeTerm::TypeID(dict.get_typeid(&"LengthPrefix".into()).unwrap())
                        {
                            let _length_type = args[1].clone();
                            let item_c_type : String = get_c_repr_type(dict, args[2].clone(), false)?;
                            match item_c_type.as_str() {
                                "uint8_t" => Some(format!("struct LengthPrefixUInt8Array")),
                                "uint16_t" => Some(format!("struct LengthPrefixUInt16Array")),
                                "uint32_t" => Some(format!("struct LengthPrefixUInt32Array")),
                                "uint64_t" => Some(format!("struct LengthPrefixUInt64Array")),
                                _ => None
                            }
                        }
                        else if args[0] == laddertypes::TypeTerm::TypeID(dict.get_typeid(&"ValueDelim".into()).unwrap())
                        {
                            let _delim_val = args[1].clone();
                            let c_type = get_c_repr_type(dict, args[2].clone(), false)?;
                            if skip_pointer {
                                Some(c_type)
                            } else {
                                Some(format!("{} *", c_type))
                            }
                        } else {
                            None
                        }
                    }
                    _ => None
                }
            }
        }
        None => None
    }
}

#[derive(Clone, Debug)]
struct LdmcPrimCMorphism {
    symbol: String,
    type_args: Vec<(String, String)>,
    src_type: laddertypes::TypeTerm,
    dst_type: laddertypes::TypeTerm,
    locations: Vec<String>
}

impl LdmcPrimCMorphism {
    pub fn expected_c_type_signature(&self, dict: &mut impl TypeDict) -> String {
        format!("int {} ({} const * restrict src, {} * restrict dst);",
            self.symbol,
            get_c_repr_type(dict, self.src_type.clone(), true).expect("cant get c-repr type for src type"),
            get_c_repr_type(dict, self.dst_type.clone(), true).expect("cant get c-repr type for dst type"))
    }
}

#[derive(Clone)]
enum LdmcMorphism {
    Primitive( LdmcPrimCMorphism ),
    LengthPrefixMap{
        length_prefix_type: laddertypes::TypeTerm,
        item_morph: Box<LdmcPrimCMorphism>
    },
    ValueDelimMap{
        delim: u64,
        item_morph: Box<LdmcPrimCMorphism>
    }
}

impl LdmcMorphism {
    pub fn generate_call(&self, dict: &mut impl TypeDict, i: u64) {
        match self {
            LdmcMorphism::Primitive(prim_morph) => {
                let src_c_type = get_c_repr_type(dict, prim_morph.src_type.clone(), true).expect("cant get c-repr type for src type");
                let dst_c_type = get_c_repr_type(dict, prim_morph.dst_type.clone(), true).expect("cant get c-repr type for dst type");

                let src_buf = if i%2 == 0 { "bufA" } else { "bufB" };
                let dst_buf = if i%2 == 0 { "bufB" } else { "bufA" };
                println!(r#"
    {}
            {} const * restrict src = (void*) {};
            {} * restrict dst = (void*) {};
            {} ( src, dst );
    {}"#,
                    '{',
                    src_c_type, src_buf,
                    dst_c_type, dst_buf,
                    prim_morph.symbol,
                    '}');
            }
            LdmcMorphism::LengthPrefixMap { length_prefix_type, item_morph } => {
                let src_c_type = get_c_repr_type(dict, self.get_type().src_type, true).expect("cant get c-repr type for src type");
                let dst_c_type = get_c_repr_type(dict, self.get_type().dst_type, true).expect("cant get c-repr type for dst type");

                let map_fn = match (src_c_type.as_str(), dst_c_type.as_str()) {
                    ("struct LengthPrefixUInt64Array", "struct LengthPrefixUInt64Array") => {
                        "length_prefix_array_map_64_to_64"
                    },
                    ("struct LengthPrefixUInt8Array", "struct LengthPrefixUInt64Array") => {
                        "length_prefix_array_map_8_to_64"
                    },
                    ("struct LengthPrefixUInt64Array", "struct LengthPrefixUInt8Array") => {
                        "length_prefix_array_map_64_to_8"
                    },
                    _ => {
                        "{{ ERROR: no map function implemented }}"
                    }
                };

                let src_buf = if i%2 == 0 { "bufA" } else { "bufB" };
                let dst_buf = if i%2 == 0 { "bufB" } else { "bufA" };
                println!(r#"
    {}
            {} const * restrict src = (void*) {};
            {} * restrict dst = (void*) {};
            {} ( {}, src, dst );
    {}"#,
            '{',
                    src_c_type, src_buf,
                    dst_c_type, dst_buf,
                    map_fn,
                    item_morph.symbol,
                    '}');
            }
            LdmcMorphism::ValueDelimMap { delim, item_morph } => {
                let src_c_type = get_c_repr_type(dict, item_morph.src_type.clone(), false).expect("cant get c-repr type for src type");
                let dst_c_type = get_c_repr_type(dict, item_morph.dst_type.clone(), false).expect("cant get c-repr type for dst type");

                let src_buf = if i%2 == 0 { "bufA" } else { "bufB" };
                let dst_buf = if i%2 == 0 { "bufB" } else { "bufA" };
                println!(r#"
    {}
            {} const * restrict src = (void*) {};
            {} * restrict dst = (void*) {};
            value_delim_array_map_8_to_64( {}, src, dst );
    {}"#,
            '{',
                    src_c_type, src_buf,
                    dst_c_type, dst_buf,
                    item_morph.symbol,
                    '}');
            }
        }
    }
}

impl Morphism for LdmcMorphism {
    fn weight(&self) -> u64 {
        1
    }

    fn get_type(&self) -> laddertypes::MorphismType {
        match self {
            LdmcMorphism::Primitive(prim_morph) =>
                laddertypes::MorphismType {
                    src_type: prim_morph.src_type.clone().normalize(),
                    dst_type: prim_morph.dst_type.clone().normalize()
                },
            LdmcMorphism::LengthPrefixMap{ length_prefix_type, item_morph } => {
                laddertypes::MorphismType {
                    src_type: laddertypes::TypeTerm::App(vec![ length_prefix_type.clone(), item_morph.src_type.clone() ]),
                    dst_type: laddertypes::TypeTerm::App(vec![ length_prefix_type.clone(), item_morph.dst_type.clone() ]),
                }
            },
            LdmcMorphism::ValueDelimMap{ delim, item_morph } => {
                let value_delim_type = laddertypes::TypeTerm::App(vec![]);
                laddertypes::MorphismType {
                    src_type: laddertypes::TypeTerm::App(vec![ value_delim_type.clone(), item_morph.src_type.clone() ]),
                    dst_type: laddertypes::TypeTerm::App(vec![ value_delim_type.clone(), item_morph.dst_type.clone() ]),
                }
            }
        }.normalize()
    }

    fn map_morphism(&self, seq_type: laddertypes::TypeTerm) -> Option< Self > {
        match self {
            LdmcMorphism::Primitive(prim) => {
                let item_morph = Box::new(prim.clone());
/*
                if seq_type == self.length_prefix_type {
                */
                    Some(LdmcMorphism::LengthPrefixMap{
                        length_prefix_type: seq_type,
                        item_morph,
                    })
                    /*
                } else if seq_type == self.value_delim_type {
                    Some(LdmcMorphism::ValueDelimMap { delim, item_morph })
                } else {
                    None
                }
                */
            }
            _ => None
        }
    }
}

/* morphism-base text format:
 *     NAME '(' [TYPE-ARG-NAME ':' KIND]  ')'
 *           SRC-TYPE
 *     '-->' DST-TYPE
 *     '@' [ LOCATION ':' ]
 */
fn parser(
    type_dict: Arc<RwLock< BimapTypeDict >>
) -> impl Parser<char, Vec<LdmcPrimCMorphism>, Error = Simple<char>> {

    ident().padded()
    .then(
        ident().padded()
        .then_ignore(just(':').padded())
        .then(none_of(",)").repeated().padded())
        .separated_by(just(',').padded())
        .delimited_by(just('('), just(')'))
    )
    .then_ignore(just('\n'))
    .then(
        take_until(just("-->").ignored())
        .then(take_until(just('@').ignored()))
    )
    .then(
        none_of(":\n").repeated().separated_by(just(':'))
    )
    .map(
        move |(((symbol, type_args), ((src_type, _), (dst_type, _))), locations)| {
            let mut type_dict = type_dict.write().unwrap();
            let type_args : Vec<_> = type_args.into_iter().map(|(v,k)| (v,k.into_iter().collect())).collect();
            for (var, kind) in type_args.iter() {
                type_dict.add_varname(var.clone());
            }

            let mut src_type = type_dict.parse(&src_type.iter().collect::<String>()).expect("couldnt parse src type");
            let mut dst_type = type_dict.parse(&dst_type.iter().collect::<String>()).expect("couldnt parse dst type");

            LdmcPrimCMorphism {
                symbol,
                type_args,
                src_type,
                dst_type,
                locations: locations.into_iter().map(|l| l.into_iter().collect()).collect()
            }
        })
        .separated_by(text::newline())
}

fn main() {
    let mut type_dict = Arc::new(RwLock::new(BimapTypeDict::new()));
    let mut morphism_base = laddertypes::MorphismBase::<LdmcMorphism>::new(vec![
        type_dict.parse("Seq~<ValueDelim '\\0'>").expect(""),
        type_dict.parse("Seq~<LengthPrefix x86.UInt64>").expect("")
    ]);

    let mut args = std::env::args().skip(1);
    let src_type_arg = args.next().expect("src type expected");
    let dst_type_arg = args.next().expect("dst type expected");

    for mb_path in args {
        let src = std::fs::read_to_string(mb_path).expect("read");
        let result = parser(type_dict.clone()).parse(src.clone());
        match result {
            Ok(morphisms) => {
                eprintln!("parse ok.");
                for m in morphisms {
                    morphism_base.add_morphism(LdmcMorphism::Primitive(m));
                }
            }
            Err(errs) => {
                errs.into_iter().for_each(|e| {
                        Report::build(ReportKind::Error, (), e.span().start)
                            .with_message(e.to_string())
                            .with_label(
                                Label::new(e.span())
                                    .with_message(e)
                                    .with_color(Color::Red),
                            )
                            .finish()
                            .print(Source::from(&src))
                            .unwrap()
                        });
            }
        }
    }



    let path = morphism_base.find_morphism_path(MorphismType {
        src_type: type_dict.parse( src_type_arg.as_str() ).expect(""),
        dst_type: type_dict.parse( dst_type_arg.as_str() ).expect(""),
    },
    &mut *type_dict.write().unwrap()
    );

    match path {
        Some(path) => {
            let mut i = 0;
            println!(r#"
#include <stdio.h>
#include <stdint.h>
#include <morphisms/length-prefix.h>
#include <morphisms/posint.h>

int main() {{
    uint8_t bufA[1024];
    uint8_t bufB[1024];

    scanf("%s", bufA);
                "#);
            for morph_inst in path {
                println!(r#"
    /* morph to {}

    ...with
    morph {}
     ---> {},
    subst σ = {{ {:?} }},
    halo  Ψ = {}
    */"#,
                    type_dict.unparse(&morph_inst.get_type().dst_type.param_normalize().decurry()),
                    type_dict.unparse(&morph_inst.m.get_type().src_type),
                    type_dict.unparse(&morph_inst.m.get_type().dst_type),
                    morph_inst.σ,
                    type_dict.unparse(&morph_inst.halo),
                );
                morph_inst.m.generate_call(&mut type_dict, i);
                i += 1;
            }

            let out_buf = if i%2==0 { "bufA" } else { "bufB" };
            println!(r#"
    printf("%s\n", {});

    return 0;
}}
            "#, out_buf);
        }
        None => {
            eprintln!("Error: could not find morphism path");
        }
    }
}