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

源码开发语言/平台
当前位置：首页 > 源码/资料 > Windows编程 > 多媒体编程 > 查看源码
ShaderEditorDlg.cpp
资源名称：fullMPCcode.rar　[点击查看]
上传用户：xjjlds
上传日期：2015-12-05
资源大小：22823k
文件大小：13k
源码类别：

多媒体编程

开发平台：

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"
  9. #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. CList<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(CString label, ISubPicAllocatorPresenter* pCAP, CWnd* pParent /*=NULL*/)
  195. : CResizableDialog(CShaderEditorDlg::IDD, pParent)
  196. , m_label(label)
  197. , m_pCAP(pCAP)
  198. , m_fSplitterGrabbed(false)
  199. {
  200. m_pPSC = new CPixelShaderCompiler(NULL);
  201. }
  203. CShaderEditorDlg::~CShaderEditorDlg()
  204. {
  205. delete m_pPSC;
  206. }
  208. void CShaderEditorDlg::DoDataExchange(CDataExchange* pDX)
  209. {
  210. __super::DoDataExchange(pDX);
  211. DDX_Control(pDX, IDC_COMBO1, m_labels);
  212. DDX_Control(pDX, IDC_COMBO2, m_targets);
  213. DDX_Control(pDX, IDC_EDIT1, m_srcdata);
  214. DDX_Control(pDX, IDC_EDIT2, m_output);
  215. }
  217. bool CShaderEditorDlg::HitTestSplitter(CPoint p)
  218. {
  219. CRect r, rs, ro;
  220. m_srcdata.GetWindowRect(&rs);
  221. m_output.GetWindowRect(&ro);
  222. ScreenToClient(&rs);
  223. ScreenToClient(&ro);
  224. GetClientRect(&r);
  225. r.left = ro.left;
  226. r.right = ro.right;
  227. r.top = rs.bottom;
  228. r.bottom = ro.top;
  229. return !!r.PtInRect(p);
  230. }
  232. BEGIN_MESSAGE_MAP(CShaderEditorDlg, CResizableDialog)
  233. ON_CBN_SELCHANGE(IDC_COMBO1, OnCbnSelchangeCombo1)
  234. ON_BN_CLICKED(IDC_BUTTON1, OnBnClickedButton1)
  235. ON_BN_CLICKED(IDC_BUTTON2, OnBnClickedButton2)
  236. ON_WM_TIMER()
  237. ON_WM_CLOSE()
  238. ON_WM_LBUTTONDOWN()
  239. ON_WM_LBUTTONUP()
  240. ON_WM_MOUSEMOVE()
  241. ON_WM_SETCURSOR()
  242. END_MESSAGE_MAP()
  245. // CShaderEditorDlg message handlers
  247. BOOL CShaderEditorDlg::OnInitDialog()
  248. {
  249. __super::OnInitDialog();
  251. AddAnchor(IDC_COMBO1, TOP_LEFT, TOP_RIGHT);
  252. AddAnchor(IDC_COMBO2, TOP_RIGHT);
  253. AddAnchor(IDC_BUTTON1, TOP_RIGHT);
  254. AddAnchor(IDC_BUTTON2, TOP_RIGHT);
  255. AddAnchor(IDC_EDIT1, TOP_LEFT, BOTTOM_RIGHT);
  256. AddAnchor(IDC_EDIT2, BOTTOM_LEFT, BOTTOM_RIGHT);
  258. m_srcdata.SetTabStops(16);
  260. CRect r;
  261. GetWindowRect(r);
  262. CSize s = r.Size();
  263. s.cx = 400;
  264. s.cy = 150;
  265. SetMinTrackSize(s);
  267. CMap<CString, LPCTSTR, bool, bool> targetmap;
  268. targetmap[_T("ps_1_1")] = true;
  269. targetmap[_T("ps_1_2")] = true;
  270. targetmap[_T("ps_1_3")] = true;
  271. targetmap[_T("ps_1_4")] = true;
  272. targetmap[_T("ps_2_0")] = true;
  273. targetmap[_T("ps_2_a")] = true;
  274. targetmap[_T("ps_2_sw")] = true;
  275. targetmap[_T("ps_3_0")] = true;
  276. targetmap[_T("ps_3_sw")] = true;
  278. int nSelIndex = -1;
  280. for(int i = 0; ; i++)
  281. {
  282. CString str;
  283. str.Format(_T("%d"), i);
  284. str = AfxGetApp()->GetProfileString(_T("Shaders"), str);
  286. CList<CString> sl;
  287. Explode(str, sl, '|', 3);
  288. if(sl.GetCount() != 3) break;
  290. CString label = sl.RemoveHead();
  291. CString target = sl.RemoveHead();
  292. CString srcdata = sl.RemoveHead();
  293. srcdata.Replace(_T("\n"), _T("rn"));
  294. srcdata.Replace(_T("\t"), _T("t"));
  296. targetmap[target] = false;
  298. shader_t s = {target, srcdata};
  299. m_shaders[label] = s;
  301. int nIndex = m_labels.AddString(label);
  303. if(m_label == label) 
  304. nSelIndex = nIndex;
  305. }
  307. m_labels.SetCurSel(nSelIndex);
  309. POSITION pos = targetmap.GetStartPosition();
  310. while(pos) 
  311. {
  312. CString target;
  313. bool b;
  314. targetmap.GetNextAssoc(pos, target, b);
  315. m_targets.AddString(target);
  316. }
  318. OnCbnSelchangeCombo1();
  320. m_nIDEventShader = SetTimer(1, 1000, NULL);
  322. return TRUE;  // return TRUE unless you set the focus to a control
  323. // EXCEPTION: OCX Property Pages should return FALSE
  324. }
  326. BOOL CShaderEditorDlg::PreTranslateMessage(MSG* pMsg)
  327. {
  328. if(pMsg->message == WM_KEYDOWN && pMsg->wParam == VK_RETURN
  329. && pMsg->hwnd == m_labels.m_edit.GetSafeHwnd())
  330. {
  331. CString label;
  332. m_labels.GetWindowText(label);
  334. shader_t s;
  335. m_labels.SetCurSel(!m_shaders.Lookup(label, s) 
  336. ? m_labels.AddString(label) 
  337. : m_labels.FindStringExact(0, label));
  339. OnCbnSelchangeCombo1();
  340.         
  341. return TRUE;
  342. }
  343. else if(pMsg->message == WM_KEYDOWN && pMsg->wParam == VK_TAB
  344. && pMsg->hwnd == m_srcdata.GetSafeHwnd())
  345. {
  346. int nStartChar, nEndChar;
  347. m_srcdata.GetSel(nStartChar, nEndChar);
  349. if(nStartChar == nEndChar)
  350.             m_srcdata.ReplaceSel(_T("t"));
  352. return TRUE;
  353. }
  354. else if(pMsg->message == WM_KEYDOWN && pMsg->wParam == VK_ESCAPE)
  355. {
  356. return TRUE;
  357. }
  359. return __super::PreTranslateMessage(pMsg);
  360. }
  362. void CShaderEditorDlg::OnCbnSelchangeCombo1()
  363. {
  364. int i = m_labels.GetCurSel();
  365. if(i < 0) return;
  366. CString label;
  367. m_labels.GetLBText(i, label);
  368. if(label.IsEmpty()) return;
  370. shader_t s;
  372. if(!m_shaders.Lookup(label, s))
  373. {
  374. s.target = _T("ps_2_0");
  375. s.srcdata = 
  376.             "sampler s0 : register(s0);rn"
  377. "float4 p0 : register(c0);rn"
  378.             "rn"
  379. "#define width (p0[0])rn"
  380. "#define height (p0[1])rn"
  381. "#define counter (p0[2])rn"
  382.             "#define clock (p0[3])rn"
  383. "rn"
  384. "float4 main(float2 tex : TEXCOORD0) : COLORrn"
  385. "{rn"
  386. "tfloat4 c0 = tex2D(s0, tex);rn"
  387. "t// TODOrn"
  388. "treturn c0;rn"
  389. "}rn";
  390. m_shaders[label] = s;
  391. }
  393. if(m_shaders.Lookup(label, s))
  394. {
  395. m_targets.SetWindowText(s.target);
  396. m_srcdata.SetWindowText(s.srcdata);
  397. }
  398. }
  400. void CShaderEditorDlg::OnBnClickedButton1()
  401. {
  402. CString label;
  403. m_labels.GetWindowText(label);
  404. int nIndex = m_labels.FindStringExact(0, label);
  405. if(nIndex >= 0) m_labels.DeleteString(nIndex);
  406. m_labels.SetWindowText(_T(""));
  407. m_targets.SetWindowText(_T(""));
  408. m_srcdata.SetWindowText(_T(""));
  409. m_shaders.RemoveKey(label);
  410. }
  412. void CShaderEditorDlg::OnBnClickedButton2()
  413. {
  414. CString label;
  415. m_labels.GetWindowText(label);
  416. m_targets.GetWindowText(m_shaders[label].target);
  417. m_srcdata.GetWindowText(m_shaders[label].srcdata);
  418. if(m_labels.FindStringExact(0, label) < 0)
  419. m_labels.SetCurSel(m_labels.AddString(label));
  420. }
  422. void CShaderEditorDlg::OnTimer(UINT nIDEvent)
  423. {
  424. if(nIDEvent == m_nIDEventShader)
  425. {
  426. static CString s_srcdata, s_target;
  428. CString srcdata;
  429. m_srcdata.GetWindowText(srcdata);
  430. srcdata.Trim();
  432. CString target;
  433. m_targets.GetWindowText(target);
  434. target.Trim();
  436. if(!srcdata.IsEmpty() && !target.IsEmpty() && (s_srcdata != srcdata || s_target != target))
  437. {
  438. CString disasm, errmsg(_T("Unknown Error"));
  440. HRESULT hr = m_pPSC->CompileShader(CStringA(srcdata), "main", CStringA(target), D3DXSHADER_DEBUG, NULL, &disasm, &errmsg); 
  442. if(SUCCEEDED(hr))
  443. {
  444. errmsg = _T("D3DXCompileShader succeededn");
  445. if(m_pCAP && FAILED(m_pCAP->SetPixelShader(CStringA(srcdata), CStringA(target))))
  446. errmsg += _T("SetPixelShader failedn");
  447. errmsg += _T("n");
  448. errmsg += disasm;
  449. }
  451. errmsg.Replace(_T("n"), _T("rn"));
  453. m_output.SetWindowText(errmsg);
  454. }
  456. s_srcdata = srcdata;
  457. s_target = target;
  458. }
  460. __super::OnTimer(nIDEvent);
  461. }
  463. void CShaderEditorDlg::OnClose()
  464. {
  465. if(IDYES == AfxMessageBox(_T("Save changes?"), MB_YESNO))
  466. {
  467. OnBnClickedButton2();
  469. CWinApp* pApp = AfxGetApp();
  471. pApp->WriteProfileString(_T("Shaders"), NULL, NULL);
  472. pApp->WriteProfileInt(_T("Shaders"), _T("Initialized"), 1);
  474. for(int i = 0, id = 0; i < m_labels.GetCount(); i++)
  475. {
  476. CString label;
  477. m_labels.GetLBText(i, label);
  479. shader_t s;
  480. if(!label.IsEmpty() && m_shaders.Lookup(label, s))
  481. {
  482. CString str;
  483. str.Format(_T("%d"), id++);
  484. s.srcdata.Replace(_T("r"), _T(""));
  485. s.srcdata.Replace(_T("n"), _T("\n"));
  486. s.srcdata.Replace(_T("t"), _T("\t"));
  487. AfxGetApp()->WriteProfileString(_T("Shaders"), str, label + _T("|") + s.target + _T("|") + s.srcdata);
  488. }
  489. }
  490. }
  492. __super::OnClose();
  493. }
  495. void CShaderEditorDlg::OnLButtonDown(UINT nFlags, CPoint point)
  496. {
  497. if(HitTestSplitter(point))
  498. {
  499. m_fSplitterGrabbed = true;
  500. SetCapture();
  501. }
  503. __super::OnLButtonDown(nFlags, point);
  504. }
  506. void CShaderEditorDlg::OnLButtonUp(UINT nFlags, CPoint point)
  507. {
  508. if(m_fSplitterGrabbed)
  509. {
  510. ReleaseCapture();
  511. m_fSplitterGrabbed = false;
  512. }
  514. __super::OnLButtonUp(nFlags, point);
  515. }
  517. void CShaderEditorDlg::OnMouseMove(UINT nFlags, CPoint point)
  518. {
  519. if(m_fSplitterGrabbed)
  520. {
  521. CRect r, rs, ro;
  522. GetClientRect(&r);
  523. m_srcdata.GetWindowRect(&rs);
  524. m_output.GetWindowRect(&ro);
  525. ScreenToClient(&rs);
  526. ScreenToClient(&ro);
  528. int dist = ro.top - rs.bottom;
  529. int avgdist = dist / 2;
  531. rs.bottom = min(max(point.y, rs.top + 40), ro.bottom - 40) - avgdist;
  532. ro.top = rs.bottom + dist;
  533. m_srcdata.MoveWindow(&rs);
  534. m_output.MoveWindow(&ro);
  536. int div = 100 * ((rs.bottom + ro.top) / 2) / (ro.bottom - rs.top);
  538. RemoveAnchor(IDC_EDIT1);
  539. RemoveAnchor(IDC_EDIT2);
  540. AddAnchor(IDC_EDIT1, TOP_LEFT, CSize(100, div)/*BOTTOM_RIGHT*/);
  541. AddAnchor(IDC_EDIT2, CSize(0, div)/*BOTTOM_LEFT*/, BOTTOM_RIGHT);
  542. }
  544. __super::OnMouseMove(nFlags, point);
  545. }
  547. BOOL CShaderEditorDlg::OnSetCursor(CWnd* pWnd, UINT nHitTest, UINT message)
  548. {
  549. CPoint p;
  550. GetCursorPos(&p);
  551. ScreenToClient(&p);
  552. if(HitTestSplitter(p))
  553. {
  554. ::SetCursor(AfxGetApp()->LoadStandardCursor(IDC_SIZENS));
  555. return TRUE;
  556. }
  558. return __super::OnSetCursor(pWnd, nHitTest, message);
  559. }