window-basic-preview.cpp 19 KB

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