From d0150d6b6d836ee81a1f3fa557bb89ef7ee6780d Mon Sep 17 00:00:00 2001
From: Thomas Goyne <plorkyeran@aegisub.org>
Date: Mon, 12 Mar 2012 23:34:54 +0000
Subject: [PATCH] Handle external changes in the translation assisant

As all program hotkeys are now allowed while the translation assistant
is active, it can no longer assume that it's the only thing modifying
the file.

Originally committed to SVN as r6579.
---
 aegisub/src/dialog_translation.cpp | 39 +++++++++++++++++++++++++++++-
 aegisub/src/dialog_translation.h   | 14 ++++++++++-
 2 files changed, 51 insertions(+), 2 deletions(-)

diff --git a/aegisub/src/dialog_translation.cpp b/aegisub/src/dialog_translation.cpp
index e96fe0f9d..74737b673 100644
--- a/aegisub/src/dialog_translation.cpp
+++ b/aegisub/src/dialog_translation.cpp
@@ -66,10 +66,12 @@ static bool bad_block(AssDialogueBlock *block) {
 DialogTranslation::DialogTranslation(agi::Context *c)
 : wxDialog(c->parent, -1, _("Translation Assistant"), wxDefaultPosition, wxDefaultSize, wxDEFAULT_DIALOG_STYLE | wxRESIZE_BORDER | wxMINIMIZE_BOX)
 , c(c)
+, file_change_connection(c->ass->AddCommitListener(&DialogTranslation::OnExternalCommit, this))
 , active_line(c->selectionController->GetActiveLine())
 , cur_block(0)
 , line_count(count_if(c->ass->Line.begin(), c->ass->Line.end(), cast<AssDialogue*>()))
 , line_number(count_if(c->ass->Line.begin(), find(c->ass->Line.begin(), c->ass->Line.end(), active_line), cast<AssDialogue*>()) + 1)
+, switching_lines(false)
 {
 	SetIcon(BitmapToIcon(GETIMAGE(translation_toolbutton_16)));
 
@@ -164,11 +166,40 @@ DialogTranslation::DialogTranslation(agi::Context *c)
 	}
 	else
 		UpdateDisplay();
+
+	c->selectionController->AddSelectionListener(this);
 }
 
-DialogTranslation::~DialogTranslation() { }
+DialogTranslation::~DialogTranslation() {
+	c->selectionController->RemoveSelectionListener(this);
+}
+
+void DialogTranslation::OnActiveLineChanged(AssDialogue *new_line) {
+	if (switching_lines) return;
+
+	active_line = new_line;
+	active_line->ParseASSTags();
+	cur_block = 0;
+	line_number = count_if(c->ass->Line.begin(), find(c->ass->Line.begin(), c->ass->Line.end(), active_line), cast<AssDialogue*>()) + 1;
+
+	if (bad_block(active_line->Blocks[cur_block]) && !NextBlock()) {
+		wxMessageBox(_("No more lines to translate."));
+		EndModal(1);
+	}
+}
+
+void DialogTranslation::OnExternalCommit(int commit_type) {
+	if (commit_type == AssFile::COMMIT_NEW || commit_type & AssFile::COMMIT_DIAG_ADDREM) {
+		line_count = count_if(c->ass->Line.begin(), c->ass->Line.end(), cast<AssDialogue*>());
+		line_number_display->SetLabel(wxString::Format(_("Current line: %d/%d"), (int)line_number, (int)line_count));
+	}
+
+	if (commit_type & AssFile::COMMIT_DIAG_TEXT)
+		OnActiveLineChanged(active_line);
+}
 
 bool DialogTranslation::NextBlock() {
+	switching_lines = true;
 	do {
 		if (cur_block == active_line->Blocks.size() - 1) {
 			c->selectionController->NextLine();
@@ -184,12 +215,14 @@ bool DialogTranslation::NextBlock() {
 		else
 			++cur_block;
 	} while (bad_block(active_line->Blocks[cur_block]));
+	switching_lines = false;
 
 	UpdateDisplay();
 	return true;
 }
 
 bool DialogTranslation::PrevBlock() {
+	switching_lines = true;
 	do {
 		if (cur_block == 0) {
 			c->selectionController->PrevLine();
@@ -205,6 +238,7 @@ bool DialogTranslation::PrevBlock() {
 		else
 			--cur_block;
 	} while (bad_block(active_line->Blocks[cur_block]));
+	switching_lines = false;
 
 	UpdateDisplay();
 	return true;
@@ -245,7 +279,10 @@ void DialogTranslation::Commit(bool next) {
 	new_value.Replace("\n", "\\N");
 	*active_line->Blocks[cur_block] = AssDialogueBlockPlain(new_value);
 	active_line->UpdateText();
+
+	file_change_connection.Block();
 	c->ass->Commit(_("translation assistant"), AssFile::COMMIT_DIAG_TEXT);
+	file_change_connection.Unblock();
 
 	if (next) {
 		if (!NextBlock()) {
diff --git a/aegisub/src/dialog_translation.h b/aegisub/src/dialog_translation.h
index 7831598ba..74df386c3 100644
--- a/aegisub/src/dialog_translation.h
+++ b/aegisub/src/dialog_translation.h
@@ -26,6 +26,9 @@
 
 #include <libaegisub/exception.h>
 #include <libaegisub/scoped_ptr.h>
+#include <libaegisub/signal.h>
+
+#include "selection_controller.h"
 
 namespace agi { struct Context; }
 class AssDialogue;
@@ -39,9 +42,11 @@ class wxCheckBox;
 /// @brief Assistant for translating subtitles in one language to another language
 ///
 /// DOCME
-class DialogTranslation : public wxDialog {
+class DialogTranslation : public wxDialog, private SelectionListener<AssDialogue> {
 	agi::Context *c;
 
+	agi::signal::Connection file_change_connection;
+
 	/// The active line
 	AssDialogue *active_line;
 	/// Which dialogue block in the active line is currently being translated
@@ -52,6 +57,9 @@ class DialogTranslation : public wxDialog {
 	/// Line number of the currently active line
 	size_t line_number;
 
+	/// Should active line change announcements be ignored?
+	bool switching_lines;
+
 	wxStaticText *line_number_display;
 	ScintillaTextCtrl *original_text;
 	SubsTextEditCtrl *translated_text;
@@ -62,9 +70,13 @@ class DialogTranslation : public wxDialog {
 	void OnPlayAudioButton(wxCommandEvent &);
 	void OnPlayVideoButton(wxCommandEvent &);
 	void OnKeyDown(wxKeyEvent &evt);
+	void OnExternalCommit(int commit_type);
 
 	void UpdateDisplay();
 
+	void OnSelectedSetChanged(Selection const&, Selection const&) { }
+	void OnActiveLineChanged(AssDialogue *new_line);
+
 public:
 	DialogTranslation(agi::Context *context);
 	~DialogTranslation();