Add an operator<< for the SequenceDistanceGraphPath

And disable size modification on the interface for the vector of sdgp
bioinfologics · Aug 8, 2019 · abcb596 · abcb596
1 parent d50790b
commit abcb596
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 0 deletions.
diff --git a/interfaces/SequenceDistanceGraph.i.in b/interfaces/SequenceDistanceGraph.i.in
@@ -239,6 +239,8 @@ __commit__ = "@GIT_COMMIT_HASH@"
 
 %template(vectorNode) std::vector<Node>;
 
+%ignore std::vector<SequenceDistanceGraphPath>::vector(size_type);
+%ignore std::vector<SequenceDistanceGraphPath>::resize;
 %template(vectorSequenceDistanceGraphPath) std::vector<SequenceDistanceGraphPath>;
 %extend std::vector<SequenceDistanceGraphPath> {
     const std::string __repr__() {

diff --git a/src/sdglib/graph/SequenceDistanceGraphPath.cc b/src/sdglib/graph/SequenceDistanceGraphPath.cc
@@ -153,3 +153,8 @@ bool SequenceDistanceGraphPath::is_unitig() {
     }
     return true;
 }
+
+std::ostream &operator<<(std::ostream &os, const SequenceDistanceGraphPath &sdgp) {
+    os << "SequenceDistanceGraphPath (" << sdgp.sg.name << "): " << sdgp.nodes.size() << " nodes";
+    return os;
+}
diff --git a/src/sdglib/graph/SequenceDistanceGraphPath.hpp b/src/sdglib/graph/SequenceDistanceGraphPath.hpp
@@ -21,6 +21,8 @@ class SequenceDistanceGraphPath {
 
     SequenceDistanceGraphPath& operator=(const SequenceDistanceGraphPath &other);
 
+    friend std::ostream& operator<<(std::ostream &os, const SequenceDistanceGraphPath &sdgp);
+
     std::string get_fasta_header(bool use_oldnames = false) const;
     std::string sequence() const;
     size_t get_sequence_size_fast() const;