window-basic-preview.cpp 58 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288
  1. #include <QGuiApplication>
  2. #include <QMouseEvent>
  3. #include <cmath>
  4. #include <string>
  5. #include <graphics/vec4.h>
  6. #include <graphics/matrix4.h>
  7. #include "window-basic-preview.hpp"
  8. #include "window-basic-main.hpp"
  9. #include "obs-app.hpp"
  10. #include "platform.hpp"
  11. #define HANDLE_RADIUS 4.0f
  12. #define HANDLE_SEL_RADIUS (HANDLE_RADIUS * 1.5f)
  13. /* TODO: make C++ math classes and clean up code here later */
  14. OBSBasicPreview::OBSBasicPreview(QWidget *parent, Qt::WindowFlags flags)
  15. : OBSQTDisplay(parent, flags)
  16. {
  17. ResetScrollingOffset();
  18. setMouseTracking(true);
  19. }
  20. OBSBasicPreview::~OBSBasicPreview()
  21. {
  22. obs_enter_graphics();
  23. if (overflow)
  24. gs_texture_destroy(overflow);
  25. if (rectFill)
  26. gs_vertexbuffer_destroy(rectFill);
  27. if (circleFill)
  28. gs_vertexbuffer_destroy(circleFill);
  29. obs_leave_graphics();
  30. }
  31. vec2 OBSBasicPreview::GetMouseEventPos(QMouseEvent *event)
  32. {
  33. OBSBasic *main = reinterpret_cast<OBSBasic *>(App()->GetMainWindow());
  34. float pixelRatio = main->devicePixelRatioF();
  35. float scale = pixelRatio / main->previewScale;
  36. QPoint qtPos = event->pos();
  37. vec2 pos;
  38. vec2_set(&pos, (qtPos.x() - main->previewX / pixelRatio) * scale,
  39. (qtPos.y() - main->previewY / pixelRatio) * scale);
  40. return pos;
  41. }
  42. static void RotatePos(vec2 *pos, float rot)
  43. {
  44. float cosR = cos(rot);
  45. float sinR = sin(rot);
  46. vec2 newPos;
  47. newPos.x = cosR * pos->x - sinR * pos->y;
  48. newPos.y = sinR * pos->x + cosR * pos->y;
  49. vec2_copy(pos, &newPos);
  50. }
  51. struct SceneFindData {
  52. const vec2 &pos;
  53. OBSSceneItem item;
  54. bool selectBelow;
  55. obs_sceneitem_t *group = nullptr;
  56. SceneFindData(const SceneFindData &) = delete;
  57. SceneFindData(SceneFindData &&) = delete;
  58. SceneFindData &operator=(const SceneFindData &) = delete;
  59. SceneFindData &operator=(SceneFindData &&) = delete;
  60. inline SceneFindData(const vec2 &pos_, bool selectBelow_)
  61. : pos(pos_), selectBelow(selectBelow_)
  62. {
  63. }
  64. };
  65. struct SceneFindBoxData {
  66. const vec2 &startPos;
  67. const vec2 &pos;
  68. std::vector<obs_sceneitem_t *> sceneItems;
  69. SceneFindBoxData(const SceneFindData &) = delete;
  70. SceneFindBoxData(SceneFindData &&) = delete;
  71. SceneFindBoxData &operator=(const SceneFindData &) = delete;
  72. SceneFindBoxData &operator=(SceneFindData &&) = delete;
  73. inline SceneFindBoxData(const vec2 &startPos_, const vec2 &pos_)
  74. : startPos(startPos_), pos(pos_)
  75. {
  76. }
  77. };
  78. static bool SceneItemHasVideo(obs_sceneitem_t *item)
  79. {
  80. obs_source_t *source = obs_sceneitem_get_source(item);
  81. uint32_t flags = obs_source_get_output_flags(source);
  82. return (flags & OBS_SOURCE_VIDEO) != 0;
  83. }
  84. static bool CloseFloat(float a, float b, float epsilon = 0.01)
  85. {
  86. using std::abs;
  87. return abs(a - b) <= epsilon;
  88. }
  89. static bool FindItemAtPos(obs_scene_t *scene, obs_sceneitem_t *item,
  90. void *param)
  91. {
  92. SceneFindData *data = reinterpret_cast<SceneFindData *>(param);
  93. matrix4 transform;
  94. matrix4 invTransform;
  95. vec3 transformedPos;
  96. vec3 pos3;
  97. vec3 pos3_;
  98. if (!SceneItemHasVideo(item))
  99. return true;
  100. if (obs_sceneitem_locked(item))
  101. return true;
  102. vec3_set(&pos3, data->pos.x, data->pos.y, 0.0f);
  103. obs_sceneitem_get_box_transform(item, &transform);
  104. matrix4_inv(&invTransform, &transform);
  105. vec3_transform(&transformedPos, &pos3, &invTransform);
  106. vec3_transform(&pos3_, &transformedPos, &transform);
  107. if (CloseFloat(pos3.x, pos3_.x) && CloseFloat(pos3.y, pos3_.y) &&
  108. transformedPos.x >= 0.0f && transformedPos.x <= 1.0f &&
  109. transformedPos.y >= 0.0f && transformedPos.y <= 1.0f) {
  110. if (data->selectBelow && obs_sceneitem_selected(item)) {
  111. if (data->item)
  112. return false;
  113. else
  114. data->selectBelow = false;
  115. }
  116. data->item = item;
  117. }
  118. UNUSED_PARAMETER(scene);
  119. return true;
  120. }
  121. static vec3 GetTransformedPos(float x, float y, const matrix4 &mat)
  122. {
  123. vec3 result;
  124. vec3_set(&result, x, y, 0.0f);
  125. vec3_transform(&result, &result, &mat);
  126. return result;
  127. }
  128. static inline vec2 GetOBSScreenSize()
  129. {
  130. obs_video_info ovi;
  131. vec2 size;
  132. vec2_zero(&size);
  133. if (obs_get_video_info(&ovi)) {
  134. size.x = float(ovi.base_width);
  135. size.y = float(ovi.base_height);
  136. }
  137. return size;
  138. }
  139. vec3 OBSBasicPreview::GetSnapOffset(const vec3 &tl, const vec3 &br)
  140. {
  141. OBSBasic *main = reinterpret_cast<OBSBasic *>(App()->GetMainWindow());
  142. vec2 screenSize = GetOBSScreenSize();
  143. vec3 clampOffset;
  144. vec3_zero(&clampOffset);
  145. const bool snap = config_get_bool(GetGlobalConfig(), "BasicWindow",
  146. "SnappingEnabled");
  147. if (snap == false)
  148. return clampOffset;
  149. const bool screenSnap = config_get_bool(
  150. GetGlobalConfig(), "BasicWindow", "ScreenSnapping");
  151. const bool centerSnap = config_get_bool(
  152. GetGlobalConfig(), "BasicWindow", "CenterSnapping");
  153. const float clampDist = config_get_double(GetGlobalConfig(),
  154. "BasicWindow",
  155. "SnapDistance") /
  156. main->previewScale;
  157. const float centerX = br.x - (br.x - tl.x) / 2.0f;
  158. const float centerY = br.y - (br.y - tl.y) / 2.0f;
  159. // Left screen edge.
  160. if (screenSnap && fabsf(tl.x) < clampDist)
  161. clampOffset.x = -tl.x;
  162. // Right screen edge.
  163. if (screenSnap && fabsf(clampOffset.x) < EPSILON &&
  164. fabsf(screenSize.x - br.x) < clampDist)
  165. clampOffset.x = screenSize.x - br.x;
  166. // Horizontal center.
  167. if (centerSnap && fabsf(screenSize.x - (br.x - tl.x)) > clampDist &&
  168. fabsf(screenSize.x / 2.0f - centerX) < clampDist)
  169. clampOffset.x = screenSize.x / 2.0f - centerX;
  170. // Top screen edge.
  171. if (screenSnap && fabsf(tl.y) < clampDist)
  172. clampOffset.y = -tl.y;
  173. // Bottom screen edge.
  174. if (screenSnap && fabsf(clampOffset.y) < EPSILON &&
  175. fabsf(screenSize.y - br.y) < clampDist)
  176. clampOffset.y = screenSize.y - br.y;
  177. // Vertical center.
  178. if (centerSnap && fabsf(screenSize.y - (br.y - tl.y)) > clampDist &&
  179. fabsf(screenSize.y / 2.0f - centerY) < clampDist)
  180. clampOffset.y = screenSize.y / 2.0f - centerY;
  181. return clampOffset;
  182. }
  183. OBSSceneItem OBSBasicPreview::GetItemAtPos(const vec2 &pos, bool selectBelow)
  184. {
  185. OBSBasic *main = reinterpret_cast<OBSBasic *>(App()->GetMainWindow());
  186. OBSScene scene = main->GetCurrentScene();
  187. if (!scene)
  188. return OBSSceneItem();
  189. SceneFindData data(pos, selectBelow);
  190. obs_scene_enum_items(scene, FindItemAtPos, &data);
  191. return data.item;
  192. }
  193. static bool CheckItemSelected(obs_scene_t *scene, obs_sceneitem_t *item,
  194. void *param)
  195. {
  196. SceneFindData *data = reinterpret_cast<SceneFindData *>(param);
  197. matrix4 transform;
  198. vec3 transformedPos;
  199. vec3 pos3;
  200. if (!SceneItemHasVideo(item))
  201. return true;
  202. if (obs_sceneitem_is_group(item)) {
  203. data->group = item;
  204. obs_sceneitem_group_enum_items(item, CheckItemSelected, param);
  205. data->group = nullptr;
  206. if (data->item) {
  207. return false;
  208. }
  209. }
  210. vec3_set(&pos3, data->pos.x, data->pos.y, 0.0f);
  211. obs_sceneitem_get_box_transform(item, &transform);
  212. if (data->group) {
  213. matrix4 parent_transform;
  214. obs_sceneitem_get_draw_transform(data->group,
  215. &parent_transform);
  216. matrix4_mul(&transform, &transform, &parent_transform);
  217. }
  218. matrix4_inv(&transform, &transform);
  219. vec3_transform(&transformedPos, &pos3, &transform);
  220. if (transformedPos.x >= 0.0f && transformedPos.x <= 1.0f &&
  221. transformedPos.y >= 0.0f && transformedPos.y <= 1.0f) {
  222. if (obs_sceneitem_selected(item)) {
  223. data->item = item;
  224. return false;
  225. }
  226. }
  227. UNUSED_PARAMETER(scene);
  228. return true;
  229. }
  230. bool OBSBasicPreview::SelectedAtPos(const vec2 &pos)
  231. {
  232. OBSBasic *main = reinterpret_cast<OBSBasic *>(App()->GetMainWindow());
  233. OBSScene scene = main->GetCurrentScene();
  234. if (!scene)
  235. return false;
  236. SceneFindData data(pos, false);
  237. obs_scene_enum_items(scene, CheckItemSelected, &data);
  238. return !!data.item;
  239. }
  240. struct HandleFindData {
  241. const vec2 &pos;
  242. const float radius;
  243. matrix4 parent_xform;
  244. OBSSceneItem item;
  245. ItemHandle handle = ItemHandle::None;
  246. float angle = 0.0f;
  247. vec2 rotatePoint;
  248. vec2 offsetPoint;
  249. float angleOffset = 0.0f;
  250. HandleFindData(const HandleFindData &) = delete;
  251. HandleFindData(HandleFindData &&) = delete;
  252. HandleFindData &operator=(const HandleFindData &) = delete;
  253. HandleFindData &operator=(HandleFindData &&) = delete;
  254. inline HandleFindData(const vec2 &pos_, float scale)
  255. : pos(pos_), radius(HANDLE_SEL_RADIUS / scale)
  256. {
  257. matrix4_identity(&parent_xform);
  258. }
  259. inline HandleFindData(const HandleFindData &hfd,
  260. obs_sceneitem_t *parent)
  261. : pos(hfd.pos),
  262. radius(hfd.radius),
  263. item(hfd.item),
  264. handle(hfd.handle),
  265. angle(hfd.angle),
  266. rotatePoint(hfd.rotatePoint),
  267. offsetPoint(hfd.offsetPoint)
  268. {
  269. obs_sceneitem_get_draw_transform(parent, &parent_xform);
  270. }
  271. };
  272. static bool FindHandleAtPos(obs_scene_t *scene, obs_sceneitem_t *item,
  273. void *param)
  274. {
  275. HandleFindData &data = *reinterpret_cast<HandleFindData *>(param);
  276. if (!obs_sceneitem_selected(item)) {
  277. if (obs_sceneitem_is_group(item)) {
  278. HandleFindData newData(data, item);
  279. newData.angleOffset = obs_sceneitem_get_rot(item);
  280. obs_sceneitem_group_enum_items(item, FindHandleAtPos,
  281. &newData);
  282. data.item = newData.item;
  283. data.handle = newData.handle;
  284. data.angle = newData.angle;
  285. data.rotatePoint = newData.rotatePoint;
  286. data.offsetPoint = newData.offsetPoint;
  287. }
  288. return true;
  289. }
  290. matrix4 transform;
  291. vec3 pos3;
  292. float closestHandle = data.radius;
  293. vec3_set(&pos3, data.pos.x, data.pos.y, 0.0f);
  294. obs_sceneitem_get_box_transform(item, &transform);
  295. auto TestHandle = [&](float x, float y, ItemHandle handle) {
  296. vec3 handlePos = GetTransformedPos(x, y, transform);
  297. vec3_transform(&handlePos, &handlePos, &data.parent_xform);
  298. float dist = vec3_dist(&handlePos, &pos3);
  299. if (dist < data.radius) {
  300. if (dist < closestHandle) {
  301. closestHandle = dist;
  302. data.handle = handle;
  303. data.item = item;
  304. }
  305. }
  306. };
  307. TestHandle(0.0f, 0.0f, ItemHandle::TopLeft);
  308. TestHandle(0.5f, 0.0f, ItemHandle::TopCenter);
  309. TestHandle(1.0f, 0.0f, ItemHandle::TopRight);
  310. TestHandle(0.0f, 0.5f, ItemHandle::CenterLeft);
  311. TestHandle(1.0f, 0.5f, ItemHandle::CenterRight);
  312. TestHandle(0.0f, 1.0f, ItemHandle::BottomLeft);
  313. TestHandle(0.5f, 1.0f, ItemHandle::BottomCenter);
  314. TestHandle(1.0f, 1.0f, ItemHandle::BottomRight);
  315. vec2 rotHandleOffset;
  316. vec2_set(&rotHandleOffset, 0.0f,
  317. HANDLE_RADIUS * data.radius * 1.5 - data.radius);
  318. RotatePos(&rotHandleOffset, atan2(transform.x.y, transform.x.x));
  319. RotatePos(&rotHandleOffset, RAD(data.angleOffset));
  320. vec3 handlePos = GetTransformedPos(0.5f, 0.0f, transform);
  321. vec3_transform(&handlePos, &handlePos, &data.parent_xform);
  322. handlePos.x -= rotHandleOffset.x;
  323. handlePos.y -= rotHandleOffset.y;
  324. float dist = vec3_dist(&handlePos, &pos3);
  325. if (dist < data.radius) {
  326. if (dist < closestHandle) {
  327. closestHandle = dist;
  328. data.item = item;
  329. data.angle = obs_sceneitem_get_rot(item);
  330. data.handle = ItemHandle::Rot;
  331. vec2_set(&data.rotatePoint,
  332. transform.t.x + transform.x.x / 2 +
  333. transform.y.x / 2,
  334. transform.t.y + transform.x.y / 2 +
  335. transform.y.y / 2);
  336. obs_sceneitem_get_pos(item, &data.offsetPoint);
  337. data.offsetPoint.x -= data.rotatePoint.x;
  338. data.offsetPoint.y -= data.rotatePoint.y;
  339. RotatePos(&data.offsetPoint,
  340. -RAD(obs_sceneitem_get_rot(item)));
  341. }
  342. }
  343. UNUSED_PARAMETER(scene);
  344. return true;
  345. }
  346. static vec2 GetItemSize(obs_sceneitem_t *item)
  347. {
  348. obs_bounds_type boundsType = obs_sceneitem_get_bounds_type(item);
  349. vec2 size;
  350. if (boundsType != OBS_BOUNDS_NONE) {
  351. obs_sceneitem_get_bounds(item, &size);
  352. } else {
  353. obs_source_t *source = obs_sceneitem_get_source(item);
  354. obs_sceneitem_crop crop;
  355. vec2 scale;
  356. obs_sceneitem_get_scale(item, &scale);
  357. obs_sceneitem_get_crop(item, &crop);
  358. size.x = float(obs_source_get_width(source) - crop.left -
  359. crop.right) *
  360. scale.x;
  361. size.y = float(obs_source_get_height(source) - crop.top -
  362. crop.bottom) *
  363. scale.y;
  364. }
  365. return size;
  366. }
  367. void OBSBasicPreview::GetStretchHandleData(const vec2 &pos, bool ignoreGroup)
  368. {
  369. OBSBasic *main = reinterpret_cast<OBSBasic *>(App()->GetMainWindow());
  370. OBSScene scene = main->GetCurrentScene();
  371. if (!scene)
  372. return;
  373. float scale = main->previewScale / main->devicePixelRatioF();
  374. vec2 scaled_pos = pos;
  375. vec2_divf(&scaled_pos, &scaled_pos, scale);
  376. HandleFindData data(scaled_pos, scale);
  377. obs_scene_enum_items(scene, FindHandleAtPos, &data);
  378. stretchItem = std::move(data.item);
  379. stretchHandle = data.handle;
  380. rotateAngle = data.angle;
  381. rotatePoint = data.rotatePoint;
  382. offsetPoint = data.offsetPoint;
  383. if (stretchHandle != ItemHandle::None) {
  384. matrix4 boxTransform;
  385. vec3 itemUL;
  386. float itemRot;
  387. stretchItemSize = GetItemSize(stretchItem);
  388. obs_sceneitem_get_box_transform(stretchItem, &boxTransform);
  389. itemRot = obs_sceneitem_get_rot(stretchItem);
  390. vec3_from_vec4(&itemUL, &boxTransform.t);
  391. /* build the item space conversion matrices */
  392. matrix4_identity(&itemToScreen);
  393. matrix4_rotate_aa4f(&itemToScreen, &itemToScreen, 0.0f, 0.0f,
  394. 1.0f, RAD(itemRot));
  395. matrix4_translate3f(&itemToScreen, &itemToScreen, itemUL.x,
  396. itemUL.y, 0.0f);
  397. matrix4_identity(&screenToItem);
  398. matrix4_translate3f(&screenToItem, &screenToItem, -itemUL.x,
  399. -itemUL.y, 0.0f);
  400. matrix4_rotate_aa4f(&screenToItem, &screenToItem, 0.0f, 0.0f,
  401. 1.0f, RAD(-itemRot));
  402. obs_sceneitem_get_crop(stretchItem, &startCrop);
  403. obs_sceneitem_get_pos(stretchItem, &startItemPos);
  404. obs_source_t *source = obs_sceneitem_get_source(stretchItem);
  405. cropSize.x = float(obs_source_get_width(source) -
  406. startCrop.left - startCrop.right);
  407. cropSize.y = float(obs_source_get_height(source) -
  408. startCrop.top - startCrop.bottom);
  409. stretchGroup = obs_sceneitem_get_group(scene, stretchItem);
  410. if (stretchGroup && !ignoreGroup) {
  411. obs_sceneitem_get_draw_transform(stretchGroup,
  412. &invGroupTransform);
  413. matrix4_inv(&invGroupTransform, &invGroupTransform);
  414. obs_sceneitem_defer_group_resize_begin(stretchGroup);
  415. }
  416. }
  417. }
  418. void OBSBasicPreview::keyPressEvent(QKeyEvent *event)
  419. {
  420. if (!IsFixedScaling() || event->isAutoRepeat()) {
  421. OBSQTDisplay::keyPressEvent(event);
  422. return;
  423. }
  424. switch (event->key()) {
  425. case Qt::Key_Space:
  426. setCursor(Qt::OpenHandCursor);
  427. scrollMode = true;
  428. break;
  429. }
  430. OBSQTDisplay::keyPressEvent(event);
  431. }
  432. void OBSBasicPreview::keyReleaseEvent(QKeyEvent *event)
  433. {
  434. if (event->isAutoRepeat()) {
  435. OBSQTDisplay::keyReleaseEvent(event);
  436. return;
  437. }
  438. switch (event->key()) {
  439. case Qt::Key_Space:
  440. scrollMode = false;
  441. setCursor(Qt::ArrowCursor);
  442. break;
  443. }
  444. OBSQTDisplay::keyReleaseEvent(event);
  445. }
  446. void OBSBasicPreview::wheelEvent(QWheelEvent *event)
  447. {
  448. if (scrollMode && IsFixedScaling()) {
  449. const int delta = event->angleDelta().y();
  450. if (delta != 0) {
  451. if (delta > 0)
  452. SetScalingLevel(scalingLevel + 1);
  453. else
  454. SetScalingLevel(scalingLevel - 1);
  455. emit DisplayResized();
  456. }
  457. }
  458. OBSQTDisplay::wheelEvent(event);
  459. }
  460. void OBSBasicPreview::mousePressEvent(QMouseEvent *event)
  461. {
  462. #if QT_VERSION >= QT_VERSION_CHECK(6, 0, 0)
  463. QPointF pos = event->position();
  464. #else
  465. QPointF pos = event->localPos();
  466. #endif
  467. if (scrollMode && IsFixedScaling() &&
  468. event->button() == Qt::LeftButton) {
  469. setCursor(Qt::ClosedHandCursor);
  470. scrollingFrom.x = pos.x();
  471. scrollingFrom.y = pos.x();
  472. return;
  473. }
  474. if (event->button() == Qt::RightButton) {
  475. scrollMode = false;
  476. setCursor(Qt::ArrowCursor);
  477. }
  478. if (locked) {
  479. OBSQTDisplay::mousePressEvent(event);
  480. return;
  481. }
  482. OBSBasic *main = reinterpret_cast<OBSBasic *>(App()->GetMainWindow());
  483. float pixelRatio = main->devicePixelRatioF();
  484. float x = pos.x() - main->previewX / pixelRatio;
  485. float y = pos.y() - main->previewY / pixelRatio;
  486. Qt::KeyboardModifiers modifiers = QGuiApplication::keyboardModifiers();
  487. bool altDown = (modifiers & Qt::AltModifier);
  488. bool shiftDown = (modifiers & Qt::ShiftModifier);
  489. bool ctrlDown = (modifiers & Qt::ControlModifier);
  490. OBSQTDisplay::mousePressEvent(event);
  491. if (event->button() != Qt::LeftButton &&
  492. event->button() != Qt::RightButton)
  493. return;
  494. if (event->button() == Qt::LeftButton)
  495. mouseDown = true;
  496. {
  497. std::lock_guard<std::mutex> lock(selectMutex);
  498. selectedItems.clear();
  499. }
  500. if (altDown)
  501. cropping = true;
  502. if (altDown || shiftDown || ctrlDown) {
  503. vec2 s;
  504. SceneFindBoxData data(s, s);
  505. obs_scene_enum_items(main->GetCurrentScene(), FindSelected,
  506. &data);
  507. std::lock_guard<std::mutex> lock(selectMutex);
  508. selectedItems = data.sceneItems;
  509. }
  510. vec2_set(&startPos, x, y);
  511. GetStretchHandleData(startPos, false);
  512. vec2_divf(&startPos, &startPos, main->previewScale / pixelRatio);
  513. startPos.x = std::round(startPos.x);
  514. startPos.y = std::round(startPos.y);
  515. mouseOverItems = SelectedAtPos(startPos);
  516. vec2_zero(&lastMoveOffset);
  517. mousePos = startPos;
  518. wrapper =
  519. obs_scene_save_transform_states(main->GetCurrentScene(), true);
  520. changed = false;
  521. }
  522. void OBSBasicPreview::UpdateCursor(uint32_t &flags)
  523. {
  524. if (obs_sceneitem_locked(stretchItem)) {
  525. unsetCursor();
  526. return;
  527. }
  528. if (!flags && (cursor().shape() != Qt::OpenHandCursor || !scrollMode))
  529. unsetCursor();
  530. if (cursor().shape() != Qt::ArrowCursor)
  531. return;
  532. if ((flags & ITEM_LEFT && flags & ITEM_TOP) ||
  533. (flags & ITEM_RIGHT && flags & ITEM_BOTTOM))
  534. setCursor(Qt::SizeFDiagCursor);
  535. else if ((flags & ITEM_LEFT && flags & ITEM_BOTTOM) ||
  536. (flags & ITEM_RIGHT && flags & ITEM_TOP))
  537. setCursor(Qt::SizeBDiagCursor);
  538. else if (flags & ITEM_LEFT || flags & ITEM_RIGHT)
  539. setCursor(Qt::SizeHorCursor);
  540. else if (flags & ITEM_TOP || flags & ITEM_BOTTOM)
  541. setCursor(Qt::SizeVerCursor);
  542. else if (flags & ITEM_ROT)
  543. setCursor(Qt::OpenHandCursor);
  544. }
  545. static bool select_one(obs_scene_t *scene, obs_sceneitem_t *item, void *param)
  546. {
  547. obs_sceneitem_t *selectedItem =
  548. reinterpret_cast<obs_sceneitem_t *>(param);
  549. if (obs_sceneitem_is_group(item))
  550. obs_sceneitem_group_enum_items(item, select_one, param);
  551. obs_sceneitem_select(item, (selectedItem == item));
  552. UNUSED_PARAMETER(scene);
  553. return true;
  554. }
  555. void OBSBasicPreview::DoSelect(const vec2 &pos)
  556. {
  557. OBSBasic *main = reinterpret_cast<OBSBasic *>(App()->GetMainWindow());
  558. OBSScene scene = main->GetCurrentScene();
  559. OBSSceneItem item = GetItemAtPos(pos, true);
  560. obs_scene_enum_items(scene, select_one, (obs_sceneitem_t *)item);
  561. }
  562. void OBSBasicPreview::DoCtrlSelect(const vec2 &pos)
  563. {
  564. OBSSceneItem item = GetItemAtPos(pos, false);
  565. if (!item)
  566. return;
  567. bool selected = obs_sceneitem_selected(item);
  568. obs_sceneitem_select(item, !selected);
  569. }
  570. void OBSBasicPreview::ProcessClick(const vec2 &pos)
  571. {
  572. Qt::KeyboardModifiers modifiers = QGuiApplication::keyboardModifiers();
  573. if (modifiers & Qt::ControlModifier)
  574. DoCtrlSelect(pos);
  575. else
  576. DoSelect(pos);
  577. }
  578. void OBSBasicPreview::mouseReleaseEvent(QMouseEvent *event)
  579. {
  580. if (scrollMode)
  581. setCursor(Qt::OpenHandCursor);
  582. if (locked) {
  583. OBSQTDisplay::mouseReleaseEvent(event);
  584. return;
  585. }
  586. if (mouseDown) {
  587. vec2 pos = GetMouseEventPos(event);
  588. if (!mouseMoved)
  589. ProcessClick(pos);
  590. if (selectionBox) {
  591. Qt::KeyboardModifiers modifiers =
  592. QGuiApplication::keyboardModifiers();
  593. bool altDown = modifiers & Qt::AltModifier;
  594. bool shiftDown = modifiers & Qt::ShiftModifier;
  595. bool ctrlDown = modifiers & Qt::ControlModifier;
  596. std::lock_guard<std::mutex> lock(selectMutex);
  597. if (altDown || ctrlDown || shiftDown) {
  598. for (size_t i = 0; i < selectedItems.size();
  599. i++) {
  600. obs_sceneitem_select(selectedItems[i],
  601. true);
  602. }
  603. }
  604. for (size_t i = 0; i < hoveredPreviewItems.size();
  605. i++) {
  606. bool select = true;
  607. obs_sceneitem_t *item = hoveredPreviewItems[i];
  608. if (altDown) {
  609. select = false;
  610. } else if (ctrlDown) {
  611. select = !obs_sceneitem_selected(item);
  612. }
  613. obs_sceneitem_select(hoveredPreviewItems[i],
  614. select);
  615. }
  616. }
  617. if (stretchGroup) {
  618. obs_sceneitem_defer_group_resize_end(stretchGroup);
  619. }
  620. stretchItem = nullptr;
  621. stretchGroup = nullptr;
  622. mouseDown = false;
  623. mouseMoved = false;
  624. cropping = false;
  625. selectionBox = false;
  626. unsetCursor();
  627. OBSSceneItem item = GetItemAtPos(pos, true);
  628. std::lock_guard<std::mutex> lock(selectMutex);
  629. hoveredPreviewItems.clear();
  630. hoveredPreviewItems.push_back(item);
  631. selectedItems.clear();
  632. }
  633. OBSBasic *main = reinterpret_cast<OBSBasic *>(App()->GetMainWindow());
  634. OBSDataAutoRelease rwrapper =
  635. obs_scene_save_transform_states(main->GetCurrentScene(), true);
  636. auto undo_redo = [](const std::string &data) {
  637. OBSDataAutoRelease dat =
  638. obs_data_create_from_json(data.c_str());
  639. OBSSourceAutoRelease source = obs_get_source_by_name(
  640. obs_data_get_string(dat, "scene_name"));
  641. reinterpret_cast<OBSBasic *>(App()->GetMainWindow())
  642. ->SetCurrentScene(source.Get(), true);
  643. obs_scene_load_transform_states(data.c_str());
  644. };
  645. if (wrapper && rwrapper) {
  646. std::string undo_data(obs_data_get_json(wrapper));
  647. std::string redo_data(obs_data_get_json(rwrapper));
  648. if (changed && undo_data.compare(redo_data) != 0)
  649. main->undo_s.add_action(
  650. QTStr("Undo.Transform")
  651. .arg(obs_source_get_name(
  652. main->GetCurrentSceneSource())),
  653. undo_redo, undo_redo, undo_data, redo_data);
  654. }
  655. wrapper = nullptr;
  656. }
  657. struct SelectedItemBounds {
  658. bool first = true;
  659. vec3 tl, br;
  660. };
  661. static bool AddItemBounds(obs_scene_t *scene, obs_sceneitem_t *item,
  662. void *param)
  663. {
  664. SelectedItemBounds *data =
  665. reinterpret_cast<SelectedItemBounds *>(param);
  666. vec3 t[4];
  667. auto add_bounds = [data, &t]() {
  668. for (const vec3 &v : t) {
  669. if (data->first) {
  670. vec3_copy(&data->tl, &v);
  671. vec3_copy(&data->br, &v);
  672. data->first = false;
  673. } else {
  674. vec3_min(&data->tl, &data->tl, &v);
  675. vec3_max(&data->br, &data->br, &v);
  676. }
  677. }
  678. };
  679. if (obs_sceneitem_is_group(item)) {
  680. SelectedItemBounds sib;
  681. obs_sceneitem_group_enum_items(item, AddItemBounds, &sib);
  682. if (!sib.first) {
  683. matrix4 xform;
  684. obs_sceneitem_get_draw_transform(item, &xform);
  685. vec3_set(&t[0], sib.tl.x, sib.tl.y, 0.0f);
  686. vec3_set(&t[1], sib.tl.x, sib.br.y, 0.0f);
  687. vec3_set(&t[2], sib.br.x, sib.tl.y, 0.0f);
  688. vec3_set(&t[3], sib.br.x, sib.br.y, 0.0f);
  689. vec3_transform(&t[0], &t[0], &xform);
  690. vec3_transform(&t[1], &t[1], &xform);
  691. vec3_transform(&t[2], &t[2], &xform);
  692. vec3_transform(&t[3], &t[3], &xform);
  693. add_bounds();
  694. }
  695. }
  696. if (!obs_sceneitem_selected(item))
  697. return true;
  698. matrix4 boxTransform;
  699. obs_sceneitem_get_box_transform(item, &boxTransform);
  700. t[0] = GetTransformedPos(0.0f, 0.0f, boxTransform);
  701. t[1] = GetTransformedPos(1.0f, 0.0f, boxTransform);
  702. t[2] = GetTransformedPos(0.0f, 1.0f, boxTransform);
  703. t[3] = GetTransformedPos(1.0f, 1.0f, boxTransform);
  704. add_bounds();
  705. UNUSED_PARAMETER(scene);
  706. return true;
  707. }
  708. struct OffsetData {
  709. float clampDist;
  710. vec3 tl, br, offset;
  711. };
  712. static bool GetSourceSnapOffset(obs_scene_t *scene, obs_sceneitem_t *item,
  713. void *param)
  714. {
  715. OffsetData *data = reinterpret_cast<OffsetData *>(param);
  716. if (obs_sceneitem_selected(item))
  717. return true;
  718. matrix4 boxTransform;
  719. obs_sceneitem_get_box_transform(item, &boxTransform);
  720. vec3 t[4] = {GetTransformedPos(0.0f, 0.0f, boxTransform),
  721. GetTransformedPos(1.0f, 0.0f, boxTransform),
  722. GetTransformedPos(0.0f, 1.0f, boxTransform),
  723. GetTransformedPos(1.0f, 1.0f, boxTransform)};
  724. bool first = true;
  725. vec3 tl, br;
  726. vec3_zero(&tl);
  727. vec3_zero(&br);
  728. for (const vec3 &v : t) {
  729. if (first) {
  730. vec3_copy(&tl, &v);
  731. vec3_copy(&br, &v);
  732. first = false;
  733. } else {
  734. vec3_min(&tl, &tl, &v);
  735. vec3_max(&br, &br, &v);
  736. }
  737. }
  738. // Snap to other source edges
  739. #define EDGE_SNAP(l, r, x, y) \
  740. do { \
  741. double dist = fabsf(l.x - data->r.x); \
  742. if (dist < data->clampDist && \
  743. fabsf(data->offset.x) < EPSILON && data->tl.y < br.y && \
  744. data->br.y > tl.y && \
  745. (fabsf(data->offset.x) > dist || \
  746. data->offset.x < EPSILON)) \
  747. data->offset.x = l.x - data->r.x; \
  748. } while (false)
  749. EDGE_SNAP(tl, br, x, y);
  750. EDGE_SNAP(tl, br, y, x);
  751. EDGE_SNAP(br, tl, x, y);
  752. EDGE_SNAP(br, tl, y, x);
  753. #undef EDGE_SNAP
  754. UNUSED_PARAMETER(scene);
  755. return true;
  756. }
  757. void OBSBasicPreview::SnapItemMovement(vec2 &offset)
  758. {
  759. OBSBasic *main = reinterpret_cast<OBSBasic *>(App()->GetMainWindow());
  760. OBSScene scene = main->GetCurrentScene();
  761. SelectedItemBounds data;
  762. obs_scene_enum_items(scene, AddItemBounds, &data);
  763. data.tl.x += offset.x;
  764. data.tl.y += offset.y;
  765. data.br.x += offset.x;
  766. data.br.y += offset.y;
  767. vec3 snapOffset = GetSnapOffset(data.tl, data.br);
  768. const bool snap = config_get_bool(GetGlobalConfig(), "BasicWindow",
  769. "SnappingEnabled");
  770. const bool sourcesSnap = config_get_bool(
  771. GetGlobalConfig(), "BasicWindow", "SourceSnapping");
  772. if (snap == false)
  773. return;
  774. if (sourcesSnap == false) {
  775. offset.x += snapOffset.x;
  776. offset.y += snapOffset.y;
  777. return;
  778. }
  779. const float clampDist = config_get_double(GetGlobalConfig(),
  780. "BasicWindow",
  781. "SnapDistance") /
  782. main->previewScale;
  783. OffsetData offsetData;
  784. offsetData.clampDist = clampDist;
  785. offsetData.tl = data.tl;
  786. offsetData.br = data.br;
  787. vec3_copy(&offsetData.offset, &snapOffset);
  788. obs_scene_enum_items(scene, GetSourceSnapOffset, &offsetData);
  789. if (fabsf(offsetData.offset.x) > EPSILON ||
  790. fabsf(offsetData.offset.y) > EPSILON) {
  791. offset.x += offsetData.offset.x;
  792. offset.y += offsetData.offset.y;
  793. } else {
  794. offset.x += snapOffset.x;
  795. offset.y += snapOffset.y;
  796. }
  797. }
  798. static bool move_items(obs_scene_t *scene, obs_sceneitem_t *item, void *param)
  799. {
  800. if (obs_sceneitem_locked(item))
  801. return true;
  802. bool selected = obs_sceneitem_selected(item);
  803. vec2 *offset = reinterpret_cast<vec2 *>(param);
  804. if (obs_sceneitem_is_group(item) && !selected) {
  805. matrix4 transform;
  806. vec3 new_offset;
  807. vec3_set(&new_offset, offset->x, offset->y, 0.0f);
  808. obs_sceneitem_get_draw_transform(item, &transform);
  809. vec4_set(&transform.t, 0.0f, 0.0f, 0.0f, 1.0f);
  810. matrix4_inv(&transform, &transform);
  811. vec3_transform(&new_offset, &new_offset, &transform);
  812. obs_sceneitem_group_enum_items(item, move_items, &new_offset);
  813. }
  814. if (selected) {
  815. vec2 pos;
  816. obs_sceneitem_get_pos(item, &pos);
  817. vec2_add(&pos, &pos, offset);
  818. obs_sceneitem_set_pos(item, &pos);
  819. }
  820. UNUSED_PARAMETER(scene);
  821. return true;
  822. }
  823. void OBSBasicPreview::MoveItems(const vec2 &pos)
  824. {
  825. Qt::KeyboardModifiers modifiers = QGuiApplication::keyboardModifiers();
  826. OBSBasic *main = reinterpret_cast<OBSBasic *>(App()->GetMainWindow());
  827. OBSScene scene = main->GetCurrentScene();
  828. vec2 offset, moveOffset;
  829. vec2_sub(&offset, &pos, &startPos);
  830. vec2_sub(&moveOffset, &offset, &lastMoveOffset);
  831. if (!(modifiers & Qt::ControlModifier))
  832. SnapItemMovement(moveOffset);
  833. vec2_add(&lastMoveOffset, &lastMoveOffset, &moveOffset);
  834. obs_scene_enum_items(scene, move_items, &moveOffset);
  835. }
  836. static bool CounterClockwise(float x1, float x2, float x3, float y1, float y2,
  837. float y3)
  838. {
  839. return (y3 - y1) * (x2 - x1) > (y2 - y1) * (x3 - x1);
  840. }
  841. static bool IntersectLine(float x1, float x2, float x3, float x4, float y1,
  842. float y2, float y3, float y4)
  843. {
  844. bool a = CounterClockwise(x1, x2, x3, y1, y2, y3);
  845. bool b = CounterClockwise(x1, x2, x4, y1, y2, y4);
  846. bool c = CounterClockwise(x3, x4, x1, y3, y4, y1);
  847. bool d = CounterClockwise(x3, x4, x2, y3, y4, y2);
  848. return (a != b) && (c != d);
  849. }
  850. static bool IntersectBox(matrix4 transform, float x1, float x2, float y1,
  851. float y2)
  852. {
  853. float x3, x4, y3, y4;
  854. x3 = transform.t.x;
  855. y3 = transform.t.y;
  856. x4 = x3 + transform.x.x;
  857. y4 = y3 + transform.x.y;
  858. if (IntersectLine(x1, x1, x3, x4, y1, y2, y3, y4) ||
  859. IntersectLine(x1, x2, x3, x4, y1, y1, y3, y4) ||
  860. IntersectLine(x2, x2, x3, x4, y1, y2, y3, y4) ||
  861. IntersectLine(x1, x2, x3, x4, y2, y2, y3, y4))
  862. return true;
  863. x4 = x3 + transform.y.x;
  864. y4 = y3 + transform.y.y;
  865. if (IntersectLine(x1, x1, x3, x4, y1, y2, y3, y4) ||
  866. IntersectLine(x1, x2, x3, x4, y1, y1, y3, y4) ||
  867. IntersectLine(x2, x2, x3, x4, y1, y2, y3, y4) ||
  868. IntersectLine(x1, x2, x3, x4, y2, y2, y3, y4))
  869. return true;
  870. x3 = transform.t.x + transform.x.x;
  871. y3 = transform.t.y + transform.x.y;
  872. x4 = x3 + transform.y.x;
  873. y4 = y3 + transform.y.y;
  874. if (IntersectLine(x1, x1, x3, x4, y1, y2, y3, y4) ||
  875. IntersectLine(x1, x2, x3, x4, y1, y1, y3, y4) ||
  876. IntersectLine(x2, x2, x3, x4, y1, y2, y3, y4) ||
  877. IntersectLine(x1, x2, x3, x4, y2, y2, y3, y4))
  878. return true;
  879. x3 = transform.t.x + transform.y.x;
  880. y3 = transform.t.y + transform.y.y;
  881. x4 = x3 + transform.x.x;
  882. y4 = y3 + transform.x.y;
  883. if (IntersectLine(x1, x1, x3, x4, y1, y2, y3, y4) ||
  884. IntersectLine(x1, x2, x3, x4, y1, y1, y3, y4) ||
  885. IntersectLine(x2, x2, x3, x4, y1, y2, y3, y4) ||
  886. IntersectLine(x1, x2, x3, x4, y2, y2, y3, y4))
  887. return true;
  888. return false;
  889. }
  890. #undef PI
  891. bool OBSBasicPreview::FindSelected(obs_scene_t *scene, obs_sceneitem_t *item,
  892. void *param)
  893. {
  894. SceneFindBoxData *data = reinterpret_cast<SceneFindBoxData *>(param);
  895. if (obs_sceneitem_selected(item))
  896. data->sceneItems.push_back(item);
  897. UNUSED_PARAMETER(scene);
  898. return true;
  899. }
  900. static bool FindItemsInBox(obs_scene_t *scene, obs_sceneitem_t *item,
  901. void *param)
  902. {
  903. SceneFindBoxData *data = reinterpret_cast<SceneFindBoxData *>(param);
  904. matrix4 transform;
  905. matrix4 invTransform;
  906. vec3 transformedPos;
  907. vec3 pos3;
  908. vec3 pos3_;
  909. vec2 pos_min, pos_max;
  910. vec2_min(&pos_min, &data->startPos, &data->pos);
  911. vec2_max(&pos_max, &data->startPos, &data->pos);
  912. const float x1 = pos_min.x;
  913. const float x2 = pos_max.x;
  914. const float y1 = pos_min.y;
  915. const float y2 = pos_max.y;
  916. if (!SceneItemHasVideo(item))
  917. return true;
  918. if (obs_sceneitem_locked(item))
  919. return true;
  920. if (!obs_sceneitem_visible(item))
  921. return true;
  922. vec3_set(&pos3, data->pos.x, data->pos.y, 0.0f);
  923. obs_sceneitem_get_box_transform(item, &transform);
  924. matrix4_inv(&invTransform, &transform);
  925. vec3_transform(&transformedPos, &pos3, &invTransform);
  926. vec3_transform(&pos3_, &transformedPos, &transform);
  927. if (CloseFloat(pos3.x, pos3_.x) && CloseFloat(pos3.y, pos3_.y) &&
  928. transformedPos.x >= 0.0f && transformedPos.x <= 1.0f &&
  929. transformedPos.y >= 0.0f && transformedPos.y <= 1.0f) {
  930. data->sceneItems.push_back(item);
  931. return true;
  932. }
  933. if (transform.t.x > x1 && transform.t.x < x2 && transform.t.y > y1 &&
  934. transform.t.y < y2) {
  935. data->sceneItems.push_back(item);
  936. return true;
  937. }
  938. if (transform.t.x + transform.x.x > x1 &&
  939. transform.t.x + transform.x.x < x2 &&
  940. transform.t.y + transform.x.y > y1 &&
  941. transform.t.y + transform.x.y < y2) {
  942. data->sceneItems.push_back(item);
  943. return true;
  944. }
  945. if (transform.t.x + transform.y.x > x1 &&
  946. transform.t.x + transform.y.x < x2 &&
  947. transform.t.y + transform.y.y > y1 &&
  948. transform.t.y + transform.y.y < y2) {
  949. data->sceneItems.push_back(item);
  950. return true;
  951. }
  952. if (transform.t.x + transform.x.x + transform.y.x > x1 &&
  953. transform.t.x + transform.x.x + transform.y.x < x2 &&
  954. transform.t.y + transform.x.y + transform.y.y > y1 &&
  955. transform.t.y + transform.x.y + transform.y.y < y2) {
  956. data->sceneItems.push_back(item);
  957. return true;
  958. }
  959. if (transform.t.x + 0.5 * (transform.x.x + transform.y.x) > x1 &&
  960. transform.t.x + 0.5 * (transform.x.x + transform.y.x) < x2 &&
  961. transform.t.y + 0.5 * (transform.x.y + transform.y.y) > y1 &&
  962. transform.t.y + 0.5 * (transform.x.y + transform.y.y) < y2) {
  963. data->sceneItems.push_back(item);
  964. return true;
  965. }
  966. if (IntersectBox(transform, x1, x2, y1, y2)) {
  967. data->sceneItems.push_back(item);
  968. return true;
  969. }
  970. UNUSED_PARAMETER(scene);
  971. return true;
  972. }
  973. void OBSBasicPreview::BoxItems(const vec2 &startPos, const vec2 &pos)
  974. {
  975. OBSBasic *main = reinterpret_cast<OBSBasic *>(App()->GetMainWindow());
  976. OBSScene scene = main->GetCurrentScene();
  977. if (!scene)
  978. return;
  979. if (cursor().shape() != Qt::CrossCursor)
  980. setCursor(Qt::CrossCursor);
  981. SceneFindBoxData data(startPos, pos);
  982. obs_scene_enum_items(scene, FindItemsInBox, &data);
  983. std::lock_guard<std::mutex> lock(selectMutex);
  984. hoveredPreviewItems = data.sceneItems;
  985. }
  986. vec3 OBSBasicPreview::CalculateStretchPos(const vec3 &tl, const vec3 &br)
  987. {
  988. uint32_t alignment = obs_sceneitem_get_alignment(stretchItem);
  989. vec3 pos;
  990. vec3_zero(&pos);
  991. if (alignment & OBS_ALIGN_LEFT)
  992. pos.x = tl.x;
  993. else if (alignment & OBS_ALIGN_RIGHT)
  994. pos.x = br.x;
  995. else
  996. pos.x = (br.x - tl.x) * 0.5f + tl.x;
  997. if (alignment & OBS_ALIGN_TOP)
  998. pos.y = tl.y;
  999. else if (alignment & OBS_ALIGN_BOTTOM)
  1000. pos.y = br.y;
  1001. else
  1002. pos.y = (br.y - tl.y) * 0.5f + tl.y;
  1003. return pos;
  1004. }
  1005. void OBSBasicPreview::ClampAspect(vec3 &tl, vec3 &br, vec2 &size,
  1006. const vec2 &baseSize)
  1007. {
  1008. float baseAspect = baseSize.x / baseSize.y;
  1009. float aspect = size.x / size.y;
  1010. uint32_t stretchFlags = (uint32_t)stretchHandle;
  1011. if (stretchHandle == ItemHandle::TopLeft ||
  1012. stretchHandle == ItemHandle::TopRight ||
  1013. stretchHandle == ItemHandle::BottomLeft ||
  1014. stretchHandle == ItemHandle::BottomRight) {
  1015. if (aspect < baseAspect) {
  1016. if ((size.y >= 0.0f && size.x >= 0.0f) ||
  1017. (size.y <= 0.0f && size.x <= 0.0f))
  1018. size.x = size.y * baseAspect;
  1019. else
  1020. size.x = size.y * baseAspect * -1.0f;
  1021. } else {
  1022. if ((size.y >= 0.0f && size.x >= 0.0f) ||
  1023. (size.y <= 0.0f && size.x <= 0.0f))
  1024. size.y = size.x / baseAspect;
  1025. else
  1026. size.y = size.x / baseAspect * -1.0f;
  1027. }
  1028. } else if (stretchHandle == ItemHandle::TopCenter ||
  1029. stretchHandle == ItemHandle::BottomCenter) {
  1030. if ((size.y >= 0.0f && size.x >= 0.0f) ||
  1031. (size.y <= 0.0f && size.x <= 0.0f))
  1032. size.x = size.y * baseAspect;
  1033. else
  1034. size.x = size.y * baseAspect * -1.0f;
  1035. } else if (stretchHandle == ItemHandle::CenterLeft ||
  1036. stretchHandle == ItemHandle::CenterRight) {
  1037. if ((size.y >= 0.0f && size.x >= 0.0f) ||
  1038. (size.y <= 0.0f && size.x <= 0.0f))
  1039. size.y = size.x / baseAspect;
  1040. else
  1041. size.y = size.x / baseAspect * -1.0f;
  1042. }
  1043. size.x = std::round(size.x);
  1044. size.y = std::round(size.y);
  1045. if (stretchFlags & ITEM_LEFT)
  1046. tl.x = br.x - size.x;
  1047. else if (stretchFlags & ITEM_RIGHT)
  1048. br.x = tl.x + size.x;
  1049. if (stretchFlags & ITEM_TOP)
  1050. tl.y = br.y - size.y;
  1051. else if (stretchFlags & ITEM_BOTTOM)
  1052. br.y = tl.y + size.y;
  1053. }
  1054. void OBSBasicPreview::SnapStretchingToScreen(vec3 &tl, vec3 &br)
  1055. {
  1056. uint32_t stretchFlags = (uint32_t)stretchHandle;
  1057. vec3 newTL = GetTransformedPos(tl.x, tl.y, itemToScreen);
  1058. vec3 newTR = GetTransformedPos(br.x, tl.y, itemToScreen);
  1059. vec3 newBL = GetTransformedPos(tl.x, br.y, itemToScreen);
  1060. vec3 newBR = GetTransformedPos(br.x, br.y, itemToScreen);
  1061. vec3 boundingTL;
  1062. vec3 boundingBR;
  1063. vec3_copy(&boundingTL, &newTL);
  1064. vec3_min(&boundingTL, &boundingTL, &newTR);
  1065. vec3_min(&boundingTL, &boundingTL, &newBL);
  1066. vec3_min(&boundingTL, &boundingTL, &newBR);
  1067. vec3_copy(&boundingBR, &newTL);
  1068. vec3_max(&boundingBR, &boundingBR, &newTR);
  1069. vec3_max(&boundingBR, &boundingBR, &newBL);
  1070. vec3_max(&boundingBR, &boundingBR, &newBR);
  1071. vec3 offset = GetSnapOffset(boundingTL, boundingBR);
  1072. vec3_add(&offset, &offset, &newTL);
  1073. vec3_transform(&offset, &offset, &screenToItem);
  1074. vec3_sub(&offset, &offset, &tl);
  1075. if (stretchFlags & ITEM_LEFT)
  1076. tl.x += offset.x;
  1077. else if (stretchFlags & ITEM_RIGHT)
  1078. br.x += offset.x;
  1079. if (stretchFlags & ITEM_TOP)
  1080. tl.y += offset.y;
  1081. else if (stretchFlags & ITEM_BOTTOM)
  1082. br.y += offset.y;
  1083. }
  1084. static float maxfunc(float x, float y)
  1085. {
  1086. return x > y ? x : y;
  1087. }
  1088. static float minfunc(float x, float y)
  1089. {
  1090. return x < y ? x : y;
  1091. }
  1092. void OBSBasicPreview::CropItem(const vec2 &pos)
  1093. {
  1094. obs_bounds_type boundsType = obs_sceneitem_get_bounds_type(stretchItem);
  1095. uint32_t stretchFlags = (uint32_t)stretchHandle;
  1096. uint32_t align = obs_sceneitem_get_alignment(stretchItem);
  1097. vec3 tl, br, pos3;
  1098. vec3_zero(&tl);
  1099. vec3_set(&br, stretchItemSize.x, stretchItemSize.y, 0.0f);
  1100. vec3_set(&pos3, pos.x, pos.y, 0.0f);
  1101. vec3_transform(&pos3, &pos3, &screenToItem);
  1102. obs_sceneitem_crop crop = startCrop;
  1103. vec2 scale;
  1104. obs_sceneitem_get_scale(stretchItem, &scale);
  1105. vec2 max_tl;
  1106. vec2 max_br;
  1107. vec2_set(&max_tl, float(-crop.left) * scale.x,
  1108. float(-crop.top) * scale.y);
  1109. vec2_set(&max_br, stretchItemSize.x + crop.right * scale.x,
  1110. stretchItemSize.y + crop.bottom * scale.y);
  1111. typedef std::function<float(float, float)> minmax_func_t;
  1112. minmax_func_t min_x = scale.x < 0.0f ? maxfunc : minfunc;
  1113. minmax_func_t min_y = scale.y < 0.0f ? maxfunc : minfunc;
  1114. minmax_func_t max_x = scale.x < 0.0f ? minfunc : maxfunc;
  1115. minmax_func_t max_y = scale.y < 0.0f ? minfunc : maxfunc;
  1116. pos3.x = min_x(pos3.x, max_br.x);
  1117. pos3.x = max_x(pos3.x, max_tl.x);
  1118. pos3.y = min_y(pos3.y, max_br.y);
  1119. pos3.y = max_y(pos3.y, max_tl.y);
  1120. if (stretchFlags & ITEM_LEFT) {
  1121. float maxX = stretchItemSize.x - (2.0 * scale.x);
  1122. pos3.x = tl.x = min_x(pos3.x, maxX);
  1123. } else if (stretchFlags & ITEM_RIGHT) {
  1124. float minX = (2.0 * scale.x);
  1125. pos3.x = br.x = max_x(pos3.x, minX);
  1126. }
  1127. if (stretchFlags & ITEM_TOP) {
  1128. float maxY = stretchItemSize.y - (2.0 * scale.y);
  1129. pos3.y = tl.y = min_y(pos3.y, maxY);
  1130. } else if (stretchFlags & ITEM_BOTTOM) {
  1131. float minY = (2.0 * scale.y);
  1132. pos3.y = br.y = max_y(pos3.y, minY);
  1133. }
  1134. #define ALIGN_X (ITEM_LEFT | ITEM_RIGHT)
  1135. #define ALIGN_Y (ITEM_TOP | ITEM_BOTTOM)
  1136. vec3 newPos;
  1137. vec3_zero(&newPos);
  1138. uint32_t align_x = (align & ALIGN_X);
  1139. uint32_t align_y = (align & ALIGN_Y);
  1140. if (align_x == (stretchFlags & ALIGN_X) && align_x != 0)
  1141. newPos.x = pos3.x;
  1142. else if (align & ITEM_RIGHT)
  1143. newPos.x = stretchItemSize.x;
  1144. else if (!(align & ITEM_LEFT))
  1145. newPos.x = stretchItemSize.x * 0.5f;
  1146. if (align_y == (stretchFlags & ALIGN_Y) && align_y != 0)
  1147. newPos.y = pos3.y;
  1148. else if (align & ITEM_BOTTOM)
  1149. newPos.y = stretchItemSize.y;
  1150. else if (!(align & ITEM_TOP))
  1151. newPos.y = stretchItemSize.y * 0.5f;
  1152. #undef ALIGN_X
  1153. #undef ALIGN_Y
  1154. crop = startCrop;
  1155. if (stretchFlags & ITEM_LEFT)
  1156. crop.left += int(std::round(tl.x / scale.x));
  1157. else if (stretchFlags & ITEM_RIGHT)
  1158. crop.right +=
  1159. int(std::round((stretchItemSize.x - br.x) / scale.x));
  1160. if (stretchFlags & ITEM_TOP)
  1161. crop.top += int(std::round(tl.y / scale.y));
  1162. else if (stretchFlags & ITEM_BOTTOM)
  1163. crop.bottom +=
  1164. int(std::round((stretchItemSize.y - br.y) / scale.y));
  1165. vec3_transform(&newPos, &newPos, &itemToScreen);
  1166. newPos.x = std::round(newPos.x);
  1167. newPos.y = std::round(newPos.y);
  1168. #if 0
  1169. vec3 curPos;
  1170. vec3_zero(&curPos);
  1171. obs_sceneitem_get_pos(stretchItem, (vec2*)&curPos);
  1172. blog(LOG_DEBUG, "curPos {%d, %d} - newPos {%d, %d}",
  1173. int(curPos.x), int(curPos.y),
  1174. int(newPos.x), int(newPos.y));
  1175. blog(LOG_DEBUG, "crop {%d, %d, %d, %d}",
  1176. crop.left, crop.top,
  1177. crop.right, crop.bottom);
  1178. #endif
  1179. obs_sceneitem_defer_update_begin(stretchItem);
  1180. obs_sceneitem_set_crop(stretchItem, &crop);
  1181. if (boundsType == OBS_BOUNDS_NONE)
  1182. obs_sceneitem_set_pos(stretchItem, (vec2 *)&newPos);
  1183. obs_sceneitem_defer_update_end(stretchItem);
  1184. }
  1185. void OBSBasicPreview::StretchItem(const vec2 &pos)
  1186. {
  1187. Qt::KeyboardModifiers modifiers = QGuiApplication::keyboardModifiers();
  1188. obs_bounds_type boundsType = obs_sceneitem_get_bounds_type(stretchItem);
  1189. uint32_t stretchFlags = (uint32_t)stretchHandle;
  1190. bool shiftDown = (modifiers & Qt::ShiftModifier);
  1191. vec3 tl, br, pos3;
  1192. vec3_zero(&tl);
  1193. vec3_set(&br, stretchItemSize.x, stretchItemSize.y, 0.0f);
  1194. vec3_set(&pos3, pos.x, pos.y, 0.0f);
  1195. vec3_transform(&pos3, &pos3, &screenToItem);
  1196. if (stretchFlags & ITEM_LEFT)
  1197. tl.x = pos3.x;
  1198. else if (stretchFlags & ITEM_RIGHT)
  1199. br.x = pos3.x;
  1200. if (stretchFlags & ITEM_TOP)
  1201. tl.y = pos3.y;
  1202. else if (stretchFlags & ITEM_BOTTOM)
  1203. br.y = pos3.y;
  1204. if (!(modifiers & Qt::ControlModifier))
  1205. SnapStretchingToScreen(tl, br);
  1206. obs_source_t *source = obs_sceneitem_get_source(stretchItem);
  1207. vec2 baseSize;
  1208. vec2_set(&baseSize, float(obs_source_get_width(source)),
  1209. float(obs_source_get_height(source)));
  1210. vec2 size;
  1211. vec2_set(&size, br.x - tl.x, br.y - tl.y);
  1212. if (boundsType != OBS_BOUNDS_NONE) {
  1213. if (shiftDown)
  1214. ClampAspect(tl, br, size, baseSize);
  1215. if (tl.x > br.x)
  1216. std::swap(tl.x, br.x);
  1217. if (tl.y > br.y)
  1218. std::swap(tl.y, br.y);
  1219. vec2_abs(&size, &size);
  1220. obs_sceneitem_set_bounds(stretchItem, &size);
  1221. } else {
  1222. obs_sceneitem_crop crop;
  1223. obs_sceneitem_get_crop(stretchItem, &crop);
  1224. baseSize.x -= float(crop.left + crop.right);
  1225. baseSize.y -= float(crop.top + crop.bottom);
  1226. if (!shiftDown)
  1227. ClampAspect(tl, br, size, baseSize);
  1228. vec2_div(&size, &size, &baseSize);
  1229. obs_sceneitem_set_scale(stretchItem, &size);
  1230. }
  1231. pos3 = CalculateStretchPos(tl, br);
  1232. vec3_transform(&pos3, &pos3, &itemToScreen);
  1233. vec2 newPos;
  1234. vec2_set(&newPos, std::round(pos3.x), std::round(pos3.y));
  1235. obs_sceneitem_set_pos(stretchItem, &newPos);
  1236. }
  1237. void OBSBasicPreview::RotateItem(const vec2 &pos)
  1238. {
  1239. OBSBasic *main = reinterpret_cast<OBSBasic *>(App()->GetMainWindow());
  1240. OBSScene scene = main->GetCurrentScene();
  1241. Qt::KeyboardModifiers modifiers = QGuiApplication::keyboardModifiers();
  1242. bool shiftDown = (modifiers & Qt::ShiftModifier);
  1243. bool ctrlDown = (modifiers & Qt::ControlModifier);
  1244. vec2 pos2;
  1245. vec2_copy(&pos2, &pos);
  1246. float angle =
  1247. atan2(pos2.y - rotatePoint.y, pos2.x - rotatePoint.x) + RAD(90);
  1248. #define ROT_SNAP(rot, thresh) \
  1249. if (abs(angle - RAD(rot)) < RAD(thresh)) { \
  1250. angle = RAD(rot); \
  1251. }
  1252. if (shiftDown) {
  1253. for (int i = 0; i <= 360 / 15; i++) {
  1254. ROT_SNAP(i * 15 - 90, 7.5);
  1255. }
  1256. } else if (!ctrlDown) {
  1257. ROT_SNAP(rotateAngle, 5)
  1258. ROT_SNAP(-90, 5)
  1259. ROT_SNAP(-45, 5)
  1260. ROT_SNAP(0, 5)
  1261. ROT_SNAP(45, 5)
  1262. ROT_SNAP(90, 5)
  1263. ROT_SNAP(135, 5)
  1264. ROT_SNAP(180, 5)
  1265. ROT_SNAP(225, 5)
  1266. ROT_SNAP(270, 5)
  1267. ROT_SNAP(315, 5)
  1268. }
  1269. #undef ROT_SNAP
  1270. vec2 pos3;
  1271. vec2_copy(&pos3, &offsetPoint);
  1272. RotatePos(&pos3, angle);
  1273. pos3.x += rotatePoint.x;
  1274. pos3.y += rotatePoint.y;
  1275. obs_sceneitem_set_rot(stretchItem, DEG(angle));
  1276. obs_sceneitem_set_pos(stretchItem, &pos3);
  1277. }
  1278. void OBSBasicPreview::mouseMoveEvent(QMouseEvent *event)
  1279. {
  1280. OBSBasic *main = reinterpret_cast<OBSBasic *>(App()->GetMainWindow());
  1281. changed = true;
  1282. #if QT_VERSION >= QT_VERSION_CHECK(6, 0, 0)
  1283. QPointF qtPos = event->position();
  1284. #else
  1285. QPointF qtPos = event->localPos();
  1286. #endif
  1287. float pixelRatio = main->devicePixelRatioF();
  1288. if (scrollMode && event->buttons() == Qt::LeftButton) {
  1289. scrollingOffset.x += pixelRatio * (qtPos.x() - scrollingFrom.x);
  1290. scrollingOffset.y += pixelRatio * (qtPos.y() - scrollingFrom.y);
  1291. scrollingFrom.x = qtPos.x();
  1292. scrollingFrom.y = qtPos.y();
  1293. emit DisplayResized();
  1294. return;
  1295. }
  1296. if (locked)
  1297. return;
  1298. bool updateCursor = false;
  1299. if (mouseDown) {
  1300. vec2 pos = GetMouseEventPos(event);
  1301. if (!mouseMoved && !mouseOverItems &&
  1302. stretchHandle == ItemHandle::None) {
  1303. ProcessClick(startPos);
  1304. mouseOverItems = SelectedAtPos(startPos);
  1305. }
  1306. pos.x = std::round(pos.x);
  1307. pos.y = std::round(pos.y);
  1308. if (stretchHandle != ItemHandle::None) {
  1309. if (obs_sceneitem_locked(stretchItem))
  1310. return;
  1311. selectionBox = false;
  1312. OBSScene scene = main->GetCurrentScene();
  1313. obs_sceneitem_t *group =
  1314. obs_sceneitem_get_group(scene, stretchItem);
  1315. if (group) {
  1316. vec3 group_pos;
  1317. vec3_set(&group_pos, pos.x, pos.y, 0.0f);
  1318. vec3_transform(&group_pos, &group_pos,
  1319. &invGroupTransform);
  1320. pos.x = group_pos.x;
  1321. pos.y = group_pos.y;
  1322. }
  1323. if (stretchHandle == ItemHandle::Rot) {
  1324. RotateItem(pos);
  1325. setCursor(Qt::ClosedHandCursor);
  1326. } else if (cropping)
  1327. CropItem(pos);
  1328. else
  1329. StretchItem(pos);
  1330. } else if (mouseOverItems) {
  1331. if (cursor().shape() != Qt::SizeAllCursor)
  1332. setCursor(Qt::SizeAllCursor);
  1333. selectionBox = false;
  1334. MoveItems(pos);
  1335. } else {
  1336. selectionBox = true;
  1337. if (!mouseMoved)
  1338. DoSelect(startPos);
  1339. BoxItems(startPos, pos);
  1340. }
  1341. mouseMoved = true;
  1342. mousePos = pos;
  1343. } else {
  1344. vec2 pos = GetMouseEventPos(event);
  1345. OBSSceneItem item = GetItemAtPos(pos, true);
  1346. std::lock_guard<std::mutex> lock(selectMutex);
  1347. hoveredPreviewItems.clear();
  1348. hoveredPreviewItems.push_back(item);
  1349. if (!mouseMoved && hoveredPreviewItems.size() > 0) {
  1350. mousePos = pos;
  1351. OBSBasic *main = reinterpret_cast<OBSBasic *>(
  1352. App()->GetMainWindow());
  1353. float scale = main->devicePixelRatioF();
  1354. float x = qtPos.x() - main->previewX / scale;
  1355. float y = qtPos.y() - main->previewY / scale;
  1356. vec2_set(&startPos, x, y);
  1357. updateCursor = true;
  1358. }
  1359. }
  1360. if (updateCursor) {
  1361. GetStretchHandleData(startPos, true);
  1362. uint32_t stretchFlags = (uint32_t)stretchHandle;
  1363. UpdateCursor(stretchFlags);
  1364. }
  1365. }
  1366. void OBSBasicPreview::leaveEvent(QEvent *)
  1367. {
  1368. std::lock_guard<std::mutex> lock(selectMutex);
  1369. if (!selectionBox)
  1370. hoveredPreviewItems.clear();
  1371. }
  1372. static void DrawLine(float x1, float y1, float x2, float y2, float thickness,
  1373. vec2 scale)
  1374. {
  1375. float ySide = (y1 == y2) ? (y1 < 0.5f ? 1.0f : -1.0f) : 0.0f;
  1376. float xSide = (x1 == x2) ? (x1 < 0.5f ? 1.0f : -1.0f) : 0.0f;
  1377. gs_render_start(true);
  1378. gs_vertex2f(x1, y1);
  1379. gs_vertex2f(x1 + (xSide * (thickness / scale.x)),
  1380. y1 + (ySide * (thickness / scale.y)));
  1381. gs_vertex2f(x2 + (xSide * (thickness / scale.x)),
  1382. y2 + (ySide * (thickness / scale.y)));
  1383. gs_vertex2f(x2, y2);
  1384. gs_vertex2f(x1, y1);
  1385. gs_vertbuffer_t *line = gs_render_save();
  1386. gs_load_vertexbuffer(line);
  1387. gs_draw(GS_TRISTRIP, 0, 0);
  1388. gs_vertexbuffer_destroy(line);
  1389. }
  1390. static void DrawSquareAtPos(float x, float y)
  1391. {
  1392. OBSBasic *main = OBSBasic::Get();
  1393. float pixelRatio = main->devicePixelRatioF();
  1394. struct vec3 pos;
  1395. vec3_set(&pos, x, y, 0.0f);
  1396. struct matrix4 matrix;
  1397. gs_matrix_get(&matrix);
  1398. vec3_transform(&pos, &pos, &matrix);
  1399. gs_matrix_push();
  1400. gs_matrix_identity();
  1401. gs_matrix_translate(&pos);
  1402. gs_matrix_translate3f(-HANDLE_RADIUS * pixelRatio,
  1403. -HANDLE_RADIUS * pixelRatio, 0.0f);
  1404. gs_matrix_scale3f(HANDLE_RADIUS * pixelRatio * 2,
  1405. HANDLE_RADIUS * pixelRatio * 2, 1.0f);
  1406. gs_draw(GS_TRISTRIP, 0, 0);
  1407. gs_matrix_pop();
  1408. }
  1409. static void DrawRotationHandle(gs_vertbuffer_t *circle, float rot)
  1410. {
  1411. OBSBasic *main = OBSBasic::Get();
  1412. float pixelRatio = main->devicePixelRatioF();
  1413. struct vec3 pos;
  1414. vec3_set(&pos, 0.5f, 0.0f, 0.0f);
  1415. struct matrix4 matrix;
  1416. gs_matrix_get(&matrix);
  1417. vec3_transform(&pos, &pos, &matrix);
  1418. gs_render_start(true);
  1419. gs_vertex2f(0.5f - 0.34f / HANDLE_RADIUS, 0.5f);
  1420. gs_vertex2f(0.5f - 0.34f / HANDLE_RADIUS, -2.0f);
  1421. gs_vertex2f(0.5f + 0.34f / HANDLE_RADIUS, -2.0f);
  1422. gs_vertex2f(0.5f + 0.34f / HANDLE_RADIUS, 0.5f);
  1423. gs_vertex2f(0.5f - 0.34f / HANDLE_RADIUS, 0.5f);
  1424. gs_vertbuffer_t *line = gs_render_save();
  1425. gs_load_vertexbuffer(line);
  1426. gs_matrix_push();
  1427. gs_matrix_identity();
  1428. gs_matrix_translate(&pos);
  1429. gs_matrix_rotaa4f(0.0f, 0.0f, 1.0f, RAD(rot));
  1430. gs_matrix_translate3f(-HANDLE_RADIUS * 1.5 * pixelRatio,
  1431. -HANDLE_RADIUS * 1.5 * pixelRatio, 0.0f);
  1432. gs_matrix_scale3f(HANDLE_RADIUS * 3 * pixelRatio,
  1433. HANDLE_RADIUS * 3 * pixelRatio, 1.0f);
  1434. gs_draw(GS_TRISTRIP, 0, 0);
  1435. gs_matrix_translate3f(0.0f, -HANDLE_RADIUS * 2 / 3, 0.0f);
  1436. gs_load_vertexbuffer(circle);
  1437. gs_draw(GS_TRISTRIP, 0, 0);
  1438. gs_matrix_pop();
  1439. gs_vertexbuffer_destroy(line);
  1440. }
  1441. static void DrawStripedLine(float x1, float y1, float x2, float y2,
  1442. float thickness, vec2 scale)
  1443. {
  1444. float ySide = (y1 == y2) ? (y1 < 0.5f ? 1.0f : -1.0f) : 0.0f;
  1445. float xSide = (x1 == x2) ? (x1 < 0.5f ? 1.0f : -1.0f) : 0.0f;
  1446. float dist =
  1447. sqrt(pow((x1 - x2) * scale.x, 2) + pow((y1 - y2) * scale.y, 2));
  1448. float offX = (x2 - x1) / dist;
  1449. float offY = (y2 - y1) / dist;
  1450. for (int i = 0, l = ceil(dist / 15); i < l; i++) {
  1451. gs_render_start(true);
  1452. float xx1 = x1 + i * 15 * offX;
  1453. float yy1 = y1 + i * 15 * offY;
  1454. float dx;
  1455. float dy;
  1456. if (x1 < x2) {
  1457. dx = std::min(xx1 + 7.5f * offX, x2);
  1458. } else {
  1459. dx = std::max(xx1 + 7.5f * offX, x2);
  1460. }
  1461. if (y1 < y2) {
  1462. dy = std::min(yy1 + 7.5f * offY, y2);
  1463. } else {
  1464. dy = std::max(yy1 + 7.5f * offY, y2);
  1465. }
  1466. gs_vertex2f(xx1, yy1);
  1467. gs_vertex2f(xx1 + (xSide * (thickness / scale.x)),
  1468. yy1 + (ySide * (thickness / scale.y)));
  1469. gs_vertex2f(dx, dy);
  1470. gs_vertex2f(dx + (xSide * (thickness / scale.x)),
  1471. dy + (ySide * (thickness / scale.y)));
  1472. gs_vertbuffer_t *line = gs_render_save();
  1473. gs_load_vertexbuffer(line);
  1474. gs_draw(GS_TRISTRIP, 0, 0);
  1475. gs_vertexbuffer_destroy(line);
  1476. }
  1477. }
  1478. static void DrawRect(float thickness, vec2 scale)
  1479. {
  1480. gs_render_start(true);
  1481. gs_vertex2f(0.0f, 0.0f);
  1482. gs_vertex2f(0.0f + (thickness / scale.x), 0.0f);
  1483. gs_vertex2f(0.0f, 1.0f);
  1484. gs_vertex2f(0.0f + (thickness / scale.x), 1.0f);
  1485. gs_vertex2f(0.0f, 1.0f - (thickness / scale.y));
  1486. gs_vertex2f(1.0f, 1.0f);
  1487. gs_vertex2f(1.0f, 1.0f - (thickness / scale.y));
  1488. gs_vertex2f(1.0f - (thickness / scale.x), 1.0f);
  1489. gs_vertex2f(1.0f, 0.0f);
  1490. gs_vertex2f(1.0f - (thickness / scale.x), 0.0f);
  1491. gs_vertex2f(1.0f, 0.0f + (thickness / scale.y));
  1492. gs_vertex2f(0.0f, 0.0f);
  1493. gs_vertex2f(0.0f, 0.0f + (thickness / scale.y));
  1494. gs_vertbuffer_t *rect = gs_render_save();
  1495. gs_load_vertexbuffer(rect);
  1496. gs_draw(GS_TRISTRIP, 0, 0);
  1497. gs_vertexbuffer_destroy(rect);
  1498. }
  1499. static inline bool crop_enabled(const obs_sceneitem_crop *crop)
  1500. {
  1501. return crop->left > 0 || crop->top > 0 || crop->right > 0 ||
  1502. crop->bottom > 0;
  1503. }
  1504. bool OBSBasicPreview::DrawSelectedOverflow(obs_scene_t *scene,
  1505. obs_sceneitem_t *item, void *param)
  1506. {
  1507. if (obs_sceneitem_locked(item))
  1508. return true;
  1509. if (!SceneItemHasVideo(item))
  1510. return true;
  1511. bool select = config_get_bool(GetGlobalConfig(), "BasicWindow",
  1512. "OverflowSelectionHidden");
  1513. if (!select && !obs_sceneitem_visible(item))
  1514. return true;
  1515. if (obs_sceneitem_is_group(item)) {
  1516. matrix4 mat;
  1517. obs_sceneitem_get_draw_transform(item, &mat);
  1518. gs_matrix_push();
  1519. gs_matrix_mul(&mat);
  1520. obs_sceneitem_group_enum_items(item, DrawSelectedOverflow,
  1521. param);
  1522. gs_matrix_pop();
  1523. }
  1524. bool always = config_get_bool(GetGlobalConfig(), "BasicWindow",
  1525. "OverflowAlwaysVisible");
  1526. if (!always && !obs_sceneitem_selected(item))
  1527. return true;
  1528. OBSBasicPreview *prev = reinterpret_cast<OBSBasicPreview *>(param);
  1529. matrix4 boxTransform;
  1530. matrix4 invBoxTransform;
  1531. obs_sceneitem_get_box_transform(item, &boxTransform);
  1532. matrix4_inv(&invBoxTransform, &boxTransform);
  1533. vec3 bounds[] = {
  1534. {{{0.f, 0.f, 0.f}}},
  1535. {{{1.f, 0.f, 0.f}}},
  1536. {{{0.f, 1.f, 0.f}}},
  1537. {{{1.f, 1.f, 0.f}}},
  1538. };
  1539. bool visible = std::all_of(
  1540. std::begin(bounds), std::end(bounds), [&](const vec3 &b) {
  1541. vec3 pos;
  1542. vec3_transform(&pos, &b, &boxTransform);
  1543. vec3_transform(&pos, &pos, &invBoxTransform);
  1544. return CloseFloat(pos.x, b.x) && CloseFloat(pos.y, b.y);
  1545. });
  1546. if (!visible)
  1547. return true;
  1548. GS_DEBUG_MARKER_BEGIN(GS_DEBUG_COLOR_DEFAULT, "DrawSelectedOverflow");
  1549. obs_transform_info info;
  1550. obs_sceneitem_get_info(item, &info);
  1551. gs_effect_t *solid = obs_get_base_effect(OBS_EFFECT_REPEAT);
  1552. gs_eparam_t *image = gs_effect_get_param_by_name(solid, "image");
  1553. gs_eparam_t *scale = gs_effect_get_param_by_name(solid, "scale");
  1554. vec2 s;
  1555. vec2_set(&s, boxTransform.x.x / 96, boxTransform.y.y / 96);
  1556. gs_effect_set_vec2(scale, &s);
  1557. gs_effect_set_texture(image, prev->overflow);
  1558. gs_matrix_push();
  1559. gs_matrix_mul(&boxTransform);
  1560. obs_sceneitem_crop crop;
  1561. obs_sceneitem_get_crop(item, &crop);
  1562. while (gs_effect_loop(solid, "Draw")) {
  1563. gs_draw_sprite(prev->overflow, 0, 1, 1);
  1564. }
  1565. gs_matrix_pop();
  1566. GS_DEBUG_MARKER_END();
  1567. UNUSED_PARAMETER(scene);
  1568. return true;
  1569. }
  1570. bool OBSBasicPreview::DrawSelectedItem(obs_scene_t *scene,
  1571. obs_sceneitem_t *item, void *param)
  1572. {
  1573. if (obs_sceneitem_locked(item))
  1574. return true;
  1575. if (!SceneItemHasVideo(item))
  1576. return true;
  1577. OBSBasicPreview *prev = reinterpret_cast<OBSBasicPreview *>(param);
  1578. if (obs_sceneitem_is_group(item)) {
  1579. matrix4 mat;
  1580. obs_transform_info groupInfo;
  1581. obs_sceneitem_get_draw_transform(item, &mat);
  1582. obs_sceneitem_get_info(item, &groupInfo);
  1583. prev->groupRot = groupInfo.rot;
  1584. gs_matrix_push();
  1585. gs_matrix_mul(&mat);
  1586. obs_sceneitem_group_enum_items(item, DrawSelectedItem, prev);
  1587. gs_matrix_pop();
  1588. prev->groupRot = 0.0f;
  1589. }
  1590. OBSBasic *main = OBSBasic::Get();
  1591. float pixelRatio = main->devicePixelRatioF();
  1592. bool hovered = false;
  1593. {
  1594. std::lock_guard<std::mutex> lock(prev->selectMutex);
  1595. for (size_t i = 0; i < prev->hoveredPreviewItems.size(); i++) {
  1596. if (prev->hoveredPreviewItems[i] == item) {
  1597. hovered = true;
  1598. break;
  1599. }
  1600. }
  1601. }
  1602. bool selected = obs_sceneitem_selected(item);
  1603. if (!selected && !hovered)
  1604. return true;
  1605. matrix4 boxTransform;
  1606. matrix4 invBoxTransform;
  1607. obs_sceneitem_get_box_transform(item, &boxTransform);
  1608. matrix4_inv(&invBoxTransform, &boxTransform);
  1609. vec3 bounds[] = {
  1610. {{{0.f, 0.f, 0.f}}},
  1611. {{{1.f, 0.f, 0.f}}},
  1612. {{{0.f, 1.f, 0.f}}},
  1613. {{{1.f, 1.f, 0.f}}},
  1614. };
  1615. main->GetCameraIcon();
  1616. QColor selColor = main->GetSelectionColor();
  1617. QColor cropColor = main->GetCropColor();
  1618. QColor hoverColor = main->GetHoverColor();
  1619. vec4 red;
  1620. vec4 green;
  1621. vec4 blue;
  1622. vec4_set(&red, selColor.redF(), selColor.greenF(), selColor.blueF(),
  1623. 1.0f);
  1624. vec4_set(&green, cropColor.redF(), cropColor.greenF(),
  1625. cropColor.blueF(), 1.0f);
  1626. vec4_set(&blue, hoverColor.redF(), hoverColor.greenF(),
  1627. hoverColor.blueF(), 1.0f);
  1628. bool visible = std::all_of(
  1629. std::begin(bounds), std::end(bounds), [&](const vec3 &b) {
  1630. vec3 pos;
  1631. vec3_transform(&pos, &b, &boxTransform);
  1632. vec3_transform(&pos, &pos, &invBoxTransform);
  1633. return CloseFloat(pos.x, b.x) && CloseFloat(pos.y, b.y);
  1634. });
  1635. if (!visible)
  1636. return true;
  1637. GS_DEBUG_MARKER_BEGIN(GS_DEBUG_COLOR_DEFAULT, "DrawSelectedItem");
  1638. matrix4 curTransform;
  1639. vec2 boxScale;
  1640. gs_matrix_get(&curTransform);
  1641. obs_sceneitem_get_box_scale(item, &boxScale);
  1642. boxScale.x *= curTransform.x.x;
  1643. boxScale.y *= curTransform.y.y;
  1644. obs_transform_info info;
  1645. obs_sceneitem_get_info(item, &info);
  1646. gs_matrix_push();
  1647. gs_matrix_mul(&boxTransform);
  1648. obs_sceneitem_crop crop;
  1649. obs_sceneitem_get_crop(item, &crop);
  1650. gs_effect_t *eff = gs_get_effect();
  1651. gs_eparam_t *colParam = gs_effect_get_param_by_name(eff, "color");
  1652. gs_effect_set_vec4(colParam, &red);
  1653. if (info.bounds_type == OBS_BOUNDS_NONE && crop_enabled(&crop)) {
  1654. #define DRAW_SIDE(side, x1, y1, x2, y2) \
  1655. if (hovered && !selected) { \
  1656. gs_effect_set_vec4(colParam, &blue); \
  1657. DrawLine(x1, y1, x2, y2, HANDLE_RADIUS *pixelRatio / 2, \
  1658. boxScale); \
  1659. } else if (crop.side > 0) { \
  1660. gs_effect_set_vec4(colParam, &green); \
  1661. DrawStripedLine(x1, y1, x2, y2, HANDLE_RADIUS *pixelRatio / 2, \
  1662. boxScale); \
  1663. } else { \
  1664. DrawLine(x1, y1, x2, y2, HANDLE_RADIUS *pixelRatio / 2, \
  1665. boxScale); \
  1666. } \
  1667. gs_effect_set_vec4(colParam, &red);
  1668. DRAW_SIDE(left, 0.0f, 0.0f, 0.0f, 1.0f);
  1669. DRAW_SIDE(top, 0.0f, 0.0f, 1.0f, 0.0f);
  1670. DRAW_SIDE(right, 1.0f, 0.0f, 1.0f, 1.0f);
  1671. DRAW_SIDE(bottom, 0.0f, 1.0f, 1.0f, 1.0f);
  1672. #undef DRAW_SIDE
  1673. } else {
  1674. if (!selected) {
  1675. gs_effect_set_vec4(colParam, &blue);
  1676. DrawRect(HANDLE_RADIUS * pixelRatio / 2, boxScale);
  1677. } else {
  1678. DrawRect(HANDLE_RADIUS * pixelRatio / 2, boxScale);
  1679. }
  1680. }
  1681. gs_load_vertexbuffer(main->box);
  1682. gs_effect_set_vec4(colParam, &red);
  1683. if (selected) {
  1684. DrawSquareAtPos(0.0f, 0.0f);
  1685. DrawSquareAtPos(0.0f, 1.0f);
  1686. DrawSquareAtPos(1.0f, 0.0f);
  1687. DrawSquareAtPos(1.0f, 1.0f);
  1688. DrawSquareAtPos(0.5f, 0.0f);
  1689. DrawSquareAtPos(0.0f, 0.5f);
  1690. DrawSquareAtPos(0.5f, 1.0f);
  1691. DrawSquareAtPos(1.0f, 0.5f);
  1692. if (!prev->circleFill) {
  1693. gs_render_start(true);
  1694. float angle = 180;
  1695. for (int i = 0, l = 40; i < l; i++) {
  1696. gs_vertex2f(sin(RAD(angle)) / 2 + 0.5f,
  1697. cos(RAD(angle)) / 2 + 0.5f);
  1698. angle += 360 / l;
  1699. gs_vertex2f(sin(RAD(angle)) / 2 + 0.5f,
  1700. cos(RAD(angle)) / 2 + 0.5f);
  1701. gs_vertex2f(0.5f, 1.0f);
  1702. }
  1703. prev->circleFill = gs_render_save();
  1704. }
  1705. DrawRotationHandle(prev->circleFill, info.rot + prev->groupRot);
  1706. }
  1707. gs_matrix_pop();
  1708. GS_DEBUG_MARKER_END();
  1709. UNUSED_PARAMETER(scene);
  1710. return true;
  1711. }
  1712. bool OBSBasicPreview::DrawSelectionBox(float x1, float y1, float x2, float y2,
  1713. gs_vertbuffer_t *rectFill)
  1714. {
  1715. OBSBasic *main = OBSBasic::Get();
  1716. float pixelRatio = main->devicePixelRatioF();
  1717. x1 = std::round(x1);
  1718. x2 = std::round(x2);
  1719. y1 = std::round(y1);
  1720. y2 = std::round(y2);
  1721. gs_effect_t *eff = gs_get_effect();
  1722. gs_eparam_t *colParam = gs_effect_get_param_by_name(eff, "color");
  1723. vec4 fillColor;
  1724. vec4_set(&fillColor, 0.7f, 0.7f, 0.7f, 0.5f);
  1725. vec4 borderColor;
  1726. vec4_set(&borderColor, 1.0f, 1.0f, 1.0f, 1.0f);
  1727. vec2 scale;
  1728. vec2_set(&scale, std::abs(x2 - x1), std::abs(y2 - y1));
  1729. gs_matrix_push();
  1730. gs_matrix_identity();
  1731. gs_matrix_translate3f(x1, y1, 0.0f);
  1732. gs_matrix_scale3f(x2 - x1, y2 - y1, 1.0f);
  1733. gs_effect_set_vec4(colParam, &fillColor);
  1734. gs_load_vertexbuffer(rectFill);
  1735. gs_draw(GS_TRISTRIP, 0, 0);
  1736. gs_effect_set_vec4(colParam, &borderColor);
  1737. DrawRect(HANDLE_RADIUS * pixelRatio / 2, scale);
  1738. gs_matrix_pop();
  1739. return true;
  1740. }
  1741. void OBSBasicPreview::DrawOverflow()
  1742. {
  1743. if (locked)
  1744. return;
  1745. bool hidden = config_get_bool(GetGlobalConfig(), "BasicWindow",
  1746. "OverflowHidden");
  1747. if (hidden)
  1748. return;
  1749. GS_DEBUG_MARKER_BEGIN(GS_DEBUG_COLOR_DEFAULT, "DrawOverflow");
  1750. if (!overflow) {
  1751. std::string path;
  1752. GetDataFilePath("images/overflow.png", path);
  1753. overflow = gs_texture_create_from_file(path.c_str());
  1754. }
  1755. OBSBasic *main = reinterpret_cast<OBSBasic *>(App()->GetMainWindow());
  1756. OBSScene scene = main->GetCurrentScene();
  1757. if (scene) {
  1758. gs_matrix_push();
  1759. gs_matrix_scale3f(main->previewScale, main->previewScale, 1.0f);
  1760. obs_scene_enum_items(scene, DrawSelectedOverflow, this);
  1761. gs_matrix_pop();
  1762. }
  1763. gs_load_vertexbuffer(nullptr);
  1764. GS_DEBUG_MARKER_END();
  1765. }
  1766. void OBSBasicPreview::DrawSceneEditing()
  1767. {
  1768. if (locked)
  1769. return;
  1770. GS_DEBUG_MARKER_BEGIN(GS_DEBUG_COLOR_DEFAULT, "DrawSceneEditing");
  1771. OBSBasic *main = reinterpret_cast<OBSBasic *>(App()->GetMainWindow());
  1772. gs_effect_t *solid = obs_get_base_effect(OBS_EFFECT_SOLID);
  1773. gs_technique_t *tech = gs_effect_get_technique(solid, "Solid");
  1774. gs_technique_begin(tech);
  1775. gs_technique_begin_pass(tech, 0);
  1776. OBSScene scene = main->GetCurrentScene();
  1777. if (scene) {
  1778. gs_matrix_push();
  1779. gs_matrix_scale3f(main->previewScale, main->previewScale, 1.0f);
  1780. obs_scene_enum_items(scene, DrawSelectedItem, this);
  1781. gs_matrix_pop();
  1782. }
  1783. if (selectionBox) {
  1784. if (!rectFill) {
  1785. gs_render_start(true);
  1786. gs_vertex2f(0.0f, 0.0f);
  1787. gs_vertex2f(1.0f, 0.0f);
  1788. gs_vertex2f(0.0f, 1.0f);
  1789. gs_vertex2f(1.0f, 1.0f);
  1790. rectFill = gs_render_save();
  1791. }
  1792. DrawSelectionBox(startPos.x * main->previewScale,
  1793. startPos.y * main->previewScale,
  1794. mousePos.x * main->previewScale,
  1795. mousePos.y * main->previewScale, rectFill);
  1796. }
  1797. gs_load_vertexbuffer(nullptr);
  1798. gs_technique_end_pass(tech);
  1799. gs_technique_end(tech);
  1800. GS_DEBUG_MARKER_END();
  1801. }
  1802. void OBSBasicPreview::ResetScrollingOffset()
  1803. {
  1804. vec2_zero(&scrollingOffset);
  1805. }
  1806. void OBSBasicPreview::SetScalingLevel(int32_t newScalingLevelVal)
  1807. {
  1808. float newScalingAmountVal =
  1809. pow(ZOOM_SENSITIVITY, float(newScalingLevelVal));
  1810. scalingLevel = newScalingLevelVal;
  1811. SetScalingAmount(newScalingAmountVal);
  1812. }
  1813. void OBSBasicPreview::SetScalingAmount(float newScalingAmountVal)
  1814. {
  1815. scrollingOffset.x *= newScalingAmountVal / scalingAmount;
  1816. scrollingOffset.y *= newScalingAmountVal / scalingAmount;
  1817. scalingAmount = newScalingAmountVal;
  1818. }
  1819. OBSBasicPreview *OBSBasicPreview::Get()
  1820. {
  1821. return OBSBasic::Get()->ui->preview;
  1822. }