370 lines
14 KiB
Rust
370 lines
14 KiB
Rust
use {
|
||
laddertypes::{TypeDict, TypeTerm, parser::*, unparser::*, morphism::{Morphism, MorphismInstance}},
|
||
crate::morphism::{LdmcPrimCMorphism, LdmcMorphism}
|
||
};
|
||
|
||
/*
|
||
*/
|
||
pub fn get_c_repr_kind(kind: String) -> Option<String> {
|
||
match kind.as_str() {
|
||
"ℤ" => Some("uint64_t".into()),
|
||
_ => None
|
||
}
|
||
}
|
||
|
||
/* 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 if t == &dict.parse("native.Float").expect("parse") {
|
||
Some("float".into())
|
||
} else if t == &dict.parse("native.Double").expect("parse") {
|
||
Some("double".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(&"ValueTerminated".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 if args[0] == laddertypes::TypeTerm::TypeID(dict.get_typeid(&"MsbCont".into()).unwrap())
|
||
{
|
||
let c_type = get_c_repr_type(dict, args[1].clone(), false)?;
|
||
if skip_pointer {
|
||
Some(c_type)
|
||
} else {
|
||
Some(format!("{} *", c_type))
|
||
}
|
||
}
|
||
else {
|
||
None
|
||
}
|
||
}
|
||
|
||
_ => None
|
||
}
|
||
}
|
||
}
|
||
None => None
|
||
}
|
||
}
|
||
|
||
|
||
pub fn encode_type_to_symbol(dict: &mut impl TypeDict, t: &laddertypes::TypeTerm)-> String {
|
||
match t {
|
||
laddertypes::TypeTerm::Char(c) => {
|
||
match c {
|
||
'.' => format!("_dot_"),
|
||
_ =>
|
||
format!("{}", (*c as u64))
|
||
}
|
||
},
|
||
laddertypes::TypeTerm::Num(n) => {
|
||
format!("{}", n)
|
||
}
|
||
laddertypes::TypeTerm::TypeID(ty_id) => {
|
||
if let Some(name) = dict.get_typename(ty_id) {
|
||
format!("{}", name.replace(".", "_dot_"))
|
||
} else {
|
||
format!("")
|
||
}
|
||
}
|
||
laddertypes::TypeTerm::Ladder(rs) |
|
||
laddertypes::TypeTerm::App(rs) => {
|
||
let mut s = String::new();
|
||
for r in rs {
|
||
s.push('_');
|
||
s.push_str(&encode_type_to_symbol(dict, r));
|
||
}
|
||
s
|
||
}
|
||
}
|
||
}
|
||
|
||
pub fn encode_type_to_value(dict: &mut impl TypeDict, t: &laddertypes::TypeTerm) -> String {
|
||
dict.unparse(t)
|
||
}
|
||
|
||
|
||
pub fn generate_main(type_dict: &mut impl TypeDict, path: Vec<MorphismInstance<LdmcMorphism>>, src_type: TypeTerm, dst_type: TypeTerm) {
|
||
let mut i = 0;
|
||
|
||
/* todo: collect include files from morphism base */
|
||
println!(r#"
|
||
#include <stdio.h>
|
||
#include <unistd.h>
|
||
#include <string.h>
|
||
#include <stdint.h>
|
||
#include <stdbool.h>
|
||
"#);
|
||
|
||
let mut existing_instantiations = Vec::new();
|
||
for morph_inst in path.iter() {
|
||
match &morph_inst.m {
|
||
LdmcMorphism::Primitive( item_morph ) => {
|
||
let name = item_morph.instantiated_symbol_name(type_dict, &morph_inst.σ);
|
||
if ! existing_instantiations.contains(&name) {
|
||
if let Some(s) = item_morph.generate_instantiation(type_dict, &morph_inst.σ) {
|
||
println!("{}", s);
|
||
} else {
|
||
eprintln!("couldnt generate instance {}", name);
|
||
}
|
||
existing_instantiations.push( name );
|
||
}
|
||
}
|
||
LdmcMorphism::LengthPrefixMap { length_prefix_type, item_morph } => {
|
||
let name = item_morph.instantiated_symbol_name(type_dict, &morph_inst.σ);
|
||
if ! existing_instantiations.contains(&name) {
|
||
if let Some(s) = item_morph.generate_instantiation(type_dict, &morph_inst.σ) {
|
||
println!("{}", s);
|
||
} else {
|
||
eprintln!("couldnt generate instance {}", name);
|
||
}
|
||
|
||
existing_instantiations.push( name );
|
||
}
|
||
}
|
||
_ => {}
|
||
}
|
||
}
|
||
|
||
println!(r#"
|
||
int main() {{
|
||
uint8_t bufA[128];
|
||
uint8_t bufB[128];
|
||
|
||
memset(bufA, 0, sizeof(bufA));
|
||
memset(bufB, 0, sizeof(bufB));
|
||
|
||
char in_str[] = "read :: {} \n";
|
||
char out_str[]= "write:: {} \n";
|
||
write(2, in_str, strlen(in_str));
|
||
write(2, out_str, strlen(out_str));
|
||
|
||
int l = read(0, bufA, sizeof(bufA));
|
||
fprintf(stderr, "read %d bytes\n", l);
|
||
|
||
"#,
|
||
type_dict.unparse(&src_type).replace("\\", "\\\\"),
|
||
type_dict.unparse(&dst_type).replace("\\", "\\\\") );
|
||
|
||
for morph_inst in path.iter() {
|
||
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(type_dict, &morph_inst.σ, i);
|
||
i += 1;
|
||
}
|
||
|
||
let out_buf = if i%2==0 { "bufA" } else { "bufB" };
|
||
|
||
let is_string = false;
|
||
if let Ok((halo, σ)) = laddertypes::subtype_unify(
|
||
&dst_type,
|
||
&type_dict.parse("<Seq~<ValueTerminated 0> x86.UInt8>").unwrap()
|
||
) {
|
||
println!(r#"
|
||
printf("%s\n", {});"#, out_buf);
|
||
} else if let Ok((halo, σ)) = laddertypes::subtype_unify(
|
||
&dst_type,
|
||
&type_dict.parse("<Seq~<LengthPrefix x86.UInt64> x86.UInt8>").unwrap()
|
||
) {
|
||
println!(r#"
|
||
/* write output
|
||
*/
|
||
{{
|
||
struct LengthPrefixUInt8Array * buf = (void*){};
|
||
write(1, {}, sizeof(uint64_t) + buf->len);
|
||
}}"#, out_buf, out_buf);
|
||
} else {
|
||
println!(r#"
|
||
write(1, {}, {});"#,
|
||
out_buf,
|
||
"sizeof(bufA)"
|
||
);
|
||
}
|
||
|
||
|
||
println!(r#"
|
||
return 0;
|
||
}}
|
||
"#);
|
||
|
||
eprintln!("Success: generated C code");
|
||
}
|
||
|
||
impl LdmcPrimCMorphism {
|
||
pub fn instantiated_symbol_name(&self, dict: &mut impl TypeDict,
|
||
σ: &std::collections::HashMap<laddertypes::TypeID, laddertypes::TypeTerm>) -> String {
|
||
let mut s = self.symbol.clone();
|
||
for (k_id,v) in self.type_args.iter() {
|
||
if let Some(val_trm) = σ.get(k_id) {
|
||
if let laddertypes::TypeID::Var(var_id) = k_id {
|
||
let name = dict.get_varname(*var_id).unwrap();
|
||
let val_str = encode_type_to_symbol(dict, val_trm);
|
||
s.push_str(&format!("_{}_{}", name, val_str));
|
||
}
|
||
} else {
|
||
if let laddertypes::TypeID::Var(var_id) = k_id {
|
||
let k = dict.get_varname(*var_id).unwrap();
|
||
s.push_str(&format!("_{:?}_MISSING", k));
|
||
eprintln!("INCOMPLETE MORPHISM INSTANTIATION, missing {} ({})", k, var_id);
|
||
}
|
||
}
|
||
}
|
||
s
|
||
}
|
||
|
||
pub fn expected_c_type_signature(&self, dict: &mut impl TypeDict,
|
||
σ: &std::collections::HashMap<laddertypes::TypeID, laddertypes::TypeTerm>) -> String {
|
||
format!("int {} ({} const * restrict src, {} * restrict dst);",
|
||
self.instantiated_symbol_name(dict, σ),
|
||
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"))
|
||
}
|
||
|
||
pub fn generate_instantiation(&self, dict: &mut impl TypeDict, σ: &std::collections::HashMap<laddertypes::TypeID, laddertypes::TypeTerm>) -> Option<String> {
|
||
let mut s = String::new();
|
||
|
||
s.push_str(&format!(r#"
|
||
int {} ( {} const * restrict src, {} * restrict dst ) {{
|
||
"#,
|
||
self.instantiated_symbol_name(dict, &σ),
|
||
get_c_repr_type(dict, self.src_type.clone().apply_substitution(&|k| σ.get(k).cloned()).clone(), true).expect("cant get c-repr-type"),
|
||
get_c_repr_type(dict, self.dst_type.clone().apply_substitution(&|k| σ.get(k).cloned()).clone(), true).expect("cant get c-repr-type"),
|
||
));
|
||
for (ty_id, kind) in self.type_args.iter() {
|
||
if let Some(val) = σ.get(ty_id) {
|
||
s.push_str(&format!(" #define {} {}\n", dict.get_typename(&ty_id).unwrap(), encode_type_to_value(dict, val)));
|
||
}
|
||
}
|
||
|
||
s.push_str(&self.c_source);
|
||
|
||
for (ty_id, kind) in self.type_args.iter() {
|
||
s.push_str(&format!("\n #undef {}", dict.get_typename(&ty_id).unwrap()));
|
||
}
|
||
|
||
s.push_str(&format!(r#"
|
||
}}
|
||
"#));
|
||
Some(s)
|
||
}
|
||
}
|
||
|
||
impl LdmcMorphism {
|
||
pub fn generate_call(&self, dict: &mut impl TypeDict, σ: &std::collections::HashMap<laddertypes::TypeID, laddertypes::TypeTerm>, i: u64) {
|
||
match self {
|
||
LdmcMorphism::Primitive(prim_morph) => {
|
||
let src_c_type = get_c_repr_type(dict, prim_morph.src_type.clone().apply_substitution(&|k| σ.get(k).cloned()).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().apply_substitution(&|k| σ.get(k).cloned()).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.instantiated_symbol_name(dict, σ),
|
||
'}');
|
||
}
|
||
LdmcMorphism::LengthPrefixMap { length_prefix_type, item_morph } => {
|
||
let src_c_type = get_c_repr_type(dict, self.get_type().src_type.clone().apply_substitution(&|k| σ.get(k).cloned()).clone(), true).expect("cant get c-repr type for src type");
|
||
let dst_c_type = get_c_repr_type(dict, self.get_type().dst_type.clone().apply_substitution(&|k| σ.get(k).cloned()).clone(), 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",
|
||
("struct LengthPrefixUInt8Array", "struct LengthPrefixUInt8Array") => "length_prefix_array_map_8_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.instantiated_symbol_name(dict, σ),
|
||
'}');
|
||
}
|
||
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.instantiated_symbol_name(dict, σ),
|
||
'}');
|
||
}
|
||
}
|
||
}
|
||
}
|