Skip to content
Snippets Groups Projects
mwereader.cpp 1.54 KiB
Newer Older
Bartosz Broda's avatar
Bartosz Broda committed
#include "mwereader.h"
#include <boost/algorithm/string.hpp>
Bartosz Broda's avatar
Bartosz Broda committed

namespace Corpus2{

bool MWEReader::registered = TokenReader::register_path_reader<MWEReader>(
Bartosz Broda's avatar
Bartosz Broda committed
	"mwereader","inner,mwepath"); // TODO more help?
Bartosz Broda's avatar
Bartosz Broda committed

	MWEReader::MWEReader(const Tagset &tagset, const std::string &filename)
		: TokenReader(tagset), inner_filename_(filename)
Bartosz Broda's avatar
Bartosz Broda committed
	{
		// TODO implementataion
	}

	MWEReader::~MWEReader()
	{
		// TODO implementataion
	}

	Token* MWEReader::get_next_token()
	{
		// TODO MWE stuff
Bartosz Broda's avatar
Bartosz Broda committed
		// get whole sentence -> process it -> return token by token
		return inner_reader_->get_next_token();
Bartosz Broda's avatar
Bartosz Broda committed
	}

	Sentence::Ptr MWEReader::get_next_sentence()
	{
		// TODO MWE stuff
		return inner_reader_->get_next_sentence();
Bartosz Broda's avatar
Bartosz Broda committed
	}

	boost::shared_ptr<Chunk> MWEReader::get_next_chunk()
	{
		// TODO MWE stuff
Bartosz Broda's avatar
Bartosz Broda committed
		// get whole chunk -> process sentences -> return processed chunk
		return inner_reader_->get_next_chunk();
Bartosz Broda's avatar
Bartosz Broda committed
	}

	void MWEReader::set_option(const std::string& option)
	{
Bartosz Broda's avatar
Bartosz Broda committed
		if(boost::algorithm::starts_with(option, "inner:")) {
			std::string inner = option.substr(6);
			inner_reader_ = create_path_reader(inner, this->tagset(),
											   inner_filename_);
		}

Bartosz Broda's avatar
Bartosz Broda committed

		// TODO MWE stuff
Bartosz Broda's avatar
Bartosz Broda committed
	}

	void MWEReader::validate()
	{
		if(inner_reader_ == NULL)
			throw Corpus2Error("Inner reader not initialised.");
		// TODO MWE stuff
Bartosz Broda's avatar
Bartosz Broda committed
	}

	std::string MWEReader::get_option(const std::string& option) const
	{
		if(boost::algorithm::starts_with(option, "inner:")
			&& inner_reader_ != NULL)
			return option;
		// TODO options for MWE
		return inner_reader_->get_option(option);
Bartosz Broda's avatar
Bartosz Broda committed
	}


}// ns Corpus2