RustTraitGetterSetter.cpp

Go to the documentation of this file.

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

RustTraitGetterSetter::RustTraitGetterSetter()
        :PAbstractRustTraitBackend()
{}

RustTraitGetterSetter::~RustTraitGetterSetter(){}


void RustTraitGetterSetter::implMethodDeclaration(std::ofstream & fs, const PClassConfig & classConfig, const GeneratorMode & mode) const{
        if(classConfig.getIsEnum()){return;}
        const PVecClassAttribute & vecAttr(classConfig.getListAttribute());
        for(PVecClassAttribute::const_iterator it(vecAttr.begin()); it != vecAttr.end(); ++it){
                PString varType(it->getType()), fieldName(it->getName().toSnakeCase());
                if(getIsPointerType(varType)){
                        PString pointeeType(getRustType(varType.eraseFirstLastChar(" \n\t*")));
                        if(mode.enableGetter){
                                fs << "\t///Gets the " << it->getName() << " of the " << classConfig.getName() << "\n";
                                fs << "\tpub fn get_" << fieldName << "(&self) -> Option<&" << pointeeType << ">{\n";
                                fs << "\t\tself." << fieldName << ".as_deref()\n";
                                fs << "\t}\n\n";
                                fs << "\t///Gets a mutable reference to the " << it->getName() << " of the " << classConfig.getName() << "\n";
                                fs << "\tpub fn get_" << fieldName << "_mut(&mut self) -> Option<&mut " << pointeeType << ">{\n";
                                fs << "\t\tself." << fieldName << ".as_deref_mut()\n";
                                fs << "\t}\n\n";
                        }
                        if(mode.enableSetter){
                                fs << "\t///Sets the " << it->getName() << " of the " << classConfig.getName() << "\n";
                                fs << "\tpub fn set_" << fieldName << "(&mut self, " << fieldName << ": " << pointeeType << "){\n";
                                fs << "\t\tself." << fieldName << " = Some(Box::new(" << fieldName << "));\n";
                                fs << "\t}\n\n";
                        }
                        continue;
                }
                PString rustType(getRustType(varType));
                bool isCopy(getRustTypeIsCopy(varType));
                if(mode.enableGetter){
                        fs << "\t///Gets the " << it->getName() << " of the " << classConfig.getName() << "\n";
                        if(isCopy){
                                fs << "\tpub fn get_" << fieldName << "(&self) -> " << rustType << "{\n";
                                fs << "\t\tself." << fieldName << "\n";
                        }else{
                                fs << "\tpub fn get_" << fieldName << "(&self) -> &" << rustType << "{\n";
                                fs << "\t\t&self." << fieldName << "\n";
                        }
                        fs << "\t}\n\n";
                        if(!isCopy){
                                fs << "\t///Gets a mutable reference to the " << it->getName() << " of the " << classConfig.getName() << "\n";
                                fs << "\tpub fn get_" << fieldName << "_mut(&mut self) -> &mut " << rustType << "{\n";
                                fs << "\t\t&mut self." << fieldName << "\n";
                                fs << "\t}\n\n";
                        }
                }
                if(mode.enableSetter){
                        fs << "\t///Sets the " << it->getName() << " of the " << classConfig.getName() << "\n";
                        fs << "\tpub fn set_" << fieldName << "(&mut self, " << fieldName << ": " << rustType << "){\n";
                        fs << "\t\tself." << fieldName << " = " << fieldName << ";\n";
                        fs << "\t}\n\n";
                }
        }
}


void RustTraitGetterSetter::testFunction(std::ofstream & fs, const PClassConfig & classConfig, const GeneratorMode & mode) const{
        if(classConfig.getIsEnum() || !mode.enableUnitTest || (!mode.enableGetter && !mode.enableSetter)){return;}
        PString name(classConfig.getName());
        fs << "#[test]\n";
        fs << "fn test_" << name.toSnakeCase() << "_getter_setter(){\n";
        fs << "\tlet mut obj = " << name << "::default();\n";
        const PVecClassAttribute & vecAttr(classConfig.getListAttribute());
        for(PVecClassAttribute::const_iterator it(vecAttr.begin()); it != vecAttr.end(); ++it){
                PString varType(it->getType()), fieldName(it->getName().toSnakeCase());
                if(getIsPointerType(varType)){
                        PString pointeeType(getRustType(varType.eraseFirstLastChar(" \n\t*")));
                        fs << "\tlet next_" << fieldName << " = " << pointeeType << "::default();\n";
                        fs << "\tobj.set_" << fieldName << "(next_" << fieldName << ".clone());\n";
                        fs << "\tassert_eq!(*obj.get_" << fieldName << "().unwrap(), next_" << fieldName << ");\n";
                        continue;
                }
                PString testValue(getRustTestValue(*it));
                if(testValue == ""){continue;} //list/nested type : no generic literal test value, skip
                bool isCopy(getRustTypeIsCopy(varType));
                fs << "\tobj.set_" << fieldName << "(" << testValue << ");\n";
                if(isCopy){
                        fs << "\tassert_eq!(obj.get_" << fieldName << "(), " << testValue << ");\n";
                }else{
                        fs << "\tassert_eq!(*obj.get_" << fieldName << "(), " << testValue << ");\n";
                }
        }
        fs << "}\n\n";
}