py/sequence.cpp

   1 #include <boost/python.hpp>
   2 #include <boost/iterator.hpp>
   3 #include <boost/python/return_internal_reference.hpp>
   4 #include <boost/python/return_by_value.hpp>
   5 #include <boost/python/return_value_policy.hpp>
   6 #include <boost/python/register_ptr_to_python.hpp>
   7 using namespace boost::python;
   8
   9 #include <boost/shared_ptr.hpp>
  10
  11 #include <string>
  12 #include "alg/glsequence.hpp"
  13
  14 PyObject* seq_to_string(const Sequence& s) {
  15   ::PyString_FromStringAndSize(s.data(), s.size());
  16 }
  17
  18 void export_sequence()
  19 {
  20   class_<Sequence>("Sequence")
  21     .def(init<std::string>())
  22     .def("__str__", &seq_to_string)
  23     .def("size", &Sequence::size)
  24     .def("__len__", &Sequence::size)
  25     .add_property("header", &Sequence::get_fasta_header, &Sequence::set_fasta_header)
  26     .add_property("species", &Sequence::get_species, &Sequence::set_species)
  27     .def("rcseq", &Sequence::rev_comp, return_value_policy<return_by_value>())
  28   ;
  29
  30   register_ptr_to_python< boost::shared_ptr<Sequence> >();
  31 }