MenuWrapper.cpp

#include "stdafx.h"
#include "souistd.h"
#include "MenuWrapper.h"
#include "helper\smenuex.h"

namespace SOUI
{

    // ------------------------------------------------------------------------------
    //
    // impl MenuItemWrapper
    //
    // ------------------------------------------------------------------------------

    MenuItemWrapper::MenuItemWrapper(pugi::xml_node node) : _menuItemNode(node)
    {
    }

    void MenuItemWrapper::SetText(const SStringW& str)
    {
        if (_menuItemNode)
        {
            _menuItemNode.text().set((LPCWSTR)str);
        }
    }

    pugi::xml_attribute MenuItemWrapper::AppendItemAttr(LPCWSTR lpszAttr)
    {
        pugi::xml_attribute attr;

        if (_menuItemNode)
        {
            attr = _menuItemNode.attribute(lpszAttr);
            if (!attr)
            {
                attr = _menuItemNode.append_attribute(lpszAttr);
            }
        }

        return attr;
    }

    void MenuItemWrapper::SetId(int id)
    {
        AppendItemAttr(L"id").set_value(id);
    }

    void MenuItemWrapper::SetEnable(BOOL bEnable)
    {
        AppendItemAttr(L"enable").set_value(bEnable);
    }

    void MenuItemWrapper::SetCheck(BOOL bCheck)
    {
        AppendItemAttr(L"check").set_value(bCheck);
    }

	void MenuItemWrapper::SetIcon(int nIcon)
	{
		AppendItemAttr(L"icon").set_value(nIcon);
	}

    void MenuItemWrapper::SetFont(const SStringW& font)
    {
        AppendItemAttr(L"font").set_value((LPCWSTR)font);
    }

    int  MenuItemWrapper::GetId()
    {
        return AppendItemAttr(L"id").as_int(-1);
    }

	bool MenuItemWrapper::GetCheck()
	{
		return AppendItemAttr(L"check").as_bool(-1);
	}

    // ------------------------------------------------------------------------------
    //
    // impl MenuWrapper
    //
    // ------------------------------------------------------------------------------

    MenuWrapper::MenuWrapper(const SStringW& xml, const SStringW& type)
    {
		SStringW strRes=xml + L":" + type;
        LOADXML(_menuDoc, strRes);
        _menuRoot = _menuDoc.child(L"menuRoot");
    }

    MenuWrapper::~MenuWrapper()
    {
    }

    MenuItemWrapper* MenuWrapper::AddMenu(const SStringW& str,
        int id,
        BOOL bEnable, /*=TRUE*/
        BOOL bCheck)  /*=FALSE*/
    {
        if (!_menuRoot)
        {
            return NULL;
        }

        pugi::xml_node node = _menuRoot.append_child(L"menuItem");

        node.append_attribute(L"colorTextDisable").set_value(L"#c0c0c0");

        MenuItemWrapper menuItem(node);
        menuItem.SetText(str);
        menuItem.SetId(id);
        menuItem.SetEnable(bEnable);
        menuItem.SetCheck(bCheck);
        menuItem.SetFont(L"size:12");

        _menuItems.push_back(menuItem);
        VecMenuItems::iterator it = _menuItems.end() - 1;

        return &(*it);
    }

	void MenuWrapper::AddSeperator()
	{
		if (!_menuRoot)
			return;

		pugi::xml_node node = _menuRoot.append_child(L"sep");
		MenuItemWrapper menuItem(node);
		_menuItems.push_back(menuItem);
		VecMenuItems::iterator it = _menuItems.end() - 1;
	}

    MenuItemWrapper* MenuWrapper::GetMenuItemById(int id)
    {
        VecMenuItems::iterator it = _menuItems.begin();

        for (; it != _menuItems.end(); ++it)
        {
            if (it->GetId() == id)
            {
                return &(*it);
            }
        }

        if (_menuRoot)
        {
            pugi::xml_node node = GetNodeById(_menuRoot.first_child(), id);

            if (node)
            {
                MenuItemWrapper menuItem(node);
                _menuItems.push_back(menuItem);

                it = _menuItems.end() - 1;
                return &(*it);
            }
        }

        return NULL;
    }

    void MenuWrapper::RemoveItem(int id)
    {
        VecMenuItems::iterator it = _menuItems.begin();

        for (; it != _menuItems.end(); ++it)
        {
            if (it->GetId() == id)
            {
                _menuItems.erase(it);
                break;
            }
        }

        pugi::xml_node node = GetNodeById(_menuRoot.first_child(), id);
        if (node)
        {
            _menuRoot.remove_child(node);
        }
    }

    void MenuWrapper::SetItemSkin(const SStringW& skin)
    {

    }

    void MenuWrapper::SetCheckSkin(const SStringW& skin)
    {

    }

    int MenuWrapper::ShowMenu(int op, int x, int y, HWND hHost)
    {
        SMenuEx menu;
        menu.LoadMenu(_menuRoot);
        return menu.TrackPopupMenu(op, x, y, hHost);
    }

    pugi::xml_node MenuWrapper::GetNodeById(pugi::xml_node node, int id)
    {
        if (!node)
        {
            return pugi::xml_node();
        }

        if (node.attribute(L"id").as_int(-1) == id)
        {
            return node;
        }

        pugi::xml_node matched;
        pugi::xml_node sibling;

        matched = GetNodeById(node.next_sibling(), id);
        if (matched)
        {
            return matched;
        }

        return GetNodeById(node.child(L"menuItem"), id);
    }

	void MenuWrapper::AddChildMenu(const SStringW& str,
		int id,
		BOOL bEnable, /*=TRUE*/
		BOOL bCheck)  /*=FALSE*/
	{
		if (!_menuRoot)
			return;
		pugi::xml_node parentNode = _menuRoot.last_child();
		if (!parentNode)
			return;
		pugi::xml_node node = parentNode.append_child(L"menuItem");

		node.append_attribute(L"colorTextDisable").set_value(L"#c0c0c0");

		MenuItemWrapper menuItem(node);
		menuItem.SetText(str);
		menuItem.SetId(id);
		menuItem.SetEnable(bEnable);
		menuItem.SetCheck(bCheck);
		menuItem.SetFont(L"size:12");

		//MenuItemWrapper menuParentItem(parentNode);
		//int nCount = map_menuChildItems.count(menuParentItem);
		//if (0 == nCount)
		//{
		//	VecMenuItems menuChildItems;
		//	menuChildItems.push_back(menuItem);
		//	map_menuChildItems.insert(std::make_pair(menuParentItem, menuChildItems));
		//}
		//else
		//{
		//	map_menuChildItems[menuParentItem].push_back(menuItem);
		//}
		
		/*std::map<MenuItemWrapper, VecMenuItems>::iterator it = map_menuChildItems.end() - 1;
		int count = it->second.size();
		return &(it->second[count]);*/
	}

};