* a bit of code clean up

* some fixes to make screen_blanker wait_for_thread() instead of
  killing the screen saver thread, BScreenSaver::Stop() is now called
  as well as its destructor


git-svn-id: file:///srv/svn/repos/haiku/haiku/trunk@17206 a95241bf-73f2-0310-859d-f6bbb57e9c96

headers/private/screen_saver/ScreenSaverThread.h

@@ -12,16 +12,21 @@ class BScreenSaver;
 class BView;
 class ScreenSaverPrefs;
 
-class ScreenSaverThread 
+class ScreenSaverThread  {
-{
  public:
-	ScreenSaverThread(BWindow *wnd, BView *vw, ScreenSaverPrefs *p);
+								ScreenSaverThread(BWindow* window,
+												  BView* view,
+												  ScreenSaverPrefs* prefs);
+								~ScreenSaverThread();
+
 			void				Thread();
 			BScreenSaver*		LoadAddOn();
-	void Quit();
+			void				Quit(thread_id thread);
 
 	static	int32				ThreadFunc(void* data);
  private:
+			void				_CleanUp();
+
 			BScreenSaver*		fSaver;
 			BWindow*			fWin;
 			BDirectWindow*		fDWin;
@@ -31,6 +36,8 @@ private:
 			long				fFrame;
 			int					fSnoozeCount;
 			image_id			fAddonImage;
+
+	volatile bool				fQuitting;
 };
 
 #endif //SCREEN_SAVER_THREAD_H

src/bin/screen_blanker/ScreenSaverApp.cpp

@@ -114,6 +114,7 @@ ScreenSaverApp::MessageReceived(BMessage *message)
 			} else  {
 				PRINT(("Quitting!\n"));
 				_Shutdown();
+				Quit();
 			}
 			break;
 		}
@@ -144,26 +145,29 @@ ScreenSaverApp::QuitRequested()
 		&& (system_time() - fBlankTime > (fPrefs.PasswordTime() - fPrefs.BlankTime()))) {
 		_ShowPasswordWindow();
 		return false;
-	} else
+	}
-		_Shutdown();
 
+	_Shutdown();
 	return true;
 }
 
 
 void
-ScreenSaverApp::_Shutdown(void) 
+ScreenSaverApp::_Shutdown() 
 {
-	if (fWindow)
+	if (fThread) {
-		fWindow->Hide();
+		fThread->Quit(fThreadID);
-	if (fThreadID > -1) {
+		delete fThread;
-		// TODO: kill_thread()? How about doing this right??
+	} else if (fThreadID > -1) {
+		// ?!?
 		kill_thread(fThreadID);
 	}
-	if (fThread)
-		delete fThread;
 
-	Quit();
+	fThread = NULL;
+	fThreadID = -1;
+
+	if (fWindow)
+		fWindow->Hide();
 }
 
 

src/kits/screensaver/ScreenSaverThread.cpp

@@ -23,26 +23,40 @@ ScreenSaverThread::ThreadFunc(void *data)
 }
 
 
-ScreenSaverThread::ScreenSaverThread(BWindow *wnd, BView *vw, ScreenSaverPrefs *p) 
+ScreenSaverThread::ScreenSaverThread(BWindow* window,
+									 BView* view,
+									 ScreenSaverPrefs* prefs) 
 	:	fSaver(NULL),
-		fWin(wnd),
+		fWin(window),
-		fView(vw), 
+		fView(view),
-		fPref(p),
+		fPref(prefs),
 		fFrame(0),
 		fSnoozeCount(0),
-		fAddonImage(0) 
+		fAddonImage(-1),
+
+		fQuitting(false)
 {
-	fDWin = dynamic_cast<BDirectWindow *>(wnd);
+	fDWin = dynamic_cast<BDirectWindow *>(fWin);
+}
+
+
+ScreenSaverThread::~ScreenSaverThread()
+{
+	_CleanUp();
 }
 
 
 void 
-ScreenSaverThread::Quit() 
+ScreenSaverThread::Quit(thread_id id)
 {
+	fQuitting = true;
+	if (id >= 0) {
+		status_t threadRetValue;
+		wait_for_thread(id, &threadRetValue);
+	}
+
 	if (fSaver)
 		fSaver->StopSaver();
-	if (fWin)
-		fWin->Hide();
 }
 
 
@@ -56,15 +70,18 @@ ScreenSaverThread::Thread()
 			fSaver->StartSaver(fView, false);
 		fWin->Unlock();
 	}
-	while (1) {
+	while (!fQuitting) {
 		snooze(fSaver->TickSize());
-		if (fSnoozeCount) { // If we are sleeping, do nothing
+		if (fSnoozeCount) {
+			// if we are sleeping, do nothing
 			fSnoozeCount--;
 			return;
 		} else if (fSaver->LoopOnCount() && (fFrame >= fSaver->LoopOnCount())) { // Time to nap
 			fFrame = 0;
 			fSnoozeCount = fSaver->LoopOffCount();
 		} else if (fWin->Lock()) {
+			// NOTE: R5 really calls DirectDraw()
+			// and then Draw() for the same frame
 			if (fDWin)
 				fSaver->DirectDraw(fFrame);
 			fSaver->Draw(fView, fFrame);
@@ -81,10 +98,11 @@ ScreenSaverThread::LoadAddOn()
 	if (strcmp("", fPref->ModuleName()) == 0)
 		return NULL;
 
+	// This is a new set of preferences. Free up what we did have
+	_CleanUp();
+
 	BScreenSaver* (*instantiate)(BMessage*, image_id);
-	if (fAddonImage) { // This is a new set of preferences. Free up what we did have 
+
-		unload_add_on(fAddonImage);
-	}
 	char temp[B_PATH_NAME_LENGTH]; 
 	if (B_OK==find_directory(B_BEOS_ADDONS_DIRECTORY, 0, false, temp, B_PATH_NAME_LENGTH)) { 
 		sprintf (temp,"%s/Screen Savers/%s", temp, fPref->ModuleName());
@@ -132,3 +150,14 @@ ScreenSaverThread::LoadAddOn()
 	return fSaver;
 }
 
+
+void
+ScreenSaverThread::_CleanUp()
+{
+	delete fSaver;
+	fSaver = NULL;
+	if (fAddonImage >= 0) {
+		unload_add_on(fAddonImage);
+		fAddonImage = -1;
+	}
+}