#ifdef _MSC_VER
#pragma warning(disable:4786)
#endif

#include <algorithm>
#include <fstream>
#include <strstream>

#include "eoState.h"
#include "eoObject.h"
#include "eoPersistent.h"

using namespace std;

void removeComment(string& str, string comment)
{
    string::size_type pos = str.find(comment);

    if (pos != string::npos)
    {
        str.erase(pos, str.size());
    }
}

bool is_section(const string& str, string& name) 
{
    string::size_type pos = str.find("\\section{");

    if (pos == string::npos)
        return false;
    //else

    string::size_type end = str.find("}");

    if (end == string::npos)
        return false;
    // else

    name = str.substr(pos + 9, end-9);

    return true;
}
    
void eoState::registerObject(eoPersistent& registrant)
{
    string name = createObjectName(dynamic_cast<eoObject*>(&registrant));
 
    objectMap[name] = &registrant; 
}

void eoState::load(const string& _filename)
{
    ifstream is (_filename.c_str());

    if (is.fail())
    {
        string str = "Could not open file " + _filename;
        throw runtime_error(str);
    }
    
    string str;
    string name;

    getline(is, str);
    
    while(is)
    { // parse section header
        if (is_section(str, name))
        {
            string fullString;
            ObjectMap::iterator it = objectMap.find(name);

            if (it == objectMap.end())
            { // ignore
                while (getline(is, str))
                {
                    if (is_section(str, name))
                        break;
                }
            }
            else
            {
            
                eoPersistent* object = it->second;

                // now we have the object, get lines, remove comments etc.

                string fullstring;

                while (getline(is, str))
                {
                    if (is_section(str, name))
                        break;

                    removeComment(str, getCommentString());
                    fullstring += str + "\n";
                }

                istrstream the_stream(fullstring.c_str(), fullstring.size());

                object->readFrom(the_stream);
            }
        }
    }

}

void eoState::save(const string& filename)
{
    ofstream os(filename.c_str());

    for (ObjectMap::iterator it = objectMap.begin(); it != objectMap.end(); ++it)
    {
        os << "\\section{" << it->first << "}\n";
        it->second->printOn(os);
        os << '\n';
    }
}

string eoState::createObjectName(eoObject* obj)
{
    if (obj == 0)
    {
        ostrstream os;
        os << objectMap.size();
        return os.str();
    }
    // else
    
    string name = obj->className();
    ObjectMap::const_iterator it = objectMap.find(name);

    unsigned count = 1;
    while (it != objectMap.end())
    {
        ostrstream os;
        os << obj->className().c_str() << count++ << ends;

        name = os.str();

        it = objectMap.find(name);
    }

    return name;
}