NanobindTraitDataStream.cpp

Go to the documentation of this file.

/***************************************
        Auteur : Thibaut Oprinsen
        Mail : thibaut.oprinsen@lapp.in2p3.fr
        Licence : CeCILL-C
****************************************/
 
#include "NanobindTraitDataStream.h"

NanobindTraitDataStream::NanobindTraitDataStream()
: PAbstractNanobindTraitBackend()
{}

NanobindTraitDataStream::~NanobindTraitDataStream(){}

void registerDeserializationMethods(std::ofstream & fs, const PClassConfig & classConfig, const GeneratorMode & mode){
        PString name(classConfig.getName());
        fs << std::endl << std::endl << "\t\t///Deserialization method for " << name;
        fs << std::endl << "\t\t.def(\"from_bytes\", []( " << name << " & self, const nb::bytearray &b) -> bool {";
        fs << std::endl << "\t\t\tDataStreamIter iter((DataStreamIter)b.data());;";
        fs << std::endl << "\t\t\treturn data_message_load(iter, self);";
        fs << std::endl << "\t\t}, ";
        fs << std::endl << "\t\t\"Load a " << name << " from a bytearray\"";
        fs << std::endl << "\t\t)";
}

void registerSerializationMethods(std::ofstream & fs, const PClassConfig & classConfig, const GeneratorMode & mode){
        PString name(classConfig.getName());
        fs << std::endl << std::endl << "\t\t///Serialization method for " << name;
        fs << std::endl << "\t\t.def(\"to_bytes\", [](const " << name << " & self) -> nb::bytearray {";
        fs << std::endl << "\t\t\tsize_t size = data_size(self);";
        fs << std::endl << "\t\t\tnb::bytearray msg(NULL, size);";
        fs << std::endl << "\t\t\tDataStreamIter iter((DataStreamIter)msg.data());";
        fs << std::endl << "\t\t\tbool success = data_message_save(iter, self);";
        fs << std::endl << "\t\t\tif(success){";
        fs << std::endl << "\t\t\t\treturn msg;";
        fs << std::endl << "\t\t\t}else{";
        fs << std::endl << "\t\t\t\tthrow nb::buffer_error(\"Failed to serialize " << name << "\");";
        fs << std::endl << "\t\t\t}";
        fs << std::endl << "\t\t}, ";
        fs << std::endl << "\t\t\"Serialize the " << name << " to bytearray\"";
        fs << std::endl << "\t\t)";
}

void registerSizeMethod(std::ofstream & fs, const PClassConfig & classConfig, const GeneratorMode & mode){
        PString name(classConfig.getName());
        fs << std::endl << std::endl << "\t\t///Size method for " << name;
        fs << std::endl << "\t\t.def(\"get_size\", [](const " << name << " & self) -> size_t {";
        fs << std::endl << "\t\t\treturn data_size(self);";
        fs << std::endl << "\t\t}, ";
        fs << std::endl << "\t\t\"Get the size of the " << name << " in bytes\"";
        fs << std::endl << "\t\t)";
}

void NanobindTraitDataStream::registerMethod(std::ofstream & fs, const PClassConfig & classConfig, const GeneratorMode & mode) const{
        if(!mode.enableDataStream){return;}
        PString name(classConfig.getName());
 
        registerSerializationMethods(fs, classConfig, mode);
        registerDeserializationMethods(fs, classConfig, mode);
        registerSizeMethod(fs, classConfig, mode);
}

void NanobindTraitDataStream::testFunction(std::ofstream & fs, const PClassConfig & classConfig, const GeneratorMode & mode, const PString & baseFileName) const{
        if(!mode.enableDataStream){return;}
        PString name(classConfig.getName());
        PString moduleName = baseFileName.toLower() + "_module";
        fs << "def test_data_stream_conversion():" << std::endl;
        fs << "\t\"\"\"Test data stream conversion\"\"\"" << std::endl;
        fs << "\t" << name.toLower() << " = " << moduleName << "." << name << "()" << std::endl;
        fs << "\t# Set some example values" << std::endl;
        const PVecClassAttribute & listAttr(classConfig.getListAttribute());
        for(PVecClassAttribute::const_iterator it(listAttr.begin()); it != listAttr.end(); ++it){
                PString defaultValue = getTestDefaultValueTypeInPython(it->getType());
                fs << "\t" << name.toLower() << "." << it->getName() << " = " << defaultValue << std::endl;
        }
        fs << std::endl;
        fs << "\tdata_stream = " << name.toLower() << ".to_bytes()" << std::endl;
        fs << std::endl;
        fs << "\tnew_" << name.toLower() << " = " << moduleName << "." << name << "()" << std::endl;
        fs << "\tsuccess = new_" << name.toLower() << ".from_bytes(data_stream)" << std::endl;
        fs << std::endl;
        fs << "\tassert success" << std::endl;
        for(PVecClassAttribute::const_iterator it(listAttr.begin()); it != listAttr.end(); ++it){
                PString defaultValue = getTestDefaultValueTypeInPython(it->getType());
                fs << "\tassert new_" << name.toLower() << "." << it->getName() << " == " << defaultValue << std::endl;
        }
        fs << std::endl;
        fs << "def test_serialization_type_errors():" << std::endl;
        fs << "\t\"\"\"Test erro types for from_bytes\"\"\"" << std::endl;
        fs << std::endl;
        fs << "\t" << name.toLower() << " = " << moduleName << "." << name << "()" << std::endl;
        fs << "\ttry:" << std::endl;
        fs << "\t\tresult = " << name.toLower() << ".from_bytes(\"A string is not a bytearray !!!\")  # deserialize string instead of bytearray" << std::endl;
        fs << "\t\tassert False" << std::endl;
        fs << "\texcept TypeError:" << std::endl;
        fs << "\t\tpass" << std::endl;
        fs << std::endl;
        fs << "\ttry:" << std::endl;
        fs << "\t\tresult = " << name.toLower() << ".from_bytes(42)  #  deserialize instead of bytearray" << std::endl;
        fs << "\t\tassert False" << std::endl;
        fs << "\texcept TypeError:" << std::endl;
        fs << "\t\tpass" << std::endl;
        fs << std::endl;
        fs << "\ttry:" << std::endl;
        fs << "\t\tresult = " << name.toLower() << ".from_bytes([1, 2, 3, 4])  # deserialize list instead of bytearray" << std::endl;
        fs << "\t\tassert False" << std::endl;
        fs << "\texcept TypeError:" << std::endl;
        fs << "\t\tpass" << std::endl << std::endl;
}
        
void NanobindTraitDataStream::testCallFunction(std::ofstream & fs, const PClassConfig & classConfig, const GeneratorMode & mode) const{
        if(!mode.enableDataStream){return;}
        PString name(classConfig.getName());
        fs << "\ttest_data_stream_conversion()" << std::endl;
        fs << "\ttest_serialization_type_errors()" << std::endl;
}