class BeeToolConfigParser:
def __init__(self,device):
self.xml=QXmlStreamReader()
self.xml.setNamespaceProcessing(False)
self.xml.setDevice(device)
self.curtool=None
def loadToToolBox(self,toolbox):
self.toolbox=toolbox
while not self.xml.atEnd():
tokentype=self.xml.readNext()
if tokentype==QXmlStreamReader.StartElement:
self.processStartElement()
def processStartElement(self):
name=self.xml.name()
attrs=self.xml.attributes()
if name=="toolconfig":
toolname="%s" % attrs.value('name').toString()
self.curtool=self.toolbox.getToolDescByName(toolname)
elif name=="option":
if self.curtool:
valname="%s" % attrs.value('name').toString()
(value,ok)=attrs.value('value').toString().toInt()
self.curtool.options[valname]=value
def __init__(self,device):
self.xml=QXmlStreamReader()
self.xml.setNamespaceProcessing(False)
self.xml.setDevice(device)
self.swatchsize=15
self.colors=[[]]
class PaletteParser:
def __init__(self,device):
self.xml=QXmlStreamReader()
self.xml.setNamespaceProcessing(False)
self.xml.setDevice(device)
self.swatchsize=15
self.colors=[[]]
def getColors(self):
while not self.xml.atEnd():
tokentype=self.xml.readNext()
if tokentype==QXmlStreamReader.StartElement:
self.processStartElement()
return self.colors
def processStartElement(self):
name=self.xml.name()
attrs=self.xml.attributes()
if name == "beepalette":
(self.rows,ok)=attrs.value('rows').toString().toInt()
(self.columns,ok)=attrs.value('columns').toString().toInt()
if attrs.value('swatchsize'):
(self.swatchsize,ok)=attrs.value('swatchsize').toString().toInt()
elif name == "color":
(r,ok)=attrs.value('r').toString().toInt()
(g,ok)=attrs.value('g').toString().toInt()
(b,ok)=attrs.value('b').toString().toInt()
if len(self.colors[-1]) >= self.columns:
self.colors.append([])
self.colors[-1].append((r,g,b))
def __init__(self,device,window,stepdelay,type=ThreadTypes.animation,id=0):
self.xml=QXmlStreamReader()
#turn off namespace processing
self.xml.setNamespaceProcessing(False)
if device:
self.xml.setDevice(device)
self.id=id
self.window=window
self.type=type
self.stepdelay=stepdelay
self.keymap={}
self.imagestarted=False
self.imagedata=None
if type==ThreadTypes.animation:
self.layertype=LayerTypes.animation
else:
self.layertype=LayerTypes.network
def __init__(self,device): self.xml=QXmlStreamReader() self.xml.setNamespaceProcessing(False) self.xml.setDevice(device) self.curtool=None
def __init__(self, path):
QXmlStreamReader.__init__(self, open(path).read())
def __init__(self, path):
QXmlStreamReader.__init__(self, open(path).read())
class XmlToQueueEventsConverter:
""" Represents a parser to to turn an incomming xml stream into drawing events
"""
def __init__(self,device,window,stepdelay,type=ThreadTypes.animation,id=0):
self.xml=QXmlStreamReader()
#turn off namespace processing
self.xml.setNamespaceProcessing(False)
if device:
self.xml.setDevice(device)
self.id=id
self.window=window
self.type=type
self.stepdelay=stepdelay
self.keymap={}
self.imagestarted=False
self.imagedata=None
self.image=None
if type==ThreadTypes.animation:
self.layertype=LayerTypes.animation
else:
self.layertype=LayerTypes.network
def translateKey(self,key):
""" only needs to do something in animation threads
"""
return key
def addKeyTranslation(self,key,dockey):
""" only needs to do something in animation threads
"""
pass
def read(self):
""" Read tokens in the xml document until the end or until an error occurs, this function serves as a switchboard to call other functions based on the type of token
"""
while not self.xml.atEnd():
tokentype=self.xml.readNext()
if tokentype==QXmlStreamReader.StartElement:
self.processStartElement()
elif tokentype==QXmlStreamReader.EndElement:
self.processEndElement()
elif tokentype==QXmlStreamReader.Characters:
self.processCharacterData()
# if it's an error that might actually be a problem then print it out
if self.xml.hasError() and self.xml.error() != QXmlStreamReader.PrematureEndOfDocumentError:
print_debug("error while parsing XML: %s" % self.xml.errorString())
return self.xml.error()
def processStartElement(self):
""" Handle any type of starting XML tag and turn it into a drawing event if needed
"""
type=self.type
name=self.xml.name()
attrs=self.xml.attributes()
if name == 'createdoc':
(width,ok)=attrs.value('width').toString().toInt()
(height,ok)=attrs.value('height').toString().toInt()
self.window.addSetCanvasSizeRequestToQueue(width,height,type)
elif name == 'addlayer':
if self.type==ThreadTypes.server:
# create our own key data in this case
key=self.window.nextLayerKey()
else:
(key,ok)=attrs.value("key").toString().toInt()
(pos,ok)=attrs.value("position").toString().toInt()
# if this is the server don't trust the client to give the right ID, instead pull it from the ID given to this thread
if self.type==ThreadTypes.server:
owner=self.id
# otherwise trust the ID in the message
else:
(owner,ok)=attrs.value("owner").toString().toInt()
if self.type!=ThreadTypes.animation:
dockey=key
# if it's an animation I need to map the key to a local one
else:
dockey=self.window.nextLayerKey()
self.addKeyTranslation(key,dockey)
self.window.addInsertLayerEventToQueue(pos,dockey,self.image,self.type,owner=owner)
elif name == 'sublayer':
(key,ok)=attrs.value("index").toString().toInt()
self.window.addRemoveLayerRequestToQueue(key,type)
elif name == 'movelayer':
(change,ok)=attrs.value("change").toString().toInt()
(index,ok)=attrs.value("index").toString().toInt()
if change==1:
self.window.addLayerUpToQueue(index,type)
else:
self.window.addLayerDownToQueue(index,type)
elif name == 'layeralpha':
(key,ok)=attrs.value("key").toString().toInt()
(opacity,ok)=attrs.value("alpha").toString().toFloat()
self.window.addOpacityChangeToQueue(key,opacity,type)
self.window.addOpacityDoneToQueue(key,type)
elif name == 'layermode':
time.sleep(self.stepdelay)
(key,ok)=attrs.value('key').toString().toInt()
mode=BlendTranslations.nameToMode(attrs.value('mode').toString())
self.window.addBlendModeChangeToQueue(self.translateKey(key),mode,type)
elif name == 'undo':
(owner,ok)=attrs.value('owner').toString().toInt()
self.window.addUndoToQueue(owner,type)
elif name == 'redo':
(owner,ok)=attrs.value('owner').toString().toInt()
self.window.addRedoToQueue(owner,type)
elif name == 'event':
self.clippath=None
self.image=None
elif name == 'toolevent':
self.image=None
self.strokestart=False
toolname="%s" % attrs.value('name').toString()
(layerkey,ok)=attrs.value('layerkey').toString().toInt()
(owner,ok)=attrs.value('owner').toString().toInt()
self.curlayer=self.translateKey(layerkey)
tool=self.window.master.getToolClassByName(toolname.strip())
#print error if we can't find the tool
if tool == None:
print_debug("Error, couldn't find tool with name: %s" % toolname)
return
self.curtool=tool.setupTool(self.window,self.curlayer)
self.curtool.clippath=self.clippath
self.curtool.layerkey=self.curlayer
self.curtool.owner=owner
elif name == 'fgcolor':
(r,ok)=attrs.value('r').toString().toInt()
(g,ok)=attrs.value('g').toString().toInt()
(b,ok)=attrs.value('b').toString().toInt()
self.curtool.fgcolor=qtgui.QColor(r,g,b)
elif name == 'bgcolor':
(r,ok)=attrs.value('r').toString().toInt()
(g,ok)=attrs.value('g').toString().toInt()
(b,ok)=attrs.value('b').toString().toInt()
self.curtool.bgcolor=qtgui.QColor(r,g,b)
elif name == 'clippath':
self.clippoints=[]
elif name == 'polypoint':
(x,ok)=attrs.value('x').toString().toInt()
(y,ok)=attrs.value('y').toString().toInt()
self.clippoints.append(qtcore.QPointF(x,y))
elif name == 'toolparam':
(value,ok)=attrs.value('value').toString().toInt()
self.curtool.setOption("%s" % attrs.value('name').toString(),value)
elif name == 'image':
self.imagestarted=True
elif name == 'rawevent':
self.raweventargs=[]
xstr=attrs.value('x').toString()
(self.x,ok)=xstr.toFloat()
(self.y,ok)=attrs.value('y').toString().toFloat()
(layerkey,ok)=attrs.value('layerkey').toString().toInt()
mode=qtgui.QPainter.CompositionMode_SourceOver
if attrs.value('compmode'):
modename=attrs.value('compmode')
mode=BlendTranslations.nameToMode(modename.toString())
if not mode:
mode=qtgui.QPainter.CompositionMode_SourceOver
self.layerkey=self.translateKey(layerkey)
if self.image:
self.window.addRawEventToQueue(self.layerkey,self.image,self.x,self.y,self.clippath,source=type,compmode=mode)
elif name == 'pointslist':
if self.strokestart==True:
self.window.addPenLeaveToQueue(self.curlayer,type)
self.window.addPenEnterToQueue(self.curlayer,type)
elif name == 'point':
time.sleep(self.stepdelay)
(x,ok)=attrs.value('x').toString().toFloat()
(y,ok)=attrs.value('y').toString().toFloat()
self.lastx=x
self.lasty=y
(pressure,ok)=attrs.value('pressure').toString().toFloat()
if self.strokestart == False:
self.window.addPenDownToQueue(x,y,pressure,self.curlayer,self.curtool,type)
self.strokestart=True
else:
self.window.addPenMotionToQueue(x,y,pressure,self.curlayer,type)
elif name == 'resyncrequest':
self.window.addResyncRequestToQueue(self.id)
elif name == 'resyncstart':
(width,ok)=attrs.value('width').toString().toInt()
(height,ok)=attrs.value('height').toString().toInt()
(remoteid,ok)=attrs.value('remoteid').toString().toInt()
self.window.addResyncStartToQueue(remoteid,width,height)
elif name == 'giveuplayer':
(layerkey,ok)=attrs.value('key').toString().toInt()
# make sure command is legit from this source
layer=self.window.getLayerForKey(layerkey)
# make sure layer is there
if layer:
proplock=qtcore.QReadLocker(layer.propertieslock)
if layer.owner!=self.id:
print_debug("ERROR: got bad give up layer command from client: %d for layer key: %d" % (self.id,layerkey))
else:
self.window.addGiveUpLayerToQueue(layerkey,self.id,type)
elif name == 'changelayerowner':
(layerkey,ok)=attrs.value('key').toString().toInt()
(owner,ok)=attrs.value('owner').toString().toInt()
self.window.addChangeLayerOwnerToQueue(layerkey,owner,type)
elif name == 'layerrequest':
(layerkey,ok)=attrs.value('key').toString().toInt()
self.window.addLayerRequestToQueue(layerkey,self.id,type)
elif name == 'fatalerror':
errormessage="%s" % attrs.value('errormessage').toString()
self.window.addFatalErrorNotificationToQueue(0,errormessage,type)
elif name == 'sketchlog':
print_debug("DEBUG: got document start tag")
elif name == 'networkhistorysize':
(newsize,ok)=attrs.value('newsize').toString().toInt()
self.window.setNetworkHistorySize(newsize)
elif name == 'null':
pass
else:
print_debug("WARNING: Don't know how to handle tag: %s" % name.toString())
def processEndElement(self):
name=self.xml.name()
if name == 'toolevent':
#print_debug("Adding end tool event to queue on layer %d" % self.curlayer)
self.window.addPenUpToQueue(self.lastx,self.lasty,self.curlayer,type)
self.curtool=None
elif name == 'clippath':
poly=qtgui.QPolygonF(self.clippoints)
self.clippath=qtgui.QPainterPath()
self.clippath.addPolygon(poly)
def processCharacterData(self):
if self.imagestarted:
#print_debug("processing image data")
self.imagestarted=False
rawstring=self.xml.text().toString()
data=qtcore.QByteArray()
data=data.append(rawstring)
data=qtcore.QByteArray.fromBase64(data)
data=qtcore.qUncompress(data)
self.image=qtgui.QImage.fromData(data,"PNG")