
    Mg                     D    S SK J r   S SKJr  S rS r\ " S5      S 5       rg)    )plugin)Forec                    / n[        U5       GH  nU R                  SR                  U5      5      n/ nUR                  5        H)  n [	        U5        UR                  [	        U5      5        M+     [        U5      U:w  a  U R                  SR                  U5      5        U R                  SR                  U5      5      nUR                  5        H)  n [	        U5        UR                  [	        U5      5        M+     [        U5      U:w  a  M  UR                  U5        GM     U$ ! [         a    UR                  S5         GM  f = f! [         a    UR                  S5         M  f = f)Nzenter row #{:d}: r   zRow length should be {:d})	rangeinputformatsplitintappendBaseExceptionlensay)jarvisrcmatrixirow_strcur_rowns           @/Users/admin/workspace/ai/Jarvis/jarviscli/plugins/matrix_add.py
get_matrixr      s/   F1X,,299!<=A'A s1v& ! 'laJJ299!<=ll#6#=#=a#@AG]]_+F NN3q6* %	 'la 	g+ , M! ! "q!!" % &NN1%&s$   D&E	&EE	E('E(c                 B    [        U5       H  n[        X   5        M     g )N)r   print)r   r   r   r   s       r   print_matrixr       s    1Xfi     z
matrix addc                 :   U R                  S[        R                  5        [        U R	                  S5      5      n[        U R	                  S5      5      n[        XU5      n U R                  S5      nUR                  5       S:X  a  Ow[        XU5      n[        U5       H)  n[        U5       H  nXG   U==   Xg   U   -  ss'   M     M+     U R                  S5      n	U	R                  5       S:w  a  [        XBU5        M  U R                  S5        [        XBU5        g )Nz%Sum of matrices with dimensions M x NzEnter M (rows): zEnter N (cols): zContinue with next matrix? nozPrint current sum matrix? zFinal sum matrix:)
r   r   GREENr
   input_numberr   r   lowerr   r   )
r   srowcolinitialnext_decisionnext_matrixr   jcurrent_decisions
             r   
matrix_addr*   %   s    
JJ6

C
f!!"45
6C
f!!"45
6Cc*G
%BC D( c2sA3Z
1!22    "<<(DE!!#t+s+  JJ"#s#r   N)r   coloramar   r   r   r*    r   r   <module>r-      s.     6
 $ $r   