/* Anakrino - .NET Reflection Browser
 * Copyright (C) 2001-2002  Jay Freeman (saurik)
*/

/*
 *        Redistribution and use in source and binary
 * forms, with or without modification, are permitted
 * provided that the following conditions are met:
 * 
 * 1. Redistributions of source code must retain the
 *    above copyright notice, this list of conditions
 *    and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the
 *    above copyright notice, this list of conditions
 *    and the following disclaimer in the documentation
 *    and/or other materials provided with the
 *    distribution.
 * 3. The name of the author may not be used to endorse
 *    or promote products derived from this software
 *    without specific prior written permission.
 * 
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS''
 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING,
 * BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE
 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
 * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
 * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR
 * TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
 * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/

#ifndef ANAKRINO_CALLGRAPHVIEWER_H
#define ANAKRINO_CALLGRAPHVIEWER_H

#if _MSC_VER >= 1000
#pragma once
#endif // _MSC_VER >= 1000

#include "anakrino/IViewer.h"
#include "anakrino/TreeBrowser.h"

#include "theoretic/IEngine.h"

namespace Anakrino {

class CallGraphViewer :
	public CWindowImpl<CallGraphViewer, CTreeViewCtrlEx>,
	public Diapexis::Unknown,
	public IViewer
{

	DIAPEXIS_OBJECT(CallGraphViewer)

	DIAPEXIS_QUERY_BEGIN(CallGraphViewer)
	DIAPEXIS_QUERY_MAP(Diapexis::Unknown)
	DIAPEXIS_QUERY_MAP(IViewer)
	DIAPEXIS_QUERY_END()

  protected:
	IBinder *binder;
	Theoretic::IEngine *engine;

  public:
	BOOL PreTranslateMessage(MSG* /*pMsg*/) {
		return FALSE;
	}

	BEGIN_MSG_MAP(CallGraphViewer)
		/*MSG_WM_CONTEXTMENU(OnContextMenu)
		REFLECTED_NOTIFY_CODE_HANDLER(NM_RCLICK, OnRClick)*/
		REFLECTED_NOTIFY_CODE_HANDLER(TVN_GETDISPINFO, OnGetDispInfo)
		REFLECTED_NOTIFY_CODE_HANDLER(TVN_ITEMEXPANDING, OnItemExpanding)
		/*REFLECTED_NOTIFY_CODE_HANDLER(TVN_ITEMEXPANDED, OnItemExpanded)
		REFLECTED_NOTIFY_CODE_HANDLER(TVN_SELCHANGED, OnSelChanged)*/
	END_MSG_MAP()

	virtual operator HWND() const { return m_hWnd; }

	HWND Initialize(IBinder *binder, HWND parent) {
		this->binder = binder;
		engine = Theoretic::IEngine::GetInstance(L"msil");

		Create(parent, rcDefault, NULL, WS_CHILD | WS_VISIBLE | WS_CLIPSIBLINGS | WS_CLIPCHILDREN | TVS_HASBUTTONS | TVS_HASLINES | TVS_LINESATROOT | TVS_SHOWSELALWAYS | TVS_TRACKSELECT, WS_EX_STATICEDGE);
		SetImageList(binder->GetImageList(), TVSIL_NORMAL);
		return m_hWnd;
	}

	void Destroy(CTreeItem node) {
        if (node == NULL) return;
		delete (Node *) node.GetData();
		for (CTreeItem child = node.GetChild(); child != NULL; child = child.GetNextSibling())
			Destroy(child);
	}

	void Destroy() {
        Destroy(GetRootItem());
	}

	bool Supports(const Metallurgy::Token *token) {
		if (token == NULL)
			return false;
		switch (token->getTokenType()) {
			case Chordata::Tokens::mdtMethodDef:
				return true;
			default:
				return false;
		}
	}

	wcstr_t GetName() {
		return L"Call Graph";
	}

	void Select(const Metallurgy::Token *token) {
        Destroy(GetRootItem());
        DeleteItem(GetRootItem());

		if (token != NULL)
			switch (token->getTokenType()) {
				case Chordata::Tokens::mdtMethodDef: {
                    const Metallurgy::Method *mMethod = dynamic_cast<const Metallurgy::Method *>(token);
	                uint32_t icon = binder->GetIcon(mMethod);
	                CTreeItem tMethod = InsertItem(LPSTR_TEXTCALLBACK, icon, icon, NULL, NULL);
	                tMethod.SetData(reinterpret_cast<uint32_t>(new Node(mMethod)));
	                tMethod.SetChildren(TRUE);
				} break;
			}
	}

	void TokenSort(HTREEITEM item) {
		TVSORTCB sort;
		sort.hParent = item;
        sort.lpfnCompare = TreeBrowser::TokenCompare;
		sort.lParam = (LPARAM) binder;
		SortChildrenCB(&sort);
	}

    LRESULT OnGetDispInfo(int /*idCtrl*/, LPNMHDR pnmh, BOOL& /*bHandled*/) {
	    LPNMTVDISPINFO lptvdi = (LPNMTVDISPINFO) pnmh;

	    CTreeItem tDisplay(lptvdi->item.hItem, this);
	    Node *node = reinterpret_cast<Node *>(tDisplay.GetData());

	    if (node == NULL || node->token == NULL)
            return FALSE;

	    static wchar_t buff[102400];
	    _snwprintf(buff, 102400, L"%s", binder->GetDisplayText(node->token).c_str());
	    lptvdi->item.pszText = buff;

		return TRUE;
    }

    LRESULT OnItemExpanding(int /*idCtrl*/, LPNMHDR pnmh, BOOL& /*bHandled*/) {
	    LPNMTREEVIEW pmtv = (LPNMTREEVIEW) pnmh;

	    if (pmtv->action == TVE_COLLAPSE)
		    return FALSE;

	    CTreeItem tExpanding(pmtv->itemNew.hItem, this);
	    Node *node = reinterpret_cast<Node *>(tExpanding.GetData());

	    if (node == NULL || node->state != Irrelevant)
		    return FALSE;

	    bool gotChild(false);
	    if (node->token == NULL) {
		    return FALSE;
	    } else {
		    switch (node->token->getTokenType()) {
			    case Chordata::Tokens::mdtMethodDef: {
                    const Metallurgy::Method *mMethod = dynamic_cast<const Metallurgy::Method *>(node->token);
                    const Metallurgy::Module *mModule = mMethod->getModule();

		            const Metallurgy::ILMethodImpl *impl = dynamic_cast<const Metallurgy::ILMethodImpl *>(mMethod->getImpl());
                    if (impl == NULL)
                        break;

		            uint8_t *data = impl->getOffset();
		            uint32_t length(impl->getLength());

                    Theoretic::OpGraph *graph = engine->Parse(impl->getData());

                    typedef std::set<const Metallurgy::Member *> MemberSet;
                    MemberSet members;

                    const Theoretic::OperSet &ops = graph->GetOps();
                    for (Theoretic::OperSet::const_iterator op = ops.begin(); op != ops.end(); ++op)
                        if ((*op)->GetName() == L"msil:call") {
                            double token = (*op)->operator [](L"method")->num();
                            members.insert(dynamic_cast<const Metallurgy::Member *>(mModule->ResolveToken(token)));
                        }

                    for (MemberSet::iterator member = members.begin(); member != members.end(); ++member, gotChild = true) {
	                    uint32_t icon = binder->GetIcon(*member);
	                    CTreeItem tCall = InsertItem(LPSTR_TEXTCALLBACK, icon, icon, tExpanding, NULL);
	                    tCall.SetData(reinterpret_cast<uint32_t>(new Node(*member)));
                        if ((*member)->getTokenType() == Chordata::Tokens::mdtMethodDef)
                            tCall.SetChildren(TRUE);
                    }

                    delete graph;
                } break;
            }
	    }

	    if (gotChild) {
		    TokenSort(tExpanding);
		    node->state = Stretched;
		    return FALSE;
	    } else {
		    tExpanding.SetChildren(FALSE);
		    node->state = Shrunk;
		    return TRUE;
	    }
    }
};

}

#endif//ANAKRINO_CALLGRAPHVIEWER_H