开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > Windows编程 > 多媒体编程 > 查看源码
ShaderEditorDlg.cpp
资源名称：bfyy.rar　[点击查看]
上传用户：tangyu_668
上传日期：2014-02-27
资源大小：678k
文件大小：10k
源码类别：

多媒体编程

开发平台：

Visual C++

ShaderEditorDlg.cpp：源码内容
  1. // ShaderEditorDlg.cpp : implementation file
  2. //
  4. #include "stdafx.h"
  5. #include "mplayerc.h"
  6. #include "PixelShaderCompiler.h"
  7. #include "ShaderEditorDlg.h"
  8. #include "MainFrm.h"
  10. #undef SubclassWindow
  12. // CShaderLabelComboBox
  14. BEGIN_MESSAGE_MAP(CShaderLabelComboBox, CComboBox)
  15. ON_WM_CTLCOLOR()
  16. ON_WM_DESTROY()
  17. END_MESSAGE_MAP()
  19. HBRUSH CShaderLabelComboBox::OnCtlColor(CDC* pDC, CWnd* pWnd, UINT nCtlColor)
  20. {
  21. if(nCtlColor == CTLCOLOR_EDIT)
  22. {
  23. if(m_edit.GetSafeHwnd() == NULL)
  24.             m_edit.SubclassWindow(pWnd->GetSafeHwnd());
  25. }
  27. return __super::OnCtlColor(pDC, pWnd, nCtlColor);
  28. }
  30. void CShaderLabelComboBox::OnDestroy()
  31. {
  32. if(m_edit.GetSafeHwnd() != NULL)
  33. m_edit.UnsubclassWindow();
  35. __super::OnDestroy();
  36. }
  38. // CShaderEdit
  40. CShaderEdit::CShaderEdit()
  41. {
  42. m_acdlg.Create(CShaderAutoCompleteDlg::IDD, NULL);
  44. m_nEndChar = -1;
  45. m_nIDEvent = -1;
  46. }
  48. CShaderEdit::~CShaderEdit()
  49. {
  50. m_acdlg.DestroyWindow();
  51. }
  53. BOOL CShaderEdit::PreTranslateMessage(MSG* pMsg)
  54. {
  55. if(m_acdlg.IsWindowVisible() 
  56. && pMsg->message == WM_KEYDOWN 
  57. && (pMsg->wParam == VK_UP || pMsg->wParam == VK_DOWN
  58. || pMsg->wParam == VK_PRIOR || pMsg->wParam == VK_NEXT
  59. || pMsg->wParam == VK_RETURN || pMsg->wParam == VK_ESCAPE))
  60. {
  61. int i = m_acdlg.m_list.GetCurSel();
  63. if(pMsg->wParam == VK_RETURN && i >= 0)
  64. {
  65. CString str;
  66. m_acdlg.m_list.GetText(i, str);
  67. i = str.Find('(')+1;
  68. if(i > 0) str = str.Left(i);
  70. int nStartChar = 0, nEndChar = -1;
  71. GetSel(nStartChar, nEndChar);
  73. CString text;
  74. GetWindowText(text);
  75. while(nStartChar > 0 && _istalnum(text.GetAt(nStartChar-1)))
  76. nStartChar--;
  78. SetSel(nStartChar, nEndChar);
  79. ReplaceSel(str, TRUE);
  80. }
  81. else if(pMsg->wParam == VK_ESCAPE)
  82. {
  83. m_acdlg.ShowWindow(SW_HIDE);
  84. }
  85. else
  86. {
  87. m_acdlg.m_list.SendMessage(pMsg->message, pMsg->wParam, pMsg->lParam);
  88. }
  90. return TRUE;
  91. }
  93. return __super::PreTranslateMessage(pMsg);
  94. }
  96. BEGIN_MESSAGE_MAP(CShaderEdit, CLineNumberEdit)
  97. ON_CONTROL_REFLECT(EN_UPDATE, OnUpdate)
  98. ON_WM_KILLFOCUS()
  99. ON_WM_TIMER()
  100. END_MESSAGE_MAP()
  102. void CShaderEdit::OnUpdate()
  103. {
  104. if(m_nIDEvent == -1)
  105. {
  106. m_nIDEvent = SetTimer(1, 100, NULL); 
  107. }
  109. CString text;
  110. int nStartChar = 0, nEndChar = -1;
  111. GetSel(nStartChar, nEndChar);
  113. if(nStartChar == nEndChar)
  114. {
  115. GetWindowText(text);
  116. while(nStartChar > 0 && _istalnum(text.GetAt(nStartChar-1)))
  117. nStartChar--;
  118. }
  120. if(nStartChar < nEndChar)
  121. {
  122. text = text.Mid(nStartChar, nEndChar - nStartChar);
  123. text.TrimRight('(');
  124. text.MakeLower();
  126. m_acdlg.m_list.ResetContent();
  128. CString key, value;
  129. POSITION pos = m_acdlg.m_inst.GetStartPosition();
  130. while(pos)
  131. {
  132. POSITION cur = pos;
  133. m_acdlg.m_inst.GetNextAssoc(pos, key, value);
  135. if(key.Find(text) == 0)
  136. {
  137. CAtlList<CString> sl;
  138. Explode(value, sl, '|', 2);
  139. if(sl.GetCount() != 2) continue;
  140. CString name = sl.RemoveHead();
  141. CString description = sl.RemoveHead();
  142. int i = m_acdlg.m_list.AddString(name);
  143. m_acdlg.m_list.SetItemDataPtr(i, cur);
  144. }
  145. }
  147. if(m_acdlg.m_list.GetCount() > 0)
  148. {
  149. int lineheight = GetLineHeight();
  151. CPoint p = PosFromChar(nStartChar);
  152. p.y += lineheight;
  153. ClientToScreen(&p);
  154. CRect r(p, CSize(100, 100));
  156. m_acdlg.MoveWindow(r);
  157. m_acdlg.SetWindowPos(&wndTopMost, 0, 0, 0, 0, SWP_NOMOVE|SWP_NOSIZE|SWP_NOACTIVATE);
  158. m_acdlg.ShowWindow(SW_SHOWNOACTIVATE);
  160. m_nEndChar = nEndChar;
  162. return;
  163. }
  164. }
  166. m_acdlg.ShowWindow(SW_HIDE);
  167. }
  169. void CShaderEdit::OnKillFocus(CWnd* pNewWnd)
  170. {
  171. CString text;
  172. GetWindowText(text);
  173. __super::OnKillFocus(pNewWnd);
  174. GetWindowText(text);
  176. m_acdlg.ShowWindow(SW_HIDE);
  177. }
  179. void CShaderEdit::OnTimer(UINT nIDEvent)
  180. {
  181. if(m_nIDEvent == nIDEvent)
  182. {
  183. int nStartChar = 0, nEndChar = -1;
  184. GetSel(nStartChar, nEndChar);
  185. if(nStartChar != nEndChar || m_nEndChar != nEndChar)
  186. m_acdlg.ShowWindow(SW_HIDE);
  187. }
  189. __super::OnTimer(nIDEvent);
  190. }
  192. // CShaderEditorDlg dialog
  194. CShaderEditorDlg::CShaderEditorDlg()
  195. : CResizableDialog(CShaderEditorDlg::IDD, NULL)
  196. , m_fSplitterGrabbed(false)
  197. , m_pPSC(NULL)
  198. , m_pShader(NULL)
  199. {
  200. }
  202. CShaderEditorDlg::~CShaderEditorDlg()
  203. {
  204. delete m_pPSC;
  205. }
  207. BOOL CShaderEditorDlg::Create(CWnd* pParent)
  208. {
  209. if(!__super::Create(IDD, pParent))
  210. return FALSE;
  212. AddAnchor(IDC_COMBO1, TOP_LEFT, TOP_RIGHT);
  213. AddAnchor(IDC_COMBO2, TOP_RIGHT);
  214. AddAnchor(IDC_EDIT1, TOP_LEFT, BOTTOM_RIGHT);
  215. AddAnchor(IDC_EDIT2, BOTTOM_LEFT, BOTTOM_RIGHT);
  216. AddAnchor(IDC_BUTTON1, TOP_RIGHT);
  218. m_srcdata.SetTabStops(16);
  220. SetMinTrackSize(CSize(250, 40));
  222. m_targets.AddString(_T("ps_1_1"));
  223. m_targets.AddString(_T("ps_1_2"));
  224. m_targets.AddString(_T("ps_1_3"));
  225. m_targets.AddString(_T("ps_1_4"));
  226. m_targets.AddString(_T("ps_2_0"));
  227. m_targets.AddString(_T("ps_2_a"));
  228. m_targets.AddString(_T("ps_2_sw"));
  229. m_targets.AddString(_T("ps_3_0"));
  230. m_targets.AddString(_T("ps_3_sw"));
  232. POSITION pos = AfxGetAppSettings().m_shaders.GetHeadPosition();
  233. while(pos)
  234. {
  235. const AppSettings::Shader& s = AfxGetAppSettings().m_shaders.GetNext(pos);
  236. m_labels.SetItemDataPtr(m_labels.AddString(s.label), (void*)&s);
  237. }
  239. m_nIDEventShader = SetTimer(1, 1000, NULL);
  241. return TRUE;
  242. }
  244. void CShaderEditorDlg::DoDataExchange(CDataExchange* pDX)
  245. {
  246. __super::DoDataExchange(pDX);
  247. DDX_Control(pDX, IDC_COMBO1, m_labels);
  248. DDX_Control(pDX, IDC_COMBO2, m_targets);
  249. DDX_Control(pDX, IDC_EDIT1, m_srcdata);
  250. DDX_Control(pDX, IDC_EDIT2, m_output);
  251. }
  253. bool CShaderEditorDlg::HitTestSplitter(CPoint p)
  254. {
  255. CRect r, rs, ro;
  256. m_srcdata.GetWindowRect(&rs);
  257. m_output.GetWindowRect(&ro);
  258. ScreenToClient(&rs);
  259. ScreenToClient(&ro);
  260. GetClientRect(&r);
  261. r.left = ro.left;
  262. r.right = ro.right;
  263. r.top = rs.bottom;
  264. r.bottom = ro.top;
  265. return !!r.PtInRect(p);
  266. }
  268. BEGIN_MESSAGE_MAP(CShaderEditorDlg, CResizableDialog)
  269. ON_CBN_SELCHANGE(IDC_COMBO1, OnCbnSelchangeCombo1)
  270. ON_BN_CLICKED(IDC_BUTTON1, OnBnClickedButton2)
  271. ON_WM_TIMER()
  272. ON_WM_LBUTTONDOWN()
  273. ON_WM_LBUTTONUP()
  274. ON_WM_MOUSEMOVE()
  275. ON_WM_SETCURSOR()
  276. END_MESSAGE_MAP()
  278. // CShaderEditorDlg message handlers
  280. BOOL CShaderEditorDlg::PreTranslateMessage(MSG* pMsg)
  281. {
  282. if(pMsg->message == WM_KEYDOWN && pMsg->wParam == VK_RETURN
  283. && pMsg->hwnd == m_labels.m_edit.GetSafeHwnd())
  284. {
  285. OnCbnSelchangeCombo1();
  286.         
  287. return TRUE;
  288. }
  289. else if(pMsg->message == WM_KEYDOWN && pMsg->wParam == VK_TAB
  290. && pMsg->hwnd == m_srcdata.GetSafeHwnd())
  291. {
  292. int nStartChar, nEndChar;
  293. m_srcdata.GetSel(nStartChar, nEndChar);
  294. if(nStartChar == nEndChar) m_srcdata.ReplaceSel(_T("t"));
  295. return TRUE;
  296. }
  297. else if(pMsg->message == WM_KEYDOWN && pMsg->wParam == VK_ESCAPE)
  298. {
  299. return TRUE;
  300. }
  302. return __super::PreTranslateMessage(pMsg);
  303. }
  305. void CShaderEditorDlg::OnCbnSelchangeCombo1()
  306. {
  307. int i = m_labels.GetCurSel();
  309. if(i < 0)
  310. {
  311. CString label;
  312. m_labels.GetWindowText(label);
  313. label.Trim();
  315. if(label.IsEmpty()) return;
  317. CStringA srcdata;
  318. LoadResource(IDF_SHADER_EMPTY, srcdata, _T("FILE"));
  320. AppSettings::Shader s;
  321. s.label = label;
  322. s.target = _T("ps_2_0");
  323. s.srcdata = CString(srcdata);
  325. POSITION pos = AfxGetAppSettings().m_shaders.AddTail(s);
  327. i = m_labels.AddString(s.label);
  328. m_labels.SetCurSel(i);
  329. m_labels.SetItemDataPtr(i, (void*)&AfxGetAppSettings().m_shaders.GetAt(pos));
  330. }
  332. m_pShader = (AppSettings::Shader*)m_labels.GetItemDataPtr(i);
  334. m_targets.SetWindowText(m_pShader->target);
  336. CString srcdata = m_pShader->srcdata;
  337. srcdata.Replace(_T("n"), _T("rn"));
  338. m_srcdata.SetWindowText(srcdata);
  340. ((CMainFrame*)AfxGetMainWnd())->UpdateShaders(m_pShader->label);
  341. }
  343. void CShaderEditorDlg::OnBnClickedButton2()
  344. {
  345. if(!m_pShader) return;
  347. if(IDYES != AfxMessageBox(_T("Are you sure you want to delete this shader?"), MB_YESNO))
  348. return;
  350. AppSettings& s = AfxGetAppSettings();
  352. for(POSITION pos = s.m_shaders.GetHeadPosition(); pos; s.m_shaders.GetNext(pos))
  353. {
  354. if(m_pShader == &s.m_shaders.GetAt(pos))
  355. {
  356. m_pShader = NULL;
  357. s.m_shaders.RemoveAt(pos);
  358. int i = m_labels.GetCurSel();
  359. if(i >= 0) m_labels.DeleteString(i);
  360. m_labels.SetWindowText(_T(""));
  361. m_targets.SetWindowText(_T(""));
  362. m_srcdata.SetWindowText(_T(""));
  363. m_output.SetWindowText(_T(""));
  364. ((CMainFrame*)AfxGetMainWnd())->UpdateShaders(_T(""));
  365. break;
  366. }
  367. }
  368. }
  370. void CShaderEditorDlg::OnTimer(UINT nIDEvent)
  371. {
  372. if(nIDEvent == m_nIDEventShader && IsWindowVisible() && m_pShader)
  373. {
  374. CString srcdata;
  375. m_srcdata.GetWindowText(srcdata);
  376. srcdata.Replace(_T("r"), _T(""));
  377. srcdata.Trim();
  379. CString target;
  380. m_targets.GetWindowText(target);
  381. target.Trim();
  383. if(!srcdata.IsEmpty() && !target.IsEmpty() && (m_pShader->srcdata != srcdata || m_pShader->target != target))
  384. {
  385. KillTimer(m_nIDEventShader);
  387. m_pShader->srcdata = srcdata;
  388. m_pShader->target = target;
  390. if(!m_pPSC) m_pPSC = new CPixelShaderCompiler(NULL);
  392. CString disasm, errmsg;
  393. HRESULT hr = m_pPSC->CompileShader(CStringA(srcdata), "main", CStringA(target), D3DXSHADER_DEBUG, NULL, &disasm, &errmsg); 
  395. if(SUCCEEDED(hr))
  396. {
  397. errmsg = _T("D3DXCompileShader succeededn");
  398. errmsg += _T("n");
  399. errmsg += disasm;
  401. ((CMainFrame*)AfxGetMainWnd())->UpdateShaders(m_pShader->label);
  402. }
  404. errmsg.Replace(_T("n"), _T("rn"));
  406. m_output.SetWindowText(errmsg);
  408. // TODO: autosave
  410. m_nIDEventShader = SetTimer(1, 1000, NULL);
  411. }
  412. }
  414. __super::OnTimer(nIDEvent);
  415. }
  417. void CShaderEditorDlg::OnLButtonDown(UINT nFlags, CPoint point)
  418. {
  419. if(HitTestSplitter(point))
  420. {
  421. m_fSplitterGrabbed = true;
  422. SetCapture();
  423. }
  425. __super::OnLButtonDown(nFlags, point);
  426. }
  428. void CShaderEditorDlg::OnLButtonUp(UINT nFlags, CPoint point)
  429. {
  430. if(m_fSplitterGrabbed)
  431. {
  432. ReleaseCapture();
  433. m_fSplitterGrabbed = false;
  434. }
  436. __super::OnLButtonUp(nFlags, point);
  437. }
  439. void CShaderEditorDlg::OnMouseMove(UINT nFlags, CPoint point)
  440. {
  441. if(m_fSplitterGrabbed)
  442. {
  443. CRect r, rs, ro;
  444. GetClientRect(&r);
  445. m_srcdata.GetWindowRect(&rs);
  446. m_output.GetWindowRect(&ro);
  447. ScreenToClient(&rs);
  448. ScreenToClient(&ro);
  450. int dist = ro.top - rs.bottom;
  451. int avgdist = dist / 2;
  453. rs.bottom = min(max(point.y, rs.top + 40), ro.bottom - 40) - avgdist;
  454. ro.top = rs.bottom + dist;
  455. m_srcdata.MoveWindow(&rs);
  456. m_output.MoveWindow(&ro);
  458. int div = 100 * ((rs.bottom + ro.top) / 2) / (ro.bottom - rs.top);
  460. RemoveAnchor(IDC_EDIT1);
  461. RemoveAnchor(IDC_EDIT2);
  462. AddAnchor(IDC_EDIT1, TOP_LEFT, CSize(100, div)/*BOTTOM_RIGHT*/);
  463. AddAnchor(IDC_EDIT2, CSize(0, div)/*BOTTOM_LEFT*/, BOTTOM_RIGHT);
  464. }
  466. __super::OnMouseMove(nFlags, point);
  467. }
  469. BOOL CShaderEditorDlg::OnSetCursor(CWnd* pWnd, UINT nHitTest, UINT message)
  470. {
  471. CPoint p;
  472. GetCursorPos(&p);
  473. ScreenToClient(&p);
  474. if(HitTestSplitter(p))
  475. {
  476. ::SetCursor(AfxGetApp()->LoadStandardCursor(IDC_SIZENS));
  477. return TRUE;
  478. }
  480. return __super::OnSetCursor(pWnd, nHitTest, message);
  481. }