csgo/cstrike15_src/utils/FileSystemOpenDialog/FileSystemOpenDlg.cpp

//===== Copyright � 1996-2005, Valve Corporation, All rights reserved. ======//
//
// Purpose: 
//
// $NoKeywords: $
//
//===========================================================================//
// FileSystemOpenDlg.cpp : implementation file
//

#include "stdafx.h"
#include "FileSystemOpenDlg.h"
#include "jpeglib/jpeglib.h"
#include "utldict.h"
#include "resource.h"
#include "tier2/tier2.h"
#include "ifilesystemopendialog.h"
#include "smartptr.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif


CFileInfo::CFileInfo()
{
	m_pBitmap = NULL;
}


CFileInfo::~CFileInfo()
{
}


/////////////////////////////////////////////////////////////////////////////
// This caches the thumbnail bitmaps we generate to speed up browsing.
/////////////////////////////////////////////////////////////////////////////

class CBitmapCache
{
public:
	CBitmapCache()
	{
		m_CurMemoryUsage = 0;
		m_MaxMemoryUsage = 1024 * 1024 * 6;
	}

	void AddToCache( CBitmap *pBitmap, const char *pName, int memoryUsage, bool bLock )
	{
		Assert( m_Bitmaps.Find( pName ) == -1 );
		m_CurMemoryUsage += memoryUsage;

		CBitmapCacheEntry newEntry;
		newEntry.m_pBitmap = pBitmap;
		newEntry.m_MemoryUsage = memoryUsage;
		newEntry.m_bLocked = bLock;
		m_Bitmaps.Insert( pName, newEntry );

		EnsureMemoryUsage();
	}

	CBitmap* Find( const char *pName )
	{
		int i = m_Bitmaps.Find( pName );
		if ( i == -1 )
			return NULL;
		else
			return m_Bitmaps[i].m_pBitmap;
	}

	void UnlockAll()
	{
		for ( int i=m_Bitmaps.First(); i != m_Bitmaps.InvalidIndex(); i=m_Bitmaps.Next( i ) )
		{
			m_Bitmaps[i].m_bLocked = false;
		}
	}

private:

	void EnsureMemoryUsage()
	{
		while ( m_CurMemoryUsage > m_MaxMemoryUsage )
		{
			// Free something.
			bool bFreed = false;
			for ( int i=m_Bitmaps.First(); i != m_Bitmaps.InvalidIndex(); i=m_Bitmaps.Next( i ) )
			{
				if ( !m_Bitmaps[i].m_bLocked )
				{
					delete m_Bitmaps[i].m_pBitmap;
					m_CurMemoryUsage -= m_Bitmaps[i].m_MemoryUsage;
					m_Bitmaps.RemoveAt( i );
					break;
				}
			}
			
			// Nothing left to free?			
			if ( !bFreed )
				return;
		}
	}

private:

	class CBitmapCacheEntry
	{
	public:
		CBitmap *m_pBitmap;
		int m_MemoryUsage;
		bool m_bLocked;
	};
	
	CUtlDict<CBitmapCacheEntry,int> m_Bitmaps;
	int m_CurMemoryUsage;
	int m_MaxMemoryUsage;
};

CBitmapCache g_BitmapCache;


/////////////////////////////////////////////////////////////////////////////
// CFileSystemOpenDlg dialog

CFileSystemOpenDlg::CFileSystemOpenDlg(CreateInterfaceFn factory, CWnd* pParent )
	: CDialog(CFileSystemOpenDlg::IDD, pParent)
{
	//{{AFX_DATA_INIT(CFileSystemOpenDlg)
	//}}AFX_DATA_INIT
	m_pFileSystem = (IFileSystem*)factory( FILESYSTEM_INTERFACE_VERSION, NULL );
	if ( !m_pFileSystem )
	{
		Error( "Unable to connect to %s!\n", FILESYSTEM_INTERFACE_VERSION );
	}

	m_bFilterMdlAndJpgFiles = false;
}

void CFileSystemOpenDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CFileSystemOpenDlg)
	DDX_Control(pDX, IDC_FILENAME_LABEL, m_FilenameLabel);
	DDX_Control(pDX, IDC_FILENAME, m_FilenameControl);
	DDX_Control(pDX, IDC_LOOKIN, m_LookInLabel);
	DDX_Control(pDX, IDC_FILE_LIST, m_FileList);
	//}}AFX_DATA_MAP
}


BEGIN_MESSAGE_MAP(CFileSystemOpenDlg, CDialog)
	//{{AFX_MSG_MAP(CFileSystemOpenDlg)
	ON_WM_CREATE()
	ON_WM_SIZE()
	ON_NOTIFY(NM_DBLCLK, IDC_FILE_LIST, OnDblclkFileList)
	ON_BN_CLICKED(IDC_UP_BUTTON, OnUpButton)
	ON_NOTIFY(LVN_ITEMCHANGED, IDC_FILE_LIST, OnItemchangedFileList)
	ON_WM_KEYDOWN()
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()


/////////////////////////////////////////////////////////////////////////////
// CFileSystemOpenDlg message handlers

void CFileSystemOpenDlg::OnOK() 
{
	// Make sure it's a valid filename.
	CString testFilename;
	m_FilenameControl.GetWindowText( testFilename );

	CString fullFilename = m_CurrentDir + "\\" + testFilename;
 	if ( m_pFileSystem->IsDirectory( fullFilename, GetPathID() ) )
	{
		m_CurrentDir = fullFilename;
		PopulateListControl();
	}
	else if ( m_pFileSystem->FileExists( fullFilename, GetPathID() ) )
	{
		m_Filename = fullFilename;
		
		// Translate .jpg to .mdl?
		if ( m_bFilterMdlAndJpgFiles )
		{
			char tempFilename[MAX_PATH];
			Q_strncpy( tempFilename, fullFilename, sizeof( tempFilename ) );
			char *pPos = strrchr( tempFilename, '.' );
			if ( pPos )
			{
				if ( Q_stricmp( pPos, ".jpeg" ) == 0 || Q_stricmp( pPos, ".jpg" ) == 0 )
				{
					Q_strncpy( pPos, ".mdl", 5 );
					m_Filename = tempFilename;
				}
			}
		}

		EndDialog( IDOK );
	}
	else
	{
		// No file or directory here.
		CString str;
		str.FormatMessage( "File %1!s! doesn't exist.", (const char*)fullFilename );
		AfxMessageBox( str, MB_OK );
	}
}

void CFileSystemOpenDlg::SetInitialDir( const char *pDir, const char *pPathID )
{
	m_CurrentDir = pDir;
	if ( pPathID )
		m_PathIDString = pPathID;
	else
		m_PathIDString = "";
}

CString CFileSystemOpenDlg::GetFilename() const
{
	return m_Filename;
}


BOOL CFileSystemOpenDlg::OnInitDialog() 
{
	CDialog::OnInitDialog();
	
	// Setup our anchor list.
	AddAnchor( IDC_FILE_LIST, 2, 2 );
	AddAnchor( IDC_FILE_LIST, 3, 3 );
	
	AddAnchor( IDC_FILENAME, 1, 3 );
	AddAnchor( IDC_FILENAME, 3, 3 );
	AddAnchor( IDC_FILENAME, 2, 2 );

	AddAnchor( IDC_FILENAME_LABEL, 0, 0 );
	AddAnchor( IDC_FILENAME_LABEL, 2, 0 );
	AddAnchor( IDC_FILENAME_LABEL, 1, 3 );
	AddAnchor( IDC_FILENAME_LABEL, 3, 3 );

	AddAnchor( IDOK, 0, 2 );
	AddAnchor( IDOK, 2, 2 );
	AddAnchor( IDOK, 1, 3 );
	AddAnchor( IDOK, 3, 3 );
	
	AddAnchor( IDCANCEL, 0, 2 );
	AddAnchor( IDCANCEL, 2, 2 );
	AddAnchor( IDCANCEL, 1, 3 );
	AddAnchor( IDCANCEL, 3, 3 );

	AddAnchor( IDC_LOOKIN, 2, 2 );

	AddAnchor( IDC_UP_BUTTON, 0, 2 );
	AddAnchor( IDC_UP_BUTTON, 2, 2 );


	// Setup our image list.
	m_ImageList.Create( PREVIEW_IMAGE_SIZE, PREVIEW_IMAGE_SIZE, ILC_COLOR32, 0, 512 );
	
	m_BitmapFolder.LoadBitmap( IDB_LABEL_FOLDER );
	m_iLabel_Folder = m_ImageList.Add( &m_BitmapFolder, (CBitmap*)NULL );

	m_BitmapMdl.LoadBitmap( IDB_LABEL_MDL );
	m_iLabel_Mdl = m_ImageList.Add( &m_BitmapMdl, (CBitmap*)NULL );

	m_BitmapFile.LoadBitmap( IDB_LABEL_FILE );
	m_iLabel_File = m_ImageList.Add( &m_BitmapFile, (CBitmap*)NULL );

	m_FileList.SetImageList( &m_ImageList, LVSIL_NORMAL );

	// Populate the list with the contents of our current directory.
	PopulateListControl();
	
	return TRUE;  // return TRUE unless you set the focus to a control
	              // EXCEPTION: OCX Property Pages should return FALSE
}

void CFileSystemOpenDlg::GetEntries( const char *pMask, CUtlVector<CString> &entries, GetEntriesMode_t mode )
{
	CString searchStr = m_CurrentDir + "\\" + pMask;

	// Workaround Steam bug.
	if ( searchStr == ".\\*.*" )
		searchStr = "*.*";
	
	FileFindHandle_t handle;	
	const char *pFile = m_pFileSystem->FindFirst( searchStr, &handle );
	while ( pFile )
	{
		bool bIsDir = m_pFileSystem->FindIsDirectory( handle );
		if ( (mode == GETENTRIES_DIRECTORIES_ONLY && bIsDir) || (mode == GETENTRIES_FILES_ONLY && !bIsDir) )
		{
			entries.AddToTail( pFile );
		}

		pFile = m_pFileSystem->FindNext( handle );
	}
	m_pFileSystem->FindClose( handle );
}



class CJpegSourceMgr : public jpeg_source_mgr
{
public:
	CJpegSourceMgr()
	{
		this->init_source = &CJpegSourceMgr::imp_init_source;
		this->fill_input_buffer = &CJpegSourceMgr::imp_fill_input_buffer;
		this->skip_input_data = &CJpegSourceMgr::imp_skip_input_data;
		this->resync_to_restart = &CJpegSourceMgr::imp_resync_to_restart;
		this->term_source = &CJpegSourceMgr::imp_term_source;

		this->next_input_byte = 0;
		this->bytes_in_buffer = 0;
	}

	bool Init( IFileSystem *pFileSystem, FileHandle_t fp )
	{
		m_Data.SetSize( pFileSystem->Size( fp ) );
		return pFileSystem->Read( m_Data.Base(), m_Data.Count(), fp ) == m_Data.Count();
	}

	static void imp_init_source(j_decompress_ptr cinfo)
	{
	}

	static boolean imp_fill_input_buffer(j_decompress_ptr cinfo)
	{
		Assert( false ); // They should never need to call these functions since we give them all the data up front.
		return 0;
	}

	static void imp_skip_input_data(j_decompress_ptr cinfo, long num_bytes)
	{
		Assert( false ); // They should never need to call these functions since we give them all the data up front.
	}

	static boolean imp_resync_to_restart(j_decompress_ptr cinfo, int desired)
	{
		Assert( false ); // They should never need to call these functions since we give them all the data up front.
		return false;
	}

	static void imp_term_source(j_decompress_ptr cinfo)
	{
	}

public:
	CUtlVector<char> m_Data;
};


bool ReadJpeg( IFileSystem *pFileSystem, const char *pFilename, CUtlVector<unsigned char> &buf, int &width, int &height, const char *pPathID )
{
	// Read the data.
	FileHandle_t fp = pFileSystem->Open( pFilename, "rb", pPathID );
	if ( fp == FILESYSTEM_INVALID_HANDLE )
		return false;

	CJpegSourceMgr sourceMgr;
	bool bRet = sourceMgr.Init( pFileSystem, fp );
	pFileSystem->Close( fp );
	if ( !bRet )
		return false;

	sourceMgr.bytes_in_buffer = sourceMgr.m_Data.Count();
	sourceMgr.next_input_byte = (unsigned char*)sourceMgr.m_Data.Base();

	// Load the jpeg.
	struct jpeg_decompress_struct jpegInfo;
	struct jpeg_error_mgr jerr;

	memset( &jpegInfo, 0, sizeof( jpegInfo ) );
	jpegInfo.err = jpeg_std_error(&jerr);
	jpeg_create_decompress(&jpegInfo);
	jpegInfo.src = &sourceMgr;

	if (jpeg_read_header(&jpegInfo, TRUE) != JPEG_HEADER_OK)
	{
		return false;
	}	

	// start the decompress with the jpeg engine.
	if (jpeg_start_decompress(&jpegInfo) != TRUE || jpegInfo.output_components != 3)
	{
		jpeg_destroy_decompress(&jpegInfo);
		return false;
	}

	// now that we've started the decompress with the jpeg lib, we have the attributes of the
	// image ready to be read out of the decompress struct.
	int row_stride = jpegInfo.output_width * jpegInfo.output_components;
	int mem_required = jpegInfo.image_height * jpegInfo.image_width * jpegInfo.output_components;
	JSAMPROW row_pointer[1];
	int cur_row = 0;
	
	width = jpegInfo.output_width;
	height = jpegInfo.output_height;

	// allocate the memory to read the image data into.
	buf.SetSize( mem_required );

	// read in all the scan lines of the image into our image data buffer.
	bool working = true;
	while (working && (jpegInfo.output_scanline < jpegInfo.output_height))
	{
		row_pointer[0] = &(buf[cur_row * row_stride]);
		if (jpeg_read_scanlines(&jpegInfo, row_pointer, 1) != TRUE)
		{
			working = false;
		}
		++cur_row;
	}

	if (!working)
	{
		jpeg_destroy_decompress(&jpegInfo);
		return false;
	}

	jpeg_finish_decompress(&jpegInfo);
	return true;
}

void DownsampleRGBToRGBAImage( 
	CUtlVector<unsigned char> &srcData,
	int srcWidth,
	int srcHeight,
	CUtlVector<unsigned char> &destData,
	int destWidth,
	int destHeight )
{
	int srcPixelSize = 3;
	int destPixelSize = 4;
	destData.SetSize( destWidth * destHeight * destPixelSize );
	memset( destData.Base(), 0xFF, destWidth * destHeight * destPixelSize );

	// This preserves the aspect ratio of the image.
	int scaledDestWidth = destWidth;
	int scaledDestHeight = destHeight;
	int destOffsetX = 0, destOffsetY = 0;
	if ( srcWidth > srcHeight )
	{
		scaledDestHeight = (srcHeight * destHeight) / srcWidth;
		destOffsetY = (destHeight - scaledDestHeight) / 2;
	}
	else if ( srcHeight > srcWidth )
	{
		scaledDestWidth = (srcWidth * destWidth) / srcHeight;
		destOffsetX = (destWidth - scaledDestWidth) / 2;
	}

	for ( int destY=0; destY < scaledDestHeight; destY++ )
	{
		unsigned char *pDestLine = &destData[(destY + destOffsetY) * destWidth * destPixelSize + (destOffsetX * destPixelSize)];
		unsigned char *pDestPos = pDestLine;

		float destYPercent = (float)destY / (scaledDestHeight-1);
		int srcY = (int)( destYPercent * (srcHeight-1) );

		for ( int destX=0; destX < scaledDestWidth; destX++ )
		{
			float destXPercent = (float)destX / (scaledDestWidth-1);

			int srcX = (int)( destXPercent * (srcWidth-1) );
			unsigned char *pSrcPos = &srcData[(srcY * srcWidth + srcX) * srcPixelSize];
			pDestPos[0] = pSrcPos[2];
			pDestPos[1] = pSrcPos[1];
			pDestPos[2] = pSrcPos[0];
			pDestPos[3] = 255;

			pDestPos += destPixelSize;
		}
	}
}


CBitmap* SetupJpegLabel( IFileSystem *pFileSystem, CString filename, int labelSize, const char *pPathID )
{
	CBitmap *pBitmap = g_BitmapCache.Find( filename );
	if ( pBitmap )
		return pBitmap;

	CUtlVector<unsigned char> data;
	int width, height;
	if ( !ReadJpeg( pFileSystem, filename, data, width, height, pPathID ) )
		return NULL;

	CUtlVector<unsigned char> downsampled;
	DownsampleRGBToRGBAImage( data, width, height, downsampled, labelSize, labelSize );

	pBitmap = new CBitmap;
	if ( pBitmap->CreateBitmap( labelSize, labelSize, 1, 32, downsampled.Base() ) )
	{
		g_BitmapCache.AddToCache( pBitmap, filename, downsampled.Count(), true );
		return pBitmap;
	}
	else
	{
		delete pBitmap;
		return NULL;
	}
}

int CFileSystemOpenDlg::SetupLabelImage( CFileInfo *pInfo, CString name, bool bIsDir )
{
	if ( bIsDir )
		return m_iLabel_Folder;

	CString extension = name.Right( 4 );
	extension.MakeLower();
	if ( extension == ".jpg" || extension == ".jpeg" )
	{
		pInfo->m_pBitmap = SetupJpegLabel( m_pFileSystem, m_CurrentDir + "\\" + name, PREVIEW_IMAGE_SIZE, GetPathID() );
		if ( pInfo->m_pBitmap )
			return m_ImageList.Add( pInfo->m_pBitmap, (CBitmap*)NULL );
		else
			return m_iLabel_File;
	}
	else
	{
		return (extension == ".mdl") ? m_iLabel_Mdl : m_iLabel_File;
	}
}

void FilterMdlAndJpgFiles( CUtlVector<CString> &files )
{
	// Build a dictionary with all the .jpeg files.
	CUtlDict<int,int> jpgFiles;
	for ( int i=0; i < files.Count(); i++ )
	{
		CString extension = files[i].Right( 4 );
		extension.MakeLower();
		if ( extension == ".jpg" || extension == ".jpeg" )
		{
			CString base = files[i].Left( files[i].GetLength() - 4 );
			jpgFiles.Insert( base, 1 );
		}
	}

	// Now look for all mdls and remove them if they have a jpg.
	for ( int i=0; i < files.Count(); i++ )
	{
		CString extension = files[i].Right( 4 );
		extension.MakeLower();
		if ( extension == ".mdl" )
		{
			CString base = files[i].Left( files[i].GetLength() - 4 );
			if ( jpgFiles.Find( base ) != -1 )
			{
				files.Remove( i );
				--i;
			}
		}		
	}
}


int CALLBACK FileListSortCallback( LPARAM lParam1, LPARAM lParam2, LPARAM lParamSort )
{
	CFileSystemOpenDlg *pDlg = (CFileSystemOpenDlg*)lParamSort;
	CFileInfo *pInfo1 = &pDlg->m_FileInfos[lParam1];
	CFileInfo *pInfo2 = &pDlg->m_FileInfos[lParam2];
	if ( pInfo1->m_bIsDir != pInfo2->m_bIsDir )
		return pInfo1->m_bIsDir ? -1 : 1;

	return Q_stricmp( pInfo1->m_Name, pInfo2->m_Name );
}


void RemoveDuplicates( CUtlVector<CString> &files )
{
	CUtlDict<int,int> uniqueFilenames;
	for ( int i=0; i < files.Count(); i++ )
	{
		int iPreviousIndex = uniqueFilenames.Find( files[i] );
		if ( iPreviousIndex == -1 )
		{
			uniqueFilenames.Insert( files[i], i );
		}
		else
		{
			files.Remove( i );
			--i;
		}
	}
}	
		

void CFileSystemOpenDlg::PopulateListControl()
{
	m_FileList.DeleteAllItems();
	g_BitmapCache.UnlockAll();
	m_LookInLabel.SetWindowText( CString( "[ROOT]\\" ) + m_CurrentDir );

	int iItem = 0;

	// First add directories at the top.
	CUtlVector<CString> directories;
	GetEntries( "*.*", directories, GETENTRIES_DIRECTORIES_ONLY );
	RemoveDuplicates( directories );

	for ( int i=0; i < directories.Count(); i++ )
	{
		if ( directories[i] == "." || directories[i] == ".." )
			continue;

		LVITEM item;
		item.mask = LVIF_TEXT | LVIF_IMAGE | LVIF_PARAM;
		item.iItem = iItem++;
		item.iSubItem = 0;
		item.pszText = directories[i].GetBuffer(0);
		
		item.lParam = m_FileInfos.AddToTail();
		m_FileInfos[item.lParam].m_bIsDir = true;
		m_FileInfos[item.lParam].m_Name = directories[i];
		item.iImage = SetupLabelImage( &m_FileInfos[item.lParam], directories[i], true );

		m_FileList.InsertItem( &item );
	}

	CUtlVector<CString> files;
	for ( int iMask=0; iMask < m_FileMasks.Count(); iMask++ )
	{
		GetEntries( m_FileMasks[iMask], files, GETENTRIES_FILES_ONLY );
	}

	RemoveDuplicates( files );
	if ( m_bFilterMdlAndJpgFiles )
		FilterMdlAndJpgFiles( files );

	for ( int i=0; i < files.Count(); i++ )
	{
		LVITEM item;
		item.mask = LVIF_TEXT | LVIF_IMAGE | LVIF_PARAM;
		item.iItem = iItem++;
		item.iSubItem = 0;
		item.iImage = m_iLabel_Mdl;
		item.pszText = files[i].GetBuffer(0);

		item.lParam = m_FileInfos.AddToTail();
		m_FileInfos[item.lParam].m_bIsDir = false;
		m_FileInfos[item.lParam].m_Name = files[i];
		item.iImage = SetupLabelImage( &m_FileInfos[item.lParam], files[i], false );

		m_FileList.InsertItem( &item );
	}

	m_FileList.SortItems( FileListSortCallback, (DWORD)this );
}

void CFileSystemOpenDlg::AddFileMask( const char *pMask )
{
	m_FileMasks.AddToTail( pMask );
}


BOOL CFileSystemOpenDlg::Create(LPCTSTR lpszClassName, LPCTSTR lpszWindowName, DWORD dwStyle, const RECT& rect, CWnd* pParentWnd, UINT nID, CCreateContext* pContext) 
{
	return CDialog::Create(IDD, pParentWnd);
}

int CFileSystemOpenDlg::OnCreate(LPCREATESTRUCT lpCreateStruct) 
{
	if (CDialog::OnCreate(lpCreateStruct) == -1)
		return -1;

	return 0;
}


LONG& GetSideCoord( RECT &rect, int iSide )
{
	if ( iSide == 0 )
		return rect.left;
	else if ( iSide == 1 )
		return rect.top;
	else if ( iSide == 2 )
		return rect.right;
	else
		return rect.bottom;
}


LONG GetSideScreenCoord( CWnd *pWnd, int iSide )
{
	RECT rect;
	pWnd->GetWindowRect( &rect );
	return GetSideCoord( rect, iSide );
}


void CFileSystemOpenDlg::ProcessAnchor( CWindowAnchor *pAnchor )
{
	RECT rect, parentRect;
	GetWindowRect( &parentRect );
	pAnchor->m_pWnd->GetWindowRect( &rect );
	
	GetSideCoord( rect, pAnchor->m_Side ) = GetSideCoord( parentRect, pAnchor->m_ParentSide ) + pAnchor->m_OriginalDist;

	ScreenToClient( &rect );
	pAnchor->m_pWnd->MoveWindow( &rect );
}


void CFileSystemOpenDlg::AddAnchor( int iDlgItem, int iSide, int iParentSide )
{
	CWnd *pItem = GetDlgItem( iDlgItem );
	if ( !pItem )
		return;

	CWindowAnchor *pAnchor = &m_Anchors[m_Anchors.AddToTail()];
	pAnchor->m_pWnd = pItem;
	pAnchor->m_Side = iSide;
	pAnchor->m_ParentSide = iParentSide;
	pAnchor->m_OriginalDist = GetSideScreenCoord( pItem, iSide ) - GetSideScreenCoord( this, iParentSide );
}



void CFileSystemOpenDlg::OnSize(UINT nType, int cx, int cy) 
{
	CDialog::OnSize(nType, cx, cy);
	
	for ( int i=0; i < m_Anchors.Count(); i++ )
		ProcessAnchor( &m_Anchors[i] );	

	if ( m_FileList.GetSafeHwnd() )
		PopulateListControl();
}

void CFileSystemOpenDlg::OnDblclkFileList(NMHDR* pNMHDR, LRESULT* pResult) 
{
	/*int iSelected = m_FileList.GetNextItem( -1, LVNI_SELECTED );
	if ( iSelected != -1 )
	{
		DWORD iItem = m_FileList.GetItemData( iSelected );
		if ( iItem < (DWORD)m_FileInfos.Count() )
		{
			CFileInfo *pInfo = &m_FileInfos[iItem];
			if ( pInfo->m_bIsDir )
			{
				m_CurrentDir += "\\" + m_FileInfos[iItem].m_Name;
				PopulateListControl();
			}
			else
			{
				m_Filename = m_CurrentDir + "\\" + m_FileInfos[iItem].m_Name;
				EndDialog( IDOK );
			}
		}
		else
		{
			Assert( false );
		}
	}*/
	OnOK();
	
	*pResult = 0;
}

void CFileSystemOpenDlg::OnUpButton() 
{
	char str[MAX_PATH];
	Q_strncpy( str, m_CurrentDir, sizeof( str ) );
	Q_StripLastDir( str, sizeof( str ) );

	if ( str[0] == 0 )
		Q_strncpy( str, ".", sizeof( str ) );
	
	if ( str[strlen(str)-1] == '\\' || str[strlen(str)-1] == '/' )
		str[strlen(str)-1] = 0;

	m_CurrentDir = str;
	PopulateListControl();
}

void CFileSystemOpenDlg::OnItemchangedFileList(NMHDR* pNMHDR, LRESULT* pResult) 
{
	NM_LISTVIEW* pNMListView = (NM_LISTVIEW*)pNMHDR;

	DWORD iItem = m_FileList.GetItemData( pNMListView->iItem );
	if ( iItem < (DWORD)m_FileInfos.Count() )
	{
		CFileInfo *pInfo = &m_FileInfos[iItem];
		if ( (pNMListView->uChanged & LVIF_STATE) && 
			 (pNMListView->uNewState & LVIS_SELECTED) )
		{
			m_FilenameControl.SetWindowText( pInfo->m_Name );
		}
	}
	
	*pResult = 0;
}

void CFileSystemOpenDlg::SetFilterMdlAndJpgFiles( bool bFilter )
{
	m_bFilterMdlAndJpgFiles = bFilter;
}

const char* CFileSystemOpenDlg::GetPathID()
{
	if ( m_PathIDString == "" )
		return NULL;
	else
		return (const char*)m_PathIDString;
}



// ------------------------------------------------------------------------------------------------ //
// Implementation of IFileSystemOpenDialog.
// ------------------------------------------------------------------------------------------------ //

// IFileSystemOpenDialog implementation.
class CFileSystemOpenDialogWrapper : public IFileSystemOpenDialog
{
public:
	CFileSystemOpenDialogWrapper()
	{
		m_pDialog = 0;
		m_bLastModalWasWindowsDialog = false;
		m_bAllowMultiSelect = false;
		m_RelativeFilename = NULL;
	}

	~CFileSystemOpenDialogWrapper()
	{
		delete m_RelativeFilename;
	}

	virtual void Release()
	{
		AFX_MANAGE_STATE(AfxGetStaticModuleState());
		delete m_pDialog;
		delete this;
	}

	// You must call this first to set the hwnd.
	virtual void Init( CreateInterfaceFn factory, void *parentHwnd )
	{
		AFX_MANAGE_STATE(AfxGetStaticModuleState());
		Assert( !m_pDialog );

		m_hParentWnd = (HWND)parentHwnd;
		m_pDialog = new CFileSystemOpenDlg( factory, CWnd::FromHandle( m_hParentWnd ) );
	}

	virtual void AddFileMask( const char *pMask )
	{
		AFX_MANAGE_STATE(AfxGetStaticModuleState());
		Assert( m_pDialog );
		m_pDialog->AddFileMask( pMask );
	}

	virtual void SetInitialDir( const char *pDir, const char *pPathID )
	{
		AFX_MANAGE_STATE(AfxGetStaticModuleState());
		Assert( m_pDialog );
		m_pDialog->SetInitialDir( pDir, pPathID );
	}

	virtual void SetFilterMdlAndJpgFiles( bool bFilter )
	{
		AFX_MANAGE_STATE(AfxGetStaticModuleState());
		Assert( m_pDialog );
		m_pDialog->SetFilterMdlAndJpgFiles( bFilter );
	}

	virtual void GetFilename( char *pOut, int outLen ) const
	{
		AFX_MANAGE_STATE(AfxGetStaticModuleState());
		Assert( m_pDialog );
		
		if ( m_bLastModalWasWindowsDialog )
		{
			Q_strncpy( pOut, m_RelativeFilename ? m_RelativeFilename : "", outLen );
		}
		else
		{
			Q_strncpy( pOut, m_pDialog->GetFilename(), outLen );
		}
	}

	virtual int GetFilenameBufferSize() const
	{
		if ( m_bLastModalWasWindowsDialog )
			return m_RelativeFilename ? strlen( m_RelativeFilename ) + 1 : 1;
		else
			return m_pDialog->GetFilename().GetLength() + 1;
	}

	virtual bool DoModal()
	{
		AFX_MANAGE_STATE(AfxGetStaticModuleState());
		Assert( m_pDialog );

		m_bLastModalWasWindowsDialog = false;
		return m_pDialog->DoModal() == IDOK;
	}

	virtual bool DoModal_WindowsDialog()
	{
		AFX_MANAGE_STATE(AfxGetStaticModuleState());

		Assert( m_pDialog );

		m_bLastModalWasWindowsDialog = true;

		// Get the full filename, then make sure it's a relative path.
		char defExt[MAX_PATH] = {0};
		if ( m_pDialog->m_FileMasks.Count() > 0 )
		{
			CString ext = m_pDialog->m_FileMasks[m_pDialog->m_FileMasks.Count()-1].Right( 4 );
			const char *pStr = ext;
			if ( pStr[0] == '.' )
				Q_strncpy( defExt, pStr+1, sizeof( defExt ) );
		}

		char pFileNameBuf[MAX_PATH];
		const char *pFileName = m_pDialog->m_pFileSystem->RelativePathToFullPath( m_pDialog->m_CurrentDir, m_pDialog->m_PathIDString, pFileNameBuf, MAX_PATH );
		Q_strcat( pFileNameBuf, "\\", sizeof(pFileNameBuf) );
	
		// Build the list of file filters.
		char filters[1024];
		if ( m_pDialog->m_FileMasks.Count() == 0 )
		{
			Q_strncpy( filters, "All Files (*.*)|*.*||", sizeof( filters ) );
		}
		else
		{
			filters[0] = 0;
			for ( int i=0; i < m_pDialog->m_FileMasks.Count(); i++ )
			{
				if ( i > 0 )
					Q_strncat( filters, "|", sizeof( filters ), COPY_ALL_CHARACTERS );

				Q_strncat( filters, m_pDialog->m_FileMasks[i], sizeof( filters ), COPY_ALL_CHARACTERS );
				Q_strncat( filters, "|", sizeof( filters ), COPY_ALL_CHARACTERS );
				Q_strncat( filters, m_pDialog->m_FileMasks[i], sizeof( filters ), COPY_ALL_CHARACTERS );
				if ( pFileName )
				{
					Q_strncat( pFileNameBuf, m_pDialog->m_FileMasks[i], sizeof( filters ), COPY_ALL_CHARACTERS );
					Q_strcat( pFileNameBuf, ";", sizeof(pFileNameBuf) );
				}

			}
			Q_strncat( filters, "||", sizeof( filters ), COPY_ALL_CHARACTERS );
		}

		DWORD dwDlgFlags = OFN_ENABLESIZING;
		if ( m_bAllowMultiSelect )
		{
			dwDlgFlags |= OFN_ALLOWMULTISELECT;
		}

		CFileDialog dlg( 
			true,								// open dialog?
			defExt[0]==0 ? NULL : defExt,		// default file extension
			pFileName,							// initial filename
			dwDlgFlags,							// flags
			filters,
			CWnd::FromHandle( m_hParentWnd ) );

		CArrayAutoPtr< char > spMultiSelectBuffer;
		if ( m_bAllowMultiSelect )
		{
			dlg.m_ofn.nMaxFile = 128 * 1024;
			spMultiSelectBuffer.Attach( new char[ dlg.m_ofn.nMaxFile ] );
			memset( spMultiSelectBuffer.Get(), 0, dlg.m_ofn.nMaxFile );
			dlg.m_ofn.lpstrFile = spMultiSelectBuffer.Get();
		}

		while ( dlg.DoModal() == IDOK )
		{
			CStringList strPathList;
			int numCharsTotal = MAX_PATH;

			if ( m_bAllowMultiSelect )
			{
				for ( POSITION pos = dlg.GetStartPosition(); pos; )
				{
					strPathList.AddTail( dlg.GetNextPathName( pos ) );
					numCharsTotal += strPathList.GetTail().GetLength();
				}
			}
			else
			{
				strPathList.AddTail( dlg.GetPathName() );
				numCharsTotal += strPathList.GetTail().GetLength();
			}
			numCharsTotal += 2 * strPathList.GetCount();

			if ( strPathList.IsEmpty() )
			{
				AfxMessageBox( IDS_NO_RELATIVE_PATH );
				continue;
			}

			// Allocate the buffer
			delete m_RelativeFilename;
			m_RelativeFilename = new char[ numCharsTotal ];

			char *pchFill = m_RelativeFilename;
			char chBuffer[ MAX_PATH ];

			bool bFailed = false;
			for ( POSITION pos = strPathList.GetHeadPosition(); pos; )
			{
				CString const &strPath = strPathList.GetNext( pos );

				if ( pchFill != m_RelativeFilename )
					*( pchFill ++ ) = ' ';

				// Make sure we can make this into a relative path.
				if ( m_pDialog->m_pFileSystem->FullPathToRelativePath( strPath,
					chBuffer, sizeof( chBuffer ) ) )
				{
					// Replace .jpg or .jpeg extension with .mdl?
					char *pEnd = chBuffer;
					while ( Q_stristr( pEnd+1, ".jpeg" ) || Q_stristr( pEnd+1, ".jpg" ) )
						pEnd = max( Q_stristr( pEnd, ".jpeg" ), Q_stristr( pEnd, ".jpg" ) );

					if ( pEnd && pEnd != chBuffer )
						Q_strncpy( pEnd, ".mdl", sizeof( chBuffer ) - (pEnd - chBuffer) );

					strcpy( pchFill, chBuffer );
					pchFill += strlen( pchFill );
				}
				else
				{
					AfxMessageBox( IDS_NO_RELATIVE_PATH );
					bFailed = true;
					break;
				}
			}

			if ( !bFailed )
				return true;
		}

		return false;
	}

	virtual void AllowMultiSelect( bool bAllow )
	{
		m_bAllowMultiSelect = bAllow;
	}

private:
	CFileSystemOpenDlg *m_pDialog;
	HWND m_hParentWnd;

	char *m_RelativeFilename;
	bool m_bLastModalWasWindowsDialog;
	bool m_bAllowMultiSelect;
};

EXPOSE_INTERFACE( CFileSystemOpenDialogWrapper, IFileSystemOpenDialog, FILESYSTEMOPENDIALOG_VERSION );