Aegisub/SSATool/SaveFileDialogWithEncoding.cs

using System;
using System.Collections;
using System.Windows.Forms;
using System.Data;
using System.Text;
using System.Runtime.InteropServices;
using System.Drawing;



public enum EncodingType {
	//UTF8=0,
	//UTF8WithPreamble,
	//Unicode,
	//Ansi
	ANSI,
	BigEndianUnicode,
	Unicode,
	UTF8,
	Shift_JIS
};

public class SaveFileDialogWithEncoding {
	private delegate int OFNHookProcDelegate(int hdlg, int msg, int wParam, int lParam);

	private int m_LabelHandle=0;
	private int m_ComboHandle=0;

	private string m_Filter="";
	private string m_DefaultExt="";

	private string m_FileName="";

	private EncodingType m_EncodingType;
	private Screen m_ActiveScreen;

	[StructLayout(LayoutKind.Sequential, CharSet=CharSet.Auto)]
		private struct OPENFILENAME {
		public int lStructSize; 
		public IntPtr hwndOwner; 
		public int hInstance; 
		[MarshalAs(UnmanagedType.LPTStr)] public string lpstrFilter;
		[MarshalAs(UnmanagedType.LPTStr)] public string lpstrCustomFilter; 
		public int nMaxCustFilter; 
		public int nFilterIndex; 
		[MarshalAs(UnmanagedType.LPTStr)] public string lpstrFile; 
		public int nMaxFile; 
		[MarshalAs(UnmanagedType.LPTStr)] public string lpstrFileTitle; 
		public int nMaxFileTitle; 
		[MarshalAs(UnmanagedType.LPTStr)] public string lpstrInitialDir; 
		[MarshalAs(UnmanagedType.LPTStr)] public string lpstrTitle; 
		public int Flags; 
		public short nFileOffset; 
		public short nFileExtension; 
		[MarshalAs(UnmanagedType.LPTStr)] public string lpstrDefExt; 
		public int lCustData; 
		public OFNHookProcDelegate lpfnHook;
		[MarshalAs(UnmanagedType.LPTStr)] public string lpTemplateName;
		//only if on nt 5.0 or higher
		public int pvReserved;
		public int dwReserved;
		public int FlagsEx;
	}

	[DllImport("Comdlg32.dll", CharSet=CharSet.Auto, SetLastError=true)]
	private static extern bool GetSaveFileName(ref OPENFILENAME lpofn);

	[DllImport("Comdlg32.dll")]
	private static extern int CommDlgExtendedError();

	[DllImport("user32.dll")]
	private static extern bool SetWindowPos(int hWnd, int hWndInsertAfter, int X, int Y, int cx, int cy, uint uFlags);

	private struct RECT {
		public int Left;
		public int Top;
		public int Right;
		public int Bottom;
	}

	private struct POINT {
		public int X;
		public int Y;
	}

	private struct NMHDR {
		public int HwndFrom;
		public int IdFrom;
		public int Code;
	}

	[DllImport("user32.dll")]
	private static extern bool GetWindowRect(int hWnd, ref RECT lpRect);

	[DllImport("user32.dll")]
	private static extern int GetParent(int hWnd);

	[DllImport("user32.dll", CharSet=CharSet.Auto)]
	private static extern bool SetWindowText(int hWnd, string lpString);

	[DllImport("user32.dll")]
	private static extern int SendMessage(int hWnd, int Msg, int wParam, int lParam);

	[DllImport("user32.dll", CharSet=CharSet.Auto)]
	private static extern int SendMessage(int hWnd, int Msg, int wParam, string lParam);

	[DllImport("user32.dll")]
	private static extern bool DestroyWindow(int hwnd);

	private const int OFN_ENABLEHOOK=0x00000020;
	private const int OFN_EXPLORER=0x00080000;
	private const int OFN_FILEMUSTEXIST=0x00001000;
	private const int OFN_HIDEREADONLY=0x00000004;
	private const int OFN_CREATEPROMPT=0x00002000;
	private const int OFN_NOTESTFILECREATE=0x00010000;
	private const int OFN_OVERWRITEPROMPT=0x00000002;
	private const int OFN_PATHMUSTEXIST=0x00000800;

	private const int SWP_NOSIZE=0x0001;
	private const int SWP_NOMOVE=0x0002;
	private const int SWP_NOZORDER=0x0004;

	private const int WM_INITDIALOG=0x110;
	private const int WM_DESTROY=0x2;
	private const int WM_SETFONT=0x0030;
	private const int WM_GETFONT=0x0031;

	private const int CBS_DROPDOWNLIST=0x0003;
	private const int CBS_HASSTRINGS=0x0200;
	private const int CB_ADDSTRING=0x0143;
	private const int CB_SETCURSEL=0x014E;
	private const int CB_GETCURSEL=0x0147;

	private const uint WS_VISIBLE=0x10000000;
	private const uint WS_CHILD=0x40000000;
	private const uint WS_TABSTOP=0x00010000;

	private const int CDN_FILEOK=-606;
	private const int WM_NOTIFY=0x004E;

	[DllImport("user32.dll", CharSet=CharSet.Auto)]
	private static extern int GetDlgItem(int hDlg, int nIDDlgItem);

	[DllImport("user32.dll", CharSet=CharSet.Auto)]
	private static extern int CreateWindowEx(int dwExStyle, string lpClassName, string lpWindowName, uint dwStyle, int x, int y, int nWidth, int nHeight, int hWndParent, int hMenu, int hInstance, int lpParam);

	[DllImport("user32.dll")]
	private static extern bool ScreenToClient(int hWnd, ref POINT lpPoint);

	private int HookProc(int hdlg, int msg, int wParam, int lParam) {
		switch (msg) {
			case WM_INITDIALOG:

				//we need to centre the dialog
				Rectangle sr=m_ActiveScreen.Bounds;
				RECT cr=new RECT();
				int parent=GetParent(hdlg);
				GetWindowRect(parent, ref cr);

				int x=(sr.Right + sr.Left - (cr.Right-cr.Left))/2;
				int y=(sr.Bottom + sr.Top - (cr.Bottom-cr.Top))/2;

				SetWindowPos(parent, 0, x, y, cr.Right-cr.Left, cr.Bottom - cr.Top + 32, SWP_NOZORDER);
				

				//we need to find the label to position our new label under

				int fileTypeWindow=GetDlgItem(parent, 0x441);

				RECT aboveRect=new RECT();
				GetWindowRect(fileTypeWindow, ref aboveRect);

				//now convert the label's screen co-ordinates to client co-ordinates
				POINT point=new POINT();
				point.X=aboveRect.Left;
				point.Y=aboveRect.Bottom;

				ScreenToClient(parent, ref point);

				//create the label
				int labelHandle=CreateWindowEx(0, "STATIC", "mylabel", WS_VISIBLE | WS_CHILD | WS_TABSTOP, point.X, point.Y + 12, 200, 100, parent, 0, 0, 0);
				SetWindowText(labelHandle, "&Encoding:");

				int fontHandle=SendMessage(fileTypeWindow, WM_GETFONT, 0, 0);
				SendMessage(labelHandle, WM_SETFONT, fontHandle, 0);

				//we now need to find the combo-box to position the new combo-box under

				int fileComboWindow=GetDlgItem(parent, 0x470);
				aboveRect=new RECT();
				GetWindowRect(fileComboWindow, ref aboveRect);

				point=new POINT();
				point.X=aboveRect.Left;
				point.Y=aboveRect.Bottom;
				ScreenToClient(parent, ref point);

				POINT rightPoint=new POINT();
				rightPoint.X=aboveRect.Right;
				rightPoint.Y=aboveRect.Top;

				ScreenToClient(parent, ref rightPoint);

				//we create the new combobox

				int comboHandle=CreateWindowEx(0, "ComboBox", "mycombobox", WS_VISIBLE | WS_CHILD | CBS_HASSTRINGS | CBS_DROPDOWNLIST | WS_TABSTOP, point.X, point.Y + 8, rightPoint.X-point.X, 100, parent, 0, 0, 0);
				SendMessage(comboHandle, WM_SETFONT, fontHandle, 0);

				//and add the encodings we want to offer
				SendMessage(comboHandle, CB_ADDSTRING, 0, "ANSI");
				SendMessage(comboHandle, CB_ADDSTRING, 0, "BigEndianUnicode");
				SendMessage(comboHandle, CB_ADDSTRING, 0, "Unicode");
				SendMessage(comboHandle, CB_ADDSTRING, 0, "UTF-8");
				SendMessage(comboHandle, CB_ADDSTRING, 0, "SJIS");
				
				SendMessage(comboHandle, CB_SETCURSEL, (int)m_EncodingType, 0);

				//remember the handles of the controls we have created so we can destroy them after
				m_LabelHandle=labelHandle;
				m_ComboHandle=comboHandle;

				break;
			case WM_DESTROY:
				//destroy the handles we have created
				if (m_ComboHandle!=0) {
					DestroyWindow(m_ComboHandle);
				}

				if (m_LabelHandle!=0) {
					DestroyWindow(m_LabelHandle);
				}
				break;
			case WM_NOTIFY:

				//we need to intercept the CDN_FILEOK message
				//which is sent when the user selects a filename

				NMHDR nmhdr=(NMHDR)Marshal.PtrToStructure(new IntPtr(lParam), typeof(NMHDR));

				if (nmhdr.Code==CDN_FILEOK) {
					//a file has been selected
					//we need to get the encoding

					m_EncodingType=(EncodingType)SendMessage(m_ComboHandle, CB_GETCURSEL, 0, 0);
				}
				break;

		}
		return 0;
	}

	public string DefaultExt {
		get { return m_DefaultExt; }
		set { m_DefaultExt=value; }
	}

	public string Filter {
		get { return m_Filter; }
		set { m_Filter=value; }
	}

	public string FileName {
		get { return m_FileName; }
		set { m_FileName=value;	}
	}

	public EncodingType EncodingType {
		get { return m_EncodingType; }
		set { m_EncodingType=value;	}
	}

	public DialogResult ShowDialog(IntPtr ownerhwnd, Screen ownerscreen) {
		return ShowDialog(ownerhwnd,ownerscreen,"Unicode");
	}

	public DialogResult ShowDialog(IntPtr ownerhwnd, Screen ownerscreen, string encoding) {

		//m_EncodingType = stringToEncodingType(encoding);

		//set up the struct and populate it

		OPENFILENAME ofn=new OPENFILENAME();

		ofn.lStructSize= Marshal.SizeOf( ofn );
		ofn.lpstrFilter= m_Filter.Replace('|', '\0') + '\0';

		ofn.lpstrFile = m_FileName + new string(' ', 512);
		ofn.nMaxFile= ofn.lpstrFile.Length;
		ofn.lpstrFileTitle= System.IO.Path.GetFileName(m_FileName) + new string(' ', 512);
		ofn.nMaxFileTitle = ofn.lpstrFileTitle.Length;
		ofn.lpstrTitle= "Save file as";
		ofn.lpstrDefExt=m_DefaultExt;

		//position the dialog above the active window
		ofn.hwndOwner=ownerhwnd;
		
		//we need to find out the active screen so the dialog box is
		//centred on the correct display

		m_ActiveScreen=ownerscreen;

		//set up some sensible flags
		ofn.Flags=OFN_EXPLORER |  OFN_PATHMUSTEXIST | OFN_NOTESTFILECREATE | OFN_ENABLEHOOK | OFN_HIDEREADONLY | OFN_OVERWRITEPROMPT;
		
		//this is where the hook is set. Note that we can use a C# delegate in place of a C function pointer
		ofn.lpfnHook=new OFNHookProcDelegate(HookProc);

		//if we're running on Windows 98/ME then the struct is smaller
		if (System.Environment.OSVersion.Platform!=PlatformID.Win32NT) {
			ofn.lStructSize-=12;
		}

		//show the dialog

		if (!GetSaveFileName(ref ofn)) {
			int ret=CommDlgExtendedError();

			if (ret!=0) {
				throw new ApplicationException("Couldn't show file open dialog - " + ret.ToString());
			}

			return DialogResult.Cancel;
		}

		m_FileName=ofn.lpstrFile;

		return DialogResult.OK;
	}

	public static System.Text.Encoding stringToEncodingType(string input) {
		unsafe {
			switch (input) {
				case "ASCII" :              return System.Text.Encoding.ASCII;
				case "ANSI" :               return System.Text.Encoding.GetEncoding(1252);
				case "BigEndianUnicode" :   return System.Text.Encoding.BigEndianUnicode;
				case "SJIS" : 
                case "Shift_JIS" :          return System.Text.Encoding.GetEncoding("shift_jis");
				case "UTF8" :
				case "UTF-8" :              return System.Text.Encoding.UTF8;
                case "Unicode" :
				default :                   return System.Text.Encoding.Unicode;
			}
		}
	}
}