boinc/clientgui/ViewResources.cpp

353 lines
10 KiB
C++

// This file is part of BOINC.
// http://boinc.berkeley.edu
// Copyright (C) 2008 University of California
//
// BOINC 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 3 of the License, or (at your option) any later version.
//
// BOINC 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.
//
// You should have received a copy of the GNU Lesser General Public License
// along with BOINC. If not, see <http://www.gnu.org/licenses/>.
#if defined(__GNUG__) && !defined(__APPLE__)
#pragma implementation "ViewResources.h"
#endif
#include "stdwx.h"
#include "BOINCGUIApp.h"
#include "BOINCBaseFrame.h"
#include "MainDocument.h"
#include "AdvancedFrame.h"
#include "BOINCTaskCtrl.h"
#include "BOINCListCtrl.h"
#include "ViewResources.h"
#include "Events.h"
#include <wx/arrimpl.cpp>
#include "res/usage.xpm"
WX_DEFINE_OBJARRAY(wxArrayColour);
IMPLEMENT_DYNAMIC_CLASS(CViewResources, CBOINCBaseView)
BEGIN_EVENT_TABLE (CViewResources, CBOINCBaseView)
END_EVENT_TABLE ()
CViewResources::CViewResources()
{}
CViewResources::CViewResources(wxNotebook* pNotebook) :
CBOINCBaseView(pNotebook)
{
m_BOINCwasEmpty=false;
wxGridSizer* itemGridSizer = new wxGridSizer(2, 0, 3);
wxASSERT(itemGridSizer);
// create pie chart ctrl for total disk usage
m_pieCtrlTotal = new wxPieCtrl(this, ID_PIECTRL_RESOURCEUTILIZATIONVIEWTOTAL, wxDefaultPosition, wxDefaultSize);
wxASSERT(m_pieCtrlTotal);
// setup the legend
m_pieCtrlTotal->SetTransparent(true);
m_pieCtrlTotal->SetHorLegendBorder(10);
m_pieCtrlTotal->SetLabelFont(*wxSWISS_FONT);
m_pieCtrlTotal->SetLabelColour(wxColour(0,0,0));
m_pieCtrlTotal->SetLabel(_("Total disk usage"));
// initialize pie control
m_pieCtrlTotal->m_Series.Clear();
wxPiePart CtrlTotalPart;
CtrlTotalPart.SetLabel(wxEmptyString);
CtrlTotalPart.SetValue(1);
CtrlTotalPart.SetColour(wxColour(255,255,255));
m_pieCtrlTotal->m_Series.Add(CtrlTotalPart);
m_pieCtrlTotal->Refresh();
// create pie chart ctrl for BOINC disk usage
m_pieCtrlBOINC = new wxPieCtrl(this, ID_PIECTRL_RESOURCEUTILIZATIONVIEW, wxDefaultPosition, wxDefaultSize);
wxASSERT(m_pieCtrlBOINC);
//setup the legend
m_pieCtrlBOINC->SetTransparent(true);
m_pieCtrlBOINC->SetHorLegendBorder(10);
m_pieCtrlBOINC->SetLabelFont(*wxSWISS_FONT);
m_pieCtrlBOINC->SetLabelColour(wxColour(0,0,0));
m_pieCtrlBOINC->SetLabel(_("Disk usage by BOINC projects"));
// initialize pie control
m_pieCtrlBOINC->m_Series.Clear();
wxPiePart CtrlBOINCPart;
CtrlBOINCPart.SetLabel(wxEmptyString);
CtrlBOINCPart.SetValue(1);
CtrlBOINCPart.SetColour(wxColour(255,255,255));
m_pieCtrlBOINC->m_Series.Add(CtrlBOINCPart);
m_pieCtrlBOINC->Refresh();
//init the flexGrid
itemGridSizer->Add(m_pieCtrlTotal, 1, wxGROW|wxALL,1);
itemGridSizer->Add(m_pieCtrlBOINC, 1, wxGROW|wxALL,1);
SetSizer(itemGridSizer);
UpdateSelection();
}
CViewResources::~CViewResources() {
EmptyTasks();
}
wxString& CViewResources::GetViewName() {
static wxString strViewName(wxT("Disk"));
return strViewName;
}
wxString& CViewResources::GetViewDisplayName() {
static wxString strViewName(_("Disk"));
return strViewName;
}
const char** CViewResources::GetViewIcon() {
return usage_xpm;
}
const int CViewResources::GetViewRefreshRate() {
return 10;
}
const int CViewResources::GetViewCurrentViewPage() {
return VW_DISK;
}
void CViewResources::UpdateSelection() {
CBOINCBaseView::PreUpdateSelection();
}
wxInt32 CViewResources::FormatProjectName(PROJECT* project, wxString& strBuffer) const {
CMainDocument* doc = wxGetApp().GetDocument();
std::string project_name;
wxASSERT(doc);
wxASSERT(wxDynamicCast(doc, CMainDocument));
if (project) {
PROJECT* state_project = doc->state.lookup_project(project->master_url);
if (state_project) {
state_project->get_name(project_name);
strBuffer = HtmlEntityDecode(wxString(project_name.c_str(), wxConvUTF8));
}
}
return 0;
}
bool CViewResources::OnSaveState(wxConfigBase* /*pConfig*/) {
return true;
}
bool CViewResources::OnRestoreState(wxConfigBase* /*pConfig*/) {
return true;
}
void hsv2rgb(double h, double s, double v, double& r, double& g, double& b) {
double m, n, f;
int i = floor(h);
f = h - i;
if (!(i&1)) f = 1 - f;
m = v * (1 - s);
n = v * (1 - s*f);
switch (i) {
case 6:
case 0: r = v; g = n; b = m; return;
case 1: r = n; g = v; b = m; return;
case 2: r = m; g = v; b = n; return;
case 3: r = m; g = n; b = v; return;
case 4: r = n; g = m; b = v; return;
case 5: r = v; g = m; b = n; return;
}
}
void CViewResources::OnListRender( wxTimerEvent& WXUNUSED(event) ) {
CMainDocument* pDoc = wxGetApp().GetDocument();
wxString diskspace;
static double project_total=0.0;
unsigned int i;
wxASSERT(pDoc);
wxASSERT(wxDynamicCast(pDoc, CMainDocument));
//get data for BOINC projects disk usage
pDoc->CachedDiskUsageUpdate();
pDoc->CachedStateUpdate();
bool refreshBOINC=false;
if (pDoc->disk_usage.projects.size()>0) {
m_BOINCwasEmpty=false;
//check for changes worth a refresh
if(pDoc->disk_usage.projects.size() != m_pieCtrlBOINC->m_Series.size()) {
refreshBOINC=true;
} else {
for (i=0; i<pDoc->disk_usage.projects.size(); i++) {
wxString oldValue;
wxString newValue;
FormatDiskSpace(pDoc->DiskUsageProject(i)->disk_usage, newValue);
FormatDiskSpace(m_pieCtrlBOINC->m_Series.Item(i).GetValue(), oldValue);
if(newValue.Cmp(oldValue)!=0) {
refreshBOINC=true;
break;
}
}
}
//only refresh when worthy changes
if(refreshBOINC) {
m_pieCtrlBOINC->m_Series.Clear();
project_total = 0;
for (i=0; i<pDoc->disk_usage.projects.size(); i++) {
//update data for boinc projects pie chart
PROJECT* project = pDoc->DiskUsageProject(i);
wxString projectname;
FormatProjectName(project, projectname);
FormatDiskSpace(project->disk_usage, diskspace);
double usage = project->disk_usage;
project_total += usage;
wxPiePart part;
part.SetLabel(projectname + wxT(": ") + diskspace);
part.SetValue(usage);
double h = (double)i/(double)pDoc->disk_usage.projects.size();
double r, g, b;
double v = .5 + (i % 3)*.2;
// cycle through 3 different brightnesses
hsv2rgb(h*6, .5, v, r, g, b);
unsigned char cr = (unsigned char) (r*256);
unsigned char cg = (unsigned char) (g*256);
unsigned char cb = (unsigned char) (b*256);
part.SetColour(wxColour(cr, cg, cb));
m_pieCtrlBOINC->m_Series.Add(part);
}
m_pieCtrlBOINC->Refresh();
}
} else {
if(!m_BOINCwasEmpty) {
//paint an empty black pie
m_pieCtrlBOINC->m_Series.Clear();
wxPiePart part;
part.SetLabel(_("no projects: 0 bytes used"));
part.SetValue(1);
part.SetColour(wxColour(0,0,0));
m_pieCtrlBOINC->m_Series.Add(part);
m_pieCtrlBOINC->Refresh();
m_BOINCwasEmpty=true;
refreshBOINC=true;
}
}
//pDoc->disk_usage.d_allowed = 0;
//data for pie chart 2 (total disk usage)
//
// good source of color palettes:
// http://www.siteprocentral.com/cgi-bin/feed/feed.cgi
//
bool refreshTotal=false;
double free = pDoc->disk_usage.d_free;
double total = pDoc->disk_usage.d_total;
if(m_pieCtrlTotal->m_Series.size()>0) {
wxString oldFree;
wxString newFree;
FormatDiskSpace(free, newFree);
FormatDiskSpace(m_pieCtrlTotal->m_Series.Item(0).GetValue(), oldFree);
if(oldFree.Cmp(newFree)!=0) {
refreshTotal=true;
}
} else {
refreshTotal=true;
}
if(refreshBOINC || refreshTotal) {
m_pieCtrlTotal->m_Series.Clear();
wxPiePart part;
// used by BOINC
double boinc_total = project_total + pDoc->disk_usage.d_boinc;
FormatDiskSpace(boinc_total, diskspace);
part.SetLabel(_("used by BOINC: ") + diskspace);
part.SetValue(boinc_total);
part.SetColour(wxColour(0,0,0));
m_pieCtrlTotal->m_Series.Add(part);
if (pDoc->disk_usage.d_allowed > 0) {
double avail = pDoc->disk_usage.d_allowed - boinc_total;
if (avail > 0) {
if (avail > free) avail = free;
FormatDiskSpace(avail, diskspace);
part.SetLabel(_("free, available to BOINC: ") + diskspace);
part.SetValue(avail);
part.SetColour(wxColour(128, 128, 128));
m_pieCtrlTotal->m_Series.Add(part);
} else {
avail = 0;
}
double not_avail = free - avail;
if (not_avail > 0) {
FormatDiskSpace(not_avail, diskspace);
part.SetLabel(_("free, not available to BOINC: ") + diskspace);
part.SetValue(not_avail);
part.SetColour(wxColour(238,238,238));
m_pieCtrlTotal->m_Series.Add(part);
}
} else {
// if d_allowed is zero, we must be talking to a pre-6.3 client.
// Just show free space
//
FormatDiskSpace(free, diskspace);
part.SetLabel(_("free: ") + diskspace);
part.SetValue(free);
part.SetColour(wxColour(238,238,238));
m_pieCtrlTotal->m_Series.Add(part);
}
// used by others
double used_by_others = total-boinc_total-free;
FormatDiskSpace(used_by_others, diskspace);
part.SetLabel(_("used by other programs: ") + diskspace);
part.SetValue(used_by_others);
part.SetColour(wxColour(192,192,192));
m_pieCtrlTotal->m_Series.Add(part);
m_pieCtrlTotal->Refresh();
}
}
wxInt32 CViewResources::FormatDiskSpace(double bytes, wxString& strBuffer) const {
double xTera = 1099511627776.0;
double xGiga = 1073741824.0;
double xMega = 1048576.0;
double xKilo = 1024.0;
if (bytes >= xTera) {
strBuffer.Printf(wxT("%0.2f TB"), bytes/xTera);
} else if (bytes >= xGiga) {
strBuffer.Printf(wxT("%0.2f GB"), bytes/xGiga);
} else if (bytes >= xMega) {
strBuffer.Printf(wxT("%0.2f MB"), bytes/xMega);
} else {
strBuffer.Printf(wxT("%0.2f KB"), bytes/xKilo);
}
return 0;
}