Fix end-of-stringstream logic to not miss the last line.

[libsub.git] / src / reader.h

diff --git a/src/reader.h b/src/reader.h
index 8aa9fd891443ec26dd1d88186da157220062a4f2..08474f64d0a42201ba09816a70817603c635b613 100644 (file)
--- a/src/reader.h
+++ b/src/reader.h
@@ -20,17 +20,24 @@
 #ifndef LIBSUB_READER_H
 #define LIBSUB_READER_H
 
-#include "subtitle.h"
+#include "raw_subtitle.h"
 #include <list>
 #include <map>
 #include <string>
 
+struct subrip_reader_convert_line_test;
+
 namespace sub {
 
+/** @class Reader
+ *  @brief Parent for classes which can read particular subtitle formats.
+ */
 class Reader
 {
 public:
-       std::list<Subtitle> subtitles () const {
+       virtual ~Reader () {}
+
+       std::list<RawSubtitle> subtitles () const {
                return _subs;
        }
 
@@ -39,9 +46,11 @@ public:
        }
 
 protected:
+       friend struct ::subrip_reader_convert_line_test;
+
        void warn (std::string) const;
 
-       std::list<Subtitle> _subs;
+       std::list<RawSubtitle> _subs;
 };
 
 }