/Users/paul/Documents/phlogo/core/DataSelectorVerbs.cpp

/*
 *  DataSelectorVerbs.cpp
 *
 *  Author: Paul Hamilton
 *      Date: 13 Jun 2011
 *
 */

#include "Verbs.h"

#include "Interpreter.h"
#include "World.h"
#include "Exceptions.h"
#include "List.h"
#include "Thing.h"
#include "Array.h"

#include <boost/lexical_cast.hpp>
#include <set>

using namespace std;
using namespace boost;

namespace phlogo {

void Verbs::first(TokenStream *ts) {

        pThing thing;
        _itp->getNextThing(_world, ts, &thing);
        
        // if it's a word, then the first char, otherwise the first element.
        if (thing->isWord())
                _world->allocResult()->set(string(1, thing->getWord()[0]));
        else if (thing->isList())
                _world->setResult(thing->getList().get(0));
        else
                _world->setResult(thing->getArray().get(0));

}

void Verbs::last(TokenStream *ts) {

        pThing thing;
        _itp->getNextThing(_world, ts, &thing);
        
        // if it's a word, then the first char, otherwise the first element.
        if (thing->isWord()) {
                string s = thing->getWord();
                _world->allocResult()->set(string(1, s[s.length()-1]));
        }
        else if (thing->isList())
                _world->setResult(thing->getList().get(thing->getList().size()-1));
        else
                _world->setResult(thing->getArray().get(thing->getArray().size()-1));

}

void Verbs::firsts(TokenStream *ts) {

        pThing thing;
        _itp->getNextThing(_world, ts, &thing);
        
        if (!thing->isList())
                BOOST_THROW_EXCEPTION( no_list_in_thing_exception() );

        // get all the firsts of the elements of the lists as lists.
        List rlist = _world->allocResult()->allocList();
        for (tListArray::iterator i=thing->getList().begin(); i != thing->getList().end(); i++) {
                if ((*i)->isWord())
                        rlist.push_back(*i);
                else {
                        if ((*i)->getList().size() == 0)
                                BOOST_THROW_EXCEPTION( empty_list_exception() );
                        rlist.push_back((*i)->getList().get(0));
                }
        }

}

void Verbs::butfirst(TokenStream *ts) {

        pThing thing;
        _itp->getNextThing(_world, ts, &thing);
        
        // get all the firsts
        // if it's a word, then the first char, otherwise the first element.
        if (thing->isWord()) {
                string s = thing->getWord();
                _world->allocResult()->set(s.substr(1));
        }
        else if (thing->isList()) {
                List list = _world->allocResult()->allocList();
        List l1 = thing->getList();
                tListArray::iterator i = l1.begin();
                i++;
                for (; i != l1.end(); i++) {
            list.add(*i);               
        }
        }
        else {
                BOOST_THROW_EXCEPTION( no_list_in_thing_exception() );
        }

}

void Verbs::butlast(TokenStream *ts) {

        pThing thing;
        _itp->getNextThing(_world, ts, &thing);
        
        // if it's a word, then the first char, otherwise the first element.
        if (thing->isWord()) {
        
                string s = thing->getWord();
                _world->allocResult()->set(s.substr(0, s.length()-1));
                
        }
        else if (thing->isList()) {
        
                List list = _world->allocResult()->allocList();
                tListArray::iterator i = thing->getList().begin();
                for (; (i+1) != thing->getList().end(); i++)
                        list.add(*i);

        }
        else {
                BOOST_THROW_EXCEPTION( no_list_in_thing_exception() );
        }

}

void Verbs::butfirsts(TokenStream *ts) {

        pThing thing;
        _itp->getNextThing(_world, ts, &thing);
        
        if (!thing->isList())
                BOOST_THROW_EXCEPTION( no_list_in_thing_exception() );
                
        // get all the firsts of the elements of the lists as lists.
        List rlist = _world->allocResult()->allocList();
        for (tListArray::iterator i=thing->getList().begin(); i != thing->getList().end(); i++) {
                if (!(*i)->isList())
                        BOOST_THROW_EXCEPTION( no_list_exception() );
                if ((*i)->getList().size() == 0)
                        BOOST_THROW_EXCEPTION( empty_list_exception() );
                pThing thing(new Thing());
                List list = thing->allocList();
                tListArray::iterator j = (*i)->getList().begin();
                j++;
                for (; j != (*i)->getList().end(); j++)
                        list.add(*j);
                rlist.push_back(thing);
        }

}

void Verbs::item(TokenStream *ts) {

        long index = getLong(ts);

        pThing thing;
        _itp->getNextThing(_world, ts, &thing);
        if (thing->isWord()) {
                _world->allocResult()->set(thing->getWord().substr(index-1, 1));                
        }
        else if (thing->isList()) {
                _world->setResult(thing->getList().get(index-1));               
        }
        else if (thing->isArray()) {
                _world->setResult(thing->getArray().get(index-1));              
        }

}

void Verbs::mditem(TokenStream *ts) {

        pThing thing;
        _itp->getNextThing(_world, ts, &thing);
        
        if (!thing->isList())
                BOOST_THROW_EXCEPTION( no_list_in_thing_exception() );
        if (thing->getList().size() != 2)
                BOOST_THROW_EXCEPTION( incorrect_list_size_exception() );

        int x = boost::lexical_cast<int>(thing->getList().get(0)->getWord());
        int y = boost::lexical_cast<int>(thing->getList().get(1)->getWord());

        _itp->getNextThing(_world, ts, &thing);
        
        if ((int)thing->getList().size() < x)
                BOOST_THROW_EXCEPTION( bad_index_exception() );
        if ((int)thing->getList().get(x-1)->getList().size() < y)
                BOOST_THROW_EXCEPTION( bad_index_exception() );
        
        _world->setResult(thing->getList().get(x-1)->getList().get(y-1));               

}

void Verbs::pick(TokenStream *ts) {

        pThing thing;
        _itp->getNextThing(_world, ts, &thing);
        
        if (!thing->isList())
                BOOST_THROW_EXCEPTION( no_list_in_thing_exception() );

        _world->setResult(thing->getList().get(0));             

}

void Verbs::remove(TokenStream *ts) {

        string s = getWord(ts);

        pThing thing;
        _itp->getNextThing(_world, ts, &thing);
        if (!thing->isList())
                BOOST_THROW_EXCEPTION( no_list_in_thing_exception() );

        List list = _world->allocResult()->allocList();
        for (tListArray::iterator i=thing->getList().begin(); i != thing->getList().end(); i++) {
                if (!(*i)->isWord())
                        BOOST_THROW_EXCEPTION( no_word_in_thing_exception() );
                if ((*i)->getWord() != s)
                        list.add(*i);
        }

}

void Verbs::remdup(TokenStream *ts) {

        pThing thing;
        _itp->getNextThing(_world, ts, &thing);
        
        if (!thing->isList())
                BOOST_THROW_EXCEPTION( no_list_in_thing_exception() );

        set<string> sofar;
        List list = _world->allocResult()->allocList();
        for (tListArray::iterator i=thing->getList().begin(); i != thing->getList().end(); i++) {
                if (!(*i)->isWord())
                        BOOST_THROW_EXCEPTION( no_word_in_thing_exception() );
                string s = (*i)->getWord();
                if (sofar.find(s) == sofar.end()) {
                        sofar.insert(s);
                        list.add(*i);
                }
        }

}

void Verbs::quoted(TokenStream *ts) {

        pThing thing;
        _itp->getNextThing(_world, ts, &thing);
        
        if (thing->isWord())
                _world->allocResult()->set("\"" + thing->getWord());
        else
                _world->setResult(thing);

}

}