NanobindTraitProperty.cpp

Go to the documentation of this file.

/***************************************
        Auteur : Thibaut Oprinsen, Pierre Aubert
        Mail : thibaut.oprinsen@lapp.in2p3.fr, pierre.aubert@lapp.in2p3.fr
        Licence : CeCILL-C
****************************************/
 
#include "NanobindTraitProperty.h"

NanobindTraitProperty::NanobindTraitProperty()
: PAbstractNanobindTraitBackend()
{}

NanobindTraitProperty::~NanobindTraitProperty(){}


PString createGetterMethodName(const PString & varName){
        return "get" + varName.firstToUpper();
}


PString createSetterMethodName(const PString & varName){
        return "set" + varName.firstToUpper();
}

void NanobindTraitProperty::registerProperty(std::ofstream & fs, const PClassConfig & classConfig, const GeneratorMode & mode) const{
        const PVecClassAttribute & listAttr(classConfig.getListAttribute());
        if(classConfig.getIsEnum()){
                PString enumName = classConfig.getName();
                for(PVecClassAttribute::const_iterator it(listAttr.begin()); it != listAttr.end(); ++it){
                        PString attributeValue(it->getName());
                        fs << "\t\t.value(\"" << it->getName() << "\"," << enumName << "::" << attributeValue << ")" << std::endl;
                }
                return;
        }
        if(!mode.enableSetter || !mode.enableGetter){return;}
        for(PVecClassAttribute::const_iterator it(listAttr.begin()); it != listAttr.end(); ++it){
                fs << "\t\t.def_prop_rw(\"" << it->getName() << "\",";
                fs << std::endl << "\t\t\t[]( " << classConfig.getName() << " &c) { return c." << createGetterMethodName(it->getName()) << "(); },";
                fs << std::endl << "\t\t\t[]( " << classConfig.getName() << " &c, " << makeVarType(it->getType(), true, false, false, it->getIsPointer()) << " value) { c." << createSetterMethodName(it->getName()) << "(value); })";
                if(it != listAttr.end() - 1){
                        fs << std::endl;
                }
        }
}

void NanobindTraitProperty::testFunction(std::ofstream & fs, const PClassConfig & classConfig, const GeneratorMode & mode, const PString & baseFileName) const{
        if(!mode.enableSetter || !mode.enableSetter || classConfig.getIsEnum()){return;}
        PString moduleName = baseFileName.toLower();
        PString className = classConfig.getName().toLower();
        fs << "#Test getters and setters of the " << classConfig.getName() << " property trait" << std::endl;
        fs << "def test_attributes_setters_getters():" << std::endl;
        fs << "\t\"\"\"Test setters and getters of attributes generated by Nanobind\"\"\"" << std::endl;
        fs << "\t" << "# Test default constructor" << std::endl;
        fs << "\t" << className << " = " << moduleName << "." << classConfig.getName() << "()" << std::endl;
        fs << "\t" << "# Test properties" << std::endl;
        const PVecClassAttribute & listAttr(classConfig.getListAttribute());
        for(PVecClassAttribute::const_iterator it(listAttr.begin()); it != listAttr.end(); ++it){
                PString defaultValue = getTestDefaultValueTypeInPython(it->getType());
                if(defaultValue != ""){
                        fs << "\t" << className << "." << it->getName() << " = " << defaultValue << std::endl;
                        fs << "\t" << "assert " << className << "." << it->getName() << " == " << defaultValue << std::endl;
                }
        }
        fs << std::endl;
}

void NanobindTraitProperty::testCallFunction(std::ofstream & fs, const PClassConfig & classConfig, const GeneratorMode & mode) const{
        if(!mode.enableSetter || !mode.enableSetter || classConfig.getIsEnum()){return;}
        
        fs << "\ttest_attributes_setters_getters()" << std::endl;
}