ó Z™xTc@ s_ddlmZddlZd„Zd„Zd„Zd„Zd„Zd„Z d „Z d „Z d „Z d „Z d „Zd„Zd„Zddd„ƒYZedkr[ddddddgZeddƒZeddƒZxreD]gZx]eD]UZededgZedGedGx'eD]ZeeƒdƒjeƒGq+WHqúWHqíWndS(iÿÿÿÿ(tdivisionNcC s8itd6td6td6td6td6td6}||S(sF Easy way to get a function constructor based on its name tspheretackleyt michalewiczt rosenbrockt rastriginteasom(tSpheretAckleyt Michalewiczt Rosenbrockt RastrigintEasom(tnametfuncs((sG/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/scitbx/examples/functions.pytFunctions  cC s(ttdg|dg|dg|ƒS(sN returns the sphere objective function with mins and maxs defined gÂõ(\ÀgÂõ(\@g(tObjFunct_sphere(tdim((sG/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/scitbx/examples/functions.pyRscC s(ttdg|dg|dg|ƒS(sN returns the ackley objective function with mins and maxs defined g>Àg>@g(Rt_ackley(R((sG/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/scitbx/examples/functions.pyRscC s+ttdg|tjg|dg|ƒS(sS returns the michalewicz objective function with mins and maxs defined g(Rt _michalewicztmtpi(R((sG/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/scitbx/examples/functions.pyR scC s(ttdg|dg|dg|ƒS(sQ returns the Rastrigin objective function with mins and maxs defined gÀg@g(Rt _rastrigin(R((sG/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/scitbx/examples/functions.pyR 'scC s(ttdg|dg|dg|ƒS(sR returns the Rosenbrock objective function with mins and maxs defined gÀg$@gð?(Rt _rosenbrock(R((sG/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/scitbx/examples/functions.pyR .scC s=|dkr‚nttddgddgtjtjgƒS(sM returns the Easom objective function with mins and maxs defined ig>Àg>@(Rt_easomRR(R((sG/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/scitbx/examples/functions.pyR 5s cC s!tg|D]}|d^q ƒS(s" the sphere (sphere) function g@(tsum(tcoordstx((sG/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/scitbx/examples/functions.pyR>scC s­t|ƒ}d}d}dtj}d}}x3|D]+}||d7}|tj||ƒ7}q6W| tj| tjd||ƒƒtjd||ƒ|tjS(s the ackley function g4@gš™™™™™É?g@ggð?(tlenRRtcostexptsqrtte(Rtntatbtcts1ts2ti((sG/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/scitbx/examples/functions.pyRDs    c C sAdtg|D],}|ddtjdtj|ƒ^q ƒS(s rastrigin function ig@g$@(RRRR(RR((sG/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/scitbx/examples/functions.pyRUscC s}t|ƒ}|d}|d|!}||}d}xCt||ƒD]2\}}|d|||d|dd7}qCW|S(s rosenbrock function iiidg@gð?(Rtzip(RRtpRtytresulttxxtyy((sG/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/scitbx/examples/functions.pyR[s    *cC s]d}tgt|ƒD]?\}}tj|ƒtj||dtjƒd|^qƒ S(s michalewicz function g$@g@(Rt enumerateRtsinR(RtrR(tcoord((sG/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/scitbx/examples/functions.pyRgscC sW|d}|d}tj|ƒ tj|ƒtj|tjd |tjdƒS(s easom function iig@(RRRR(Rtx1tx2((sG/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/scitbx/examples/functions.pyRns  RcB s eZdZd„Zd„ZRS(s\ This is our objective function. It contains the function as well as its bounds cC s(||_||_||_||_dS(N(tfunctminstmaxstxstar(tselfR5R6R7R8((sG/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/scitbx/examples/functions.pyt__init__zs   cC s |j|ƒS(s Evaluates the function (R5(R9R((sG/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/scitbx/examples/functions.pyteval€s(t__name__t __module__t__doc__R:R;(((sG/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/scitbx/examples/functions.pyRus t__main__RRRRRRiØÿÿÿi)g$@i((t __future__RtmathRRRRR R R R RRRRRRRR<RtrangeRR+R-R.ttmpR R;(((sG/public2/ftp/pub/ccp4/6.5/ccp4-6.5/lib/py2/scitbx/examples/functions.pyts@