j<ddlZddlZddlmZmZddlmZmZddlm Z m Z m Z m Z m Z mZmZmZmZmZddlmZddlmZmZddlmZmZmZmZej:ZGdd eZy) N)Qt pyqtSignal)QActionQColor) QAbstractItemView QCheckBox QHBoxLayout QHeaderViewQMenu QPushButtonQStyle QTableWidgetQTableWidgetItemQWidget)PARTICLE_GROUP_MAPPING)calculate_particle_availabilityexpand_group_selections) BG_DEFAULTBUTTON_STYLE_ALTCODE_BGFONT_SIZE_NORMALceZdZeeZdfd ZdZdZdZ dZ dZ dZ dZ d Zd Zd Zd Zd ZdZdZdZdZdZdZdZdZxZS)ConflictMatrixct||jjtj j |jjtj j |jjtjjtjjz|jtjj ||_d|_i|_g|_|jj+tj,j.|jj0j3|j4|j7t8j:j<|j?}d}|jA|y)NF)!super__init__horizontalHeadersetSectionResizeModer ResizeModeResizeToContentsverticalHeadersetDefaultAlignmentr AlignmentFlag AlignRight AlignVCentersetSelectionModer SelectionMode NoSelectionsettings simple_modemod_particles_cacheall_particles_cachesetContextMenuPolicyContextMenuPolicyCustomContextMenucustomContextMenuRequestedconnectshow_mod_context_menusetHorizontalScrollModer ScrollModeScrollPerPixelhorizontalScrollBar setSingleStep)selfr* h_scrollbarnew_pixel_step __class__s FC:\Users\pivot\Documents\cueki\casual-preloader\gui\conflict_matrix.pyrzConflictMatrix.__init__ sF  44[5K5K5\5\] 22;3I3I3Z3Z[ 11"2B2B2M2MPRP`P`PmPm2mn l88DDE   #% #%  2223G3G3Y3YZ 88@@A[A[\ $$%6%A%A%P%PQ..0 !!.1c* j}|j|}|dkryj|}|sy|jj dd t }j jjtjj}t|d}|jj fd|j||j!jjtjj"}t|d} | jj fd|j| |j%|j'|y)Nr (z Open mod pagec&jSN) open_mod_urlmod_namer9sr=z6ConflictMatrix.show_mod_context_menu..Es1B1B81Lr>Deletec:jjSrB)mod_delete_requestedemitrDsr=rFz6ConflictMatrix.show_mod_context_menu..Ls0I0I0N0Nx0Xr>)r"logicalIndexAtverticalHeaderItemtextsplitr get_mod_urlstyle standardIconr StandardPixmapSP_FileDialogInfoViewr triggeredr2 addAction addSeparator SP_TrashIconexec mapToGlobal) r9positionheaderrow header_itemmenuurl_icon url_action delete_icon delete_actionrEs ` @r=r3z$ConflictMatrix.show_mod_context_menu3s@$$&##H- 7 --c2  ##%++D1!4T{   H % 1zz|001F1F1\1\]H ?DAJ  ( ()L M NN: &    jjl//0E0E0R0RS  Xt< ''(XY }% &$$X./r>cz ddlm}|j|S#t$rtj dYywxYw)Nr)mod_urlszError loading mod URLs)core.util.datardget Exceptionlog exception)r9rErds r=rOzConflictMatrix.get_mod_urlQs8  /<<) )  MM2 3 s ::c|j|}| tj|yy#t$rtj d|YywxYw)NzError opening URL for )rO webbrowseropenrgrhri)r9rEurls r=rCzConflictMatrix.open_mod_urlYsUx( ? C$  C 6xjAB Cs,!AAc|jr8|jr|jjS|jjSiSrB)r*r+matrix_selections_simplematrix_selectionsr9s r=load_selectionszConflictMatrix.load_selectionsbs7 ==}}===}}666 r>c i}td|jD]B}|j|}|stj d|0|j j }t|jD]}|j||}|s|j}|s+|jdkDs?|jd}|sS|j} t| tst| js|j!|} | stj d|| j j#dd} | ||<BEt%||j&|j(}|S)NzMissing header item for column rz%Missing vertical header item for row r@)range columnCounthorizontalHeaderItemrhwarningrMstriprowCount cellWidgetlayoutcountitemAtwidget isinstancer isCheckedrLrNrr,r+) r9 selectionscolr] column_namer\ cell_widgetr| widget_itemcheckbox v_header_itemrEs r=_get_current_selections_dictz+ConflictMatrix._get_current_selections_dictjsZ D,,./C33C8K =cUCD%**,224KT]]_-"ooc37 (//1F&,,.1"4&,mmA&6 &'2'9'9';H)(I>8CUCUCW040G0G0L '4$'KK2WX[W\0]$^$,+8+=+=+?+E+Ed+KA+N:B ; 7 %.02- 00$2B2B r>c|jsy|j}|jr||j_y||j_yrB)r*rr+rorpr9rs r=save_selectionszConflictMatrix.save_selectionss;}} 668   5?DMM 2.8DMM +r>c&|j}|SrB)rrs r=get_selected_particlesz%ConflictMatrix.get_selected_particless668 r>cdi|_g|_|jd|jdyNr)r,r- setRowCountsetColumnCountrqs r= clear_matrixzConflictMatrix.clear_matrixs.#% #%   Ar>c||_|jr^t|jj}|jr|j |y|j ||j yyrB)r+r,listkeysupdate_matrix_simpleupdate_matrix_advancedr-)r9enabledmodss r=set_simple_modezConflictMatrix.set_simple_modes^"  # #005578D))$/++D$2J2JK $r>c||_ddlm}|\}}||_|jr|j |y|j ||y)Nr)get_mod_particles)r-core.util.pcf_path_walkrr,r+rr)r9r pcf_filesr mod_particles_s r= update_matrixzConflictMatrix.update_matrixsK#,  >,. q#0     % %d +  ' 'i 8r>cn|jttj}|j t |dz|j t |dg|Dcgc]}d|d c}z}|j||j||j||ycc}w)Nrt Select All ) clearContentsrrrrlenrsetHorizontalHeaderLabelssetVerticalHeaderLabels_setup_matrix_cells)r9rgroupsgroupheaderss r=rz#ConflictMatrix.update_matrix_simples ,1134 CK!O, T# .f#EfUawaLf#EE &&w/ $$T*   v. $Fs+ B2c|j|jt|dz|jt|dg|z}|j ||j ||j ||y)Nrtr)rrrrrrr)r9rrrs r=rz%ConflictMatrix.update_matrix_advancedss  C NQ./ T# .9, &&w/ $$T*   y1r>c g}j}t|D]d\}}td}|jtdt dz|j jd|ffd j|d|ftjD]}|dzdk(rtnt} j|} | r| jt| tj!D]b}t#} | jt| | j%t&j(j*j-||| dt|D]&\}}t/j0j3|g} t|D]\} }| dz}t5||j6| |\}}|s*|dzdk(rtnt} t9}|jd | d t;|}|j=t&j>j@|jCddddjE||}|r|jG||jI|j|||)|D]5}|jKd |jMd |jKd7dd l'm(}|jSdjTy) Nrz+QPushButton { padding: 4px 8px; font-size: z; }Fc&j|SrB)select_all_row)checkedrr9s r=rFz4ConflictMatrix._setup_matrix_cells..s4K^K^_`Kar>rrtz background: ;T)QTimer)+rr enumerater setStyleSheetrrclickedr2 setCellWidgetrurvrrrw setBackgroundrrzrsetFlagsrItemFlag NoItemFlagssetItemsetr,rfrr+rr setAlignmentr$ AlignCentersetContentsMarginscreate_checkboxappend addWidget blockSignals setChecked PyQt6.QtCorer singleShotupdate_select_all_buttons)r9rcolumnssaved_checkboxes_to_checksaved_selectionsr\modselect_all_buttonrbg_colorr]itemmod_particles_setcol_idxr should_enable should_checkrr|rrs` r=rz"ConflictMatrix._setup_matrix_cellss$&!//1"$HC +L 9   + +,  *""3[95);(@2H  ! !$ '    %  ! !% (2 (!T;;chd}g}td|jD]~}|j||}|s|jj dj }|sH|j sY|j||f|js}d}| }d}|D]f\}}|j} |} | | k7s| r|j|||jd|j| |jdd}h|r!|j|jyyNFrtrT)rurvr{r|r~r isEnabledrruncheck_column_exceptrrrr) r9r\ any_checkedenabled_checkboxes_in_rowrrrrsomething_changed current_state target_states r=rzConflictMatrix.select_all_rows, $&!D,,./C//#s3K&--/66q9@@B 2 2 4-44h_E))+&* 0' !6MHc$..0M'L ,..sC8%%d+##L1%%e,$(!7   "  * * , r>cd}t|jD]}td|jD]}|j||}|s|j j dj }|sH|jsY|jd|jd|jdd}|r!|j|jyyr) rurzrvr{r|r~rrrrrr)r9rr\rrrs r= deselect_allzConflictMatrix.deselect_all@s!)CQ 0 0 23"ooc37 *113::1=DDFHH$6$6$8 --d3 ++E2 --e4,0)4*   "  * * , r>c|jdt|jD]}|j||j yr)resizeColumnToContentsrurzresizeRowToContents_update_row_counts)r9r\s r=rz(ConflictMatrix.update_select_all_buttonsQs? ##A&)C  $ $S )* !r>c ht|jD]}|j|}|s|jj dd}d}d}td|j D]}|j ||}|s|j}|s+|jdkDs?|jdj} t| tso|dz }| js|dz }|dkDr|j|d|d|d|j|y)Nr@rrt/))rurzrLrMrNrvr{r|r}r~rrrrsetText) r9r\v_header base_nametotalrrrr|rs r=rz!ConflictMatrix._update_row_countsXs)C..s3H! --d3A6IEGQ 0 0 23"ooc37 (//1F&,,.1"4#)==#3#:#:#<%h :!QJE'113 '1 4qy  I;b 5'!CD  +1*r>ct|jD]}||k7s |j||}|s|j}|s1|j dkDsE|j dj }t|tsu|js|jd|jd|jdy)NrTF) rurzr{r|r}r~rrrrrr)r9r target_rowr\rr|rs r=rz$ConflictMatrix.uncheck_column_exceptss)Cj "ooc37 (//1F&,,.1"4$*MM!$4$;$;$=&x;@R@R@T$11$7$//6$11%8*r>c^t}|jj||ffd |S)Nc*j|||SrB)on_checkbox_state_changed)statercr9s r=rFz0ConflictMatrix.create_checkbox..s(F(FuaQR(Sr>)r stateChangedr2)r9r\rrs` r=rzConflictMatrix.create_checkboxs-;%%3 S r>c|j}t|tsy|tjj j k(r|j|||j|jyrB) senderrrr CheckStateCheckedvaluerrr)r9rr\rrs r=rz(ConflictMatrix.on_checkbox_state_changedsZ;;=(I.  BMM))// /  & &sC 0  &&(r>rB)__name__ __module__ __qualname__rstrrIrr3rOrCrrrrrrrrrrrrrrrrrr __classcell__)r<s@r=rrs|%c?2&0<C D9L 9/" 2A=F#-J-"",6 9 )r>r) loggingrkrrr PyQt6.QtGuirrPyQt6.QtWidgetsrrr r r r r rrrcore.constantsrcore.services.particlesrr gui.themerrrr getLoggerrhrr>r=rsQ''   2NMgu)\u)r>