/*==LICENSE==*

CyanWorlds.com Engine - MMOG client, server and tools
Copyright (C) 2011  Cyan Worlds, Inc.

This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program.  If not, see <http://www.gnu.org/licenses/>.

You can contact Cyan Worlds, Inc. by email legal@cyan.com
 or by snail mail at:
      Cyan Worlds, Inc.
      14617 N Newport Hwy
      Mead, WA   99021

*==LICENSE==*/
#ifndef plPickNodeBase_h_inc
#define plPickNodeBase_h_inc

#include "max.h"
#include <vector>

class IParamBlock2;
class plMaxNode;
class plComponentBase;

class plPickNodeBase
{
protected:
    IParamBlock2* fPB;
    int fNodeParamID;

    static BOOL CALLBACK IDlgProc(HWND hDlg, UINT msg, WPARAM wParam, LPARAM lParam);

    void IInitDlg(HWND hDlg);

    void IGetNodesRecur(plMaxNode* node, HWND hList, plMaxNode* curSelNode);

    void ISetNodeValue(plMaxNode* node);
    void IGetSelNode(HWND hList);
    
    virtual bool ICheckNode(plMaxNode* node)=0;

    virtual void IAddUserType(HWND hList) {}
    virtual void ISetUserType(plMaxNode* node, const char* userType) {}

public:
    plPickNodeBase(IParamBlock2* pb, int nodeParamID);
    virtual ~plPickNodeBase() {}

    bool DoPick();
};

class plPickNode : public plPickNodeBase
{
protected:
    std::vector<Class_ID> fCIDs;
    bool fCanConvertToType;

    bool ICanConvertToType(Object *obj);
    virtual bool ICheckNode(plMaxNode* node);

public:
    plPickNode(IParamBlock2* pb, int nodeParamID);
};

//
// Pick one of the nodes a material is attached to
//
class plPickMtlNode : public plPickNodeBase
{
protected:
    Mtl* fMtl;

    bool ICheckNode(plMaxNode* node);

public:
    plPickMtlNode(IParamBlock2* pb, int nodeParamID, Mtl* mtl);
};

//
// Pick one of the nodes a component is attached to
//
class plPickCompNode : public plPickNodeBase
{
protected:
    plComponentBase* fComp;

    bool ICheckNode(plMaxNode* node);

public:
    plPickCompNode(IParamBlock2* pb, int nodeParamID, plComponentBase* comp);
};

#endif // plPickNodeBase_h_inc