// Berkeley Open Infrastructure for Network Computing
// http://boinc.berkeley.edu
// Copyright (C) 2005 University of California
//
// This is free software; you can redistribute it and/or
// modify it under the terms of the GNU Lesser General Public
// License as published by the Free Software Foundation;
// either version 2.1 of the License, or (at your option) any later version.
//
// This software 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 Lesser General Public License for more details.
//
// To view the GNU Lesser General Public License visit
// http://www.gnu.org/copyleft/lesser.html
// or write to the Free Software Foundation, Inc.,
// 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

#if defined(__GNUG__) && !defined(__APPLE__)
#pragma implementation "ViewResources.h"
#endif

#include "stdwx.h"
#include "BOINCGUIApp.h"
#include "MainDocument.h"
#include "BOINCTaskCtrl.h"
#include "BOINCListCtrl.h"
#include "ViewResources.h"
#include "Events.h"


#include "res/usage.xpm"


#define COLUMN_PROJECT              0
#define COLUMN_DISKSPACE            1


CResource::CResource() {
}


CResource::~CResource() {
    m_strProjectName.Clear();
    m_strDiskSpace.Clear();
}


IMPLEMENT_DYNAMIC_CLASS(CViewResources, CBOINCBaseView)

BEGIN_EVENT_TABLE (CViewResources, CBOINCBaseView)
    EVT_LIST_ITEM_SELECTED(ID_LIST_RESOURCEUTILIZATIONVIEW, CViewResources::OnListSelected)
    EVT_LIST_ITEM_DESELECTED(ID_LIST_RESOURCEUTILIZATIONVIEW, CViewResources::OnListDeselected)
END_EVENT_TABLE ()


CViewResources::CViewResources()
{}


CViewResources::CViewResources(wxNotebook* pNotebook) :
    CBOINCBaseView(pNotebook, ID_TASK_RESOURCEUTILIZATIONVIEW, DEFAULT_TASK_FLAGS, ID_LIST_RESOURCEUTILIZATIONVIEW, DEFAULT_LIST_SINGLE_SEL_FLAGS)
{
    wxASSERT(m_pTaskPane);
    wxASSERT(m_pListPane);

    //
    // Setup View
    //

    // Create Task Pane Items
    m_pTaskPane->UpdateControls();

    // Create List Pane Items
    m_pListPane->InsertColumn(COLUMN_PROJECT, _("Project"), wxLIST_FORMAT_LEFT, -1);
    m_pListPane->InsertColumn(COLUMN_DISKSPACE, _("Disk Space"), wxLIST_FORMAT_RIGHT, -1);

    UpdateSelection();
}


CViewResources::~CViewResources() {
    EmptyCache();
    EmptyTasks();
}


wxString& CViewResources::GetViewName() {
    static wxString strViewName(_("Disk"));
    return strViewName;
}


const char** CViewResources::GetViewIcon() {
    return usage_xpm;
}


wxInt32 CViewResources::GetDocCount() {
    return wxGetApp().GetDocument()->GetResourceCount();
}


wxString CViewResources::OnListGetItemText(long item, long column) const {
    CResource* resource   = NULL;
    wxString   strBuffer  = wxEmptyString;

    try {
        resource = m_ResourceCache.at(item);
    } catch ( std::out_of_range ) {
        resource = NULL;
    }

    if (resource) {
        switch(column)
        {
            case COLUMN_PROJECT:
                strBuffer = resource->m_strProjectName;
                break;
            case COLUMN_DISKSPACE:
                strBuffer = resource->m_strDiskSpace;
                break;
        }
    }

    return strBuffer;
}


wxString CViewResources::OnDocGetItemText(long item, long column) const {
    wxString       strBuffer = wxEmptyString;

    switch(column) {
    case COLUMN_PROJECT:
        FormatProjectName(item, strBuffer);
        break;
    case COLUMN_DISKSPACE:
        FormatDiskSpace(item, strBuffer);
        break;
    }
    return strBuffer;
}


wxInt32 CViewResources::AddCacheElement() {
    CResource* pItem = new CResource();
    wxASSERT(pItem);
    if (pItem) {
        m_ResourceCache.push_back(pItem);
        return 0;
    }
    return -1;
}


wxInt32 CViewResources::EmptyCache()
{
    unsigned int i;
    for (i=0; i<m_ResourceCache.size(); i++) {
        delete m_ResourceCache[i];
    }
    m_ResourceCache.clear();
    return 0;
}


wxInt32 CViewResources::GetCacheCount() {
    return m_ResourceCache.size();
}


wxInt32 CViewResources::RemoveCacheElement() {
    delete m_ResourceCache.back();
    m_ResourceCache.erase(m_ResourceCache.end() - 1);
    return 0;
}


wxInt32 CViewResources::UpdateCache(long item, long column, wxString& strNewData) {
    CResource* resource   = m_ResourceCache.at(item);

    switch(column) {
    case COLUMN_PROJECT:
        resource->m_strProjectName = strNewData;
        break;
    case COLUMN_DISKSPACE:
        resource->m_strDiskSpace = strNewData;
        break;
    }
    return 0;
}


void CViewResources::UpdateSelection() {
    CBOINCBaseView::PreUpdateSelection();
    CBOINCBaseView::PostUpdateSelection();
}


wxInt32 CViewResources::FormatProjectName(wxInt32 item, wxString& strBuffer) const {
    CMainDocument* doc = wxGetApp().GetDocument();
    PROJECT* resource = wxGetApp().GetDocument()->resource(item);
    PROJECT* state_project = NULL;
    std::string project_name;

    wxASSERT(doc);
    wxASSERT(wxDynamicCast(doc, CMainDocument));

    if (resource) {
        state_project = doc->state.lookup_project(resource->master_url);
        if (state_project) {
            state_project->get_name(project_name);
            strBuffer = wxString(project_name.c_str(), wxConvUTF8);
        }
    }

    return 0;
}


wxInt32 CViewResources::FormatDiskSpace(wxInt32 item, wxString& strBuffer) const {
    float          fBuffer = 0;
    double         xTera = 1099511627776.0;
    double         xGiga = 1073741824.0;
    double         xMega = 1048576.0;
    double         xKilo = 1024.0;
    PROJECT* resource = wxGetApp().GetDocument()->resource(item);

    if (resource) {
        fBuffer = resource->disk_usage;
    }

    if (fBuffer >= xTera) {
        strBuffer.Printf(wxT("%0.2f TB"), fBuffer/xTera);
    } else if (fBuffer >= xGiga) {
        strBuffer.Printf(wxT("%0.2f GB"), fBuffer/xGiga);
    } else if (fBuffer >= xMega) {
        strBuffer.Printf(wxT("%0.2f MB"), fBuffer/xMega);
    } else if (fBuffer >= xKilo) {
        strBuffer.Printf(wxT("%0.2f KB"), fBuffer/xKilo);
    } else {
        strBuffer.Printf(wxT("%0.0f bytes"), fBuffer);
    }

    return 0;
}


const char *BOINC_RCSID_5a37b46a6e = "$Id: ViewResources.cpp,v 1.51.2.1 2006/04/18 00:49:47 rwalton Exp $";