
     hc                     h    d dl mZ ddlmZ ddlmZ ddlmZ  G d de      Zej                  fdZ	y	)
    )Enum   )PointCloudOptimizer)ModularPointCloudOptimizer)
PairViewerc                       e Zd ZdZdZdZy)GlobalAlignerModer   r   r   N)__name__
__module____qualname__r   r   r        G/home/cameronsmith/repos/controll3r/dust3r/dust3r/cloud_opt/__init__.pyr	   r	      s    /!=Jr   r	   c                    dj                         D cg c]  }| |   	 c}\  }}}}|t        j                  k(  r t        ||||fi |j                  |      }	|	S |t        j                  k(  r t	        ||||fi |j                  |      }	|	S |t        j
                  k(  r t        ||||fi |j                  |      }	|	S t        d|       c c}w )Nzview1 view2 pred1 pred2zUnknown mode )splitr	   r   tor   r   NotImplementedError)
dust3r_outputdevicemodeoptim_kwkview1view2pred1pred2nets
             r   global_alignerr      s    <U<[<[<]!^q-"2!^E5% 444!%uIILLVT J 
"==	=(uePxPSSTZ[ J 
"--	-ue@x@CCFK J "M$"899 "_s   CN)
enumr   	optimizerr   modular_optimizerr   pair_viewerr   r	   r   r   r   r   <module>r#      s0     * 9 #  0A/T/T r   