Refactor BaseGrid painting a bit

Originally committed to SVN as r5829.
This commit is contained in:
Thomas Goyne 2011-11-08 00:24:53 +00:00
parent 8e81fd07ed
commit dc380c5848
2 changed files with 83 additions and 82 deletions

View file

@ -70,6 +70,15 @@ enum {
MENU_SHOW_COL = 1250 // Needs 15 IDs after this MENU_SHOW_COL = 1250 // Needs 15 IDs after this
}; };
enum RowColor {
COLOR_DEFAULT = 0,
COLOR_HEADER,
COLOR_SELECTION,
COLOR_COMMENT,
COLOR_VISIBLE,
COLOR_SELECTED_COMMENT
};
template<class S1, class S2, class D> template<class S1, class S2, class D>
static inline void set_difference(const S1 &src1, const S2 &src2, D &dst) { static inline void set_difference(const S1 &src1, const S2 &src2, D &dst) {
std::set_difference( std::set_difference(
@ -422,19 +431,15 @@ wxArrayInt BaseGrid::GetSelection() const {
void BaseGrid::OnPaint(wxPaintEvent &event) { void BaseGrid::OnPaint(wxPaintEvent &event) {
// Get size and pos // Get size and pos
int w = 0; wxSize cs = GetClientSize();
int h = 0; cs.SetWidth(cs.GetWidth() - scrollBar->GetSize().GetWidth());
GetClientSize(&w,&h);
w -= scrollBar->GetSize().GetWidth();
// Prepare bitmap // Prepare bitmap
if (bmp) { if (bmp && (bmp->GetWidth() < cs.GetWidth() || bmp->GetHeight() < cs.GetHeight())) {
if (bmp->GetWidth() < w || bmp->GetHeight() < h) {
delete bmp; delete bmp;
bmp = 0; bmp = 0;
} }
} if (!bmp) bmp = new wxBitmap(cs);
if (!bmp) bmp = new wxBitmap(w,h);
// Find which columns need to be repainted // Find which columns need to be repainted
bool paint_columns[11]; bool paint_columns[11];
@ -456,7 +461,7 @@ void BaseGrid::OnPaint(wxPaintEvent &event) {
DrawImage(bmpDC, paint_columns); DrawImage(bmpDC, paint_columns);
wxPaintDC dc(this); wxPaintDC dc(this);
dc.Blit(0,0,w,h,&bmpDC,0,0); dc.Blit(wxPoint(0, 0), cs, &bmpDC, wxPoint(0, 0));
} }
void BaseGrid::DrawImage(wxDC &dc, bool paint_columns[]) { void BaseGrid::DrawImage(wxDC &dc, bool paint_columns[]) {
@ -500,95 +505,43 @@ void BaseGrid::DrawImage(wxDC &dc, bool paint_columns[]) {
dc.DrawLine(0, 0, w, 0); dc.DrawLine(0, 0, w, 0);
dc.SetPen(*wxTRANSPARENT_PEN); dc.SetPen(*wxTRANSPARENT_PEN);
// Draw rows wxString strings[] = {
wxString strings[11] = { _("#"), _("L"), _("Start"), _("End"), _("Style"), _("Actor"),
_("#"), _("Effect"), _("Left"), _("Right"), _("Vert"), _("Text")
_("L"),
_("Start"),
_("End"),
_("Style"),
_("Actor"),
_("Effect"),
_("Left"),
_("Right"),
_("Vert"),
_("Text")
}; };
for (int i = 0; i < nDraw + 1; i++) { for (int i = 0; i < nDraw + 1; i++) {
int curRow = i + yPos - 1; int curRow = i + yPos - 1;
int curColor = 0; RowColor curColor = COLOR_DEFAULT;
// Header // Header
if (i == 0) { if (i == 0) {
curColor = 1; curColor = COLOR_HEADER;
dc.SetTextForeground(text_standard); dc.SetTextForeground(text_standard);
} }
// Lines // Lines
else if (AssDialogue *curDiag = GetDialogue(curRow)) { else if (AssDialogue *curDiag = GetDialogue(curRow)) {
// Set fields GetRowStrings(curRow, curDiag, paint_columns, strings);
if (paint_columns[0]) strings[0] = wxString::Format("%i", curRow + 1);
if (paint_columns[1]) strings[1] = wxString::Format("%i", curDiag->Layer);
if (byFrame) {
if (paint_columns[2]) strings[2] = wxString::Format("%i", context->videoController->FrameAtTime(curDiag->Start.GetMS(), agi::vfr::START));
if (paint_columns[3]) strings[3] = wxString::Format("%i", context->videoController->FrameAtTime(curDiag->End.GetMS(), agi::vfr::END));
}
else {
if (paint_columns[2]) strings[2] = curDiag->Start.GetASSFormated();
if (paint_columns[3]) strings[3] = curDiag->End.GetASSFormated();
}
if (paint_columns[4]) strings[4] = curDiag->Style;
if (paint_columns[5]) strings[5] = curDiag->Actor;
if (paint_columns[6]) strings[6] = curDiag->Effect;
if (paint_columns[7]) strings[7] = curDiag->GetMarginString(0);
if (paint_columns[8]) strings[8] = curDiag->GetMarginString(1);
if (paint_columns[9]) strings[9] = curDiag->GetMarginString(2);
// Set text
if (paint_columns[10]) {
strings[10].clear();
int mode = OPT_GET("Subtitle/Grid/Hide Overrides")->GetInt();
// Hidden overrides
if (mode == 1 || mode == 2) {
wxString replaceWith = lagi_wxString(OPT_GET("Subtitle/Grid/Hide Overrides Char")->GetString());
strings[10].reserve(curDiag->Text.size());
size_t start = 0, pos;
while ((pos = curDiag->Text.find('{', start)) != wxString::npos) {
strings[10] += curDiag->Text.Mid(start, pos - start);
strings[10] += replaceWith;
start = curDiag->Text.find('}', pos);
if (start != wxString::npos) ++start;
}
strings[10] += curDiag->Text.Mid(start);
}
// Show overrides
else strings[10] = curDiag->Text;
// Cap length and set text
if (strings[10].size() > 512) strings[10] = strings[10].Left(512) + "...";
}
// Set color
curColor = 0;
bool inSel = !!selection.count(curDiag); bool inSel = !!selection.count(curDiag);
if (inSel && curDiag->Comment) curColor = 5; if (inSel && curDiag->Comment)
else if (inSel) curColor = 2; curColor = COLOR_SELECTED_COMMENT;
else if (curDiag->Comment) curColor = 3; else if (inSel)
else if (OPT_GET("Subtitle/Grid/Highlight Subtitles in Frame")->GetBool() && IsDisplayed(curDiag)) curColor = 4; curColor = COLOR_SELECTION;
else if (curDiag->Comment)
curColor = COLOR_COMMENT;
else if (OPT_GET("Subtitle/Grid/Highlight Subtitles in Frame")->GetBool() && IsDisplayed(curDiag))
curColor = COLOR_VISIBLE;
else
curColor = COLOR_DEFAULT;
if (active_line != curDiag && curDiag->CollidesWith(active_line)) { if (active_line != curDiag && curDiag->CollidesWith(active_line))
dc.SetTextForeground(text_collision); dc.SetTextForeground(text_collision);
} else if (inSel)
else if (inSel) {
dc.SetTextForeground(text_selection); dc.SetTextForeground(text_selection);
} else
else {
dc.SetTextForeground(text_standard); dc.SetTextForeground(text_standard);
} }
}
else { else {
assert(false); assert(false);
} }
@ -647,6 +600,52 @@ void BaseGrid::DrawImage(wxDC &dc, bool paint_columns[]) {
} }
} }
void BaseGrid::GetRowStrings(int row, AssDialogue *line, bool *paint_columns, wxString *strings) const {
if (paint_columns[0]) strings[0] = wxString::Format("%d", row + 1);
if (paint_columns[1]) strings[1] = wxString::Format("%d", line->Layer);
if (byFrame) {
if (paint_columns[2]) strings[2] = wxString::Format("%d", context->videoController->FrameAtTime(line->Start.GetMS(), agi::vfr::START));
if (paint_columns[3]) strings[3] = wxString::Format("%d", context->videoController->FrameAtTime(line->End.GetMS(), agi::vfr::END));
}
else {
if (paint_columns[2]) strings[2] = line->Start.GetASSFormated();
if (paint_columns[3]) strings[3] = line->End.GetASSFormated();
}
if (paint_columns[4]) strings[4] = line->Style;
if (paint_columns[5]) strings[5] = line->Actor;
if (paint_columns[6]) strings[6] = line->Effect;
if (paint_columns[7]) strings[7] = line->GetMarginString(0);
if (paint_columns[8]) strings[8] = line->GetMarginString(1);
if (paint_columns[9]) strings[9] = line->GetMarginString(2);
if (paint_columns[10]) {
strings[10].clear();
int mode = OPT_GET("Subtitle/Grid/Hide Overrides")->GetInt();
// Hidden overrides
if (mode == 1 || mode == 2) {
wxString replaceWith = lagi_wxString(OPT_GET("Subtitle/Grid/Hide Overrides Char")->GetString());
strings[10].reserve(line->Text.size());
size_t start = 0, pos;
while ((pos = line->Text.find('{', start)) != wxString::npos) {
strings[10] += line->Text.Mid(start, pos - start);
strings[10] += replaceWith;
start = line->Text.find('}', pos);
if (start != wxString::npos) ++start;
}
strings[10] += line->Text.Mid(start);
}
// Show overrides
else
strings[10] = line->Text;
// Cap length and set text
if (strings[10].size() > 512)
strings[10] = strings[10].Left(512) + "...";
}
}
void BaseGrid::OnSize(wxSizeEvent &event) { void BaseGrid::OnSize(wxSizeEvent &event) {
AdjustScrollbar(); AdjustScrollbar();
SetColumnWidths(); SetColumnWidths();

View file

@ -112,6 +112,8 @@ class BaseGrid : public wxWindow, public BaseSelectionController<AssDialogue> {
void OnSubtitlesOpen(); void OnSubtitlesOpen();
void DrawImage(wxDC &dc, bool paint_columns[]); void DrawImage(wxDC &dc, bool paint_columns[]);
void GetRowStrings(int row, AssDialogue *line, bool *paint_columns, wxString *strings) const;
void ScrollTo(int y); void ScrollTo(int y);
int colWidth[16]; ///< Width in pixels of each column int colWidth[16]; ///< Width in pixels of each column