#include "tordelo.hpp"

using namespace std;

string Tordelo::elvalasztok = " ,.?!-";

Tordelo::Tordelo(const string& input)
               : s(input),
                 allapot(0),
                 szokar(false)
               {}

int Tordelo::darab() const {

    unsigned int i;
    int db = 0;

    allapot = 0;

    for (i = 0; i < s.length(); i++) {

        szokar = elvalasztok.find(s[i]) == string::npos;

        if ((allapot == 0) && szokar)
            allapot = 1;

        if ((allapot == 1) && !szokar) {
            allapot = 0;
            db++;
        }
    }

    if (allapot == 1) db++;

    return db;
}

string Tordelo::szo(int hanyadik) const {

    unsigned int i, j, kezd;

    allapot = 0;

    j = 0;

    for (i = 0; i < s.length(); i++) {

        szokar = elvalasztok.find(s[i]) == string::npos;

        if ((allapot == 0) && szokar) {
            allapot = 1;
            kezd = i;
        }

        if ((allapot == 1) && !szokar) {
            allapot = 0;
            if (j == hanyadik)
                return s.substr(kezd, i - kezd);
            j++;
        }
    }
    if ((allapot == 1) && (j == hanyadik))
        return s.substr(kezd, s.length() - kezd);

    return "";
}