window-basic-preview.cpp 68 KB

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