diff --git a/libcorpus2/io/docreader.cpp b/libcorpus2/io/docreader.cpp index 7d9052bb372b8a9ab87ede0a16962e0bb719841e..cccb62583ecb230d48d5c0cf1cc2e30ba747b2d5 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 d1b27a6537acbe54f90ab2c76f84508a639e2cad..fe683c9cfd44b981b0d0a8b0c5a47a92114af9c6 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 df1a87d5dff263fce45240072caa2627584491ce..69229e8265c99d60496f8997b66fa2a16d9b62ba 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(); };