֭j% dZddlZddlZddlZddlmZmZddlmZdZ dZ dZ Gddej Z d ed ed ed ed eejf dZddeded eejfdZ ddeded eejfdZdS)uThư viện dùng chung chụp màn hình an toàn tài nguyên GDI cho AutoPOE2. Module này cung cấp các engine chụp màn hình hiệu năng cao dựa trên Win32 GDI BitBlt và PrintWindow API. Tuân thủ nghiêm ngặt nguyên tắc Zero Resource Leak (Rule 5 & Rule 16): Mọi đối tượng đồ họa Windows (DC, HBITMAP) đều BẮT BUỘC phải được giải phóng qua DeleteDC, DeleteObject và ReleaseDC trong các khối finally. N)OptionalTuple)Imagei c 0eZdZdejjfdejjfdejjfdejjfdejjfdejjfdejjfdejjfd ejjfd ejjfd ejjfg Zd S) _BITMAPINFOHEADERbiSizebiWidthbiHeightbiPlanes biBitCount biCompression biSizeImagebiXPelsPerMeterbiYPelsPerMeter biClrUsedbiClrImportantN) __name__ __module__ __qualname__ctypeswintypesDWORDLONGWORD_fields_1C:\Projects\AutoPOE2\src\common\screen_capture.pyrrs 6?() FO() V_)* V_)* v+, &//0 -. FO01 FO01 fo+, 6?01 HHHrrlefttopwidthheightreturnc  tjdks |dks|dkrdS ddlm}|n#t$rYnwxYwt jj}t jj}d}d}d} d} | d}|se |r| r| || | r| | |r| ||r| d|dSdS||}|se |r| r| || | r| | |r| ||r| d|dSdS||||} | se |r| r| || | r| | |r| ||r| d|dSdS| || } | se |r| r| || | r| | |r| ||r| d|dSdS||dd|||||t" se |r| r| || | r| | |r| ||r| d|dSdSt%} t jt$| _|| _| | _d| _d| _t2| _t j||zdz} ||| d|| t j| d} | dkre |r| r| || | r| | |r| ||r| d|dSdSt=jd||ftA| d d dd}|!d |r| r| || | r| | |r| ||r| d|SS#t$rfY|r| r| || | r| | |r| ||r| d|dSdSwxYw#|r| r| || | r| | |r| ||r| d|wwxYw) uxChụp vùng chữ nhật xác định trên màn hình thông qua GDI BitBlt siêu tốc. Đảm bảo 100% tài nguyên GDI (Device Context, Compatible Bitmap) được dọn dẹp triệt để trong khối finally ngay cả khi xảy ra ngoại lệ không mong muốn. Args: left: Tọa độ X góc trên bên trái theo pixel màn hình. top: Tọa độ Y góc trên bên trái theo pixel màn hình. width: Chiều rộng vùng chụp (pixel). Phải lớn hơn 0. height: Chiều cao vùng chụp (pixel). Phải lớn hơn 0. Returns: Optional[Image.Image]: Đối tượng PIL Image RGB nếu thành công, hoặc None nếu thất bại. Thread-safety: Thread-safe. Example: >>> img = capture_rect_bitblt(0, 0, 1920, 1080) >>> if img: ... img.save("fullscreen.png") ntrN)attach_to_default_desktop RGBArawBGRARGB)"osnamesrc.common.win32_windowr' Exceptionrwindlluser32gdi32GetDC SelectObject DeleteObjectDeleteDC ReleaseDCCreateCompatibleDCCreateCompatibleBitmapBitBltSRCCOPYrsizeofr r r r r BI_RGBrcreate_string_buffer GetDIBitsbyrefr frombufferbytesconvert)r r!r"r#r'r4r5desk_dcmem_dcbmpold_bmpbmibuflinesraw_imgs rcapture_rect_bitbltrO%s 2 w$%1**! t EEEEEE!!####      ] !F M EG F CG/),,q// J  0g 0   vw / / /  $   s # # #  # NN6 " " "  )   Q ( ( ( ( ( ) )S))'22 B  0g 0   vw / / /  $   s # # #  # NN6 " " "  )   Q ( ( ( ( ( ) )K**7E6BB :  0g 0   vw / / /  $   s # # #  # NN6 " " "  )   Q ( ( ( ( ( ) )C$$VS11 2  0g 0   vw / / /  $   s # # #  # NN6 " " "  )   Q ( ( ( ( ( ) )9||FAq%$WUU *  0g 0   vw / / /  $   s # # #  # NN6 " " "  )   Q ( ( ( ( ( ) )3 !!]#455  w  ")%&.1*<==QV\#=N=NPQRR A::  0g 0   vw / / /  $   s # # #  # NN6 " " "  )   Q ( ( ( ( ( ) )"6E6?E#JJvWXZ[\\u%%  0g 0   vw / / /  $   s # # #  # NN6 " " "  )   Q ( ( ( ( )   0g 0   vw / / /  $   s # # #  # NN6 " " "  )   Q ( ( ( ( ( ) )  0g 0   vw / / /  $   s # # #  # NN6 " " "  )   Q ( ( ( ( )sV1 >>,R (R $R "R "R &B"R -=R S<S?;S<<S??A#U"Thwnd client_onlyc~tjdks|sdS ddlm}||n#t$rYnwxYwt jj}||sdS|rt j }| |t j |sdS|j |jz }|j|jz }t j dd}||t j |sdS|j|j} }ntt j }||t j |sdS|j |jz }|j|jz }|j|j} }|dks|dkrdS t-|d} | e| j\} } |rW| |ks| |krKt1d| |z dz} t1d| |z | z }| | || |z||zfS| Sn#t$rYnwxYwt5|| ||S)uChụp cửa sổ game bằng phương pháp GDI BitBlt. Tự động xác định tọa độ vùng vẽ thực tế (Client Rect) hoặc toàn bộ khung cửa sổ (Window Rect) và thực hiện chụp nhanh. Đảm bảo giải phóng toàn bộ GDI handle trong finally. Args: hwnd: Handle cửa sổ Win32 cần chụp. client_only: Nếu True, chỉ chụp vùng hiển thị nội dung (loại trừ title bar và viền). Nếu False, chụp toàn bộ bounding box của cửa sổ. Returns: Optional[Image.Image]: Ảnh chụp dưới dạng PIL Image RGB, hoặc None nếu cửa sổ không hợp lệ. Thread-safety: Thread-safe. Example: >>> img = capture_window_bitblt(poe_hwnd, client_only=True) r&Nrattach_to_window_desktopT)render_full_contentr)r/r0r1rTr2rr3r4IsWindowrRECT GetClientRectrCrightr bottomr!POINTClientToScreenxy GetWindowRectcapture_window_printsizemaxcroprO)rPrQrTr4rectwhptr r! img_printpwphbxbys rcapture_window_bitbltrms( w$dt DDDDDD  &&&&      ] !F ??4 t(##%%##D&,t*<*<== 4 J " K$( " _ " "1a ( ($$T6<+;+;<< 4D"$c##%%##D&,t*<*<== 4 J " K$( "ItxcAvvat  (4HHH  ^FB @Q"q&&R!VM**BFRK(( ~~r2rAvrAv&>???  !       tS!Q / //s%( 55$A5HH H*)H*rUc  tjdks|sdS ddlm}||n#t$rYnwxYwt jj}t jj}| |sdSt j }| |t j |sdS|j|jz }|j|jz }|dks|dkrdSd}d} d} d} |d}|se | r| r|| | | r|| | r|| |r|d|dSdS||} | se | r| r|| | | r|| | r|| |r|d|dSdS||||} | se | r| r|| | | r|| | r|| |r|d|dSdS|| | } | se | r| r|| | | r|| | r|| |r|d|dSdS|rt2nd} ||| | } | s|r||| d} | se | r| r|| | | r|| | r|| |r|d|dSdSt7}t jt6|_||_| |_d|_ d|_!tD|_#t j$||zdz}|%| | d||t j |d}|dkre | r| r|| | | r|| | r|| |r|d|dSdStMj'd||ftQ|d d dd}|)d | r| r|| | | r|| | r|| |r|d|SS#t$rfY| r| r|| | | r|| | r|| |r|d|dSdSwxYw#| r| r|| | | r|| | r|| |r|d|wwxYw) uChụp cửa sổ game thông qua Win32 PrintWindow API. Phương pháp này hỗ trợ lấy ảnh ngay cả khi cửa sổ bị che khuất bởi các cửa sổ khác nhờ cờ PW_RENDERFULLCONTENT (0x02). Toàn bộ DC và Bitmap đều được giải phóng an toàn trong khối finally. Args: hwnd: Handle cửa sổ Win32 cần chụp. render_full_content: Nếu True, sử dụng cờ PW_RENDERFULLCONTENT để DWM render đầy đủ nội dung. Returns: Optional[Image.Image]: Đối tượng PIL Image RGB, hoặc None nếu thất bại. Thread-safety: Thread-safe. Example: >>> img = capture_window_print(poe_hwnd) r&NrrSr(r)r*r+r,r-r.)*r/r0r1rTr2rr3r4r5rVrrWr_rCrYr rZr!r6r7r8r9r:r;r<PW_RENDERFULLCONTENT PrintWindowrr?r r r r r r@rrArBrrDrErF)rPrUrTr4r5rdrerfrGrHrIrJflagssuccessrKrLrMrNs rr`r`s, w$dt DDDDDD  &&&&      ] !F M E ??4 t ?   ! !D   fl4&8&8 9 9t TYA dhAAvvatG F CG3),,q// R  0g 0   vw / / /  $   s # # #  # NN6 " " "  )   Q ( ( ( ( ( ) )[))'22 J  0g 0   vw / / /  $   s # # #  # NN6 " " "  )   Q ( ( ( ( ( ) )S**7Aq99 B  0g 0   vw / / /  $   s # # #  # NN6 " " "  )   Q ( ( ( ( ( ) )K$$VS11 :  0g 0   vw / / /  $   s # # #  # NN6 " " "  )   Q ( ( ( ( ( ) )C)<B$$$$T6599 :. :((vq99G (  0g 0   vw / / /  $   s # # #  # NN6 " " "  )   Q ( ( ( ( ( ) )1 !!]#455  r  ")!a%!)44Q3 S8I8I1MM A::  0g 0   vw / / /  $   s # # #  # NN6 " " "  )   Q ( ( ( ( ( ) )"6Aq65::ufaQRSSu%%  0g 0   vw / / /  $   s # # #  # NN6 " " "  )   Q ( ( ( ( )   0g 0   vw / / /  $   s # # #  # NN6 " " "  )   Q ( ( ( ( ( ) )  0g 0   vw / / /  $   s # # #  # NN6 " " "  )   Q ( ( ( ( )sV( 55.T+*T+&T+$T+!?T+B"T+ =T++ V5VVVA#X)T)__doc__rctypes.wintypesr/typingrrPILrr>ror@ StructurerintrOboolrmr`rrrrzsr  """""""" !      (    Y) Y)Y) #Y)-0Y) ekY)Y)Y)Y)xD0D0D0$D0(5;BWD0D0D0D0P,0f)f) f)$(f) ekf)f)f)f)f)f)r