1
0

window-basic-preview.cpp 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747
  1. #include <QGuiApplication>
  2. #include <QMouseEvent>
  3. #include <cmath>
  4. #include <graphics/vec4.h>
  5. #include <graphics/matrix4.h>
  6. #include "window-basic-preview.hpp"
  7. #include "window-basic-main.hpp"
  8. #include "obs-app.hpp"
  9. #define HANDLE_RADIUS 4.0f
  10. #define HANDLE_SEL_RADIUS (HANDLE_RADIUS * 1.5f)
  11. #define CLAMP_DISTANCE 10.0f
  12. /* TODO: make C++ math classes and clean up code here later */
  13. OBSBasicPreview::OBSBasicPreview(QWidget *parent, Qt::WindowFlags flags)
  14. : OBSQTDisplay(parent, flags)
  15. {
  16. setMouseTracking(true);
  17. }
  18. vec2 OBSBasicPreview::GetMouseEventPos(QMouseEvent *event)
  19. {
  20. OBSBasic *main = reinterpret_cast<OBSBasic*>(App()->GetMainWindow());
  21. vec2 pos;
  22. vec2_set(&pos,
  23. (float(event->x()) - main->previewX) / main->previewScale,
  24. (float(event->y()) - main->previewY) / main->previewScale);
  25. return pos;
  26. }
  27. struct SceneFindData {
  28. const vec2 &pos;
  29. OBSSceneItem item;
  30. bool selectBelow;
  31. SceneFindData(const SceneFindData &) = delete;
  32. SceneFindData(SceneFindData &&) = delete;
  33. SceneFindData& operator=(const SceneFindData &) = delete;
  34. SceneFindData& operator=(SceneFindData &&) = delete;
  35. inline SceneFindData(const vec2 &pos_, bool selectBelow_)
  36. : pos (pos_),
  37. selectBelow (selectBelow_)
  38. {}
  39. };
  40. static bool SceneItemHasVideo(obs_sceneitem_t item)
  41. {
  42. obs_source_t source = obs_sceneitem_getsource(item);
  43. uint32_t flags = obs_source_get_output_flags(source);
  44. return (flags & OBS_SOURCE_VIDEO) != 0;
  45. }
  46. static bool FindItemAtPos(obs_scene_t scene, obs_sceneitem_t item, void *param)
  47. {
  48. SceneFindData *data = reinterpret_cast<SceneFindData*>(param);
  49. matrix4 transform;
  50. vec3 transformedPos;
  51. vec3 pos3;
  52. if (!SceneItemHasVideo(item))
  53. return true;
  54. vec3_set(&pos3, data->pos.x, data->pos.y, 0.0f);
  55. obs_sceneitem_get_box_transform(item, &transform);
  56. matrix4_inv(&transform, &transform);
  57. vec3_transform(&transformedPos, &pos3, &transform);
  58. if (transformedPos.x >= 0.0f && transformedPos.x <= 1.0f &&
  59. transformedPos.y >= 0.0f && transformedPos.y <= 1.0f) {
  60. if (data->selectBelow && obs_sceneitem_selected(item)) {
  61. if (data->item)
  62. return false;
  63. else
  64. data->selectBelow = false;
  65. }
  66. data->item = item;
  67. }
  68. UNUSED_PARAMETER(scene);
  69. return true;
  70. }
  71. static vec3 GetTransformedPos(float x, float y, const matrix4 &mat)
  72. {
  73. vec3 result;
  74. vec3_set(&result, x, y, 0.0f);
  75. vec3_transform(&result, &result, &mat);
  76. return result;
  77. }
  78. static vec3 GetTransformedPosScaled(float x, float y, const matrix4 &mat,
  79. float scale)
  80. {
  81. vec3 result;
  82. vec3_set(&result, x, y, 0.0f);
  83. vec3_transform(&result, &result, &mat);
  84. vec3_mulf(&result, &result, scale);
  85. return result;
  86. }
  87. static inline vec2 GetOBSScreenSize()
  88. {
  89. obs_video_info ovi;
  90. vec2 size;
  91. vec2_zero(&size);
  92. if (obs_get_video_info(&ovi)) {
  93. size.x = float(ovi.base_width);
  94. size.y = float(ovi.base_height);
  95. }
  96. return size;
  97. }
  98. vec3 OBSBasicPreview::GetScreenSnapOffset(const vec3 &tl, const vec3 &br)
  99. {
  100. OBSBasic *main = reinterpret_cast<OBSBasic*>(App()->GetMainWindow());
  101. vec2 screenSize = GetOBSScreenSize();
  102. vec3 clampOffset;
  103. vec3_zero(&clampOffset);
  104. const float clampDist = CLAMP_DISTANCE / main->previewScale;
  105. if (fabsf(tl.x) < clampDist)
  106. clampOffset.x = -tl.x;
  107. if (fabsf(clampOffset.x) < EPSILON &&
  108. fabsf(screenSize.x - br.x) < clampDist)
  109. clampOffset.x = screenSize.x - br.x;
  110. if (fabsf(tl.y) < clampDist)
  111. clampOffset.y = -tl.y;
  112. if (fabsf(clampOffset.y) < EPSILON &&
  113. fabsf(screenSize.y - br.y) < clampDist)
  114. clampOffset.y = screenSize.y - br.y;
  115. return clampOffset;
  116. }
  117. OBSSceneItem OBSBasicPreview::GetItemAtPos(const vec2 &pos, bool selectBelow)
  118. {
  119. OBSBasic *main = reinterpret_cast<OBSBasic*>(App()->GetMainWindow());
  120. OBSScene scene = main->GetCurrentScene();
  121. if (!scene)
  122. return OBSSceneItem();
  123. SceneFindData data(pos, selectBelow);
  124. obs_scene_enum_items(scene, FindItemAtPos, &data);
  125. return data.item;
  126. }
  127. static bool CheckItemSelected(obs_scene_t scene, obs_sceneitem_t item,
  128. void *param)
  129. {
  130. SceneFindData *data = reinterpret_cast<SceneFindData*>(param);
  131. matrix4 transform;
  132. vec3 transformedPos;
  133. vec3 pos3;
  134. if (!SceneItemHasVideo(item))
  135. return true;
  136. vec3_set(&pos3, data->pos.x, data->pos.y, 0.0f);
  137. obs_sceneitem_get_box_transform(item, &transform);
  138. matrix4_inv(&transform, &transform);
  139. vec3_transform(&transformedPos, &pos3, &transform);
  140. if (transformedPos.x >= 0.0f && transformedPos.x <= 1.0f &&
  141. transformedPos.y >= 0.0f && transformedPos.y <= 1.0f) {
  142. if (obs_sceneitem_selected(item)) {
  143. data->item = item;
  144. return false;
  145. }
  146. }
  147. UNUSED_PARAMETER(scene);
  148. return true;
  149. }
  150. bool OBSBasicPreview::SelectedAtPos(const vec2 &pos)
  151. {
  152. OBSBasic *main = reinterpret_cast<OBSBasic*>(App()->GetMainWindow());
  153. OBSScene scene = main->GetCurrentScene();
  154. if (!scene)
  155. return false;
  156. SceneFindData data(pos, false);
  157. obs_scene_enum_items(scene, CheckItemSelected, &data);
  158. return !!data.item;
  159. }
  160. struct HandleFindData {
  161. const vec2 &pos;
  162. const float scale;
  163. OBSSceneItem item;
  164. ItemHandle handle = ItemHandle::None;
  165. HandleFindData(const HandleFindData &) = delete;
  166. HandleFindData(HandleFindData &&) = delete;
  167. HandleFindData& operator=(const HandleFindData &) = delete;
  168. HandleFindData& operator=(HandleFindData &&) = delete;
  169. inline HandleFindData(const vec2 &pos_, float scale_)
  170. : pos (pos_),
  171. scale (scale_)
  172. {}
  173. };
  174. static bool FindHandleAtPos(obs_scene_t scene, obs_sceneitem_t item,
  175. void *param)
  176. {
  177. if (!obs_sceneitem_selected(item))
  178. return true;
  179. HandleFindData *data = reinterpret_cast<HandleFindData*>(param);
  180. matrix4 transform;
  181. vec3 pos3;
  182. float closestHandle = HANDLE_SEL_RADIUS;
  183. vec3_set(&pos3, data->pos.x, data->pos.y, 0.0f);
  184. obs_sceneitem_get_box_transform(item, &transform);
  185. auto TestHandle = [&] (float x, float y, ItemHandle handle)
  186. {
  187. vec3 handlePos = GetTransformedPosScaled(x, y, transform,
  188. data->scale);
  189. float dist = vec3_dist(&handlePos, &pos3);
  190. if (dist < HANDLE_SEL_RADIUS) {
  191. if (dist < closestHandle) {
  192. closestHandle = dist;
  193. data->handle = handle;
  194. data->item = item;
  195. }
  196. }
  197. };
  198. TestHandle(0.0f, 0.0f, ItemHandle::TopLeft);
  199. TestHandle(0.5f, 0.0f, ItemHandle::TopCenter);
  200. TestHandle(1.0f, 0.0f, ItemHandle::TopRight);
  201. TestHandle(0.0f, 0.5f, ItemHandle::CenterLeft);
  202. TestHandle(1.0f, 0.5f, ItemHandle::CenterRight);
  203. TestHandle(0.0f, 1.0f, ItemHandle::BottomLeft);
  204. TestHandle(0.5f, 1.0f, ItemHandle::BottomCenter);
  205. TestHandle(1.0f, 1.0f, ItemHandle::BottomRight);
  206. UNUSED_PARAMETER(scene);
  207. return true;
  208. }
  209. static vec2 GetItemSize(obs_sceneitem_t item)
  210. {
  211. obs_bounds_type boundsType = obs_sceneitem_get_bounds_type(item);
  212. vec2 size;
  213. if (boundsType != OBS_BOUNDS_NONE) {
  214. obs_sceneitem_get_bounds(item, &size);
  215. } else {
  216. obs_source_t source = obs_sceneitem_getsource(item);
  217. vec2 scale;
  218. obs_sceneitem_getscale(item, &scale);
  219. size.x = float(obs_source_getwidth(source)) * scale.x;
  220. size.y = float(obs_source_getheight(source)) * scale.y;
  221. }
  222. return size;
  223. }
  224. void OBSBasicPreview::GetStretchHandleData(const vec2 &pos)
  225. {
  226. OBSBasic *main = reinterpret_cast<OBSBasic*>(App()->GetMainWindow());
  227. OBSScene scene = main->GetCurrentScene();
  228. if (!scene)
  229. return;
  230. HandleFindData data(pos, main->previewScale);
  231. obs_scene_enum_items(scene, FindHandleAtPos, &data);
  232. stretchItem = std::move(data.item);
  233. stretchHandle = data.handle;
  234. if (stretchHandle != ItemHandle::None) {
  235. matrix4 boxTransform;
  236. vec3 itemUL;
  237. float itemRot;
  238. stretchItemSize = GetItemSize(stretchItem);
  239. obs_sceneitem_get_box_transform(stretchItem, &boxTransform);
  240. itemRot = obs_sceneitem_getrot(stretchItem);
  241. vec3_from_vec4(&itemUL, &boxTransform.t);
  242. /* build the item space conversion matrices */
  243. matrix4_identity(&itemToScreen);
  244. matrix4_rotate_aa4f(&itemToScreen, &itemToScreen,
  245. 0.0f, 0.0f, 1.0f, RAD(itemRot));
  246. matrix4_translate3f(&itemToScreen, &itemToScreen,
  247. itemUL.x, itemUL.y, 0.0f);
  248. matrix4_identity(&screenToItem);
  249. matrix4_translate3f(&screenToItem, &screenToItem,
  250. -itemUL.x, -itemUL.y, 0.0f);
  251. matrix4_rotate_aa4f(&screenToItem, &screenToItem,
  252. 0.0f, 0.0f, 1.0f, RAD(-itemRot));
  253. }
  254. }
  255. void OBSBasicPreview::mousePressEvent(QMouseEvent *event)
  256. {
  257. OBSBasic *main = reinterpret_cast<OBSBasic*>(App()->GetMainWindow());
  258. float x = float(event->x()) - main->previewX;
  259. float y = float(event->y()) - main->previewY;
  260. if (event->button() != Qt::LeftButton)
  261. return;
  262. mouseDown = true;
  263. vec2_set(&startPos, x, y);
  264. GetStretchHandleData(startPos);
  265. vec2_divf(&startPos, &startPos, main->previewScale);
  266. startPos.x = std::round(startPos.x);
  267. startPos.y = std::round(startPos.y);
  268. mouseOverItems = SelectedAtPos(startPos);
  269. vec2_zero(&lastMoveOffset);
  270. }
  271. static bool select_one(obs_scene_t scene, obs_sceneitem_t item, void *param)
  272. {
  273. obs_sceneitem_t selectedItem = reinterpret_cast<obs_sceneitem_t>(param);
  274. obs_sceneitem_select(item, (selectedItem == item));
  275. UNUSED_PARAMETER(scene);
  276. return true;
  277. }
  278. void OBSBasicPreview::DoSelect(const vec2 &pos)
  279. {
  280. OBSBasic *main = reinterpret_cast<OBSBasic*>(App()->GetMainWindow());
  281. OBSScene scene = main->GetCurrentScene();
  282. OBSSceneItem item = GetItemAtPos(pos, true);
  283. obs_scene_enum_items(scene, select_one, (obs_sceneitem_t)item);
  284. }
  285. void OBSBasicPreview::DoCtrlSelect(const vec2 &pos)
  286. {
  287. OBSSceneItem item = GetItemAtPos(pos, false);
  288. if (!item)
  289. return;
  290. bool selected = obs_sceneitem_selected(item);
  291. obs_sceneitem_select(item, !selected);
  292. }
  293. void OBSBasicPreview::ProcessClick(const vec2 &pos)
  294. {
  295. Qt::KeyboardModifiers modifiers = QGuiApplication::keyboardModifiers();
  296. if (modifiers & Qt::ControlModifier)
  297. DoCtrlSelect(pos);
  298. else
  299. DoSelect(pos);
  300. }
  301. void OBSBasicPreview::mouseReleaseEvent(QMouseEvent *event)
  302. {
  303. if (mouseDown) {
  304. vec2 pos = GetMouseEventPos(event);
  305. if (!mouseMoved)
  306. ProcessClick(pos);
  307. stretchItem = nullptr;
  308. mouseDown = false;
  309. mouseMoved = false;
  310. }
  311. }
  312. struct SelectedItemBounds {
  313. bool first = true;
  314. vec3 tl, br;
  315. };
  316. static bool AddItemBounds(obs_scene_t scene, obs_sceneitem_t item,
  317. void *param)
  318. {
  319. SelectedItemBounds *data = reinterpret_cast<SelectedItemBounds*>(param);
  320. if (!obs_sceneitem_selected(item))
  321. return true;
  322. matrix4 boxTransform;
  323. obs_sceneitem_get_box_transform(item, &boxTransform);
  324. vec3 t[4] = {
  325. GetTransformedPos(0.0f, 0.0f, boxTransform),
  326. GetTransformedPos(1.0f, 0.0f, boxTransform),
  327. GetTransformedPos(0.0f, 1.0f, boxTransform),
  328. GetTransformedPos(1.0f, 1.0f, boxTransform)
  329. };
  330. for (const vec3 &v : t) {
  331. if (data->first) {
  332. vec3_copy(&data->tl, &v);
  333. vec3_copy(&data->br, &v);
  334. data->first = false;
  335. } else {
  336. vec3_min(&data->tl, &data->tl, &v);
  337. vec3_max(&data->br, &data->br, &v);
  338. }
  339. }
  340. UNUSED_PARAMETER(scene);
  341. return true;
  342. }
  343. void OBSBasicPreview::SnapItemMovement(vec2 &offset)
  344. {
  345. OBSBasic *main = reinterpret_cast<OBSBasic*>(App()->GetMainWindow());
  346. OBSScene scene = main->GetCurrentScene();
  347. SelectedItemBounds data;
  348. obs_scene_enum_items(scene, AddItemBounds, &data);
  349. data.tl.x += offset.x;
  350. data.tl.y += offset.y;
  351. data.br.x += offset.x;
  352. data.br.y += offset.y;
  353. vec3 snapOffset = GetScreenSnapOffset(data.tl, data.br);
  354. offset.x += snapOffset.x;
  355. offset.y += snapOffset.y;
  356. }
  357. static bool move_items(obs_scene_t scene, obs_sceneitem_t item, void *param)
  358. {
  359. vec2 *offset = reinterpret_cast<vec2*>(param);
  360. if (obs_sceneitem_selected(item)) {
  361. vec2 pos;
  362. obs_sceneitem_getpos(item, &pos);
  363. vec2_add(&pos, &pos, offset);
  364. obs_sceneitem_setpos(item, &pos);
  365. }
  366. UNUSED_PARAMETER(scene);
  367. return true;
  368. }
  369. void OBSBasicPreview::MoveItems(const vec2 &pos)
  370. {
  371. Qt::KeyboardModifiers modifiers = QGuiApplication::keyboardModifiers();
  372. OBSBasic *main = reinterpret_cast<OBSBasic*>(App()->GetMainWindow());
  373. OBSScene scene = main->GetCurrentScene();
  374. vec2 offset, moveOffset;
  375. vec2_sub(&offset, &pos, &startPos);
  376. vec2_sub(&moveOffset, &offset, &lastMoveOffset);
  377. if (!(modifiers & Qt::ControlModifier))
  378. SnapItemMovement(moveOffset);
  379. vec2_add(&lastMoveOffset, &lastMoveOffset, &moveOffset);
  380. obs_scene_enum_items(scene, move_items, &moveOffset);
  381. }
  382. vec3 OBSBasicPreview::CalculateStretchPos(const vec3 &tl, const vec3 &br)
  383. {
  384. uint32_t alignment = obs_sceneitem_getalignment(stretchItem);
  385. vec3 pos;
  386. vec3_zero(&pos);
  387. if (alignment & OBS_ALIGN_LEFT)
  388. pos.x = tl.x;
  389. else if (alignment & OBS_ALIGN_RIGHT)
  390. pos.x = br.x;
  391. else
  392. pos.x = (br.x - tl.x) * 0.5f + tl.x;
  393. if (alignment & OBS_ALIGN_TOP)
  394. pos.y = tl.y;
  395. else if (alignment & OBS_ALIGN_BOTTOM)
  396. pos.y = br.y;
  397. else
  398. pos.y = (br.y - tl.y) * 0.5f + tl.y;
  399. return pos;
  400. }
  401. void OBSBasicPreview::ClampAspect(vec3 &tl, vec3 &br, vec2 &size,
  402. const vec2 &baseSize)
  403. {
  404. float baseAspect = baseSize.x / baseSize.y;
  405. float aspect = size.x / size.y;
  406. uint32_t stretchFlags = (uint32_t)stretchHandle;
  407. if (stretchHandle == ItemHandle::TopLeft ||
  408. stretchHandle == ItemHandle::TopRight ||
  409. stretchHandle == ItemHandle::BottomLeft ||
  410. stretchHandle == ItemHandle::BottomRight) {
  411. if (aspect < baseAspect)
  412. size.x = size.y * baseAspect;
  413. else
  414. size.y = size.x / baseAspect;
  415. } else if (stretchHandle == ItemHandle::TopCenter ||
  416. stretchHandle == ItemHandle::BottomCenter) {
  417. size.x = size.y * baseAspect;
  418. } else if (stretchHandle == ItemHandle::CenterLeft ||
  419. stretchHandle == ItemHandle::CenterRight) {
  420. size.y = size.x / baseAspect;
  421. }
  422. size.x = std::round(size.x);
  423. size.y = std::round(size.y);
  424. if (stretchFlags & ITEM_LEFT)
  425. tl.x = br.x - size.x;
  426. else if (stretchFlags & ITEM_RIGHT)
  427. br.x = tl.x + size.x;
  428. if (stretchFlags & ITEM_TOP)
  429. tl.y = br.y - size.y;
  430. else if (stretchFlags & ITEM_BOTTOM)
  431. br.y = tl.y + size.y;
  432. }
  433. void OBSBasicPreview::SnapStretchingToScreen(vec3 &tl, vec3 &br)
  434. {
  435. uint32_t stretchFlags = (uint32_t)stretchHandle;
  436. vec3 newTL = GetTransformedPos(tl.x, tl.y, itemToScreen);
  437. vec3 newTR = GetTransformedPos(br.x, tl.y, itemToScreen);
  438. vec3 newBL = GetTransformedPos(tl.x, br.y, itemToScreen);
  439. vec3 newBR = GetTransformedPos(br.x, br.y, itemToScreen);
  440. vec3 boundingTL;
  441. vec3 boundingBR;
  442. vec3_copy(&boundingTL, &newTL);
  443. vec3_min(&boundingTL, &boundingTL, &newTR);
  444. vec3_min(&boundingTL, &boundingTL, &newBL);
  445. vec3_min(&boundingTL, &boundingTL, &newBR);
  446. vec3_copy(&boundingBR, &newTL);
  447. vec3_max(&boundingBR, &boundingBR, &newTR);
  448. vec3_max(&boundingBR, &boundingBR, &newBL);
  449. vec3_max(&boundingBR, &boundingBR, &newBR);
  450. vec3 offset = GetScreenSnapOffset(boundingTL, boundingBR);
  451. vec3_add(&offset, &offset, &newTL);
  452. vec3_transform(&offset, &offset, &screenToItem);
  453. vec3_sub(&offset, &offset, &tl);
  454. if (stretchFlags & ITEM_LEFT)
  455. tl.x += offset.x;
  456. else if (stretchFlags & ITEM_RIGHT)
  457. br.x += offset.x;
  458. if (stretchFlags & ITEM_TOP)
  459. tl.y += offset.y;
  460. else if (stretchFlags & ITEM_BOTTOM)
  461. br.y += offset.y;
  462. }
  463. void OBSBasicPreview::StretchItem(const vec2 &pos)
  464. {
  465. Qt::KeyboardModifiers modifiers = QGuiApplication::keyboardModifiers();
  466. obs_bounds_type boundsType = obs_sceneitem_get_bounds_type(stretchItem);
  467. uint32_t stretchFlags = (uint32_t)stretchHandle;
  468. bool shiftDown = (modifiers & Qt::ShiftModifier);
  469. vec3 tl, br, pos3;
  470. vec3_zero(&tl);
  471. vec3_set(&br, stretchItemSize.x, stretchItemSize.y, 0.0f);
  472. vec3_set(&pos3, pos.x, pos.y, 0.0f);
  473. vec3_transform(&pos3, &pos3, &screenToItem);
  474. if (stretchFlags & ITEM_LEFT)
  475. tl.x = pos3.x;
  476. else if (stretchFlags & ITEM_RIGHT)
  477. br.x = pos3.x;
  478. if (stretchFlags & ITEM_TOP)
  479. tl.y = pos3.y;
  480. else if (stretchFlags & ITEM_BOTTOM)
  481. br.y = pos3.y;
  482. if (!(modifiers & Qt::ControlModifier))
  483. SnapStretchingToScreen(tl, br);
  484. obs_source_t source = obs_sceneitem_getsource(stretchItem);
  485. vec2 baseSize;
  486. vec2_set(&baseSize,
  487. float(obs_source_getwidth(source)),
  488. float(obs_source_getheight(source)));
  489. vec2 size;
  490. vec2_set(&size,br. x - tl.x, br.y - tl.y);
  491. if (boundsType != OBS_BOUNDS_NONE) {
  492. if (shiftDown)
  493. ClampAspect(tl, br, size, baseSize);
  494. if (tl.x > br.x) std::swap(tl.x, br.x);
  495. if (tl.y > br.y) std::swap(tl.y, br.y);
  496. vec2_abs(&size, &size);
  497. obs_sceneitem_set_bounds(stretchItem, &size);
  498. } else {
  499. if (!shiftDown)
  500. ClampAspect(tl, br, size, baseSize);
  501. vec2_div(&size, &size, &baseSize);
  502. obs_sceneitem_setscale(stretchItem, &size);
  503. }
  504. pos3 = CalculateStretchPos(tl, br);
  505. vec3_transform(&pos3, &pos3, &itemToScreen);
  506. vec2 newPos;
  507. vec2_set(&newPos, std::round(pos3.x), std::round(pos3.y));
  508. obs_sceneitem_setpos(stretchItem, &newPos);
  509. }
  510. void OBSBasicPreview::mouseMoveEvent(QMouseEvent *event)
  511. {
  512. if (mouseDown) {
  513. vec2 pos = GetMouseEventPos(event);
  514. if (!mouseMoved && !mouseOverItems &&
  515. stretchHandle == ItemHandle::None) {
  516. ProcessClick(startPos);
  517. mouseOverItems = SelectedAtPos(startPos);
  518. }
  519. pos.x = std::round(pos.x);
  520. pos.y = std::round(pos.y);
  521. if (stretchHandle != ItemHandle::None)
  522. StretchItem(pos);
  523. else if (mouseOverItems)
  524. MoveItems(pos);
  525. mouseMoved = true;
  526. }
  527. }
  528. static void DrawCircleAtPos(float x, float y, matrix4 &matrix,
  529. float previewScale)
  530. {
  531. struct vec3 pos;
  532. vec3_set(&pos, x, y, 0.0f);
  533. vec3_transform(&pos, &pos, &matrix);
  534. vec3_mulf(&pos, &pos, previewScale);
  535. gs_matrix_push();
  536. gs_matrix_translate(&pos);
  537. gs_draw(GS_LINESTRIP, 0, 0);
  538. gs_matrix_pop();
  539. }
  540. bool OBSBasicPreview::DrawSelectedItem(obs_scene_t scene, obs_sceneitem_t item,
  541. void *param)
  542. {
  543. if (!obs_sceneitem_selected(item))
  544. return true;
  545. OBSBasic *main = reinterpret_cast<OBSBasic*>(App()->GetMainWindow());
  546. gs_load_vertexbuffer(main->circle);
  547. matrix4 boxTransform;
  548. obs_sceneitem_get_box_transform(item, &boxTransform);
  549. gs_matrix_push();
  550. gs_matrix_scale3f(HANDLE_RADIUS, HANDLE_RADIUS, 1.0f);
  551. DrawCircleAtPos(0.0f, 0.0f, boxTransform, main->previewScale);
  552. DrawCircleAtPos(0.0f, 1.0f, boxTransform, main->previewScale);
  553. DrawCircleAtPos(1.0f, 0.0f, boxTransform, main->previewScale);
  554. DrawCircleAtPos(1.0f, 1.0f, boxTransform, main->previewScale);
  555. DrawCircleAtPos(0.5f, 0.0f, boxTransform, main->previewScale);
  556. DrawCircleAtPos(0.0f, 0.5f, boxTransform, main->previewScale);
  557. DrawCircleAtPos(0.5f, 1.0f, boxTransform, main->previewScale);
  558. DrawCircleAtPos(1.0f, 0.5f, boxTransform, main->previewScale);
  559. gs_matrix_pop();
  560. gs_load_vertexbuffer(main->box);
  561. gs_matrix_push();
  562. gs_matrix_set(&boxTransform);
  563. gs_matrix_scale3f(main->previewScale, main->previewScale, 1.0f);
  564. gs_draw(GS_LINESTRIP, 0, 0);
  565. gs_matrix_pop();
  566. UNUSED_PARAMETER(scene);
  567. UNUSED_PARAMETER(param);
  568. return true;
  569. }
  570. void OBSBasicPreview::DrawSceneEditing()
  571. {
  572. OBSBasic *main = reinterpret_cast<OBSBasic*>(App()->GetMainWindow());
  573. effect_t solid = obs_get_solid_effect();
  574. technique_t tech = effect_gettechnique(solid, "Solid");
  575. vec4 color;
  576. vec4_set(&color, 1.0f, 0.0f, 0.0f, 1.0f);
  577. effect_setvec4(effect_getparambyname(solid, "color"), &color);
  578. technique_begin(tech);
  579. technique_beginpass(tech, 0);
  580. OBSScene scene = main->GetCurrentScene();
  581. if (scene)
  582. obs_scene_enum_items(scene, DrawSelectedItem, this);
  583. gs_load_vertexbuffer(nullptr);
  584. technique_endpass(tech);
  585. technique_end(tech);
  586. }