# Cythonize the encode machinery?

import colorsys

from .rate_control import rate

import platform

from math import sqrt, tan, pi

import time

# vpython provides clock in its namespace

clock = time.perf_counter

def sign(x): # for compatibility with Web VPython

if x > 0: return 1

if x < 0: return -1

return 0

import sys

from . import __version__, __gs_version__

from ._notebook_helpers import _isnotebook

from ._vector_import_helper import (vector, mag, norm, cross, dot, adjust_up,

adjust_axis, object_rotate)

def Exit():

# no infinite loop here so build processs can finish.

# print("in atexit")

pass

import atexit

if platform.system() == 'Windows':

atexit.register(Exit)

# List of names that will be imported from this file with import *

__all__ = ['Camera', 'GlowWidget', 'version', 'GSversion', 'Mouse', 'arrow', 'attach_arrow',

'attach_light', 'attach_trail', 'baseObj', 'box', 'bumpmaps', 'button',

'canvas', 'checkbox', 'clock', 'color', 'combin', 'compound', 'cone', 'controls',

'curve', 'curveMethods', 'cylinder', 'distant_light', 'ellipsoid',

'event_return', 'extrusion', 'faces', 'frame', 'gcurve', 'gdots',

'ghbars', 'gobj', 'graph', 'gvbars', 'helix', 'label',

'local_light', 'menu', 'meta_canvas', 'points', 'pyramid',

'quad', 'radio', 'ring', 'set_browser', 'simple_sphere', 'sleep', 'slider', 'sphere',

'standardAttributes', 'text', 'textures', 'triangle', 'vertex',

'wtext', 'winput', 'keysdown', 'sign']

from inspect import signature # Python 3; needed to allow zero arguments in a bound function

# __version__ is the version number of the Jupyter VPython installer, generated in building the installer.

version = [__version__, 'jupyter']

GSversion = [__gs_version__, 'glowscript']

keysdownlist = [] # list of keys currently pressed

# To print immediately, do this:

# print(.....)

# sys.stdout.flush()

##def eprint(*args, **kwargs): # this may output when ordinary print won't

## print(*args, file=sys.stderr, **kwargs)

##

##def jprint(s): # prints to terminal, for debugging purposes

## os.write(1, json.dumps(s, separators=(',', ':')).encode('utf_8')+ b'\n')

# scalar attribute: { <'a' or 'b'>: string }

# string is str(idx)+attrs[<attributename>]+str(attributevalue)

# for example: {'a':'23K1.72'} thickness of object 23 is 1.72

# vector attrribute:

# string is str(idx)+ attrs[<attributename>]+str(val.x)+','+str(val.y)+','+str(val.z)

# for example: {'a':'37f23.54678,32.12345,-65.00123'} axis of object 37

# text attribute:

# string is str(value)

# boolean values are integers (0,1)

# method: {'m': str(idx)+methods[<methodname>]+<value(s) as above>}

# attrs are X in {'a': '23X....'}

__attrs = {'pos':'a', 'up':'b', 'color':'c', 'trail_color':'d', # don't use single and double quotes; available: comma, but maybe that would cause trouble

'ambient':'e', 'axis':'f', 'size':'g', 'origin':'h',

'direction':'j', 'linecolor':'k', 'bumpaxis':'l', 'dot_color':'m',

'foreground':'n', 'background':'o', 'ray':'p', 'center':'E', 'forward':'#', 'resizable':'+',

# scalar attributes

'graph':'q', 'canvas':'r', 'trail_radius':'s',

'visible':'t', 'opacity':'u','shininess':'v', 'emissive':'w',

'make_trail':'x', 'trail_type':'y', 'interval':'z', 'pps':'A', 'retain':'B',

'red':'C', 'green':'D', 'blue':'F','length':'G', 'width':'H', 'height':'I', 'radius':'J',

'thickness':'K', 'shaftwidth':'L', 'headwidth':'M', 'headlength':'N', 'pickable':'O',

'coils':'P', 'xoffset':'Q', 'yoffset':'R',

'border':'S', 'line':'T', 'box':'U', 'space':'V', 'linewidth':'W',

'xmin':'X', 'xmax':'Y', 'ymin':'Z', 'ymax':'`',

'ctrl':'~', 'shift':'!', 'alt':'@',

# text attributes:

'text':'$', 'align':'%', 'caption':'^',

'fast':'-', 'title':'&', 'xtitle':'*', 'ytitle':'(',

# Miscellany:

'lights':')', 'objects':'_', 'bind':'=',

'pixel_pos':'[', 'texpos':']',

'v0':'{', 'v1':'}', 'v2':';', 'v3':':', 'vs':'<', 'type':'>',

'font':'?', 'texture':'/'}

# attrsb are X in {'b': '23X....'}; ran out of easily typable one-character codes

__attrsb = {'userzoom':'a', 'userspin':'b', 'range':'c', 'autoscale':'d', 'fov':'e',

'normal':'f', 'data':'g', 'checked':'h', 'disabled':'i', 'selected':'j',

'vertical':'k', 'min':'l', 'max':'m', 'step':'n', 'value':'o', 'left':'p',

'right':'q', 'top':'r', 'bottom':'s', '_cloneid':'t',

'logx':'u', 'logy':'v', 'dot':'w', 'dot_radius':'x',

'markers':'y', 'legend':'z', 'label':'A', 'delta':'B', 'marker_color':'C',

'size_units':'D', 'userpan':'E', 'scroll':'F', 'choices':'G', 'depth':'H',

'round':'I', 'name':'J', 'offset':'K', 'attach_idx':'L'}

# methods are X in {'m': '23X....'}

# pos is normally updated as an attribute, but for interval-based trails, it is updated (multiply) as a method

__methods = {'select':'a', 'pos':'b', 'start':'c', 'stop':'d', 'clear':'f', # unused eghijklmnopvxyzCDFABu

'plot':'q', 'add_to_trail':'s',

'follow':'t', 'clear_trail':'w',

'bind':'G', 'unbind':'H', 'waitfor':'I', 'pause':'J', 'pick':'K', 'GSprint':'L',

'delete':'M', 'capture':'N'}

__vecattrs = ['pos', 'up', 'color', 'trail_color', 'axis', 'size', 'origin',

'direction', 'linecolor', 'bumpaxis', 'dot_color', 'ambient', 'add_to_trail',

'foreground', 'background', 'ray', 'ambient', 'center', 'forward', 'normal',

'marker_color', 'offset']

__textattrs = ['text', 'align', 'caption', 'title', 'xtitle', 'ytitle', 'selected', 'label', 'capture', 'name',

'append_to_caption', 'append_to_title', 'bind', 'unbind', 'pause', 'GSprint', 'choices']

def _encode_attr2(sendval, val, ismethods):

s = ''

if sendval in __vecattrs: # it would be good to do some kind of compression of doubles

s += "{:.16G},{:.16G},{:.16G}".format(val[0], val[1], val[2])

elif sendval in __textattrs:

# '\n' doesn't survive JSON transmission, so we replace '\n' with '<br>' (and convert back in glowcomm)

if not isinstance(val, str): val = print_to_string(val)

val = val.replace('\n', '<br>')

s += val

elif sendval == 'rotate':

for p in val:

s += "{:.16G},".format(p)

s = s[:-1]

elif sendval == 'plot' or sendval == 'data':

# for p in val:

# s += "{:.16G},{:.16G},".format(p[0], p[1])

if sendval == 'data' and len(val) == 0: s += "None, None,"

else:

for p in val:

s += "{:.16G},{:.16G},".format(p[0], p[1])

s = s[:-1]

elif sendval in ['v0', 'v1', 'v2', 'v3']: # val is the vertex object referenced by triangle or quad

s += str(val.idx)

elif sendval == 'vs':

if len(val) == 3:

s += "{d},{d},{d}".format(val[0].idx, val[1].idx, val[2].idx)

else:

s += "{d},{d},{d},{d}".format(val[0].idx, val[1].idx, val[2].idx, val[3].idx)

elif ismethods:

#if sendval in ['follow', 'waitfor', 'delete']: val = str(val) # scene.camera.follow(object idx)

s += str(val)

else:

s += "{:.16G}".format(val)

return s

def _encode_attr(D, ismethods): # ismethods is True if a list of method operations

# If ismethods is True, D is [ [idx, method, data], [idx, method, data], etc. ]

# If ismethods is False: D is {idx:{'pos':vec, etc.}, idx:{'pos':vec, etc.}, etc.}

# For 'attrs', convert {'opacity': 0.5, 'idx': 3} to 'a(idx)X(value)' or 'b(idx)X(value)'

# where (idx) is a number, X is the attribute code in attrs, (value) is a number or x,y,z or a string

# For a method, we have {idx, method, value} to be converted to 'm(idx)X(value)'

out = []

if ismethods: # methods; a list

for d in D:

s = 'm'+str(d[0])+__methods[d[1]]

s += _encode_attr2(d[1], d[2], True)

out.append(s)

else: # attributes, a dictionary

for idx,d in D.items():

for sendval,val in d.items():

if sendval in __attrs:

s = 'a'+str(idx)+__attrs[sendval]

else:

s = 'b'+str(idx)+__attrsb[sendval]

s += _encode_attr2(sendval, val, False)

out.append(s)

return out

if sys.version > '3':

long = int

def list_to_vec(L):

return vector(L[0], L[1], L[2])

class baseObj(object):

glow = None

objCnt = 0

sent = True # set to True by a render in the non-notebook case

_view_constructed = False

_canvas_constructing = False

# 'cmds': list of constructors,

# 'attrs': {idx:{'pos':vec, etc.}, idx:{'pos':vec, etc.}, etc.},

# 'methods': [ [idx, method, date], [idx, method, date], etc. ]

updates = {'cmds':[], 'methods':[], 'attrs':{}}

object_registry = {} ## idx -> instance

attach_arrows = []

attach_trails = [] # needed only for functions

follow_objects = [] # entries are [invisible object to follow, function to call for pos, prevous pos]

attrs = set() # each element is (idx, attr name)

@classmethod

def initialize(cls):

cls.updates = {'cmds':[], 'methods':[], 'attrs':{}}

cls.attrs = set()

@classmethod

def empty(cls):

b = cls.updates

return b['cmds'] == [] and b['methods'] == [] and len(cls.attrs) == 0

@classmethod

def handle_attach(cls): # called when about to send data to the browser

## update every attach_arrow if relevant vector has changed

for aa in cls.attach_arrows:

if not aa._run: continue

obj = baseObj.object_registry[aa._object]

if not hasattr(obj, aa.attr): # no longer an attribute of the object

continue

vval = getattr(obj, aa.attr) # could be 'velocity', for example

if not isinstance(vval, vector):

raise AttributeError('attach_arrow attribute "'+aa.attr+'" value must be a vector.')

if aa._last_pos.equals(obj._pos) and aa._last_val.equals(vval): continue

aa._last_val = vector(vval) # keep copies of last vectors

aa._last_pos = vector(obj._pos)

aa.pos = obj._pos

aa.axis = aa._scale*vval

## update every attach_trail that depends on a function

for aa in cls.attach_trails:

if aa._obj == '_func':

fval = aa._func()

if not isinstance(fval, vector):

raise AttributeError('attach_trail value must be a vector')

aa.addmethod('add_to_trail', fval.value)

else:

fval = cls.object_registry[aa._obj].pos

if ( isinstance(aa._last_val, vector) and aa._last_val.equals(fval) ):

continue

aa._last_val = fval

## update every scene.camera.follow(function)

for aa in cls.follow_objects:

obj = aa[0]

val = aa[1]()

lastpos = aa[2]

if val != lastpos:

aa[2] = val

obj.pos = val

def __init__(self, **kwargs):

if not (baseObj._view_constructed or

baseObj._canvas_constructing):

if _isnotebook:

from .with_notebook import _

else:

from .no_notebook import _

baseObj._view_constructed = True

self.idx = baseObj.objCnt ## an integer

self.object_registry[self.idx] = self

if kwargs is not None:

for key, value in kwargs.items():

object.__setattr__(self, key, value)

baseObj.incrObjCnt()

def delete(self):

# Currently delete decrements the object count but doesn't

# delete any objects in the browser....so, let's not decrement

# the count for now.

# baseObj.decrObjCnt()

# The body of this and __del__ are the same other than the

# decrement here, so let's just call __del__.

self.__del__()

def appendcmd(self,cmd):

# The following code makes sure that constructors are sent to the front end first.

cmd['idx'] = self.idx

while not baseObj.sent: # baseObj.sent is always True in the notebook case

time.sleep(0.001)

baseObj.updates['cmds'].append(cmd) # this is an "atomic" (uninterruptable) operation

def addmethod(self, method, data):

while not baseObj.sent: # baseObj.sent is always True in the notebook case

time.sleep(0.001)

baseObj.updates['methods'].append((self.idx, method, data)) # this is an "atomic" (uninterruptable) operation

def addattr(self, attr):

while not baseObj.sent: # baseObj.sent is always True in the notebook case

time.sleep(0.001)

baseObj.attrs.add((self.idx, attr)) # this is an "atomic" (uninterruptable) operation

@classmethod

def package(cls, objdata): # package up the data to send to the browser

ret = {}

m = None

if objdata['cmds'] != []: ret['cmds'] = objdata['cmds']

if objdata['methods'] != []:

m = _encode_attr(objdata['methods'], True)

if objdata['attrs'] != {}:

ret['attrs'] = _encode_attr(objdata['attrs'], False)

if m is not None: ret['attrs'] += m

elif m is not None:

ret['attrs'] = m

return ret

@classmethod

def trigger(cls): # isnotebook; called by a canvas update event from browser, coming from GlowWidget.handle_msg

if cls.empty():

objdata = 'trigger' # handshake with browser

else:

cls.handle_attach()

for a in baseObj.attrs:

idx, attr = a

val = getattr(baseObj.object_registry[idx], attr)

if type(val) is vector: val = [val.x, val.y, val.z]

if idx in baseObj.updates['attrs']:

baseObj.updates['attrs'][idx][attr] = val

else:

baseObj.updates['attrs'][idx] = {attr:val}

objdata = cls.package(baseObj.updates)

sender(objdata)

cls.initialize()

baseObj.sent = True

@classmethod

def incrObjCnt(cls):

cls.objCnt += 1

@classmethod

def decrObjCnt(cls):

cls.objCnt -= 1

def __del__(self):

cmd = {"cmd": "delete", "idx": self.idx}

if (baseObj.glow is not None and sender is not None):

sender([cmd])

else:

self.appendcmd(cmd)

# Jupyter does not immediately transmit data to the browser from a thread,

# which made for an awkward thread in early versions of Jupyter VPython, and

# even caused display mistakes due to losing portions of data sent to the browser

# from within a thread.

# Now there is no threading in Jupyter VPython. Data is sent to the

# browser from the trigger() function, which is called by a

# canvas_update event sent to Python from the browser (glowcomm.js), currently

# every 33 milliseconds. When trigger() is called, it immediately signals

# the browser to set a timeout of 33 ms to send another signal to Python.

# Note that a typical VPython program starts out by creating objects (constructors) and

# specifying their attributes. The 33 ms signal from the browser is adequate to ensure

# prompt data transmissions to the browser.

# The situation with non-notebook use is similar, but the http server is threaded,

# in order to serve glowcomm.html, jpg texture files, and font files, and the

# websocket is also threaded.

# In both the notebook and non-notebook cases output is buffered in baseObj.updates

# and sent as a block to the browser at render times.

class GlowWidget(object):

def __init__(self, wsport=None, wsuri=None):

global sender

baseObj.glow = self

if _isnotebook:

from ipykernel.comm import Comm

if (wsport):

self.comm = Comm(target_name='glow', data={'wsport':wsport, 'wsuri':wsuri})

else:

self.comm = Comm(target_name='glow')

self.comm.on_close(self.handle_close)

self.comm.on_msg(self.handle_msg)

sender = self.comm.send

self.show = True

else:

sender = None

## baseObj.object_registry = {}

## idx -> instance

def handle_msg(self, msg):

events = msg['content']['data'] # this is a list of time-ordered events

for evt in events:

if 'widget' in evt:

obj = baseObj.object_registry[evt['idx']]

if evt['widget'] == 'button':

pass

elif evt['widget'] == 'slider':

obj._value = evt['value']

elif evt['widget'] == 'menu':

obj._selected = obj._choices[evt['value']]

elif evt['widget'] == 'checkbox':

obj._checked = evt['value']

elif evt['widget'] == 'radio':

obj.checked = evt['value']

elif evt['widget'] == 'winput':

obj._text = evt['text']

obj._number = evt['value']

# inspect the bound function and see what it's expecting

a = signature(obj._bind)

if str(a) != '()':

obj._bind( obj )

else:

obj._bind()

else: ## a canvas event

if 'trigger' not in evt:

cvs = baseObj.object_registry[evt['canvas']]

cvs.handle_event(evt)

if _isnotebook:

baseObj.trigger()

def handle_close(self, data):

print ("comm closed")

def _wait(cvs): # wait for an event

cvs._waitfor = None

if _isnotebook: baseObj.trigger() # in notebook environment must send methods immediately

while cvs._waitfor is None:

rate(30)

return cvs._waitfor

class color(object):

black = vector(0,0,0)

white = vector(1,1,1)

red = vector(1,0,0)

green = vector(0,1,0)

blue = vector(0,0,1)

yellow = vector(1,1,0)

cyan = vector(0,1,1)

magenta = vector(1,0,1)

orange = vector(1,0.6,0)

purple = vector(0.4,0.2,0.6)

@classmethod

def gray(cls,luminance):

return vector(luminance,luminance,luminance)

@classmethod

def rgb_to_hsv(cls,v):

T = [v.x, v.y, v.z]

c = colorsys.rgb_to_hsv(*T)

return list_to_vec(c)

@classmethod

def hsv_to_rgb(cls,v):

T = [v.x, v.y, v.z]

c = colorsys.hsv_to_rgb(*T)

return list_to_vec(c)

@classmethod

def rgb_to_grayscale(cls,v):

luminance = 0.21*v.x + 0.71*v.y + 0.07*v.z

return vector(luminance, luminance, luminance)

class textures(object):

flower = ":flower_texture.jpg"

granite=":granite_texture.jpg"

gravel=":gravel_texture.jpg"

earth=":earth_texture.jpg"

metal=":metal_texture.jpg"

rock=":rock_texture.jpg"

rough=":rough_texture.jpg"

rug=":rug_texture.jpg"

stones=":stones_texture.jpg"

stucco=":stucco_texture.jpg"

wood=":wood_texture.jpg"

wood_old=":wood_old_texture.jpg"

class bumpmaps(object):

gravel=":gravel_bumpmap.jpg"

rock=":rock_bumpmap.jpg"

stones=":stones_bumpmap.jpg"

stucco=":stucco_bumpmap.jpg"

wood_old=":wood_old_bumpmap.jpg"

class standardAttributes(baseObj):

# vector-no-interactions, vector-interactions, scalar-no-interactions, scalar-interactions

attrLists = {'box':[['pos', 'color', 'trail_color'],

['axis', 'size', 'up'],

['visible', 'opacity','shininess', 'emissive',

'make_trail', 'trail_type', 'interval',

'retain', 'trail_color', 'trail_radius', 'texture', 'pickable'],

['red', 'green', 'blue','length', 'width', 'height']],

'sphere':[['pos', 'color', 'trail_color'],

['axis', 'size', 'up'],

['visible', 'opacity','shininess', 'emissive',

'make_trail', 'trail_type', 'interval',

'retain', 'trail_color', 'trail_radius', 'texture', 'pickable'],

['red', 'green', 'blue','length', 'width', 'height', 'radius']],

'simple_sphere':[['pos', 'color', 'trail_color'],

['axis', 'size', 'up'],

['visible', 'opacity','shininess', 'emissive',

'make_trail', 'trail_type', 'interval',

'retain', 'trail_color', 'trail_radius', 'texture', 'pickable'],

['red', 'green', 'blue','length', 'width', 'height', 'radius']],

'arrow':[['pos', 'color', 'trail_color'],

['axis', 'size', 'up'],

['visible', 'opacity',

'shininess', 'emissive', 'texture', 'frame', 'material',

'make_trail', 'trail_type', 'interval',

'retain', 'trail_color', 'trail_radius', 'texture',

'round', 'shaftwidth', 'headwidth', 'headlength', 'pickable'],

['red', 'green', 'blue','length', 'width', 'height']],

'ring':[['pos', 'color', 'trail_color', 'size'],

['axis', 'up'],

['visible', 'opacity','shininess', 'emissive',

'make_trail', 'trail_type', 'interval',

'retain', 'trail_color', 'trail_radius', 'texture', 'pickable'],

['red', 'green', 'blue','length', 'width', 'height', 'thickness']],

'helix':[['pos', 'color', 'trail_color'],

['axis', 'size', 'up'],

['visible', 'opacity','shininess', 'emissive',

'make_trail', 'trail_type', 'interval',

'retain', 'trail_color', 'trail_radius', 'coils', 'thickness', 'pickable'],

['red', 'green', 'blue','length', 'width', 'height', 'radius']],

'curve':[['origin', 'color'],

['axis', 'size', 'up'],

['visible', 'shininess', 'emissive', 'radius', 'retain', 'pickable'],

['red', 'green', 'blue','length', 'width', 'height']],

'points':[['color'],

[],

['visible', 'opacity', 'shininess', 'emissive', 'radius', 'retain', 'pickable', 'size_units'],

['red', 'green', 'blue']],

'label':[['pos', 'color', 'background', 'linecolor'],

[],

['visible', 'xoffset', 'yoffset', 'font', 'height', 'opacity',

'border', 'line', 'box', 'space', 'align', 'linewidth', 'pixel_pos'],

['text']],

'local_light':[['pos', 'color', 'offset'],

[],

['visible', 'attach_idx'],

[]],

'distant_light':[['direction', 'color'],

[],

['visible'],

[]],

'compound':[['pos', 'color', 'trail_color'],

['axis', 'size', 'up', 'origin'],

['visible', 'opacity','shininess', 'emissive',

'make_trail', 'trail_type', 'interval', 'texture',

'retain', 'trail_color', 'trail_radius', 'obj_idxs', 'pickable'],

['red', 'green', 'blue', 'length', 'width', 'height']],

'vertex':[['pos', 'color', 'normal', 'bumpaxis', 'texpos'],

[],

['visible', 'opacity','shininess', 'emissive'],

['red', 'green', 'blue']],

'triangle': [ [],

[],

['texture', 'bumpmap', 'visible', 'pickable'],

['v0', 'v1', 'v2'] ],

'quad': [ [],

[],

['texture', 'bumpmap', 'visible', 'pickable'],

['v0', 'v1', 'v2', 'v3'] ],

'attach_trail': [ ['color'],

[],

['radius', 'pps', 'retain', 'type', '_obj'],

[] ],

'wtext': [[],

[],

['location', 'text'],

[]],

'extrusion':[ ['pos', 'color', 'start_face_color', 'end_face_color',

'start_normal', 'end_normal'],

[ 'axis', 'size', 'up' ],

['path', 'shape', 'visible', 'opacity','shininess', 'emissive',

'show_start_face', 'show_end_face', 'smooth', 'smooth_joints', 'sharp_joints',

'scale', 'xscale', 'yscale', 'twist',

'make_trail', 'trail_type', 'interval', 'show_start_face', 'show_end_face',

'retain', 'trail_color', 'trail_radius', 'texture', 'pickable' ],

['red', 'green', 'blue','length', 'width', 'height'] ],

'text': [ ['pos', 'up', 'color', 'start_face_color', 'end_face_color', 'axis'],

[],

['visible', 'opacity','shininess', 'emissive',

'make_trail', 'trail_type', 'interval',

'retain', 'trail_color', 'trail_radius', 'obj_idxs', 'pickable',

'align', 'text', 'font', 'billboard', 'show_start_face', 'show_end_face', 'descender',

'vertical_spacing', 'depth', 'length', 'width', 'height'],

['red', 'green', 'blue'] ]

}

attrLists['pyramid'] = attrLists['box']

attrLists['cylinder'] = attrLists['sphere']

attrLists['cone'] = attrLists['sphere']

attrLists['ellipsoid'] = attrLists['sphere']

def setup(self, args):

super(standardAttributes, self).__init__()

self._constructing = True ## calls to setters are from constructor

objName = self._objName = args['_objName'] ## identifies object type

if objName[:8] == 'compound': objName = 'compound'

del args['_objName']

# default values

self._sizing = True # axis/size connection is the default; False for sphere, ring, text, compound

self._pos = vector(0,0,0)

self._axis = vector(1,0,0)

self._up = vector(0,1,0)

self._color = vector(1,1,1)

defaultSize = args['_default_size']

if defaultSize is not None: # is not points or vertex or triangle or quad

self._size = defaultSize

del args['_default_size']

self._texture = None

self._opacity = 1.0

self._shininess = 0.6

self._emissive = False

self._make_trail = False

self._trail_type = 'curve'

self._trail_color = self._color

self._interval = 0 # means no interval set

self._retain = -1

self._trail_radius = None # set by default after size set

self._canvas = None

if 'canvas' in args: ## converted to idx below

self._canvas = args['canvas']

self._visible = True

self._frame = None

self._size_units = 'pixels'

self._texture = None

self._pickable = True

self._offset = vector(0,0,0)

self._attach_idx = None

self._save_oldaxis = None # used in linking axis and up

self._save_oldup = None # used in linking axis and up

_special_clone = None

if '_cloneid' in args: # text, extrusion, or compound is being cloned

_special_clone = args['_cloneid']

del args['_cloneid']

argsToSend = [] ## send to GlowScript only attributes specified in constructor

## override defaults for vector attributes without side effects

attrs = standardAttributes.attrLists[objName][0]

for a in attrs:

if a in args:

argsToSend.append(a)

val = args[a]

if objName == 'extrusion' and a == 'color':

setattr(self, '_'+a, val) # '_' bypasses setters

else:

if isinstance(val, vector): setattr(self, '_'+a, vector(val)) # '_' bypasses setters; copy of val

else: raise AttributeError(a+' must be a vector')

del args[a]

# Track side effects of modifying size, axis, or up

# For consistency with GlowScript, axis is listed before up in the attrLists,

# so that setting axis may affect up, but then setting up can affect axis afterwards.

attrs = standardAttributes.attrLists[objName][1] # vector attributes with interactions

for a in attrs:

if a in args:

val = args[a]

if isinstance(val, vector):

setattr(self, a, vector(val))

if a not in argsToSend:

argsToSend.append(a)

if a == 'size':

self._axis = self._axis.norm()*val.x

elif a == 'axis':

self._size.x = mag(val)

self.axis = val # this will have the side effect of modifying up

elif a == 'up':

self.up = val # this will have the side effect of modifying axis

else: raise AttributeError(a+' must be a vector')

del args[a]

if defaultSize is not None:

self._trail_radius = 0.1 * self._size.y ## default depends on size

elif objName == 'points':

self._trail_radius = self._radius # points object

# override defaults for scalar attributes without side effects

attrs = standardAttributes.attrLists[objName][2]

for a in attrs:

if a in args:

argsToSend.append(a)

setattr(self, '_'+a, args[a]) ## by-pass setters

del args[a]

scalarInteractions={'red':'color', 'green':'color', 'blue':'color', 'radius':'size', 'thickness':'size',

'length':'size', 'height':'size', 'width':'size', 'v0':'v0', 'v1':'v1',

'v2':'v2', 'v3':'v3', 'text':'text'}

# override defaults for scalar attributes with side effects

attrs = standardAttributes.attrLists[objName][3]

for a in attrs:

if a in args:

setattr(self, a, args[a]) ## use setter to take care of side effects

if scalarInteractions[a] not in argsToSend:

argsToSend.append(scalarInteractions[a]) # e.g. if a is radius, send size

# The following makes sure that size.x is nonzero.

# It will be reset when axis is no longer zero.

if defaultSize is not None and self._size._x == 0: self._size._x = 1

del args[a]

# set values of user-defined attributes

for key, value in args.items(): # Assign all other properties

setattr(self, key, value)

cmd = {"cmd": objName, "idx": self.idx}

# now put all args to send into cmd

nosize = ['compound', 'extrusion', 'text'] # size will be computed by the constuctor

for a in argsToSend:

aval = getattr(self,a)

if isinstance(aval, vector):

aval = aval.value

elif isinstance(aval, vertex):

aval = aval.idx

if objName in nosize and a == 'size': continue # do not send superfluous size

cmd[a] = aval

# set canvas

if self.canvas is None: ## not specified in constructor

self.canvas = canvas.get_selected()

#cmd["attrs"].append({"attr": 'canvas', "value": self.canvas.idx})

cmd['canvas'] = self.canvas.idx

# Lights are listed in canvas.lights, not canvas.objects

if not (objName == 'distant_light' or objName == 'local_light'):

self.canvas.objz(self,'add')

self._constructing = False ## from now on any setter call will not be from constructor

# _special_clone is True if text, extrusion, or compound objects

if _special_clone is not None: cmd["_cloneid"] = _special_clone

self.appendcmd(cmd)

# attribute vectors have these methods which call self.addattr()

# The vector class calls a change function when there's a change in x, y, or z.

noSize = ['points', 'label', 'vertex', 'triangle', 'quad', 'attach_trail']

if not (objName == 'extrusion'): #

self._color.on_change = self._on_color_change

if objName not in noSize:

self._axis.on_change = self._on_axis_change

self._size.on_change = self._on_size_change

self._up.on_change = self._on_up_change

noPos = ['curve', 'points', 'triangle', 'quad']

if objName not in noPos:

self._pos.on_change = self._on_pos_change

elif objName == 'curve':

self._origin.on_change = self._on_origin_change

if objName == 'vertex':

self._bumpaxis.on_change = self._on_bumpaxis_change

self._normal.on_change = self._on_normal_change

# Ensure that if axis or up is <0,0,0> in constructor, we'll recover eventually:

if self._axis.mag2 == 0: self._save_oldaxis = vector(1,0,0)

if self._up.mag2 == 0: self._save_oldup = vector(0,1,0)

@property

def pos(self):

return self._pos

@pos.setter

def pos(self,value):

self._pos.value = value

if not self._constructing:

if self._make_trail and self._interval > 0:

self.addmethod('pos', value.value)

else:

self.addattr('pos')

@property

def up(self):

return self._up

@up.setter

def up(self,value):

self._save_oldup = adjust_axis(self._up, value, self._axis, self._save_oldup) # this sets self._axis and self._up

if not self._constructing:

self.addattr('up')

@property

def axis(self):

return self._axis

@axis.setter

def axis(self,value): # sphere or ring or text or compound have no axis/size link

if value.mag2 == 0:

if self._save_oldaxis is None: self._save_oldaxis = self._axis

self._axis = value

else:

self._save_oldaxis = adjust_up(self._axis, value, self._up, self._save_oldaxis) # this sets self._axis and self._up

if not self._constructing:

self.addattr('axis')

if self._sizing:

self._size._x = value.mag # changing axis length changes size.x

@property

def size(self):

return self._size

@size.setter

def size(self,value): # sphere or ring or text or compound have no axis/size link

currentaxis = self._axis

self._size = value

if value.x == 0:

if self._save_oldaxis is not None:

currentaxis = self._save_oldaxis

self._save_oldaxis = None

else:

currentaxis = vector(1,0,0)

if not self._constructing:

self.addattr('size')

if self._sizing:

self._axis = currentaxis.norm()*value.x

@property

def length(self):

return self._size.x

@length.setter

def length(self,value):

if value == 0:

if self._save_oldaxis is None: self._save_oldaxis = vector(self._axis.x, self._axis.y, self._axis.z)

self._axis = vector(0,0,0)

self._size._x = 0

else:

if self._save_oldaxis is not None:

self._axis = self._save_oldaxis

self._save_oldaxis = None

if self._size._x == 0:

self.axis = vector(value, 0, 0)

else:

if self._sizing:

self.axis = value*self._axis.norm() # this will set length if self._sizing

else:

self._size._x = value # for objects whose axis and size are not linked

if not self._constructing:

self.addattr('axis')

self.addattr('size')

@property

def height(self):

return self._size.y

@height.setter

def height(self,value):

self._size._y = value

if not self._constructing:

self.addattr('size')

@property

def width(self):

return self._size.z

@width.setter

def width(self,value):

self._size._z = value

if not self._constructing:

self.addattr('size')

@property

def color(self):

return self._color

@color.setter

def color(self,value):

if isinstance(self._color, list): # may be a list of extrusion colors of the form [ [1,0,0], [0,1,0] ]

self._color = vector(0,0,0) # change list to a vector

self._color.value = value # modify the vector

self._color.value = value

if not self._constructing:

self.addattr('color')

@property

def red(self):

return self._color.x

@red.setter

def red(self,value):

self._color = (value,self.green,self.blue)

if not self._constructing:

self.addattr('color')

@property

def green(self):

return self._color.y

@green.setter

def green(self,value):

self._color = (self.red,value,self.blue)

if not self._constructing:

self.addattr('color')

@property

def blue(self):

return self._color.z

@blue.setter

def blue(self,value):

self._color = (self.red,self.green,value)

if not self._constructing:

self.addattr('color')

@property

def visible(self):

return self._visible

@visible.setter

def visible(self,value):

self._visible = value

if not self._constructing:

self.addattr('visible')

@property

def pickable(self):

return self._pickable

@pickable.setter

def pickable(self,value):

self._pickable = value

if not self._constructing:

self.addattr('pickable')

@property

def canvas(self):

return self._canvas

@canvas.setter

def canvas(self,value):

self._canvas = value

if not self._constructing:

raise AttributeError('canvas cannot be modified')

@property

def opacity(self):

return self._opacity

@opacity.setter

def opacity(self,value):

self._opacity = value

if not self._constructing:

self.addattr('opacity')

@property

def emissive(self):

return self._emissive

@emissive.setter

def emissive(self,value):

self._emissive = value

if not self._constructing:

self.addattr('emissive')

@property

def texture(self):

return self._texture

@texture.setter

def texture(self,value):

self._texture = value

if not self._constructing:

self.appendcmd({"texture":value})

@property

def shininess(self):

return self._shininess

@shininess.setter

def shininess(self,value):

self._shininess = value

if not self._constructing:

self.addattr('shininess')

@property

def make_trail(self):

return self._make_trail

@make_trail.setter

def make_trail(self,value):

self._make_trail = value

if not self._constructing:

self.addattr('make_trail')

@property

def trail_type(self):

return self._trail_type

@trail_type.setter

def trail_type(self,value):

if (value not in ['curve', 'points']):

raise Exception("ArgumentError: trail_type must be 'curve' or 'points'")

self._trail_type = value

if not self._constructing:

self.addattr('trail_type')

@property

def trail_color(self):

return self._trail_color

@trail_color.setter

def trail_color(self, value):

if isinstance(value, vector):

self._trail_color.value = value

else:

raise TypeError('trail_color must be a vector')

if not self._constructing:

self.addattr('trail_color')

@property

def interval(self):