EpisodeListView: Use FileView consistently.

1 files changed, 87 insertions, 27 deletions

diff --git a/c/episodelistview.cpp b/c/episodelistview.cpp
index 3f2104f..0451132 100644
--- a/c/episodelistview.cpp
+++ b/c/episodelistview.cpp
@@ -13,7 +13,8 @@
 #include "win.h"
 
 EpisodeListView::EpisodeListView(const HWND hWndParent)
-	: ListView(hWndParent, reinterpret_cast<HMENU>(IDC_EPISODELISTVIEW), 0)
+	: ListView(hWndParent, reinterpret_cast<HMENU>(IDC_EPISODELISTVIEW), 0),
+	  m_fv(L"elvdata.dat", sizeof(ElvDataA)*8192)
 {
 	LVCOLUMN lvc;
 
@@ -45,6 +46,69 @@ void EpisodeListView::EnsureFocusVisible()
 		ListView_EnsureVisible(hWnd, iEpFocus, TRUE);
 }
 
+void EpisodeListView::HandleContextMenu(const WORD command)
+{
+	int cNotFound = 0;
+
+	/* Look through selected items, applying the
+	 * selected command to each one. */
+
+	LVITEM lvi = {LVIF_PARAM, -1};
+	while (FindNextItem(&lvi, LVNI_SELECTED)) {
+		ElvDataA& e = static_cast<ElvDataA*>(m_fv)[lvi.lParam-1];
+
+		if (ID_SUBGROUP(command) == IDG_CTX_RATE) {
+			/* Process rate commands. */
+			if (e.rating = ID_RATING(command))
+				Swprintf(e.sRating, L"%d", e.rating);
+			else
+				e.sRating[0] = 0;
+		} else {
+			/* Process other commands. */
+			switch (command) {
+			case IDM_WATCH_LOCALLY:
+				if (!Pl("local_episode","open_episode_locally",lvi.lParam))
+					cNotFound++;
+				break;
+
+			case IDM_WATCH_ONLINE:
+				Pl("local_episode","open_episode_online",lvi.lParam);
+				break;
+
+			case IDM_TOGGLE:
+				e.bWatched = !e.bWatched;
+				break;
+
+			case IDM_FORGET:
+				Pl("track_episodes","forget_episode",lvi.lParam);
+				Pl("track_episodes","update_tracked_episodes");
+				break;
+
+			case IDM_WIKI:
+				Pl("episode_data","open_episode_wiki",lvi.lParam);
+				break;
+			}
+		}
+	}
+
+	Redraw();
+
+	if (ID_SUBGROUP(command) == IDG_CTX_RATE) {
+		/* If ratings changed, the episodes may need to be resorted. */
+		Sort();
+		ShowFocus();
+	} else if (cNotFound) {
+		/* Show warning if local episodes were not found. */
+		if (cNotFound == 1)
+			EBMessageBox(L"Episode could not be opened locally.", L"Error", MB_ICONWARNING);
+		else {
+			wchar_t msg[64] = {0};
+			Swprintf(msg, L"%d episodes could not be opened locally.", cNotFound);
+			EBMessageBox(msg, L"Error", MB_ICONWARNING);
+		}
+	}
+}
+
 LRESULT EpisodeListView::HandleNotify(const LPARAM lParam)
 {
 	const NMLISTVIEW* const nm = reinterpret_cast<NMLISTVIEW*>(lParam);
@@ -53,7 +117,7 @@ LRESULT EpisodeListView::HandleNotify(const LPARAM lParam)
 	case LVN_GETDISPINFO:	/* Display item. */
 	    {
 		NMLVDISPINFO* const nm = reinterpret_cast<NMLVDISPINFO*>(lParam);
-		ElvDataA& e = m_vData.at(nm->item.lParam-1);
+		ElvDataA& e = reinterpret_cast<ElvDataA*>(m_fv.view)[nm->item.lParam-1];
 		wchar_t* vs[] = {e.siEp, e.title, e.sRating}; /* ELVSIEPISODE, ELVSITITLE, ELVSIRATING */
 		nm->item.pszText = vs[nm->item.iSubItem];
 		return 0;
@@ -101,12 +165,15 @@ LRESULT EpisodeListView::HandleNotify(const LPARAM lParam)
 			return CDRF_NOTIFYITEMDRAW;
 			break;
 		case CDDS_ITEMPREPAINT:
-			if (!Pl("track_episodes","watched",nm->nmcd.lItemlParam)) {
+		    {
+			const ElvDataA& e = static_cast<ElvDataA*>(m_fv)[nm->nmcd.lItemlParam-1];
+			if (!e.bWatched) {
 				extern HFONT g_hfBold;
 				Require(SelectObject(nm->nmcd.hdc, g_hfBold));
 				return CDRF_NEWFONT;
 			}
 			break;
+		    }
 		}
 		return 0;
 	    }
@@ -218,7 +285,7 @@ void EpisodeListView::SelectUnwatched(int dir)
 	lvfi.lParam = lviFocus.lParam;
 
 	do {
-		if (!Pl("track_episodes",dir > 0? "push_unwatched": "previous_unwatched",
+		if (!Pl("track_episodes",dir > 0? "next_unwatched": "previous_unwatched",
 			lvfi.lParam,&iEpNew))
 			return;
 
@@ -262,30 +329,26 @@ int CALLBACK EpisodeListView::SortProc(const LPARAM iItem1, const LPARAM iItem2,
 	 * If m_iSortCol is negative, the order is descending. */
 	const int order = Cmp(elv->m_iSortCol, 0);
 
+	const ElvDataA& e1 = static_cast<ElvDataA*>(elv->m_fv)[lvi1.lParam-1];
+	const ElvDataA& e2 = static_cast<ElvDataA*>(elv->m_fv)[lvi2.lParam-1];
+
 	switch (abs(elv->m_iSortCol)-1) {
 	case ELVSIEPISODE:
 		return order*Cmp(lvi1.lParam, lvi2.lParam);
 	case ELVSIRATING:
 	    {
-		int rating1, rating2;
-		rating1 = elv->m_iSortCol > 0? 99: -1;
-		rating2 = elv->m_iSortCol > 0? 99: -1;
-		Pl("episode_data","episode_rating",lvi1.lParam,&rating1);
-		Pl("episode_data","episode_rating",lvi2.lParam,&rating2);
+		int rating1 = e1.rating;
+		int rating2 = e2.rating;
+		if (!rating1)
+			rating1 = elv->m_iSortCol > 0? 99: -1;
+		if (!rating2)
+			rating2 = elv->m_iSortCol > 0? 99: -1;
 		if (rating1 == rating2)
 			return Cmp(lvi1.lParam, lvi2.lParam);
 		return order*Cmp(rating1, rating2);
 	    }
 	case ELVSITITLE:
-	    {
-		Mark m;
-		WcharPtr s1, s2;
-		if (!Pl("episode_data","episode_title",lvi1.lParam,&s1))
-			return 0;
-		if (!Pl("episode_data","episode_title",lvi2.lParam,&s2))
-			return 0;
-		return order*_wcsicmp(s1, s2);
-	    }
+		return order*_wcsicmp(e1.title, e2.title);
 	default:
 		return 0;
 	}
@@ -333,24 +396,23 @@ void EpisodeListView::Update()
 
 		/* Retrieve episode data and add list view items. */
 		SendMessage(hWnd, WM_SETREDRAW, FALSE, 0);
-		m_vData.clear();
 		ListView_DeleteAllItems(hWnd);
 		for (int iEp = 1; iEp <= cEp; iEp++) {
-			ElvDataA e;
+			ElvDataA e = static_cast<ElvDataA*>(m_fv)[iEp-1];
 
-			if (!FromProlog(iEp, e))
-				goto push;
+			if (!e.siEp[0])
+				continue;
 
 			if (extern char g_limitScreenwriter[];
 			    g_limitScreenwriter[0] && !Pl("episode_data","episode_datum",iEp,
 			    "Screenwriter",g_limitScreenwriter))
-				goto push;
+				continue;
 
 			if (extern int g_bViewWatched; !g_bViewWatched && e.bWatched)
-				goto push;
+				continue;
 
 			if (extern int g_bViewTVOriginal; !g_bViewTVOriginal && e.bTVOriginal)
-				goto push;
+				continue;
 
 			/* Insert item. */
 			lviEpisode.iItem = cItem++;
@@ -361,8 +423,6 @@ void EpisodeListView::Update()
 
 			ListView_SetItemText(hWnd, lviEpisode.iItem, ELVSITITLE, LPSTR_TEXTCALLBACK);
 			ListView_SetItemText(hWnd, lviEpisode.iItem, ELVSIRATING, LPSTR_TEXTCALLBACK);
-
-		push:	m_vData.push_back(std::move(e));
 		}
 
 		/* Show number of displayed items in status bar. */