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