From aa315d516f3379c67000466c2b9c84ad50e25e23 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Pawe=C5=82=20K=C4=99dzia?= <Pawel.Kedzia@pwr.wroc.pl>
Date: Tue, 29 Nov 2011 10:27:33 +0100
Subject: [PATCH] Added method to set option for document reader.

---
 libcorpus2/io/docreader.cpp | 7 +++++++
 libcorpus2/io/docreader.h   | 7 +++++++
 swig/documentreader.i       | 2 ++
 3 files changed, 16 insertions(+)

diff --git a/libcorpus2/io/docreader.cpp b/libcorpus2/io/docreader.cpp
index 7d9052b..cccb625 100644
--- a/libcorpus2/io/docreader.cpp
+++ b/libcorpus2/io/docreader.cpp
@@ -57,4 +57,11 @@ namespace Corpus2 {
 		return document;
 	}
 
+	void DocumentReader::set_option(const std::string& option)
+	{
+		if (option == "autogen_sent_id") {
+			ccl_reader_->set_option("autogen_sent_id");
+		}
+	}
+
 } /* end ns Corpus2 */
diff --git a/libcorpus2/io/docreader.h b/libcorpus2/io/docreader.h
index d1b27a6..fe683c9 100644
--- a/libcorpus2/io/docreader.h
+++ b/libcorpus2/io/docreader.h
@@ -52,6 +52,13 @@ public:
 	 */
 	boost::shared_ptr<Document> read();
 
+	/**
+	 * Sets options for readers (relation reader and/or ccl reader).
+	 * Available options:
+	 *  - autogen_sent_id -- for automatically generation identifiers of sentences
+	 */
+	void set_option(const std::string& option);
+
 private:
 	/**
 	 * Makes CclReader and RelationReader for given paths to files.
diff --git a/swig/documentreader.i b/swig/documentreader.i
index df1a87d..69229e8 100644
--- a/swig/documentreader.i
+++ b/swig/documentreader.i
@@ -33,6 +33,8 @@ namespace Corpus2 {
     }
     boost::shared_ptr<Document> read();
 
+    void set_option(const std::string& option);
+
     /* --------------------------------------------------------------------- */
     ~DocumentReader();
   };
-- 
GitLab