window-basic-preview.cpp 19 KB

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