1

devtools/syncfrommirror/Redirect.cpp

@@ -0,0 +1,152 @@
+//========= Copyright Valve Corporation, All rights reserved. ============//
+//
+// Purpose: 
+//
+// $NoKeywords: $
+//
+//=============================================================================//
+//------------------------------------------------------------------------------
+// Redirect.cpp : implementation file
+//
+// Creates a child process that runs a user-specified command and redirects its
+// standard output and standard error to a CEdit control.
+//
+// Written by Matt Brunk (brunk@gorge.net)
+// Copyright (C) 1999 Matt Brunk
+// All rights reserved.
+//
+// This code may be used in compiled form in any way. This file may be
+// distributed by any means providing it is not sold for profit without
+// the written consent of the author, and providing that this notice and the
+// author's name is included in the distribution. If the compiled form of the
+// source code in this file is used in a commercial application, an e-mail to
+// the author would be appreciated.
+//
+// Thanks to Dima Shamroni (dima@abirnet.co.il) for providing the essential
+// code for this class.
+//
+// Thanks to Chris Maunder (chrismaunder@codeguru.com) for the PeekAndPump()
+// function (from his CProgressWnd class).
+//
+// Initial Release Feb 8, 1999
+//------------------------------------------------------------------------------
+
+
+#include "stdafx.h"
+#include <string.h>
+#include <string>
+#include "Redirect.h"
+
+const int BUF_SIZE = 8192;
+
+CRedirect::~CRedirect()
+{
+}
+
+void CRedirect::Run( LPCTSTR szCommand, CEdit *pEdit, LPCTSTR pszCurrentDirectory )
+{
+	m_pEdit = pEdit;
+	m_bStopped = false;
+
+	if ( !Open( szCommand, pszCurrentDirectory ) )
+	{
+		AppendText("\nFAILED TO SPAWN PROCESS\n");
+		return;
+	}
+
+	for (;;)
+	{ 
+		//------------------------------------------------------------------
+		//	If the child process has completed, break out.
+		//------------------------------------------------------------------
+		if ( !m_hThread || MsgWaitForMultipleObjects( 1, &m_hThread, FALSE, 20, QS_ALLINPUT) == WAIT_OBJECT_0 )
+		{
+			break;
+		}
+
+		//------------------------------------------------------------------
+		//	Peek and pump messages.
+		//------------------------------------------------------------------
+		PeekAndPump();
+
+		//------------------------------------------------------------------
+		//	If the user cancelled the operation, terminate the process.
+		//------------------------------------------------------------------
+		if ( m_bStopped )
+		{
+			Close();
+		}
+	}
+}
+
+
+void CRedirect::PeekAndPump()
+{
+    MSG Msg;
+    while (::PeekMessage(&Msg, NULL, 0, 0, PM_NOREMOVE)) 
+    {
+        (void)AfxGetApp()->PumpMessage(); //lint !e1924 (warning about C-style cast)
+    }
+}
+
+void CRedirect::Stop()
+{
+	m_bStopped = true;
+}
+
+void CRedirect::AppendText(LPCTSTR Text)
+{
+	char *pszCurrent = const_cast<char *>(Text);
+	int Length;
+
+	Length = m_pEdit->GetWindowTextLength();
+	if ( Length + strlen(Text) > m_pEdit->GetLimitText() )
+		m_pEdit->SetLimitText( m_pEdit->GetLimitText() * 2 );
+
+	if ( *pszCurrent == '\n' )
+		pszCurrent++;
+
+	while ( pszCurrent && *pszCurrent )
+	{
+		char *pszLineEnd = strchr( pszCurrent, '\r' );
+		if ( pszLineEnd )
+			*pszLineEnd = 0;
+
+		Length = m_pEdit->GetWindowTextLength();
+		m_pEdit->SetSel(Length, Length);
+		m_pEdit->ReplaceSel(pszCurrent);
+		//OutputDebugString(pszCurrent);
+
+		if ( pszLineEnd )
+		{
+			Length = m_pEdit->GetWindowTextLength();
+			m_pEdit->SetSel(Length, Length);
+			m_pEdit->ReplaceSel("\r\n");
+			//OutputDebugString("\r\n");
+		}
+
+		m_pEdit->LineScroll( m_pEdit->GetLineCount() );
+		PeekAndPump();
+
+		if ( pszLineEnd )
+		{
+			*pszLineEnd = '\r';
+			pszCurrent = pszLineEnd + 1;
+			if ( *pszCurrent == '\n' )
+				pszCurrent++;
+		}
+		else
+			pszCurrent = NULL;
+	};
+}
+
+
+void CRedirect::WriteStdOut(LPCSTR pszOutput)
+{
+	AppendText( pszOutput );
+}
+
+void CRedirect::WriteStdError(LPCSTR pszError)
+{
+	AppendText( CString(pszError) );
+}