window-basic-preview.cpp 67 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637
  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. #include "display-helpers.hpp"
  12. #define HANDLE_RADIUS 4.0f
  13. #define HANDLE_SEL_RADIUS (HANDLE_RADIUS * 1.5f)
  14. #define HELPER_ROT_BREAKPONT 45.0f
  15. /* TODO: make C++ math classes and clean up code here later */
  16. OBSBasicPreview::OBSBasicPreview(QWidget *parent, Qt::WindowFlags flags)
  17. : OBSQTDisplay(parent, flags)
  18. {
  19. ResetScrollingOffset();
  20. setMouseTracking(true);
  21. }
  22. OBSBasicPreview::~OBSBasicPreview()
  23. {
  24. obs_enter_graphics();
  25. if (overflow)
  26. gs_texture_destroy(overflow);
  27. if (rectFill)
  28. gs_vertexbuffer_destroy(rectFill);
  29. if (circleFill)
  30. gs_vertexbuffer_destroy(circleFill);
  31. obs_leave_graphics();
  32. }
  33. vec2 OBSBasicPreview::GetMouseEventPos(QMouseEvent *event)
  34. {
  35. OBSBasic *main = reinterpret_cast<OBSBasic *>(App()->GetMainWindow());
  36. float pixelRatio = main->GetDevicePixelRatio();
  37. float scale = pixelRatio / main->previewScale;
  38. QPoint qtPos = event->pos();
  39. vec2 pos;
  40. vec2_set(&pos, (qtPos.x() - main->previewX / pixelRatio) * scale,
  41. (qtPos.y() - main->previewY / pixelRatio) * scale);
  42. return pos;
  43. }
  44. static void RotatePos(vec2 *pos, float rot)
  45. {
  46. float cosR = cos(rot);
  47. float sinR = sin(rot);
  48. vec2 newPos;
  49. newPos.x = cosR * pos->x - sinR * pos->y;
  50. newPos.y = sinR * pos->x + cosR * pos->y;
  51. vec2_copy(pos, &newPos);
  52. }
  53. struct SceneFindData {
  54. const vec2 &pos;
  55. OBSSceneItem item;
  56. bool selectBelow;
  57. obs_sceneitem_t *group = nullptr;
  58. SceneFindData(const SceneFindData &) = delete;
  59. SceneFindData(SceneFindData &&) = delete;
  60. SceneFindData &operator=(const SceneFindData &) = delete;
  61. SceneFindData &operator=(SceneFindData &&) = delete;
  62. inline SceneFindData(const vec2 &pos_, bool selectBelow_)
  63. : pos(pos_), selectBelow(selectBelow_)
  64. {
  65. }
  66. };
  67. struct SceneFindBoxData {
  68. const vec2 &startPos;
  69. const vec2 &pos;
  70. std::vector<obs_sceneitem_t *> sceneItems;
  71. SceneFindBoxData(const SceneFindData &) = delete;
  72. SceneFindBoxData(SceneFindData &&) = delete;
  73. SceneFindBoxData &operator=(const SceneFindData &) = delete;
  74. SceneFindBoxData &operator=(SceneFindData &&) = delete;
  75. inline SceneFindBoxData(const vec2 &startPos_, const vec2 &pos_)
  76. : startPos(startPos_), pos(pos_)
  77. {
  78. }
  79. };
  80. static bool SceneItemHasVideo(obs_sceneitem_t *item)
  81. {
  82. obs_source_t *source = obs_sceneitem_get_source(item);
  83. uint32_t flags = obs_source_get_output_flags(source);
  84. return (flags & OBS_SOURCE_VIDEO) != 0;
  85. }
  86. static bool CloseFloat(float a, float b, float epsilon = 0.01)
  87. {
  88. using std::abs;
  89. return abs(a - b) <= epsilon;
  90. }
  91. static bool FindItemAtPos(obs_scene_t * /* scene */, obs_sceneitem_t *item,
  92. void *param)
  93. {
  94. SceneFindData *data = reinterpret_cast<SceneFindData *>(param);
  95. matrix4 transform;
  96. matrix4 invTransform;
  97. vec3 transformedPos;
  98. vec3 pos3;
  99. vec3 pos3_;
  100. if (!SceneItemHasVideo(item))
  101. return true;
  102. if (obs_sceneitem_locked(item))
  103. return true;
  104. vec3_set(&pos3, data->pos.x, data->pos.y, 0.0f);
  105. obs_sceneitem_get_box_transform(item, &transform);
  106. matrix4_inv(&invTransform, &transform);
  107. vec3_transform(&transformedPos, &pos3, &invTransform);
  108. vec3_transform(&pos3_, &transformedPos, &transform);
  109. if (CloseFloat(pos3.x, pos3_.x) && CloseFloat(pos3.y, pos3_.y) &&
  110. transformedPos.x >= 0.0f && transformedPos.x <= 1.0f &&
  111. transformedPos.y >= 0.0f && transformedPos.y <= 1.0f) {
  112. if (data->selectBelow && obs_sceneitem_selected(item)) {
  113. if (data->item)
  114. return false;
  115. else
  116. data->selectBelow = false;
  117. }
  118. data->item = item;
  119. }
  120. return true;
  121. }
  122. static vec3 GetTransformedPos(float x, float y, const matrix4 &mat)
  123. {
  124. vec3 result;
  125. vec3_set(&result, x, y, 0.0f);
  126. vec3_transform(&result, &result, &mat);
  127. return result;
  128. }
  129. static inline vec2 GetOBSScreenSize()
  130. {
  131. obs_video_info ovi;
  132. vec2 size;
  133. vec2_zero(&size);
  134. if (obs_get_video_info(&ovi)) {
  135. size.x = float(ovi.base_width);
  136. size.y = float(ovi.base_height);
  137. }
  138. return size;
  139. }
  140. vec3 OBSBasicPreview::GetSnapOffset(const vec3 &tl, const vec3 &br)
  141. {
  142. OBSBasic *main = reinterpret_cast<OBSBasic *>(App()->GetMainWindow());
  143. vec2 screenSize = GetOBSScreenSize();
  144. vec3 clampOffset;
  145. vec3_zero(&clampOffset);
  146. const bool snap = config_get_bool(GetGlobalConfig(), "BasicWindow",
  147. "SnappingEnabled");
  148. if (snap == false)
  149. return clampOffset;
  150. const bool screenSnap = config_get_bool(
  151. GetGlobalConfig(), "BasicWindow", "ScreenSnapping");
  152. const bool centerSnap = config_get_bool(
  153. GetGlobalConfig(), "BasicWindow", "CenterSnapping");
  154. const float clampDist = config_get_double(GetGlobalConfig(),
  155. "BasicWindow",
  156. "SnapDistance") /
  157. main->previewScale;
  158. const float centerX = br.x - (br.x - tl.x) / 2.0f;
  159. const float centerY = br.y - (br.y - tl.y) / 2.0f;
  160. // Left screen edge.
  161. if (screenSnap && fabsf(tl.x) < clampDist)
  162. clampOffset.x = -tl.x;
  163. // Right screen edge.
  164. if (screenSnap && fabsf(clampOffset.x) < EPSILON &&
  165. fabsf(screenSize.x - br.x) < clampDist)
  166. clampOffset.x = screenSize.x - br.x;
  167. // Horizontal center.
  168. if (centerSnap && fabsf(screenSize.x - (br.x - tl.x)) > clampDist &&
  169. fabsf(screenSize.x / 2.0f - centerX) < clampDist)
  170. clampOffset.x = screenSize.x / 2.0f - centerX;
  171. // Top screen edge.
  172. if (screenSnap && fabsf(tl.y) < clampDist)
  173. clampOffset.y = -tl.y;
  174. // Bottom screen edge.
  175. if (screenSnap && fabsf(clampOffset.y) < EPSILON &&
  176. fabsf(screenSize.y - br.y) < clampDist)
  177. clampOffset.y = screenSize.y - br.y;
  178. // Vertical center.
  179. if (centerSnap && fabsf(screenSize.y - (br.y - tl.y)) > clampDist &&
  180. fabsf(screenSize.y / 2.0f - centerY) < clampDist)
  181. clampOffset.y = screenSize.y / 2.0f - centerY;
  182. return clampOffset;
  183. }
  184. OBSSceneItem OBSBasicPreview::GetItemAtPos(const vec2 &pos, bool selectBelow)
  185. {
  186. OBSBasic *main = reinterpret_cast<OBSBasic *>(App()->GetMainWindow());
  187. OBSScene scene = main->GetCurrentScene();
  188. if (!scene)
  189. return OBSSceneItem();
  190. SceneFindData data(pos, selectBelow);
  191. obs_scene_enum_items(scene, FindItemAtPos, &data);
  192. return data.item;
  193. }
  194. static bool CheckItemSelected(obs_scene_t * /* scene */, obs_sceneitem_t *item,
  195. void *param)
  196. {
  197. SceneFindData *data = reinterpret_cast<SceneFindData *>(param);
  198. matrix4 transform;
  199. vec3 transformedPos;
  200. vec3 pos3;
  201. if (!SceneItemHasVideo(item))
  202. return true;
  203. if (obs_sceneitem_is_group(item)) {
  204. data->group = item;
  205. obs_sceneitem_group_enum_items(item, CheckItemSelected, param);
  206. data->group = nullptr;
  207. if (data->item) {
  208. return false;
  209. }
  210. }
  211. vec3_set(&pos3, data->pos.x, data->pos.y, 0.0f);
  212. obs_sceneitem_get_box_transform(item, &transform);
  213. if (data->group) {
  214. matrix4 parent_transform;
  215. obs_sceneitem_get_draw_transform(data->group,
  216. &parent_transform);
  217. matrix4_mul(&transform, &transform, &parent_transform);
  218. }
  219. matrix4_inv(&transform, &transform);
  220. vec3_transform(&transformedPos, &pos3, &transform);
  221. if (transformedPos.x >= 0.0f && transformedPos.x <= 1.0f &&
  222. transformedPos.y >= 0.0f && transformedPos.y <= 1.0f) {
  223. if (obs_sceneitem_selected(item)) {
  224. data->item = item;
  225. return false;
  226. }
  227. }
  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. vec2 scale;
  321. obs_sceneitem_get_scale(item, &scale);
  322. bool invert = scale.y < 0.0f;
  323. vec3 handlePos =
  324. GetTransformedPos(0.5f, invert ? 1.0f : 0.0f, transform);
  325. vec3_transform(&handlePos, &handlePos, &data.parent_xform);
  326. handlePos.x -= rotHandleOffset.x;
  327. if (scale.x < 0.0f)
  328. handlePos.y += rotHandleOffset.y;
  329. else
  330. handlePos.y -= rotHandleOffset.y;
  331. float dist = vec3_dist(&handlePos, &pos3);
  332. if (dist < data.radius) {
  333. if (dist < closestHandle) {
  334. closestHandle = dist;
  335. data.item = item;
  336. data.angle = obs_sceneitem_get_rot(item);
  337. data.handle = ItemHandle::Rot;
  338. vec2_set(&data.rotatePoint,
  339. transform.t.x + transform.x.x / 2 +
  340. transform.y.x / 2,
  341. transform.t.y + transform.x.y / 2 +
  342. transform.y.y / 2);
  343. obs_sceneitem_get_pos(item, &data.offsetPoint);
  344. data.offsetPoint.x -= data.rotatePoint.x;
  345. data.offsetPoint.y -= data.rotatePoint.y;
  346. RotatePos(&data.offsetPoint,
  347. -RAD(obs_sceneitem_get_rot(item)));
  348. }
  349. }
  350. return true;
  351. }
  352. static vec2 GetItemSize(obs_sceneitem_t *item)
  353. {
  354. obs_bounds_type boundsType = obs_sceneitem_get_bounds_type(item);
  355. vec2 size;
  356. if (boundsType != OBS_BOUNDS_NONE) {
  357. obs_sceneitem_get_bounds(item, &size);
  358. } else {
  359. obs_source_t *source = obs_sceneitem_get_source(item);
  360. obs_sceneitem_crop crop;
  361. vec2 scale;
  362. obs_sceneitem_get_scale(item, &scale);
  363. obs_sceneitem_get_crop(item, &crop);
  364. size.x = float(obs_source_get_width(source) - crop.left -
  365. crop.right) *
  366. scale.x;
  367. size.y = float(obs_source_get_height(source) - crop.top -
  368. crop.bottom) *
  369. scale.y;
  370. }
  371. return size;
  372. }
  373. void OBSBasicPreview::GetStretchHandleData(const vec2 &pos, bool ignoreGroup)
  374. {
  375. OBSBasic *main = reinterpret_cast<OBSBasic *>(App()->GetMainWindow());
  376. OBSScene scene = main->GetCurrentScene();
  377. if (!scene)
  378. return;
  379. float scale = main->previewScale / main->GetDevicePixelRatio();
  380. vec2 scaled_pos = pos;
  381. vec2_divf(&scaled_pos, &scaled_pos, scale);
  382. HandleFindData data(scaled_pos, scale);
  383. obs_scene_enum_items(scene, FindHandleAtPos, &data);
  384. stretchItem = std::move(data.item);
  385. stretchHandle = data.handle;
  386. rotateAngle = data.angle;
  387. rotatePoint = data.rotatePoint;
  388. offsetPoint = data.offsetPoint;
  389. if (stretchHandle != ItemHandle::None) {
  390. matrix4 boxTransform;
  391. vec3 itemUL;
  392. float itemRot;
  393. stretchItemSize = GetItemSize(stretchItem);
  394. obs_sceneitem_get_box_transform(stretchItem, &boxTransform);
  395. itemRot = obs_sceneitem_get_rot(stretchItem);
  396. vec3_from_vec4(&itemUL, &boxTransform.t);
  397. /* build the item space conversion matrices */
  398. matrix4_identity(&itemToScreen);
  399. matrix4_rotate_aa4f(&itemToScreen, &itemToScreen, 0.0f, 0.0f,
  400. 1.0f, RAD(itemRot));
  401. matrix4_translate3f(&itemToScreen, &itemToScreen, itemUL.x,
  402. itemUL.y, 0.0f);
  403. matrix4_identity(&screenToItem);
  404. matrix4_translate3f(&screenToItem, &screenToItem, -itemUL.x,
  405. -itemUL.y, 0.0f);
  406. matrix4_rotate_aa4f(&screenToItem, &screenToItem, 0.0f, 0.0f,
  407. 1.0f, RAD(-itemRot));
  408. obs_sceneitem_get_crop(stretchItem, &startCrop);
  409. obs_sceneitem_get_pos(stretchItem, &startItemPos);
  410. obs_source_t *source = obs_sceneitem_get_source(stretchItem);
  411. cropSize.x = float(obs_source_get_width(source) -
  412. startCrop.left - startCrop.right);
  413. cropSize.y = float(obs_source_get_height(source) -
  414. startCrop.top - startCrop.bottom);
  415. stretchGroup = obs_sceneitem_get_group(scene, stretchItem);
  416. if (stretchGroup && !ignoreGroup) {
  417. obs_sceneitem_get_draw_transform(stretchGroup,
  418. &invGroupTransform);
  419. matrix4_inv(&invGroupTransform, &invGroupTransform);
  420. obs_sceneitem_defer_group_resize_begin(stretchGroup);
  421. }
  422. }
  423. }
  424. void OBSBasicPreview::keyPressEvent(QKeyEvent *event)
  425. {
  426. if (!IsFixedScaling() || event->isAutoRepeat()) {
  427. OBSQTDisplay::keyPressEvent(event);
  428. return;
  429. }
  430. switch (event->key()) {
  431. case Qt::Key_Space:
  432. setCursor(Qt::OpenHandCursor);
  433. scrollMode = true;
  434. break;
  435. }
  436. OBSQTDisplay::keyPressEvent(event);
  437. }
  438. void OBSBasicPreview::keyReleaseEvent(QKeyEvent *event)
  439. {
  440. if (event->isAutoRepeat()) {
  441. OBSQTDisplay::keyReleaseEvent(event);
  442. return;
  443. }
  444. switch (event->key()) {
  445. case Qt::Key_Space:
  446. scrollMode = false;
  447. setCursor(Qt::ArrowCursor);
  448. break;
  449. }
  450. OBSQTDisplay::keyReleaseEvent(event);
  451. }
  452. void OBSBasicPreview::wheelEvent(QWheelEvent *event)
  453. {
  454. if (scrollMode && IsFixedScaling()) {
  455. const int delta = event->angleDelta().y();
  456. if (delta != 0) {
  457. if (delta > 0)
  458. SetScalingLevel(scalingLevel + 1);
  459. else
  460. SetScalingLevel(scalingLevel - 1);
  461. emit DisplayResized();
  462. }
  463. }
  464. OBSQTDisplay::wheelEvent(event);
  465. }
  466. void OBSBasicPreview::mousePressEvent(QMouseEvent *event)
  467. {
  468. #if QT_VERSION >= QT_VERSION_CHECK(6, 0, 0)
  469. QPointF pos = event->position();
  470. #else
  471. QPointF pos = event->localPos();
  472. #endif
  473. if (scrollMode && IsFixedScaling() &&
  474. event->button() == Qt::LeftButton) {
  475. setCursor(Qt::ClosedHandCursor);
  476. scrollingFrom.x = pos.x();
  477. scrollingFrom.y = pos.y();
  478. return;
  479. }
  480. if (event->button() == Qt::RightButton) {
  481. scrollMode = false;
  482. setCursor(Qt::ArrowCursor);
  483. }
  484. if (locked) {
  485. OBSQTDisplay::mousePressEvent(event);
  486. return;
  487. }
  488. OBSBasic *main = reinterpret_cast<OBSBasic *>(App()->GetMainWindow());
  489. float pixelRatio = main->GetDevicePixelRatio();
  490. float x = pos.x() - main->previewX / pixelRatio;
  491. float y = pos.y() - main->previewY / pixelRatio;
  492. Qt::KeyboardModifiers modifiers = QGuiApplication::keyboardModifiers();
  493. bool altDown = (modifiers & Qt::AltModifier);
  494. bool shiftDown = (modifiers & Qt::ShiftModifier);
  495. bool ctrlDown = (modifiers & Qt::ControlModifier);
  496. OBSQTDisplay::mousePressEvent(event);
  497. if (event->button() != Qt::LeftButton &&
  498. event->button() != Qt::RightButton)
  499. return;
  500. if (event->button() == Qt::LeftButton)
  501. mouseDown = true;
  502. {
  503. std::lock_guard<std::mutex> lock(selectMutex);
  504. selectedItems.clear();
  505. }
  506. if (altDown)
  507. cropping = true;
  508. if (altDown || shiftDown || ctrlDown) {
  509. vec2 s;
  510. SceneFindBoxData data(s, s);
  511. obs_scene_enum_items(main->GetCurrentScene(), FindSelected,
  512. &data);
  513. std::lock_guard<std::mutex> lock(selectMutex);
  514. selectedItems = data.sceneItems;
  515. }
  516. vec2_set(&startPos, x, y);
  517. GetStretchHandleData(startPos, false);
  518. vec2_divf(&startPos, &startPos, main->previewScale / pixelRatio);
  519. startPos.x = std::round(startPos.x);
  520. startPos.y = std::round(startPos.y);
  521. mouseOverItems = SelectedAtPos(startPos);
  522. vec2_zero(&lastMoveOffset);
  523. mousePos = startPos;
  524. wrapper =
  525. obs_scene_save_transform_states(main->GetCurrentScene(), true);
  526. changed = false;
  527. }
  528. void OBSBasicPreview::UpdateCursor(uint32_t &flags)
  529. {
  530. if (obs_sceneitem_locked(stretchItem)) {
  531. unsetCursor();
  532. return;
  533. }
  534. if (!flags && (cursor().shape() != Qt::OpenHandCursor || !scrollMode))
  535. unsetCursor();
  536. if (cursor().shape() != Qt::ArrowCursor)
  537. return;
  538. if ((flags & ITEM_LEFT && flags & ITEM_TOP) ||
  539. (flags & ITEM_RIGHT && flags & ITEM_BOTTOM))
  540. setCursor(Qt::SizeFDiagCursor);
  541. else if ((flags & ITEM_LEFT && flags & ITEM_BOTTOM) ||
  542. (flags & ITEM_RIGHT && flags & ITEM_TOP))
  543. setCursor(Qt::SizeBDiagCursor);
  544. else if (flags & ITEM_LEFT || flags & ITEM_RIGHT)
  545. setCursor(Qt::SizeHorCursor);
  546. else if (flags & ITEM_TOP || flags & ITEM_BOTTOM)
  547. setCursor(Qt::SizeVerCursor);
  548. else if (flags & ITEM_ROT)
  549. setCursor(Qt::OpenHandCursor);
  550. }
  551. static bool select_one(obs_scene_t * /* scene */, obs_sceneitem_t *item,
  552. void *param)
  553. {
  554. obs_sceneitem_t *selectedItem =
  555. reinterpret_cast<obs_sceneitem_t *>(param);
  556. if (obs_sceneitem_is_group(item))
  557. obs_sceneitem_group_enum_items(item, select_one, param);
  558. obs_sceneitem_select(item, (selectedItem == item));
  559. return true;
  560. }
  561. void OBSBasicPreview::DoSelect(const vec2 &pos)
  562. {
  563. OBSBasic *main = reinterpret_cast<OBSBasic *>(App()->GetMainWindow());
  564. OBSScene scene = main->GetCurrentScene();
  565. OBSSceneItem item = GetItemAtPos(pos, true);
  566. obs_scene_enum_items(scene, select_one, (obs_sceneitem_t *)item);
  567. }
  568. void OBSBasicPreview::DoCtrlSelect(const vec2 &pos)
  569. {
  570. OBSSceneItem item = GetItemAtPos(pos, false);
  571. if (!item)
  572. return;
  573. bool selected = obs_sceneitem_selected(item);
  574. obs_sceneitem_select(item, !selected);
  575. }
  576. void OBSBasicPreview::ProcessClick(const vec2 &pos)
  577. {
  578. Qt::KeyboardModifiers modifiers = QGuiApplication::keyboardModifiers();
  579. if (modifiers & Qt::ControlModifier)
  580. DoCtrlSelect(pos);
  581. else
  582. DoSelect(pos);
  583. }
  584. void OBSBasicPreview::mouseReleaseEvent(QMouseEvent *event)
  585. {
  586. if (scrollMode)
  587. setCursor(Qt::OpenHandCursor);
  588. if (locked) {
  589. OBSQTDisplay::mouseReleaseEvent(event);
  590. return;
  591. }
  592. if (mouseDown) {
  593. vec2 pos = GetMouseEventPos(event);
  594. if (!mouseMoved)
  595. ProcessClick(pos);
  596. if (selectionBox) {
  597. Qt::KeyboardModifiers modifiers =
  598. QGuiApplication::keyboardModifiers();
  599. bool altDown = modifiers & Qt::AltModifier;
  600. bool shiftDown = modifiers & Qt::ShiftModifier;
  601. bool ctrlDown = modifiers & Qt::ControlModifier;
  602. std::lock_guard<std::mutex> lock(selectMutex);
  603. if (altDown || ctrlDown || shiftDown) {
  604. for (size_t i = 0; i < selectedItems.size();
  605. i++) {
  606. obs_sceneitem_select(selectedItems[i],
  607. true);
  608. }
  609. }
  610. for (size_t i = 0; i < hoveredPreviewItems.size();
  611. i++) {
  612. bool select = true;
  613. obs_sceneitem_t *item = hoveredPreviewItems[i];
  614. if (altDown) {
  615. select = false;
  616. } else if (ctrlDown) {
  617. select = !obs_sceneitem_selected(item);
  618. }
  619. obs_sceneitem_select(hoveredPreviewItems[i],
  620. select);
  621. }
  622. }
  623. if (stretchGroup) {
  624. obs_sceneitem_defer_group_resize_end(stretchGroup);
  625. }
  626. stretchItem = nullptr;
  627. stretchGroup = nullptr;
  628. mouseDown = false;
  629. mouseMoved = false;
  630. cropping = false;
  631. selectionBox = false;
  632. unsetCursor();
  633. OBSSceneItem item = GetItemAtPos(pos, true);
  634. std::lock_guard<std::mutex> lock(selectMutex);
  635. hoveredPreviewItems.clear();
  636. hoveredPreviewItems.push_back(item);
  637. selectedItems.clear();
  638. }
  639. OBSBasic *main = reinterpret_cast<OBSBasic *>(App()->GetMainWindow());
  640. OBSDataAutoRelease rwrapper =
  641. obs_scene_save_transform_states(main->GetCurrentScene(), true);
  642. auto undo_redo = [](const std::string &data) {
  643. OBSDataAutoRelease dat =
  644. obs_data_create_from_json(data.c_str());
  645. OBSSourceAutoRelease source = obs_get_source_by_name(
  646. obs_data_get_string(dat, "scene_name"));
  647. reinterpret_cast<OBSBasic *>(App()->GetMainWindow())
  648. ->SetCurrentScene(source.Get(), true);
  649. obs_scene_load_transform_states(data.c_str());
  650. };
  651. if (wrapper && rwrapper) {
  652. std::string undo_data(obs_data_get_json(wrapper));
  653. std::string redo_data(obs_data_get_json(rwrapper));
  654. if (changed && undo_data.compare(redo_data) != 0)
  655. main->undo_s.add_action(
  656. QTStr("Undo.Transform")
  657. .arg(obs_source_get_name(
  658. main->GetCurrentSceneSource())),
  659. undo_redo, undo_redo, undo_data, redo_data);
  660. }
  661. wrapper = nullptr;
  662. }
  663. struct SelectedItemBounds {
  664. bool first = true;
  665. vec3 tl, br;
  666. };
  667. static bool AddItemBounds(obs_scene_t * /* scene */, obs_sceneitem_t *item,
  668. void *param)
  669. {
  670. SelectedItemBounds *data =
  671. reinterpret_cast<SelectedItemBounds *>(param);
  672. vec3 t[4];
  673. auto add_bounds = [data, &t]() {
  674. for (const vec3 &v : t) {
  675. if (data->first) {
  676. vec3_copy(&data->tl, &v);
  677. vec3_copy(&data->br, &v);
  678. data->first = false;
  679. } else {
  680. vec3_min(&data->tl, &data->tl, &v);
  681. vec3_max(&data->br, &data->br, &v);
  682. }
  683. }
  684. };
  685. if (obs_sceneitem_is_group(item)) {
  686. SelectedItemBounds sib;
  687. obs_sceneitem_group_enum_items(item, AddItemBounds, &sib);
  688. if (!sib.first) {
  689. matrix4 xform;
  690. obs_sceneitem_get_draw_transform(item, &xform);
  691. vec3_set(&t[0], sib.tl.x, sib.tl.y, 0.0f);
  692. vec3_set(&t[1], sib.tl.x, sib.br.y, 0.0f);
  693. vec3_set(&t[2], sib.br.x, sib.tl.y, 0.0f);
  694. vec3_set(&t[3], sib.br.x, sib.br.y, 0.0f);
  695. vec3_transform(&t[0], &t[0], &xform);
  696. vec3_transform(&t[1], &t[1], &xform);
  697. vec3_transform(&t[2], &t[2], &xform);
  698. vec3_transform(&t[3], &t[3], &xform);
  699. add_bounds();
  700. }
  701. }
  702. if (!obs_sceneitem_selected(item))
  703. return true;
  704. matrix4 boxTransform;
  705. obs_sceneitem_get_box_transform(item, &boxTransform);
  706. t[0] = GetTransformedPos(0.0f, 0.0f, boxTransform);
  707. t[1] = GetTransformedPos(1.0f, 0.0f, boxTransform);
  708. t[2] = GetTransformedPos(0.0f, 1.0f, boxTransform);
  709. t[3] = GetTransformedPos(1.0f, 1.0f, boxTransform);
  710. add_bounds();
  711. return true;
  712. }
  713. struct OffsetData {
  714. float clampDist;
  715. vec3 tl, br, offset;
  716. };
  717. static bool GetSourceSnapOffset(obs_scene_t * /* scene */,
  718. obs_sceneitem_t *item, void *param)
  719. {
  720. OffsetData *data = reinterpret_cast<OffsetData *>(param);
  721. if (obs_sceneitem_selected(item))
  722. return true;
  723. matrix4 boxTransform;
  724. obs_sceneitem_get_box_transform(item, &boxTransform);
  725. vec3 t[4] = {GetTransformedPos(0.0f, 0.0f, boxTransform),
  726. GetTransformedPos(1.0f, 0.0f, boxTransform),
  727. GetTransformedPos(0.0f, 1.0f, boxTransform),
  728. GetTransformedPos(1.0f, 1.0f, boxTransform)};
  729. bool first = true;
  730. vec3 tl, br;
  731. vec3_zero(&tl);
  732. vec3_zero(&br);
  733. for (const vec3 &v : t) {
  734. if (first) {
  735. vec3_copy(&tl, &v);
  736. vec3_copy(&br, &v);
  737. first = false;
  738. } else {
  739. vec3_min(&tl, &tl, &v);
  740. vec3_max(&br, &br, &v);
  741. }
  742. }
  743. // Snap to other source edges
  744. #define EDGE_SNAP(l, r, x, y) \
  745. do { \
  746. double dist = fabsf(l.x - data->r.x); \
  747. if (dist < data->clampDist && \
  748. fabsf(data->offset.x) < EPSILON && data->tl.y < br.y && \
  749. data->br.y > tl.y && \
  750. (fabsf(data->offset.x) > dist || \
  751. data->offset.x < EPSILON)) \
  752. data->offset.x = l.x - data->r.x; \
  753. } while (false)
  754. EDGE_SNAP(tl, br, x, y);
  755. EDGE_SNAP(tl, br, y, x);
  756. EDGE_SNAP(br, tl, x, y);
  757. EDGE_SNAP(br, tl, y, x);
  758. #undef EDGE_SNAP
  759. return true;
  760. }
  761. void OBSBasicPreview::SnapItemMovement(vec2 &offset)
  762. {
  763. OBSBasic *main = reinterpret_cast<OBSBasic *>(App()->GetMainWindow());
  764. OBSScene scene = main->GetCurrentScene();
  765. SelectedItemBounds data;
  766. obs_scene_enum_items(scene, AddItemBounds, &data);
  767. data.tl.x += offset.x;
  768. data.tl.y += offset.y;
  769. data.br.x += offset.x;
  770. data.br.y += offset.y;
  771. vec3 snapOffset = GetSnapOffset(data.tl, data.br);
  772. const bool snap = config_get_bool(GetGlobalConfig(), "BasicWindow",
  773. "SnappingEnabled");
  774. const bool sourcesSnap = config_get_bool(
  775. GetGlobalConfig(), "BasicWindow", "SourceSnapping");
  776. if (snap == false)
  777. return;
  778. if (sourcesSnap == false) {
  779. offset.x += snapOffset.x;
  780. offset.y += snapOffset.y;
  781. return;
  782. }
  783. const float clampDist = config_get_double(GetGlobalConfig(),
  784. "BasicWindow",
  785. "SnapDistance") /
  786. main->previewScale;
  787. OffsetData offsetData;
  788. offsetData.clampDist = clampDist;
  789. offsetData.tl = data.tl;
  790. offsetData.br = data.br;
  791. vec3_copy(&offsetData.offset, &snapOffset);
  792. obs_scene_enum_items(scene, GetSourceSnapOffset, &offsetData);
  793. if (fabsf(offsetData.offset.x) > EPSILON ||
  794. fabsf(offsetData.offset.y) > EPSILON) {
  795. offset.x += offsetData.offset.x;
  796. offset.y += offsetData.offset.y;
  797. } else {
  798. offset.x += snapOffset.x;
  799. offset.y += snapOffset.y;
  800. }
  801. }
  802. static bool move_items(obs_scene_t * /* scene */, obs_sceneitem_t *item,
  803. void *param)
  804. {
  805. if (obs_sceneitem_locked(item))
  806. return true;
  807. bool selected = obs_sceneitem_selected(item);
  808. vec2 *offset = reinterpret_cast<vec2 *>(param);
  809. if (obs_sceneitem_is_group(item) && !selected) {
  810. matrix4 transform;
  811. vec3 new_offset;
  812. vec3_set(&new_offset, offset->x, offset->y, 0.0f);
  813. obs_sceneitem_get_draw_transform(item, &transform);
  814. vec4_set(&transform.t, 0.0f, 0.0f, 0.0f, 1.0f);
  815. matrix4_inv(&transform, &transform);
  816. vec3_transform(&new_offset, &new_offset, &transform);
  817. obs_sceneitem_group_enum_items(item, move_items, &new_offset);
  818. }
  819. if (selected) {
  820. vec2 pos;
  821. obs_sceneitem_get_pos(item, &pos);
  822. vec2_add(&pos, &pos, offset);
  823. obs_sceneitem_set_pos(item, &pos);
  824. }
  825. return true;
  826. }
  827. void OBSBasicPreview::MoveItems(const vec2 &pos)
  828. {
  829. Qt::KeyboardModifiers modifiers = QGuiApplication::keyboardModifiers();
  830. OBSBasic *main = reinterpret_cast<OBSBasic *>(App()->GetMainWindow());
  831. OBSScene scene = main->GetCurrentScene();
  832. vec2 offset, moveOffset;
  833. vec2_sub(&offset, &pos, &startPos);
  834. vec2_sub(&moveOffset, &offset, &lastMoveOffset);
  835. if (!(modifiers & Qt::ControlModifier))
  836. SnapItemMovement(moveOffset);
  837. vec2_add(&lastMoveOffset, &lastMoveOffset, &moveOffset);
  838. obs_scene_enum_items(scene, move_items, &moveOffset);
  839. }
  840. static bool CounterClockwise(float x1, float x2, float x3, float y1, float y2,
  841. float y3)
  842. {
  843. return (y3 - y1) * (x2 - x1) > (y2 - y1) * (x3 - x1);
  844. }
  845. static bool IntersectLine(float x1, float x2, float x3, float x4, float y1,
  846. float y2, float y3, float y4)
  847. {
  848. bool a = CounterClockwise(x1, x2, x3, y1, y2, y3);
  849. bool b = CounterClockwise(x1, x2, x4, y1, y2, y4);
  850. bool c = CounterClockwise(x3, x4, x1, y3, y4, y1);
  851. bool d = CounterClockwise(x3, x4, x2, y3, y4, y2);
  852. return (a != b) && (c != d);
  853. }
  854. static bool IntersectBox(matrix4 transform, float x1, float x2, float y1,
  855. float y2)
  856. {
  857. float x3, x4, y3, y4;
  858. x3 = transform.t.x;
  859. y3 = transform.t.y;
  860. x4 = x3 + transform.x.x;
  861. y4 = y3 + transform.x.y;
  862. if (IntersectLine(x1, x1, x3, x4, y1, y2, y3, y4) ||
  863. IntersectLine(x1, x2, x3, x4, y1, y1, y3, y4) ||
  864. IntersectLine(x2, x2, x3, x4, y1, y2, y3, y4) ||
  865. IntersectLine(x1, x2, x3, x4, y2, y2, y3, y4))
  866. return true;
  867. x4 = x3 + transform.y.x;
  868. y4 = y3 + transform.y.y;
  869. if (IntersectLine(x1, x1, x3, x4, y1, y2, y3, y4) ||
  870. IntersectLine(x1, x2, x3, x4, y1, y1, y3, y4) ||
  871. IntersectLine(x2, x2, x3, x4, y1, y2, y3, y4) ||
  872. IntersectLine(x1, x2, x3, x4, y2, y2, y3, y4))
  873. return true;
  874. x3 = transform.t.x + transform.x.x;
  875. y3 = transform.t.y + transform.x.y;
  876. x4 = x3 + transform.y.x;
  877. y4 = y3 + transform.y.y;
  878. if (IntersectLine(x1, x1, x3, x4, y1, y2, y3, y4) ||
  879. IntersectLine(x1, x2, x3, x4, y1, y1, y3, y4) ||
  880. IntersectLine(x2, x2, x3, x4, y1, y2, y3, y4) ||
  881. IntersectLine(x1, x2, x3, x4, y2, y2, y3, y4))
  882. return true;
  883. x3 = transform.t.x + transform.y.x;
  884. y3 = transform.t.y + transform.y.y;
  885. x4 = x3 + transform.x.x;
  886. y4 = y3 + transform.x.y;
  887. if (IntersectLine(x1, x1, x3, x4, y1, y2, y3, y4) ||
  888. IntersectLine(x1, x2, x3, x4, y1, y1, y3, y4) ||
  889. IntersectLine(x2, x2, x3, x4, y1, y2, y3, y4) ||
  890. IntersectLine(x1, x2, x3, x4, y2, y2, y3, y4))
  891. return true;
  892. return false;
  893. }
  894. #undef PI
  895. bool OBSBasicPreview::FindSelected(obs_scene_t *, obs_sceneitem_t *item,
  896. void *param)
  897. {
  898. SceneFindBoxData *data = reinterpret_cast<SceneFindBoxData *>(param);
  899. if (obs_sceneitem_selected(item))
  900. data->sceneItems.push_back(item);
  901. return true;
  902. }
  903. static bool FindItemsInBox(obs_scene_t * /* scene */, obs_sceneitem_t *item,
  904. void *param)
  905. {
  906. SceneFindBoxData *data = reinterpret_cast<SceneFindBoxData *>(param);
  907. matrix4 transform;
  908. matrix4 invTransform;
  909. vec3 transformedPos;
  910. vec3 pos3;
  911. vec3 pos3_;
  912. vec2 pos_min, pos_max;
  913. vec2_min(&pos_min, &data->startPos, &data->pos);
  914. vec2_max(&pos_max, &data->startPos, &data->pos);
  915. const float x1 = pos_min.x;
  916. const float x2 = pos_max.x;
  917. const float y1 = pos_min.y;
  918. const float y2 = pos_max.y;
  919. if (!SceneItemHasVideo(item))
  920. return true;
  921. if (obs_sceneitem_locked(item))
  922. return true;
  923. if (!obs_sceneitem_visible(item))
  924. return true;
  925. vec3_set(&pos3, data->pos.x, data->pos.y, 0.0f);
  926. obs_sceneitem_get_box_transform(item, &transform);
  927. matrix4_inv(&invTransform, &transform);
  928. vec3_transform(&transformedPos, &pos3, &invTransform);
  929. vec3_transform(&pos3_, &transformedPos, &transform);
  930. if (CloseFloat(pos3.x, pos3_.x) && CloseFloat(pos3.y, pos3_.y) &&
  931. transformedPos.x >= 0.0f && transformedPos.x <= 1.0f &&
  932. transformedPos.y >= 0.0f && transformedPos.y <= 1.0f) {
  933. data->sceneItems.push_back(item);
  934. return true;
  935. }
  936. if (transform.t.x > x1 && transform.t.x < x2 && transform.t.y > y1 &&
  937. transform.t.y < y2) {
  938. data->sceneItems.push_back(item);
  939. return true;
  940. }
  941. if (transform.t.x + transform.x.x > x1 &&
  942. transform.t.x + transform.x.x < x2 &&
  943. transform.t.y + transform.x.y > y1 &&
  944. transform.t.y + transform.x.y < y2) {
  945. data->sceneItems.push_back(item);
  946. return true;
  947. }
  948. if (transform.t.x + transform.y.x > x1 &&
  949. transform.t.x + transform.y.x < x2 &&
  950. transform.t.y + transform.y.y > y1 &&
  951. transform.t.y + transform.y.y < y2) {
  952. data->sceneItems.push_back(item);
  953. return true;
  954. }
  955. if (transform.t.x + transform.x.x + transform.y.x > x1 &&
  956. transform.t.x + transform.x.x + transform.y.x < x2 &&
  957. transform.t.y + transform.x.y + transform.y.y > y1 &&
  958. transform.t.y + transform.x.y + transform.y.y < y2) {
  959. data->sceneItems.push_back(item);
  960. return true;
  961. }
  962. if (transform.t.x + 0.5 * (transform.x.x + transform.y.x) > x1 &&
  963. transform.t.x + 0.5 * (transform.x.x + transform.y.x) < x2 &&
  964. transform.t.y + 0.5 * (transform.x.y + transform.y.y) > y1 &&
  965. transform.t.y + 0.5 * (transform.x.y + transform.y.y) < y2) {
  966. data->sceneItems.push_back(item);
  967. return true;
  968. }
  969. if (IntersectBox(transform, x1, x2, y1, y2)) {
  970. data->sceneItems.push_back(item);
  971. return true;
  972. }
  973. return true;
  974. }
  975. void OBSBasicPreview::BoxItems(const vec2 &startPos, const vec2 &pos)
  976. {
  977. OBSBasic *main = reinterpret_cast<OBSBasic *>(App()->GetMainWindow());
  978. OBSScene scene = main->GetCurrentScene();
  979. if (!scene)
  980. return;
  981. if (cursor().shape() != Qt::CrossCursor)
  982. setCursor(Qt::CrossCursor);
  983. SceneFindBoxData data(startPos, pos);
  984. obs_scene_enum_items(scene, FindItemsInBox, &data);
  985. std::lock_guard<std::mutex> lock(selectMutex);
  986. hoveredPreviewItems = data.sceneItems;
  987. }
  988. vec3 OBSBasicPreview::CalculateStretchPos(const vec3 &tl, const vec3 &br)
  989. {
  990. uint32_t alignment = obs_sceneitem_get_alignment(stretchItem);
  991. vec3 pos;
  992. vec3_zero(&pos);
  993. if (alignment & OBS_ALIGN_LEFT)
  994. pos.x = tl.x;
  995. else if (alignment & OBS_ALIGN_RIGHT)
  996. pos.x = br.x;
  997. else
  998. pos.x = (br.x - tl.x) * 0.5f + tl.x;
  999. if (alignment & OBS_ALIGN_TOP)
  1000. pos.y = tl.y;
  1001. else if (alignment & OBS_ALIGN_BOTTOM)
  1002. pos.y = br.y;
  1003. else
  1004. pos.y = (br.y - tl.y) * 0.5f + tl.y;
  1005. return pos;
  1006. }
  1007. void OBSBasicPreview::ClampAspect(vec3 &tl, vec3 &br, vec2 &size,
  1008. const vec2 &baseSize)
  1009. {
  1010. float baseAspect = baseSize.x / baseSize.y;
  1011. float aspect = size.x / size.y;
  1012. uint32_t stretchFlags = (uint32_t)stretchHandle;
  1013. if (stretchHandle == ItemHandle::TopLeft ||
  1014. stretchHandle == ItemHandle::TopRight ||
  1015. stretchHandle == ItemHandle::BottomLeft ||
  1016. stretchHandle == ItemHandle::BottomRight) {
  1017. if (aspect < baseAspect) {
  1018. if ((size.y >= 0.0f && size.x >= 0.0f) ||
  1019. (size.y <= 0.0f && size.x <= 0.0f))
  1020. size.x = size.y * baseAspect;
  1021. else
  1022. size.x = size.y * baseAspect * -1.0f;
  1023. } else {
  1024. if ((size.y >= 0.0f && size.x >= 0.0f) ||
  1025. (size.y <= 0.0f && size.x <= 0.0f))
  1026. size.y = size.x / baseAspect;
  1027. else
  1028. size.y = size.x / baseAspect * -1.0f;
  1029. }
  1030. } else if (stretchHandle == ItemHandle::TopCenter ||
  1031. stretchHandle == ItemHandle::BottomCenter) {
  1032. if ((size.y >= 0.0f && size.x >= 0.0f) ||
  1033. (size.y <= 0.0f && size.x <= 0.0f))
  1034. size.x = size.y * baseAspect;
  1035. else
  1036. size.x = size.y * baseAspect * -1.0f;
  1037. } else if (stretchHandle == ItemHandle::CenterLeft ||
  1038. stretchHandle == ItemHandle::CenterRight) {
  1039. if ((size.y >= 0.0f && size.x >= 0.0f) ||
  1040. (size.y <= 0.0f && size.x <= 0.0f))
  1041. size.y = size.x / baseAspect;
  1042. else
  1043. size.y = size.x / baseAspect * -1.0f;
  1044. }
  1045. size.x = std::round(size.x);
  1046. size.y = std::round(size.y);
  1047. if (stretchFlags & ITEM_LEFT)
  1048. tl.x = br.x - size.x;
  1049. else if (stretchFlags & ITEM_RIGHT)
  1050. br.x = tl.x + size.x;
  1051. if (stretchFlags & ITEM_TOP)
  1052. tl.y = br.y - size.y;
  1053. else if (stretchFlags & ITEM_BOTTOM)
  1054. br.y = tl.y + size.y;
  1055. }
  1056. void OBSBasicPreview::SnapStretchingToScreen(vec3 &tl, vec3 &br)
  1057. {
  1058. uint32_t stretchFlags = (uint32_t)stretchHandle;
  1059. vec3 newTL = GetTransformedPos(tl.x, tl.y, itemToScreen);
  1060. vec3 newTR = GetTransformedPos(br.x, tl.y, itemToScreen);
  1061. vec3 newBL = GetTransformedPos(tl.x, br.y, itemToScreen);
  1062. vec3 newBR = GetTransformedPos(br.x, br.y, itemToScreen);
  1063. vec3 boundingTL;
  1064. vec3 boundingBR;
  1065. vec3_copy(&boundingTL, &newTL);
  1066. vec3_min(&boundingTL, &boundingTL, &newTR);
  1067. vec3_min(&boundingTL, &boundingTL, &newBL);
  1068. vec3_min(&boundingTL, &boundingTL, &newBR);
  1069. vec3_copy(&boundingBR, &newTL);
  1070. vec3_max(&boundingBR, &boundingBR, &newTR);
  1071. vec3_max(&boundingBR, &boundingBR, &newBL);
  1072. vec3_max(&boundingBR, &boundingBR, &newBR);
  1073. vec3 offset = GetSnapOffset(boundingTL, boundingBR);
  1074. vec3_add(&offset, &offset, &newTL);
  1075. vec3_transform(&offset, &offset, &screenToItem);
  1076. vec3_sub(&offset, &offset, &tl);
  1077. if (stretchFlags & ITEM_LEFT)
  1078. tl.x += offset.x;
  1079. else if (stretchFlags & ITEM_RIGHT)
  1080. br.x += offset.x;
  1081. if (stretchFlags & ITEM_TOP)
  1082. tl.y += offset.y;
  1083. else if (stretchFlags & ITEM_BOTTOM)
  1084. br.y += offset.y;
  1085. }
  1086. static float maxfunc(float x, float y)
  1087. {
  1088. return x > y ? x : y;
  1089. }
  1090. static float minfunc(float x, float y)
  1091. {
  1092. return x < y ? x : y;
  1093. }
  1094. void OBSBasicPreview::CropItem(const vec2 &pos)
  1095. {
  1096. obs_bounds_type boundsType = obs_sceneitem_get_bounds_type(stretchItem);
  1097. uint32_t stretchFlags = (uint32_t)stretchHandle;
  1098. uint32_t align = obs_sceneitem_get_alignment(stretchItem);
  1099. vec3 tl, br, pos3;
  1100. vec3_zero(&tl);
  1101. vec3_set(&br, stretchItemSize.x, stretchItemSize.y, 0.0f);
  1102. vec3_set(&pos3, pos.x, pos.y, 0.0f);
  1103. vec3_transform(&pos3, &pos3, &screenToItem);
  1104. obs_sceneitem_crop crop = startCrop;
  1105. vec2 scale;
  1106. obs_sceneitem_get_scale(stretchItem, &scale);
  1107. vec2 max_tl;
  1108. vec2 max_br;
  1109. vec2_set(&max_tl, float(-crop.left) * scale.x,
  1110. float(-crop.top) * scale.y);
  1111. vec2_set(&max_br, stretchItemSize.x + crop.right * scale.x,
  1112. stretchItemSize.y + crop.bottom * scale.y);
  1113. typedef std::function<float(float, float)> minmax_func_t;
  1114. minmax_func_t min_x = scale.x < 0.0f ? maxfunc : minfunc;
  1115. minmax_func_t min_y = scale.y < 0.0f ? maxfunc : minfunc;
  1116. minmax_func_t max_x = scale.x < 0.0f ? minfunc : maxfunc;
  1117. minmax_func_t max_y = scale.y < 0.0f ? minfunc : maxfunc;
  1118. pos3.x = min_x(pos3.x, max_br.x);
  1119. pos3.x = max_x(pos3.x, max_tl.x);
  1120. pos3.y = min_y(pos3.y, max_br.y);
  1121. pos3.y = max_y(pos3.y, max_tl.y);
  1122. if (stretchFlags & ITEM_LEFT) {
  1123. float maxX = stretchItemSize.x - (2.0 * scale.x);
  1124. pos3.x = tl.x = min_x(pos3.x, maxX);
  1125. } else if (stretchFlags & ITEM_RIGHT) {
  1126. float minX = (2.0 * scale.x);
  1127. pos3.x = br.x = max_x(pos3.x, minX);
  1128. }
  1129. if (stretchFlags & ITEM_TOP) {
  1130. float maxY = stretchItemSize.y - (2.0 * scale.y);
  1131. pos3.y = tl.y = min_y(pos3.y, maxY);
  1132. } else if (stretchFlags & ITEM_BOTTOM) {
  1133. float minY = (2.0 * scale.y);
  1134. pos3.y = br.y = max_y(pos3.y, minY);
  1135. }
  1136. #define ALIGN_X (ITEM_LEFT | ITEM_RIGHT)
  1137. #define ALIGN_Y (ITEM_TOP | ITEM_BOTTOM)
  1138. vec3 newPos;
  1139. vec3_zero(&newPos);
  1140. uint32_t align_x = (align & ALIGN_X);
  1141. uint32_t align_y = (align & ALIGN_Y);
  1142. if (align_x == (stretchFlags & ALIGN_X) && align_x != 0)
  1143. newPos.x = pos3.x;
  1144. else if (align & ITEM_RIGHT)
  1145. newPos.x = stretchItemSize.x;
  1146. else if (!(align & ITEM_LEFT))
  1147. newPos.x = stretchItemSize.x * 0.5f;
  1148. if (align_y == (stretchFlags & ALIGN_Y) && align_y != 0)
  1149. newPos.y = pos3.y;
  1150. else if (align & ITEM_BOTTOM)
  1151. newPos.y = stretchItemSize.y;
  1152. else if (!(align & ITEM_TOP))
  1153. newPos.y = stretchItemSize.y * 0.5f;
  1154. #undef ALIGN_X
  1155. #undef ALIGN_Y
  1156. crop = startCrop;
  1157. if (stretchFlags & ITEM_LEFT)
  1158. crop.left += int(std::round(tl.x / scale.x));
  1159. else if (stretchFlags & ITEM_RIGHT)
  1160. crop.right +=
  1161. int(std::round((stretchItemSize.x - br.x) / scale.x));
  1162. if (stretchFlags & ITEM_TOP)
  1163. crop.top += int(std::round(tl.y / scale.y));
  1164. else if (stretchFlags & ITEM_BOTTOM)
  1165. crop.bottom +=
  1166. int(std::round((stretchItemSize.y - br.y) / scale.y));
  1167. vec3_transform(&newPos, &newPos, &itemToScreen);
  1168. newPos.x = std::round(newPos.x);
  1169. newPos.y = std::round(newPos.y);
  1170. #if 0
  1171. vec3 curPos;
  1172. vec3_zero(&curPos);
  1173. obs_sceneitem_get_pos(stretchItem, (vec2*)&curPos);
  1174. blog(LOG_DEBUG, "curPos {%d, %d} - newPos {%d, %d}",
  1175. int(curPos.x), int(curPos.y),
  1176. int(newPos.x), int(newPos.y));
  1177. blog(LOG_DEBUG, "crop {%d, %d, %d, %d}",
  1178. crop.left, crop.top,
  1179. crop.right, crop.bottom);
  1180. #endif
  1181. obs_sceneitem_defer_update_begin(stretchItem);
  1182. obs_sceneitem_set_crop(stretchItem, &crop);
  1183. if (boundsType == OBS_BOUNDS_NONE)
  1184. obs_sceneitem_set_pos(stretchItem, (vec2 *)&newPos);
  1185. obs_sceneitem_defer_update_end(stretchItem);
  1186. }
  1187. void OBSBasicPreview::StretchItem(const vec2 &pos)
  1188. {
  1189. Qt::KeyboardModifiers modifiers = QGuiApplication::keyboardModifiers();
  1190. obs_bounds_type boundsType = obs_sceneitem_get_bounds_type(stretchItem);
  1191. uint32_t stretchFlags = (uint32_t)stretchHandle;
  1192. bool shiftDown = (modifiers & Qt::ShiftModifier);
  1193. vec3 tl, br, pos3;
  1194. vec3_zero(&tl);
  1195. vec3_set(&br, stretchItemSize.x, stretchItemSize.y, 0.0f);
  1196. vec3_set(&pos3, pos.x, pos.y, 0.0f);
  1197. vec3_transform(&pos3, &pos3, &screenToItem);
  1198. if (stretchFlags & ITEM_LEFT)
  1199. tl.x = pos3.x;
  1200. else if (stretchFlags & ITEM_RIGHT)
  1201. br.x = pos3.x;
  1202. if (stretchFlags & ITEM_TOP)
  1203. tl.y = pos3.y;
  1204. else if (stretchFlags & ITEM_BOTTOM)
  1205. br.y = pos3.y;
  1206. if (!(modifiers & Qt::ControlModifier))
  1207. SnapStretchingToScreen(tl, br);
  1208. obs_source_t *source = obs_sceneitem_get_source(stretchItem);
  1209. uint32_t source_cx = obs_source_get_width(source);
  1210. uint32_t source_cy = obs_source_get_height(source);
  1211. /* if the source's internal size has been set to 0 for whatever reason
  1212. * while resizing, do not update transform, otherwise source will be
  1213. * stuck invisible until a complete transform reset */
  1214. if (!source_cx || !source_cy)
  1215. return;
  1216. vec2 baseSize;
  1217. vec2_set(&baseSize, float(source_cx), float(source_cy));
  1218. vec2 size;
  1219. vec2_set(&size, br.x - tl.x, br.y - tl.y);
  1220. if (boundsType != OBS_BOUNDS_NONE) {
  1221. if (shiftDown)
  1222. ClampAspect(tl, br, size, baseSize);
  1223. if (tl.x > br.x)
  1224. std::swap(tl.x, br.x);
  1225. if (tl.y > br.y)
  1226. std::swap(tl.y, br.y);
  1227. vec2_abs(&size, &size);
  1228. obs_sceneitem_set_bounds(stretchItem, &size);
  1229. } else {
  1230. obs_sceneitem_crop crop;
  1231. obs_sceneitem_get_crop(stretchItem, &crop);
  1232. baseSize.x -= float(crop.left + crop.right);
  1233. baseSize.y -= float(crop.top + crop.bottom);
  1234. if (!shiftDown)
  1235. ClampAspect(tl, br, size, baseSize);
  1236. vec2_div(&size, &size, &baseSize);
  1237. obs_sceneitem_set_scale(stretchItem, &size);
  1238. }
  1239. pos3 = CalculateStretchPos(tl, br);
  1240. vec3_transform(&pos3, &pos3, &itemToScreen);
  1241. vec2 newPos;
  1242. vec2_set(&newPos, std::round(pos3.x), std::round(pos3.y));
  1243. obs_sceneitem_set_pos(stretchItem, &newPos);
  1244. }
  1245. void OBSBasicPreview::RotateItem(const vec2 &pos)
  1246. {
  1247. OBSBasic *main = reinterpret_cast<OBSBasic *>(App()->GetMainWindow());
  1248. OBSScene scene = main->GetCurrentScene();
  1249. Qt::KeyboardModifiers modifiers = QGuiApplication::keyboardModifiers();
  1250. bool shiftDown = (modifiers & Qt::ShiftModifier);
  1251. bool ctrlDown = (modifiers & Qt::ControlModifier);
  1252. vec2 pos2;
  1253. vec2_copy(&pos2, &pos);
  1254. float angle =
  1255. atan2(pos2.y - rotatePoint.y, pos2.x - rotatePoint.x) + RAD(90);
  1256. #define ROT_SNAP(rot, thresh) \
  1257. if (abs(angle - RAD(rot)) < RAD(thresh)) { \
  1258. angle = RAD(rot); \
  1259. }
  1260. if (shiftDown) {
  1261. for (int i = 0; i <= 360 / 15; i++) {
  1262. ROT_SNAP(i * 15 - 90, 7.5);
  1263. }
  1264. } else if (!ctrlDown) {
  1265. ROT_SNAP(rotateAngle, 5)
  1266. ROT_SNAP(-90, 5)
  1267. ROT_SNAP(-45, 5)
  1268. ROT_SNAP(0, 5)
  1269. ROT_SNAP(45, 5)
  1270. ROT_SNAP(90, 5)
  1271. ROT_SNAP(135, 5)
  1272. ROT_SNAP(180, 5)
  1273. ROT_SNAP(225, 5)
  1274. ROT_SNAP(270, 5)
  1275. ROT_SNAP(315, 5)
  1276. }
  1277. #undef ROT_SNAP
  1278. vec2 pos3;
  1279. vec2_copy(&pos3, &offsetPoint);
  1280. RotatePos(&pos3, angle);
  1281. pos3.x += rotatePoint.x;
  1282. pos3.y += rotatePoint.y;
  1283. obs_sceneitem_set_rot(stretchItem, DEG(angle));
  1284. obs_sceneitem_set_pos(stretchItem, &pos3);
  1285. }
  1286. void OBSBasicPreview::mouseMoveEvent(QMouseEvent *event)
  1287. {
  1288. OBSBasic *main = reinterpret_cast<OBSBasic *>(App()->GetMainWindow());
  1289. changed = true;
  1290. #if QT_VERSION >= QT_VERSION_CHECK(6, 0, 0)
  1291. QPointF qtPos = event->position();
  1292. #else
  1293. QPointF qtPos = event->localPos();
  1294. #endif
  1295. float pixelRatio = main->GetDevicePixelRatio();
  1296. if (scrollMode && event->buttons() == Qt::LeftButton) {
  1297. scrollingOffset.x += pixelRatio * (qtPos.x() - scrollingFrom.x);
  1298. scrollingOffset.y += pixelRatio * (qtPos.y() - scrollingFrom.y);
  1299. scrollingFrom.x = qtPos.x();
  1300. scrollingFrom.y = qtPos.y();
  1301. emit DisplayResized();
  1302. return;
  1303. }
  1304. if (locked)
  1305. return;
  1306. bool updateCursor = false;
  1307. if (mouseDown) {
  1308. vec2 pos = GetMouseEventPos(event);
  1309. if (!mouseMoved && !mouseOverItems &&
  1310. stretchHandle == ItemHandle::None) {
  1311. ProcessClick(startPos);
  1312. mouseOverItems = SelectedAtPos(startPos);
  1313. }
  1314. pos.x = std::round(pos.x);
  1315. pos.y = std::round(pos.y);
  1316. if (stretchHandle != ItemHandle::None) {
  1317. if (obs_sceneitem_locked(stretchItem))
  1318. return;
  1319. selectionBox = false;
  1320. OBSScene scene = main->GetCurrentScene();
  1321. obs_sceneitem_t *group =
  1322. obs_sceneitem_get_group(scene, stretchItem);
  1323. if (group) {
  1324. vec3 group_pos;
  1325. vec3_set(&group_pos, pos.x, pos.y, 0.0f);
  1326. vec3_transform(&group_pos, &group_pos,
  1327. &invGroupTransform);
  1328. pos.x = group_pos.x;
  1329. pos.y = group_pos.y;
  1330. }
  1331. if (stretchHandle == ItemHandle::Rot) {
  1332. RotateItem(pos);
  1333. setCursor(Qt::ClosedHandCursor);
  1334. } else if (cropping)
  1335. CropItem(pos);
  1336. else
  1337. StretchItem(pos);
  1338. } else if (mouseOverItems) {
  1339. if (cursor().shape() != Qt::SizeAllCursor)
  1340. setCursor(Qt::SizeAllCursor);
  1341. selectionBox = false;
  1342. MoveItems(pos);
  1343. } else {
  1344. selectionBox = true;
  1345. if (!mouseMoved)
  1346. DoSelect(startPos);
  1347. BoxItems(startPos, pos);
  1348. }
  1349. mouseMoved = true;
  1350. mousePos = pos;
  1351. } else {
  1352. vec2 pos = GetMouseEventPos(event);
  1353. OBSSceneItem item = GetItemAtPos(pos, true);
  1354. std::lock_guard<std::mutex> lock(selectMutex);
  1355. hoveredPreviewItems.clear();
  1356. hoveredPreviewItems.push_back(item);
  1357. if (!mouseMoved && hoveredPreviewItems.size() > 0) {
  1358. mousePos = pos;
  1359. OBSBasic *main = reinterpret_cast<OBSBasic *>(
  1360. App()->GetMainWindow());
  1361. float scale = main->GetDevicePixelRatio();
  1362. float x = qtPos.x() - main->previewX / scale;
  1363. float y = qtPos.y() - main->previewY / scale;
  1364. vec2_set(&startPos, x, y);
  1365. updateCursor = true;
  1366. }
  1367. }
  1368. if (updateCursor) {
  1369. GetStretchHandleData(startPos, true);
  1370. uint32_t stretchFlags = (uint32_t)stretchHandle;
  1371. UpdateCursor(stretchFlags);
  1372. }
  1373. }
  1374. void OBSBasicPreview::leaveEvent(QEvent *)
  1375. {
  1376. std::lock_guard<std::mutex> lock(selectMutex);
  1377. if (!selectionBox)
  1378. hoveredPreviewItems.clear();
  1379. }
  1380. static void DrawLine(float x1, float y1, float x2, float y2, float thickness,
  1381. vec2 scale)
  1382. {
  1383. float ySide = (y1 == y2) ? (y1 < 0.5f ? 1.0f : -1.0f) : 0.0f;
  1384. float xSide = (x1 == x2) ? (x1 < 0.5f ? 1.0f : -1.0f) : 0.0f;
  1385. gs_render_start(true);
  1386. gs_vertex2f(x1, y1);
  1387. gs_vertex2f(x1 + (xSide * (thickness / scale.x)),
  1388. y1 + (ySide * (thickness / scale.y)));
  1389. gs_vertex2f(x2 + (xSide * (thickness / scale.x)),
  1390. y2 + (ySide * (thickness / scale.y)));
  1391. gs_vertex2f(x2, y2);
  1392. gs_vertex2f(x1, y1);
  1393. gs_vertbuffer_t *line = gs_render_save();
  1394. gs_load_vertexbuffer(line);
  1395. gs_draw(GS_TRISTRIP, 0, 0);
  1396. gs_vertexbuffer_destroy(line);
  1397. }
  1398. static void DrawSquareAtPos(float x, float y, float pixelRatio)
  1399. {
  1400. struct vec3 pos;
  1401. vec3_set(&pos, x, y, 0.0f);
  1402. struct matrix4 matrix;
  1403. gs_matrix_get(&matrix);
  1404. vec3_transform(&pos, &pos, &matrix);
  1405. gs_matrix_push();
  1406. gs_matrix_identity();
  1407. gs_matrix_translate(&pos);
  1408. gs_matrix_translate3f(-HANDLE_RADIUS * pixelRatio,
  1409. -HANDLE_RADIUS * pixelRatio, 0.0f);
  1410. gs_matrix_scale3f(HANDLE_RADIUS * pixelRatio * 2,
  1411. HANDLE_RADIUS * pixelRatio * 2, 1.0f);
  1412. gs_draw(GS_TRISTRIP, 0, 0);
  1413. gs_matrix_pop();
  1414. }
  1415. static void DrawRotationHandle(gs_vertbuffer_t *circle, float rot,
  1416. float pixelRatio, bool invert)
  1417. {
  1418. struct vec3 pos;
  1419. vec3_set(&pos, 0.5f, invert ? 1.0f : 0.0f, 0.0f);
  1420. struct matrix4 matrix;
  1421. gs_matrix_get(&matrix);
  1422. vec3_transform(&pos, &pos, &matrix);
  1423. gs_render_start(true);
  1424. gs_vertex2f(0.5f - 0.34f / HANDLE_RADIUS, 0.5f);
  1425. gs_vertex2f(0.5f - 0.34f / HANDLE_RADIUS, -2.0f);
  1426. gs_vertex2f(0.5f + 0.34f / HANDLE_RADIUS, -2.0f);
  1427. gs_vertex2f(0.5f + 0.34f / HANDLE_RADIUS, 0.5f);
  1428. gs_vertex2f(0.5f - 0.34f / HANDLE_RADIUS, 0.5f);
  1429. gs_vertbuffer_t *line = gs_render_save();
  1430. gs_load_vertexbuffer(line);
  1431. gs_matrix_push();
  1432. gs_matrix_identity();
  1433. gs_matrix_translate(&pos);
  1434. gs_matrix_rotaa4f(0.0f, 0.0f, 1.0f, RAD(rot));
  1435. gs_matrix_translate3f(-HANDLE_RADIUS * 1.5 * pixelRatio,
  1436. -HANDLE_RADIUS * 1.5 * pixelRatio, 0.0f);
  1437. gs_matrix_scale3f(HANDLE_RADIUS * 3 * pixelRatio,
  1438. HANDLE_RADIUS * 3 * pixelRatio, 1.0f);
  1439. gs_draw(GS_TRISTRIP, 0, 0);
  1440. gs_matrix_translate3f(0.0f, -HANDLE_RADIUS * 2 / 3, 0.0f);
  1441. gs_load_vertexbuffer(circle);
  1442. gs_draw(GS_TRISTRIP, 0, 0);
  1443. gs_matrix_pop();
  1444. gs_vertexbuffer_destroy(line);
  1445. }
  1446. static void DrawStripedLine(float x1, float y1, float x2, float y2,
  1447. float thickness, vec2 scale)
  1448. {
  1449. float ySide = (y1 == y2) ? (y1 < 0.5f ? 1.0f : -1.0f) : 0.0f;
  1450. float xSide = (x1 == x2) ? (x1 < 0.5f ? 1.0f : -1.0f) : 0.0f;
  1451. float dist =
  1452. sqrt(pow((x1 - x2) * scale.x, 2) + pow((y1 - y2) * scale.y, 2));
  1453. float offX = (x2 - x1) / dist;
  1454. float offY = (y2 - y1) / dist;
  1455. for (int i = 0, l = ceil(dist / 15); i < l; i++) {
  1456. gs_render_start(true);
  1457. float xx1 = x1 + i * 15 * offX;
  1458. float yy1 = y1 + i * 15 * offY;
  1459. float dx;
  1460. float dy;
  1461. if (x1 < x2) {
  1462. dx = std::min(xx1 + 7.5f * offX, x2);
  1463. } else {
  1464. dx = std::max(xx1 + 7.5f * offX, x2);
  1465. }
  1466. if (y1 < y2) {
  1467. dy = std::min(yy1 + 7.5f * offY, y2);
  1468. } else {
  1469. dy = std::max(yy1 + 7.5f * offY, y2);
  1470. }
  1471. gs_vertex2f(xx1, yy1);
  1472. gs_vertex2f(xx1 + (xSide * (thickness / scale.x)),
  1473. yy1 + (ySide * (thickness / scale.y)));
  1474. gs_vertex2f(dx, dy);
  1475. gs_vertex2f(dx + (xSide * (thickness / scale.x)),
  1476. dy + (ySide * (thickness / scale.y)));
  1477. gs_vertbuffer_t *line = gs_render_save();
  1478. gs_load_vertexbuffer(line);
  1479. gs_draw(GS_TRISTRIP, 0, 0);
  1480. gs_vertexbuffer_destroy(line);
  1481. }
  1482. }
  1483. static void DrawRect(float thickness, vec2 scale)
  1484. {
  1485. gs_render_start(true);
  1486. gs_vertex2f(0.0f, 0.0f);
  1487. gs_vertex2f(0.0f + (thickness / scale.x), 0.0f);
  1488. gs_vertex2f(0.0f, 1.0f);
  1489. gs_vertex2f(0.0f + (thickness / scale.x), 1.0f);
  1490. gs_vertex2f(0.0f, 1.0f - (thickness / scale.y));
  1491. gs_vertex2f(1.0f, 1.0f);
  1492. gs_vertex2f(1.0f, 1.0f - (thickness / scale.y));
  1493. gs_vertex2f(1.0f - (thickness / scale.x), 1.0f);
  1494. gs_vertex2f(1.0f, 0.0f);
  1495. gs_vertex2f(1.0f - (thickness / scale.x), 0.0f);
  1496. gs_vertex2f(1.0f, 0.0f + (thickness / scale.y));
  1497. gs_vertex2f(0.0f, 0.0f);
  1498. gs_vertex2f(0.0f, 0.0f + (thickness / scale.y));
  1499. gs_vertbuffer_t *rect = gs_render_save();
  1500. gs_load_vertexbuffer(rect);
  1501. gs_draw(GS_TRISTRIP, 0, 0);
  1502. gs_vertexbuffer_destroy(rect);
  1503. }
  1504. static inline bool crop_enabled(const obs_sceneitem_crop *crop)
  1505. {
  1506. return crop->left > 0 || crop->top > 0 || crop->right > 0 ||
  1507. crop->bottom > 0;
  1508. }
  1509. bool OBSBasicPreview::DrawSelectedOverflow(obs_scene_t *, obs_sceneitem_t *item,
  1510. void *param)
  1511. {
  1512. if (obs_sceneitem_locked(item))
  1513. return true;
  1514. if (!SceneItemHasVideo(item))
  1515. return true;
  1516. OBSBasicPreview *prev = reinterpret_cast<OBSBasicPreview *>(param);
  1517. if (!prev->GetOverflowSelectionHidden() && !obs_sceneitem_visible(item))
  1518. return true;
  1519. if (obs_sceneitem_is_group(item)) {
  1520. matrix4 mat;
  1521. obs_sceneitem_get_draw_transform(item, &mat);
  1522. gs_matrix_push();
  1523. gs_matrix_mul(&mat);
  1524. obs_sceneitem_group_enum_items(item, DrawSelectedOverflow,
  1525. param);
  1526. gs_matrix_pop();
  1527. }
  1528. if (!prev->GetOverflowAlwaysVisible() && !obs_sceneitem_selected(item))
  1529. return true;
  1530. matrix4 boxTransform;
  1531. matrix4 invBoxTransform;
  1532. obs_sceneitem_get_box_transform(item, &boxTransform);
  1533. matrix4_inv(&invBoxTransform, &boxTransform);
  1534. vec3 bounds[] = {
  1535. {{{0.f, 0.f, 0.f}}},
  1536. {{{1.f, 0.f, 0.f}}},
  1537. {{{0.f, 1.f, 0.f}}},
  1538. {{{1.f, 1.f, 0.f}}},
  1539. };
  1540. bool visible = std::all_of(
  1541. std::begin(bounds), std::end(bounds), [&](const vec3 &b) {
  1542. vec3 pos;
  1543. vec3_transform(&pos, &b, &boxTransform);
  1544. vec3_transform(&pos, &pos, &invBoxTransform);
  1545. return CloseFloat(pos.x, b.x) && CloseFloat(pos.y, b.y);
  1546. });
  1547. if (!visible)
  1548. return true;
  1549. GS_DEBUG_MARKER_BEGIN(GS_DEBUG_COLOR_DEFAULT, "DrawSelectedOverflow");
  1550. obs_transform_info info;
  1551. obs_sceneitem_get_info(item, &info);
  1552. gs_effect_t *solid = obs_get_base_effect(OBS_EFFECT_REPEAT);
  1553. gs_eparam_t *image = gs_effect_get_param_by_name(solid, "image");
  1554. gs_eparam_t *scale = gs_effect_get_param_by_name(solid, "scale");
  1555. vec2 s;
  1556. vec2_set(&s, boxTransform.x.x / 96, boxTransform.y.y / 96);
  1557. gs_effect_set_vec2(scale, &s);
  1558. gs_effect_set_texture(image, prev->overflow);
  1559. gs_matrix_push();
  1560. gs_matrix_mul(&boxTransform);
  1561. obs_sceneitem_crop crop;
  1562. obs_sceneitem_get_crop(item, &crop);
  1563. while (gs_effect_loop(solid, "Draw")) {
  1564. gs_draw_sprite(prev->overflow, 0, 1, 1);
  1565. }
  1566. gs_matrix_pop();
  1567. GS_DEBUG_MARKER_END();
  1568. return true;
  1569. }
  1570. bool OBSBasicPreview::DrawSelectedItem(obs_scene_t *, obs_sceneitem_t *item,
  1571. 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->GetDevicePixelRatio();
  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, pixelRatio);
  1685. DrawSquareAtPos(0.0f, 1.0f, pixelRatio);
  1686. DrawSquareAtPos(1.0f, 0.0f, pixelRatio);
  1687. DrawSquareAtPos(1.0f, 1.0f, pixelRatio);
  1688. DrawSquareAtPos(0.5f, 0.0f, pixelRatio);
  1689. DrawSquareAtPos(0.0f, 0.5f, pixelRatio);
  1690. DrawSquareAtPos(0.5f, 1.0f, pixelRatio);
  1691. DrawSquareAtPos(1.0f, 0.5f, pixelRatio);
  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. bool invert = info.scale.y < 0.0f;
  1706. DrawRotationHandle(prev->circleFill, info.rot + prev->groupRot,
  1707. pixelRatio, invert);
  1708. }
  1709. gs_matrix_pop();
  1710. GS_DEBUG_MARKER_END();
  1711. return true;
  1712. }
  1713. bool OBSBasicPreview::DrawSelectionBox(float x1, float y1, float x2, float y2,
  1714. gs_vertbuffer_t *rectFill)
  1715. {
  1716. OBSBasic *main = OBSBasic::Get();
  1717. float pixelRatio = main->GetDevicePixelRatio();
  1718. x1 = std::round(x1);
  1719. x2 = std::round(x2);
  1720. y1 = std::round(y1);
  1721. y2 = std::round(y2);
  1722. gs_effect_t *eff = gs_get_effect();
  1723. gs_eparam_t *colParam = gs_effect_get_param_by_name(eff, "color");
  1724. vec4 fillColor;
  1725. vec4_set(&fillColor, 0.7f, 0.7f, 0.7f, 0.5f);
  1726. vec4 borderColor;
  1727. vec4_set(&borderColor, 1.0f, 1.0f, 1.0f, 1.0f);
  1728. vec2 scale;
  1729. vec2_set(&scale, std::abs(x2 - x1), std::abs(y2 - y1));
  1730. gs_matrix_push();
  1731. gs_matrix_identity();
  1732. gs_matrix_translate3f(x1, y1, 0.0f);
  1733. gs_matrix_scale3f(x2 - x1, y2 - y1, 1.0f);
  1734. gs_effect_set_vec4(colParam, &fillColor);
  1735. gs_load_vertexbuffer(rectFill);
  1736. gs_draw(GS_TRISTRIP, 0, 0);
  1737. gs_effect_set_vec4(colParam, &borderColor);
  1738. DrawRect(HANDLE_RADIUS * pixelRatio / 2, scale);
  1739. gs_matrix_pop();
  1740. return true;
  1741. }
  1742. void OBSBasicPreview::DrawOverflow()
  1743. {
  1744. if (locked)
  1745. return;
  1746. if (overflowHidden)
  1747. return;
  1748. GS_DEBUG_MARKER_BEGIN(GS_DEBUG_COLOR_DEFAULT, "DrawOverflow");
  1749. if (!overflow) {
  1750. std::string path;
  1751. GetDataFilePath("images/overflow.png", path);
  1752. overflow = gs_texture_create_from_file(path.c_str());
  1753. }
  1754. OBSBasic *main = reinterpret_cast<OBSBasic *>(App()->GetMainWindow());
  1755. OBSScene scene = main->GetCurrentScene();
  1756. if (scene) {
  1757. gs_matrix_push();
  1758. gs_matrix_scale3f(main->previewScale, main->previewScale, 1.0f);
  1759. obs_scene_enum_items(scene, DrawSelectedOverflow, this);
  1760. gs_matrix_pop();
  1761. }
  1762. gs_load_vertexbuffer(nullptr);
  1763. GS_DEBUG_MARKER_END();
  1764. }
  1765. void OBSBasicPreview::DrawSceneEditing()
  1766. {
  1767. if (locked)
  1768. return;
  1769. GS_DEBUG_MARKER_BEGIN(GS_DEBUG_COLOR_DEFAULT, "DrawSceneEditing");
  1770. OBSBasic *main = reinterpret_cast<OBSBasic *>(App()->GetMainWindow());
  1771. gs_effect_t *solid = obs_get_base_effect(OBS_EFFECT_SOLID);
  1772. gs_technique_t *tech = gs_effect_get_technique(solid, "Solid");
  1773. gs_technique_begin(tech);
  1774. gs_technique_begin_pass(tech, 0);
  1775. OBSScene scene = main->GetCurrentScene();
  1776. if (scene) {
  1777. gs_matrix_push();
  1778. gs_matrix_scale3f(main->previewScale, main->previewScale, 1.0f);
  1779. obs_scene_enum_items(scene, DrawSelectedItem, this);
  1780. gs_matrix_pop();
  1781. }
  1782. if (selectionBox) {
  1783. if (!rectFill) {
  1784. gs_render_start(true);
  1785. gs_vertex2f(0.0f, 0.0f);
  1786. gs_vertex2f(1.0f, 0.0f);
  1787. gs_vertex2f(0.0f, 1.0f);
  1788. gs_vertex2f(1.0f, 1.0f);
  1789. rectFill = gs_render_save();
  1790. }
  1791. DrawSelectionBox(startPos.x * main->previewScale,
  1792. startPos.y * main->previewScale,
  1793. mousePos.x * main->previewScale,
  1794. mousePos.y * main->previewScale, rectFill);
  1795. }
  1796. gs_load_vertexbuffer(nullptr);
  1797. gs_technique_end_pass(tech);
  1798. gs_technique_end(tech);
  1799. GS_DEBUG_MARKER_END();
  1800. }
  1801. void OBSBasicPreview::ResetScrollingOffset()
  1802. {
  1803. vec2_zero(&scrollingOffset);
  1804. }
  1805. void OBSBasicPreview::SetScalingLevel(int32_t newScalingLevelVal)
  1806. {
  1807. newScalingLevelVal = std::clamp(newScalingLevelVal, -MAX_SCALING_LEVEL,
  1808. MAX_SCALING_LEVEL);
  1809. float newScalingAmountVal =
  1810. pow(ZOOM_SENSITIVITY, float(newScalingLevelVal));
  1811. scalingLevel = newScalingLevelVal;
  1812. SetScalingAmount(newScalingAmountVal);
  1813. }
  1814. void OBSBasicPreview::SetScalingAmount(float newScalingAmountVal)
  1815. {
  1816. scrollingOffset.x *= newScalingAmountVal / scalingAmount;
  1817. scrollingOffset.y *= newScalingAmountVal / scalingAmount;
  1818. scalingAmount = newScalingAmountVal;
  1819. }
  1820. OBSBasicPreview *OBSBasicPreview::Get()
  1821. {
  1822. return OBSBasic::Get()->ui->preview;
  1823. }
  1824. static obs_source_t *CreateLabel(float pixelRatio)
  1825. {
  1826. OBSDataAutoRelease settings = obs_data_create();
  1827. OBSDataAutoRelease font = obs_data_create();
  1828. #if defined(_WIN32)
  1829. obs_data_set_string(font, "face", "Arial");
  1830. #elif defined(__APPLE__)
  1831. obs_data_set_string(font, "face", "Helvetica");
  1832. #else
  1833. obs_data_set_string(font, "face", "Monospace");
  1834. #endif
  1835. obs_data_set_int(font, "flags", 1); // Bold text
  1836. obs_data_set_int(font, "size", 16 * pixelRatio);
  1837. obs_data_set_obj(settings, "font", font);
  1838. obs_data_set_bool(settings, "outline", true);
  1839. #ifdef _WIN32
  1840. obs_data_set_int(settings, "outline_color", 0x000000);
  1841. obs_data_set_int(settings, "outline_size", 3);
  1842. const char *text_source_id = "text_gdiplus";
  1843. #else
  1844. const char *text_source_id = "text_ft2_source";
  1845. #endif
  1846. return obs_source_create_private(text_source_id, NULL, settings);
  1847. }
  1848. static void SetLabelText(int sourceIndex, int px)
  1849. {
  1850. OBSBasicPreview *prev = OBSBasicPreview::Get();
  1851. if (px == prev->spacerPx[sourceIndex])
  1852. return;
  1853. std::string text = std::to_string(px) + " px";
  1854. obs_source_t *source = prev->spacerLabel[sourceIndex];
  1855. OBSDataAutoRelease settings = obs_source_get_settings(source);
  1856. obs_data_set_string(settings, "text", text.c_str());
  1857. obs_source_update(source, settings);
  1858. prev->spacerPx[sourceIndex] = px;
  1859. }
  1860. static void DrawLabel(OBSSource source, vec3 &pos, vec3 &viewport)
  1861. {
  1862. if (!source)
  1863. return;
  1864. vec3_mul(&pos, &pos, &viewport);
  1865. gs_matrix_push();
  1866. gs_matrix_identity();
  1867. gs_matrix_translate(&pos);
  1868. obs_source_video_render(source);
  1869. gs_matrix_pop();
  1870. }
  1871. static void DrawSpacingLine(vec3 &start, vec3 &end, vec3 &viewport,
  1872. float pixelRatio)
  1873. {
  1874. OBSBasic *main = OBSBasic::Get();
  1875. matrix4 transform;
  1876. matrix4_identity(&transform);
  1877. transform.x.x = viewport.x;
  1878. transform.y.y = viewport.y;
  1879. gs_effect_t *solid = obs_get_base_effect(OBS_EFFECT_SOLID);
  1880. gs_technique_t *tech = gs_effect_get_technique(solid, "Solid");
  1881. QColor selColor = main->GetSelectionColor();
  1882. vec4 color;
  1883. vec4_set(&color, selColor.redF(), selColor.greenF(), selColor.blueF(),
  1884. 1.0f);
  1885. gs_effect_set_vec4(gs_effect_get_param_by_name(solid, "color"), &color);
  1886. gs_technique_begin(tech);
  1887. gs_technique_begin_pass(tech, 0);
  1888. gs_matrix_push();
  1889. gs_matrix_mul(&transform);
  1890. vec2 scale;
  1891. vec2_set(&scale, viewport.x, viewport.y);
  1892. DrawLine(start.x, start.y, end.x, end.y,
  1893. pixelRatio * (HANDLE_RADIUS / 2), scale);
  1894. gs_matrix_pop();
  1895. gs_load_vertexbuffer(nullptr);
  1896. gs_technique_end_pass(tech);
  1897. gs_technique_end(tech);
  1898. }
  1899. static void RenderSpacingHelper(int sourceIndex, vec3 &start, vec3 &end,
  1900. vec3 &viewport, float pixelRatio)
  1901. {
  1902. bool horizontal = (sourceIndex == 2 || sourceIndex == 3);
  1903. // If outside of preview, don't render
  1904. if (!((horizontal && (end.x >= start.x)) ||
  1905. (!horizontal && (end.y >= start.y))))
  1906. return;
  1907. float length = vec3_dist(&start, &end);
  1908. obs_video_info ovi;
  1909. obs_get_video_info(&ovi);
  1910. float px;
  1911. if (horizontal) {
  1912. px = length * ovi.base_width;
  1913. } else {
  1914. px = length * ovi.base_height;
  1915. }
  1916. if (px <= 0.0f)
  1917. return;
  1918. OBSBasicPreview *prev = OBSBasicPreview::Get();
  1919. obs_source_t *source = prev->spacerLabel[sourceIndex];
  1920. vec3 labelSize, labelPos;
  1921. vec3_set(&labelSize, obs_source_get_width(source),
  1922. obs_source_get_height(source), 1.0f);
  1923. vec3_div(&labelSize, &labelSize, &viewport);
  1924. vec3 labelMargin;
  1925. vec3_set(&labelMargin, SPACER_LABEL_MARGIN * pixelRatio,
  1926. SPACER_LABEL_MARGIN * pixelRatio, 1.0f);
  1927. vec3_div(&labelMargin, &labelMargin, &viewport);
  1928. vec3_set(&labelPos, end.x, end.y, end.z);
  1929. if (horizontal) {
  1930. labelPos.x -= (end.x - start.x) / 2;
  1931. labelPos.x -= labelSize.x / 2;
  1932. labelPos.y -= labelMargin.y + (labelSize.y / 2) +
  1933. (HANDLE_RADIUS / viewport.y);
  1934. } else {
  1935. labelPos.y -= (end.y - start.y) / 2;
  1936. labelPos.y -= labelSize.y / 2;
  1937. labelPos.x += labelMargin.x;
  1938. }
  1939. DrawSpacingLine(start, end, viewport, pixelRatio);
  1940. SetLabelText(sourceIndex, (int)px);
  1941. DrawLabel(source, labelPos, viewport);
  1942. }
  1943. void OBSBasicPreview::DrawSpacingHelpers()
  1944. {
  1945. if (locked)
  1946. return;
  1947. OBSBasic *main = OBSBasic::Get();
  1948. vec2 s;
  1949. SceneFindBoxData data(s, s);
  1950. obs_scene_enum_items(main->GetCurrentScene(), FindSelected, &data);
  1951. if (data.sceneItems.size() > 1)
  1952. return;
  1953. OBSSceneItem item = main->GetCurrentSceneItem();
  1954. if (!item)
  1955. return;
  1956. if (obs_sceneitem_locked(item))
  1957. return;
  1958. vec2 itemSize = GetItemSize(item);
  1959. if (itemSize.x == 0.0f || itemSize.y == 0.0f)
  1960. return;
  1961. obs_sceneitem_t *parentGroup =
  1962. obs_sceneitem_get_group(main->GetCurrentScene(), item);
  1963. if (parentGroup && obs_sceneitem_locked(parentGroup))
  1964. return;
  1965. matrix4 boxTransform;
  1966. obs_sceneitem_get_box_transform(item, &boxTransform);
  1967. obs_transform_info oti;
  1968. obs_sceneitem_get_info(item, &oti);
  1969. obs_video_info ovi;
  1970. obs_get_video_info(&ovi);
  1971. vec3 size;
  1972. vec3_set(&size, ovi.base_width, ovi.base_height, 1.0f);
  1973. // Init box transform side locations
  1974. vec3 left, right, top, bottom;
  1975. vec3_set(&left, 0.0f, 0.5f, 1.0f);
  1976. vec3_set(&right, 1.0f, 0.5f, 1.0f);
  1977. vec3_set(&top, 0.5f, 0.0f, 1.0f);
  1978. vec3_set(&bottom, 0.5f, 1.0f, 1.0f);
  1979. // Decide which side to use with box transform, based on rotation
  1980. // Seems hacky, probably a better way to do it
  1981. float rot = oti.rot;
  1982. if (parentGroup) {
  1983. obs_transform_info groupOti;
  1984. obs_sceneitem_get_info(parentGroup, &groupOti);
  1985. //Correct the scene item rotation angle
  1986. rot = oti.rot + groupOti.rot;
  1987. // Correct the scene item box transform
  1988. // Based on scale, rotation angle, position of parent's group
  1989. matrix4_scale3f(&boxTransform, &boxTransform, groupOti.scale.x,
  1990. groupOti.scale.y, 1.0f);
  1991. matrix4_rotate_aa4f(&boxTransform, &boxTransform, 0.0f, 0.0f,
  1992. 1.0f, RAD(groupOti.rot));
  1993. matrix4_translate3f(&boxTransform, &boxTransform,
  1994. groupOti.pos.x, groupOti.pos.y, 0.0f);
  1995. }
  1996. // Switch top/bottom or right/left if scale is negative
  1997. if (oti.scale.x < 0.0f) {
  1998. vec3 l = left;
  1999. vec3 r = right;
  2000. vec3_copy(&left, &r);
  2001. vec3_copy(&right, &l);
  2002. }
  2003. if (oti.scale.y < 0.0f) {
  2004. vec3 t = top;
  2005. vec3 b = bottom;
  2006. vec3_copy(&top, &b);
  2007. vec3_copy(&bottom, &t);
  2008. }
  2009. if (rot >= HELPER_ROT_BREAKPONT) {
  2010. for (float i = HELPER_ROT_BREAKPONT; i <= 360.0f; i += 90.0f) {
  2011. if (rot < i)
  2012. break;
  2013. vec3 l = left;
  2014. vec3 r = right;
  2015. vec3 t = top;
  2016. vec3 b = bottom;
  2017. vec3_copy(&top, &l);
  2018. vec3_copy(&right, &t);
  2019. vec3_copy(&bottom, &r);
  2020. vec3_copy(&left, &b);
  2021. }
  2022. } else if (rot <= -HELPER_ROT_BREAKPONT) {
  2023. for (float i = -HELPER_ROT_BREAKPONT; i >= -360.0f;
  2024. i -= 90.0f) {
  2025. if (rot > i)
  2026. break;
  2027. vec3 l = left;
  2028. vec3 r = right;
  2029. vec3 t = top;
  2030. vec3 b = bottom;
  2031. vec3_copy(&top, &r);
  2032. vec3_copy(&right, &b);
  2033. vec3_copy(&bottom, &l);
  2034. vec3_copy(&left, &t);
  2035. }
  2036. }
  2037. // Get sides of box transform
  2038. left = GetTransformedPos(left.x, left.y, boxTransform);
  2039. right = GetTransformedPos(right.x, right.y, boxTransform);
  2040. top = GetTransformedPos(top.x, top.y, boxTransform);
  2041. bottom = GetTransformedPos(bottom.x, bottom.y, boxTransform);
  2042. bottom.y = size.y - bottom.y;
  2043. right.x = size.x - right.x;
  2044. // Init viewport
  2045. vec3 viewport;
  2046. vec3_set(&viewport, main->previewCX, main->previewCY, 1.0f);
  2047. vec3_div(&left, &left, &viewport);
  2048. vec3_div(&right, &right, &viewport);
  2049. vec3_div(&top, &top, &viewport);
  2050. vec3_div(&bottom, &bottom, &viewport);
  2051. vec3_mulf(&left, &left, main->previewScale);
  2052. vec3_mulf(&right, &right, main->previewScale);
  2053. vec3_mulf(&top, &top, main->previewScale);
  2054. vec3_mulf(&bottom, &bottom, main->previewScale);
  2055. // Draw spacer lines and labels
  2056. vec3 start, end;
  2057. float pixelRatio = main->GetDevicePixelRatio();
  2058. for (int i = 0; i < 4; i++) {
  2059. if (!spacerLabel[i])
  2060. spacerLabel[i] = CreateLabel(pixelRatio);
  2061. }
  2062. vec3_set(&start, top.x, 0.0f, 1.0f);
  2063. vec3_set(&end, top.x, top.y, 1.0f);
  2064. RenderSpacingHelper(0, start, end, viewport, pixelRatio);
  2065. vec3_set(&start, bottom.x, 1.0f - bottom.y, 1.0f);
  2066. vec3_set(&end, bottom.x, 1.0f, 1.0f);
  2067. RenderSpacingHelper(1, start, end, viewport, pixelRatio);
  2068. vec3_set(&start, 0.0f, left.y, 1.0f);
  2069. vec3_set(&end, left.x, left.y, 1.0f);
  2070. RenderSpacingHelper(2, start, end, viewport, pixelRatio);
  2071. vec3_set(&start, 1.0f - right.x, right.y, 1.0f);
  2072. vec3_set(&end, 1.0f, right.y, 1.0f);
  2073. RenderSpacingHelper(3, start, end, viewport, pixelRatio);
  2074. }
  2075. void OBSBasicPreview::ClampScrollingOffsets()
  2076. {
  2077. obs_video_info ovi;
  2078. obs_get_video_info(&ovi);
  2079. QSize targetSize = GetPixelSize(this);
  2080. vec3 target, offset;
  2081. vec3_set(&target, (float)targetSize.width(), (float)targetSize.height(),
  2082. 1.0f);
  2083. vec3_set(&offset, (float)ovi.base_width, (float)ovi.base_height, 1.0f);
  2084. vec3_mulf(&offset, &offset, scalingAmount);
  2085. vec3_sub(&offset, &offset, &target);
  2086. vec3_mulf(&offset, &offset, 0.5f);
  2087. vec3_maxf(&offset, &offset, 0.0f);
  2088. scrollingOffset.x = std::clamp(scrollingOffset.x, -offset.x, offset.x);
  2089. scrollingOffset.y = std::clamp(scrollingOffset.y, -offset.y, offset.y);
  2090. }