window-basic-preview.cpp 37 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488
  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. gs_texture_destroy(overflow);
  28. }
  29. }
  30. vec2 OBSBasicPreview::GetMouseEventPos(QMouseEvent *event)
  31. {
  32. OBSBasic *main = reinterpret_cast<OBSBasic*>(App()->GetMainWindow());
  33. #ifdef SUPPORTS_FRACTIONAL_SCALING
  34. float pixelRatio = main->devicePixelRatioF();
  35. #else
  36. float pixelRatio = main->devicePixelRatio();
  37. #endif
  38. float scale = pixelRatio / main->previewScale;
  39. vec2 pos;
  40. vec2_set(&pos,
  41. (float(event->x()) - main->previewX / pixelRatio) * scale,
  42. (float(event->y()) - main->previewY / pixelRatio) * scale);
  43. return pos;
  44. }
  45. struct SceneFindData {
  46. const vec2 &pos;
  47. OBSSceneItem item;
  48. bool selectBelow;
  49. obs_sceneitem_t *group = nullptr;
  50. SceneFindData(const SceneFindData &) = delete;
  51. SceneFindData(SceneFindData &&) = delete;
  52. SceneFindData& operator=(const SceneFindData &) = delete;
  53. SceneFindData& operator=(SceneFindData &&) = delete;
  54. inline SceneFindData(const vec2 &pos_, bool selectBelow_)
  55. : pos (pos_),
  56. selectBelow (selectBelow_)
  57. {}
  58. };
  59. static bool SceneItemHasVideo(obs_sceneitem_t *item)
  60. {
  61. obs_source_t *source = obs_sceneitem_get_source(item);
  62. uint32_t flags = obs_source_get_output_flags(source);
  63. return (flags & OBS_SOURCE_VIDEO) != 0;
  64. }
  65. static bool CloseFloat(float a, float b, float epsilon=0.01)
  66. {
  67. using std::abs;
  68. return abs(a-b) <= epsilon;
  69. }
  70. static bool FindItemAtPos(obs_scene_t *scene, obs_sceneitem_t *item,
  71. void *param)
  72. {
  73. SceneFindData *data = reinterpret_cast<SceneFindData*>(param);
  74. matrix4 transform;
  75. matrix4 invTransform;
  76. vec3 transformedPos;
  77. vec3 pos3;
  78. vec3 pos3_;
  79. if (!SceneItemHasVideo(item))
  80. return true;
  81. if (obs_sceneitem_locked(item))
  82. return true;
  83. vec3_set(&pos3, data->pos.x, data->pos.y, 0.0f);
  84. obs_sceneitem_get_box_transform(item, &transform);
  85. matrix4_inv(&invTransform, &transform);
  86. vec3_transform(&transformedPos, &pos3, &invTransform);
  87. vec3_transform(&pos3_, &transformedPos, &transform);
  88. if (CloseFloat(pos3.x, pos3_.x) && CloseFloat(pos3.y, pos3_.y) &&
  89. transformedPos.x >= 0.0f && transformedPos.x <= 1.0f &&
  90. transformedPos.y >= 0.0f && transformedPos.y <= 1.0f) {
  91. if (data->selectBelow && obs_sceneitem_selected(item)) {
  92. if (data->item)
  93. return false;
  94. else
  95. data->selectBelow = false;
  96. }
  97. data->item = item;
  98. }
  99. UNUSED_PARAMETER(scene);
  100. return true;
  101. }
  102. static vec3 GetTransformedPos(float x, float y, const matrix4 &mat)
  103. {
  104. vec3 result;
  105. vec3_set(&result, x, y, 0.0f);
  106. vec3_transform(&result, &result, &mat);
  107. return result;
  108. }
  109. static inline vec2 GetOBSScreenSize()
  110. {
  111. obs_video_info ovi;
  112. vec2 size;
  113. vec2_zero(&size);
  114. if (obs_get_video_info(&ovi)) {
  115. size.x = float(ovi.base_width);
  116. size.y = float(ovi.base_height);
  117. }
  118. return size;
  119. }
  120. vec3 OBSBasicPreview::GetSnapOffset(const vec3 &tl, const vec3 &br)
  121. {
  122. OBSBasic *main = reinterpret_cast<OBSBasic*>(App()->GetMainWindow());
  123. vec2 screenSize = GetOBSScreenSize();
  124. vec3 clampOffset;
  125. vec3_zero(&clampOffset);
  126. const bool snap = config_get_bool(GetGlobalConfig(),
  127. "BasicWindow", "SnappingEnabled");
  128. if (snap == false)
  129. return clampOffset;
  130. const bool screenSnap = config_get_bool(GetGlobalConfig(),
  131. "BasicWindow", "ScreenSnapping");
  132. const bool centerSnap = config_get_bool(GetGlobalConfig(),
  133. "BasicWindow", "CenterSnapping");
  134. const float clampDist = config_get_double(GetGlobalConfig(),
  135. "BasicWindow", "SnapDistance") / main->previewScale;
  136. const float centerX = br.x - (br.x - tl.x) / 2.0f;
  137. const float centerY = br.y - (br.y - tl.y) / 2.0f;
  138. // Left screen edge.
  139. if (screenSnap &&
  140. fabsf(tl.x) < clampDist)
  141. clampOffset.x = -tl.x;
  142. // Right screen edge.
  143. if (screenSnap &&
  144. fabsf(clampOffset.x) < EPSILON &&
  145. fabsf(screenSize.x - br.x) < clampDist)
  146. clampOffset.x = screenSize.x - br.x;
  147. // Horizontal center.
  148. if (centerSnap &&
  149. fabsf(screenSize.x - (br.x - tl.x)) > clampDist &&
  150. fabsf(screenSize.x / 2.0f - centerX) < clampDist)
  151. clampOffset.x = screenSize.x / 2.0f - centerX;
  152. // Top screen edge.
  153. if (screenSnap &&
  154. fabsf(tl.y) < clampDist)
  155. clampOffset.y = -tl.y;
  156. // Bottom screen edge.
  157. if (screenSnap &&
  158. fabsf(clampOffset.y) < EPSILON &&
  159. fabsf(screenSize.y - br.y) < clampDist)
  160. clampOffset.y = screenSize.y - br.y;
  161. // Vertical center.
  162. if (centerSnap &&
  163. fabsf(screenSize.y - (br.y - tl.y)) > clampDist &&
  164. fabsf(screenSize.y / 2.0f - centerY) < clampDist)
  165. clampOffset.y = screenSize.y / 2.0f - centerY;
  166. return clampOffset;
  167. }
  168. OBSSceneItem OBSBasicPreview::GetItemAtPos(const vec2 &pos, bool selectBelow)
  169. {
  170. OBSBasic *main = reinterpret_cast<OBSBasic*>(App()->GetMainWindow());
  171. OBSScene scene = main->GetCurrentScene();
  172. if (!scene)
  173. return OBSSceneItem();
  174. SceneFindData data(pos, selectBelow);
  175. obs_scene_enum_items(scene, FindItemAtPos, &data);
  176. return data.item;
  177. }
  178. static bool CheckItemSelected(obs_scene_t *scene, obs_sceneitem_t *item,
  179. void *param)
  180. {
  181. SceneFindData *data = reinterpret_cast<SceneFindData*>(param);
  182. matrix4 transform;
  183. vec3 transformedPos;
  184. vec3 pos3;
  185. if (!SceneItemHasVideo(item))
  186. return true;
  187. if (obs_sceneitem_is_group(item)) {
  188. data->group = item;
  189. obs_sceneitem_group_enum_items(item, CheckItemSelected, param);
  190. data->group = nullptr;
  191. if (data->item) {
  192. return false;
  193. }
  194. }
  195. vec3_set(&pos3, data->pos.x, data->pos.y, 0.0f);
  196. obs_sceneitem_get_box_transform(item, &transform);
  197. if (data->group) {
  198. matrix4 parent_transform;
  199. obs_sceneitem_get_draw_transform(data->group, &parent_transform);
  200. matrix4_mul(&transform, &transform, &parent_transform);
  201. }
  202. matrix4_inv(&transform, &transform);
  203. vec3_transform(&transformedPos, &pos3, &transform);
  204. if (transformedPos.x >= 0.0f && transformedPos.x <= 1.0f &&
  205. transformedPos.y >= 0.0f && transformedPos.y <= 1.0f) {
  206. if (obs_sceneitem_selected(item)) {
  207. data->item = item;
  208. return false;
  209. }
  210. }
  211. UNUSED_PARAMETER(scene);
  212. return true;
  213. }
  214. bool OBSBasicPreview::SelectedAtPos(const vec2 &pos)
  215. {
  216. OBSBasic *main = reinterpret_cast<OBSBasic*>(App()->GetMainWindow());
  217. OBSScene scene = main->GetCurrentScene();
  218. if (!scene)
  219. return false;
  220. SceneFindData data(pos, false);
  221. obs_scene_enum_items(scene, CheckItemSelected, &data);
  222. return !!data.item;
  223. }
  224. struct HandleFindData {
  225. const vec2 &pos;
  226. const float radius;
  227. matrix4 parent_xform;
  228. OBSSceneItem item;
  229. ItemHandle handle = ItemHandle::None;
  230. HandleFindData(const HandleFindData &) = delete;
  231. HandleFindData(HandleFindData &&) = delete;
  232. HandleFindData& operator=(const HandleFindData &) = delete;
  233. HandleFindData& operator=(HandleFindData &&) = delete;
  234. inline HandleFindData(const vec2 &pos_, float scale)
  235. : pos (pos_),
  236. radius (HANDLE_SEL_RADIUS / scale)
  237. {
  238. matrix4_identity(&parent_xform);
  239. }
  240. inline HandleFindData(const HandleFindData &hfd,
  241. obs_sceneitem_t *parent)
  242. : pos (hfd.pos),
  243. radius (hfd.radius),
  244. item (hfd.item),
  245. handle (hfd.handle)
  246. {
  247. obs_sceneitem_get_draw_transform(parent, &parent_xform);
  248. }
  249. };
  250. static bool FindHandleAtPos(obs_scene_t *scene, obs_sceneitem_t *item,
  251. void *param)
  252. {
  253. HandleFindData &data = *reinterpret_cast<HandleFindData*>(param);
  254. if (!obs_sceneitem_selected(item)) {
  255. if (obs_sceneitem_is_group(item)) {
  256. HandleFindData newData(data, item);
  257. obs_sceneitem_group_enum_items(item, FindHandleAtPos,
  258. &newData);
  259. data.item = newData.item;
  260. data.handle = newData.handle;
  261. }
  262. return true;
  263. }
  264. matrix4 transform;
  265. vec3 pos3;
  266. float closestHandle = data.radius;
  267. vec3_set(&pos3, data.pos.x, data.pos.y, 0.0f);
  268. obs_sceneitem_get_box_transform(item, &transform);
  269. auto TestHandle = [&] (float x, float y, ItemHandle handle)
  270. {
  271. vec3 handlePos = GetTransformedPos(x, y, transform);
  272. vec3_transform(&handlePos, &handlePos, &data.parent_xform);
  273. float dist = vec3_dist(&handlePos, &pos3);
  274. if (dist < data.radius) {
  275. if (dist < closestHandle) {
  276. closestHandle = dist;
  277. data.handle = handle;
  278. data.item = item;
  279. }
  280. }
  281. };
  282. TestHandle(0.0f, 0.0f, ItemHandle::TopLeft);
  283. TestHandle(0.5f, 0.0f, ItemHandle::TopCenter);
  284. TestHandle(1.0f, 0.0f, ItemHandle::TopRight);
  285. TestHandle(0.0f, 0.5f, ItemHandle::CenterLeft);
  286. TestHandle(1.0f, 0.5f, ItemHandle::CenterRight);
  287. TestHandle(0.0f, 1.0f, ItemHandle::BottomLeft);
  288. TestHandle(0.5f, 1.0f, ItemHandle::BottomCenter);
  289. TestHandle(1.0f, 1.0f, ItemHandle::BottomRight);
  290. UNUSED_PARAMETER(scene);
  291. return true;
  292. }
  293. static vec2 GetItemSize(obs_sceneitem_t *item)
  294. {
  295. obs_bounds_type boundsType = obs_sceneitem_get_bounds_type(item);
  296. vec2 size;
  297. if (boundsType != OBS_BOUNDS_NONE) {
  298. obs_sceneitem_get_bounds(item, &size);
  299. } else {
  300. obs_source_t *source = obs_sceneitem_get_source(item);
  301. obs_sceneitem_crop crop;
  302. vec2 scale;
  303. obs_sceneitem_get_scale(item, &scale);
  304. obs_sceneitem_get_crop(item, &crop);
  305. size.x = float(obs_source_get_width(source) -
  306. crop.left - crop.right) * scale.x;
  307. size.y = float(obs_source_get_height(source) -
  308. crop.top - crop.bottom) * scale.y;
  309. }
  310. return size;
  311. }
  312. void OBSBasicPreview::GetStretchHandleData(const vec2 &pos)
  313. {
  314. OBSBasic *main = reinterpret_cast<OBSBasic*>(App()->GetMainWindow());
  315. OBSScene scene = main->GetCurrentScene();
  316. if (!scene)
  317. return;
  318. #ifdef SUPPORTS_FRACTIONAL_SCALING
  319. float scale = main->previewScale / main->devicePixelRatioF();
  320. #else
  321. float scale = main->previewScale / main->devicePixelRatio();
  322. #endif
  323. vec2 scaled_pos = pos;
  324. vec2_divf(&scaled_pos, &scaled_pos, scale);
  325. HandleFindData data(scaled_pos, scale);
  326. obs_scene_enum_items(scene, FindHandleAtPos, &data);
  327. stretchItem = std::move(data.item);
  328. stretchHandle = data.handle;
  329. if (stretchHandle != ItemHandle::None) {
  330. matrix4 boxTransform;
  331. vec3 itemUL;
  332. float itemRot;
  333. stretchItemSize = GetItemSize(stretchItem);
  334. obs_sceneitem_get_box_transform(stretchItem, &boxTransform);
  335. itemRot = obs_sceneitem_get_rot(stretchItem);
  336. vec3_from_vec4(&itemUL, &boxTransform.t);
  337. /* build the item space conversion matrices */
  338. matrix4_identity(&itemToScreen);
  339. matrix4_rotate_aa4f(&itemToScreen, &itemToScreen,
  340. 0.0f, 0.0f, 1.0f, RAD(itemRot));
  341. matrix4_translate3f(&itemToScreen, &itemToScreen,
  342. itemUL.x, itemUL.y, 0.0f);
  343. matrix4_identity(&screenToItem);
  344. matrix4_translate3f(&screenToItem, &screenToItem,
  345. -itemUL.x, -itemUL.y, 0.0f);
  346. matrix4_rotate_aa4f(&screenToItem, &screenToItem,
  347. 0.0f, 0.0f, 1.0f, RAD(-itemRot));
  348. obs_sceneitem_get_crop(stretchItem, &startCrop);
  349. obs_sceneitem_get_pos(stretchItem, &startItemPos);
  350. obs_source_t *source = obs_sceneitem_get_source(stretchItem);
  351. cropSize.x = float(obs_source_get_width(source) -
  352. startCrop.left - startCrop.right);
  353. cropSize.y = float(obs_source_get_height(source) -
  354. startCrop.top - startCrop.bottom);
  355. stretchGroup = obs_sceneitem_get_group(scene, stretchItem);
  356. if (stretchGroup) {
  357. obs_sceneitem_get_draw_transform(stretchGroup,
  358. &invGroupTransform);
  359. matrix4_inv(&invGroupTransform,
  360. &invGroupTransform);
  361. obs_sceneitem_defer_group_resize_begin(stretchGroup);
  362. }
  363. }
  364. }
  365. void OBSBasicPreview::keyPressEvent(QKeyEvent *event)
  366. {
  367. if (!IsFixedScaling() || event->isAutoRepeat()) {
  368. OBSQTDisplay::keyPressEvent(event);
  369. return;
  370. }
  371. switch (event->key()) {
  372. case Qt::Key_Space:
  373. setCursor(Qt::OpenHandCursor);
  374. scrollMode = true;
  375. break;
  376. }
  377. OBSQTDisplay::keyPressEvent(event);
  378. }
  379. void OBSBasicPreview::keyReleaseEvent(QKeyEvent *event)
  380. {
  381. if (event->isAutoRepeat()) {
  382. OBSQTDisplay::keyReleaseEvent(event);
  383. return;
  384. }
  385. switch (event->key()) {
  386. case Qt::Key_Space:
  387. scrollMode = false;
  388. setCursor(Qt::ArrowCursor);
  389. break;
  390. }
  391. OBSQTDisplay::keyReleaseEvent(event);
  392. }
  393. void OBSBasicPreview::wheelEvent(QWheelEvent *event)
  394. {
  395. if (scrollMode && IsFixedScaling()
  396. && event->orientation() == Qt::Vertical) {
  397. if (event->delta() > 0)
  398. SetScalingLevel(scalingLevel + 1);
  399. else if (event->delta() < 0)
  400. SetScalingLevel(scalingLevel - 1);
  401. emit DisplayResized();
  402. }
  403. OBSQTDisplay::wheelEvent(event);
  404. }
  405. void OBSBasicPreview::mousePressEvent(QMouseEvent *event)
  406. {
  407. if (scrollMode && IsFixedScaling() &&
  408. event->button() == Qt::LeftButton) {
  409. setCursor(Qt::ClosedHandCursor);
  410. scrollingFrom.x = event->x();
  411. scrollingFrom.y = event->y();
  412. return;
  413. }
  414. if (event->button() == Qt::RightButton) {
  415. scrollMode = false;
  416. setCursor(Qt::ArrowCursor);
  417. }
  418. if (locked) {
  419. OBSQTDisplay::mousePressEvent(event);
  420. return;
  421. }
  422. OBSBasic *main = reinterpret_cast<OBSBasic*>(App()->GetMainWindow());
  423. #ifdef SUPPORTS_FRACTIONAL_SCALING
  424. float pixelRatio = main->devicePixelRatioF();
  425. #else
  426. float pixelRatio = main->devicePixelRatio();
  427. #endif
  428. float x = float(event->x()) - main->previewX / pixelRatio;
  429. float y = float(event->y()) - main->previewY / pixelRatio;
  430. Qt::KeyboardModifiers modifiers = QGuiApplication::keyboardModifiers();
  431. bool altDown = (modifiers & Qt::AltModifier);
  432. OBSQTDisplay::mousePressEvent(event);
  433. if (event->button() != Qt::LeftButton &&
  434. event->button() != Qt::RightButton)
  435. return;
  436. if (event->button() == Qt::LeftButton)
  437. mouseDown = true;
  438. if (altDown)
  439. cropping = true;
  440. vec2_set(&startPos, x, y);
  441. GetStretchHandleData(startPos);
  442. vec2_divf(&startPos, &startPos, main->previewScale / pixelRatio);
  443. startPos.x = std::round(startPos.x);
  444. startPos.y = std::round(startPos.y);
  445. mouseOverItems = SelectedAtPos(startPos);
  446. vec2_zero(&lastMoveOffset);
  447. }
  448. static bool select_one(obs_scene_t *scene, obs_sceneitem_t *item, void *param)
  449. {
  450. obs_sceneitem_t *selectedItem =
  451. reinterpret_cast<obs_sceneitem_t*>(param);
  452. if (obs_sceneitem_is_group(item))
  453. obs_sceneitem_group_enum_items(item, select_one, param);
  454. obs_sceneitem_select(item, (selectedItem == item));
  455. UNUSED_PARAMETER(scene);
  456. return true;
  457. }
  458. void OBSBasicPreview::DoSelect(const vec2 &pos)
  459. {
  460. OBSBasic *main = reinterpret_cast<OBSBasic*>(App()->GetMainWindow());
  461. OBSScene scene = main->GetCurrentScene();
  462. OBSSceneItem item = GetItemAtPos(pos, true);
  463. obs_scene_enum_items(scene, select_one, (obs_sceneitem_t*)item);
  464. }
  465. void OBSBasicPreview::DoCtrlSelect(const vec2 &pos)
  466. {
  467. OBSSceneItem item = GetItemAtPos(pos, false);
  468. if (!item)
  469. return;
  470. bool selected = obs_sceneitem_selected(item);
  471. obs_sceneitem_select(item, !selected);
  472. }
  473. void OBSBasicPreview::ProcessClick(const vec2 &pos)
  474. {
  475. Qt::KeyboardModifiers modifiers = QGuiApplication::keyboardModifiers();
  476. if (modifiers & Qt::ControlModifier)
  477. DoCtrlSelect(pos);
  478. else
  479. DoSelect(pos);
  480. }
  481. void OBSBasicPreview::mouseReleaseEvent(QMouseEvent *event)
  482. {
  483. if (scrollMode)
  484. setCursor(Qt::OpenHandCursor);
  485. if (locked) {
  486. OBSQTDisplay::mouseReleaseEvent(event);
  487. return;
  488. }
  489. if (mouseDown) {
  490. vec2 pos = GetMouseEventPos(event);
  491. if (!mouseMoved)
  492. ProcessClick(pos);
  493. if (stretchGroup) {
  494. obs_sceneitem_defer_group_resize_end(stretchGroup);
  495. }
  496. stretchItem = nullptr;
  497. stretchGroup = nullptr;
  498. mouseDown = false;
  499. mouseMoved = false;
  500. cropping = false;
  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 = reinterpret_cast<SelectedItemBounds*>(param);
  511. vec3 t[4];
  512. auto add_bounds = [data, &t] ()
  513. {
  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] = {
  567. GetTransformedPos(0.0f, 0.0f, boxTransform),
  568. GetTransformedPos(1.0f, 0.0f, boxTransform),
  569. GetTransformedPos(0.0f, 1.0f, boxTransform),
  570. GetTransformedPos(1.0f, 1.0f, boxTransform)
  571. };
  572. bool first = true;
  573. vec3 tl, br;
  574. vec3_zero(&tl);
  575. vec3_zero(&br);
  576. for (const vec3 &v : t) {
  577. if (first) {
  578. vec3_copy(&tl, &v);
  579. vec3_copy(&br, &v);
  580. first = false;
  581. } else {
  582. vec3_min(&tl, &tl, &v);
  583. vec3_max(&br, &br, &v);
  584. }
  585. }
  586. // Snap to other source edges
  587. #define EDGE_SNAP(l, r, x, y) \
  588. do { \
  589. double dist = fabsf(l.x - data->r.x); \
  590. if (dist < data->clampDist && \
  591. fabsf(data->offset.x) < EPSILON && \
  592. data->tl.y < br.y && \
  593. data->br.y > tl.y && \
  594. (fabsf(data->offset.x) > dist || data->offset.x < EPSILON)) \
  595. data->offset.x = l.x - data->r.x; \
  596. } while (false)
  597. EDGE_SNAP(tl, br, x, y);
  598. EDGE_SNAP(tl, br, y, x);
  599. EDGE_SNAP(br, tl, x, y);
  600. EDGE_SNAP(br, tl, y, x);
  601. #undef EDGE_SNAP
  602. UNUSED_PARAMETER(scene);
  603. return true;
  604. }
  605. void OBSBasicPreview::SnapItemMovement(vec2 &offset)
  606. {
  607. OBSBasic *main = reinterpret_cast<OBSBasic*>(App()->GetMainWindow());
  608. OBSScene scene = main->GetCurrentScene();
  609. SelectedItemBounds data;
  610. obs_scene_enum_items(scene, AddItemBounds, &data);
  611. data.tl.x += offset.x;
  612. data.tl.y += offset.y;
  613. data.br.x += offset.x;
  614. data.br.y += offset.y;
  615. vec3 snapOffset = GetSnapOffset(data.tl, data.br);
  616. const bool snap = config_get_bool(GetGlobalConfig(),
  617. "BasicWindow", "SnappingEnabled");
  618. const bool sourcesSnap = config_get_bool(GetGlobalConfig(),
  619. "BasicWindow", "SourceSnapping");
  620. if (snap == false)
  621. return;
  622. if (sourcesSnap == false) {
  623. offset.x += snapOffset.x;
  624. offset.y += snapOffset.y;
  625. return;
  626. }
  627. const float clampDist = config_get_double(GetGlobalConfig(),
  628. "BasicWindow", "SnapDistance") / 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,
  804. float(-crop.left) * scale.x,
  805. float(-crop.top) * scale.y);
  806. vec2_set(&max_br,
  807. stretchItemSize.x + crop.right * scale.x,
  808. stretchItemSize.y + crop.bottom * scale.y);
  809. typedef std::function<float (float, float)> minmax_func_t;
  810. minmax_func_t min_x = scale.x < 0.0f ? maxfunc : minfunc;
  811. minmax_func_t min_y = scale.y < 0.0f ? maxfunc : minfunc;
  812. minmax_func_t max_x = scale.x < 0.0f ? minfunc : maxfunc;
  813. minmax_func_t max_y = scale.y < 0.0f ? minfunc : maxfunc;
  814. pos3.x = min_x(pos3.x, max_br.x);
  815. pos3.x = max_x(pos3.x, max_tl.x);
  816. pos3.y = min_y(pos3.y, max_br.y);
  817. pos3.y = max_y(pos3.y, max_tl.y);
  818. if (stretchFlags & ITEM_LEFT) {
  819. float maxX = stretchItemSize.x - (2.0 * scale.x);
  820. pos3.x = tl.x = min_x(pos3.x, maxX);
  821. } else if (stretchFlags & ITEM_RIGHT) {
  822. float minX = (2.0 * scale.x);
  823. pos3.x = br.x = max_x(pos3.x, minX);
  824. }
  825. if (stretchFlags & ITEM_TOP) {
  826. float maxY = stretchItemSize.y - (2.0 * scale.y);
  827. pos3.y = tl.y = min_y(pos3.y, maxY);
  828. } else if (stretchFlags & ITEM_BOTTOM) {
  829. float minY = (2.0 * scale.y);
  830. pos3.y = br.y = max_y(pos3.y, minY);
  831. }
  832. #define ALIGN_X (ITEM_LEFT|ITEM_RIGHT)
  833. #define ALIGN_Y (ITEM_TOP|ITEM_BOTTOM)
  834. vec3 newPos;
  835. vec3_zero(&newPos);
  836. uint32_t align_x = (align & ALIGN_X);
  837. uint32_t align_y = (align & ALIGN_Y);
  838. if (align_x == (stretchFlags & ALIGN_X) && align_x != 0)
  839. newPos.x = pos3.x;
  840. else if (align & ITEM_RIGHT)
  841. newPos.x = stretchItemSize.x;
  842. else if (!(align & ITEM_LEFT))
  843. newPos.x = stretchItemSize.x * 0.5f;
  844. if (align_y == (stretchFlags & ALIGN_Y) && align_y != 0)
  845. newPos.y = pos3.y;
  846. else if (align & ITEM_BOTTOM)
  847. newPos.y = stretchItemSize.y;
  848. else if (!(align & ITEM_TOP))
  849. newPos.y = stretchItemSize.y * 0.5f;
  850. #undef ALIGN_X
  851. #undef ALIGN_Y
  852. crop = startCrop;
  853. if (stretchFlags & ITEM_LEFT)
  854. crop.left += int(std::round(tl.x / scale.x));
  855. else if (stretchFlags & ITEM_RIGHT)
  856. crop.right += int(std::round((stretchItemSize.x - br.x) / scale.x));
  857. if (stretchFlags & ITEM_TOP)
  858. crop.top += int(std::round(tl.y / scale.y));
  859. else if (stretchFlags & ITEM_BOTTOM)
  860. crop.bottom += 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,
  905. float(obs_source_get_width(source)),
  906. float(obs_source_get_height(source)));
  907. vec2 size;
  908. vec2_set(&size,br. x - tl.x, br.y - tl.y);
  909. if (boundsType != OBS_BOUNDS_NONE) {
  910. if (shiftDown)
  911. ClampAspect(tl, br, size, baseSize);
  912. if (tl.x > br.x) std::swap(tl.x, br.x);
  913. if (tl.y > br.y) std::swap(tl.y, br.y);
  914. vec2_abs(&size, &size);
  915. obs_sceneitem_set_bounds(stretchItem, &size);
  916. } else {
  917. obs_sceneitem_crop crop;
  918. obs_sceneitem_get_crop(stretchItem, &crop);
  919. baseSize.x -= float(crop.left + crop.right);
  920. baseSize.y -= float(crop.top + crop.bottom);
  921. if (!shiftDown)
  922. ClampAspect(tl, br, size, baseSize);
  923. vec2_div(&size, &size, &baseSize);
  924. obs_sceneitem_set_scale(stretchItem, &size);
  925. }
  926. pos3 = CalculateStretchPos(tl, br);
  927. vec3_transform(&pos3, &pos3, &itemToScreen);
  928. vec2 newPos;
  929. vec2_set(&newPos, std::round(pos3.x), std::round(pos3.y));
  930. obs_sceneitem_set_pos(stretchItem, &newPos);
  931. }
  932. void OBSBasicPreview::mouseMoveEvent(QMouseEvent *event)
  933. {
  934. if (scrollMode && event->buttons() == Qt::LeftButton) {
  935. scrollingOffset.x += event->x() - scrollingFrom.x;
  936. scrollingOffset.y += event->y() - scrollingFrom.y;
  937. scrollingFrom.x = event->x();
  938. scrollingFrom.y = event->y();
  939. emit DisplayResized();
  940. return;
  941. }
  942. if (locked)
  943. return;
  944. if (mouseDown) {
  945. vec2 pos = GetMouseEventPos(event);
  946. if (!mouseMoved && !mouseOverItems &&
  947. stretchHandle == ItemHandle::None) {
  948. ProcessClick(startPos);
  949. mouseOverItems = SelectedAtPos(startPos);
  950. }
  951. pos.x = std::round(pos.x);
  952. pos.y = std::round(pos.y);
  953. if (stretchHandle != ItemHandle::None) {
  954. OBSBasic *main = reinterpret_cast<OBSBasic*>(
  955. App()->GetMainWindow());
  956. OBSScene scene = main->GetCurrentScene();
  957. obs_sceneitem_t *group = obs_sceneitem_get_group(
  958. scene, stretchItem);
  959. if (group) {
  960. vec3 group_pos;
  961. vec3_set(&group_pos, pos.x, pos.y, 0.0f);
  962. vec3_transform(&group_pos, &group_pos,
  963. &invGroupTransform);
  964. pos.x = group_pos.x;
  965. pos.y = group_pos.y;
  966. }
  967. if (cropping)
  968. CropItem(pos);
  969. else
  970. StretchItem(pos);
  971. } else if (mouseOverItems) {
  972. MoveItems(pos);
  973. }
  974. mouseMoved = true;
  975. }
  976. }
  977. static void DrawCircleAtPos(float x, float y)
  978. {
  979. struct vec3 pos;
  980. vec3_set(&pos, x, y, 0.0f);
  981. struct matrix4 matrix;
  982. gs_matrix_get(&matrix);
  983. vec3_transform(&pos, &pos, &matrix);
  984. gs_matrix_push();
  985. gs_matrix_identity();
  986. gs_matrix_translate(&pos);
  987. gs_matrix_scale3f(HANDLE_RADIUS, HANDLE_RADIUS, 1.0f);
  988. gs_draw(GS_LINESTRIP, 0, 0);
  989. gs_matrix_pop();
  990. }
  991. static inline bool crop_enabled(const obs_sceneitem_crop *crop)
  992. {
  993. return crop->left > 0 ||
  994. crop->top > 0 ||
  995. crop->right > 0 ||
  996. crop->bottom > 0;
  997. }
  998. bool OBSBasicPreview::DrawSelectedOverflow(obs_scene_t *scene,
  999. obs_sceneitem_t *item, void *param)
  1000. {
  1001. if (obs_sceneitem_locked(item))
  1002. return true;
  1003. if (!SceneItemHasVideo(item))
  1004. return true;
  1005. bool select = config_get_bool(GetGlobalConfig(), "BasicWindow",
  1006. "OverflowSelectionHidden");
  1007. if (!select && !obs_sceneitem_visible(item))
  1008. return true;
  1009. if (obs_sceneitem_is_group(item)) {
  1010. matrix4 mat;
  1011. obs_sceneitem_get_draw_transform(item, &mat);
  1012. gs_matrix_push();
  1013. gs_matrix_mul(&mat);
  1014. obs_sceneitem_group_enum_items(item, DrawSelectedOverflow, param);
  1015. gs_matrix_pop();
  1016. }
  1017. bool always = config_get_bool(GetGlobalConfig(), "BasicWindow",
  1018. "OverflowAlwaysVisible");
  1019. if (!always && !obs_sceneitem_selected(item))
  1020. return true;
  1021. OBSBasicPreview *prev = reinterpret_cast<OBSBasicPreview*>(param);
  1022. matrix4 boxTransform;
  1023. matrix4 invBoxTransform;
  1024. obs_sceneitem_get_box_transform(item, &boxTransform);
  1025. matrix4_inv(&invBoxTransform, &boxTransform);
  1026. vec3 bounds[] = {
  1027. {{{0.f, 0.f, 0.f}}},
  1028. {{{1.f, 0.f, 0.f}}},
  1029. {{{0.f, 1.f, 0.f}}},
  1030. {{{1.f, 1.f, 0.f}}},
  1031. };
  1032. bool visible = std::all_of(std::begin(bounds), std::end(bounds),
  1033. [&](const vec3 &b)
  1034. {
  1035. vec3 pos;
  1036. vec3_transform(&pos, &b, &boxTransform);
  1037. vec3_transform(&pos, &pos, &invBoxTransform);
  1038. return CloseFloat(pos.x, b.x) && CloseFloat(pos.y, b.y);
  1039. });
  1040. if (!visible)
  1041. return true;
  1042. obs_transform_info info;
  1043. obs_sceneitem_get_info(item, &info);
  1044. gs_effect_t *solid = obs_get_base_effect(OBS_EFFECT_REPEAT);
  1045. gs_eparam_t *image = gs_effect_get_param_by_name(solid, "image");
  1046. gs_eparam_t *scale = gs_effect_get_param_by_name(solid, "scale");
  1047. vec2 s;
  1048. vec2_set(&s, boxTransform.x.x / 96, boxTransform.y.y / 96);
  1049. gs_effect_set_vec2(scale, &s);
  1050. gs_effect_set_texture(image, prev->overflow);
  1051. gs_matrix_push();
  1052. gs_matrix_mul(&boxTransform);
  1053. obs_sceneitem_crop crop;
  1054. obs_sceneitem_get_crop(item, &crop);
  1055. while (gs_effect_loop(solid, "Draw")) {
  1056. gs_draw_sprite(prev->overflow, 0, 1, 1);
  1057. }
  1058. gs_matrix_pop();
  1059. UNUSED_PARAMETER(scene);
  1060. return true;
  1061. }
  1062. bool OBSBasicPreview::DrawSelectedItem(obs_scene_t *scene,
  1063. obs_sceneitem_t *item, void *param)
  1064. {
  1065. if (obs_sceneitem_locked(item))
  1066. return true;
  1067. if (!SceneItemHasVideo(item))
  1068. return true;
  1069. if (obs_sceneitem_is_group(item)) {
  1070. matrix4 mat;
  1071. obs_sceneitem_get_draw_transform(item, &mat);
  1072. gs_matrix_push();
  1073. gs_matrix_mul(&mat);
  1074. obs_sceneitem_group_enum_items(item, DrawSelectedItem, param);
  1075. gs_matrix_pop();
  1076. }
  1077. if (!obs_sceneitem_selected(item))
  1078. return true;
  1079. OBSBasic *main = reinterpret_cast<OBSBasic*>(App()->GetMainWindow());
  1080. matrix4 boxTransform;
  1081. matrix4 invBoxTransform;
  1082. obs_sceneitem_get_box_transform(item, &boxTransform);
  1083. matrix4_inv(&invBoxTransform, &boxTransform);
  1084. vec3 bounds[] = {
  1085. {{{0.f, 0.f, 0.f}}},
  1086. {{{1.f, 0.f, 0.f}}},
  1087. {{{0.f, 1.f, 0.f}}},
  1088. {{{1.f, 1.f, 0.f}}},
  1089. };
  1090. bool visible = std::all_of(std::begin(bounds), std::end(bounds),
  1091. [&](const vec3 &b)
  1092. {
  1093. vec3 pos;
  1094. vec3_transform(&pos, &b, &boxTransform);
  1095. vec3_transform(&pos, &pos, &invBoxTransform);
  1096. return CloseFloat(pos.x, b.x) && CloseFloat(pos.y, b.y);
  1097. });
  1098. if (!visible)
  1099. return true;
  1100. obs_transform_info info;
  1101. obs_sceneitem_get_info(item, &info);
  1102. gs_load_vertexbuffer(main->circle);
  1103. gs_matrix_push();
  1104. gs_matrix_mul(&boxTransform);
  1105. DrawCircleAtPos(0.0f, 0.0f);
  1106. DrawCircleAtPos(0.0f, 1.0f);
  1107. DrawCircleAtPos(1.0f, 0.0f);
  1108. DrawCircleAtPos(1.0f, 1.0f);
  1109. DrawCircleAtPos(0.5f, 0.0f);
  1110. DrawCircleAtPos(0.0f, 0.5f);
  1111. DrawCircleAtPos(0.5f, 1.0f);
  1112. DrawCircleAtPos(1.0f, 0.5f);
  1113. obs_sceneitem_crop crop;
  1114. obs_sceneitem_get_crop(item, &crop);
  1115. if (info.bounds_type == OBS_BOUNDS_NONE && crop_enabled(&crop)) {
  1116. vec4 color;
  1117. gs_effect_t *eff = gs_get_effect();
  1118. gs_eparam_t *param = gs_effect_get_param_by_name(eff, "color");
  1119. #define DRAW_SIDE(side, vb) \
  1120. if (crop.side > 0) \
  1121. vec4_set(&color, 0.0f, 1.0f, 0.0f, 1.0f); \
  1122. else \
  1123. vec4_set(&color, 1.0f, 0.0f, 0.0f, 1.0f); \
  1124. gs_effect_set_vec4(param, &color); \
  1125. gs_load_vertexbuffer(main->vb); \
  1126. gs_draw(GS_LINESTRIP, 0, 0);
  1127. DRAW_SIDE(left, boxLeft);
  1128. DRAW_SIDE(top, boxTop);
  1129. DRAW_SIDE(right, boxRight);
  1130. DRAW_SIDE(bottom, boxBottom);
  1131. #undef DRAW_SIDE
  1132. } else {
  1133. gs_load_vertexbuffer(main->box);
  1134. gs_draw(GS_LINESTRIP, 0, 0);
  1135. }
  1136. gs_matrix_pop();
  1137. UNUSED_PARAMETER(scene);
  1138. UNUSED_PARAMETER(param);
  1139. return true;
  1140. }
  1141. void OBSBasicPreview::DrawOverflow()
  1142. {
  1143. if (locked)
  1144. return;
  1145. bool hidden = config_get_bool(GetGlobalConfig(), "BasicWindow",
  1146. "OverflowHidden");
  1147. if (hidden)
  1148. return;
  1149. if (!overflow) {
  1150. std::string path;
  1151. GetDataFilePath("images/overflow.png", path);
  1152. overflow = gs_texture_create_from_file(path.c_str());
  1153. }
  1154. OBSBasic *main = reinterpret_cast<OBSBasic*>(App()->GetMainWindow());
  1155. OBSScene scene = main->GetCurrentScene();
  1156. if (scene) {
  1157. gs_matrix_push();
  1158. gs_matrix_scale3f(main->previewScale, main->previewScale, 1.0f);
  1159. obs_scene_enum_items(scene, DrawSelectedOverflow, this);
  1160. gs_matrix_pop();
  1161. }
  1162. gs_load_vertexbuffer(nullptr);
  1163. }
  1164. void OBSBasicPreview::DrawSceneEditing()
  1165. {
  1166. if (locked)
  1167. return;
  1168. OBSBasic *main = reinterpret_cast<OBSBasic*>(App()->GetMainWindow());
  1169. gs_effect_t *solid = obs_get_base_effect(OBS_EFFECT_SOLID);
  1170. gs_technique_t *tech = gs_effect_get_technique(solid, "Solid");
  1171. vec4 color;
  1172. vec4_set(&color, 1.0f, 0.0f, 0.0f, 1.0f);
  1173. gs_effect_set_vec4(gs_effect_get_param_by_name(solid, "color"), &color);
  1174. gs_technique_begin(tech);
  1175. gs_technique_begin_pass(tech, 0);
  1176. OBSScene scene = main->GetCurrentScene();
  1177. if (scene) {
  1178. gs_matrix_push();
  1179. gs_matrix_scale3f(main->previewScale, main->previewScale, 1.0f);
  1180. obs_scene_enum_items(scene, DrawSelectedItem, this);
  1181. gs_matrix_pop();
  1182. }
  1183. gs_load_vertexbuffer(nullptr);
  1184. gs_technique_end_pass(tech);
  1185. gs_technique_end(tech);
  1186. }
  1187. void OBSBasicPreview::ResetScrollingOffset()
  1188. {
  1189. vec2_zero(&scrollingOffset);
  1190. }
  1191. void OBSBasicPreview::SetScalingLevel(int32_t newScalingLevelVal) {
  1192. float newScalingAmountVal = pow(ZOOM_SENSITIVITY, float(newScalingLevelVal));
  1193. scalingLevel = newScalingLevelVal;
  1194. SetScalingAmount(newScalingAmountVal);
  1195. }
  1196. void OBSBasicPreview::SetScalingAmount(float newScalingAmountVal) {
  1197. scrollingOffset.x *= newScalingAmountVal / scalingAmount;
  1198. scrollingOffset.y *= newScalingAmountVal / scalingAmount;
  1199. scalingAmount = newScalingAmountVal;
  1200. }