MZdddlmZdZdZy)wrapsc|fd}|S)z} Memo decorator for sequences defined by recurrence See usage examples e.g. in the specfun/combinatorial module c2tfd}|S)Nct}||dz kr|St||dzD]}j|dS)Nlenrangeappend)nLicachefs =/usr/lib/python3/dist-packages/sympy/utilities/memoization.pygz-recurrence_memo..decorator..g sUE AAEzQx1a!e_ * Qq%[) *9 r)rrrs` r decoratorz"recurrence_memo..decorator s q   r)initialrrs @rrecurrence_memors E  rcgfd}|S)z Memo decorator for associated sequences defined by recurrence starting from base base_seq(n) -- callable to get base sequence elements XXX works only for Pn0 = base_seq(0) cases XXX works only for m <= n cases c4tfd}|S)Nct }||kr ||St||dzD]M}|}|g} j|td|dzD]} || }|j|O ||S)Nrr ) rmrrF_i0 F_i_cachejF_ijbase_seqrrs rrz3assoc_recurrence_memo..decorator..g&sE A1uQx{"1a!e_ +{!F  Y'q!a%+AQ5>D$$T*+ +8A; rr)rrr"rs` rrz(assoc_recurrence_memo..decorator%s q   &rr)r"rrs` @rassoc_recurrence_memor#s E, rN) functoolsrrr#rrrr%s*"r