source: trunk/src/advtabs/Peers.hpp @ 531

//         Copyright Eóin O'Callaghan 2006 - 2008.
// Distributed under the Boost Software License, Version 1.0.
//    (See accompanying file LICENSE_1_0.txt or copy at
//          http://www.boost.org/LICENSE_1_0.txt)

#pragma once

#include "stdAfx.hpp"
#include "Halite.hpp"

#include "DdxEx.hpp"
#include "global/string_conv.hpp"

#include "../HaliteTabPage.hpp"
#include "../HaliteDialogBase.hpp"
#include "../HaliteListManager.hpp"

class PeerListView :
        public CHaliteSortListViewCtrl<PeerListView, const hal::PeerDetail>,
        public hal::IniBase<PeerListView>,
        private boost::noncopyable
{
protected:
        typedef PeerListView thisClass;
        typedef hal::IniBase<thisClass> iniClass;
        typedef CHaliteSortListViewCtrl<thisClass, const hal::PeerDetail> listClass;
        typedef const hal::PeerDetail pD;

        friend class listClass;
        
        struct ColumnAdapters
        {
        
        typedef listClass::ColumnAdapter ColAdapter_t;
        
        struct SpeedDown : public ColAdapter_t
        {
                virtual int compare(pD& l, pD& r)       { return hal::compare(l.speed.first, r.speed.first); }          
                virtual std::wstring print(pD& p) 
                {
                        return (hal::wform(L"%1$.2fkb/s") % (p.speed.first/1024)).str(); 
                }               
        };
        
        struct SpeedUp : public ColAdapter_t
        {
                virtual int compare(pD& l, pD& r)       { return hal::compare(l.speed.second, r.speed.second); }                
                virtual std::wstring print(pD& p) 
                {
                        return (hal::wform(L"%1$.2fkb/s") % (p.speed.second/1024)).str(); 
                }               
        };
        
        };

public: 
        enum { 
                LISTVIEW_ID_MENU = 0,
                LISTVIEW_ID_COLUMNNAMES = HAL_DIALOGPEER_LISTVIEW_ADV,
                LISTVIEW_ID_COLUMNWIDTHS = 0
        };
        
        BEGIN_MSG_MAP_EX(thisClass)
                MSG_WM_DESTROY(OnDestroy)

                CHAIN_MSG_MAP(listClass)
                DEFAULT_REFLECTION_HANDLER()
        END_MSG_MAP()

        thisClass() :
                iniClass("listviews/advPeers", "PeerListView")
        {}
        
        void saveSettings()
        {
                GetListViewDetails();
                save_to_ini();
        }
        
        bool SubclassWindow(HWND hwnd)
        {
                if(!listClass::SubclassWindow(hwnd))
                        return false;

                InitialSetup(); 
                
                std::vector<wstring> names;     
                wstring column_names = hal::app().res_wstr(LISTVIEW_ID_COLUMNNAMES);

                // "Peer;Country;Download;Upload;Type;Client,Status"
                boost::split(names, column_names, boost::is_any_of(L";"));
                
                array<int, 7> widths = {100,20,70,70,70,100,200};
                array<int, 7> order = {0,1,2,3,4,5,6};
                array<bool, 7> visible = {true,true,true,true,true,true,true};

                for (int i=0, e=7; i < e; ++i)
                {
                        AddColumn(names[i].c_str(), i, visible[i], widths[i]);
                }       
                                        
                load_from_ini();
                
                SetColumnSortType(2, WTL::LVCOLSORT_CUSTOM, new ColumnAdapters::SpeedDown());
                SetColumnSortType(3, WTL::LVCOLSORT_CUSTOM, new ColumnAdapters::SpeedUp());
                
                return true;
        }
        
        void OnDestroy()
        {
                saveSettings();
        }
        
        friend class boost::serialization::access;
        template<class Archive>
        void serialize(Archive& ar, const unsigned int version)
        {
                ar & boost::serialization::make_nvp("listview", 
                        boost::serialization::base_object<listClass>(*this));
        }
        
        pD CustomItemConversion(LVCompareParam* param, int iSortCol)
        {                       
                return peerDetails_[param->dwItemData];
        }               

        void uiUpdate(const hal::torrent_details_manager& tD);
        
private:
        hal::PeerDetails peerDetails_;
};

class AdvPeerDialog :
        public CHalTabPageImpl<AdvPeerDialog>,
        public CHaliteDialogBase<AdvPeerDialog>,
        public WTL::CDialogResize<AdvPeerDialog>
{
protected:
        typedef AdvPeerDialog thisClass;
        typedef CHalTabPageImpl<thisClass> baseClass;
        typedef WTL::CDialogResize<thisClass> resizeClass;
        typedef CHaliteDialogBase<AdvPeerDialog> dialogBaseClass;

public:
        enum { IDD = HAL_ADVPEER };

        AdvPeerDialog(HaliteWindow& halWindow) :
                dialogBaseClass(halWindow)
        {}

        BOOL PreTranslateMessage(MSG* pMsg)
        {
                return this->IsDialogMessage(pMsg);
        }

        BEGIN_MSG_MAP_EX(thisClass)
                MSG_WM_INITDIALOG(OnInitDialog)
                MSG_WM_CLOSE(OnClose)
        
                if (uMsg == WM_FORWARDMSG)
                        if (PreTranslateMessage((LPMSG)lParam)) return TRUE;

                CHAIN_MSG_MAP(dialogBaseClass)
                CHAIN_MSG_MAP(resizeClass)
                CHAIN_MSG_MAP(baseClass)
                REFLECT_NOTIFICATIONS()
        END_MSG_MAP()

        BEGIN_DLGRESIZE_MAP(thisClass)
                DLGRESIZE_CONTROL(HAL_PEERLIST, DLSZ_SIZE_X|DLSZ_SIZE_Y)
        END_DLGRESIZE_MAP()

        LRESULT OnInitDialog(HWND, LPARAM);
        void OnClose();
        
        void uiUpdate(const hal::torrent_details_manager& tD);

protected:
        PeerListView peerList_;
};