#include "MixSegment.h"

namespace CppJieba
{
	MixSegment::MixSegment()
	{
	}
	
	MixSegment::~MixSegment()
	{
	}

	bool MixSegment::init(const char* const mpSegDict, const char* const hmmSegDict)
	{
        if(!_mpSeg.init(mpSegDict))
        {
            LogError("_mpSeg init");
            return false;
        }
        if(!_hmmSeg.init(hmmSegDict))
        {
            LogError("_hmmSeg init");
            return false;
        }
		return true;
	}
	
	bool MixSegment::dispose()
	{
        _mpSeg.dispose();
        _hmmSeg.dispose();
		return true;
	}

    bool MixSegment::cut(const string& str, vector<string>& res)
    {
        vector<TrieNodeInfo> infos;
        if(!_mpSeg.cut(str, infos))
        {
            LogError("_mpSeg cutDAG failed.");
            return false;
        }
        res.clear();
        Unicode unico;
        vector<Unicode> hmmRes;
        for(uint i= 0; i < infos.size(); i++)
        {
            if(1 == infos[i].word.size())
            {
                unico.push_back(infos[i].word[0]);
            }
            else
            {
                if(!unico.empty())
                {
                    if(!_hmmSeg.cut(unico, hmmRes))
                    {
                        LogError("_hmmSeg cut failed.");
                        return false;
                    }
                    for(uint j = 0; j < hmmRes.size(); j++)
                    {
                        res.push_back(TransCode::encode(hmmRes[j]));
                    }
                }
                unico.clear();

                res.push_back(TransCode::encode(infos[i].word));
            }
            
        }
        if(!unico.empty())
        {
            if(!_hmmSeg.cut(unico, hmmRes))
            {
                LogError("_hmmSeg cut failed.");
                return false;
            }
            for(uint j = 0; j < hmmRes.size(); j++)
            {
                res.push_back(TransCode::encode(hmmRes[j]));
            }
        }
        
        return true;
    }
}

#ifdef MIXSEGMENT_UT
using namespace CppJieba;

int main()
{
	return 0;
}

#endif