mirror of
https://github.com/levinsv/pgadmin3.git
synced 2026-05-15 06:05:49 -06:00
342 lines
8.8 KiB
C++
342 lines
8.8 KiB
C++
//////////////////////////////////////////////////////////////////////////
|
||
//
|
||
// pgAdmin III - PostgreSQL Tools
|
||
//
|
||
// Copyright (C) 2002 - 2016, The pgAdmin Development Team
|
||
// This software is released under the PostgreSQL Licence
|
||
//
|
||
// ctlSQLGrid.h - SQL Data Display Grid
|
||
//
|
||
//////////////////////////////////////////////////////////////////////////
|
||
|
||
#ifndef CTLSQLGRID_H
|
||
#define CTLSQLGRID_H
|
||
|
||
// wxWindows headers
|
||
#include <wx/grid.h>
|
||
#include <wx/brush.h>
|
||
class GroupRows;
|
||
|
||
class ctlSQLGrid : public wxGrid
|
||
{
|
||
public:
|
||
ctlSQLGrid(wxWindow *parent, wxWindowID id, const wxPoint &pos, const wxSize &size);
|
||
ctlSQLGrid();
|
||
|
||
wxString GetExportLine(int row);
|
||
wxString GetExportLine(int row, wxArrayInt cols);
|
||
wxString GetExportLine(int row, int col1, int col2);
|
||
virtual bool IsColText(int col)
|
||
{
|
||
return false;
|
||
}
|
||
int CopyTableToHtml(wxString htmlquery);
|
||
int Copy(int gensql);
|
||
|
||
virtual bool CheckRowPresent(int row)
|
||
{
|
||
return true;
|
||
};
|
||
bool IsSort()
|
||
{
|
||
return isSort;
|
||
};
|
||
void SetSort(bool flag)
|
||
{
|
||
isSort=flag;
|
||
};
|
||
|
||
wxString GetColumnName(int colNum);
|
||
wxSize GetBestSize(int row, int col);
|
||
void OnLabelDoubleClick(wxGridEvent &event);
|
||
void OnGridSelectCell(wxGridEvent& evt);
|
||
void OnLabelClick(wxGridEvent &event);
|
||
void OnCellRightClick(wxGridEvent &event);
|
||
bool FullArrayCollapseRowsPlan(bool clear);
|
||
void AutoSizeColumn(int col, bool setAsMin = false, bool doLimit = true);
|
||
void AutoSizeColumns(bool setAsMin);
|
||
wxString GetRowLabelValue( int row );
|
||
void SetRowGroup(int row);
|
||
GroupRows *grp;
|
||
int generatesql; // 0 -<2D><><EFBFBD>, 1 - insert , 2 - in_list
|
||
wxString sqlquerytext;
|
||
// Fast searh
|
||
wxString searchStr;
|
||
|
||
WX_DECLARE_STRING_HASH_MAP( int, ColKeySizeHashMap );
|
||
|
||
DECLARE_DYNAMIC_CLASS(ctlSQLGrid)
|
||
DECLARE_EVENT_TABLE()
|
||
|
||
private:
|
||
void setresizedpi();
|
||
void OnCopy(wxCommandEvent &event);
|
||
void OnMouseWheel(wxMouseEvent &event);
|
||
void OnGridColSize(wxGridSizeEvent &event);
|
||
void DrawColLabel( wxDC& dc, int col );
|
||
void DrawRowLabel(wxDC& dc, int row);
|
||
wxString GetColKeyValue(int col);
|
||
void AppendColumnHeader(wxString &str, int start, int end);
|
||
void AppendColumnHeader(wxString &str, wxArrayInt columns);
|
||
// Stores sizes of colums explicitly resized by user
|
||
ColKeySizeHashMap colSizes;
|
||
// Max size for each column
|
||
wxArrayInt colMaxSizes;
|
||
bool isSort;
|
||
|
||
};
|
||
|
||
class GroupRows
|
||
{
|
||
public:
|
||
GroupRows( ctlSQLGrid *grid) {
|
||
g=grid;
|
||
rowsGroup.Clear();
|
||
rowsGroup.Add(0,g->GetNumberRows());
|
||
end.Clear();
|
||
end.Add(-1,g->GetNumberRows());
|
||
run.Clear();
|
||
run.Add(0.0,g->GetNumberRows());
|
||
};
|
||
void AddGroup( int rowgroup, int lastrowgroup, double actualtime) {
|
||
// default group open
|
||
rowsGroup[rowgroup]=-rowgroup;
|
||
//beg[rowgroup]=rowgroup;
|
||
wxASSERT_MSG(lastrowgroup>end.Count()," out of bounds");
|
||
end[rowgroup]=lastrowgroup;
|
||
run[rowgroup]=actualtime;
|
||
};
|
||
void VisibleGroup(int row, bool visible) {
|
||
int endg=end[row];
|
||
int grp=IsGroupRow(row);
|
||
int gg;
|
||
int r=row+1;
|
||
if (grp!=0) {
|
||
|
||
if (!visible) {
|
||
// hide group
|
||
rowsGroup[row]*=-1;
|
||
for(int i=r;i<(endg+1);i++) {
|
||
g->HideRow(i);
|
||
}
|
||
wxGridCellAttr* pAttrg = new wxGridCellAttr;
|
||
pAttrg->SetBackgroundColour(wxColour(200,191,232));
|
||
g->SetRowAttr(row,pAttrg);
|
||
} else
|
||
{
|
||
// show group
|
||
rowsGroup[row]*=-1;
|
||
int sizerow=g->GetDefaultRowSize();
|
||
//pAttr->SetBackgroundColour(wxColour(0,162,232));
|
||
wxGridCellAttr* pAttrg = new wxGridCellAttr;
|
||
pAttrg->SetBackgroundColour(wxColour(248,240,130));
|
||
g->SetRowAttr(row,pAttrg);
|
||
for(int i=r;i<(endg+1);i++) {
|
||
gg=IsGroupRow(i);
|
||
if (gg<=0) {
|
||
g->SetRowSize(i,sizerow);
|
||
}
|
||
else
|
||
{
|
||
g->SetRowSize(i,sizerow);
|
||
i=end[i];
|
||
}
|
||
|
||
}
|
||
|
||
}
|
||
}
|
||
};
|
||
// 0 - no group, -int - open group ,+int - close gruop
|
||
int IsGroupRow(int row) {
|
||
if (end[row]!=-1) {
|
||
// is group
|
||
return rowsGroup[row];
|
||
} else return 0;
|
||
};
|
||
void ColoriseRow(int row,const wxColour& color) {
|
||
//g->GetTable()->SetRowLabelValue(row-1,s);
|
||
wxGridCellAttr* pAttr = new wxGridCellAttr;
|
||
pAttr->SetBackgroundColour(color);
|
||
g->SetRowAttr(row,pAttr);
|
||
};
|
||
void CalcTime() {
|
||
//g->GetTable()->SetRowLabelValue(row-1,s);
|
||
double sum=0;
|
||
double total=1;
|
||
for(int i=0;i<run.Count();i++) {
|
||
double t=run[i];
|
||
if (i==0) {
|
||
total=abs(t);
|
||
g->GetTable()->SetRowLabelValue(i,"100");
|
||
} else
|
||
{
|
||
wxString str;
|
||
if (total==0||t==0) {
|
||
str="";
|
||
} else
|
||
{
|
||
|
||
t=(t/total)*100;
|
||
str.Printf(wxT("%5.2f"), t);
|
||
}
|
||
g->GetTable()->SetRowLabelValue(i,str);
|
||
}
|
||
|
||
}
|
||
|
||
|
||
};
|
||
|
||
private:
|
||
ctlSQLGrid *g;
|
||
wxArrayInt rowsGroup, end;
|
||
wxArrayDouble run;
|
||
};
|
||
//#define TEST_wxGridCellAutoWrapStringRenderer
|
||
#ifndef TEST_wxGridCellAutoWrapStringRenderer
|
||
class CursorCellRenderer : public wxGridCellStringRenderer
|
||
{
|
||
public:
|
||
virtual void Draw(wxGrid& grid, wxGridCellAttr& attr, wxDC& dc,
|
||
const wxRect& rect, int row, int col, bool isSelected)
|
||
{
|
||
int hAlign, vAlign;
|
||
int sPos=-1;
|
||
bool multiline = false;
|
||
attr.GetAlignment(&hAlign, &vAlign);
|
||
//////////////////////////////////////////////////////////////////////////////
|
||
//CursorCellRenderer::Draw(grid, attr, dc, rect, row, col, isSelected); //
|
||
dc.SetBackgroundMode( wxSOLID );
|
||
wxString text=grid.GetCellValue(row, col);
|
||
// grey out fields if the grid is disabled
|
||
if ( grid.IsEnabled() )
|
||
{
|
||
if ( isSelected )
|
||
{
|
||
wxColour clr;
|
||
if ( wxWindow::FindFocus() == grid.GetGridWindow() )
|
||
clr = grid.GetSelectionBackground();
|
||
else
|
||
clr = wxSystemSettings::GetColour(wxSYS_COLOUR_BTNSHADOW);
|
||
|
||
|
||
dc.SetBrush( *wxTheBrushList->FindOrCreateBrush(clr));
|
||
}
|
||
else
|
||
{
|
||
wxColor color;
|
||
color.Set(239, 228, 176);
|
||
if (sPos = text.Find(wxT('\n')) != wxNOT_FOUND) {
|
||
dc.SetBrush(*wxTheBrushList->FindOrCreateBrush(color));
|
||
multiline = true;
|
||
}
|
||
else
|
||
dc.SetBrush(*wxTheBrushList->FindOrCreateBrush(attr.GetBackgroundColour()));
|
||
}
|
||
}
|
||
else
|
||
{
|
||
dc.SetBrush(*wxTheBrushList->FindOrCreateBrush(wxSystemSettings::GetColour(wxSYS_COLOUR_BTNFACE)));
|
||
}
|
||
|
||
dc.SetPen( *wxTRANSPARENT_PEN );
|
||
dc.DrawRectangle(rect);
|
||
//////////////////////////////////////////////////////////////////////////////
|
||
SetTextColoursAndFont(grid, attr, dc, isSelected);
|
||
ctlSQLGrid* ctrl = static_cast<ctlSQLGrid*>(&grid);
|
||
if (!ctrl->searchStr.IsEmpty()) {
|
||
if (sPos == -1) sPos = text.Len();
|
||
int pp;
|
||
|
||
pp = text.Find(ctrl->searchStr);
|
||
if (pp >= 0) {
|
||
wxArrayString lines;
|
||
grid.StringToLines(text, lines);
|
||
wxRect r;
|
||
r.y = rect.y;
|
||
dc.SetBrush(*wxYELLOW_BRUSH);
|
||
size_t nLines = lines.GetCount();
|
||
for (size_t l = 0; l < nLines; l++)
|
||
{
|
||
const wxString& line = lines[l];
|
||
pp = line.Find(ctrl->searchStr);
|
||
if (line.empty() || (pp==-1))
|
||
{
|
||
r.y += dc.GetCharHeight();
|
||
continue;
|
||
}
|
||
|
||
int lineWidth, lineWidthP, lineHeight,start=0;
|
||
wxString pref;
|
||
if (hAlign == wxALIGN_RIGHT) {
|
||
start = pp + ctrl->searchStr.Len();
|
||
pref = line.substr(start);
|
||
} else
|
||
pref = line.substr(start, pp);
|
||
|
||
dc.GetTextExtent(pref, &lineWidthP, &lineHeight);
|
||
r.x=rect.x+lineWidthP;
|
||
pref= line.substr(pp, ctrl->searchStr.Len());
|
||
dc.GetTextExtent(pref, &lineWidth, &lineHeight);
|
||
r.width = lineWidth;
|
||
r.height = lineHeight;
|
||
if (hAlign == wxALIGN_RIGHT) r.x = rect.x + (rect.width - lineWidth- lineWidthP);
|
||
if (!(r.y < (rect.y + rect.height))) { r.y = rect.y + rect.height - 5; r.height = 5; }
|
||
if (!((r.x < (rect.x + rect.width)))) { r.x = rect.x + rect.width - 5; r.width = 5; }
|
||
dc.DrawRoundedRectangle(r, 3);
|
||
break;
|
||
}
|
||
}
|
||
}
|
||
|
||
if (!multiline) {
|
||
//int textWidth = dc.GetTextExtent(text).GetWidth();
|
||
wxEllipsizeMode mode(wxELLIPSIZE_END);
|
||
if (hAlign == wxALIGN_RIGHT) mode = wxELLIPSIZE_START;
|
||
const wxString& ellipsizedText = wxControl::Ellipsize
|
||
(
|
||
text,
|
||
dc,
|
||
mode,
|
||
rect.GetWidth() - 2,
|
||
wxELLIPSIZE_FLAGS_NONE
|
||
);
|
||
if (ellipsizedText != text) text=ellipsizedText;
|
||
}
|
||
grid.DrawTextRectangle(dc, text,
|
||
rect, hAlign, vAlign);
|
||
}
|
||
#else
|
||
class CursorCellRenderer : public wxGridCellAutoWrapStringRenderer
|
||
{
|
||
public:
|
||
virtual void Draw(wxGrid& grid,
|
||
wxGridCellAttr& attr,
|
||
wxDC& dc,
|
||
const wxRect& rectCell,
|
||
int row, int col,
|
||
bool isSelected) {
|
||
bool f = false;
|
||
wxString text = grid.GetCellValue(row, col);
|
||
wxColor prev;
|
||
if (!isSelected) {
|
||
wxColor color;
|
||
color.Set(239, 228, 176);
|
||
if (text.Find(wxT('\n')) != wxNOT_FOUND)
|
||
{
|
||
//dc.SetBrush(wxBrush(color, wxSOLID));
|
||
prev = attr.GetBackgroundColour();
|
||
attr.SetBackgroundColour(color);
|
||
f = true;
|
||
}
|
||
|
||
}
|
||
|
||
wxGridCellAutoWrapStringRenderer::Draw(grid, attr, dc, rectCell, row, col, isSelected);
|
||
if (f) attr.SetBackgroundColour(prev);
|
||
}
|
||
#endif
|
||
};
|
||
|
||
#endif
|