window-basic-preview.cpp 40 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607
  1. #include <QGuiApplication>
  2. #include <QMouseEvent>
  3. #include <algorithm>
  4. #include <cmath>
  5. #include <string>
  6. #include <graphics/vec4.h>
  7. #include <graphics/matrix4.h>
  8. #include "window-basic-preview.hpp"
  9. #include "window-basic-main.hpp"
  10. #include "obs-app.hpp"
  11. #include "platform.hpp"
  12. #define HANDLE_RADIUS 4.0f
  13. #define HANDLE_SEL_RADIUS (HANDLE_RADIUS * 1.5f)
  14. #if QT_VERSION >= QT_VERSION_CHECK(5, 6, 0)
  15. #define SUPPORTS_FRACTIONAL_SCALING
  16. #endif
  17. /* TODO: make C++ math classes and clean up code here later */
  18. OBSBasicPreview::OBSBasicPreview(QWidget *parent, Qt::WindowFlags flags)
  19. : OBSQTDisplay(parent, flags)
  20. {
  21. ResetScrollingOffset();
  22. setMouseTracking(true);
  23. }
  24. OBSBasicPreview::~OBSBasicPreview()
  25. {
  26. if (overflow) {
  27. obs_enter_graphics();
  28. gs_texture_destroy(overflow);
  29. obs_leave_graphics();
  30. }
  31. }
  32. vec2 OBSBasicPreview::GetMouseEventPos(QMouseEvent *event)
  33. {
  34. OBSBasic *main = reinterpret_cast<OBSBasic *>(App()->GetMainWindow());
  35. #ifdef SUPPORTS_FRACTIONAL_SCALING
  36. float pixelRatio = main->devicePixelRatioF();
  37. #else
  38. float pixelRatio = main->devicePixelRatio();
  39. #endif
  40. float scale = pixelRatio / main->previewScale;
  41. vec2 pos;
  42. vec2_set(&pos,
  43. (float(event->x()) - main->previewX / pixelRatio) * scale,
  44. (float(event->y()) - main->previewY / pixelRatio) * scale);
  45. return pos;
  46. }
  47. struct SceneFindData {
  48. const vec2 &pos;
  49. OBSSceneItem item;
  50. bool selectBelow;
  51. obs_sceneitem_t *group = nullptr;
  52. SceneFindData(const SceneFindData &) = delete;
  53. SceneFindData(SceneFindData &&) = delete;
  54. SceneFindData &operator=(const SceneFindData &) = delete;
  55. SceneFindData &operator=(SceneFindData &&) = delete;
  56. inline SceneFindData(const vec2 &pos_, bool selectBelow_)
  57. : pos(pos_), selectBelow(selectBelow_)
  58. {
  59. }
  60. };
  61. static bool SceneItemHasVideo(obs_sceneitem_t *item)
  62. {
  63. obs_source_t *source = obs_sceneitem_get_source(item);
  64. uint32_t flags = obs_source_get_output_flags(source);
  65. return (flags & OBS_SOURCE_VIDEO) != 0;
  66. }
  67. static bool CloseFloat(float a, float b, float epsilon = 0.01)
  68. {
  69. using std::abs;
  70. return abs(a - b) <= epsilon;
  71. }
  72. static bool FindItemAtPos(obs_scene_t *scene, obs_sceneitem_t *item,
  73. void *param)
  74. {
  75. SceneFindData *data = reinterpret_cast<SceneFindData *>(param);
  76. matrix4 transform;
  77. matrix4 invTransform;
  78. vec3 transformedPos;
  79. vec3 pos3;
  80. vec3 pos3_;
  81. if (!SceneItemHasVideo(item))
  82. return true;
  83. if (obs_sceneitem_locked(item))
  84. return true;
  85. vec3_set(&pos3, data->pos.x, data->pos.y, 0.0f);
  86. obs_sceneitem_get_box_transform(item, &transform);
  87. matrix4_inv(&invTransform, &transform);
  88. vec3_transform(&transformedPos, &pos3, &invTransform);
  89. vec3_transform(&pos3_, &transformedPos, &transform);
  90. if (CloseFloat(pos3.x, pos3_.x) && CloseFloat(pos3.y, pos3_.y) &&
  91. transformedPos.x >= 0.0f && transformedPos.x <= 1.0f &&
  92. transformedPos.y >= 0.0f && transformedPos.y <= 1.0f) {
  93. if (data->selectBelow && obs_sceneitem_selected(item)) {
  94. if (data->item)
  95. return false;
  96. else
  97. data->selectBelow = false;
  98. }
  99. data->item = item;
  100. }
  101. UNUSED_PARAMETER(scene);
  102. return true;
  103. }
  104. static vec3 GetTransformedPos(float x, float y, const matrix4 &mat)
  105. {
  106. vec3 result;
  107. vec3_set(&result, x, y, 0.0f);
  108. vec3_transform(&result, &result, &mat);
  109. return result;
  110. }
  111. static inline vec2 GetOBSScreenSize()
  112. {
  113. obs_video_info ovi;
  114. vec2 size;
  115. vec2_zero(&size);
  116. if (obs_get_video_info(&ovi)) {
  117. size.x = float(ovi.base_width);
  118. size.y = float(ovi.base_height);
  119. }
  120. return size;
  121. }
  122. vec3 OBSBasicPreview::GetSnapOffset(const vec3 &tl, const vec3 &br)
  123. {
  124. OBSBasic *main = reinterpret_cast<OBSBasic *>(App()->GetMainWindow());
  125. vec2 screenSize = GetOBSScreenSize();
  126. vec3 clampOffset;
  127. vec3_zero(&clampOffset);
  128. const bool snap = config_get_bool(GetGlobalConfig(), "BasicWindow",
  129. "SnappingEnabled");
  130. if (snap == false)
  131. return clampOffset;
  132. const bool screenSnap = config_get_bool(
  133. GetGlobalConfig(), "BasicWindow", "ScreenSnapping");
  134. const bool centerSnap = config_get_bool(
  135. GetGlobalConfig(), "BasicWindow", "CenterSnapping");
  136. const float clampDist = config_get_double(GetGlobalConfig(),
  137. "BasicWindow",
  138. "SnapDistance") /
  139. main->previewScale;
  140. const float centerX = br.x - (br.x - tl.x) / 2.0f;
  141. const float centerY = br.y - (br.y - tl.y) / 2.0f;
  142. // Left screen edge.
  143. if (screenSnap && fabsf(tl.x) < clampDist)
  144. clampOffset.x = -tl.x;
  145. // Right screen edge.
  146. if (screenSnap && fabsf(clampOffset.x) < EPSILON &&
  147. fabsf(screenSize.x - br.x) < clampDist)
  148. clampOffset.x = screenSize.x - br.x;
  149. // Horizontal center.
  150. if (centerSnap && fabsf(screenSize.x - (br.x - tl.x)) > clampDist &&
  151. fabsf(screenSize.x / 2.0f - centerX) < clampDist)
  152. clampOffset.x = screenSize.x / 2.0f - centerX;
  153. // Top screen edge.
  154. if (screenSnap && fabsf(tl.y) < clampDist)
  155. clampOffset.y = -tl.y;
  156. // Bottom screen edge.
  157. if (screenSnap && fabsf(clampOffset.y) < EPSILON &&
  158. fabsf(screenSize.y - br.y) < clampDist)
  159. clampOffset.y = screenSize.y - br.y;
  160. // Vertical center.
  161. if (centerSnap && fabsf(screenSize.y - (br.y - tl.y)) > clampDist &&
  162. fabsf(screenSize.y / 2.0f - centerY) < clampDist)
  163. clampOffset.y = screenSize.y / 2.0f - centerY;
  164. return clampOffset;
  165. }
  166. OBSSceneItem OBSBasicPreview::GetItemAtPos(const vec2 &pos, bool selectBelow)
  167. {
  168. OBSBasic *main = reinterpret_cast<OBSBasic *>(App()->GetMainWindow());
  169. OBSScene scene = main->GetCurrentScene();
  170. if (!scene)
  171. return OBSSceneItem();
  172. SceneFindData data(pos, selectBelow);
  173. obs_scene_enum_items(scene, FindItemAtPos, &data);
  174. return data.item;
  175. }
  176. static bool CheckItemSelected(obs_scene_t *scene, obs_sceneitem_t *item,
  177. void *param)
  178. {
  179. SceneFindData *data = reinterpret_cast<SceneFindData *>(param);
  180. matrix4 transform;
  181. vec3 transformedPos;
  182. vec3 pos3;
  183. if (!SceneItemHasVideo(item))
  184. return true;
  185. if (obs_sceneitem_is_group(item)) {
  186. data->group = item;
  187. obs_sceneitem_group_enum_items(item, CheckItemSelected, param);
  188. data->group = nullptr;
  189. if (data->item) {
  190. return false;
  191. }
  192. }
  193. vec3_set(&pos3, data->pos.x, data->pos.y, 0.0f);
  194. obs_sceneitem_get_box_transform(item, &transform);
  195. if (data->group) {
  196. matrix4 parent_transform;
  197. obs_sceneitem_get_draw_transform(data->group,
  198. &parent_transform);
  199. matrix4_mul(&transform, &transform, &parent_transform);
  200. }
  201. matrix4_inv(&transform, &transform);
  202. vec3_transform(&transformedPos, &pos3, &transform);
  203. if (transformedPos.x >= 0.0f && transformedPos.x <= 1.0f &&
  204. transformedPos.y >= 0.0f && transformedPos.y <= 1.0f) {
  205. if (obs_sceneitem_selected(item)) {
  206. data->item = item;
  207. return false;
  208. }
  209. }
  210. UNUSED_PARAMETER(scene);
  211. return true;
  212. }
  213. bool OBSBasicPreview::SelectedAtPos(const vec2 &pos)
  214. {
  215. OBSBasic *main = reinterpret_cast<OBSBasic *>(App()->GetMainWindow());
  216. OBSScene scene = main->GetCurrentScene();
  217. if (!scene)
  218. return false;
  219. SceneFindData data(pos, false);
  220. obs_scene_enum_items(scene, CheckItemSelected, &data);
  221. return !!data.item;
  222. }
  223. struct HandleFindData {
  224. const vec2 &pos;
  225. const float radius;
  226. matrix4 parent_xform;
  227. OBSSceneItem item;
  228. ItemHandle handle = ItemHandle::None;
  229. HandleFindData(const HandleFindData &) = delete;
  230. HandleFindData(HandleFindData &&) = delete;
  231. HandleFindData &operator=(const HandleFindData &) = delete;
  232. HandleFindData &operator=(HandleFindData &&) = delete;
  233. inline HandleFindData(const vec2 &pos_, float scale)
  234. : pos(pos_), radius(HANDLE_SEL_RADIUS / scale)
  235. {
  236. matrix4_identity(&parent_xform);
  237. }
  238. inline HandleFindData(const HandleFindData &hfd,
  239. obs_sceneitem_t *parent)
  240. : pos(hfd.pos),
  241. radius(hfd.radius),
  242. item(hfd.item),
  243. handle(hfd.handle)
  244. {
  245. obs_sceneitem_get_draw_transform(parent, &parent_xform);
  246. }
  247. };
  248. static bool FindHandleAtPos(obs_scene_t *scene, obs_sceneitem_t *item,
  249. void *param)
  250. {
  251. HandleFindData &data = *reinterpret_cast<HandleFindData *>(param);
  252. if (!obs_sceneitem_selected(item)) {
  253. if (obs_sceneitem_is_group(item)) {
  254. HandleFindData newData(data, item);
  255. obs_sceneitem_group_enum_items(item, FindHandleAtPos,
  256. &newData);
  257. data.item = newData.item;
  258. data.handle = newData.handle;
  259. }
  260. return true;
  261. }
  262. matrix4 transform;
  263. vec3 pos3;
  264. float closestHandle = data.radius;
  265. vec3_set(&pos3, data.pos.x, data.pos.y, 0.0f);
  266. obs_sceneitem_get_box_transform(item, &transform);
  267. auto TestHandle = [&](float x, float y, ItemHandle handle) {
  268. vec3 handlePos = GetTransformedPos(x, y, transform);
  269. vec3_transform(&handlePos, &handlePos, &data.parent_xform);
  270. float dist = vec3_dist(&handlePos, &pos3);
  271. if (dist < data.radius) {
  272. if (dist < closestHandle) {
  273. closestHandle = dist;
  274. data.handle = handle;
  275. data.item = item;
  276. }
  277. }
  278. };
  279. TestHandle(0.0f, 0.0f, ItemHandle::TopLeft);
  280. TestHandle(0.5f, 0.0f, ItemHandle::TopCenter);
  281. TestHandle(1.0f, 0.0f, ItemHandle::TopRight);
  282. TestHandle(0.0f, 0.5f, ItemHandle::CenterLeft);
  283. TestHandle(1.0f, 0.5f, ItemHandle::CenterRight);
  284. TestHandle(0.0f, 1.0f, ItemHandle::BottomLeft);
  285. TestHandle(0.5f, 1.0f, ItemHandle::BottomCenter);
  286. TestHandle(1.0f, 1.0f, ItemHandle::BottomRight);
  287. UNUSED_PARAMETER(scene);
  288. return true;
  289. }
  290. static vec2 GetItemSize(obs_sceneitem_t *item)
  291. {
  292. obs_bounds_type boundsType = obs_sceneitem_get_bounds_type(item);
  293. vec2 size;
  294. if (boundsType != OBS_BOUNDS_NONE) {
  295. obs_sceneitem_get_bounds(item, &size);
  296. } else {
  297. obs_source_t *source = obs_sceneitem_get_source(item);
  298. obs_sceneitem_crop crop;
  299. vec2 scale;
  300. obs_sceneitem_get_scale(item, &scale);
  301. obs_sceneitem_get_crop(item, &crop);
  302. size.x = float(obs_source_get_width(source) - crop.left -
  303. crop.right) *
  304. scale.x;
  305. size.y = float(obs_source_get_height(source) - crop.top -
  306. crop.bottom) *
  307. scale.y;
  308. }
  309. return size;
  310. }
  311. void OBSBasicPreview::GetStretchHandleData(const vec2 &pos)
  312. {
  313. OBSBasic *main = reinterpret_cast<OBSBasic *>(App()->GetMainWindow());
  314. OBSScene scene = main->GetCurrentScene();
  315. if (!scene)
  316. return;
  317. #ifdef SUPPORTS_FRACTIONAL_SCALING
  318. float scale = main->previewScale / main->devicePixelRatioF();
  319. #else
  320. float scale = main->previewScale / main->devicePixelRatio();
  321. #endif
  322. vec2 scaled_pos = pos;
  323. vec2_divf(&scaled_pos, &scaled_pos, scale);
  324. HandleFindData data(scaled_pos, scale);
  325. obs_scene_enum_items(scene, FindHandleAtPos, &data);
  326. stretchItem = std::move(data.item);
  327. stretchHandle = data.handle;
  328. if (stretchHandle != ItemHandle::None) {
  329. matrix4 boxTransform;
  330. vec3 itemUL;
  331. float itemRot;
  332. stretchItemSize = GetItemSize(stretchItem);
  333. obs_sceneitem_get_box_transform(stretchItem, &boxTransform);
  334. itemRot = obs_sceneitem_get_rot(stretchItem);
  335. vec3_from_vec4(&itemUL, &boxTransform.t);
  336. /* build the item space conversion matrices */
  337. matrix4_identity(&itemToScreen);
  338. matrix4_rotate_aa4f(&itemToScreen, &itemToScreen, 0.0f, 0.0f,
  339. 1.0f, RAD(itemRot));
  340. matrix4_translate3f(&itemToScreen, &itemToScreen, itemUL.x,
  341. itemUL.y, 0.0f);
  342. matrix4_identity(&screenToItem);
  343. matrix4_translate3f(&screenToItem, &screenToItem, -itemUL.x,
  344. -itemUL.y, 0.0f);
  345. matrix4_rotate_aa4f(&screenToItem, &screenToItem, 0.0f, 0.0f,
  346. 1.0f, RAD(-itemRot));
  347. obs_sceneitem_get_crop(stretchItem, &startCrop);
  348. obs_sceneitem_get_pos(stretchItem, &startItemPos);
  349. obs_source_t *source = obs_sceneitem_get_source(stretchItem);
  350. cropSize.x = float(obs_source_get_width(source) -
  351. startCrop.left - startCrop.right);
  352. cropSize.y = float(obs_source_get_height(source) -
  353. startCrop.top - startCrop.bottom);
  354. stretchGroup = obs_sceneitem_get_group(scene, stretchItem);
  355. if (stretchGroup) {
  356. obs_sceneitem_get_draw_transform(stretchGroup,
  357. &invGroupTransform);
  358. matrix4_inv(&invGroupTransform, &invGroupTransform);
  359. obs_sceneitem_defer_group_resize_begin(stretchGroup);
  360. }
  361. }
  362. }
  363. void OBSBasicPreview::keyPressEvent(QKeyEvent *event)
  364. {
  365. if (!IsFixedScaling() || event->isAutoRepeat()) {
  366. OBSQTDisplay::keyPressEvent(event);
  367. return;
  368. }
  369. switch (event->key()) {
  370. case Qt::Key_Space:
  371. setCursor(Qt::OpenHandCursor);
  372. scrollMode = true;
  373. break;
  374. }
  375. OBSQTDisplay::keyPressEvent(event);
  376. }
  377. void OBSBasicPreview::keyReleaseEvent(QKeyEvent *event)
  378. {
  379. if (event->isAutoRepeat()) {
  380. OBSQTDisplay::keyReleaseEvent(event);
  381. return;
  382. }
  383. switch (event->key()) {
  384. case Qt::Key_Space:
  385. scrollMode = false;
  386. setCursor(Qt::ArrowCursor);
  387. break;
  388. }
  389. OBSQTDisplay::keyReleaseEvent(event);
  390. }
  391. void OBSBasicPreview::wheelEvent(QWheelEvent *event)
  392. {
  393. if (scrollMode && IsFixedScaling() &&
  394. event->orientation() == Qt::Vertical) {
  395. if (event->delta() > 0)
  396. SetScalingLevel(scalingLevel + 1);
  397. else if (event->delta() < 0)
  398. SetScalingLevel(scalingLevel - 1);
  399. emit DisplayResized();
  400. }
  401. OBSQTDisplay::wheelEvent(event);
  402. }
  403. void OBSBasicPreview::mousePressEvent(QMouseEvent *event)
  404. {
  405. if (scrollMode && IsFixedScaling() &&
  406. event->button() == Qt::LeftButton) {
  407. setCursor(Qt::ClosedHandCursor);
  408. scrollingFrom.x = event->x();
  409. scrollingFrom.y = event->y();
  410. return;
  411. }
  412. if (event->button() == Qt::RightButton) {
  413. scrollMode = false;
  414. setCursor(Qt::ArrowCursor);
  415. }
  416. if (locked) {
  417. OBSQTDisplay::mousePressEvent(event);
  418. return;
  419. }
  420. OBSBasic *main = reinterpret_cast<OBSBasic *>(App()->GetMainWindow());
  421. #ifdef SUPPORTS_FRACTIONAL_SCALING
  422. float pixelRatio = main->devicePixelRatioF();
  423. #else
  424. float pixelRatio = main->devicePixelRatio();
  425. #endif
  426. float x = float(event->x()) - main->previewX / pixelRatio;
  427. float y = float(event->y()) - main->previewY / pixelRatio;
  428. Qt::KeyboardModifiers modifiers = QGuiApplication::keyboardModifiers();
  429. bool altDown = (modifiers & Qt::AltModifier);
  430. OBSQTDisplay::mousePressEvent(event);
  431. if (event->button() != Qt::LeftButton &&
  432. event->button() != Qt::RightButton)
  433. return;
  434. if (event->button() == Qt::LeftButton)
  435. mouseDown = true;
  436. if (altDown)
  437. cropping = true;
  438. vec2_set(&startPos, x, y);
  439. GetStretchHandleData(startPos);
  440. vec2_divf(&startPos, &startPos, main->previewScale / pixelRatio);
  441. startPos.x = std::round(startPos.x);
  442. startPos.y = std::round(startPos.y);
  443. mouseOverItems = SelectedAtPos(startPos);
  444. vec2_zero(&lastMoveOffset);
  445. }
  446. static bool select_one(obs_scene_t *scene, obs_sceneitem_t *item, void *param)
  447. {
  448. obs_sceneitem_t *selectedItem =
  449. reinterpret_cast<obs_sceneitem_t *>(param);
  450. if (obs_sceneitem_is_group(item))
  451. obs_sceneitem_group_enum_items(item, select_one, param);
  452. obs_sceneitem_select(item, (selectedItem == item));
  453. UNUSED_PARAMETER(scene);
  454. return true;
  455. }
  456. void OBSBasicPreview::DoSelect(const vec2 &pos)
  457. {
  458. OBSBasic *main = reinterpret_cast<OBSBasic *>(App()->GetMainWindow());
  459. OBSScene scene = main->GetCurrentScene();
  460. OBSSceneItem item = GetItemAtPos(pos, true);
  461. obs_scene_enum_items(scene, select_one, (obs_sceneitem_t *)item);
  462. }
  463. void OBSBasicPreview::DoCtrlSelect(const vec2 &pos)
  464. {
  465. OBSSceneItem item = GetItemAtPos(pos, false);
  466. if (!item)
  467. return;
  468. bool selected = obs_sceneitem_selected(item);
  469. obs_sceneitem_select(item, !selected);
  470. }
  471. void OBSBasicPreview::ProcessClick(const vec2 &pos)
  472. {
  473. Qt::KeyboardModifiers modifiers = QGuiApplication::keyboardModifiers();
  474. if (modifiers & Qt::ControlModifier)
  475. DoCtrlSelect(pos);
  476. else
  477. DoSelect(pos);
  478. }
  479. void OBSBasicPreview::mouseReleaseEvent(QMouseEvent *event)
  480. {
  481. if (scrollMode)
  482. setCursor(Qt::OpenHandCursor);
  483. if (locked) {
  484. OBSQTDisplay::mouseReleaseEvent(event);
  485. return;
  486. }
  487. if (mouseDown) {
  488. vec2 pos = GetMouseEventPos(event);
  489. if (!mouseMoved)
  490. ProcessClick(pos);
  491. if (stretchGroup) {
  492. obs_sceneitem_defer_group_resize_end(stretchGroup);
  493. }
  494. stretchItem = nullptr;
  495. stretchGroup = nullptr;
  496. mouseDown = false;
  497. mouseMoved = false;
  498. cropping = false;
  499. OBSSceneItem item = GetItemAtPos(pos, true);
  500. hoveredPreviewItem = item;
  501. }
  502. }
  503. struct SelectedItemBounds {
  504. bool first = true;
  505. vec3 tl, br;
  506. };
  507. static bool AddItemBounds(obs_scene_t *scene, obs_sceneitem_t *item,
  508. void *param)
  509. {
  510. SelectedItemBounds *data =
  511. reinterpret_cast<SelectedItemBounds *>(param);
  512. vec3 t[4];
  513. auto add_bounds = [data, &t]() {
  514. for (const vec3 &v : t) {
  515. if (data->first) {
  516. vec3_copy(&data->tl, &v);
  517. vec3_copy(&data->br, &v);
  518. data->first = false;
  519. } else {
  520. vec3_min(&data->tl, &data->tl, &v);
  521. vec3_max(&data->br, &data->br, &v);
  522. }
  523. }
  524. };
  525. if (obs_sceneitem_is_group(item)) {
  526. SelectedItemBounds sib;
  527. obs_sceneitem_group_enum_items(item, AddItemBounds, &sib);
  528. if (!sib.first) {
  529. matrix4 xform;
  530. obs_sceneitem_get_draw_transform(item, &xform);
  531. vec3_set(&t[0], sib.tl.x, sib.tl.y, 0.0f);
  532. vec3_set(&t[1], sib.tl.x, sib.br.y, 0.0f);
  533. vec3_set(&t[2], sib.br.x, sib.tl.y, 0.0f);
  534. vec3_set(&t[3], sib.br.x, sib.br.y, 0.0f);
  535. vec3_transform(&t[0], &t[0], &xform);
  536. vec3_transform(&t[1], &t[1], &xform);
  537. vec3_transform(&t[2], &t[2], &xform);
  538. vec3_transform(&t[3], &t[3], &xform);
  539. add_bounds();
  540. }
  541. }
  542. if (!obs_sceneitem_selected(item))
  543. return true;
  544. matrix4 boxTransform;
  545. obs_sceneitem_get_box_transform(item, &boxTransform);
  546. t[0] = GetTransformedPos(0.0f, 0.0f, boxTransform);
  547. t[1] = GetTransformedPos(1.0f, 0.0f, boxTransform);
  548. t[2] = GetTransformedPos(0.0f, 1.0f, boxTransform);
  549. t[3] = GetTransformedPos(1.0f, 1.0f, boxTransform);
  550. add_bounds();
  551. UNUSED_PARAMETER(scene);
  552. return true;
  553. }
  554. struct OffsetData {
  555. float clampDist;
  556. vec3 tl, br, offset;
  557. };
  558. static bool GetSourceSnapOffset(obs_scene_t *scene, obs_sceneitem_t *item,
  559. void *param)
  560. {
  561. OffsetData *data = reinterpret_cast<OffsetData *>(param);
  562. if (obs_sceneitem_selected(item))
  563. return true;
  564. matrix4 boxTransform;
  565. obs_sceneitem_get_box_transform(item, &boxTransform);
  566. vec3 t[4] = {GetTransformedPos(0.0f, 0.0f, boxTransform),
  567. GetTransformedPos(1.0f, 0.0f, boxTransform),
  568. GetTransformedPos(0.0f, 1.0f, boxTransform),
  569. GetTransformedPos(1.0f, 1.0f, boxTransform)};
  570. bool first = true;
  571. vec3 tl, br;
  572. vec3_zero(&tl);
  573. vec3_zero(&br);
  574. for (const vec3 &v : t) {
  575. if (first) {
  576. vec3_copy(&tl, &v);
  577. vec3_copy(&br, &v);
  578. first = false;
  579. } else {
  580. vec3_min(&tl, &tl, &v);
  581. vec3_max(&br, &br, &v);
  582. }
  583. }
  584. // Snap to other source edges
  585. #define EDGE_SNAP(l, r, x, y) \
  586. do { \
  587. double dist = fabsf(l.x - data->r.x); \
  588. if (dist < data->clampDist && \
  589. fabsf(data->offset.x) < EPSILON && data->tl.y < br.y && \
  590. data->br.y > tl.y && \
  591. (fabsf(data->offset.x) > dist || \
  592. data->offset.x < EPSILON)) \
  593. data->offset.x = l.x - data->r.x; \
  594. } while (false)
  595. EDGE_SNAP(tl, br, x, y);
  596. EDGE_SNAP(tl, br, y, x);
  597. EDGE_SNAP(br, tl, x, y);
  598. EDGE_SNAP(br, tl, y, x);
  599. #undef EDGE_SNAP
  600. UNUSED_PARAMETER(scene);
  601. return true;
  602. }
  603. void OBSBasicPreview::SnapItemMovement(vec2 &offset)
  604. {
  605. OBSBasic *main = reinterpret_cast<OBSBasic *>(App()->GetMainWindow());
  606. OBSScene scene = main->GetCurrentScene();
  607. SelectedItemBounds data;
  608. obs_scene_enum_items(scene, AddItemBounds, &data);
  609. data.tl.x += offset.x;
  610. data.tl.y += offset.y;
  611. data.br.x += offset.x;
  612. data.br.y += offset.y;
  613. vec3 snapOffset = GetSnapOffset(data.tl, data.br);
  614. const bool snap = config_get_bool(GetGlobalConfig(), "BasicWindow",
  615. "SnappingEnabled");
  616. const bool sourcesSnap = config_get_bool(
  617. GetGlobalConfig(), "BasicWindow", "SourceSnapping");
  618. if (snap == false)
  619. return;
  620. if (sourcesSnap == false) {
  621. offset.x += snapOffset.x;
  622. offset.y += snapOffset.y;
  623. return;
  624. }
  625. const float clampDist = config_get_double(GetGlobalConfig(),
  626. "BasicWindow",
  627. "SnapDistance") /
  628. main->previewScale;
  629. OffsetData offsetData;
  630. offsetData.clampDist = clampDist;
  631. offsetData.tl = data.tl;
  632. offsetData.br = data.br;
  633. vec3_copy(&offsetData.offset, &snapOffset);
  634. obs_scene_enum_items(scene, GetSourceSnapOffset, &offsetData);
  635. if (fabsf(offsetData.offset.x) > EPSILON ||
  636. fabsf(offsetData.offset.y) > EPSILON) {
  637. offset.x += offsetData.offset.x;
  638. offset.y += offsetData.offset.y;
  639. } else {
  640. offset.x += snapOffset.x;
  641. offset.y += snapOffset.y;
  642. }
  643. }
  644. static bool move_items(obs_scene_t *scene, obs_sceneitem_t *item, void *param)
  645. {
  646. if (obs_sceneitem_locked(item))
  647. return true;
  648. bool selected = obs_sceneitem_selected(item);
  649. vec2 *offset = reinterpret_cast<vec2 *>(param);
  650. if (obs_sceneitem_is_group(item) && !selected) {
  651. matrix4 transform;
  652. vec3 new_offset;
  653. vec3_set(&new_offset, offset->x, offset->y, 0.0f);
  654. obs_sceneitem_get_draw_transform(item, &transform);
  655. vec4_set(&transform.t, 0.0f, 0.0f, 0.0f, 1.0f);
  656. matrix4_inv(&transform, &transform);
  657. vec3_transform(&new_offset, &new_offset, &transform);
  658. obs_sceneitem_group_enum_items(item, move_items, &new_offset);
  659. }
  660. if (selected) {
  661. vec2 pos;
  662. obs_sceneitem_get_pos(item, &pos);
  663. vec2_add(&pos, &pos, offset);
  664. obs_sceneitem_set_pos(item, &pos);
  665. }
  666. UNUSED_PARAMETER(scene);
  667. return true;
  668. }
  669. void OBSBasicPreview::MoveItems(const vec2 &pos)
  670. {
  671. Qt::KeyboardModifiers modifiers = QGuiApplication::keyboardModifiers();
  672. OBSBasic *main = reinterpret_cast<OBSBasic *>(App()->GetMainWindow());
  673. OBSScene scene = main->GetCurrentScene();
  674. vec2 offset, moveOffset;
  675. vec2_sub(&offset, &pos, &startPos);
  676. vec2_sub(&moveOffset, &offset, &lastMoveOffset);
  677. if (!(modifiers & Qt::ControlModifier))
  678. SnapItemMovement(moveOffset);
  679. vec2_add(&lastMoveOffset, &lastMoveOffset, &moveOffset);
  680. obs_scene_enum_items(scene, move_items, &moveOffset);
  681. }
  682. vec3 OBSBasicPreview::CalculateStretchPos(const vec3 &tl, const vec3 &br)
  683. {
  684. uint32_t alignment = obs_sceneitem_get_alignment(stretchItem);
  685. vec3 pos;
  686. vec3_zero(&pos);
  687. if (alignment & OBS_ALIGN_LEFT)
  688. pos.x = tl.x;
  689. else if (alignment & OBS_ALIGN_RIGHT)
  690. pos.x = br.x;
  691. else
  692. pos.x = (br.x - tl.x) * 0.5f + tl.x;
  693. if (alignment & OBS_ALIGN_TOP)
  694. pos.y = tl.y;
  695. else if (alignment & OBS_ALIGN_BOTTOM)
  696. pos.y = br.y;
  697. else
  698. pos.y = (br.y - tl.y) * 0.5f + tl.y;
  699. return pos;
  700. }
  701. void OBSBasicPreview::ClampAspect(vec3 &tl, vec3 &br, vec2 &size,
  702. const vec2 &baseSize)
  703. {
  704. float baseAspect = baseSize.x / baseSize.y;
  705. float aspect = size.x / size.y;
  706. uint32_t stretchFlags = (uint32_t)stretchHandle;
  707. if (stretchHandle == ItemHandle::TopLeft ||
  708. stretchHandle == ItemHandle::TopRight ||
  709. stretchHandle == ItemHandle::BottomLeft ||
  710. stretchHandle == ItemHandle::BottomRight) {
  711. if (aspect < baseAspect) {
  712. if ((size.y >= 0.0f && size.x >= 0.0f) ||
  713. (size.y <= 0.0f && size.x <= 0.0f))
  714. size.x = size.y * baseAspect;
  715. else
  716. size.x = size.y * baseAspect * -1.0f;
  717. } else {
  718. if ((size.y >= 0.0f && size.x >= 0.0f) ||
  719. (size.y <= 0.0f && size.x <= 0.0f))
  720. size.y = size.x / baseAspect;
  721. else
  722. size.y = size.x / baseAspect * -1.0f;
  723. }
  724. } else if (stretchHandle == ItemHandle::TopCenter ||
  725. stretchHandle == ItemHandle::BottomCenter) {
  726. if ((size.y >= 0.0f && size.x >= 0.0f) ||
  727. (size.y <= 0.0f && size.x <= 0.0f))
  728. size.x = size.y * baseAspect;
  729. else
  730. size.x = size.y * baseAspect * -1.0f;
  731. } else if (stretchHandle == ItemHandle::CenterLeft ||
  732. stretchHandle == ItemHandle::CenterRight) {
  733. if ((size.y >= 0.0f && size.x >= 0.0f) ||
  734. (size.y <= 0.0f && size.x <= 0.0f))
  735. size.y = size.x / baseAspect;
  736. else
  737. size.y = size.x / baseAspect * -1.0f;
  738. }
  739. size.x = std::round(size.x);
  740. size.y = std::round(size.y);
  741. if (stretchFlags & ITEM_LEFT)
  742. tl.x = br.x - size.x;
  743. else if (stretchFlags & ITEM_RIGHT)
  744. br.x = tl.x + size.x;
  745. if (stretchFlags & ITEM_TOP)
  746. tl.y = br.y - size.y;
  747. else if (stretchFlags & ITEM_BOTTOM)
  748. br.y = tl.y + size.y;
  749. }
  750. void OBSBasicPreview::SnapStretchingToScreen(vec3 &tl, vec3 &br)
  751. {
  752. uint32_t stretchFlags = (uint32_t)stretchHandle;
  753. vec3 newTL = GetTransformedPos(tl.x, tl.y, itemToScreen);
  754. vec3 newTR = GetTransformedPos(br.x, tl.y, itemToScreen);
  755. vec3 newBL = GetTransformedPos(tl.x, br.y, itemToScreen);
  756. vec3 newBR = GetTransformedPos(br.x, br.y, itemToScreen);
  757. vec3 boundingTL;
  758. vec3 boundingBR;
  759. vec3_copy(&boundingTL, &newTL);
  760. vec3_min(&boundingTL, &boundingTL, &newTR);
  761. vec3_min(&boundingTL, &boundingTL, &newBL);
  762. vec3_min(&boundingTL, &boundingTL, &newBR);
  763. vec3_copy(&boundingBR, &newTL);
  764. vec3_max(&boundingBR, &boundingBR, &newTR);
  765. vec3_max(&boundingBR, &boundingBR, &newBL);
  766. vec3_max(&boundingBR, &boundingBR, &newBR);
  767. vec3 offset = GetSnapOffset(boundingTL, boundingBR);
  768. vec3_add(&offset, &offset, &newTL);
  769. vec3_transform(&offset, &offset, &screenToItem);
  770. vec3_sub(&offset, &offset, &tl);
  771. if (stretchFlags & ITEM_LEFT)
  772. tl.x += offset.x;
  773. else if (stretchFlags & ITEM_RIGHT)
  774. br.x += offset.x;
  775. if (stretchFlags & ITEM_TOP)
  776. tl.y += offset.y;
  777. else if (stretchFlags & ITEM_BOTTOM)
  778. br.y += offset.y;
  779. }
  780. static float maxfunc(float x, float y)
  781. {
  782. return x > y ? x : y;
  783. }
  784. static float minfunc(float x, float y)
  785. {
  786. return x < y ? x : y;
  787. }
  788. void OBSBasicPreview::CropItem(const vec2 &pos)
  789. {
  790. obs_bounds_type boundsType = obs_sceneitem_get_bounds_type(stretchItem);
  791. uint32_t stretchFlags = (uint32_t)stretchHandle;
  792. uint32_t align = obs_sceneitem_get_alignment(stretchItem);
  793. vec3 tl, br, pos3;
  794. vec3_zero(&tl);
  795. vec3_set(&br, stretchItemSize.x, stretchItemSize.y, 0.0f);
  796. vec3_set(&pos3, pos.x, pos.y, 0.0f);
  797. vec3_transform(&pos3, &pos3, &screenToItem);
  798. obs_sceneitem_crop crop = startCrop;
  799. vec2 scale;
  800. obs_sceneitem_get_scale(stretchItem, &scale);
  801. vec2 max_tl;
  802. vec2 max_br;
  803. vec2_set(&max_tl, float(-crop.left) * scale.x,
  804. float(-crop.top) * scale.y);
  805. vec2_set(&max_br, stretchItemSize.x + crop.right * scale.x,
  806. stretchItemSize.y + crop.bottom * scale.y);
  807. typedef std::function<float(float, float)> minmax_func_t;
  808. minmax_func_t min_x = scale.x < 0.0f ? maxfunc : minfunc;
  809. minmax_func_t min_y = scale.y < 0.0f ? maxfunc : minfunc;
  810. minmax_func_t max_x = scale.x < 0.0f ? minfunc : maxfunc;
  811. minmax_func_t max_y = scale.y < 0.0f ? minfunc : maxfunc;
  812. pos3.x = min_x(pos3.x, max_br.x);
  813. pos3.x = max_x(pos3.x, max_tl.x);
  814. pos3.y = min_y(pos3.y, max_br.y);
  815. pos3.y = max_y(pos3.y, max_tl.y);
  816. if (stretchFlags & ITEM_LEFT) {
  817. float maxX = stretchItemSize.x - (2.0 * scale.x);
  818. pos3.x = tl.x = min_x(pos3.x, maxX);
  819. } else if (stretchFlags & ITEM_RIGHT) {
  820. float minX = (2.0 * scale.x);
  821. pos3.x = br.x = max_x(pos3.x, minX);
  822. }
  823. if (stretchFlags & ITEM_TOP) {
  824. float maxY = stretchItemSize.y - (2.0 * scale.y);
  825. pos3.y = tl.y = min_y(pos3.y, maxY);
  826. } else if (stretchFlags & ITEM_BOTTOM) {
  827. float minY = (2.0 * scale.y);
  828. pos3.y = br.y = max_y(pos3.y, minY);
  829. }
  830. #define ALIGN_X (ITEM_LEFT | ITEM_RIGHT)
  831. #define ALIGN_Y (ITEM_TOP | ITEM_BOTTOM)
  832. vec3 newPos;
  833. vec3_zero(&newPos);
  834. uint32_t align_x = (align & ALIGN_X);
  835. uint32_t align_y = (align & ALIGN_Y);
  836. if (align_x == (stretchFlags & ALIGN_X) && align_x != 0)
  837. newPos.x = pos3.x;
  838. else if (align & ITEM_RIGHT)
  839. newPos.x = stretchItemSize.x;
  840. else if (!(align & ITEM_LEFT))
  841. newPos.x = stretchItemSize.x * 0.5f;
  842. if (align_y == (stretchFlags & ALIGN_Y) && align_y != 0)
  843. newPos.y = pos3.y;
  844. else if (align & ITEM_BOTTOM)
  845. newPos.y = stretchItemSize.y;
  846. else if (!(align & ITEM_TOP))
  847. newPos.y = stretchItemSize.y * 0.5f;
  848. #undef ALIGN_X
  849. #undef ALIGN_Y
  850. crop = startCrop;
  851. if (stretchFlags & ITEM_LEFT)
  852. crop.left += int(std::round(tl.x / scale.x));
  853. else if (stretchFlags & ITEM_RIGHT)
  854. crop.right +=
  855. int(std::round((stretchItemSize.x - br.x) / scale.x));
  856. if (stretchFlags & ITEM_TOP)
  857. crop.top += int(std::round(tl.y / scale.y));
  858. else if (stretchFlags & ITEM_BOTTOM)
  859. crop.bottom +=
  860. int(std::round((stretchItemSize.y - br.y) / scale.y));
  861. vec3_transform(&newPos, &newPos, &itemToScreen);
  862. newPos.x = std::round(newPos.x);
  863. newPos.y = std::round(newPos.y);
  864. #if 0
  865. vec3 curPos;
  866. vec3_zero(&curPos);
  867. obs_sceneitem_get_pos(stretchItem, (vec2*)&curPos);
  868. blog(LOG_DEBUG, "curPos {%d, %d} - newPos {%d, %d}",
  869. int(curPos.x), int(curPos.y),
  870. int(newPos.x), int(newPos.y));
  871. blog(LOG_DEBUG, "crop {%d, %d, %d, %d}",
  872. crop.left, crop.top,
  873. crop.right, crop.bottom);
  874. #endif
  875. obs_sceneitem_defer_update_begin(stretchItem);
  876. obs_sceneitem_set_crop(stretchItem, &crop);
  877. if (boundsType == OBS_BOUNDS_NONE)
  878. obs_sceneitem_set_pos(stretchItem, (vec2 *)&newPos);
  879. obs_sceneitem_defer_update_end(stretchItem);
  880. }
  881. void OBSBasicPreview::StretchItem(const vec2 &pos)
  882. {
  883. Qt::KeyboardModifiers modifiers = QGuiApplication::keyboardModifiers();
  884. obs_bounds_type boundsType = obs_sceneitem_get_bounds_type(stretchItem);
  885. uint32_t stretchFlags = (uint32_t)stretchHandle;
  886. bool shiftDown = (modifiers & Qt::ShiftModifier);
  887. vec3 tl, br, pos3;
  888. vec3_zero(&tl);
  889. vec3_set(&br, stretchItemSize.x, stretchItemSize.y, 0.0f);
  890. vec3_set(&pos3, pos.x, pos.y, 0.0f);
  891. vec3_transform(&pos3, &pos3, &screenToItem);
  892. if (stretchFlags & ITEM_LEFT)
  893. tl.x = pos3.x;
  894. else if (stretchFlags & ITEM_RIGHT)
  895. br.x = pos3.x;
  896. if (stretchFlags & ITEM_TOP)
  897. tl.y = pos3.y;
  898. else if (stretchFlags & ITEM_BOTTOM)
  899. br.y = pos3.y;
  900. if (!(modifiers & Qt::ControlModifier))
  901. SnapStretchingToScreen(tl, br);
  902. obs_source_t *source = obs_sceneitem_get_source(stretchItem);
  903. vec2 baseSize;
  904. vec2_set(&baseSize, float(obs_source_get_width(source)),
  905. float(obs_source_get_height(source)));
  906. vec2 size;
  907. vec2_set(&size, br.x - tl.x, br.y - tl.y);
  908. if (boundsType != OBS_BOUNDS_NONE) {
  909. if (shiftDown)
  910. ClampAspect(tl, br, size, baseSize);
  911. if (tl.x > br.x)
  912. std::swap(tl.x, br.x);
  913. if (tl.y > br.y)
  914. std::swap(tl.y, br.y);
  915. vec2_abs(&size, &size);
  916. obs_sceneitem_set_bounds(stretchItem, &size);
  917. } else {
  918. obs_sceneitem_crop crop;
  919. obs_sceneitem_get_crop(stretchItem, &crop);
  920. baseSize.x -= float(crop.left + crop.right);
  921. baseSize.y -= float(crop.top + crop.bottom);
  922. if (!shiftDown)
  923. ClampAspect(tl, br, size, baseSize);
  924. vec2_div(&size, &size, &baseSize);
  925. obs_sceneitem_set_scale(stretchItem, &size);
  926. }
  927. pos3 = CalculateStretchPos(tl, br);
  928. vec3_transform(&pos3, &pos3, &itemToScreen);
  929. vec2 newPos;
  930. vec2_set(&newPos, std::round(pos3.x), std::round(pos3.y));
  931. obs_sceneitem_set_pos(stretchItem, &newPos);
  932. }
  933. void OBSBasicPreview::mouseMoveEvent(QMouseEvent *event)
  934. {
  935. if (scrollMode && event->buttons() == Qt::LeftButton) {
  936. scrollingOffset.x += event->x() - scrollingFrom.x;
  937. scrollingOffset.y += event->y() - scrollingFrom.y;
  938. scrollingFrom.x = event->x();
  939. scrollingFrom.y = event->y();
  940. emit DisplayResized();
  941. return;
  942. }
  943. if (locked)
  944. return;
  945. if (mouseDown) {
  946. hoveredPreviewItem = nullptr;
  947. vec2 pos = GetMouseEventPos(event);
  948. if (!mouseMoved && !mouseOverItems &&
  949. stretchHandle == ItemHandle::None) {
  950. ProcessClick(startPos);
  951. mouseOverItems = SelectedAtPos(startPos);
  952. }
  953. pos.x = std::round(pos.x);
  954. pos.y = std::round(pos.y);
  955. if (stretchHandle != ItemHandle::None) {
  956. OBSBasic *main = reinterpret_cast<OBSBasic *>(
  957. App()->GetMainWindow());
  958. OBSScene scene = main->GetCurrentScene();
  959. obs_sceneitem_t *group =
  960. obs_sceneitem_get_group(scene, stretchItem);
  961. if (group) {
  962. vec3 group_pos;
  963. vec3_set(&group_pos, pos.x, pos.y, 0.0f);
  964. vec3_transform(&group_pos, &group_pos,
  965. &invGroupTransform);
  966. pos.x = group_pos.x;
  967. pos.y = group_pos.y;
  968. }
  969. if (cropping)
  970. CropItem(pos);
  971. else
  972. StretchItem(pos);
  973. } else if (mouseOverItems) {
  974. MoveItems(pos);
  975. }
  976. mouseMoved = true;
  977. } else {
  978. vec2 pos = GetMouseEventPos(event);
  979. OBSSceneItem item = GetItemAtPos(pos, true);
  980. hoveredPreviewItem = item;
  981. }
  982. }
  983. void OBSBasicPreview::leaveEvent(QEvent *event)
  984. {
  985. hoveredPreviewItem = nullptr;
  986. UNUSED_PARAMETER(event);
  987. }
  988. static void DrawSquareAtPos(float x, float y)
  989. {
  990. struct vec3 pos;
  991. vec3_set(&pos, x, y, 0.0f);
  992. struct matrix4 matrix;
  993. gs_matrix_get(&matrix);
  994. vec3_transform(&pos, &pos, &matrix);
  995. gs_matrix_push();
  996. gs_matrix_identity();
  997. gs_matrix_translate(&pos);
  998. gs_matrix_translate3f(-HANDLE_RADIUS, -HANDLE_RADIUS, 0.0f);
  999. gs_matrix_scale3f(HANDLE_RADIUS * 2, HANDLE_RADIUS * 2, 1.0f);
  1000. gs_draw(GS_TRISTRIP, 0, 0);
  1001. gs_matrix_pop();
  1002. }
  1003. static void DrawLine(float x1, float y1, float x2, float y2, float thickness,
  1004. vec2 scale)
  1005. {
  1006. float ySide = (y1 == y2) ? (y1 < 0.5f ? 1.0f : -1.0f) : 0.0f;
  1007. float xSide = (x1 == x2) ? (x1 < 0.5f ? 1.0f : -1.0f) : 0.0f;
  1008. gs_render_start(true);
  1009. gs_vertex2f(x1, y1);
  1010. gs_vertex2f(x1 + (xSide * (thickness / scale.x)),
  1011. y1 + (ySide * (thickness / scale.y)));
  1012. gs_vertex2f(x2 + (xSide * (thickness / scale.x)),
  1013. y2 + (ySide * (thickness / scale.y)));
  1014. gs_vertex2f(x2, y2);
  1015. gs_vertex2f(x1, y1);
  1016. gs_vertbuffer_t *line = gs_render_save();
  1017. gs_load_vertexbuffer(line);
  1018. gs_draw(GS_TRISTRIP, 0, 0);
  1019. gs_vertexbuffer_destroy(line);
  1020. }
  1021. static void DrawRect(float thickness, vec2 scale)
  1022. {
  1023. gs_render_start(true);
  1024. gs_vertex2f(0.0f, 0.0f);
  1025. gs_vertex2f(0.0f + (thickness / scale.x), 0.0f);
  1026. gs_vertex2f(0.0f + (thickness / scale.x), 1.0f);
  1027. gs_vertex2f(0.0f, 1.0f);
  1028. gs_vertex2f(0.0f, 0.0f);
  1029. gs_vertex2f(0.0f, 1.0f);
  1030. gs_vertex2f(0.0f, 1.0f - (thickness / scale.y));
  1031. gs_vertex2f(1.0f, 1.0f - (thickness / scale.y));
  1032. gs_vertex2f(1.0f, 1.0f);
  1033. gs_vertex2f(0.0f, 1.0f);
  1034. gs_vertex2f(1.0f, 1.0f);
  1035. gs_vertex2f(1.0f - (thickness / scale.x), 1.0f);
  1036. gs_vertex2f(1.0f - (thickness / scale.x), 0.0f);
  1037. gs_vertex2f(1.0f, 0.0f);
  1038. gs_vertex2f(1.0f, 1.0f);
  1039. gs_vertex2f(1.0f, 0.0f);
  1040. gs_vertex2f(1.0f, 0.0f + (thickness / scale.y));
  1041. gs_vertex2f(0.0f, 0.0f + (thickness / scale.y));
  1042. gs_vertex2f(0.0f, 0.0f);
  1043. gs_vertex2f(1.0f, 0.0f);
  1044. gs_vertbuffer_t *rect = gs_render_save();
  1045. gs_load_vertexbuffer(rect);
  1046. gs_draw(GS_TRISTRIP, 0, 0);
  1047. gs_vertexbuffer_destroy(rect);
  1048. }
  1049. static inline bool crop_enabled(const obs_sceneitem_crop *crop)
  1050. {
  1051. return crop->left > 0 || crop->top > 0 || crop->right > 0 ||
  1052. crop->bottom > 0;
  1053. }
  1054. bool OBSBasicPreview::DrawSelectedOverflow(obs_scene_t *scene,
  1055. obs_sceneitem_t *item, void *param)
  1056. {
  1057. if (obs_sceneitem_locked(item))
  1058. return true;
  1059. if (!SceneItemHasVideo(item))
  1060. return true;
  1061. bool select = config_get_bool(GetGlobalConfig(), "BasicWindow",
  1062. "OverflowSelectionHidden");
  1063. if (!select && !obs_sceneitem_visible(item))
  1064. return true;
  1065. if (obs_sceneitem_is_group(item)) {
  1066. matrix4 mat;
  1067. obs_sceneitem_get_draw_transform(item, &mat);
  1068. gs_matrix_push();
  1069. gs_matrix_mul(&mat);
  1070. obs_sceneitem_group_enum_items(item, DrawSelectedOverflow,
  1071. param);
  1072. gs_matrix_pop();
  1073. }
  1074. bool always = config_get_bool(GetGlobalConfig(), "BasicWindow",
  1075. "OverflowAlwaysVisible");
  1076. if (!always && !obs_sceneitem_selected(item))
  1077. return true;
  1078. OBSBasicPreview *prev = reinterpret_cast<OBSBasicPreview *>(param);
  1079. matrix4 boxTransform;
  1080. matrix4 invBoxTransform;
  1081. obs_sceneitem_get_box_transform(item, &boxTransform);
  1082. matrix4_inv(&invBoxTransform, &boxTransform);
  1083. vec3 bounds[] = {
  1084. {{{0.f, 0.f, 0.f}}},
  1085. {{{1.f, 0.f, 0.f}}},
  1086. {{{0.f, 1.f, 0.f}}},
  1087. {{{1.f, 1.f, 0.f}}},
  1088. };
  1089. bool visible = std::all_of(
  1090. std::begin(bounds), std::end(bounds), [&](const vec3 &b) {
  1091. vec3 pos;
  1092. vec3_transform(&pos, &b, &boxTransform);
  1093. vec3_transform(&pos, &pos, &invBoxTransform);
  1094. return CloseFloat(pos.x, b.x) && CloseFloat(pos.y, b.y);
  1095. });
  1096. if (!visible)
  1097. return true;
  1098. GS_DEBUG_MARKER_BEGIN(GS_DEBUG_COLOR_DEFAULT, "DrawSelectedOverflow");
  1099. obs_transform_info info;
  1100. obs_sceneitem_get_info(item, &info);
  1101. gs_effect_t *solid = obs_get_base_effect(OBS_EFFECT_REPEAT);
  1102. gs_eparam_t *image = gs_effect_get_param_by_name(solid, "image");
  1103. gs_eparam_t *scale = gs_effect_get_param_by_name(solid, "scale");
  1104. vec2 s;
  1105. vec2_set(&s, boxTransform.x.x / 96, boxTransform.y.y / 96);
  1106. gs_effect_set_vec2(scale, &s);
  1107. gs_effect_set_texture(image, prev->overflow);
  1108. gs_matrix_push();
  1109. gs_matrix_mul(&boxTransform);
  1110. obs_sceneitem_crop crop;
  1111. obs_sceneitem_get_crop(item, &crop);
  1112. while (gs_effect_loop(solid, "Draw")) {
  1113. gs_draw_sprite(prev->overflow, 0, 1, 1);
  1114. }
  1115. gs_matrix_pop();
  1116. GS_DEBUG_MARKER_END();
  1117. UNUSED_PARAMETER(scene);
  1118. return true;
  1119. }
  1120. bool OBSBasicPreview::DrawSelectedItem(obs_scene_t *scene,
  1121. obs_sceneitem_t *item, void *param)
  1122. {
  1123. if (obs_sceneitem_locked(item))
  1124. return true;
  1125. if (!SceneItemHasVideo(item))
  1126. return true;
  1127. if (obs_sceneitem_is_group(item)) {
  1128. matrix4 mat;
  1129. obs_sceneitem_get_draw_transform(item, &mat);
  1130. gs_matrix_push();
  1131. gs_matrix_mul(&mat);
  1132. obs_sceneitem_group_enum_items(item, DrawSelectedItem, param);
  1133. gs_matrix_pop();
  1134. }
  1135. OBSBasicPreview *prev = reinterpret_cast<OBSBasicPreview *>(param);
  1136. OBSBasic *main = OBSBasic::Get();
  1137. bool hovered = prev->hoveredPreviewItem == item ||
  1138. prev->hoveredListItem == item;
  1139. bool selected = obs_sceneitem_selected(item);
  1140. if (!selected && !hovered)
  1141. return true;
  1142. matrix4 boxTransform;
  1143. matrix4 invBoxTransform;
  1144. obs_sceneitem_get_box_transform(item, &boxTransform);
  1145. matrix4_inv(&invBoxTransform, &boxTransform);
  1146. vec3 bounds[] = {
  1147. {{{0.f, 0.f, 0.f}}},
  1148. {{{1.f, 0.f, 0.f}}},
  1149. {{{0.f, 1.f, 0.f}}},
  1150. {{{1.f, 1.f, 0.f}}},
  1151. };
  1152. vec4 red;
  1153. vec4 green;
  1154. vec4 blue;
  1155. vec4_set(&red, 1.0f, 0.0f, 0.0f, 1.0f);
  1156. vec4_set(&green, 0.0f, 1.0f, 0.0f, 1.0f);
  1157. vec4_set(&blue, 0.0f, 0.5f, 1.0f, 1.0f);
  1158. bool visible = std::all_of(
  1159. std::begin(bounds), std::end(bounds), [&](const vec3 &b) {
  1160. vec3 pos;
  1161. vec3_transform(&pos, &b, &boxTransform);
  1162. vec3_transform(&pos, &pos, &invBoxTransform);
  1163. return CloseFloat(pos.x, b.x) && CloseFloat(pos.y, b.y);
  1164. });
  1165. if (!visible)
  1166. return true;
  1167. GS_DEBUG_MARKER_BEGIN(GS_DEBUG_COLOR_DEFAULT, "DrawSelectedItem");
  1168. matrix4 curTransform;
  1169. vec2 boxScale;
  1170. gs_matrix_get(&curTransform);
  1171. obs_sceneitem_get_box_scale(item, &boxScale);
  1172. boxScale.x *= curTransform.x.x;
  1173. boxScale.y *= curTransform.y.y;
  1174. obs_transform_info info;
  1175. obs_sceneitem_get_info(item, &info);
  1176. gs_matrix_push();
  1177. gs_matrix_mul(&boxTransform);
  1178. obs_sceneitem_crop crop;
  1179. obs_sceneitem_get_crop(item, &crop);
  1180. gs_effect_t *eff = gs_get_effect();
  1181. gs_eparam_t *colParam = gs_effect_get_param_by_name(eff, "color");
  1182. if (info.bounds_type == OBS_BOUNDS_NONE && crop_enabled(&crop)) {
  1183. #define DRAW_SIDE(side, x1, y1, x2, y2) \
  1184. if (hovered && !selected) \
  1185. gs_effect_set_vec4(colParam, &blue); \
  1186. else if (crop.side > 0) \
  1187. gs_effect_set_vec4(colParam, &green); \
  1188. DrawLine(x1, y1, x2, y2, HANDLE_RADIUS / 2, boxScale); \
  1189. gs_effect_set_vec4(colParam, &red);
  1190. DRAW_SIDE(left, 0.0f, 0.0f, 0.0f, 1.0f);
  1191. DRAW_SIDE(top, 0.0f, 0.0f, 1.0f, 0.0f);
  1192. DRAW_SIDE(right, 1.0f, 0.0f, 1.0f, 1.0f);
  1193. DRAW_SIDE(bottom, 0.0f, 1.0f, 1.0f, 1.0f);
  1194. #undef DRAW_SIDE
  1195. } else {
  1196. if (!selected) {
  1197. gs_effect_set_vec4(colParam, &blue);
  1198. DrawRect(HANDLE_RADIUS / 2, boxScale);
  1199. } else {
  1200. DrawRect(HANDLE_RADIUS / 2, boxScale);
  1201. }
  1202. }
  1203. gs_load_vertexbuffer(main->box);
  1204. gs_effect_set_vec4(colParam, &red);
  1205. if (selected) {
  1206. DrawSquareAtPos(0.0f, 0.0f);
  1207. DrawSquareAtPos(0.0f, 1.0f);
  1208. DrawSquareAtPos(1.0f, 0.0f);
  1209. DrawSquareAtPos(1.0f, 1.0f);
  1210. DrawSquareAtPos(0.5f, 0.0f);
  1211. DrawSquareAtPos(0.0f, 0.5f);
  1212. DrawSquareAtPos(0.5f, 1.0f);
  1213. DrawSquareAtPos(1.0f, 0.5f);
  1214. }
  1215. gs_matrix_pop();
  1216. GS_DEBUG_MARKER_END();
  1217. UNUSED_PARAMETER(scene);
  1218. UNUSED_PARAMETER(param);
  1219. return true;
  1220. }
  1221. void OBSBasicPreview::DrawOverflow()
  1222. {
  1223. if (locked)
  1224. return;
  1225. bool hidden = config_get_bool(GetGlobalConfig(), "BasicWindow",
  1226. "OverflowHidden");
  1227. if (hidden)
  1228. return;
  1229. GS_DEBUG_MARKER_BEGIN(GS_DEBUG_COLOR_DEFAULT, "DrawOverflow");
  1230. if (!overflow) {
  1231. std::string path;
  1232. GetDataFilePath("images/overflow.png", path);
  1233. overflow = gs_texture_create_from_file(path.c_str());
  1234. }
  1235. OBSBasic *main = reinterpret_cast<OBSBasic *>(App()->GetMainWindow());
  1236. OBSScene scene = main->GetCurrentScene();
  1237. if (scene) {
  1238. gs_matrix_push();
  1239. gs_matrix_scale3f(main->previewScale, main->previewScale, 1.0f);
  1240. obs_scene_enum_items(scene, DrawSelectedOverflow, this);
  1241. gs_matrix_pop();
  1242. }
  1243. gs_load_vertexbuffer(nullptr);
  1244. GS_DEBUG_MARKER_END();
  1245. }
  1246. void OBSBasicPreview::DrawSceneEditing()
  1247. {
  1248. if (locked)
  1249. return;
  1250. GS_DEBUG_MARKER_BEGIN(GS_DEBUG_COLOR_DEFAULT, "DrawSceneEditing");
  1251. OBSBasic *main = reinterpret_cast<OBSBasic *>(App()->GetMainWindow());
  1252. gs_effect_t *solid = obs_get_base_effect(OBS_EFFECT_SOLID);
  1253. gs_technique_t *tech = gs_effect_get_technique(solid, "Solid");
  1254. vec4 color;
  1255. vec4_set(&color, 1.0f, 0.0f, 0.0f, 1.0f);
  1256. gs_effect_set_vec4(gs_effect_get_param_by_name(solid, "color"), &color);
  1257. gs_technique_begin(tech);
  1258. gs_technique_begin_pass(tech, 0);
  1259. OBSScene scene = main->GetCurrentScene();
  1260. if (scene) {
  1261. gs_matrix_push();
  1262. gs_matrix_scale3f(main->previewScale, main->previewScale, 1.0f);
  1263. obs_scene_enum_items(scene, DrawSelectedItem, this);
  1264. gs_matrix_pop();
  1265. }
  1266. gs_load_vertexbuffer(nullptr);
  1267. gs_technique_end_pass(tech);
  1268. gs_technique_end(tech);
  1269. GS_DEBUG_MARKER_END();
  1270. }
  1271. void OBSBasicPreview::ResetScrollingOffset()
  1272. {
  1273. vec2_zero(&scrollingOffset);
  1274. }
  1275. void OBSBasicPreview::SetScalingLevel(int32_t newScalingLevelVal)
  1276. {
  1277. float newScalingAmountVal =
  1278. pow(ZOOM_SENSITIVITY, float(newScalingLevelVal));
  1279. scalingLevel = newScalingLevelVal;
  1280. SetScalingAmount(newScalingAmountVal);
  1281. }
  1282. void OBSBasicPreview::SetScalingAmount(float newScalingAmountVal)
  1283. {
  1284. scrollingOffset.x *= newScalingAmountVal / scalingAmount;
  1285. scrollingOffset.y *= newScalingAmountVal / scalingAmount;
  1286. scalingAmount = newScalingAmountVal;
  1287. }
  1288. OBSBasicPreview *OBSBasicPreview::Get()
  1289. {
  1290. return OBSBasic::Get()->ui->preview;
  1291. }