§
    Û×j§  ã                   ó®   — d dl mZmZmZmZmZmZmZmZ d dl	m
Z
mZ g d¢ZeefeefeefdœZdg d¢ddœgdg d	¢d
dœgdddgddœgdœZ G d„ d¦  «        ZdS )é   )ÚDA3_DEEPY_INFOSÚ	DA3_INFOSÚDEEPY_PROMPT_INFOSÚMAPANYTHING_DEEPY_INFOSÚMAPANYTHING_INFOSÚMOGE2_DEEPY_INFOSÚMOGE2_INFOSÚPROMPT_INFOS)ÚCUSTOM_SETTINGSÚMOGE2_MODES)Úmoge2ÚmapanythingÚ
da3_pointszphoto to 3D)úpoint cloudúdepth to 3Dz	2.5D meshzTTurns one photo into a metric point cloud or a textured mesh of its visible surface.)ÚnameÚaliasesÚdescriptionzmulti-view reconstruction)Úphotogrammetryzvideo to 3Dúcamera posesr   zOReconstructs a scene in metres with its cameras from several photos or a video.zvideo to point cloudr   r   zJFast point cloud and cameras from photos or video frames (relative scale).c                   óî   — e Zd Zed„ ¦   «         Zed„ ¦   «         Zed„ ¦   «         Zed„ ¦   «         Zed„ ¦   «         Zed„ ¦   «         Z	edd„¦   «         Z
ed	„ ¦   «         Zedd„¦   «         Zed„ ¦   «         ZdS )Úfamily_handlerc                  ó   — t           S ©N)ÚARCHITECTURES© ó    úFC:\Users\johnn\OneDrive\Documents\AI\models\recon3d\recon3d_handler.pyÚquery_supported_typesz$family_handler.query_supported_types   s   € åÐr   c                  ó
   — i i fS r   r   r   r   r   Úquery_family_mapsz family_handler.query_family_maps   s   € à�2ˆvˆr   c                  ó   — dS )NÚrecon3dr   r   r   r   Úquery_model_familyz!family_handler.query_model_family   s   € àˆyr   c                  ó
   — ddiS )Nr#   )iÖ  z3D Reconstructionr   r   r   r   Úquery_family_infosz!family_handler.query_family_infos   s   € àÐ6Ð7Ð7r   c                 ó   — | S r   r   )Úbase_model_types    r   Úget_lora_dirzfamily_handler.get_lora_dir#   s   € àÐr   c                 óÜ  — t           |          \  }}i dd“dd“dd“dd“dd“dd	“d
d“dd“dd“dd“dd“dd“dd„ t          |          D ¦   «         “dg i ddœ“d| g“d|“dt          “|t          t          |          dœ¥}| dk    rC|                     dgdddœddi t          ¥dt          t          d         ¦  «        i¥d œ¦  «         n(|                     d!d"gdddœd#d$gd%dd&d'œddd(d)œ¦  «         |S )*NÚimage_outputsTÚmesh_outputsÚdevice_explicitÚno_background_removalÚfit_into_canvas_image_refsé    Úprompt_descriptionz8Asset Name / Description (optional, used for file names)Úprompt_optionalÚno_negative_promptÚembedded_guidanceFÚinference_stepsÚlock_inference_stepsÚimage_batch_size_maxr   Úcustom_settingsc                 ó6   — g | ]}|                      ¦   «         ‘ŒS r   )Úcopy)Ú.0Úones     r   ú
<listcomp>z2family_handler.query_model_def.<locals>.<listcomp>7   s    € ÐWÐWÐW¨s §¢¡
¤
ÐWÐWÐWr   Úprompt_enhancer_defÚ )Ú	selectionÚlabelsÚdefaultÚprofiles_dirÚinfosÚprompt_infos)Údeepy_infosÚdeepy_prompt_infosÚspecialitiesr   )ÚPhotoÚIrJ   )ÚchoicesÚletters_filterrB   rK   )Úimage_ref_choicesÚone_image_ref_neededÚone_image_ref_onlyÚmodel_modes)ÚPhotosrJ   )z	No Photosr?   )zNo Videor?   )úReference VideoÚVrS   ÚVideo)rK   rL   rB   ÚlabelrR   )rM   Úguide_custom_choicesÚreference_video_enabledÚskip_video_guide_preprocessÚvideo_guide_label)ÚINFOSr   r
   r   ÚSPECIALITIESÚupdater   Úlist)r(   Ú	model_defrD   rF   Úextra_model_defs        r   Úquery_model_defzfamily_handler.query_model_def'   sü  € å" ?Ô3Ñˆˆ{ð
Ø˜Tð
à˜Dð
ð ˜tð
ð $ Tð	
ð
 )¨!ð
ð !Ð"\ð
ð ˜tð
ð ! $ð
ð   ð
ð ˜uð
ð # Dð
ð # Að
ð ÐWÐWµoÀoÔ6VÐWÑWÔWð
ð "°¸rÈbÐ#QÐ#Qð
ð ˜_Ð-ð
ð  �Uð!
ð" �Lð#
ð$ 'Ý"4Ý(¨Ô9ð)
ð 
ð 
ˆð, ˜gÒ%Ð%Ø×"Ò"Ø2@Ð1AÐUXÐehÐ%iÐ%iØ(,Ø&*ØW¥+ÐW¨y½$½{È9Ô?UÑ:VÔ:VÐWÐWð	$ð $ñ ô ð ð ð ×"Ò"Ø2AÐCTÐ1UÐilÐy|Ð%}Ð%}Ø5EÐG_Ð4`Ðtwð  EGð  RYð  )Zð  )ZØ+/Ø/3Ø%6ð$ð $ñ ô ð ð Ðr   Nc                 ó8   — |dk    rg S ddl m}  |d¬¦  «        gS )Nr   r0   )Úquery_download_defF)Úmetric)Ú&preprocessing.depth_anything_v3.assetsrb   )ÚcomputeListr(   r^   rb   s       r   Úquery_model_filesz family_handler.query_model_filesQ   s;   € à˜lÒ*Ð*ØˆIØMÐMÐMÐMÐMÐMà"Ð"¨%Ð0Ñ0Ô0Ð1Ð1r   c                 ó>   — | dk    rd|d         vrd|d         vrdS d S )Nr   rJ   Úvideo_prompt_typerS   z5Add Photos, a Reference Video or both to reconstruct.r   )r(   r^   Úinputss      r   Úvalidate_generative_settingsz+family_handler.validate_generative_settingsY   s:   € à˜gÒ%Ð%¨#°VÐ<OÔ5PÐ*PÐ*PÐUXÐ`fÐgzÔ`{ÐU{ÐU{ØJÐJØˆtr   Fc                 óB   — ddl m}  || |||¬¦  «        }|d|j        ifS )Nr   )Úmodel_factory)Úsave_quantizedÚtransformer)Úrecon3d_mainrl   rn   )Úmodel_filenameÚ
model_typer(   r^   rm   Úkwargsrl   Úmodels           r   Ú
load_modelzfamily_handler.load_model_   s@   € à/Ð/Ð/Ð/Ð/Ð/à�˜n¨j¸/ÐZhÐiÑiÔiˆØ�} eÔ&7Ð8Ð8Ð8r   c                 ól   — |                      ddddddœ¦  «         | dk    rt          d         |d<   d S d S )Nr   rJ   r0   )Ú
image_moderh   Ú
batch_sizeÚnum_inference_stepsÚremove_background_images_refr   rB   Ú
model_mode)r\   r   )r(   r^   Úui_defaultss      r   Úupdate_default_settingsz&family_handler.update_default_settingsf   sb   € à×Ò¨!À#ÐUVÐopð  STð  Uð  Uñ  	Vô  	Vð  	VØ˜gÒ%Ð%Ý(3°IÔ(>ˆK˜Ñ%Ð%Ð%ð &Ð%r   r   )F)Ú__name__Ú
__module__Ú__qualname__Ústaticmethodr   r!   r$   r&   r)   r`   rf   rj   rt   r|   r   r   r   r   r      s+  € € € € € Øðð ñ „\ðð ðð ñ „\ðð ðð ñ „\ðð ð8ð 8ñ „\ð8ð ðð ñ „\ðð ð'ð 'ñ „\ð'ðR ð2ð 2ð 2ñ „\ð2ð ðð ñ „\ðð
 ð9ð 9ð 9ñ „\ð9ð ð?ð ?ñ „\ð?ð ?ð ?r   r   N)Úpromptsr   r   r   r   r   r   r	   r
   Úsettingsr   r   r   rZ   r[   r   r   r   r   ú<module>rƒ      sv  ðð$ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $à 2Ð 2Ð 2Ð 2Ð 2Ð 2Ð 2Ð 2à6Ð6Ð6€ØÐ 1Ð2ÐDUÐWnÐCoð  AJð  L[ð  @\ð  	]ð  	]€à$Ð1\Ð1\Ð1\Øtðvð vð wà8ð  FFð  FFð  FFØ$uðwð wð xà2ÀÈ~Ð?^Ø#oðqð qð rðð €ðX?ð X?ð X?ð X?ð X?ñ X?ô X?ð X?ð X?ð X?r   