root/trunk/FileTreeCtrl.h

#if !defined(AFX_TABTREECTRL_H__AC2374B6_5047_4F92_96F4_7D5DEAE9BDCE__INCLUDED_)
#define AFX_TABTREECTRL_H__AC2374B6_5047_4F92_96F4_7D5DEAE9BDCE__INCLUDED_

#if _MSC_VER > 1000
#pragma once
#endif // _MSC_VER > 1000
// FileTreeCtrl.h : header file
//

/////////////////////////////////////////////////////////////////////////////
// CFileTreeCtrl window

class CFileTreeCtrl : public CXTTreeCtrl
{
// Construction
public:
    CFileTreeCtrl();

// Attributes
protected:
    CImageList  m_imageList;
    CBitmap     m_bitmap;
    HTREEITEM rootItem;
    map<string,HTREEITEM> projectMap;
    ProjectSettings * project;
// Operations
public:

// Overrides
    // ClassWizard generated virtual function overrides
    //{{AFX_VIRTUAL(CFileTreeCtrl)
    protected:
    virtual BOOL PreCreateWindow(CREATESTRUCT& cs);
    //}}AFX_VIRTUAL
    public:

        HTREEITEM GetFirstProjectFolder()
        {
            

            vector<string> folders;
            Split(project->GetCache("ProjectFolders"),",",folders );

            if ( !folders.size() ) return NULL;
            return projectMap[folders[0] ];
            

        }

    string GetSelFolderPath(HTREEITEM hCurSel = NULL);

    bool FolderIsSelected()
    {
        HTREEITEM hCurSel = GetNextItem(TVI_ROOT, TVGN_CARET);
        return IsFolder(hCurSel);
        
    }

    bool ProjectIsSelected()
    {
        if ( !Globals::currentProject ) return false;

        string pname = Globals::currentProject->name;
        HTREEITEM selItem = GetNextItem(TVI_ROOT, TVGN_CARET) ;

        int image1 = -1, image2 = -1;
        this->GetItemImage(selItem,image1,image2);

        if ( image1 == 1 && GetSelTreeCtrlItem(this) != pname) return true;

        return false;
    }

    bool IsProject(HTREEITEM item)
    {

        int image1 = -1, image2 = -1;
        this->GetItemImage(item,image1,image2);
        
        if ( image1 == 1 ) return true;
        
        return false;

    }

    bool IsFolder(HTREEITEM item );


    void NewFolder(const string& name, bool useDefault = false)
    {
        
        HTREEITEM selItem = NULL;
        string pname = Globals::currentProject->name;

        if ( useDefault )
        {
            selItem = GetFirstProjectFolder();
        }
        else selItem = GetNextItem(TVI_ROOT, TVGN_CARET) ;

            string parentFolder = GetSelFolderPath(selItem );
            
            CString x = GetItemText(selItem);
            string folderName = x.GetBuffer(0);
        

            HTREEITEM parent = projectMap[folderName];
            HTREEITEM newFolder = InsertItem(name.c_str(),2,3,parent);
            string folder;

            if (parentFolder == "Work Desktop" || folder == pname) 
            {
                folder = name;
            }
            else
            {
                folder = parentFolder + "//" + name;
            }
            
            

            projectMap[name] = newFolder;
            Globals::currentProject->AddFolder(folder);
            
            Expand(parent,TVE_EXPAND);

        
        
    }

    void InitTree(vector<string>& folders)
    {

        string pname = Globals::currentProject->name;
        rootItem = InsertItem(pname.c_str(),1,1);
            
        
        for ( int i = 0;i < folders.size(); i++ )
        {
            CreateTreeItem(folders[i] );
            
        }
        
        projectMap[pname] = rootItem;
        Expand(rootItem,TVE_EXPAND);
    }

    void CreateTreeItem(const string& path)
    {
        vector<string> pathList;
        Split(path,"//",pathList);
        HTREEITEM lastItem = rootItem;
        
        for ( int i = 0; i < pathList.size(); i++)
        {
            string temp = pathList[i];
            
            HTREEITEM item = NULL;          
            
            if ( projectMap.find(temp ) != NULL ) item = projectMap[temp];
            
            if ( item)
            {
            
            }
            else
            {
                item = InsertItem(temp.c_str(),2,3,lastItem);
                projectMap[temp] = item;
            
            }
            lastItem = item;
        }
        
        
        
        
    }

    void StartProject(ProjectSettings* p)
    {
        project = p;
        DeleteAllItems();
        projectMap.clear(); 

        vector<string> folders;
        Split(project->GetCache("ProjectFolders"),",",folders );
        InitTree(folders);


        vector<string> projects;
        Split(project->GetCache("AddOnProjects"),",",projects);

        InitExtraProjects(projects);

        AddInitialFiles();


    }

    void InitExtraProjects(const vector<string> & p )
    {
        for ( int i = 0 ; i < p.size(); i++ )
        {
            string name = StripFileName(p[i]);
            StripExtension(name);
            InsertProject(name.c_str() );
        }
    }

    void AddInitialFiles()
    {
        vector<string> files;
        Split(project->GetCache("files"),",",files);


        for ( int i = 0;i < files.size(); i++)
        {
            string folder = project->GetCache(files[i] + "_folderForFile");
        //  ElephantMessageBox(folder.c_str());
        //  ElephantMessageBox(files[i].c_str());
            HTREEITEM fold = projectMap[StripFileName( folder ) ];
            HTREEITEM addedItem = InsertItem(StripFileName(files[i]) .c_str(),0,0,fold);
            this->SetItemData(addedItem,( DWORD) strdup(files[i].c_str() ));
        }

    }

    void AddFile(const string& path,bool create = false, bool useDefaultFolder = false);
    void RemoveFolder();
    
    void InsertProject(const string& pname )
    {
        this->InsertItem(pname.c_str(),1,1 );
        
    }
    void RemoveFile();
    
// Implementation
public:
    virtual ~CFileTreeCtrl();

    // Generated message map functions
protected:
    //{{AFX_MSG(CFileTreeCtrl)
    afx_msg int OnCreate(LPCREATESTRUCT lpCreateStruct);
    afx_msg void OnEndlabeledit(NMHDR* pNMHDR, LRESULT* pResult);
    //}}AFX_MSG
    afx_msg void OnKeyUp(UINT nChar, UINT nRepCnt, UINT nFlags);
    afx_msg void OnRButtonDown(UINT nFlags, CPoint point);
    afx_msg void OnLButtonDblClk(UINT nFlags, CPoint point);
    DECLARE_MESSAGE_MAP()
};

/////////////////////////////////////////////////////////////////////////////

//{{AFX_INSERT_LOCATION}}
// Microsoft Visual C++ will insert additional declarations immediately before the previous line.

#endif // !defined(AFX_TABTREECTRL_H__AC2374B6_5047_4F92_96F4_7D5DEAE9BDCE__INCLUDED_)