hydrodl2 1.3.0__py3-none-any.whl

hydrodl2/core/calc/batch_jacobian.pye

@@ -0,0 +1,501 @@
+---BEGIN PYE FILE---
+UT+o!c1gk=Y+fU}0^D@+
+By!m>RnPdPRdb-Sa?`<=
+i1&B*P7FtgFp?(>M|n9l
+c5C|bQ5Usf9XrX|wF1%6
++r<2W=|5n@nEkZcr$_s5
+6aSo9i5h!gR1H=Of)r6S
+9>SKYJ|Pe0;W#nMVFB7W
+{C(MXZk#5vc&9S>bWFzH
+c%PH!wvrMK=GnW(+8o7G
++#}#H|9lG7z+&NPT|k8E
+|M=yO)E@wqv~9qHHv29a
+{zg*rI}_mphr*$gT9-Le
+#3=Y$BsB{uVs7pqk=DJD
+bT_9cOx9?Fj1vz8*GXaN
+BuB`~Z68LntV!t+J7FnH
+m#YI@>g)vu$dvw7r7}L#
+v}~P<@DnqaLOCUs*<M`!
+`F149lWLVu4r7j@+shvf
+i>C93ex$+2FY`{VJjn}+
+E=>4oXTqdim$)jty)i-L
+1DgHH!o<PeYH!iYA7#wN
+Jw!9ZWXB7l%xAUY$WpS$
+vkU7cZ%K1z(Ez?2igbl;
+QB$RiW^?RZbSGX1mjAHm
+T`B%`E?(O!&AF?1`C}KO
+XyeIW^rCSpfE9MxfA`C{
+`EY2Qe8P*bV#~KwE!srE
+7EnYr(LHgBW}R|oWbxhz
+|6~)>@vv%WG;;|l8;CMQ
+DNnr`&@-p^b)&kEh76@w
+aEoDGGfI1rUFCfDe#6AN
+i9V}4D$k8IIbOd8PR$pn
+i$S5@=t+VSqN{Qk;=6)6
+Sb7fvL(CA}qoPETJH>oZ
+4G24beT0^EihBX>a}l+6
+TEsNFbV(f^nl906(@i<b
+;^8?>?~RFun?ynA7>&Kq
+y16U1gn6Np!LQ3@%~oWt
+pe$()vteO-<`Eh2-Qy7=
+p^qb4sa*!jOZSG-?ki+J
+HuVM89C6#8D=Ei^s1hu^
+(LJc39^s^~U$7Uqcx_!;
+ZEBoNd2fJ2cYWqxn%*Es
+6AwS+sFxiWLQV`W?5&#N
+%22Mm!*729e-tfR1`Pxm
+Pe@2t7@%0e1_HEr7M{dL
+;%I`yTri!BUNQ=8-;sH{
+9f=}c83^|K-~w<G4K2;}
+?+()E)=x4*VMff_Zg{KC
+Pe$2*k|#itpPuVQb!6}m
+-7$)O#ri<DNp0>1wr!@t
+0}HCc;2y_zsmFw9m@D>`
+rblMh3w|a0>)@O)DJ_Z~
+==lOoyCF_DNK~^k3Lu&;
+-S7?oj)04)oFx|>hFrbx
+$~MR(5M7K>-XdRCX;UBo
+ymPkYk-0JAq&XO}rh+zg
+;uCz)V%oB5FU{a6LoKwK
+Ht}P<TUx|T`rF%W$LL12
+@lx{b;%g-t_gc(yxpgHv
+_}~K2g#?W(<TM^SK2KZc
+s@=T7|LE8Y6O>_oPoJCM
+wbB>aL?(e|)yuqEPBI%y
+J^VordzD)Z0n>y_xhjAG
+Z~#1qp5AbRv9A$k*2+i$
+P?vjmmt48D!h6&UND3h$
+Lan(SXzmI&U?ViK_%#3^
+;G#o_Q}Q~b1SQ;=09jaU
+OYWVx+~bK7>I3Z@QyEH3
++6M;81+9`Ww1XyW7U7^p
+-{;zQcBifY-7n}B*^7)U
+J@aOR-4}+vQ`5b276v_C
+2~a#@lJi5!sepX@LQM$s
+*JY%~JLEUVhErAhx|LHF
+>;!UpX|#;K5$_lJS#Klz
+)T|l>Ajb?`cl7=}FY*oJ
+ub6uBqECgc3D~FCfh)cO
++p#QwYc>SRK-;f>RQl0B
+LRC?CLEf5}5Zj>&pxs#G
+%@8>Rnz0)Hg$u^eDdI(t
+;S{=;8h1_VeFL~0wu7Tk
+;=)`auLthMz^1T>1AGeW
+7%ewtQp0nA#e=Q%J`hsI
+jnGzzA+YXaM1(@B0ajTd
+#d7g*LHc@#S!O0g_S_DD
+%Wnuo#t7k5CU(qAO^J_!
+Fn1unrgWfB5wMy3FLjur
+`9IyMHmP;y2M$T-&KMVY
+^ZXg2$CIczerB?cq8Y3P
+o=~1-Apej7au&0;i0cGr
+O7$Y+uBjL1URY7)1972!
+@8OU!O8MpHa3b|c0ujcj
+uW!}K8tXSlY)rzmhLjk|
+$ea3<s~K)Me^O*u_U?*-
+5Mf!9FLkuuyn3AeVCdFi
+ixQU*W+uK1_aO|Z2)i~Z
+VDQ4+lMsCPVMBI3d<QXa
+-T7=5FIEE`u3GFGjR(!y
+BV2Dr5a%6<4g~RIz!6qz
+!R{8im3KxH7?A&<WO5C%
+b166VSz}xr9M{WdSv3jR
+$|7`YqiEisTOW~TVL*Ii
+<@xd+;$*CnB{W(AKk}!E
+BuUBIbv^lOj5>G0;$rTt
++ZX`W5Yh9Kbhzo{#JbkD
+z01XeF>48(imDEgSXYb9
+{9wCQaXn>kJA?z?K4fQ*
+dPc;HO8AymQo}-z36$P=
+8+Cl-IenDFf?UE{2<t>n
+qv+-qu&^+&J+?l>z7&5?
+#m{3tfls9<2s`N7<3rr=
+c6l}zJW;9xLOFt=O;Q+S
+XIok;ACt+3lwRRUh)oTn
+D*dxU$RRD?j&wBu-&U6F
+ZiB05sa7ea%^CizZgvo@
+p=Qyp{Zk@wOum?|N{LKo
+{@rJVU3S|ih;#H(JFNK*
+P1DzZP(9_>)jvM0eknQN
+hVp@Ja|$#nV(cyL*UJ03
+MnlQsn}T<3F9myu{<0(q
+Z^Hm7O<5$Da^A~l)di|n
+n6A*HU@pBT2C)NR)5*<*
+1;<$e(K2@}L|Sti6JK}@
+KX)?mNJlj~`cDvR)K1g$
+Y`2v-+5l=|M4<h4jF7gK
+a$qp@1~#6L)2yE!8S0TR
+rjxF8ZNsPz(*Q1M?Ukjp
+3MI(FN7`I9VeUneTmP&v
+9FpuIZNp*qIx_o)62lT2
+3bvxyP8;52$kwJAQL0v>
+r9FK68=h&&^la_LO>Vm+
+Uf{$x&y7NEv4XzRLNxDU
+(YHt7*3s_y5<*F;PE9Z6
+<lS3PtQk|_+L%qc8R;XE
+$g%|Xy?XZ*G7_q9`dvN*
+^T%J10#_#%drF;8o^hTE
+qGG5Sj~I9kSHB-)&Aw*y
+r=Y2bXxfadRjy>u9v4v4
+iE>?}3ayR%*Sy|6@r-l*
+!aW|WXV4EePyEacwyS{O
+SQ9jyTL(&z6aS$mMox@E
+ZZ$yb^L)%bU_Fz9d@$VC
+62f1fr;D@#q<!<`%UtPv
+DCAUa(BKV>XNHs(Uw4Bi
+im6J!PYt_Sm-4~ft>}sg
+=5aCHWt?ZcoX!Mi1iS@!
+-2^w~JrpJNwmSP7y9Bu`
+qDrX=aouy{B4rfAQ)gY!
+PGC#$P!l>$$BvRt8HhJ4
+L|aJ-kKH;!$u{F0kp`~r
+D1rKRj@hG&V}}BDG+(S8
+(l7B!Wm=MGq}gNz67`Wv
+1+b{@aIG}~$N?q;lCXz7
+6N4sD%Jg7(-5SOjMMHUx
+J)>)%5H!XM_-8{YOc^zh
+$cN^%<z$9X#s4*l89-WN
+QdzR(d}SymrxDW80r59#
+BioT2Hd+RZM8uBmi25tM
+lHVyL;FV7H#=&!_@)e3t
+VTI3*6BgFb<zmb1*`}(Q
+fC)_uKSq8=Eu|y*>n$+<
+)AGOc%(OM8rnIkVNPI$D
+=<86uk$cZzs#sD-clf2q
+AhVd_!7o>#tYChbXvppf
+RUAdV0Gg-rq7nau55wZ5
+;@RJ;<w^K_nkL2^(z$kC
+T^esP6cTVqb<}!lUc@kH
+6d*OwBw|!Hs2Fc)x4Kd?
+TkCT1#o{cr=I{~g>5;Su
+59k48^H&@Hn@1edAM%=^
+#CyLohLV2j$e`XFFbFH!
+%ly2E;(O-1evI&b(W+$%
+0TO07k+7NW)O@G0c!Pip
+#tnJXzB7_Ft`$omeM^FT
+QF+2~sYuX5L<b6g&W2`~
+7NRey%!oKJ{YsuUSx|35
+tLn+uRi(*GE2IMT5Ptyc
+z8W%LI%%SdJ8vf7P5?-Y
+f9B^~1@{%tS3;&Jz}2MP
+x@Te_8YrOT=bN^+hAN6K
+j;7>&u9oh4`l%>?)SME{
+pfbCA5Xtc=bV@E>i4REc
+y4C)?iRFkz9N3Q-bjpk|
+ZZ3{k;*V5QmD}8Xs7rI7
+{TlVZfD}tV8C+5>KNeGc
+pia-ojA_)fydaeGK$_j{
+abKuWg2^OMBySCK>QEx-
+*q-vo{uL@UGO?xW&uCR|
+4RW}D0a<Ku*92=_-=ba1
+7Un`<&46A``Fu6KL*)1L
+-PMR<R^xDaF6E9492v9v
+f&BYaJjXs{4BvSQns4K7
+!oML8717<{BWLI3l3Qyb
+?M?+^2@VRuFMbI$>@}vo
+HPFdYjNEK1$>(35*GRRB
+uyMGaIk`j>|KfEwKf}PD
+sv9nyNxJyEn2`}nit2e;
+p6L=3)yiU>J<+q$)Fe-4
+u%VaYFZaq`1J>*^W9jz0
+)W;mUj!2q)`sc9+GR1x4
+h&*LExC`Bg$|lvWHFXtB
+-b|;u9@@gr){AmRKc2x9
+Bc<t9&mfs@a-<Dv*UZNz
+pp!KL2UcF|^?VJ1jS|^B
+6*oA+qQNZ_$CCNKHcv{`
+>R1WE(59?Lc~h`)h>}XL
+pC4hJE+}2?O?R$Q?rW`*
+8ddr=f_cl*bal_hQIxVm
+3)_E_qDl1^99a|HfiJ-k
+_;<o`)Y&B~3<hBaz-i?B
+OFFU&iXEhrE$tu(r{On~
+jyz<%wJFDy(4z@>&w*mJ
+8<$EE7HNJ*v{@zp)R?rS
+y96ne{FgXB?FC3>hKTat
+mS!ATU7ci05(j7n<}x<p
+AFx^r#cY$@^;uep{4srg
+;NcXKQo)Cb6xtuc-6D>b
+CyRFdPyuXi1B?z!5$&b2
+$`Dwe@F8+!#wmwXn!EWH
+mk8M>RA@QoZ^&%RC3XKn
+ptvZb)t0{6=GL$SPcUxV
+mka!I{yw_KFKNC(s#O?^
+5EyFo30vW=cpTfo&UmMQ
+u$4r({e>3C6x22gi$g#@
+ci`aILRjj?);u7giM%1;
+Wn^9W02ACUG%2aN6?><_
+BcjwF6%lnNky>(w=^2dD
+qH+2+j{q+Zr)DF^u#H|k
+YfW<+;An<@N??b&dwf^5
+(|RRvxlM>%h2`{R=5c39
+5G)YqMZU#p?p=$<Nh&ho
+sk659jjzQ4j;|`WPWghf
+zQcZdk$}yj`9I>V=TTkF
+f)_YO9t2)<Q>TRQFAsg)
+jkE|u$(eZpl(xxibqK>v
+(Y;v;n)LlLB8u8%AUPf2
+(7$E~)x1mBm0w}q7tILE
+B}#y2Gdn<HO#|}qDKDRK
+4B`Tn@9;ox!UL6Gfex2R
+qm$kug1VeO`p9zeCX3Uh
+qXugE9>rbsTmIFtY2>|g
+25cR>3Cmm=AN|H}uWq<3
+7W-t3wyw;0gw`YJ!BTlj
+t(QDTa*?|+_bbdIcuD%y
+r0+Crdg2!Czm(Equ0_gg
+{=isc-t^C!bvP&|5jR!c
+t?fJ|;=&_2ziiBHq??jz
+|I9=v&O%>=&I{uB*VC`g
+Bn8CMH@Hm#aoRbj2~g|Q
+>5$PGcmSI`7u>$6v@9|!
+cC9!8%$$)Tcb$afDdK28
+p{Oq*{d-#cK;4>?c1hbI
+5igMPpUJ{ehCLcKq*f-R
+(oZG$Ok)XT7w#c*OGTZ~
+usFqqdDITT&X&D|22d^L
+KqT0P>aQc(Ym3mM4*(ss
+bfpkF$h;%c@n9;1RhJ)A
+Y*p420c;}lnP`CWV2b`x
+_7C&=hTm>^jnHVU$v236
+_~<whah<pS9w`Z~ICi3l
+86_bwKlB$~89YgK5}{9f
+UwDSILDz~h2@3ds?T~A?
+>Go<@>bk;M7nBh|jynsU
+PHBq5KjOYx1_OzMUJL26
+lXUVF+X?;}N8RRDT<FN~
+_{)N(snm^u>U#gTTf)rA
+z9wu~unIxC^F)l3V$lL_
+A!o?r%a+}V*_97C&7(;C
+q`t{@g^rL4QDN^8*8)2^
+=RN&DA8Y^7nRfiET*CYG
+Jn5b!@Qz`IA@%DP%2c%;
+Q^W9+Bc?X<R1FR7m51D~
+_~4AP7H;M(934Z>s)9u9
+r{OzBt4!XE)JZ~nLL-zs
+2ErLFL;I5Z#1etmq%Lb5
+`k9%d#e1@_7F3=le%~0^
+b2%2yZ8{O|t#4L{PmtLs
+3qGNB_%n;cwS!}g7c^(r
+O^%L@E9F*y3oC54EI|(a
+t=IRctTBcr@&IzV3Rr}N
+_A)JEGk-l2=UL3TG~2NW
+YjER}-;6mx*$;2INr2S=
+o03yrH~k2!GaJ!MH^>YR
+Ue0ZLbsK2x8xaPlc$MVs
+sXdX;wCPPqE=Xv!=*#eK
++Is?+%$5%xV(Gnk8md@U
+X+R#}q2y+~+AU})j=pKE
+{rS_+Z!>W?Xd72x0UuV&
+pJWc6SsNvo%x1=6fYsI^
+{)b2^2q@dP$%l8MJG#nd
+f}k3qN<Msc5=-pnf^^Pn
+b;Hi%dg5OA@0q>jpb^MK
+v@IPrP@u;^(~^79a?ut<
+(`=1DYn-bmAOR$HBIIMM
+`{~=y!2*)!^{FReQq7Ez
+`c@6ZB;RXD=G6h{1{4;i
+&bYw2p+n>yqC!gP=fezr
+taUCw6dZPRUCvf82R|>e
+OA{GX=3eblL%+M7u^7qH
+X6(bZ$jYo?e+F$zAI~OO
+Atn_nvDDqnDiY01mD^Lz
+?%saO55dCr0hHRrIUm(I
+k=G6CVZ_0NlG53=(4H#g
+Rja+s&XB8Pgp1((^C?yu
+hdFzdkAW$(Ixf5Gl<aq2
+t?2vPxi0VOI9drYcuK1g
+KJOFQ)^oaeLi<<*5PY08
+1HHqTi}B!?M%-)7J)PXF
+9J!Ds;P6g8doq;P-9arT
+N~oju7>|!)ARwx*Y$Yjv
+e%M=*pDKynh;0IOy`@@s
+H9-_+md<9{vDEtfg4#?1
+J_FL9$J;xvDnpfXc~JP&
+5HqNOTNccO+8k`9wrjt8
+)t)acC52!)$cLJ{?5Tf9
+6G*$~H-l;l7&?*KtF;QR
+mIozHb7_l;3P_@z8WEjz
+{nuwdur9uP@!oEEq{SI!
+Wp)nL6LDMOGS>6+($N{`
+y{AVRaLWu)%Jd;hM-X$H
+M6P?Z%+GptOmI_6`A1(G
+L|4mM{Q}7VGL~>`ZlV?(
+R$4Hk*|S++@B3&KWet#F
+r(fL1BI`NbQbr)uQ9Lj|
+JCun4+5%ZD&Q}@_nHw=Q
+sCQaa{ZVhY!k8n2C@!B%
+;LCOZ0Y699X7VvVdIjf-
+ptyDq*cN@}lXLAzNo$IH
+IlksU+1b=$Q#jDv<(72@
+yK5fp67D!Orm44AH{0xq
+v&r-mnSYscfwG)c|G4Vj
+e)~x%<w*A)b)&)iF}m0)
+j2U@;pq1F}ThYoM8%mU+
+UzH9@kd%OL)%lv9-*-q|
+Q@HJ|S`>|Vo{^tO$ui`}
+m_Y$MTiAr(Te`@fpbv6N
+KlkvtQ%tPst4Y>rtzLo{
+TpndW>WguD;`+b^QHe1c
+$<KCC>2%SkrKDI5ktP0#
+XMoNA7L!Z=#i@QC0_KUu
+kWua2mY&EDe=G2Z$kvZy
+J4-adeyj-rS~D@pXbj*m
+r0)Q2^{)K2$etFPp|EMK
+{5(yB(GNq_@58gv2~tzN
+612~mm{fe@K5eAki9L-6
+TZK(Kf30BjbGJHj8}qq|
+yRrAw`rE$lT0kMsbkX%L
+{m+qP@`#Xb88olG4`F$S
+$1uKsepjD?!KqQ$(oJ(Q
+Xn9Q15_&aYKXV49aei4(
+<*u_o^FVffn6MPWYR6wr
+EcMC!qooUv_cuhcxlQVF
+J)E|GF2*+2X$4`pOKJ(!
+M(AP$nh~~AiZrWqr+N)c
+*`*y#i5KuI<)<PgAhBUK
++SkKMu#Vz<Ac1G0vi#u9
+cpGW?sFu?{_f*P*Q79+F
+uy<KBrI{AiD+C1v2pTWe
+sQtA$1^v0Z)_5xOqwFT$
+l0E<_nE+XM))pvTT$UDZ
+7c|{CIcp$rD7VmMZ=6cg
+WfbTAj`Bk}one$tB%jS=
+iUwQ78%m%0{O*{ZS|A+}
+tLYV8IV~kWMtWp!lnb*Q
+ak0asjW~mU$9nkQKUTDO
+jia3%i{_Q9>%)vBjo;u#
+*;0xSDy4DmkkYXaR+?AI
+4c$GuCq29L{vTriLtlot
+4hBVR4@s$6pGtH7<o%>r
+bBMtG<$b`((5g<_877G`
+TU}g5xp}Q_-4zV+r=CFU
+tZwK3hiKW<;a)VThu)vQ
+k~d6wZVL4vC{JP1#!Ecs
+1>!yafv6RDZNbh5f(On{
+VTMpFPTVX>k5G2TkQTz>
+M~u(KD9JJ&hnjD5VcCfi
+3*zZ=VHRZB6`z*v#O0qk
+zPjbak4E5E&)pkx3+s7v
+1^rRz3qe{_<Xp)?8hH#s
+xTm^gSW->Cis=fo=)UJQ
+4)JT?gserhN)qxNG6#A_
+=XiuwmQ@X;gUQe?9k4o_
+9ZIyD9#`KKYo5oy;IW+0
+-QCP!I|*njsGtJXwDuE-
+4GaN}ViGy~%2JK}MT~7M
+CVuE{zkn_I$VoQkpby+x
+HRBjk3d9iqF{(ZpTHO3x
+vvc?O9puf4cf0519WU{k
+cQWUn>`SS~uJ=3ld5@?2
+;kW<^?LpyOB2LG*cMaB$
+Pr(Nn!T2VF4NB&S9GU6z
+OIn{r&PNYl+1~<%O~A%x
+e4=}L@rWGm6`wTS|7S;r
+QvJ0&*obI7>85Ulo^Cc#
+v6~*+jN9=sHelcDu*HMt
+?OirQ<ta3LL_mT$sz0tx
+#sdVp9ALjonpls{;n*y;
+&@OixmA3t*bG{(BB(q#_
+Sho&3ZqK%R?wg9!qZ(V7
+WuZcp7$Vnx5`E2q@v(3X
+s_)wqI74s1yN)`+EqEA@
+4!`^A7lE+ByGgMP#RX=4
+6z$@cj7cFQv?&wf{0`YV
+hullG75j!q=zTOS-t7|}
+$En&nz4HG?8YMg1qR@%q
+Ka3_TSQE-g`c$y3oeed@
+^DtIwh~u`1I}?%_YoQ46
+JIu${vrTa>Nc|(;KlqP6
+6LOiwlD!gyfbC2lR8m^r
+|E2@$G;zPiR7YiS1j-?t
+eVk5YoeG|jR{M*)y7`oX
+CMNV82X6<ZjE-~aY>uO_
+Zo^6~tANgvlfcy!mV279
+PuvAN$}9{ff$H?&$1kU=
+6=VhQQOBg1MXV;d8?r)C
+vO4sr-jC6ui(#7Srulkl
+Y~mf8C>RL_%E0|QX3_G)
+51K;71z0x4MySc2fC&`;
+Y>Nmu0Q5HQgQ`C)lK<Y8
+mf^niz-!~3(VozSO3_{=
+X#9>@TLS_<?~>Shwk;im
+Zi($N7{fR+yB6X*-SU4H
++{fOD#c3ZxfdnhO6DPP4
+ULgXBav>FWBvp0~bb_kF
+B93UL*Ka;9E0xL_G911`
+3$Y{-$ld2`^W}XXX6FF*
+m<!rPNo74=>_EK{bb#!r
+_<_Lbn;koC*K1rhal^`6
+C82lPqlSnTIo1vs-|N$3
+rbUuRE=2+s`f<VyM5XKU
+f$&qI$ErQ!X)oqMt5se)
+z`MjdTNkx5UQL|>)jI4T
+I-+u*gb~ngp-`D0-hSKc
+ZNI+r98l$|SFj!rBG6Fv
+Gg}wTcRyEPM>P7s&*@{M
+?-Ox%9>35#DGi`^1aQY$
+i8A)rSBBK+HX^%tm&JoQ
+pB#nDCwu&RGtKQta0xtT
+0Sf{va5*mem%E-Q@Vk$R
+cUMGXy5X%va3rg6Oa(;m
+fa(^RIvAhzYV)6CK0s|o
+&|*`2<=1M=86?=rE92!H
+TGwJ@Il2dfp;1c==o6~P
+c_q`dWH^3=Dnu%j5MpkQ
+P}}xcO^sz{V(_*Z&wU+}
+)c2l+DM-VD&~j2VLfNQf
+m-bjUq9@pM!A|Po=eJsA
+9&BI{+~0)U%ARm}3ef$z
+_eE*=im}}XY(zhL;)tYi
+RX`pE1stMF6q%+wy`s0k
+diJ&OJrPkDLTgjoxkOjh
+5u1aKW>c*?`QN25OAaTz
+PUCi-v*La=%8=Hz&vmop
+Ep2N+{xoP=eQl^WN;eS{
+27?~2Lj2~jHUXoZLRHif
+^{__*1)xV7x0e}_r`s#-
+Heu0H2H&MIKSS%|B2ARE
+(Oo>4UPu%qt9D7>r*+t%
+kKs!XBkb}qv=hBEWp`io
+Y0q8?0f#|Dz}HR|zJ@k`
+v2nEbMu)?+{XSyp9*`aA
+wUn;&Y3;IY#Ht8F?-x@)
+ft_v2%}6kIKdaF450Rt=
+4<N7X>_rX(l11k>SD@0l
+rgt^+RVIi~0@lbz)AJnL
+1`58}<V<;oBlBq<z{7g*
+8fQCuKm~Ltxaw7KF!yR#
+<zO7uoQ!C+8dNxCX2Uk6
+o;tweJcLW9OQ#`Vm7rMN
+2Qyh@w8r-tdZ)i?J--+f
+OIYa93NFG2@stPgySwz?
+1ja*FX$2?aa>Vk*L70^&
++Q%Jv5URqUtDwhTM>)h*
+Pv2Dk;UZ<HmoN{uja^|a
+gDkKXZ`w{+Cvl}Bxp$iD
+|By!YPhwrB*HLX(Fi9nS
+nVYfQ-)LwvLK^A`wYoNu
+{v=`)5|i*Hi%5T3k!bXw
+iApMpO2$kdf&g4=M^QV#
+8bm(_*(e@N@$hRx=<Hgy
+??5r-2eIj#q<GG_s5_@#
+U6%u3OY+v}D20TKUVy?q
+kFtp$M;ttznpuuKF;j-4
+APmn}qjIL_QpyMWG!!Rj
+L%$3Kp+jh+7%vUx-<?s7
+GtX9z4A8-z+{3{nQn7&l
+Jfa^EHSKBxy!vt867R?@
+$I^s2Psmlgq38-7Mq(h%
+MG`^GOH&Gshp2|84Z6>l
+UYo(V^?>9Ym|Goe_VpxT
+e6B*Xfzm3ro&yhoQv+@e
++WReJx=lO}B$kHCurafG
+{)67-af>RZzp*d|gFY8o
+zbh<666~qP`|Y3qZrSpt
+%Bdo%Es|q-pow%qRei^(
+@gwkoQSB*wK|Kgb7aA)&
+=)~NYUt<&SkTV78wLPR|
+wn}XBpR0UzidJxx&nxaH
+D*
+----END PYE FILE----

hydrodl2/core/calc/fdj.py

@@ -0,0 +1,92 @@
+import torch
+
+
+def finite_difference_jacobian(G, x, p, p2, t, epsilon, auxG, perturbed_p=0):
+    """Finite difference Jacobian."""
+    nb, nx = x.shape
+    if perturbed_p == 0:
+        ny = nx
+    elif perturbed_p == 1:
+        ny = p.shape[1]
+    elif perturbed_p == 2:
+        ny = p2.shape[1]
+    else:
+        print("perturbed_p is not set corretly")
+    xE = x.repeat_interleave(ny + 1, dim=0).double()
+    pE = p.repeat_interleave(ny + 1, dim=0).double()
+    p2E = p2.repeat_interleave(ny + 1, dim=0).double()
+
+    perturbation = torch.eye(ny).unsqueeze(0).expand(nb, -1, -1) * epsilon
+    perturbation = torch.cat([torch.zeros(nb, 1, ny), perturbation], dim=1).to(x)
+
+    expand_num = ny + 1
+    # Expand x and add perturbations
+    if perturbed_p == 0:
+        xE = xE + perturbation.reshape(nb * (ny + 1), ny)
+    elif perturbed_p == 1:
+        pE = pE + perturbation.reshape(nb * (ny + 1), ny)
+    elif perturbed_p == 2:
+        p2E = p2E + perturbation.reshape(nb * (ny + 1), ny)
+    else:
+        print("perturbed_p is not set corretly")
+
+    # Compute gg for all perturbed inputs in one run
+    ggE = G(xE, pE, p2E, t, [expand_num], auxG).view(nb, ny + 1, nx)
+
+    # Extract the original and perturbed gg
+    gg_original = ggE[:, 0, :]
+    gg_perturbed = ggE[:, 1:, :]
+
+    # Compute finite differences
+    dGdx = ((gg_perturbed - gg_original.unsqueeze(1)) / epsilon).permute(0, 2, 1)
+
+    return dGdx, gg_original
+
+
+def finite_difference_jacobian_p(G, x, p, p2, t, epsilon, auxG):
+    """Finite difference Jacobian."""
+    nb, np = p.shape
+    _, np2 = p2.shape
+    _, nx = x.shape
+
+    xE = torch.cat(
+        [x.repeat_interleave(np + 1, dim=0), x.repeat_interleave(np2, dim=0)], dim=0
+    ).double()
+    pE = torch.cat(
+        [p.repeat_interleave(np + 1, dim=0), p.repeat_interleave(np2, dim=0)], dim=0
+    ).double()
+    p2E = torch.cat(
+        [p2.repeat_interleave(np + 1, dim=0), p2.repeat_interleave(np2, dim=0)], dim=0
+    ).double()
+
+    perturbation_p = torch.eye(np).unsqueeze(0).expand(nb, -1, -1) * epsilon
+    perturbation_p2 = torch.eye(np2).unsqueeze(0).expand(nb, -1, -1) * epsilon
+
+    perturbation_p = torch.cat([torch.zeros(nb, 1, np), perturbation_p], dim=1).to(x)
+
+    perturbation_p2 = perturbation_p2.to(x)
+
+    pE[: nb * (np + 1), :] = pE[: nb * (np + 1), :] + perturbation_p.reshape(
+        nb * (np + 1), np
+    )
+    p2E[nb * (np + 1) :, :] = p2E[nb * (np + 1) :, :] + perturbation_p2.reshape(
+        nb * (np2), np2
+    )
+    # Expand x and add perturbations
+    # xE = x.unsqueeze(1).expand(-1, np + 1, -1).reshape(nb * (np + 1), np) + perturbation.reshape(nb * (np + 1), np)
+
+    # Compute gg for all perturbed inputs in one run
+    ggE = G(xE, pE, p2E, t, [np + 1, np2], auxG)
+    ggE_p = ggE[: nb * (np + 1), :].view(nb, np + 1, nx)
+    ggE_p2 = ggE[nb * (np + 1) :, :].view(nb, np2, nx)
+
+    # Extract the original and perturbed gg
+    gg_original_p = ggE_p[:, 0, :]
+    gg_perturbed_p = ggE_p[:, 1:, :]
+
+    gg_perturbed_p2 = ggE_p2
+
+    # Compute finite differences
+    dGdp = ((gg_perturbed_p - gg_original_p.unsqueeze(1)) / epsilon).permute(0, 2, 1)
+    dGdp2 = ((gg_perturbed_p2 - gg_original_p.unsqueeze(1)) / epsilon).permute(0, 2, 1)
+    return dGdp, dGdp2

hydrodl2/core/calc/uh_routing.py

@@ -0,0 +1,105 @@
+import torch
+import torch.nn.functional as F
+
+
+def uh_gamma(a, b, lenF=10):
+    """Unit hydrograph [time (same all time steps), batch, var]."""
+    m = a.shape
+    lenF = min(a.shape[0], lenF)
+    w = torch.zeros([lenF, m[1], m[2]])
+    aa = (
+        F.relu(a[0:lenF, :, :]).view([lenF, m[1], m[2]]) + 0.1
+    )  # minimum 0.1. First dimension of a is repeat
+    theta = F.relu(b[0:lenF, :, :]).view([lenF, m[1], m[2]]) + 0.5  # minimum 0.5
+    t = torch.arange(0.5, lenF * 1.0).view([lenF, 1, 1]).repeat([1, m[1], m[2]])
+    t = t.to(aa.device)
+    denom = (aa.lgamma().exp()) * (theta**aa)
+    mid = t ** (aa - 1)
+    right = torch.exp(-t / theta)
+    w = 1 / denom * mid * right
+    w = w / w.sum(0)  # scale to 1 for each UH
+
+    return w
+
+
+def uh_conv(x, UH, viewmode=1):
+    r"""Unit hydrograph convolution.
+
+    UH is a vector indicating the unit hydrograph the convolved dimension will
+    be the last dimension UH convolution is.
+    Q(t)= integral(x(\tao)*UH(t-\tao))d\tao
+    conv1d does
+    integral(w(\tao)*x(t+\tao))d\tao
+    hence we flip the UH
+    https://programmer.group/pytorch-learning-conv1d-conv2d-and-conv3d.html
+    view.
+
+    x: [batch, var, time]
+    UH:[batch, var, uhLen]
+    batch needs to be accommodated by channels and we make use of groups
+    https://pytorch.org/docs/stable/generated/torch.nn.Conv1d.html
+    https://pytorch.org/docs/stable/nn.functional.html
+    """
+    mm = x.shape
+    nb = mm[0]
+    m = UH.shape[-1]
+    padd = m - 1
+    if viewmode == 1:
+        xx = x.view([1, nb, mm[-1]])
+        w = UH.view([nb, 1, m])
+        groups = nb
+
+    y = F.conv1d(
+        xx, torch.flip(w, [2]), groups=groups, padding=padd, stride=1, bias=None
+    )
+    if padd != 0:
+        y = y[:, :, 0:-padd]
+    return y.view(mm)
+
+
+def source_flow_calculation(config, flow_out, c_NN, after_routing=True):
+    """Source flow calculation."""
+    varC_NN = config['var_c_nn']
+    if 'DRAIN_SQKM' in varC_NN:
+        area_name = 'DRAIN_SQKM'
+    elif 'area_gages2' in varC_NN:
+        area_name = 'area_gages2'
+    else:
+        print("area of basins are not available among attributes dataset")
+    area = (
+        c_NN[:, varC_NN.index(area_name)]
+        .unsqueeze(0)
+        .unsqueeze(-1)
+        .repeat(flow_out['flow_sim'].shape[0], 1, 1)
+    )
+    # flow calculation. converting mm/day to m3/sec
+    if after_routing:
+        srflow = (
+            (1000 / 86400) * area * (flow_out['srflow']).repeat(1, 1, config['nmul'])
+        )  # Q_t - gw - ss
+        ssflow = (
+            (1000 / 86400) * area * (flow_out['ssflow']).repeat(1, 1, config['nmul'])
+        )  # ras
+        gwflow = (
+            (1000 / 86400) * area * (flow_out['gwflow']).repeat(1, 1, config['nmul'])
+        )
+    else:
+        srflow = (
+            (1000 / 86400)
+            * area
+            * (flow_out['srflow_no_rout']).repeat(1, 1, config['nmul'])
+        )  # Q_t - gw - ss
+        ssflow = (
+            (1000 / 86400)
+            * area
+            * (flow_out['ssflow_no_rout']).repeat(1, 1, config['nmul'])
+        )  # ras
+        gwflow = (
+            (1000 / 86400)
+            * area
+            * (flow_out['gwflow_no_rout']).repeat(1, 1, config['nmul'])
+        )
+    # srflow = torch.clamp(srflow, min=0.0)  # to remove the small negative values
+    # ssflow = torch.clamp(ssflow, min=0.0)
+    # gwflow = torch.clamp(gwflow, min=0.0)
+    return srflow, ssflow, gwflow