shapes.py 136 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700270127022703270427052706270727082709271027112712271327142715271627172718271927202721272227232724272527262727272827292730273127322733273427352736273727382739274027412742274327442745274627472748274927502751275227532754275527562757275827592760276127622763276427652766276727682769277027712772277327742775277627772778277927802781278227832784278527862787278827892790279127922793279427952796279727982799280028012802280328042805280628072808280928102811281228132814281528162817281828192820282128222823282428252826282728282829283028312832283328342835283628372838283928402841284228432844284528462847284828492850285128522853285428552856285728582859286028612862286328642865286628672868286928702871287228732874287528762877287828792880288128822883288428852886288728882889289028912892289328942895289628972898289929002901290229032904290529062907290829092910291129122913291429152916291729182919292029212922292329242925292629272928292929302931293229332934293529362937293829392940294129422943294429452946294729482949295029512952295329542955295629572958295929602961
  1. from __future__ import division
  2. from __future__ import unicode_literals
  3. from __future__ import print_function
  4. from __future__ import absolute_import
  5. from future import standard_library
  6. standard_library.install_aliases()
  7. from builtins import input
  8. from builtins import zip
  9. from builtins import str
  10. from builtins import range
  11. from builtins import *
  12. from builtins import object
  13. from numpy import linspace, sin, cos, pi, array, asarray, ndarray, sqrt, abs
  14. import pprint, copy, glob, os
  15. from math import radians
  16. from io import BytesIO
  17. from ruamel.yaml import YAML
  18. from PIL import Image
  19. import ast
  20. from .MatplotlibDraw import MatplotlibDraw
  21. drawing_tool = MatplotlibDraw()
  22. def sketch2SVG():
  23. """
  24. Save sketch as a svg string
  25. """
  26. f = BytesIO()
  27. drawing_tool.mpl.savefig(f, format="svg")
  28. return f.getvalue()
  29. def sketch2PNG():
  30. """
  31. Save sketch as a python .png image
  32. """
  33. f = BytesIO()
  34. drawing_tool.mpl.gcf().canvas.print_png(f)
  35. img = Image.open(f)
  36. return img
  37. def sVe(key, expression, container, sketch):
  38. """
  39. sVe: sketch Validate expression
  40. given an expression from a sketch, check if valid or not
  41. provides a string feedback if error else return 1
  42. """
  43. try:
  44. root = ast.parse(expression)
  45. except Exception as e:
  46. return f"{sketch}/{key}: '''{expression}''' parse error {str(e)}"
  47. names = {node.id for node in ast.walk(root) if isinstance(node, ast.Name)}
  48. for name in names:
  49. if name not in container:
  50. return f"{sketch}/{key}: {name} in {expression} is not defined"
  51. return 1
  52. def sketchParse(sketch, container):
  53. """
  54. Parse a string sketch into a container.
  55. name specifices the 'name' of the sketch
  56. A container is a name space which holds
  57. - all the libraries references needed to create pysketcher shapes
  58. - all the variable providing shapes dimensions or position
  59. - all the shapes needed to create the shapes it defines
  60. """
  61. yaml = YAML()
  62. psketch = yaml.load(sketch)
  63. sketch_name = psketch["name"]
  64. gwd = psketch['shapes']
  65. for _k in list(gwd.keys()):
  66. if _k == "stop":
  67. return True
  68. _c = gwd[_k]
  69. _t = str(type(_c))
  70. if _k == "libraries":
  71. for l in _c:
  72. _r = sVe(_k, l, container, sketch_name)
  73. if type(_r) == str:
  74. print(_r)
  75. return False
  76. exec(l,container)
  77. #print(_k, _c, _t)
  78. if _t == "<class 'ruamel.yaml.scalarfloat.ScalarFloat'>" or \
  79. _t == "<class 'str'>" or _t == "<class 'int'>":
  80. _expression = f"{_c}".replace("<bslash>","\\")
  81. _formula = f"{_k} = {_expression}"
  82. #print(_formula)
  83. _r = sVe(_k, _expression, container, sketch_name)
  84. if type(_r) == str:
  85. print(_r)
  86. return False
  87. exec(_formula,container)
  88. elif _t == "<class 'ruamel.yaml.comments.CommentedMap'>":
  89. #print(_c)
  90. _keys = list(_c.keys())
  91. #print(_keys)
  92. if 'formula' in _keys:
  93. _expression = f"{_c['formula']}".replace("<bslash>","\\")
  94. _formula = f"{_k} = {_expression}"
  95. #print(_formula)
  96. _r = sVe(_k, _expression, container, sketch_name)
  97. if type(_r) == str:
  98. print(_r)
  99. return False
  100. exec(_formula,container)
  101. # if the new object is a shape and has the sketch name, set this shape name as the sketch name
  102. if issubclass(type(container[_k]), Shape):
  103. if _k == sketch_name:
  104. container[_k].set_name(sketch_name)
  105. if 'style' in _keys:
  106. for _style in _c["style"]:
  107. # x_const.set_linestyle('dotted')
  108. _param = _c["style"][_style]
  109. __t = str(type(_param))
  110. #print(__t)
  111. if __t == "<class 'int'>":
  112. _style = f"{_k}.set_{_style}({_param})"
  113. else:
  114. _style = f"{_k}.set_{_style}('{_param}')"
  115. #print(_style)
  116. exec(_style,container)
  117. if 'transform' in _keys:
  118. #print(_c['transform'])
  119. if str(type(_c['transform'])) == "<class 'str'>":
  120. _t = f"{_k}.{_c['transform']}"
  121. #print(_t)
  122. _r = sVe(_k, _formula, container, sketch_name)
  123. if type(_r) == str:
  124. print(_r)
  125. return False
  126. exec(_t,container)
  127. else:
  128. for _transform in _c["transform"]:
  129. # x_const.rotate(-theta, contact)
  130. _t = f"{_k}.{_transform}"
  131. #print(_t)
  132. _r = sVe(_k, _t, container, sketch_name)
  133. if type(_r) == str:
  134. print(_r)
  135. return False
  136. exec(_t,container)
  137. if "action" in _keys:
  138. _action = _c["action"]
  139. #print(_action)
  140. _r = sVe(_k, _action, container, sketch_name)
  141. if type(_r) == str:
  142. print(_r)
  143. return False
  144. exec(_action,container)
  145. return True
  146. def point(x, y, check_inside=False):
  147. for obj, name in zip([x, y], ['x', 'y']):
  148. if isinstance(obj, (float,int)):
  149. pass
  150. elif isinstance(obj, ndarray):
  151. if obj.size == 1:
  152. pass
  153. else:
  154. raise TypeError('%s=%s of type %d has length=%d > 1' %
  155. (name, obj, type(obj), obj.size))
  156. else:
  157. raise TypeError('%s=%s is of wrong type %d' %
  158. (name, obj, type(obj)))
  159. if check_inside:
  160. ok, msg = drawing_tool.inside((x,y), exception=True)
  161. if not ok:
  162. print(msg)
  163. return array((x, y), dtype=float)
  164. def distance(p1, p2):
  165. p1 = arr2D(p1); p2 = arr2D(p2)
  166. d = p2 - p1
  167. return sqrt(d[0]**2 + d[1]**2)
  168. def unit_vec(x, y=None):
  169. """Return unit vector of the vector (x,y), or just x if x is a 2D point."""
  170. if isinstance(x, (float,int)) and isinstance(y, (float,int)):
  171. x = point(x, y)
  172. elif isinstance(x, (list,tuple,ndarray)) and y is None:
  173. return arr2D(x)/sqrt(x[0]**2 + x[1]**2)
  174. else:
  175. raise TypeError('x=%s is %s, must be float or ndarray 2D point' %
  176. (x, type(x)))
  177. def arr2D(x, check_inside=False):
  178. if isinstance(x, (tuple,list,ndarray)):
  179. if len(x) == 2:
  180. pass
  181. else:
  182. raise ValueError('x=%s has length %d, not 2' % (x, len(x)))
  183. else:
  184. raise TypeError('x=%s must be list/tuple/ndarray, not %s' %
  185. (x, type(x)))
  186. if check_inside:
  187. ok, msg = drawing_tool.inside(x, exception=True)
  188. if not ok:
  189. print(msg)
  190. return asarray(x, dtype=float)
  191. def _is_sequence(seq, length=None,
  192. can_be_None=False, error_message=True):
  193. if can_be_None:
  194. legal_types = (list,tuple,ndarray,None)
  195. else:
  196. legal_types = (list,tuple,ndarray)
  197. if isinstance(seq, legal_types):
  198. if length is not None:
  199. if length == len(seq):
  200. return True
  201. elif error_message:
  202. raise TypeError('sequence %s is not a sequence but %s; must be %s of length %d' %
  203. (str(seq), type(seq),
  204. ', '.join([str(t) for t in legal_types]),
  205. len(seq)))
  206. else:
  207. return False
  208. else:
  209. return True
  210. elif error_message:
  211. raise TypeError('sequence %s is not a sequence but %s, %s; must be %s' %
  212. (str(seq), seq.__class__.__name__, type(seq),
  213. ','.join([str(t)[5:-1] for t in legal_types])))
  214. else:
  215. return False
  216. def is_sequence(*sequences, **kwargs):
  217. length = kwargs.get('length', 2)
  218. can_be_None = kwargs.get('can_be_None', False)
  219. error_message = kwargs.get('error_message', True)
  220. check_inside = kwargs.get('check_inside', False)
  221. for x in sequences:
  222. _is_sequence(x, length=length, can_be_None=can_be_None,
  223. error_message=error_message)
  224. if check_inside:
  225. ok, msg = drawing_tool.inside(x, exception=True)
  226. if not ok:
  227. print(msg)
  228. def animate(fig, time_points, action, moviefiles=False,
  229. pause_per_frame=0.5, show_screen_graphics=True,
  230. title=None,
  231. **action_kwargs):
  232. if moviefiles:
  233. # Clean up old frame files
  234. framefilestem = 'tmp_frame_'
  235. framefiles = glob.glob('%s*.png' % framefilestem)
  236. for framefile in framefiles:
  237. os.remove(framefile)
  238. for n, t in enumerate(time_points):
  239. drawing_tool.erase()
  240. action(t, fig, **action_kwargs)
  241. #could demand returning fig, but in-place modifications
  242. #are done anyway
  243. #fig = action(t, fig)
  244. #if fig is None:
  245. # raise TypeError(
  246. # 'animate: action returns None, not fig\n'
  247. # '(a Shape object with the whole figure)')
  248. fig.draw()
  249. drawing_tool.display(title=title, show=show_screen_graphics)
  250. if moviefiles:
  251. drawing_tool.savefig('%s%04d.png' % (framefilestem, n),
  252. crop=False)
  253. if moviefiles:
  254. return '%s%%04d.png' % framefilestem
  255. """
  256. def save():
  257. os.system("ffmpeg -r 1 -i img%01d.png -vcodec mpeg4 -y movie.mp4")
  258. """
  259. class Shape(object):
  260. """
  261. Superclass for drawing different geometric shapes.
  262. Subclasses define shapes, but drawing, rotation, translation,
  263. etc. are done in generic functions in this superclass.
  264. """
  265. def __init__(self):
  266. """
  267. Never to be called from subclasses.
  268. """
  269. raise NotImplementedError(
  270. 'class %s must implement __init__,\nwhich defines '
  271. 'self.shapes as a dict (or list) of Shape objects\n'
  272. 'Do not call Shape.__init__!' % \
  273. self.__class__.__name__)
  274. def set_name(self, name):
  275. self.name = name
  276. return self
  277. def get_name(self):
  278. return self.name if hasattr(self, 'name') else 'no_name'
  279. def __iter__(self):
  280. # We iterate over self.shapes many places, and will
  281. # get here if self.shapes is just a Shape object and
  282. # not the assumed dict/list.
  283. print('Warning: class %s does not define self.shapes\n'\
  284. 'as a dict of Shape objects')
  285. return [self] # Make the iteration work
  286. def copy(self):
  287. return copy.deepcopy(self)
  288. def __getitem__(self, name):
  289. """
  290. Allow indexing like::
  291. obj1['name1']['name2']
  292. all the way down to ``Curve`` or ``Point`` (``Text``)
  293. objects.
  294. """
  295. if hasattr(self, 'shapes'):
  296. if name in self.shapes:
  297. return self.shapes[name]
  298. else:
  299. for shape in self.shapes:
  300. if isinstance(self.shapes[shape], (Curve,Point)):
  301. # Indexing of Curve/Point/Text is not possible
  302. raise TypeError(
  303. 'Index "%s" (%s) is illegal' %
  304. (name, self.__class__.__name__))
  305. return self.shapes[shape][name]
  306. else:
  307. raise Exception('This is a bug in __getitem__')
  308. def __setitem__(self, name, value):
  309. """
  310. Allow assignment like::
  311. obj1['name1']['name2'] = value
  312. all the way down to ``Curve`` or ``Point`` (``Text``)
  313. objects.
  314. """
  315. if hasattr(self, 'shapes'):
  316. self.shapes[name] = value
  317. else:
  318. raise Exception('Cannot assign')
  319. def _for_all_shapes(self, func, *args, **kwargs):
  320. verbose = kwargs.get('verbose', 0)
  321. if not hasattr(self, 'shapes'):
  322. # When self.shapes is lacking, we either come to
  323. # a special implementation of func or we come here
  324. # because Shape.func is just inherited. This is
  325. # an error if the class is not Curve or Point
  326. if isinstance(self, (Curve, Point)):
  327. return # ok: no shapes, but object is a curve or point end leaf
  328. else:
  329. raise AttributeError('class %s has no shapes attribute!' %
  330. self.__class__.__name__)
  331. is_dict = True if isinstance(self.shapes, dict) else False
  332. for k, shape in enumerate(self.shapes):
  333. if is_dict:
  334. shape_name = shape
  335. shape = self.shapes[shape]
  336. else:
  337. shape_name = k # use index as name if list (not dict)
  338. if not isinstance(shape, Shape):
  339. if isinstance(shape, dict):
  340. raise TypeError(
  341. 'class %s has a self.shapes member "%s" that is just\n'
  342. 'a plain dictionary,\n%s\n'
  343. 'Did you mean to embed this dict in a Composition\n'
  344. 'object?' % (self.__class__.__name__, shape_name,
  345. str(shape)))
  346. elif isinstance(shape, (list,tuple)):
  347. raise TypeError(
  348. 'class %s has self.shapes member "%s" containing\n'
  349. 'a %s object %s,\n'
  350. 'Did you mean to embed this list in a Composition\n'
  351. 'object?' % (self.__class__.__name__, shape_name,
  352. type(shape), str(shape)))
  353. elif shape is None:
  354. raise TypeError(
  355. 'class %s has a self.shapes member "%s" that is None.\n'
  356. 'Some variable name is wrong, or some function\n'
  357. 'did not return the right object...' \
  358. % (self.__class__.__name__, shape_name))
  359. else:
  360. raise TypeError(
  361. 'class %s has a self.shapes member "%s" of %s which '
  362. 'is not a Shape object\n%s' %
  363. (self.__class__.__name__, shape_name, type(shape),
  364. pprint.pformat(self.shapes)))
  365. if isinstance(shape, Curve):
  366. shape.name = shape_name
  367. if verbose > 0:
  368. print('calling %s.%s' % (shape_name, func))
  369. getattr(shape, func)(*args, **kwargs)
  370. def draw(self, verbose=0):
  371. self._for_all_shapes('draw', verbose=verbose)
  372. return self
  373. def draw_dimensions(self):
  374. if hasattr(self, 'dimensions'):
  375. for shape in self.dimensions:
  376. self.dimensions[shape].draw()
  377. return self
  378. else:
  379. #raise AttributeError('no self.dimensions dict for defining dimensions of class %s' % self.__classname__.__name__)
  380. return self
  381. def rotate(self, angle, center):
  382. is_sequence(center, length=2)
  383. self._for_all_shapes('rotate', angle, center)
  384. return self
  385. def translate(self, vec):
  386. is_sequence(vec, length=2)
  387. self._for_all_shapes('translate', vec)
  388. return self
  389. def scale(self, factor):
  390. self._for_all_shapes('scale', factor)
  391. return self
  392. def deform(self, displacement_function):
  393. self._for_all_shapes('deform', displacement_function)
  394. return self
  395. def minmax_coordinates(self, minmax=None):
  396. if minmax is None:
  397. minmax = {'xmin': 1E+20, 'xmax': -1E+20,
  398. 'ymin': 1E+20, 'ymax': -1E+20}
  399. self._for_all_shapes('minmax_coordinates', minmax)
  400. return minmax
  401. def recurse(self, name, indent=0):
  402. if not isinstance(self.shapes, dict):
  403. raise TypeError('recurse works only with dict self.shape, not %s' %
  404. type(self.shapes))
  405. space = ' '*indent
  406. print(space, '%s: %s.shapes has entries' % \
  407. (self.__class__.__name__, name), \
  408. str(list(self.shapes.keys()))[1:-1])
  409. for shape in self.shapes:
  410. print(space, end=' ')
  411. print('call %s.shapes["%s"].recurse("%s", %d)' % \
  412. (name, shape, shape, indent+2))
  413. self.shapes[shape].recurse(shape, indent+2)
  414. def graphviz_dot(self, name, classname=True):
  415. if not isinstance(self.shapes, dict):
  416. raise TypeError('recurse works only with dict self.shape, not %s' %
  417. type(self.shapes))
  418. dotfile = name + '.dot'
  419. pngfile = name + '.png'
  420. if classname:
  421. name = r"%s:\n%s" % (self.__class__.__name__, name)
  422. couplings = self._object_couplings(name, classname=classname)
  423. # Insert counter for similar names
  424. from collections import defaultdict
  425. count = defaultdict(lambda: 0)
  426. couplings2 = []
  427. for i in range(len(couplings)):
  428. parent, child = couplings[i]
  429. count[child] += 1
  430. parent += ' (%d)' % count[parent]
  431. child += ' (%d)' % count[child]
  432. couplings2.append((parent, child))
  433. print('graphviz', couplings, count)
  434. # Remove counter for names there are only one of
  435. for i in range(len(couplings)):
  436. parent2, child2 = couplings2[i]
  437. parent, child = couplings[i]
  438. if count[parent] > 1:
  439. parent = parent2
  440. if count[child] > 1:
  441. child = child2
  442. couplings[i] = (parent, child)
  443. print(couplings)
  444. f = open(dotfile, 'w')
  445. f.write('digraph G {\n')
  446. for parent, child in couplings:
  447. f.write('"%s" -> "%s";\n' % (parent, child))
  448. f.write('}\n')
  449. f.close()
  450. print('Run dot -Tpng -o %s %s' % (pngfile, dotfile))
  451. def _object_couplings(self, parent, couplings=[], classname=True):
  452. """Find all couplings of parent and child objects in a figure."""
  453. for shape in self.shapes:
  454. if classname:
  455. childname = r"%s:\n%s" % \
  456. (self.shapes[shape].__class__.__name__, shape)
  457. else:
  458. childname = shape
  459. couplings.append((parent, childname))
  460. self.shapes[shape]._object_couplings(childname, couplings,
  461. classname)
  462. return couplings
  463. def set_linestyle(self, style):
  464. styles = ('solid', 'dashed', 'dashdot', 'dotted')
  465. if style not in styles:
  466. raise ValueError('%s: style=%s must be in %s' %
  467. (self.__class__.__name__ + '.set_linestyle:',
  468. style, str(styles)))
  469. self._for_all_shapes('set_linestyle', style)
  470. return self
  471. def set_linewidth(self, width):
  472. if not isinstance(width, int) and width >= 0:
  473. raise ValueError('%s: width=%s must be positive integer' %
  474. (self.__class__.__name__ + '.set_linewidth:',
  475. width))
  476. self._for_all_shapes('set_linewidth', width)
  477. return self
  478. def set_linecolor(self, color):
  479. if color in drawing_tool.line_colors:
  480. color = drawing_tool.line_colors[color]
  481. elif color in list(drawing_tool.line_colors.values()):
  482. pass # color is ok
  483. else:
  484. raise ValueError('%s: invalid color "%s", must be in %s' %
  485. (self.__class__.__name__ + '.set_linecolor:',
  486. color, list(drawing_tool.line_colors.keys())))
  487. self._for_all_shapes('set_linecolor', color)
  488. return self
  489. def set_arrow(self, style):
  490. styles = ('->', '<-', '<->')
  491. if not style in styles:
  492. raise ValueError('%s: style=%s must be in %s' %
  493. (self.__class__.__name__ + '.set_arrow:',
  494. style, styles))
  495. self._for_all_shapes('set_arrow', style)
  496. return self
  497. def set_filled_curves(self, color='', pattern=''):
  498. if color in drawing_tool.line_colors:
  499. color = drawing_tool.line_colors[color]
  500. elif color in list(drawing_tool.line_colors.values()):
  501. pass # color is ok
  502. else:
  503. raise ValueError('%s: invalid color "%s", must be in %s' %
  504. (self.__class__.__name__ + '.set_filled_curves:',
  505. color, list(drawing_tool.line_colors.keys())))
  506. self._for_all_shapes('set_filled_curves', color, pattern)
  507. return self
  508. def set_shadow(self, pixel_displacement=3):
  509. self._for_all_shapes('set_shadow', pixel_displacement)
  510. return self
  511. def show_hierarchy(self, indent=0, format='std'):
  512. """Recursive pretty print of hierarchy of objects."""
  513. if not isinstance(self.shapes, dict):
  514. print('cannot print hierarchy when %s.shapes is not a dict' % \
  515. self.__class__.__name__)
  516. s = ''
  517. if format == 'dict':
  518. s += '{'
  519. for shape in self.shapes:
  520. if format == 'dict':
  521. shape_str = repr(shape) + ':'
  522. elif format == 'plain':
  523. shape_str = shape
  524. else:
  525. shape_str = shape + ':'
  526. if format == 'dict' or format == 'plain':
  527. class_str = ''
  528. else:
  529. class_str = ' (%s)' % \
  530. self.shapes[shape].__class__.__name__
  531. s += '\n%s%s%s %s,' % (
  532. ' '*indent,
  533. shape_str,
  534. class_str,
  535. self.shapes[shape].show_hierarchy(indent+4, format))
  536. if format == 'dict':
  537. s += '}'
  538. return s
  539. def __str__(self):
  540. """Display hierarchy with minimum information (just object names)."""
  541. return self.show_hierarchy(format='plain')
  542. def __repr__(self):
  543. """Display hierarchy as a dictionary."""
  544. return self.show_hierarchy(format='dict')
  545. #return pprint.pformat(self.shapes)
  546. class Curve(Shape):
  547. """General curve as a sequence of (x,y) coordinates."""
  548. def __init__(self, x, y):
  549. """
  550. `x`, `y`: arrays holding the coordinates of the curve.
  551. """
  552. self.x = asarray(x, dtype=float)
  553. self.y = asarray(y, dtype=float)
  554. #self.shapes must not be defined in this class
  555. #as self.shapes holds children objects:
  556. #Curve has no children (end leaf of self.shapes tree)
  557. self.linestyle = None
  558. self.linewidth = None
  559. self.linecolor = None
  560. self.fillcolor = None
  561. self.fillpattern = None
  562. self.arrow = None
  563. self.shadow = False
  564. self.name = None # name of object that this Curve represents
  565. def inside_plot_area(self, verbose=True):
  566. """Check that all coordinates are within drawing_tool's area."""
  567. xmin, xmax = self.x.min(), self.x.max()
  568. ymin, ymax = self.y.min(), self.y.max()
  569. t = drawing_tool
  570. inside = True
  571. if not hasattr(t, 'xmin'):
  572. return None # drawing area is not defined
  573. if xmin < t.xmin:
  574. inside = False
  575. if verbose:
  576. print('x_min=%g < plot area x_min=%g' % (xmin, t.xmin))
  577. if xmax > t.xmax:
  578. inside = False
  579. if verbose:
  580. print('x_max=%g > plot area x_max=%g' % (xmax, t.xmax))
  581. if ymin < t.ymin:
  582. inside = False
  583. if verbose:
  584. print('y_min=%g < plot area y_min=%g' % (ymin, t.ymin))
  585. if ymax > t.ymax:
  586. inside = False
  587. if verbose:
  588. print('y_max=%g > plot area y_max=%g' % (ymax, t.ymax))
  589. return inside
  590. def draw(self, verbose=0):
  591. """
  592. Send the curve to the plotting engine. That is, convert
  593. coordinate information in self.x and self.y, together
  594. with optional settings of linestyles, etc., to
  595. plotting commands for the chosen engine.
  596. """
  597. self.inside_plot_area()
  598. drawing_tool.plot_curve(
  599. self.x, self.y,
  600. self.linestyle, self.linewidth, self.linecolor,
  601. self.arrow, self.fillcolor, self.fillpattern,
  602. self.shadow, self.name)
  603. if verbose:
  604. print('drawing Curve object with %d points' % len(self.x))
  605. def rotate(self, angle, center):
  606. """
  607. Rotate all coordinates: `angle` is measured in degrees and
  608. (`x`,`y`) is the "origin" of the rotation.
  609. """
  610. angle = radians(angle)
  611. x, y = center
  612. c = cos(angle); s = sin(angle)
  613. xnew = x + (self.x - x)*c - (self.y - y)*s
  614. ynew = y + (self.x - x)*s + (self.y - y)*c
  615. self.x = xnew
  616. self.y = ynew
  617. return self
  618. def scale(self, factor):
  619. """Scale all coordinates by `factor`: ``x = factor*x``, etc."""
  620. self.x = factor*self.x
  621. self.y = factor*self.y
  622. return self
  623. def translate(self, vec):
  624. """Translate all coordinates by a vector `vec`."""
  625. self.x += vec[0]
  626. self.y += vec[1]
  627. return self
  628. def deform(self, displacement_function):
  629. """Displace all coordinates according to displacement_function(x,y)."""
  630. for i in range(len(self.x)):
  631. self.x[i], self.y[i] = displacement_function(self.x[i], self.y[i])
  632. return self
  633. def minmax_coordinates(self, minmax=None):
  634. if minmax is None:
  635. minmax = {'xmin': [], 'xmax': [], 'ymin': [], 'ymax': []}
  636. minmax['xmin'] = min(self.x.min(), minmax['xmin'])
  637. minmax['xmax'] = max(self.x.max(), minmax['xmax'])
  638. minmax['ymin'] = min(self.y.min(), minmax['ymin'])
  639. minmax['ymax'] = max(self.y.max(), minmax['ymax'])
  640. return minmax
  641. def recurse(self, name, indent=0):
  642. space = ' '*indent
  643. print(space, 'reached "bottom" object %s' % \
  644. self.__class__.__name__)
  645. def _object_couplings(self, parent, couplings=[], classname=True):
  646. return
  647. def set_linecolor(self, color):
  648. self.linecolor = color
  649. return self
  650. def set_linewidth(self, width):
  651. self.linewidth = width
  652. return self
  653. def set_linestyle(self, style):
  654. self.linestyle = style
  655. return self
  656. def set_arrow(self, style=None):
  657. self.arrow = style
  658. return self
  659. def set_filled_curves(self, color='', pattern=''):
  660. self.fillcolor = color
  661. self.fillpattern = pattern
  662. return self
  663. def set_shadow(self, pixel_displacement=3):
  664. self.shadow = pixel_displacement
  665. return self
  666. def show_hierarchy(self, indent=0, format='std'):
  667. if format == 'dict':
  668. return '"%s"' % str(self)
  669. elif format == 'plain':
  670. return ''
  671. else:
  672. return str(self)
  673. def __str__(self):
  674. """Compact pretty print of a Curve object."""
  675. s = '%d (x,y) coords' % self.x.size
  676. inside = self.inside_plot_area(verbose=False)
  677. if inside is None:
  678. pass # no info about the plotting area
  679. elif not inside:
  680. s += ', some coordinates are outside plotting area!\n'
  681. props = ('linecolor', 'linewidth', 'linestyle', 'arrow',
  682. 'fillcolor', 'fillpattern')
  683. for prop in props:
  684. value = getattr(self, prop)
  685. if value is not None:
  686. s += ' %s=%s' % (prop, repr(value))
  687. return s
  688. def __repr__(self):
  689. return str(self)
  690. class Spline(Shape):
  691. # Note: UnivariateSpline interpolation may not work if
  692. # the x[i] points are far from uniformly spaced
  693. def __init__(self, x, y, degree=3, resolution=501):
  694. from scipy.interpolate import UnivariateSpline
  695. self.smooth = UnivariateSpline(x, y, s=0, k=degree)
  696. self.xcoor = linspace(x[0], x[-1], resolution)
  697. ycoor = self.smooth(self.xcoor)
  698. self.shapes = {'smooth': Curve(self.xcoor, ycoor)}
  699. def geometric_features(self):
  700. s = self.shapes['smooth']
  701. return {'start': point(s.x[0], s.y[0]),
  702. 'end': point(s.x[-1], s.y[-1]),
  703. 'interval': [s.x[0], s.x[-1]]}
  704. def __call__(self, x):
  705. return self.smooth(x)
  706. # Can easily find the derivative and the integral as
  707. # self.smooth.derivative(n=1) and self.smooth.antiderivative()
  708. class SketchyFunc1(Spline):
  709. """
  710. A typical function curve used to illustrate an "arbitrary" function.
  711. """
  712. domain = [1, 6]
  713. def __init__(self, name=None, name_pos='start',
  714. xmin=0, xmax=6, ymin=0, ymax=2):
  715. x = array([0, 2, 3, 4, 5, 6])
  716. y = array([1, 1.8, 1.2, 0.7, 0.8, 0.85])
  717. #y = array([5, 3.5, 3.8, 3, 2.5, 2.4])
  718. # Scale x and y
  719. x = xmin - x.min() + x*(xmax - xmin)/(x.max()-x.min())
  720. y = ymin - y.min() + y*(ymax - ymin)/(y.max()-y.min())
  721. Spline.__init__(self, x, y)
  722. self.shapes['smooth'].set_linecolor('black')
  723. if name is not None:
  724. self.shapes['name'] = Text(name, self.geometric_features()[name_pos] + point(0,0.1))
  725. class SketchyFunc3(Spline):
  726. """
  727. A typical function curve used to illustrate an "arbitrary" function.
  728. """
  729. domain = [0, 6]
  730. def __init__(self, name=None, name_pos='start',
  731. xmin=0, xmax=6, ymin=0.5, ymax=3.8):
  732. x = array([0, 2, 3, 4, 5, 6])
  733. y = array([0.5, 3.5, 3.8, 2, 2.5, 3.5])
  734. # Scale x and y
  735. x = xmin - x.min() + x*(xmax - xmin)/(x.max()-x.min())
  736. y = ymin - y.min() + y*(ymax - ymin)/(y.max()-y.min())
  737. Spline.__init__(self, x, y)
  738. self.shapes['smooth'].set_linecolor('black')
  739. if name is not None:
  740. self.shapes['name'] = Text(name, self.geometric_features()[name_pos] + point(0,0.1))
  741. class SketchyFunc4(Spline):
  742. """
  743. A typical function curve used to illustrate an "arbitrary" function.
  744. Can be a companion function to SketchyFunc3.
  745. """
  746. domain = [1, 6]
  747. def __init__(self, name=None, name_pos='start',
  748. xmin=0, xmax=6, ymin=0.5, ymax=1.8):
  749. x = array([0, 2, 3, 4, 5, 6])
  750. y = array([1.5, 1.3, 0.7, 0.5, 0.6, 0.8])
  751. # Scale x and y
  752. x = xmin - x.min() + x*(xmax - xmin)/(x.max()-x.min())
  753. y = ymin - y.min() + y*(ymax - ymin)/(y.max()-y.min())
  754. Spline.__init__(self, x, y)
  755. self.shapes['smooth'].set_linecolor('black')
  756. if name is not None:
  757. self.shapes['name'] = Text(name, self.geometric_features()[name_pos] + point(0,0.1))
  758. class SketchyFunc2(Shape):
  759. """
  760. A typical function curve used to illustrate an "arbitrary" function.
  761. """
  762. domain = [0, 2.25]
  763. def __init__(self, name=None, name_pos='end',
  764. xmin=0, xmax=2.25, ymin=0.046679703125, ymax=1.259375):
  765. a = 0; b = 2.25
  766. resolution = 100
  767. x = linspace(a, b, resolution+1)
  768. f = self # for calling __call__
  769. y = f(x)
  770. # Scale x and y
  771. x = xmin - x.min() + x*(xmax - xmin)/(x.max()-x.min())
  772. y = ymin - y.min() + y*(ymax - ymin)/(y.max()-y.min())
  773. self.shapes = {'smooth': Curve(x, y)}
  774. self.shapes['smooth'].set_linecolor('black')
  775. pos = point(a, f(a)) if name_pos == 'start' else point(b, f(b))
  776. if name is not None:
  777. self.shapes['name'] = Text(name, pos + point(0,0.1))
  778. def __call__(self, x):
  779. return 0.5+x*(2-x)*(0.9-x) # on [0, 2.25]
  780. class Point(Shape):
  781. """A point (x,y) which can be rotated, translated, and scaled."""
  782. def __init__(self, x, y):
  783. self.x, self.y = x, y
  784. #self.shapes is not needed in this class
  785. def __add__(self, other):
  786. if isinstance(other, (list,tuple)):
  787. other = Point(other)
  788. return Point(self.x+other.x, self.y+other.y)
  789. # class Point is an abstract class - only subclasses are useful
  790. # and must implement draw
  791. def draw(self, verbose=0):
  792. raise NotImplementedError(
  793. 'class %s must implement the draw method' %
  794. self.__class__.__name__)
  795. def rotate(self, angle, center):
  796. """Rotate point an `angle` (in degrees) around (`x`,`y`)."""
  797. angle = angle*pi/180
  798. x, y = center
  799. c = cos(angle); s = sin(angle)
  800. xnew = x + (self.x - x)*c - (self.y - y)*s
  801. ynew = y + (self.x - x)*s + (self.y - y)*c
  802. self.x = xnew
  803. self.y = ynew
  804. return self
  805. def scale(self, factor):
  806. """Scale point coordinates by `factor`: ``x = factor*x``, etc."""
  807. self.x = factor*self.x
  808. self.y = factor*self.y
  809. return self
  810. def translate(self, vec):
  811. """Translate point by a vector `vec`."""
  812. self.x += vec[0]
  813. self.y += vec[1]
  814. return self
  815. def deform(self, displacement_function):
  816. """Displace coordinates according to displacement_function(x,y)."""
  817. for i in range(len(self.x)):
  818. self.x, self.y = displacement_function(self.x, self.y)
  819. return self
  820. def minmax_coordinates(self, minmax=None):
  821. if minmax is None:
  822. minmax = {'xmin': [], 'xmax': [], 'ymin': [], 'ymax': []}
  823. minmax['xmin'] = min(self.x, minmax['xmin'])
  824. minmax['xmax'] = max(self.x, minmax['xmax'])
  825. minmax['ymin'] = min(self.y, minmax['ymin'])
  826. minmax['ymax'] = max(self.y, minmax['ymax'])
  827. return minmax
  828. def recurse(self, name, indent=0):
  829. space = ' '*indent
  830. print(space, 'reached "bottom" object %s' % \
  831. self.__class__.__name__)
  832. def _object_couplings(self, parent, couplings=[], classname=True):
  833. return
  834. # No need for set_linecolor etc since self._for_all_shapes, which
  835. # is always called for these functions, makes a test and stops
  836. # calls if self.shapes is missing and the object is Point or Curve
  837. def show_hierarchy(self, indent=0, format='std'):
  838. s = '%s at (%g,%g)' % (self.__class__.__name__, self.x, self.y)
  839. if format == 'dict':
  840. return '"%s"' % s
  841. elif format == 'plain':
  842. return ''
  843. else:
  844. return s
  845. # no need to store input data as they are invalid after rotations etc.
  846. class Rectangle(Shape):
  847. """
  848. Rectangle specified by the point `lower_left_corner`, `width`,
  849. and `height`.
  850. """
  851. def __init__(self, lower_left_corner, width, height):
  852. is_sequence(lower_left_corner)
  853. p = arr2D(lower_left_corner) # short form
  854. x = [p[0], p[0] + width,
  855. p[0] + width, p[0], p[0]]
  856. y = [p[1], p[1], p[1] + height,
  857. p[1] + height, p[1]]
  858. self.shapes = {'rectangle': Curve(x,y)}
  859. # Dimensions
  860. dims = {
  861. 'width': Distance_wText(p + point(0, -height/5.),
  862. p + point(width, -height/5.),
  863. 'width'),
  864. 'height': Distance_wText(p + point(width + width/5., 0),
  865. p + point(width + width/5., height),
  866. 'height'),
  867. 'lower_left_corner': Text_wArrow('lower_left_corner',
  868. p - point(width/5., height/5.), p)
  869. }
  870. self.dimensions = dims
  871. def geometric_features(self):
  872. """
  873. Return dictionary with
  874. ==================== =============================================
  875. Attribute Description
  876. ==================== =============================================
  877. lower_left Lower left corner point.
  878. upper_left Upper left corner point.
  879. lower_right Lower right corner point.
  880. upper_right Upper right corner point.
  881. lower_mid Middle point on lower side.
  882. upper_mid Middle point on upper side.
  883. center Center point
  884. ==================== =============================================
  885. """
  886. r = self.shapes['rectangle']
  887. d = {'lower_left': point(r.x[0], r.y[0]),
  888. 'lower_right': point(r.x[1], r.y[1]),
  889. 'upper_right': point(r.x[2], r.y[2]),
  890. 'upper_left': point(r.x[3], r.y[3])}
  891. d['lower_mid'] = 0.5*(d['lower_left'] + d['lower_right'])
  892. d['upper_mid'] = 0.5*(d['upper_left'] + d['upper_right'])
  893. d['left_mid'] = 0.5*(d['lower_left'] + d['upper_left'])
  894. d['right_mid'] = 0.5*(d['lower_right'] + d['upper_right'])
  895. d['center'] = point(d['lower_mid'][0], d['left_mid'][1])
  896. return d
  897. class Triangle(Shape):
  898. """
  899. Triangle defined by its three vertices p1, p2, and p3.
  900. Recorded geometric features:
  901. ==================== =============================================
  902. Attribute Description
  903. ==================== =============================================
  904. p1, p2, p3 Corners as given to the constructor.
  905. ==================== =============================================
  906. """
  907. def __init__(self, p1, p2, p3):
  908. is_sequence(p1, p2, p3)
  909. x = [p1[0], p2[0], p3[0], p1[0]]
  910. y = [p1[1], p2[1], p3[1], p1[1]]
  911. self.shapes = {'triangle': Curve(x,y)}
  912. # Dimensions
  913. self.dimensions = {'p1': Text('p1', p1),
  914. 'p2': Text('p2', p2),
  915. 'p3': Text('p3', p3)}
  916. def geometric_features(self):
  917. t = self.shapes['triangle']
  918. return {'p1': point(t.x[0], t.y[0]),
  919. 'p2': point(t.x[1], t.y[1]),
  920. 'p3': point(t.x[2], t.y[2])}
  921. class Line(Shape):
  922. def __init__(self, start, end):
  923. is_sequence(start, end, length=2)
  924. if isinstance(start, (list,tuple)):
  925. start = array(start)
  926. if isinstance(end, (list,tuple)):
  927. end = array(end)
  928. if (start == end).all():
  929. # Introduce a very small perturbation since identical points
  930. # give drawing error
  931. end[0] = start[0] + 1E-10
  932. x = [start[0], end[0]]
  933. y = [start[1], end[1]]
  934. self.shapes = {'line': Curve(x, y)}
  935. def geometric_features(self):
  936. line = self.shapes['line']
  937. return {'start': point(line.x[0], line.y[0]),
  938. 'end': point(line.x[1], line.y[1]),}
  939. def compute_formulas(self):
  940. x, y = self.shapes['line'].x, self.shapes['line'].y
  941. # Define equations for line:
  942. # y = a*x + b, x = c*y + d
  943. try:
  944. self.a = (y[1] - y[0])/(x[1] - x[0])
  945. self.b = y[0] - self.a*x[0]
  946. except ZeroDivisionError:
  947. # Vertical line, y is not a function of x
  948. self.a = None
  949. self.b = None
  950. try:
  951. if self.a is None:
  952. self.c = 0
  953. else:
  954. self.c = 1/float(self.a)
  955. if self.b is None:
  956. self.d = x[1]
  957. except ZeroDivisionError:
  958. # Horizontal line, x is not a function of y
  959. self.c = None
  960. self.d = None
  961. def compute_formulas(self):
  962. x, y = self.shapes['line'].x, self.shapes['line'].y
  963. tol = 1E-14
  964. # Define equations for line:
  965. # y = a*x + b, x = c*y + d
  966. if abs(x[1] - x[0]) > tol:
  967. self.a = (y[1] - y[0])/(x[1] - x[0])
  968. self.b = y[0] - self.a*x[0]
  969. else:
  970. # Vertical line, y is not a function of x
  971. self.a = None
  972. self.b = None
  973. if self.a is None:
  974. self.c = 0
  975. elif abs(self.a) > tol:
  976. self.c = 1/float(self.a)
  977. self.d = x[1]
  978. else: # self.a is 0
  979. # Horizontal line, x is not a function of y
  980. self.c = None
  981. self.d = None
  982. def __call__(self, x=None, y=None):
  983. """Given x, return y on the line, or given y, return x."""
  984. self.compute_formulas()
  985. if x is not None and self.a is not None:
  986. return self.a*x + self.b
  987. elif y is not None and self.c is not None:
  988. return self.c*y + self.d
  989. else:
  990. raise ValueError(
  991. 'Line.__call__(x=%s, y=%s) not meaningful' % \
  992. (x, y))
  993. def new_interval(self, x=None, y=None):
  994. """Redefine current Line to cover interval in x or y."""
  995. if x is not None:
  996. is_sequence(x, length=2)
  997. xL, xR = x
  998. new_line = Line((xL, self(x=xL)), (xR, self(x=xR)))
  999. elif y is not None:
  1000. is_sequence(y, length=2)
  1001. yL, yR = y
  1002. new_line = Line((xL, self(y=xL)), (xR, self(y=xR)))
  1003. self.shapes['line'] = new_line['line']
  1004. return self
  1005. # First implementation of class Circle
  1006. class Circle(Shape):
  1007. def __init__(self, center, radius, resolution=180):
  1008. self.center, self.radius = center, radius
  1009. self.resolution = resolution
  1010. t = linspace(0, 2*pi, resolution+1)
  1011. x0 = center[0]; y0 = center[1]
  1012. R = radius
  1013. x = x0 + R*cos(t)
  1014. y = y0 + R*sin(t)
  1015. self.shapes = {'circle': Curve(x, y)}
  1016. def __call__(self, theta):
  1017. """
  1018. Return (x, y) point corresponding to angle theta.
  1019. Not valid after a translation, rotation, or scaling.
  1020. """
  1021. return self.center[0] + self.radius*cos(theta), \
  1022. self.center[1] + self.radius*sin(theta)
  1023. class Arc(Shape):
  1024. def __init__(self, center, radius,
  1025. start_angle, arc_angle,
  1026. resolution=180):
  1027. is_sequence(center)
  1028. # Must record some parameters for __call__
  1029. self.center = arr2D(center)
  1030. self.radius = radius
  1031. self.start_angle = radians(start_angle)
  1032. self.arc_angle = radians(arc_angle)
  1033. self.resolution = resolution
  1034. self.setCurve()
  1035. def setCurve(self):
  1036. t = linspace(self.start_angle,
  1037. self.start_angle + self.arc_angle,
  1038. self.resolution+1)
  1039. x0 = self.center[0]; y0 = self.center[1]
  1040. R = self.radius
  1041. x = x0 + R*cos(t)
  1042. y = y0 + R*sin(t)
  1043. self.shapes = {'arc': Curve(x, y)}
  1044. # Cannot set dimensions (Arc_wText recurses into this
  1045. # constructor forever). Set in test_Arc instead.
  1046. def geometric_features(self):
  1047. a = self.shapes['arc']
  1048. m = len(a.x)//2 # mid point in array
  1049. d = {'start': point(a.x[0], a.y[0]),
  1050. 'end': point(a.x[-1], a.y[-1]),
  1051. 'mid': point(a.x[m], a.y[m])}
  1052. return d
  1053. def __call__(self, theta):
  1054. """
  1055. Return (x,y) point at start_angle + theta.
  1056. Not valid after translation, rotation, or scaling.
  1057. """
  1058. theta = radians(theta)
  1059. t = self.start_angle + theta
  1060. x0 = self.center[0]
  1061. y0 = self.center[1]
  1062. R = self.radius
  1063. x = x0 + R*cos(t)
  1064. y = y0 + R*sin(t)
  1065. return (x, y)
  1066. # Alternative for small arcs: Parabola
  1067. class Parabola(Shape):
  1068. def __init__(self, start, mid, stop, resolution=21):
  1069. self.p1, self.p2, self.p3 = start, mid, stop
  1070. # y as function of x? (no point on line x=const?)
  1071. tol = 1E-14
  1072. if abs(self.p1[0] - self.p2[0]) > 1E-14 and \
  1073. abs(self.p2[0] - self.p3[0]) > 1E-14 and \
  1074. abs(self.p3[0] - self.p1[0]) > 1E-14:
  1075. self.y_of_x = True
  1076. else:
  1077. self.y_of_x = False
  1078. # x as function of y? (no point on line y=const?)
  1079. tol = 1E-14
  1080. if abs(self.p1[1] - self.p2[1]) > 1E-14 and \
  1081. abs(self.p2[1] - self.p3[1]) > 1E-14 and \
  1082. abs(self.p3[1] - self.p1[1]) > 1E-14:
  1083. self.x_of_y = True
  1084. else:
  1085. self.x_of_y = False
  1086. if self.y_of_x:
  1087. x = linspace(start[0], end[0], resolution)
  1088. y = self(x=x)
  1089. elif self.x_of_y:
  1090. y = linspace(start[1], end[1], resolution)
  1091. x = self(y=y)
  1092. else:
  1093. raise ValueError(
  1094. 'Parabola: two or more points lie on x=const '
  1095. 'or y=const - not allowed')
  1096. self.shapes = {'parabola': Curve(x, y)}
  1097. def __call__(self, x=None, y=None):
  1098. if x is not None and self.y_of_x:
  1099. return self._L2x(self.p1, self.p2)*self.p3[1] + \
  1100. self._L2x(self.p2, self.p3)*self.p1[1] + \
  1101. self._L2x(self.p3, self.p1)*self.p2[1]
  1102. elif y is not None and self.x_of_y:
  1103. return self._L2y(self.p1, self.p2)*self.p3[0] + \
  1104. self._L2y(self.p2, self.p3)*self.p1[0] + \
  1105. self._L2y(self.p3, self.p1)*self.p2[0]
  1106. else:
  1107. raise ValueError(
  1108. 'Parabola.__call__(x=%s, y=%s) not meaningful' % \
  1109. (x, y))
  1110. def _L2x(self, x, pi, pj, pk):
  1111. return (x - pi[0])*(x - pj[0])/((pk[0] - pi[0])*(pk[0] - pj[0]))
  1112. def _L2y(self, y, pi, pj, pk):
  1113. return (y - pi[1])*(y - pj[1])/((pk[1] - pi[1])*(pk[1] - pj[1]))
  1114. class Circle(Arc):
  1115. def __init__(self, center, radius, resolution=180):
  1116. Arc.__init__(self, center, radius, 0, 360, resolution)
  1117. class Wall(Shape):
  1118. """
  1119. defines an hached box given starting, ending point and thickness, filled with a pattern
  1120. """
  1121. def __init__(self, x, y, thickness, pattern='/', transparent=False):
  1122. is_sequence(x, y, length=len(x))
  1123. if isinstance(x[0], (tuple,list,ndarray)):
  1124. # x is list of curves
  1125. x1 = concatenate(x)
  1126. else:
  1127. x1 = asarray(x, float)
  1128. if isinstance(y[0], (tuple,list,ndarray)):
  1129. # x is list of curves
  1130. y1 = concatenate(y)
  1131. else:
  1132. y1 = asarray(y, float)
  1133. self.x1 = x1; self.y1 = y1
  1134. # Displaced curve (according to thickness)
  1135. x2 = x1
  1136. y2 = y1 + thickness
  1137. # Combine x1,y1 with x2,y2 reversed
  1138. from numpy import concatenate
  1139. x = concatenate((x1, x2[-1::-1]))
  1140. y = concatenate((y1, y2[-1::-1]))
  1141. wall = Curve(x, y)
  1142. wall.set_filled_curves(color='white', pattern=pattern)
  1143. x = [x1[-1]] + x2[-1::-1].tolist() + [x1[0]]
  1144. y = [y1[-1]] + y2[-1::-1].tolist() + [y1[0]]
  1145. self.shapes = {'wall': wall}
  1146. from collections import OrderedDict
  1147. self.shapes = OrderedDict()
  1148. self.shapes['wall'] = wall
  1149. if transparent:
  1150. white_eraser = Curve(x, y)
  1151. white_eraser.set_linecolor('white')
  1152. self.shapes['eraser'] = white_eraser
  1153. def geometric_features(self):
  1154. d = {'start': point(self.x1[0], self.y1[0]),
  1155. 'end': point(self.x1[-1], self.y1[-1])}
  1156. return d
  1157. class Wall2(Shape):
  1158. def __init__(self, x, y, thickness, pattern='/'):
  1159. is_sequence(x, y, length=len(x))
  1160. if isinstance(x[0], (tuple,list,ndarray)):
  1161. # x is list of curves
  1162. x1 = concatenate(x)
  1163. else:
  1164. x1 = asarray(x, float)
  1165. if isinstance(y[0], (tuple,list,ndarray)):
  1166. # x is list of curves
  1167. y1 = concatenate(y)
  1168. else:
  1169. y1 = asarray(y, float)
  1170. self.x1 = x1; self.y1 = y1
  1171. # Displaced curve (according to thickness)
  1172. x2 = x1.copy()
  1173. y2 = y1.copy()
  1174. def displace(idx, idx_m, idx_p):
  1175. # Find tangent and normal
  1176. tangent = point(x1[idx_m], y1[idx_m]) - point(x1[idx_p], y1[idx_p])
  1177. tangent = unit_vec(tangent)
  1178. normal = point(tangent[1], -tangent[0])
  1179. # Displace length "thickness" in "positive" normal direction
  1180. displaced_pt = point(x1[idx], y1[idx]) + thickness*normal
  1181. x2[idx], y2[idx] = displaced_pt
  1182. for i in range(1, len(x1)-1):
  1183. displace(i-1, i+1, i) # centered difference for normal comp.
  1184. # One-sided differences at the end points
  1185. i = 0
  1186. displace(i, i+1, i)
  1187. i = len(x1)-1
  1188. displace(i-1, i, i)
  1189. # Combine x1,y1 with x2,y2 reversed
  1190. from numpy import concatenate
  1191. x = concatenate((x1, x2[-1::-1]))
  1192. y = concatenate((y1, y2[-1::-1]))
  1193. wall = Curve(x, y)
  1194. wall.set_filled_curves(color='white', pattern=pattern)
  1195. x = [x1[-1]] + x2[-1::-1].tolist() + [x1[0]]
  1196. y = [y1[-1]] + y2[-1::-1].tolist() + [y1[0]]
  1197. self.shapes['wall'] = wall
  1198. def geometric_features(self):
  1199. d = {'start': point(self.x1[0], self.y1[0]),
  1200. 'end': point(self.x1[-1], self.y1[-1])}
  1201. return d
  1202. class VelocityProfile(Shape):
  1203. def __init__(self, start, height, profile, num_arrows, scaling=1):
  1204. # vx, vy = profile(y)
  1205. shapes = {}
  1206. # Draw left line
  1207. shapes['start line'] = Line(start, (start[0], start[1]+height))
  1208. # Draw velocity arrows
  1209. dy = float(height)/(num_arrows-1)
  1210. x = start[0]
  1211. y = start[1]
  1212. r = profile(y) # Test on return type
  1213. if not isinstance(r, (list,tuple,ndarray)) and len(r) != 2:
  1214. raise TypeError('VelocityProfile constructor: profile(y) function must return velocity vector (vx,vy), not %s' % type(r))
  1215. for i in range(num_arrows):
  1216. y = start[1] + i*dy
  1217. vx, vy = profile(y)
  1218. if abs(vx) < 1E-8:
  1219. continue
  1220. vx *= scaling
  1221. vy *= scaling
  1222. arr = Arrow1((x,y), (x+vx, y+vy), '->')
  1223. shapes['arrow%d' % i] = arr
  1224. # Draw smooth profile
  1225. xs = []
  1226. ys = []
  1227. n = 100
  1228. dy = float(height)/n
  1229. for i in range(n+2):
  1230. y = start[1] + i*dy
  1231. vx, vy = profile(y)
  1232. vx *= scaling
  1233. vy *= scaling
  1234. xs.append(x+vx)
  1235. ys.append(y+vy)
  1236. shapes['smooth curve'] = Curve(xs, ys)
  1237. self.shapes = shapes
  1238. class Arrow1(Shape):
  1239. """Draw a Line with arrow(s)."""
  1240. def __init__(self, start, end, style='->'):
  1241. arrow = Line(start, end)
  1242. arrow.set_arrow(style)
  1243. # Note:
  1244. self.shapes = {'arrow': arrow}
  1245. def geometric_features(self):
  1246. return self.shapes['arrow'].geometric_features()
  1247. class Arrow3(Shape):
  1248. """
  1249. Build a vertical line and arrow head from Line objects.
  1250. Then rotate `rotation_angle`.
  1251. """
  1252. def __init__(self, start, length, rotation_angle=0):
  1253. self.bottom = start
  1254. self.length = length
  1255. self.angle = rotation_angle
  1256. top = (self.bottom[0], self.bottom[1] + self.length)
  1257. main = Line(self.bottom, top)
  1258. #head_length = self.length/8.0
  1259. head_length = drawing_tool.xrange/50.
  1260. head_degrees = radians(30)
  1261. head_left_pt = (top[0] - head_length*sin(head_degrees),
  1262. top[1] - head_length*cos(head_degrees))
  1263. head_right_pt = (top[0] + head_length*sin(head_degrees),
  1264. top[1] - head_length*cos(head_degrees))
  1265. head_left = Line(head_left_pt, top)
  1266. head_right = Line(head_right_pt, top)
  1267. head_left.set_linestyle('solid')
  1268. head_right.set_linestyle('solid')
  1269. self.shapes = {'line': main, 'head left': head_left,
  1270. 'head right': head_right}
  1271. # rotate goes through self.shapes so self.shapes
  1272. # must be initialized first
  1273. self.rotate(rotation_angle, start)
  1274. def geometric_features(self):
  1275. return self.shapes['line'].geometric_features()
  1276. class Cross(Shape):
  1277. """
  1278. Place a cross at the (x,y) point `position`.
  1279. The cross fits in a 0.2 square which center is (x,y).
  1280. the color is black
  1281. the linewidth is 1
  1282. """
  1283. def __init__(self,c):
  1284. l = 0.1
  1285. line1 = Line(c+point(-l,l),c+point(l,-l))
  1286. line2 = Line(c+point(l,l), c+point(-l,-l))
  1287. cross = Composition({'line1': line1, 'line2': line2})
  1288. cross.set_linecolor('black')
  1289. cross.set_linewidth(1)
  1290. self.shapes = {'cross': cross}
  1291. class Text(Point):
  1292. """
  1293. Place `text` at the (x,y) point `position`, with the given
  1294. fontsize (0 indicates that the default fontsize set in drawing_tool
  1295. is to be used). The text is centered around `position` if `alignment` is
  1296. 'center'; if 'left', the text starts at `position`, and if
  1297. 'right', the right and of the text is located at `position`.
  1298. """
  1299. def __init__(self, text, position, alignment='center', fontsize=0,
  1300. bgcolor=None, fgcolor=None, fontfamily=None):
  1301. """
  1302. fontfamily can be (e.g.) 'serif' or 'monospace' (for code!).
  1303. """
  1304. is_sequence(position)
  1305. is_sequence(position, length=2, can_be_None=True)
  1306. self.text = text
  1307. self.position = position
  1308. self.alignment = alignment
  1309. self.fontsize = fontsize
  1310. self.bgcolor = bgcolor
  1311. self.fgcolor = fgcolor
  1312. self.fontfamily = fontfamily
  1313. Point.__init__(self, position[0], position[1])
  1314. #no need for self.shapes here
  1315. def draw(self, verbose=0):
  1316. drawing_tool.text(
  1317. self.text, (self.x, self.y),
  1318. self.alignment, self.fontsize,
  1319. arrow_tip=None, bgcolor=self.bgcolor, fgcolor=self.fgcolor,
  1320. fontfamily=self.fontfamily)
  1321. if verbose > 0:
  1322. print('drawing Text "%s"' % self.text)
  1323. def __str__(self):
  1324. return 'text "%s" at (%g,%g)' % (self.text, self.x, self.y)
  1325. def __repr__(self):
  1326. return repr(str(self))
  1327. class Text_wArrow(Text):
  1328. """
  1329. As class Text, but an arrow is drawn from the mid part of the text
  1330. to some point `arrow_tip`.
  1331. """
  1332. def __init__(self, text, position, arrow_tip,
  1333. alignment='center', fontsize=0):
  1334. is_sequence(arrow_tip, length=2, can_be_None=True)
  1335. is_sequence(position)
  1336. self.arrow_tip = arrow_tip
  1337. Text.__init__(self, text, position, alignment, fontsize)
  1338. def draw(self, verbose=0):
  1339. drawing_tool.text(
  1340. self.text, self.position,
  1341. self.alignment, self.fontsize,
  1342. arrow_tip=self.arrow_tip,
  1343. bgcolor=self.bgcolor, fgcolor=self.fgcolor,
  1344. fontfamily=self.fontfamily)
  1345. if verbose > 0:
  1346. print('drawing Text_wArrow "%s"' % self.text)
  1347. def __str__(self):
  1348. return 'annotation "%s" at (%g,%g) with arrow to (%g,%g)' % \
  1349. (self.text, self.x, self.y,
  1350. self.arrow_tip[0], self.arrow_tip[1])
  1351. def __repr__(self):
  1352. return repr(str(self))
  1353. class Axis(Shape):
  1354. def __init__(self, start, length, label,
  1355. rotation_angle=0, fontsize=0,
  1356. label_spacing=1./45, label_alignment='left'):
  1357. """
  1358. Draw axis from start with `length` to the right
  1359. (x axis). Place label at the end of the arrow tip.
  1360. Then return `rotation_angle` (in degrees).
  1361. The `label_spacing` denotes the space between the label
  1362. and the arrow tip as a fraction of the length of the plot
  1363. in x direction. A tuple can be given to adjust the position
  1364. in both the x and y directions (with one parameter, the
  1365. x position is adjusted).
  1366. With `label_alignment` one can place
  1367. the axis label text such that the arrow tip is to the 'left',
  1368. 'right', or 'center' with respect to the text field.
  1369. The `label_spacing` and `label_alignment`parameters can
  1370. be used to fine-tune the location of the label.
  1371. """
  1372. # Arrow is vertical arrow, make it horizontal
  1373. arrow = Arrow3(start, length, rotation_angle=-90)
  1374. arrow.rotate(rotation_angle, start)
  1375. if isinstance(label_spacing, (list,tuple)) and len(label_spacing) == 2:
  1376. x_spacing = drawing_tool.xrange*label_spacing[0]
  1377. y_spacing = drawing_tool.yrange*label_spacing[1]
  1378. elif isinstance(label_spacing, (int,float)):
  1379. # just x spacing
  1380. x_spacing = drawing_tool.xrange*label_spacing
  1381. y_spacing = 0
  1382. # should increase spacing for downward pointing axis
  1383. label_pos = [start[0] + length + x_spacing, start[1] + y_spacing]
  1384. label = Text(label, position=label_pos, fontsize=fontsize)
  1385. label.rotate(rotation_angle, start)
  1386. self.shapes = {'arrow': arrow, 'label': label}
  1387. def geometric_features(self):
  1388. return self.shapes['arrow'].geometric_features()
  1389. # Maybe Axis3 with label below/above?
  1390. class Force(Arrow1):
  1391. """
  1392. Indication of a force by an arrow and a text (symbol). Draw an
  1393. arrow, starting at `start` and with the tip at `end`. The symbol
  1394. is placed at `text_pos`, which can be 'start', 'end' or the
  1395. coordinates of a point. If 'end' or 'start', the text is placed at
  1396. a distance `text_spacing` times the width of the total plotting
  1397. area away from the specified point.
  1398. """
  1399. def __init__(self, start, end, text, text_spacing=1./60,
  1400. fontsize=0, text_pos='start', text_alignment='center'):
  1401. Arrow1.__init__(self, start, end, style='->')
  1402. if isinstance(text_spacing, (tuple,list)):
  1403. if len(text_spacing) == 2:
  1404. spacing = point(drawing_tool.xrange*text_spacing[0],
  1405. drawing_tool.xrange*text_spacing[1])
  1406. else:
  1407. spacing = drawing_tool.xrange*text_spacing[0]
  1408. else:
  1409. # just a number, this is x spacing
  1410. spacing = drawing_tool.xrange*text_spacing
  1411. start, end = arr2D(start), arr2D(end)
  1412. # Two cases: label at bottom of line or top, need more
  1413. # spacing if bottom
  1414. downward = (end-start)[1] < 0
  1415. upward = not downward # for easy code reading
  1416. if isinstance(text_pos, (str,bytes)):
  1417. if text_pos == 'start':
  1418. spacing_dir = unit_vec(start - end)
  1419. if upward:
  1420. spacing *= 1.7
  1421. if isinstance(spacing, (int, float)):
  1422. text_pos = start + spacing*spacing_dir
  1423. else:
  1424. text_pos = start + spacing
  1425. elif text_pos == 'end':
  1426. spacing_dir = unit_vec(end - start)
  1427. if downward:
  1428. spacing *= 1.7
  1429. if isinstance(spacing, (int, float)):
  1430. text_pos = end + spacing*spacing_dir
  1431. else:
  1432. text_pos = end + spacing
  1433. self.shapes['text'] = Text(text, text_pos, fontsize=fontsize,
  1434. alignment=text_alignment)
  1435. def geometric_features(self):
  1436. d = Arrow1.geometric_features(self)
  1437. d['symbol_location'] = self.shapes['text'].position
  1438. return d
  1439. class Axis2(Force):
  1440. def __init__(self, start, length, label,
  1441. rotation_angle=0, fontsize=0,
  1442. label_spacing=1./45, label_alignment='left'):
  1443. direction = point(cos(radians(rotation_angle)),
  1444. sin(radians(rotation_angle)))
  1445. Force.__init__(start=start, end=length*direction, text=label,
  1446. text_spacing=label_spacing,
  1447. fontsize=fontsize, text_pos='end',
  1448. text_alignment=label_alignment)
  1449. # Substitute text by label for axis
  1450. self.shapes['label'] = self.shapes['text']
  1451. del self.shapes['text']
  1452. # geometric features from Force is ok
  1453. class Gravity(Axis):
  1454. """Downward-pointing gravity arrow with the symbol g."""
  1455. def __init__(self, start, length, fontsize=0):
  1456. Axis.__init__(self, start, length, '$g$', below=False,
  1457. rotation_angle=-90, label_spacing=1./30,
  1458. fontsize=fontsize)
  1459. self.shapes['arrow'].set_linecolor('black')
  1460. class Gravity(Force):
  1461. """Downward-pointing gravity arrow with the symbol g."""
  1462. def __init__(self, start, length, text='$g$', fontsize=0):
  1463. Force.__init__(self, start, (start[0], start[1]-length),
  1464. text, text_spacing=1./60,
  1465. fontsize=0, text_pos='end')
  1466. self.shapes['arrow'].set_linecolor('black')
  1467. class Distance_wText(Shape):
  1468. """
  1469. Arrow <-> with text (usually a symbol) at the midpoint, used for
  1470. identifying a some distance in a figure. The text is placed
  1471. slightly to the right of vertical-like arrows, with text displaced
  1472. `text_spacing` times to total distance in x direction of the plot
  1473. area. The text is by default aligned 'left' in this case. For
  1474. horizontal-like arrows, the text is placed the same distance
  1475. above, but aligned 'center' by default (when `alignment` is None).
  1476. """
  1477. def __init__(self, start, end, text, fontsize=0, text_spacing=1/60.,
  1478. alignment=None, text_pos='mid'):
  1479. start = arr2D(start)
  1480. end = arr2D(end)
  1481. # Decide first if we have a vertical or horizontal arrow
  1482. vertical = abs(end[0]-start[0]) < 2*abs(end[1]-start[1])
  1483. if vertical:
  1484. # Assume end above start
  1485. if end[1] < start[1]:
  1486. start, end = end, start
  1487. if alignment is None:
  1488. alignment = 'left'
  1489. else: # horizontal arrow
  1490. # Assume start to the right of end
  1491. if start[0] < end[0]:
  1492. start, end = end, start
  1493. if alignment is None:
  1494. alignment = 'center'
  1495. tangent = end - start
  1496. # Tangeng goes always to the left and upward
  1497. normal = unit_vec([tangent[1], -tangent[0]])
  1498. mid = 0.5*(start + end) # midpoint of start-end line
  1499. if text_pos == 'mid':
  1500. text_pos = mid + normal*drawing_tool.xrange*text_spacing
  1501. text = Text(text, text_pos, fontsize=fontsize,
  1502. alignment=alignment)
  1503. else:
  1504. is_sequence(text_pos, length=2)
  1505. text = Text_wArrow(text, text_pos, mid, alignment='left',
  1506. fontsize=fontsize)
  1507. arrow = Arrow1(start, end, style='<->')
  1508. arrow.set_linecolor('black')
  1509. arrow.set_linewidth(1)
  1510. self.shapes = {'arrow': arrow, 'text': text}
  1511. def geometric_features(self):
  1512. d = self.shapes['arrow'].geometric_features()
  1513. d['text_position'] = self.shapes['text'].position
  1514. return d
  1515. class Arc_wText(Shape):
  1516. """
  1517. Arc with text positionned at the left of arc half-way
  1518. """
  1519. def __init__(self, text, center, radius,
  1520. start_angle, arc_angle, fontsize=0,
  1521. resolution=180, text_spacing=1/60.):
  1522. self.text = text
  1523. self.center = center
  1524. self.radius = radius
  1525. self.fontsize=fontsize
  1526. self.resolution=resolution
  1527. self.text_spacing=text_spacing
  1528. self.start_angle = start_angle
  1529. self.arc_angle = arc_angle
  1530. self.setArc()
  1531. def setArc(self):
  1532. arc = Arc(self.center, self.radius, self.start_angle, self.arc_angle,
  1533. self.resolution)
  1534. mid = arr2D(arc(self.arc_angle/2.))
  1535. normal = unit_vec(mid - arr2D(self.center))
  1536. text_pos = mid + normal*drawing_tool.xrange*self.text_spacing
  1537. if hasattr(self, 'linewidth'):
  1538. arc.set_linewidth(self.linewidth)
  1539. self.shapes = {'arc': arc,
  1540. 'text': Text(self.text, text_pos, fontsize=self.fontsize)}
  1541. def changeAngle(self,start_angle,arc_angle):
  1542. self.arc_angle = arc_angle
  1543. self.start_angle = start_angle
  1544. self.setArc()
  1545. def set_linewidth(self, width):
  1546. self.linewidth = width
  1547. self.change_linewidth()
  1548. def change_linewidth(self):
  1549. super().set_linewidth(self.linewidth)
  1550. class Composition(Shape):
  1551. def __init__(self, shapes):
  1552. """shapes: list or dict of Shape objects."""
  1553. if isinstance(shapes, (tuple,list)):
  1554. # Convert to dict using the type of the list element as key
  1555. # (add a counter to make the keys unique)
  1556. shapes = {s.__class__.__name__ + '_' + str(i): s
  1557. for i, s in enumerate(shapes)}
  1558. self.shapes = shapes
  1559. # can make help methods: Line.midpoint, Line.normal(pt, dir='left') -> (x,y)
  1560. # list annotations in each class? contains extra annotations for explaining
  1561. # important parameters to the constructor, e.g., Line.annotations holds
  1562. # start and end as Text objects. Shape.demo calls shape.draw and
  1563. # for annotation in self.demo: annotation.draw() YES!
  1564. # Can make overall demo of classes by making objects and calling demo
  1565. # Could include demo fig in each constructor
  1566. class SimplySupportedBeam(Shape):
  1567. def __init__(self, pos, size):
  1568. pos = arr2D(pos)
  1569. P0 = (pos[0] - size/2., pos[1]-size)
  1570. P1 = (pos[0] + size/2., pos[1]-size)
  1571. triangle = Triangle(P0, P1, pos)
  1572. gap = size/5.
  1573. h = size/4. # height of rectangle
  1574. P2 = (P0[0], P0[1]-gap-h)
  1575. rectangle = Rectangle(P2, size, h).set_filled_curves(pattern='/')
  1576. self.shapes = {'triangle': triangle, 'rectangle': rectangle}
  1577. self.dimensions = {'pos': Text('pos', pos),
  1578. 'size': Distance_wText((P2[0], P2[1]-size),
  1579. (P2[0]+size, P2[1]-size),
  1580. 'size')}
  1581. def geometric_features(self):
  1582. t = self.shapes['triangle']
  1583. r = self.shapes['rectangle']
  1584. d = {'pos': t.geometric_features()['p2'],
  1585. 'mid_support': r.geometric_features()['lower_mid']}
  1586. return d
  1587. class ConstantBeamLoad(Shape):
  1588. """
  1589. Downward-pointing arrows indicating a vertical load.
  1590. The arrows are of equal length and filling a rectangle
  1591. specified as in the :class:`Rectangle` class.
  1592. Recorded geometric features:
  1593. ==================== =============================================
  1594. Attribute Description
  1595. ==================== =============================================
  1596. mid_top Middle point at the top of the row of
  1597. arrows (often used for positioning a text).
  1598. ==================== =============================================
  1599. """
  1600. def __init__(self, lower_left_corner, width, height, num_arrows=10):
  1601. box = Rectangle(lower_left_corner, width, height)
  1602. self.shapes = {'box': box}
  1603. dx = float(width)/(num_arrows-1)
  1604. y_top = lower_left_corner[1] + height
  1605. y_tip = lower_left_corner[1]
  1606. for i in range(num_arrows):
  1607. x = lower_left_corner[0] + i*dx
  1608. self.shapes['arrow%d' % i] = Arrow1((x, y_top), (x, y_tip))
  1609. def geometric_features(self):
  1610. return {'mid_top': self.shapes['box'].geometric_features()['upper_mid']}
  1611. class Moment(Arc_wText):
  1612. """
  1613. defines a Moment arrow with text given text, center and radius
  1614. default direction is counter_clockwise, fontsize and text spacing as optional parameters
  1615. """
  1616. def __init__(self, text, center, radius,
  1617. left=True, counter_clockwise=True,
  1618. fontsize=0, text_spacing=1/60.):
  1619. style = '->' if counter_clockwise else '<-'
  1620. start_angle = 90 if left else -90
  1621. Arc_wText.__init__(self, text, center, radius,
  1622. start_angle=start_angle,
  1623. arc_angle=180, fontsize=fontsize,
  1624. text_spacing=text_spacing,
  1625. resolution=180)
  1626. self.shapes['arc']['arc'].set_arrow(style) # Curve object
  1627. class Wheel(Shape):
  1628. """
  1629. Hub and spokes Wheel given center, radius, spokes (default 10), inner_radius(default 1/5 of radius)
  1630. """
  1631. def __init__(self, center, radius, inner_radius=None, nlines=10):
  1632. if inner_radius is None:
  1633. inner_radius = radius/5.0
  1634. outer = Circle(center, radius)
  1635. inner = Circle(center, inner_radius)
  1636. lines = []
  1637. self.nlines = nlines
  1638. # Draw nlines+1 since the first and last coincide
  1639. # (then nlines lines will be visible)
  1640. t = linspace(0, 2*pi, self.nlines+1)
  1641. Ri = inner_radius; Ro = radius
  1642. x0 = center[0]; y0 = center[1]
  1643. xinner = x0 + Ri*cos(t)
  1644. yinner = y0 + Ri*sin(t)
  1645. xouter = x0 + Ro*cos(t)
  1646. youter = y0 + Ro*sin(t)
  1647. lines = [Line((xi,yi),(xo,yo)) for xi, yi, xo, yo in \
  1648. zip(xinner, yinner, xouter, youter)]
  1649. self.shapes = {'inner': inner, 'outer': outer,
  1650. 'spokes': Composition(
  1651. {'spoke%d' % i: lines[i]
  1652. for i in range(len(lines))})}
  1653. class SineWave(Shape):
  1654. def __init__(self, xstart, xstop,
  1655. wavelength, amplitude, mean_level):
  1656. self.xstart = xstart
  1657. self.xstop = xstop
  1658. self.wavelength = wavelength
  1659. self.amplitude = amplitude
  1660. self.mean_level = mean_level
  1661. npoints = (self.xstop - self.xstart)/(self.wavelength/61.0)
  1662. x = linspace(self.xstart, self.xstop, npoints)
  1663. k = 2*pi/self.wavelength # frequency
  1664. y = self.mean_level + self.amplitude*sin(k*x)
  1665. self.shapes = {'waves': Curve(x,y)}
  1666. class Spring(Shape):
  1667. """
  1668. Specify a *vertical* spring, starting at `start` and with `length`
  1669. as total vertical length. In the middle of the spring there are
  1670. `num_windings` circular windings to illustrate the spring. If
  1671. `teeth` is true, the spring windings look like saw teeth,
  1672. otherwise the windings are smooth circles. The parameters `width`
  1673. (total width of spring) and `bar_length` (length of first and last
  1674. bar are given sensible default values if they are not specified
  1675. (these parameters can later be extracted as attributes, see table
  1676. below).
  1677. """
  1678. spring_fraction = 1./2 # fraction of total length occupied by spring
  1679. def __init__(self, start, length, width=None, bar_length=None,
  1680. num_windings=11, teeth=False):
  1681. B = start
  1682. n = num_windings - 1 # n counts teeth intervals
  1683. if n <= 6:
  1684. n = 7
  1685. # n must be odd:
  1686. if n % 2 == 0:
  1687. n = n+1
  1688. L = length
  1689. if width is None:
  1690. w = L/10.
  1691. else:
  1692. w = width/2.0
  1693. s = bar_length
  1694. # [0, x, L-x, L], f = (L-2*x)/L
  1695. # x = L*(1-f)/2.
  1696. # B: start point
  1697. # w: half-width
  1698. # L: total length
  1699. # s: length of first bar
  1700. # P0: start of dashpot (B[0]+s)
  1701. # P1: end of dashpot
  1702. # P2: end point
  1703. shapes = {}
  1704. if s is None:
  1705. f = Spring.spring_fraction
  1706. s = L*(1-f)/2. # start of spring
  1707. self.bar_length = s # record
  1708. self.width = 2*w
  1709. P0 = (B[0], B[1] + s)
  1710. P1 = (B[0], B[1] + L-s)
  1711. P2 = (B[0], B[1] + L)
  1712. if s >= L:
  1713. raise ValueError('length of first bar: %g is larger than total length: %g' % (s, L))
  1714. shapes['bar1'] = Line(B, P0)
  1715. spring_length = L - 2*s
  1716. t = spring_length/n # height increment per winding
  1717. if teeth:
  1718. resolution = 4
  1719. else:
  1720. resolution = 90
  1721. q = linspace(0, n, n*resolution + 1)
  1722. x = P0[0] + w*sin(2*pi*q)
  1723. y = P0[1] + q*t
  1724. shapes['spiral'] = Curve(x, y)
  1725. shapes['bar2'] = Line(P1,P2)
  1726. self.shapes = shapes
  1727. # Dimensions
  1728. start = Text_wArrow('start', (B[0]-1.5*w,B[1]-1.5*w), B)
  1729. width = Distance_wText((B[0]-w, B[1]-3.5*w), (B[0]+w, B[1]-3.5*w),
  1730. 'width')
  1731. length = Distance_wText((B[0]+3*w, B[1]), (B[0]+3*w, B[1]+L),
  1732. 'length')
  1733. num_windings = Text_wArrow('num_windings',
  1734. (B[0]+2*w,P2[1]+w),
  1735. (B[0]+1.2*w, B[1]+L/2.))
  1736. blength1 = Distance_wText((B[0]-2*w, B[1]), (B[0]-2*w, P0[1]),
  1737. 'bar_length',
  1738. text_pos=(P0[0]-7*w, P0[1]+w))
  1739. blength2 = Distance_wText((P1[0]-2*w, P1[1]), (P2[0]-2*w, P2[1]),
  1740. 'bar_length',
  1741. text_pos=(P2[0]-7*w, P2[1]+w))
  1742. dims = {'start': start, 'width': width, 'length': length,
  1743. 'num_windings': num_windings, 'bar_length1': blength1,
  1744. 'bar_length2': blength2}
  1745. self.dimensions = dims
  1746. def geometric_features(self):
  1747. """
  1748. Recorded geometric features:
  1749. ==================== =============================================
  1750. Attribute Description
  1751. ==================== =============================================
  1752. start Start point of spring.
  1753. end End point of spring.
  1754. width Total width of spring.
  1755. bar_length Length of first (and last) bar part.
  1756. ==================== =============================================
  1757. """
  1758. b1 = self.shapes['bar1']
  1759. d = {'start': b1.geometric_features()['start'],
  1760. 'end': self.shapes['bar2'].geometric_features()['end'],
  1761. 'bar_length': self.bar_length,
  1762. 'width': self.width}
  1763. return d
  1764. class Dashpot(Shape):
  1765. """
  1766. Specify a vertical dashpot of height `total_length` and `start` as
  1767. bottom/starting point. The first bar part has length `bar_length`.
  1768. Then comes the dashpot as a rectangular construction of total
  1769. width `width` and height `dashpot_length`. The position of the
  1770. piston inside the rectangular dashpot area is given by
  1771. `piston_pos`, which is the distance between the first bar (given
  1772. by `bar_length`) to the piston.
  1773. If some of `dashpot_length`, `bar_length`, `width` or `piston_pos`
  1774. are not given, suitable default values are calculated. Their
  1775. values can be extracted as keys in the dict returned from
  1776. ``geometric_features``.
  1777. """
  1778. dashpot_fraction = 1./2 # fraction of total_length
  1779. piston_gap_fraction = 1./6 # fraction of width
  1780. piston_thickness_fraction = 1./8 # fraction of dashplot_length
  1781. def __init__(self, start, total_length, bar_length=None,
  1782. width=None, dashpot_length=None, piston_pos=None):
  1783. B = start
  1784. L = total_length
  1785. if width is None:
  1786. w = L/10. # total width 1/5 of length
  1787. else:
  1788. w = width/2.0
  1789. s = bar_length
  1790. # [0, x, L-x, L], f = (L-2*x)/L
  1791. # x = L*(1-f)/2.
  1792. # B: start point
  1793. # w: half-width
  1794. # L: total length
  1795. # s: length of first bar
  1796. # P0: start of dashpot (B[0]+s)
  1797. # P1: end of dashpot
  1798. # P2: end point
  1799. shapes = {}
  1800. # dashpot is P0-P1 in y and width 2*w
  1801. if dashpot_length is None:
  1802. if s is None:
  1803. f = Dashpot.dashpot_fraction
  1804. s = L*(1-f)/2. # default
  1805. P1 = (B[0], B[1]+L-s)
  1806. dashpot_length = f*L
  1807. else:
  1808. if s is None:
  1809. f = 1./2 # the bar lengths are taken as f*dashpot_length
  1810. s = f*dashpot_length # default
  1811. P1 = (B[0], B[1]+s+dashpot_length)
  1812. P0 = (B[0], B[1]+s)
  1813. P2 = (B[0], B[1]+L)
  1814. if P2[1] > P1[1] > P0[1]:
  1815. pass # ok
  1816. else:
  1817. raise ValueError('Dashpot has inconsistent dimensions! start: %g, dashpot begin: %g, dashpot end: %g, very end: %g' % (B[1], P0[1], P1[1], P2[1]))
  1818. shapes['line start'] = Line(B, P0)
  1819. shapes['pot'] = Curve([P1[0]-w, P0[0]-w, P0[0]+w, P1[0]+w],
  1820. [P1[1], P0[1], P0[1], P1[1]])
  1821. piston_thickness = dashpot_length*Dashpot.piston_thickness_fraction
  1822. if piston_pos is None:
  1823. piston_pos = 1/3.*dashpot_length
  1824. if piston_pos < 0:
  1825. piston_pos = 0
  1826. elif piston_pos > dashpot_length:
  1827. piston_pos = dashpot_length - piston_thickness
  1828. abs_piston_pos = P0[1] + piston_pos
  1829. gap = w*Dashpot.piston_gap_fraction
  1830. shapes['piston'] = Composition(
  1831. {'line': Line(P2, (B[0], abs_piston_pos + piston_thickness)),
  1832. 'rectangle': Rectangle((B[0] - w+gap, abs_piston_pos),
  1833. 2*w-2*gap, piston_thickness),
  1834. })
  1835. shapes['piston']['rectangle'].set_filled_curves(pattern='X')
  1836. self.shapes = shapes
  1837. self.bar_length = s
  1838. self.width = 2*w
  1839. self.piston_pos = piston_pos
  1840. self.dashpot_length = dashpot_length
  1841. # Dimensions
  1842. start = Text_wArrow('start', (B[0]-1.5*w,B[1]-1.5*w), B)
  1843. width = Distance_wText((B[0]-w, B[1]-3.5*w), (B[0]+w, B[1]-3.5*w),
  1844. 'width')
  1845. dplength = Distance_wText((B[0]+2*w, P0[1]), (B[0]+2*w, P1[1]),
  1846. 'dashpot_length', text_pos=(B[0]+w,B[1]-w))
  1847. blength = Distance_wText((B[0]-2*w, B[1]), (B[0]-2*w, P0[1]),
  1848. 'bar_length', text_pos=(B[0]-6*w,P0[1]-w))
  1849. ppos = Distance_wText((B[0]-2*w, P0[1]), (B[0]-2*w, P0[1]+piston_pos),
  1850. 'piston_pos', text_pos=(B[0]-6*w,P0[1]+piston_pos-w))
  1851. tlength = Distance_wText((B[0]+4*w, B[1]), (B[0]+4*w, B[1]+L),
  1852. 'total_length',
  1853. text_pos=(B[0]+4.5*w, B[1]+L-2*w))
  1854. line = Line((B[0]+w, abs_piston_pos), (B[0]+7*w, abs_piston_pos)).set_linestyle('dashed').set_linecolor('black').set_linewidth(1)
  1855. pp = Text('abs_piston_pos', (B[0]+7*w, abs_piston_pos), alignment='left')
  1856. dims = {'start': start, 'width': width, 'dashpot_length': dplength,
  1857. 'bar_length': blength, 'total_length': tlength,
  1858. 'piston_pos': ppos,}
  1859. #'abs_piston_pos': Composition({'line': line, 'text': pp})}
  1860. self.dimensions = dims
  1861. def geometric_features(self):
  1862. """
  1863. Recorded geometric features:
  1864. ==================== =============================================
  1865. Attribute Description
  1866. ==================== =============================================
  1867. start Start point of dashpot.
  1868. end End point of dashpot.
  1869. bar_length Length of first bar (from start to spring).
  1870. dashpot_length Length of dashpot middle part.
  1871. width Total width of dashpot.
  1872. piston_pos Position of piston in dashpot, relative to
  1873. start[1] + bar_length.
  1874. ==================== =============================================
  1875. """
  1876. d = {'start': self.shapes['line start'].geometric_features()['start'],
  1877. 'end': self.shapes['piston']['line'].geometric_features()['start'],
  1878. 'bar_length': self.bar_length,
  1879. 'piston_pos': self.piston_pos,
  1880. 'width': self.width,
  1881. 'dashpot_length': self.dashpot_length,
  1882. }
  1883. return d
  1884. class Wavy(Shape):
  1885. """
  1886. A wavy graph consisting of a user-given main curve y=f(x) with
  1887. additional sinusoidal waves of given (constant) amplitude,
  1888. but varying wavelength (a characteristic wavelength is specified).
  1889. """
  1890. def __init__(self, main_curve, interval, wavelength_of_perturbations,
  1891. amplitude_of_perturbations, smoothness):
  1892. """
  1893. ============================ ====================================
  1894. Name Description
  1895. ============================ ====================================
  1896. main_curve f(x) Python function
  1897. interval interval for main_curve
  1898. wavelength_of_perturbations dominant wavelength perturbed waves
  1899. amplitude_of_perturbations amplitude of perturbed waves
  1900. smoothness in [0, 1]: smooth=0, rough=1
  1901. ============================ ====================================
  1902. """
  1903. xmin, xmax = interval
  1904. L = wavelength_of_perturbations
  1905. k_0 = 2*pi/L # main frequency of waves
  1906. k_p = k_0*0.5
  1907. k_k = k_0/2*smoothness
  1908. A_0 = amplitude_of_perturbations
  1909. A_p = 0.3*A_0
  1910. A_k = k_0/2
  1911. x = linspace(xmin, xmax, 2001)
  1912. def w(x):
  1913. A = A_0 + A_p*sin(A_k*x)
  1914. k = k_0 + k_p*sin(k_k*x)
  1915. y = main_curve(x) + A*sin(k*x)
  1916. return y
  1917. self.shapes = {'wavy': Curve(x, w(x))}
  1918. # Use closure w to define __call__ - then we do not need
  1919. # to store all the parameters A_0, A_k, etc. as attributes
  1920. self.__call__ = w
  1921. class StochasticWavyCurve(object):
  1922. """
  1923. Precomputed stochastic wavy graphs.
  1924. There are three graphs with different look.
  1925. Curve 0:
  1926. ----------------------------------------------------------------------
  1927. |
  1928. |
  1929. *|
  1930. * |
  1931. * |
  1932. * |
  1933. * |
  1934. * |
  1935. * |
  1936. * |
  1937. * |
  1938. * |
  1939. * |
  1940. * |
  1941. |*
  1942. | *
  1943. | *
  1944. | *
  1945. | *
  1946. | *
  1947. | *
  1948. | *
  1949. | *
  1950. | *
  1951. | *
  1952. | *
  1953. | *
  1954. | *
  1955. | *
  1956. | *
  1957. | *
  1958. | *
  1959. | *
  1960. | *
  1961. | *
  1962. | *
  1963. | *
  1964. | *
  1965. | *
  1966. | *
  1967. | *
  1968. | *
  1969. | *
  1970. | *
  1971. | *
  1972. | *
  1973. | *
  1974. | *
  1975. | *
  1976. | *
  1977. | *
  1978. | *
  1979. | *
  1980. | *
  1981. | *
  1982. | *
  1983. | *
  1984. | *
  1985. | *
  1986. | *
  1987. | *
  1988. | *
  1989. | *
  1990. | *
  1991. | *
  1992. | *
  1993. | *
  1994. | *
  1995. | *
  1996. | *
  1997. | *
  1998. | *
  1999. |*
  2000. *|
  2001. * |
  2002. * |
  2003. * |
  2004. * |
  2005. * |
  2006. * |
  2007. * |
  2008. * |
  2009. * |
  2010. * |
  2011. * |
  2012. * |
  2013. * |
  2014. * |
  2015. * |
  2016. |*
  2017. | *
  2018. | *
  2019. | *
  2020. | *
  2021. | *
  2022. | *
  2023. | *
  2024. | *
  2025. | *
  2026. | *
  2027. | *
  2028. | *
  2029. | *
  2030. | *
  2031. | *
  2032. | *
  2033. | *
  2034. | *
  2035. | *
  2036. | *
  2037. | *
  2038. | *
  2039. | *
  2040. | *
  2041. | *
  2042. | *
  2043. | *
  2044. | *
  2045. | *
  2046. | *
  2047. | *
  2048. | *
  2049. | *
  2050. | *
  2051. | *
  2052. | *
  2053. | *
  2054. | *
  2055. | *
  2056. | *
  2057. | *
  2058. | *
  2059. | *
  2060. | *
  2061. |
  2062. * |
  2063. * |
  2064. * |
  2065. * |
  2066. * |
  2067. * |
  2068. * |
  2069. * |
  2070. * |
  2071. * |
  2072. * |
  2073. * |
  2074. * |
  2075. * |
  2076. * |
  2077. * |
  2078. * |
  2079. * |
  2080. * |
  2081. * |
  2082. * |
  2083. * |
  2084. * |
  2085. * |
  2086. * |
  2087. * |
  2088. * |
  2089. * |
  2090. * |
  2091. * |
  2092. Curve 2:
  2093. ----------------------------------------------------------------------
  2094. |
  2095. |
  2096. |
  2097. |*
  2098. |*
  2099. |*
  2100. |
  2101. |
  2102. *|
  2103. |*
  2104. | *
  2105. | *
  2106. | *
  2107. | *
  2108. | *
  2109. | *
  2110. | *
  2111. | *
  2112. | *
  2113. | *
  2114. | *
  2115. | *
  2116. | *
  2117. | *
  2118. | *
  2119. | *
  2120. | *
  2121. | *
  2122. | *
  2123. | *
  2124. | *
  2125. | *
  2126. | *
  2127. | *
  2128. | *
  2129. | *
  2130. | *
  2131. | *
  2132. | *
  2133. | *
  2134. | *
  2135. | *
  2136. | *
  2137. | *
  2138. | *
  2139. | *
  2140. | *
  2141. | *
  2142. | *
  2143. | *
  2144. | *
  2145. | *
  2146. | *
  2147. | *
  2148. | *
  2149. | *
  2150. | *
  2151. | *
  2152. | *
  2153. | *
  2154. | *
  2155. |
  2156. * |
  2157. * |
  2158. * |
  2159. * |
  2160. * |
  2161. * |
  2162. * |
  2163. * |
  2164. * |
  2165. * |
  2166. * |
  2167. * |
  2168. * |
  2169. * |
  2170. * |
  2171. * |
  2172. * |
  2173. * |
  2174. * |
  2175. * |
  2176. * |
  2177. * |
  2178. * |
  2179. * |
  2180. * |
  2181. * |
  2182. * |
  2183. * |
  2184. * |
  2185. |
  2186. | *
  2187. | *
  2188. | *
  2189. | *
  2190. | *
  2191. | *
  2192. | *
  2193. | *
  2194. | *
  2195. | *
  2196. | *
  2197. | *
  2198. | *
  2199. | *
  2200. | *
  2201. | *
  2202. | *
  2203. | *
  2204. | *
  2205. | *
  2206. | *
  2207. | *
  2208. |*
  2209. |*
  2210. |
  2211. |
  2212. |
  2213. |*
  2214. | *
  2215. | *
  2216. |*
  2217. |
  2218. *|
  2219. |*
  2220. | *
  2221. | *
  2222. | *
  2223. | *
  2224. | *
  2225. | *
  2226. | *
  2227. | *
  2228. | *
  2229. | *
  2230. | *
  2231. | *
  2232. | *
  2233. | *
  2234. | *
  2235. | *
  2236. | *
  2237. | *
  2238. | *
  2239. | *
  2240. | *
  2241. | *
  2242. | *
  2243. | *
  2244. | *
  2245. | *
  2246. | *
  2247. | *
  2248. | *
  2249. | *
  2250. | *
  2251. | *
  2252. | *
  2253. | *
  2254. | *
  2255. | *
  2256. | *
  2257. | *
  2258. | *
  2259. Curve 2:
  2260. ----------------------------------------------------------------------
  2261. |
  2262. |
  2263. |
  2264. |
  2265. |*
  2266. | *
  2267. | *
  2268. | *
  2269. | *
  2270. | *
  2271. | *
  2272. | *
  2273. | *
  2274. | *
  2275. | *
  2276. | *
  2277. | *
  2278. | *
  2279. | *
  2280. | *
  2281. | *
  2282. | *
  2283. | *
  2284. | *
  2285. | *
  2286. | *
  2287. | *
  2288. | *
  2289. |*
  2290. |
  2291. * |
  2292. * |
  2293. * |
  2294. * |
  2295. * |
  2296. * |
  2297. * |
  2298. * |
  2299. * |
  2300. * |
  2301. |*
  2302. | *
  2303. | *
  2304. | *
  2305. | *
  2306. | *
  2307. | *
  2308. | *
  2309. | *
  2310. | *
  2311. | *
  2312. | *
  2313. | *
  2314. | *
  2315. | *
  2316. | *
  2317. | *
  2318. | *
  2319. | *
  2320. | *
  2321. *|
  2322. * |
  2323. * |
  2324. * |
  2325. * |
  2326. * |
  2327. * |
  2328. * |
  2329. * |
  2330. * |
  2331. * |
  2332. * |
  2333. * |
  2334. * |
  2335. * |
  2336. * |
  2337. * |
  2338. |
  2339. | *
  2340. | *
  2341. | *
  2342. | *
  2343. | *
  2344. | *
  2345. | *
  2346. | *
  2347. | *
  2348. | *
  2349. | *
  2350. | *
  2351. | *
  2352. | *
  2353. | *
  2354. | *
  2355. | *
  2356. | *
  2357. | *
  2358. | *
  2359. | *
  2360. | *
  2361. |*
  2362. *|
  2363. * |
  2364. * |
  2365. * |
  2366. * |
  2367. * |
  2368. * |
  2369. * |
  2370. * |
  2371. * |
  2372. * |
  2373. * |
  2374. * |
  2375. * |
  2376. * |
  2377. * |
  2378. * |
  2379. * |
  2380. * |
  2381. * |
  2382. * |
  2383. * |
  2384. * |
  2385. * |
  2386. * |
  2387. * |
  2388. * |
  2389. * |
  2390. * |
  2391. * |
  2392. * |
  2393. * |
  2394. * |
  2395. * |
  2396. * |
  2397. * |
  2398. * |
  2399. * |
  2400. * |
  2401. * |
  2402. * |
  2403. * |
  2404. * |
  2405. * |
  2406. * |
  2407. * |
  2408. * |
  2409. * |
  2410. * |
  2411. * |
  2412. * |
  2413. * |
  2414. * |
  2415. * |
  2416. * |
  2417. * |
  2418. *|
  2419. |*
  2420. | *
  2421. | *
  2422. | *
  2423. | *
  2424. | *
  2425. | *
  2426. See also hplgit.github.io/pysketcher/doc/src/tut/fig-tut/StochasticWavyCurve.png (and .pdf)
  2427. """
  2428. # The curves were generated by the script generate_road_profiles.py and
  2429. # the code below were generated by plot_roads.py. Both scripts are
  2430. # found doc/src/src-bumpy in the repo git@github.com:hplgit/bumpy.git
  2431. def __init__(self, curve_no=0, percentage=100):
  2432. """
  2433. ============= ===================================================
  2434. Argument Explanation
  2435. ============= ===================================================
  2436. curve_no 0, 1, or 2: chooses one out of three shapes.
  2437. percentage The percentage of the defined curve to be used.
  2438. ============= ===================================================
  2439. """
  2440. self._define_curves()
  2441. self.curve_no = curve_no
  2442. m = int(len(self.x)/float(percentage)*100)
  2443. self.shapes = {'wavy': Curve(self.x[:m], self.y[curve_no][:m])}
  2444. def __call__(self, x):
  2445. raise NotImplementedError
  2446. def _define_curves(self):
  2447. self.x = array([0.0000, 0.0606, 0.1212, 0.1818, 0.2424, 0.3030, 0.3636, 0.4242, 0.4848, 0.5455, 0.6061, 0.6667, 0.7273, 0.7879, 0.8485, 0.9091, 0.9697, 1.0303, 1.0909, 1.1515, 1.2121, 1.2727, 1.3333, 1.3939, 1.4545, 1.5152, 1.5758, 1.6364, 1.6970, 1.7576, 1.8182, 1.8788, 1.9394, 2.0000, 2.0606, 2.1212, 2.1818, 2.2424, 2.3030, 2.3636, 2.4242, 2.4848, 2.5455, 2.6061, 2.6667, 2.7273, 2.7879, 2.8485, 2.9091, 2.9697, 3.0303, 3.0909, 3.1515, 3.2121, 3.2727, 3.3333, 3.3939, 3.4545, 3.5152, 3.5758, 3.6364, 3.6970, 3.7576, 3.8182, 3.8788, 3.9394, 4.0000, 4.0606, 4.1212, 4.1818, 4.2424, 4.3030, 4.3636, 4.4242, 4.4848, 4.5455, 4.6061, 4.6667, 4.7273, 4.7879, 4.8485, 4.9091, 4.9697, 5.0303, 5.0909, 5.1515, 5.2121, 5.2727, 5.3333, 5.3939, 5.4545, 5.5152, 5.5758, 5.6364, 5.6970, 5.7576, 5.8182, 5.8788, 5.9394, 6.0000, 6.0606, 6.1212, 6.1818, 6.2424, 6.3030, 6.3636, 6.4242, 6.4848, 6.5455, 6.6061, 6.6667, 6.7273, 6.7879, 6.8485, 6.9091, 6.9697, 7.0303, 7.0909, 7.1515, 7.2121, 7.2727, 7.3333, 7.3939, 7.4545, 7.5152, 7.5758, 7.6364, 7.6970, 7.7576, 7.8182, 7.8788, 7.9394, 8.0000, 8.0606, 8.1212, 8.1818, 8.2424, 8.3030, 8.3636, 8.4242, 8.4848, 8.5455, 8.6061, 8.6667, 8.7273, 8.7879, 8.8485, 8.9091, 8.9697, 9.0303, 9.0909, 9.1515, 9.2121, 9.2727, 9.3333, 9.3939, 9.4545, 9.5152, 9.5758, 9.6364, 9.6970, 9.7576, 9.8182, 9.8788, 9.9394, 10.0000, 10.0606, 10.1212, 10.1818, 10.2424, 10.3030, 10.3636, 10.4242, 10.4848, 10.5455, 10.6061, 10.6667, 10.7273, 10.7879, 10.8485, 10.9091, 10.9697, 11.0303, 11.0909, 11.1515, 11.2121, 11.2727, 11.3333, 11.3939, 11.4545, 11.5152, 11.5758, 11.6364, 11.6970, 11.7576, 11.8182, 11.8788, 11.9394, 12.0000, 12.0606, 12.1212, 12.1818, 12.2424, 12.3030, 12.3636, 12.4242, 12.4848, 12.5455, 12.6061, 12.6667, 12.7273, 12.7879, 12.8485, 12.9091, 12.9697, 13.0303, 13.0909, 13.1515, 13.2121, 13.2727, 13.3333, 13.3939, 13.4545, 13.5152, 13.5758, 13.6364, 13.6970, 13.7576, 13.8182, 13.8788, 13.9394, 14.0000, 14.0606, 14.1212, 14.1818, 14.2424, 14.3030, 14.3636, 14.4242, 14.4848, 14.5455, 14.6061, 14.6667, 14.7273, 14.7879, 14.8485, 14.9091, 14.9697, 15.0303, 15.0909, 15.1515, 15.2121, 15.2727, 15.3333, 15.3939, 15.4545, 15.5152, 15.5758, 15.6364, 15.6970, 15.7576, 15.8182, 15.8788, 15.9394, 16.0000, 16.0606, 16.1212, 16.1818, 16.2424, 16.3030, 16.3636, 16.4242, 16.4848, 16.5455, 16.6061, 16.6667, 16.7273, 16.7879, 16.8485, 16.9091, 16.9697, 17.0303, 17.0909, 17.1515, 17.2121, 17.2727, 17.3333, 17.3939, 17.4545, 17.5152, 17.5758, 17.6364, 17.6970, 17.7576, 17.8182, 17.8788, 17.9394, 18.0000, 18.0606, 18.1212, 18.1818, 18.2424, 18.3030, 18.3636, 18.4242, 18.4848, 18.5455, 18.6061, 18.6667, 18.7273, 18.7879, 18.8485, 18.9091, 18.9697, 19.0303, 19.0909, 19.1515, 19.2121, 19.2727, 19.3333, 19.3939, 19.4545, 19.5152, 19.5758, 19.6364, 19.6970, 19.7576, 19.8182, 19.8788, 19.9394, 20.0000, 20.0606, 20.1212, 20.1818, 20.2424, 20.3030, 20.3636, 20.4242, 20.4848, 20.5455, 20.6061, 20.6667, 20.7273, 20.7879, 20.8485, 20.9091, 20.9697, 21.0303, 21.0909, 21.1515, 21.2121, 21.2727, 21.3333, 21.3939, 21.4545, 21.5152, 21.5758, 21.6364, 21.6970, 21.7576, 21.8182, 21.8788, 21.9394, 22.0000, 22.0606, 22.1212, 22.1818, 22.2424, 22.3030, 22.3636, 22.4242, 22.4848, 22.5455, 22.6061, 22.6667, 22.7273, 22.7879, 22.8485, 22.9091, 22.9697, 23.0303, 23.0909, 23.1515, 23.2121, 23.2727, 23.3333, 23.3939, 23.4545, 23.5152, 23.5758, 23.6364, 23.6970, 23.7576, 23.8182, 23.8788, 23.9394, 24.0000, 24.0606, 24.1212, 24.1818, 24.2424, 24.3030, 24.3636, 24.4242, 24.4848, 24.5455, 24.6061, 24.6667, 24.7273, 24.7879, 24.8485, 24.9091, 24.9697, 25.0303, 25.0909, 25.1515, 25.2121, 25.2727, 25.3333, 25.3939, 25.4545, 25.5152, 25.5758, 25.6364, 25.6970, 25.7576, 25.8182, 25.8788, 25.9394, 26.0000, 26.0606, 26.1212, 26.1818, 26.2424, 26.3030, 26.3636, 26.4242, 26.4848, 26.5455, 26.6061, 26.6667, 26.7273, 26.7879, 26.8485, 26.9091, 26.9697, 27.0303, 27.0909, 27.1515, 27.2121, 27.2727, 27.3333, 27.3939, 27.4545, 27.5152, 27.5758, 27.6364, 27.6970, 27.7576, 27.8182, 27.8788, 27.9394, 28.0000, 28.0606, 28.1212, 28.1818, 28.2424, 28.3030, 28.3636, 28.4242, 28.4848, 28.5455, 28.6061, 28.6667, 28.7273, 28.7879, 28.8485, 28.9091, 28.9697, 29.0303, 29.0909, 29.1515, 29.2121, 29.2727, 29.3333, 29.3939, 29.4545, 29.5152, 29.5758, 29.6364, 29.6970, 29.7576, 29.8182, 29.8788, 29.9394, 30.0000, 30.0606, 30.1212, 30.1818, 30.2424, 30.3030, 30.3636, 30.4242, 30.4848, 30.5455, 30.6061, 30.6667, 30.7273, 30.7879, 30.8485, 30.9091, 30.9697, 31.0303, 31.0909, 31.1515, 31.2121, 31.2727, 31.3333, 31.3939, 31.4545, 31.5152, 31.5758, 31.6364, 31.6970, 31.7576, 31.8182, 31.8788, 31.9394, 32.0000, 32.0606, 32.1212, 32.1818, 32.2424, 32.3030, 32.3636, 32.4242, 32.4848, 32.5455, 32.6061, 32.6667, 32.7273, 32.7879, 32.8485, 32.9091, 32.9697, 33.0303, 33.0909, 33.1515, 33.2121, 33.2727, 33.3333, 33.3939, 33.4545, 33.5152, 33.5758, 33.6364, 33.6970, 33.7576, 33.8182, 33.8788, 33.9394, 34.0000, 34.0606, 34.1212, 34.1818, 34.2424, 34.3030, 34.3636, 34.4242, 34.4848, 34.5455, 34.6061, 34.6667, 34.7273, 34.7879, 34.8485, 34.9091, 34.9697, 35.0303, 35.0909, 35.1515, 35.2121, 35.2727, 35.3333, 35.3939, 35.4545, 35.5152, 35.5758, 35.6364, 35.6970, 35.7576, 35.8182, 35.8788, 35.9394, 36.0000, 36.0606, 36.1212, 36.1818, 36.2424, 36.3030, 36.3636, 36.4242, 36.4848, 36.5455, 36.6061, 36.6667, 36.7273, 36.7879, 36.8485, 36.9091, 36.9697, 37.0303, 37.0909, 37.1515, 37.2121, 37.2727, 37.3333, 37.3939, 37.4545, 37.5152, 37.5758, 37.6364, 37.6970, 37.7576, 37.8182, 37.8788, 37.9394, 38.0000, 38.0606, 38.1212, 38.1818, 38.2424, 38.3030, 38.3636, 38.4242, 38.4848, 38.5455, 38.6061, 38.6667, 38.7273, 38.7879, 38.8485, 38.9091, 38.9697, 39.0303, 39.0909, 39.1515, 39.2121, 39.2727, 39.3333, 39.3939, 39.4545, 39.5152, 39.5758, 39.6364, 39.6970, 39.7576, 39.8182, 39.8788, 39.9394, 40.0000, 40.0606, 40.1212, 40.1818, 40.2424, 40.3030, 40.3636, 40.4242, 40.4848, 40.5455, 40.6061, 40.6667, 40.7273, 40.7879, 40.8485, 40.9091, 40.9697, 41.0303, 41.0909, 41.1515, 41.2121, 41.2727, 41.3333, 41.3939, 41.4545, 41.5152, 41.5758, 41.6364, 41.6970, 41.7576, 41.8182, 41.8788, 41.9394, 42.0000, 42.0606, 42.1212, 42.1818, 42.2424, 42.3030, 42.3636, 42.4242, 42.4848, 42.5455, 42.6061, 42.6667, 42.7273, 42.7879, 42.8485, 42.9091, 42.9697, 43.0303, 43.0909, 43.1515, 43.2121, 43.2727, 43.3333, 43.3939, 43.4545, 43.5152, 43.5758, 43.6364, 43.6970, 43.7576, 43.8182, 43.8788, 43.9394, 44.0000, 44.0606, 44.1212, 44.1818, 44.2424, 44.3030, 44.3636, 44.4242, 44.4848, 44.5455, 44.6061, 44.6667, 44.7273, 44.7879, 44.8485, 44.9091, 44.9697, 45.0303, 45.0909, 45.1515, 45.2121, 45.2727, 45.3333, 45.3939, 45.4545, 45.5152, 45.5758, 45.6364, 45.6970, 45.7576, 45.8182, 45.8788, 45.9394, 46.0000, 46.0606, 46.1212, 46.1818, 46.2424, 46.3030, 46.3636, 46.4242, 46.4848, 46.5455, 46.6061, 46.6667, 46.7273, 46.7879, 46.8485, 46.9091, 46.9697, 47.0303, 47.0909, 47.1515, 47.2121, 47.2727, 47.3333, 47.3939, 47.4545, 47.5152, 47.5758, 47.6364, 47.6970, 47.7576, 47.8182, 47.8788, 47.9394, 48.0000, 48.0606, 48.1212, 48.1818, 48.2424, 48.3030, 48.3636, 48.4242, 48.4848, 48.5455, 48.6061, 48.6667, 48.7273, 48.7879, 48.8485, 48.9091, 48.9697, 49.0303, 49.0909, 49.1515, 49.2121, 49.2727, 49.3333, 49.3939, 49.4545, 49.5152, 49.5758, 49.6364, 49.6970, 49.7576, 49.8182, 49.8788, 49.9394, ])
  2448. self.y = [None]*3
  2449. self.y[0] = array([0.0000, 0.0005, 0.0006, 0.0004, -0.0004, -0.0007, -0.0022, -0.0027, -0.0036, -0.0042, -0.0050, -0.0049, -0.0060, -0.0072, -0.0085, -0.0092, -0.0104, -0.0116, -0.0133, -0.0148, -0.0160, -0.0177, -0.0186, -0.0191, -0.0192, -0.0187, -0.0187, -0.0187, -0.0192, -0.0198, -0.0201, -0.0208, -0.0216, -0.0227, -0.0242, -0.0260, -0.0277, -0.0299, -0.0319, -0.0328, -0.0333, -0.0338, -0.0347, -0.0360, -0.0363, -0.0365, -0.0370, -0.0373, -0.0364, -0.0355, -0.0343, -0.0329, -0.0317, -0.0312, -0.0309, -0.0306, -0.0301, -0.0290, -0.0275, -0.0259, -0.0238, -0.0222, -0.0200, -0.0176, -0.0154, -0.0130, -0.0108, -0.0081, -0.0046, -0.0001, 0.0035, 0.0061, 0.0083, 0.0105, 0.0130, 0.0156, 0.0170, 0.0181, 0.0196, 0.0212, 0.0231, 0.0247, 0.0262, 0.0277, 0.0293, 0.0309, 0.0325, 0.0336, 0.0348, 0.0360, 0.0378, 0.0401, 0.0423, 0.0443, 0.0457, 0.0473, 0.0488, 0.0500, 0.0511, 0.0518, 0.0528, 0.0534, 0.0547, 0.0561, 0.0577, 0.0585, 0.0594, 0.0606, 0.0611, 0.0614, 0.0617, 0.0612, 0.0607, 0.0608, 0.0603, 0.0599, 0.0588, 0.0577, 0.0557, 0.0543, 0.0532, 0.0520, 0.0505, 0.0496, 0.0499, 0.0490, 0.0489, 0.0496, 0.0504, 0.0504, 0.0509, 0.0512, 0.0512, 0.0504, 0.0499, 0.0498, 0.0493, 0.0491, 0.0483, 0.0478, 0.0474, 0.0468, 0.0462, 0.0460, 0.0462, 0.0467, 0.0472, 0.0476, 0.0483, 0.0491, 0.0502, 0.0510, 0.0504, 0.0503, 0.0514, 0.0527, 0.0538, 0.0547, 0.0554, 0.0561, 0.0561, 0.0558, 0.0548, 0.0540, 0.0531, 0.0524, 0.0516, 0.0513, 0.0511, 0.0520, 0.0519, 0.0513, 0.0512, 0.0525, 0.0535, 0.0545, 0.0552, 0.0566, 0.0577, 0.0591, 0.0602, 0.0605, 0.0609, 0.0615, 0.0627, 0.0638, 0.0644, 0.0652, 0.0661, 0.0670, 0.0678, 0.0692, 0.0706, 0.0729, 0.0757, 0.0786, 0.0805, 0.0825, 0.0846, 0.0870, 0.0897, 0.0921, 0.0947, 0.0968, 0.0997, 0.1018, 0.1027, 0.1025, 0.1018, 0.1004, 0.1000, 0.0994, 0.0980, 0.0972, 0.0960, 0.0941, 0.0927, 0.0916, 0.0902, 0.0896, 0.0890, 0.0892, 0.0896, 0.0908, 0.0919, 0.0922, 0.0937, 0.0948, 0.0957, 0.0960, 0.0961, 0.0963, 0.0965, 0.0970, 0.0983, 0.0994, 0.0997, 0.0993, 0.0984, 0.0965, 0.0951, 0.0934, 0.0916, 0.0897, 0.0870, 0.0840, 0.0813, 0.0791, 0.0766, 0.0751, 0.0730, 0.0707, 0.0683, 0.0644, 0.0616, 0.0592, 0.0562, 0.0545, 0.0531, 0.0519, 0.0504, 0.0490, 0.0468, 0.0451, 0.0432, 0.0414, 0.0403, 0.0394, 0.0386, 0.0380, 0.0370, 0.0364, 0.0367, 0.0374, 0.0385, 0.0390, 0.0390, 0.0381, 0.0380, 0.0377, 0.0381, 0.0380, 0.0377, 0.0374, 0.0376, 0.0378, 0.0380, 0.0382, 0.0385, 0.0381, 0.0377, 0.0373, 0.0367, 0.0365, 0.0358, 0.0351, 0.0342, 0.0336, 0.0334, 0.0326, 0.0322, 0.0329, 0.0327, 0.0321, 0.0310, 0.0297, 0.0293, 0.0290, 0.0283, 0.0279, 0.0272, 0.0271, 0.0271, 0.0279, 0.0282, 0.0302, 0.0325, 0.0351, 0.0375, 0.0393, 0.0406, 0.0416, 0.0422, 0.0428, 0.0430, 0.0434, 0.0443, 0.0447, 0.0457, 0.0465, 0.0479, 0.0494, 0.0514, 0.0527, 0.0539, 0.0557, 0.0571, 0.0572, 0.0563, 0.0539, 0.0504, 0.0469, 0.0441, 0.0412, 0.0385, 0.0359, 0.0334, 0.0308, 0.0282, 0.0260, 0.0232, 0.0211, 0.0196, 0.0180, 0.0169, 0.0154, 0.0137, 0.0121, 0.0105, 0.0088, 0.0067, 0.0044, 0.0027, -0.0000, -0.0024, -0.0048, -0.0066, -0.0082, -0.0111, -0.0136, -0.0158, -0.0179, -0.0201, -0.0218, -0.0235, -0.0242, -0.0245, -0.0236, -0.0231, -0.0237, -0.0237, -0.0233, -0.0229, -0.0233, -0.0239, -0.0241, -0.0244, -0.0247, -0.0251, -0.0259, -0.0270, -0.0288, -0.0295, -0.0305, -0.0311, -0.0322, -0.0327, -0.0343, -0.0352, -0.0361, -0.0358, -0.0362, -0.0365, -0.0365, -0.0358, -0.0353, -0.0348, -0.0355, -0.0365, -0.0373, -0.0373, -0.0375, -0.0365, -0.0345, -0.0327, -0.0322, -0.0327, -0.0335, -0.0337, -0.0337, -0.0348, -0.0359, -0.0361, -0.0364, -0.0371, -0.0366, -0.0361, -0.0358, -0.0353, -0.0348, -0.0345, -0.0335, -0.0320, -0.0300, -0.0281, -0.0257, -0.0233, -0.0208, -0.0179, -0.0146, -0.0104, -0.0062, -0.0031, -0.0007, 0.0023, 0.0049, 0.0077, 0.0099, 0.0125, 0.0147, 0.0177, 0.0202, 0.0232, 0.0264, 0.0291, 0.0322, 0.0346, 0.0365, 0.0386, 0.0403, 0.0415, 0.0425, 0.0428, 0.0436, 0.0447, 0.0457, 0.0465, 0.0475, 0.0494, 0.0516, 0.0534, 0.0555, 0.0574, 0.0591, 0.0616, 0.0638, 0.0655, 0.0660, 0.0657, 0.0656, 0.0650, 0.0647, 0.0637, 0.0623, 0.0613, 0.0611, 0.0611, 0.0618, 0.0633, 0.0652, 0.0665, 0.0677, 0.0690, 0.0700, 0.0712, 0.0713, 0.0710, 0.0709, 0.0695, 0.0675, 0.0655, 0.0626, 0.0598, 0.0566, 0.0533, 0.0501, 0.0470, 0.0437, 0.0405, 0.0372, 0.0342, 0.0323, 0.0305, 0.0289, 0.0267, 0.0250, 0.0229, 0.0204, 0.0183, 0.0164, 0.0152, 0.0148, 0.0141, 0.0135, 0.0139, 0.0148, 0.0160, 0.0179, 0.0193, 0.0210, 0.0234, 0.0266, 0.0291, 0.0314, 0.0337, 0.0358, 0.0375, 0.0395, 0.0412, 0.0417, 0.0424, 0.0428, 0.0428, 0.0418, 0.0415, 0.0398, 0.0373, 0.0347, 0.0328, 0.0318, 0.0301, 0.0284, 0.0258, 0.0240, 0.0214, 0.0188, 0.0168, 0.0148, 0.0137, 0.0122, 0.0109, 0.0101, 0.0093, 0.0093, 0.0096, 0.0089, 0.0104, 0.0123, 0.0141, 0.0150, 0.0159, 0.0165, 0.0174, 0.0185, 0.0205, 0.0225, 0.0247, 0.0269, 0.0290, 0.0308, 0.0331, 0.0357, 0.0379, 0.0392, 0.0405, 0.0423, 0.0440, 0.0465, 0.0484, 0.0503, 0.0506, 0.0506, 0.0504, 0.0504, 0.0513, 0.0521, 0.0537, 0.0555, 0.0577, 0.0595, 0.0619, 0.0636, 0.0654, 0.0666, 0.0674, 0.0672, 0.0667, 0.0664, 0.0660, 0.0648, 0.0645, 0.0642, 0.0645, 0.0651, 0.0653, 0.0643, 0.0629, 0.0621, 0.0607, 0.0597, 0.0585, 0.0574, 0.0553, 0.0539, 0.0525, 0.0516, 0.0508, 0.0505, 0.0500, 0.0500, 0.0490, 0.0475, 0.0464, 0.0452, 0.0440, 0.0427, 0.0416, 0.0403, 0.0393, 0.0377, 0.0360, 0.0349, 0.0343, 0.0332, 0.0325, 0.0313, 0.0305, 0.0287, 0.0263, 0.0237, 0.0211, 0.0194, 0.0182, 0.0180, 0.0184, 0.0186, 0.0186, 0.0192, 0.0198, 0.0203, 0.0200, 0.0196, 0.0177, 0.0156, 0.0130, 0.0110, 0.0087, 0.0064, 0.0049, 0.0029, 0.0011, -0.0016, -0.0042, -0.0064, -0.0080, -0.0100, -0.0118, -0.0133, -0.0147, -0.0160, -0.0170, -0.0176, -0.0188, -0.0206, -0.0209, -0.0201, -0.0200, -0.0198, -0.0191, -0.0181, -0.0170, -0.0165, -0.0161, -0.0159, -0.0157, -0.0156, -0.0156, -0.0150, -0.0137, -0.0131, -0.0128, -0.0129, -0.0124, -0.0119, -0.0106, -0.0096, -0.0086, -0.0084, -0.0075, -0.0069, -0.0070, -0.0075, -0.0092, -0.0109, -0.0124, -0.0137, -0.0140, -0.0146, -0.0146, -0.0144, -0.0144, -0.0155, -0.0160, -0.0167, -0.0174, -0.0187, -0.0198, -0.0208, -0.0217, -0.0226, -0.0229, -0.0226, -0.0219, -0.0194, -0.0173, -0.0155, -0.0143, -0.0129, -0.0120, -0.0126, -0.0137, -0.0148, -0.0153, -0.0166, -0.0182, -0.0197, -0.0216, -0.0230, -0.0244, -0.0255, -0.0270, -0.0275, -0.0280, -0.0277, -0.0276, -0.0272, -0.0272, -0.0276, -0.0284, -0.0292, -0.0300, -0.0299, -0.0305, -0.0315, -0.0316, -0.0313, -0.0320, -0.0324, -0.0326, -0.0339, -0.0354, -0.0368, -0.0376, -0.0379, -0.0381, -0.0377, -0.0373, -0.0374, -0.0380, -0.0382, -0.0394, -0.0402, -0.0415, -0.0429, -0.0439, -0.0442, -0.0455, -0.0472, -0.0487, -0.0504, -0.0515, -0.0537, -0.0555, -0.0570, -0.0583, -0.0589, -0.0601, -0.0606, -0.0614, -0.0617, -0.0621, -0.0620, -0.0614, -0.0616, -0.0618, -0.0620, -0.0622, -0.0628, -0.0623, -0.0617, -0.0602, -0.0590, -0.0584, -0.0577, -0.0572, -0.0555, -0.0536, -0.0511, -0.0488, -0.0468, -0.0448, -0.0427, -0.0401, -0.0378, -0.0358, ])
  2450. self.y[1] = array([0.0000, 0.0002, 0.0002, 0.0001, 0.0001, 0.0003, 0.0008, 0.0009, 0.0009, 0.0015, 0.0019, 0.0027, 0.0033, 0.0037, 0.0041, 0.0052, 0.0055, 0.0050, 0.0048, 0.0054, 0.0054, 0.0059, 0.0061, 0.0060, 0.0054, 0.0050, 0.0047, 0.0042, 0.0033, 0.0031, 0.0027, 0.0021, 0.0015, 0.0008, -0.0002, -0.0011, -0.0015, -0.0015, -0.0021, -0.0025, -0.0027, -0.0013, 0.0005, 0.0030, 0.0049, 0.0074, 0.0099, 0.0119, 0.0142, 0.0166, 0.0186, 0.0205, 0.0229, 0.0247, 0.0266, 0.0286, 0.0307, 0.0327, 0.0346, 0.0368, 0.0379, 0.0393, 0.0409, 0.0434, 0.0457, 0.0478, 0.0499, 0.0520, 0.0534, 0.0544, 0.0561, 0.0576, 0.0587, 0.0593, 0.0597, 0.0599, 0.0594, 0.0588, 0.0588, 0.0595, 0.0609, 0.0628, 0.0651, 0.0673, 0.0693, 0.0721, 0.0760, 0.0798, 0.0837, 0.0881, 0.0931, 0.0977, 0.1023, 0.1076, 0.1127, 0.1175, 0.1223, 0.1264, 0.1302, 0.1333, 0.1373, 0.1408, 0.1441, 0.1472, 0.1497, 0.1516, 0.1523, 0.1527, 0.1532, 0.1537, 0.1542, 0.1549, 0.1551, 0.1543, 0.1536, 0.1532, 0.1524, 0.1512, 0.1497, 0.1484, 0.1473, 0.1460, 0.1445, 0.1429, 0.1411, 0.1398, 0.1395, 0.1395, 0.1388, 0.1386, 0.1380, 0.1369, 0.1355, 0.1341, 0.1319, 0.1293, 0.1266, 0.1240, 0.1215, 0.1189, 0.1160, 0.1135, 0.1111, 0.1079, 0.1049, 0.1016, 0.0995, 0.0972, 0.0948, 0.0927, 0.0914, 0.0904, 0.0892, 0.0878, 0.0862, 0.0848, 0.0829, 0.0809, 0.0792, 0.0776, 0.0760, 0.0754, 0.0740, 0.0724, 0.0700, 0.0680, 0.0668, 0.0658, 0.0655, 0.0653, 0.0648, 0.0636, 0.0613, 0.0589, 0.0573, 0.0559, 0.0552, 0.0558, 0.0571, 0.0587, 0.0602, 0.0608, 0.0613, 0.0607, 0.0601, 0.0595, 0.0595, 0.0590, 0.0591, 0.0598, 0.0603, 0.0603, 0.0610, 0.0610, 0.0607, 0.0603, 0.0601, 0.0607, 0.0608, 0.0616, 0.0627, 0.0640, 0.0653, 0.0662, 0.0669, 0.0673, 0.0667, 0.0654, 0.0646, 0.0635, 0.0624, 0.0604, 0.0583, 0.0562, 0.0547, 0.0537, 0.0524, 0.0522, 0.0516, 0.0515, 0.0518, 0.0515, 0.0504, 0.0501, 0.0501, 0.0501, 0.0498, 0.0494, 0.0489, 0.0485, 0.0483, 0.0484, 0.0477, 0.0472, 0.0468, 0.0471, 0.0463, 0.0461, 0.0460, 0.0466, 0.0461, 0.0454, 0.0448, 0.0445, 0.0439, 0.0437, 0.0439, 0.0438, 0.0430, 0.0423, 0.0418, 0.0415, 0.0409, 0.0400, 0.0386, 0.0376, 0.0370, 0.0357, 0.0351, 0.0354, 0.0356, 0.0356, 0.0362, 0.0366, 0.0380, 0.0385, 0.0378, 0.0381, 0.0390, 0.0390, 0.0388, 0.0386, 0.0379, 0.0366, 0.0362, 0.0364, 0.0365, 0.0363, 0.0363, 0.0359, 0.0354, 0.0348, 0.0354, 0.0350, 0.0337, 0.0314, 0.0282, 0.0258, 0.0240, 0.0230, 0.0227, 0.0227, 0.0221, 0.0214, 0.0201, 0.0188, 0.0179, 0.0173, 0.0165, 0.0155, 0.0133, 0.0109, 0.0082, 0.0052, 0.0028, 0.0003, -0.0026, -0.0062, -0.0095, -0.0120, -0.0151, -0.0180, -0.0209, -0.0233, -0.0262, -0.0293, -0.0316, -0.0337, -0.0361, -0.0385, -0.0419, -0.0445, -0.0470, -0.0494, -0.0515, -0.0534, -0.0561, -0.0590, -0.0619, -0.0643, -0.0666, -0.0688, -0.0710, -0.0733, -0.0753, -0.0774, -0.0795, -0.0822, -0.0837, -0.0851, -0.0865, -0.0876, -0.0892, -0.0892, -0.0893, -0.0896, -0.0901, -0.0905, -0.0909, -0.0909, -0.0901, -0.0890, -0.0884, -0.0880, -0.0873, -0.0871, -0.0872, -0.0860, -0.0858, -0.0863, -0.0865, -0.0864, -0.0865, -0.0871, -0.0887, -0.0904, -0.0924, -0.0941, -0.0962, -0.0980, -0.0999, -0.1014, -0.1021, -0.1029, -0.1037, -0.1041, -0.1052, -0.1056, -0.1065, -0.1070, -0.1075, -0.1073, -0.1079, -0.1080, -0.1078, -0.1075, -0.1074, -0.1078, -0.1082, -0.1087, -0.1083, -0.1079, -0.1087, -0.1096, -0.1101, -0.1105, -0.1114, -0.1121, -0.1127, -0.1135, -0.1137, -0.1133, -0.1131, -0.1126, -0.1115, -0.1109, -0.1099, -0.1097, -0.1092, -0.1081, -0.1066, -0.1057, -0.1050, -0.1047, -0.1040, -0.1030, -0.1018, -0.1004, -0.0988, -0.0972, -0.0955, -0.0946, -0.0938, -0.0925, -0.0905, -0.0886, -0.0864, -0.0838, -0.0805, -0.0776, -0.0748, -0.0720, -0.0686, -0.0655, -0.0621, -0.0588, -0.0555, -0.0520, -0.0477, -0.0434, -0.0389, -0.0353, -0.0323, -0.0287, -0.0247, -0.0211, -0.0179, -0.0149, -0.0123, -0.0094, -0.0065, -0.0042, -0.0020, -0.0001, 0.0016, 0.0028, 0.0044, 0.0072, 0.0097, 0.0109, 0.0120, 0.0127, 0.0135, 0.0140, 0.0151, 0.0147, 0.0158, 0.0161, 0.0162, 0.0154, 0.0154, 0.0162, 0.0172, 0.0186, 0.0204, 0.0214, 0.0213, 0.0210, 0.0211, 0.0208, 0.0204, 0.0202, 0.0201, 0.0187, 0.0179, 0.0167, 0.0164, 0.0167, 0.0170, 0.0180, 0.0188, 0.0199, 0.0204, 0.0213, 0.0209, 0.0203, 0.0198, 0.0186, 0.0183, 0.0181, 0.0168, 0.0163, 0.0161, 0.0160, 0.0154, 0.0157, 0.0162, 0.0166, 0.0171, 0.0171, 0.0179, 0.0186, 0.0196, 0.0205, 0.0229, 0.0254, 0.0280, 0.0305, 0.0327, 0.0343, 0.0354, 0.0361, 0.0367, 0.0363, 0.0361, 0.0355, 0.0353, 0.0351, 0.0342, 0.0335, 0.0328, 0.0326, 0.0324, 0.0325, 0.0330, 0.0333, 0.0333, 0.0327, 0.0326, 0.0328, 0.0326, 0.0327, 0.0329, 0.0328, 0.0332, 0.0331, 0.0325, 0.0315, 0.0311, 0.0305, 0.0302, 0.0300, 0.0289, 0.0275, 0.0260, 0.0246, 0.0234, 0.0230, 0.0216, 0.0211, 0.0205, 0.0191, 0.0172, 0.0147, 0.0131, 0.0112, 0.0094, 0.0082, 0.0062, 0.0045, 0.0041, 0.0035, 0.0037, 0.0035, 0.0033, 0.0033, 0.0037, 0.0034, 0.0036, 0.0033, 0.0028, 0.0021, 0.0015, 0.0008, 0.0004, 0.0002, -0.0002, -0.0008, -0.0001, 0.0007, 0.0015, 0.0020, 0.0020, 0.0028, 0.0037, 0.0045, 0.0052, 0.0060, 0.0063, 0.0074, 0.0083, 0.0092, 0.0095, 0.0095, 0.0096, 0.0093, 0.0100, 0.0097, 0.0093, 0.0084, 0.0071, 0.0050, 0.0044, 0.0031, 0.0017, 0.0005, -0.0004, -0.0018, -0.0026, -0.0026, -0.0020, -0.0018, -0.0010, 0.0003, 0.0024, 0.0046, 0.0066, 0.0090, 0.0110, 0.0126, 0.0143, 0.0157, 0.0167, 0.0171, 0.0179, 0.0195, 0.0223, 0.0254, 0.0287, 0.0320, 0.0359, 0.0400, 0.0439, 0.0470, 0.0491, 0.0515, 0.0535, 0.0563, 0.0585, 0.0605, 0.0620, 0.0635, 0.0654, 0.0670, 0.0687, 0.0701, 0.0728, 0.0746, 0.0766, 0.0785, 0.0800, 0.0812, 0.0817, 0.0822, 0.0820, 0.0814, 0.0814, 0.0817, 0.0820, 0.0823, 0.0826, 0.0826, 0.0831, 0.0833, 0.0846, 0.0851, 0.0853, 0.0855, 0.0858, 0.0859, 0.0864, 0.0868, 0.0876, 0.0884, 0.0890, 0.0894, 0.0895, 0.0888, 0.0882, 0.0883, 0.0884, 0.0892, 0.0903, 0.0915, 0.0917, 0.0916, 0.0908, 0.0900, 0.0891, 0.0881, 0.0869, 0.0849, 0.0825, 0.0805, 0.0789, 0.0773, 0.0761, 0.0750, 0.0738, 0.0729, 0.0717, 0.0712, 0.0714, 0.0718, 0.0727, 0.0737, 0.0735, 0.0733, 0.0723, 0.0706, 0.0683, 0.0666, 0.0651, 0.0636, 0.0620, 0.0599, 0.0580, 0.0563, 0.0551, 0.0540, 0.0526, 0.0516, 0.0503, 0.0492, 0.0482, 0.0469, 0.0468, 0.0471, 0.0471, 0.0471, 0.0483, 0.0490, 0.0486, 0.0489, 0.0495, 0.0510, 0.0523, 0.0538, 0.0547, 0.0552, 0.0556, 0.0569, 0.0573, 0.0578, 0.0583, 0.0579, 0.0579, 0.0580, 0.0583, 0.0585, 0.0594, 0.0601, 0.0598, 0.0594, 0.0597, 0.0602, 0.0611, 0.0612, 0.0618, 0.0620, 0.0626, 0.0635, 0.0637, 0.0645, 0.0649, 0.0657, 0.0667, 0.0680, 0.0688, 0.0693, 0.0698, 0.0702, 0.0699, 0.0703, 0.0702, 0.0705, 0.0716, 0.0725, 0.0730, 0.0736, 0.0736, 0.0734, 0.0735, 0.0739, 0.0736, 0.0728, 0.0720, 0.0717, 0.0715, 0.0709, 0.0704, 0.0707, 0.0704, 0.0695, 0.0690, 0.0687, 0.0667, 0.0648, 0.0629, 0.0620, 0.0620, 0.0618, 0.0621, 0.0623, 0.0633, 0.0637, 0.0637, 0.0638, 0.0639, 0.0639, 0.0642, 0.0650, 0.0653, 0.0657, 0.0661, ])
  2451. self.y[2] = array([0.0000, 0.0001, -0.0002, -0.0003, 0.0004, 0.0014, 0.0021, 0.0025, 0.0025, 0.0021, 0.0018, 0.0022, 0.0016, 0.0018, 0.0018, 0.0021, 0.0027, 0.0034, 0.0046, 0.0060, 0.0076, 0.0080, 0.0084, 0.0090, 0.0100, 0.0104, 0.0098, 0.0097, 0.0100, 0.0100, 0.0105, 0.0117, 0.0124, 0.0128, 0.0133, 0.0133, 0.0133, 0.0132, 0.0132, 0.0136, 0.0144, 0.0161, 0.0179, 0.0196, 0.0222, 0.0251, 0.0265, 0.0279, 0.0287, 0.0291, 0.0297, 0.0305, 0.0316, 0.0328, 0.0340, 0.0361, 0.0382, 0.0408, 0.0425, 0.0442, 0.0460, 0.0474, 0.0489, 0.0502, 0.0512, 0.0517, 0.0526, 0.0525, 0.0525, 0.0521, 0.0508, 0.0498, 0.0487, 0.0478, 0.0472, 0.0461, 0.0446, 0.0434, 0.0415, 0.0399, 0.0388, 0.0374, 0.0360, 0.0351, 0.0339, 0.0320, 0.0308, 0.0294, 0.0286, 0.0278, 0.0255, 0.0224, 0.0194, 0.0170, 0.0147, 0.0131, 0.0123, 0.0121, 0.0110, 0.0106, 0.0100, 0.0090, 0.0089, 0.0093, 0.0100, 0.0111, 0.0132, 0.0159, 0.0179, 0.0195, 0.0207, 0.0220, 0.0229, 0.0242, 0.0261, 0.0279, 0.0290, 0.0300, 0.0303, 0.0309, 0.0316, 0.0328, 0.0335, 0.0332, 0.0323, 0.0314, 0.0300, 0.0283, 0.0268, 0.0248, 0.0225, 0.0201, 0.0172, 0.0143, 0.0119, 0.0104, 0.0088, 0.0071, 0.0059, 0.0051, 0.0038, 0.0027, 0.0017, 0.0009, 0.0007, -0.0002, -0.0008, -0.0020, -0.0034, -0.0055, -0.0069, -0.0082, -0.0086, -0.0085, -0.0089, -0.0088, -0.0092, -0.0099, -0.0110, -0.0121, -0.0137, -0.0150, -0.0157, -0.0160, -0.0155, -0.0149, -0.0140, -0.0139, -0.0134, -0.0135, -0.0136, -0.0138, -0.0145, -0.0158, -0.0155, -0.0155, -0.0153, -0.0154, -0.0159, -0.0157, -0.0152, -0.0143, -0.0140, -0.0138, -0.0142, -0.0148, -0.0154, -0.0154, -0.0147, -0.0136, -0.0122, -0.0107, -0.0102, -0.0091, -0.0082, -0.0063, -0.0042, -0.0028, -0.0007, 0.0015, 0.0038, 0.0054, 0.0075, 0.0093, 0.0110, 0.0137, 0.0156, 0.0170, 0.0184, 0.0195, 0.0199, 0.0206, 0.0209, 0.0207, 0.0201, 0.0198, 0.0196, 0.0191, 0.0192, 0.0193, 0.0196, 0.0201, 0.0206, 0.0216, 0.0229, 0.0253, 0.0278, 0.0304, 0.0321, 0.0341, 0.0356, 0.0367, 0.0372, 0.0379, 0.0383, 0.0394, 0.0396, 0.0397, 0.0391, 0.0382, 0.0362, 0.0346, 0.0334, 0.0325, 0.0325, 0.0325, 0.0323, 0.0316, 0.0321, 0.0328, 0.0338, 0.0352, 0.0366, 0.0378, 0.0389, 0.0401, 0.0403, 0.0406, 0.0419, 0.0425, 0.0424, 0.0421, 0.0415, 0.0407, 0.0406, 0.0411, 0.0413, 0.0419, 0.0420, 0.0416, 0.0415, 0.0402, 0.0388, 0.0377, 0.0368, 0.0357, 0.0353, 0.0351, 0.0348, 0.0347, 0.0340, 0.0326, 0.0314, 0.0306, 0.0295, 0.0288, 0.0275, 0.0257, 0.0234, 0.0209, 0.0188, 0.0169, 0.0151, 0.0132, 0.0114, 0.0093, 0.0076, 0.0055, 0.0026, -0.0008, -0.0038, -0.0069, -0.0097, -0.0118, -0.0132, -0.0148, -0.0164, -0.0183, -0.0202, -0.0213, -0.0228, -0.0253, -0.0280, -0.0299, -0.0315, -0.0329, -0.0343, -0.0349, -0.0364, -0.0373, -0.0383, -0.0391, -0.0400, -0.0401, -0.0404, -0.0414, -0.0422, -0.0425, -0.0425, -0.0420, -0.0419, -0.0412, -0.0412, -0.0408, -0.0400, -0.0396, -0.0397, -0.0400, -0.0401, -0.0405, -0.0412, -0.0410, -0.0401, -0.0389, -0.0377, -0.0372, -0.0369, -0.0364, -0.0356, -0.0347, -0.0345, -0.0346, -0.0353, -0.0359, -0.0367, -0.0376, -0.0385, -0.0387, -0.0391, -0.0401, -0.0400, -0.0398, -0.0395, -0.0391, -0.0385, -0.0383, -0.0377, -0.0374, -0.0371, -0.0361, -0.0357, -0.0355, -0.0342, -0.0324, -0.0303, -0.0280, -0.0257, -0.0232, -0.0205, -0.0179, -0.0147, -0.0113, -0.0091, -0.0059, -0.0019, 0.0024, 0.0071, 0.0120, 0.0177, 0.0219, 0.0248, 0.0273, 0.0290, 0.0302, 0.0318, 0.0325, 0.0331, 0.0334, 0.0344, 0.0360, 0.0387, 0.0414, 0.0437, 0.0456, 0.0471, 0.0491, 0.0511, 0.0526, 0.0536, 0.0550, 0.0570, 0.0597, 0.0622, 0.0647, 0.0672, 0.0696, 0.0710, 0.0725, 0.0748, 0.0777, 0.0806, 0.0832, 0.0854, 0.0874, 0.0893, 0.0906, 0.0914, 0.0927, 0.0937, 0.0944, 0.0959, 0.0970, 0.0970, 0.0967, 0.0955, 0.0945, 0.0934, 0.0927, 0.0921, 0.0916, 0.0909, 0.0901, 0.0893, 0.0884, 0.0881, 0.0877, 0.0870, 0.0870, 0.0871, 0.0872, 0.0867, 0.0860, 0.0845, 0.0821, 0.0800, 0.0775, 0.0752, 0.0728, 0.0703, 0.0676, 0.0655, 0.0637, 0.0618, 0.0593, 0.0558, 0.0518, 0.0483, 0.0445, 0.0414, 0.0384, 0.0365, 0.0349, 0.0338, 0.0330, 0.0323, 0.0320, 0.0319, 0.0320, 0.0323, 0.0324, 0.0324, 0.0322, 0.0327, 0.0328, 0.0320, 0.0314, 0.0295, 0.0279, 0.0258, 0.0243, 0.0229, 0.0211, 0.0195, 0.0176, 0.0163, 0.0151, 0.0136, 0.0123, 0.0108, 0.0088, 0.0064, 0.0041, 0.0021, 0.0002, -0.0020, -0.0051, -0.0078, -0.0106, -0.0137, -0.0169, -0.0195, -0.0218, -0.0235, -0.0253, -0.0269, -0.0281, -0.0290, -0.0299, -0.0315, -0.0328, -0.0336, -0.0344, -0.0352, -0.0354, -0.0353, -0.0358, -0.0355, -0.0347, -0.0344, -0.0338, -0.0335, -0.0324, -0.0319, -0.0319, -0.0322, -0.0334, -0.0351, -0.0365, -0.0381, -0.0401, -0.0424, -0.0449, -0.0472, -0.0491, -0.0506, -0.0522, -0.0533, -0.0544, -0.0553, -0.0553, -0.0557, -0.0558, -0.0563, -0.0573, -0.0587, -0.0602, -0.0617, -0.0636, -0.0666, -0.0699, -0.0734, -0.0760, -0.0791, -0.0817, -0.0840, -0.0857, -0.0869, -0.0881, -0.0882, -0.0884, -0.0888, -0.0893, -0.0889, -0.0876, -0.0869, -0.0858, -0.0851, -0.0836, -0.0825, -0.0813, -0.0807, -0.0800, -0.0802, -0.0805, -0.0807, -0.0813, -0.0818, -0.0819, -0.0822, -0.0829, -0.0825, -0.0829, -0.0838, -0.0840, -0.0836, -0.0824, -0.0810, -0.0799, -0.0789, -0.0785, -0.0785, -0.0774, -0.0770, -0.0761, -0.0755, -0.0752, -0.0746, -0.0744, -0.0741, -0.0729, -0.0718, -0.0710, -0.0694, -0.0673, -0.0649, -0.0627, -0.0606, -0.0592, -0.0582, -0.0571, -0.0555, -0.0539, -0.0527, -0.0516, -0.0508, -0.0503, -0.0508, -0.0518, -0.0527, -0.0536, -0.0544, -0.0545, -0.0541, -0.0547, -0.0553, -0.0561, -0.0573, -0.0587, -0.0600, -0.0608, -0.0616, -0.0624, -0.0626, -0.0631, -0.0638, -0.0630, -0.0621, -0.0612, -0.0603, -0.0594, -0.0577, -0.0564, -0.0551, -0.0535, -0.0511, -0.0485, -0.0455, -0.0424, -0.0396, -0.0374, -0.0360, -0.0351, -0.0343, -0.0336, -0.0324, -0.0311, -0.0296, -0.0276, -0.0259, -0.0240, -0.0223, -0.0212, -0.0205, -0.0201, -0.0203, -0.0214, -0.0232, -0.0261, -0.0285, -0.0304, -0.0320, -0.0342, -0.0365, -0.0381, -0.0394, -0.0416, -0.0434, -0.0443, -0.0450, -0.0465, -0.0482, -0.0490, -0.0497, -0.0508, -0.0506, -0.0509, -0.0514, -0.0522, -0.0525, -0.0532, -0.0547, -0.0553, -0.0575, -0.0597, -0.0629, -0.0666, -0.0702, -0.0737, -0.0773, -0.0805, -0.0835, -0.0847, -0.0875, -0.0893, -0.0913, -0.0929, -0.0935, -0.0943, -0.0949, -0.0957, -0.0962, -0.0977, -0.0994, -0.1011, -0.1031, -0.1057, -0.1086, -0.1114, -0.1141, -0.1158, -0.1174, -0.1197, -0.1219, -0.1234, -0.1250, -0.1267, -0.1273, -0.1270, -0.1268, -0.1263, -0.1257, -0.1240, -0.1220, -0.1210, -0.1197, -0.1197, -0.1188, -0.1171, -0.1147, -0.1122, -0.1101, -0.1077, -0.1052, -0.1031, -0.1020, -0.1008, -0.0983, -0.0960, -0.0934, -0.0908, -0.0889, -0.0865, -0.0836, -0.0799, -0.0767, -0.0740, -0.0716, -0.0692, -0.0663, -0.0628, -0.0596, -0.0567, -0.0535, -0.0508, -0.0472, -0.0435, -0.0394, -0.0347, -0.0301, -0.0253, -0.0210, -0.0165, -0.0126, -0.0095, -0.0072, -0.0046, -0.0020, -0.0001, 0.0023, 0.0050, 0.0077, 0.0102, 0.0135, 0.0175, 0.0212, 0.0245, 0.0280, 0.0309, 0.0336, 0.0365, 0.0397, 0.0433, 0.0474, 0.0511, 0.0546, 0.0574, 0.0602, 0.0634, 0.0663, 0.0695, 0.0729, 0.0752, 0.0762, 0.0773, 0.0781, 0.0790, 0.0806, 0.0836, 0.0857, 0.0879, 0.0896, 0.0920, 0.0949, 0.0975, 0.1002, ])
  2452. # COMPOSITE types:
  2453. # MassSpringForce: Line(horizontal), Spring, Rectangle, Arrow/Line(w/arrow)
  2454. # must be easy to find the tip of the arrow
  2455. # Maybe extra dict: self.name['mass'] = Rectangle object - YES!
  2456. class ArbitraryVolume(Shape):
  2457. """
  2458. An arbitrary closed volume with an optional normal vector and a
  2459. vector field to be used in derivation of continuum mechanical
  2460. equations.
  2461. """
  2462. def __init__(self, position, width=1,
  2463. volume_symbol='V',
  2464. volume_symbol_fontsize='18',
  2465. normal_vector_symbol='n',
  2466. vector_field_symbol=None):
  2467. """
  2468. ============================ ====================================
  2469. Name Description
  2470. ============================ ====================================
  2471. position center point of volume
  2472. width width of volume (about 3 is best)
  2473. normal_vector_symbol symbol of None (no boundary normal)
  2474. volume_symbol None (no center symbol) or character
  2475. volume_symbol_fontsize fontsize of volume symbol
  2476. vector_field_symbol None (no vector) or symbol
  2477. ============================ ====================================
  2478. """
  2479. self.position, self.width = position, width
  2480. self.vector_symbol = vector_field_symbol
  2481. self.normal_symbol = normal_vector_symbol
  2482. ellipse, normal, vector = self._perturbed_unit_ellipse()
  2483. self.shapes = {'closed_curve': ellipse}
  2484. if normal_vector_symbol:
  2485. self.shapes['normal'] = normal
  2486. if vector_field_symbol is not None:
  2487. self.shapes['vector'] = vector
  2488. # Scale and translate
  2489. self.rotate(20, (0,0))
  2490. self.scale(width/2.0)
  2491. self.translate(position)
  2492. # Must be placed at position after translation:
  2493. if volume_symbol:
  2494. self.shapes['name'] = Text('$%s$' % volume_symbol, position,
  2495. fontsize=volume_symbol_fontsize)
  2496. def _perturbed_unit_ellipse(self):
  2497. """Draw the volume as a perturbed ellipse of about unit size."""
  2498. a0 = 1.0
  2499. b0 = 0.75
  2500. eps_a = 0.2
  2501. eps_b = 0.1
  2502. a = lambda t: a0 + eps_a*sin(1*t)
  2503. b = lambda t: b0 + eps_b*cos(1*t)
  2504. x = lambda t: a(t)*cos(t)
  2505. y = lambda t: b(t)*sin(t)
  2506. t = linspace(0, 2*pi, 101) # parameter
  2507. ellipse = Curve(x(t), y(t))
  2508. # Make normal vector
  2509. tx = lambda t: eps_a*cos(t)*cos(t) - a(t)*sin(t)
  2510. ty = lambda t: -eps_b*sin(t)*sin(t) + b(t)*cos(t)
  2511. t0 = pi/5
  2512. nx = ty(t0)
  2513. ny = -tx(t0)
  2514. nx = nx/sqrt(nx**2 + ny**2)
  2515. ny = ny/sqrt(nx**2 + ny**2)
  2516. Px = x(t0)
  2517. Py = y(t0)
  2518. start = point(x(t0), y(t0))
  2519. end = start + point(0.75*b0*nx, 0.75*b0*ny)
  2520. normal = Force(start, end, '$\\boldsymbol{%s}$' % self.normal_symbol,
  2521. text_spacing=1./60,
  2522. text_pos='end',
  2523. text_alignment='center')
  2524. end = start + point(0.75*b0/3*nx, 0.75*b0*4*ny)
  2525. vector = Force(start, end, '$\\boldsymbol{%s}$' % self.vector_symbol,
  2526. text_spacing=1./60,
  2527. text_pos='end',
  2528. text_alignment='center')
  2529. return ellipse, normal, vector
  2530. def geometric_features(self):
  2531. """
  2532. Recorded geometric features:
  2533. ==================== =============================================
  2534. Attribute Description
  2535. ==================== =============================================
  2536. position center point of volume
  2537. normal_vector_start start of normal vector
  2538. normal_vector_end end of normal vector
  2539. ==================== =============================================
  2540. """
  2541. d = {'position': self.position}
  2542. if 'normal' in self.shapes:
  2543. d['normal_vector_start'] = self.shapes['normal'].geometric_features()['start']
  2544. d['normal_vector_end'] = self.shapes['normal'].geometric_features()['end']
  2545. return d
  2546. def _test1():
  2547. set_coordinate_system(xmin=0, xmax=10, ymin=0, ymax=10)
  2548. l1 = Line((0,0), (1,1))
  2549. l1.draw()
  2550. eval(input(': '))
  2551. c1 = Circle((5,2), 1)
  2552. c2 = Circle((6,2), 1)
  2553. w1 = Wheel((7,2), 1)
  2554. c1.draw()
  2555. c2.draw()
  2556. w1.draw()
  2557. hardcopy()
  2558. display() # show the plot
  2559. def _test2():
  2560. set_coordinate_system(xmin=0, xmax=10, ymin=0, ymax=10)
  2561. l1 = Line((0,0), (1,1))
  2562. l1.draw()
  2563. eval(input(': '))
  2564. c1 = Circle((5,2), 1)
  2565. c2 = Circle((6,2), 1)
  2566. w1 = Wheel((7,2), 1)
  2567. filled_curves(True)
  2568. set_linecolor('blue')
  2569. c1.draw()
  2570. set_linecolor('aqua')
  2571. c2.draw()
  2572. filled_curves(False)
  2573. set_linecolor('red')
  2574. w1.draw()
  2575. hardcopy()
  2576. display() # show the plot
  2577. def _test3():
  2578. """Test example from the book."""
  2579. set_coordinate_system(xmin=0, xmax=10, ymin=0, ymax=10)
  2580. l1 = Line(start=(0,0), stop=(1,1)) # define line
  2581. l1.draw() # make plot data
  2582. r1 = Rectangle(lower_left_corner=(0,1), width=3, height=5)
  2583. r1.draw()
  2584. Circle(center=(5,7), radius=1).draw()
  2585. Wheel(center=(6,2), radius=2, inner_radius=0.5, nlines=7).draw()
  2586. hardcopy()
  2587. display()
  2588. def _test4():
  2589. """Second example from the book."""
  2590. set_coordinate_system(xmin=0, xmax=10, ymin=0, ymax=10)
  2591. r1 = Rectangle(lower_left_corner=(0,1), width=3, height=5)
  2592. c1 = Circle(center=(5,7), radius=1)
  2593. w1 = Wheel(center=(6,2), radius=2, inner_radius=0.5, nlines=7)
  2594. c2 = Circle(center=(7,7), radius=1)
  2595. filled_curves(True)
  2596. c1.draw()
  2597. set_linecolor('blue')
  2598. r1.draw()
  2599. set_linecolor('aqua')
  2600. c2.draw()
  2601. # Add thick aqua line around rectangle:
  2602. filled_curves(False)
  2603. set_linewidth(4)
  2604. r1.draw()
  2605. set_linecolor('red')
  2606. # Draw wheel with thick lines:
  2607. w1.draw()
  2608. hardcopy('tmp_colors')
  2609. display()
  2610. def _test5():
  2611. set_coordinate_system(xmin=0, xmax=10, ymin=0, ymax=10)
  2612. c = 6. # center point of box
  2613. w = 2. # size of box
  2614. L = 3
  2615. r1 = Rectangle((c-w/2, c-w/2), w, w)
  2616. l1 = Line((c,c-w/2), (c,c-w/2-L))
  2617. linecolor('blue')
  2618. filled_curves(True)
  2619. r1.draw()
  2620. linecolor('aqua')
  2621. filled_curves(False)
  2622. l1.draw()
  2623. hardcopy()
  2624. display() # show the plot
  2625. def rolling_wheel(total_rotation_angle):
  2626. """Animation of a rotating wheel."""
  2627. set_coordinate_system(xmin=0, xmax=10, ymin=0, ymax=10)
  2628. import time
  2629. center = (6,2)
  2630. radius = 2.0
  2631. angle = 2.0
  2632. pngfiles = []
  2633. w1 = Wheel(center=center, radius=radius, inner_radius=0.5, nlines=7)
  2634. for i in range(int(total_rotation_angle/angle)):
  2635. w1.draw()
  2636. print('BIG PROBLEM WITH ANIMATE!!!')
  2637. display()
  2638. filename = 'tmp_%03d' % i
  2639. pngfiles.append(filename + '.png')
  2640. hardcopy(filename)
  2641. time.sleep(0.3) # pause
  2642. L = radius*angle*pi/180 # translation = arc length
  2643. w1.rotate(angle, center)
  2644. w1.translate((-L, 0))
  2645. center = (center[0] - L, center[1])
  2646. erase()
  2647. cmd = 'convert -delay 50 -loop 1000 %s tmp_movie.gif' \
  2648. % (' '.join(pngfiles))
  2649. print('converting PNG files to animated GIF:\n', cmd)
  2650. import subprocess
  2651. failure, output = subprocess.getstatusoutput(cmd)
  2652. if failure: print('Could not run', cmd)
  2653. if __name__ == '__main__':
  2654. #rolling_wheel(40)
  2655. #_test1()
  2656. #_test3()
  2657. funcs = [
  2658. #test_Axis,
  2659. test_inclined_plane,
  2660. ]
  2661. for func in funcs:
  2662. func()
  2663. input('Type Return: ')