U me@s ddlmZddZddZdS)wrapscs|fdd}|S)z} Memo decorator for sequences defined by recurrence See usage examples e.g. in the specfun/combinatorial module cstfdd}|S)NcsHt}||dkr|St||dD]}|q*dS)Nlenrangeappend)nLi)cachef?/tmp/pip-unpacked-wheel-rdz2gdd2/sympy/utilities/memoization.pyg s  z-recurrence_memo..decorator..grrrr rr decorator sz"recurrence_memo..decoratorr)initialrrrrrecurrence_memos rcsgfdd}|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 cstfdd}|S)Ncs~t}||kr||St||dD]F}|}|g}|td|dD]}||}||qTq*||S)Nrr)r mr r ZF_i0Z F_i_cachejZF_ij)base_seqr rrrr&s   z3assoc_recurrence_memo..decorator..grrrr rrr%sz(assoc_recurrence_memo..decoratorr)rrrrrassoc_recurrence_memos rN) functoolsrrrrrrrs