def encode(self): """Encode to binary data.""" ioBuf = IO(boolSize=32) ioBuf.u32 = self.pointType ioBuf.float32 = self.x ioBuf.float32 = self.y if self.pressure is not None: ioBuf.float32 = self.pressure if self.xTilt is not None: ioBuf.float32 = self.xTilt if self.yTilt is not None: ioBuf.float32 = self.yTilt if self.wheel is not None: ioBuf.float32 = self.wheel return ioBuf.data
def encode(self): """Convert this object to raw bytes.""" ioBuf = IO() ioBuf.float32 = self.factor ioBuf.u32 = self.numDigits ioBuf.sz754 = self.id ioBuf.sz754 = self.symbol ioBuf.sz754 = self.abbrev ioBuf.sz754 = self.sname ioBuf.sz754 = self.pname return ioBuf.data
def _propertyEncode(self, propertyType): """Encode a single property. Many properties are in the form propertyType: one of PROP_ lengthOfData: 4 data: varies but is often ioBuf.32 or ioBuf.boolean If the property is the same as the default, or not specified, returns empty array """ ioBuf = IO(boolSize=32) if propertyType == self.PROP_COLORMAP: if self.colorMap is not None and self.colorMap: ioBuf.u32 = self.PROP_COLORMAP # ioBuf.addBytes(self._colormapEncode_()) elif propertyType == self.PROP_ACTIVE_LAYER: if self.selected is not None and self.selected: ioBuf.u32 = self.PROP_ACTIVE_LAYER elif propertyType == self.PROP_ACTIVE_CHANNEL: if self.selected is not None and self.selected: ioBuf.u32 = self.PROP_ACTIVE_LAYER elif propertyType == self.PROP_SELECTION: if self.isSelection is not None and self.isSelection: ioBuf.u32 = self.PROP_SELECTION elif propertyType == self.PROP_FLOATING_SELECTION: if self.selectionAttachedTo is not None: ioBuf.u32 = self.PROP_FLOATING_SELECTION ioBuf.u32 = self.selectionAttachedTo elif propertyType == self.PROP_OPACITY: if self.opacity is not None and not isinstance(self.opacity, float): ioBuf.u32 = self.PROP_OPACITY ioBuf.u32 = self.opacity elif propertyType == self.PROP_MODE: if self.blendMode is not None: ioBuf.u32 = self.PROP_MODE ioBuf.u32 = self.blendMode elif propertyType == self.PROP_VISIBLE: if self.visible is not None: ioBuf.u32 = self.PROP_VISIBLE ioBuf.boolean = self.visible elif propertyType == self.PROP_LINKED: if self.isLinked is not None and self.isLinked: ioBuf.u32 = self.PROP_LINKED ioBuf.boolean = self.isLinked elif propertyType == self.PROP_LOCK_ALPHA: if self.lockAlpha is not None and self.lockAlpha: ioBuf.u32 = self.PROP_LOCK_ALPHA ioBuf.boolean = self.lockAlpha elif propertyType == self.PROP_APPLY_MASK: if self.applyMask is not None: ioBuf.u32 = self.PROP_APPLY_MASK ioBuf.boolean = self.applyMask elif propertyType == self.PROP_EDIT_MASK: if self.editingMask is not None and self.editingMask: ioBuf.u32 = self.PROP_EDIT_MASK ioBuf.boolean = self.editingMask elif propertyType == self.PROP_SHOW_MASK: if self.showMask is not None and self.showMask: ioBuf.u32 = self.PROP_SHOW_MASK ioBuf.boolean = self.showMask elif propertyType == self.PROP_SHOW_MASKED: if self.showMasked is not None: ioBuf.u32 = self.PROP_SHOW_MASKED ioBuf.boolean = self.showMasked elif propertyType == self.PROP_OFFSETS: if self.xOffset is not None and self.yOffset is not None: ioBuf.u32 = self.PROP_OFFSETS ioBuf.i32 = self.xOffset ioBuf.i32 = self.yOffset elif propertyType == self.PROP_COLOR: if ( self.color is not None and not isinstance(self.color[0], float) and not isinstance(self.color[1], float) and not isinstance(self.color[2], float) ): ioBuf.u32 = self.PROP_COLOR ioBuf.byte = self.color[0] ioBuf.byte = self.color[1] ioBuf.byte = self.color[2] elif propertyType == self.PROP_COMPRESSION: if self.compression is not None: ioBuf.u32 = self.PROP_COMPRESSION ioBuf.u32 = self.compression elif propertyType == self.PROP_GUIDES: if self.guidelines is not None and self.guidelines: pass # ioBuf.u32 = self.PROP_GUIDES # ioBuf.addBytes(self._guidelinesEncode_()) elif propertyType == self.PROP_RESOLUTION: if self.horizontalResolution is not None and self.verticalResolution is not None: ioBuf.u32 = self.PROP_RESOLUTION ioBuf.u32 = int(self.horizontalResolution) ioBuf.i32 = int(self.verticalResolution) elif propertyType == self.PROP_TATTOO: if self.uniqueId is not None: ioBuf.u32 = int(self.uniqueId, 16) elif propertyType == self.PROP_PARASITES: if self.parasites is not None and self.parasites: ioBuf.u32 = self.PROP_PARASITES ioBuf.addBytes(self._parasitesEncode()) elif propertyType == self.PROP_UNIT: if self.units is not None: ioBuf.u32 = self.PROP_UNIT ioBuf.u32 = self.units elif propertyType == self.PROP_PATHS: if self.paths is not None and self.paths: # ioBuf.u32 = self.PROP_PATHS # ioBuf.u32 = len(self.paths) """ for path in self.paths: ioBuf.append(self._pathEncode_(path)) """ elif propertyType == self.PROP_USER_UNIT: if self.userUnits is not None: pass # ioBuf.u32 = self.PROP_USER_UNIT # ioBuf.addBytes(self._userUnitsEncode_()) elif propertyType == self.PROP_VECTORS: if self.vectors is not None and self.vectors: pass # ioBuf.u32 = self.PROP_VECTORS # ioBuf.addBytes(self._vectorsEncode_()) elif propertyType == self.PROP_TEXT_LAYER_FLAGS: if self.textLayerFlags is not None: ioBuf.u32 = self.PROP_TEXT_LAYER_FLAGS ioBuf.u32 = self.textLayerFlags elif propertyType == self.PROP_OLD_SAMPLE_POINTS: pass elif propertyType == self.PROP_LOCK_CONTENT: if self.locked is not None and self.locked: ioBuf.u32 = self.PROP_LOCK_CONTENT ioBuf.boolean = self.locked elif propertyType == self.PROP_GROUP_ITEM: if self.isGroup is not None and self.isGroup: ioBuf.u32 = self.PROP_GROUP_ITEM elif propertyType == self.PROP_ITEM_PATH: if self.itemPath is not None: pass # ioBuf.u32 = self.PROP_ITEM_PATH # ioBuf.addBytes(self._itemPathEncode_()) elif propertyType == self.PROP_GROUP_ITEM_FLAGS: if self.groupItemFlags is not None: ioBuf.u32 = self.PROP_GROUP_ITEM_FLAGS ioBuf.u32 = self.groupItemFlags elif propertyType == self.PROP_LOCK_POSITION: if self.positionLocked is not None and self.positionLocked: ioBuf.u32 = self.PROP_LOCK_POSITION ioBuf.boolean = self.positionLocked elif propertyType == self.PROP_FLOAT_OPACITY: if self.opacity is not None and isinstance(self.opacity, float): ioBuf.u32 = self.PROP_FLOAT_OPACITY ioBuf.float32 = self.opacity elif propertyType == self.PROP_COLOR_TAG: if self.colorTag is not None: ioBuf.u32 = self.PROP_COLOR_TAG ioBuf.u32 = self.colorTag elif propertyType == self.PROP_COMPOSITE_MODE: if self.compositeMode is not None: ioBuf.u32 = self.PROP_COMPOSITE_MODE ioBuf.i32 = self.compositeMode elif propertyType == self.PROP_COMPOSITE_SPACE: if self.compositeSpace is not None: ioBuf.u32 = self.PROP_COMPOSITE_SPACE ioBuf.i32 = self.compositeSpace elif propertyType == self.PROP_BLEND_SPACE: if self.blendSpace is not None: ioBuf.u32 = self.PROP_BLEND_SPACE ioBuf.u32 = self.blendSpace elif propertyType == self.PROP_FLOAT_COLOR: if ( self.color is not None and isinstance(self.color[0], float) and isinstance(self.color[1], float) and isinstance(self.color[2], float) ): ioBuf.u32 = self.PROP_FLOAT_COLOR ioBuf.float32 = self.color[0] ioBuf.float32 = self.color[1] ioBuf.float32 = self.color[2] elif propertyType == self.PROP_SAMPLE_POINTS: if self.samplePoints is not None and self.samplePoints: pass # ioBuf.u32 = self.PROP_SAMPLE_POINTS # self.addBytes(self._samplePointsEncode_()) else: raise RuntimeError(f"Unknown property id {propertyType}") return ioBuf.data