How to stringfy a swig matrix object in python

前端 未结 2 1843
走了就别回头了
走了就别回头了 2021-01-13 22:26

I am using swig wrapper of openbabel (written in C++, and supply a python wrapper through swig)

Below i just use it to read a molecule structure file and get the unit

2条回答
  •  情话喂你
    2021-01-13 23:29

    Further to the answer from @jhoon, it seems that SWIG doesn't recognise the std::string return type so change the function to return const char*. Also, since it is a function outside the class, you can't use self but you must use SWIG's $self variable.

    So, in the SWIG .i file, if you put the following:

    %extend OpenBabel::matrix3x3 {
      const char* __str__() {
        std::ostringstream out;
        out << *$self;
        return out.str().c_str();
      }
    };
    

    you should get the desired result when calling Python's print on a matrix3x3.

    If you find yourself adding this to many classes, consider wrapping it in a macro like:

    %define __STR__()
    const char* __str__() {
      std::ostringstream out;
      out << *$self;
      return out.str().c_str();
    }
    %enddef
    

    and then adding it to the class with:

    %extend OpenBabel::matrix3x3 {
      __STR__()
    };
    

提交回复
热议问题