rust_type_utils.cpp

Go to the documentation of this file.

/***************************************
        Auteur : Pierre Aubert
        Mail : pierre.aubert@lapp.in2p3.fr
        Licence : LGPL-3.0-only
****************************************/
 
#include "type_utils.h"
#include "rust_type_utils.h"


PString getRustSimpleType(const PString & varType){
        bool isUnsigned = varType.isSameBegining("unsigned");
        PString restVarName(varType.replace("unsigned", "").eraseChar(" \n\t*&"));
        if(restVarName == "char"){return isUnsigned ? "u8" : "i8";}
        if(restVarName == "short"){return isUnsigned ? "u16" : "i16";}
        if(restVarName == "int"){return isUnsigned ? "u32" : "i32";}
        if(restVarName == "long" || restVarName == "longint"){return isUnsigned ? "u64" : "i64";}
        if(restVarName == "size_t"){return "usize";}
        if(restVarName == "ssize_t"){return "isize";}
        if(restVarName == "float"){return "f32";}
        if(restVarName == "double"){return "f64";}
        if(restVarName == "bool"){return "bool";}
        if(restVarName == "void"){return "()";}
        return "";
}


PString getRustType(const PString & varType){
        if(getIsPointerType(varType)){
                PString pointeeType(varType.eraseFirstLastChar(" \n\t*"));
                return "Option<Box<" + getRustType(pointeeType) + ">>";
        }
        if(generator_typeIsList(varType)){
                return "Vec<" + getRustType(generator_getListNestedType(varType)) + ">";
        }
        PString simpleType(getRustSimpleType(varType));
        if(simpleType != ""){return simpleType;}
        if(varType == "PString" || varType == "std::string" || varType == "PPath"){return "String";}
        //Nested generated struct or enum : the pdata type name is reused as-is for the Rust type name
        return varType;
}


bool getRustTypeIsCopy(const PString & varType){
        if(getIsPointerType(varType) || generator_typeIsList(varType)){return false;}
        if(varType == "PString" || varType == "std::string" || varType == "PPath"){return false;}
        return getRustSimpleType(varType) != "";
}


PString getRustAttributeDefaultValue(const PClassAttribute & attr){
        PString varType(attr.getType());
        bool isStringType(varType == "PString" || varType == "std::string" || varType == "PPath");
        PString simpleType(getRustSimpleType(varType));
        if(attr.getDefaultValue() != ""){
                //pdata default values are written between braces (eg "{0}", "{\"Shadoko\"}")
                PString rawValue(attr.getDefaultValue().eraseFirstLastChar("{} \n\t"));
                if(isStringType){
                        //The pdata default is a double-quoted &str literal : String has no implicit conversion from &str, unlike C++'s std::string
                        return rawValue + ".to_string()";
                }
                if((simpleType == "f32" || simpleType == "f64") && !rawValue.find(".eE")){
                        //Rust has no implicit int->float coercion, unlike the C++ NSDMI this default value was written for
                        return rawValue + ".0";
                }
                return rawValue;
        }
        if(getIsPointerType(varType)){return "None";}
        if(generator_typeIsList(varType)){return "Vec::new()";}
        if(isStringType){return "String::new()";}
        if(simpleType == "f32" || simpleType == "f64"){return "0.0";}
        if(simpleType == "bool"){return "false";}
        if(simpleType != ""){return "0";}
        //Nested generated struct or enum : rely on its own Default implementation
        return "Default::default()";
}


PString getRustTestValue(const PClassAttribute & attr){
        PString varType(attr.getType());
        if(varType == "PString" || varType == "std::string" || varType == "PPath"){return "\"Some string\".to_string()";}
        PString simpleType(getRustSimpleType(varType));
        if(simpleType == "bool"){return "true";}
        if(simpleType == "f32" || simpleType == "f64"){return "1.0";}
        if(simpleType != ""){return "42";}
        return "";
}