^jdZddlZddlZddlZddlmZddlmZmZm Z ddl m Z ddl m Z mZmZmZmZmZmZmZmZddlmZded edzfd Zd ed efd Zd e d edzfdZd ed e fdZded efdZd e d efdZ dZ!y)a build123d pickle support name: persistence.py by: Jojain & bernhard-42 date: September 8th, 2023 desc: This python module enables build123d objects to be pickled. license: Copyright 2023 Jojain & bernhard-42 Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. You may obtain a copy of the License at http://www.apache.org/licenses/LICENSE-2.0 Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions and limitations under the License. N)BinTools) gp_Quaterniongp_Trsfgp_Vec)TopLoc_Location) TopoDS_CompoundTopoDS_CompSolid TopoDS_Edge TopoDS_Face TopoDS_Shape TopoDS_Shell TopoDS_Solid TopoDS_Vertex TopoDS_Wire)downcastshapereturnc|ytj}tj|||j }|S)zj Serialize a OCP shape, this method can be used to provide a custom serialization algo for pickle N)ioBytesIOrWrite_sgetvalue)rbiobuffers N/opt/ringagent/.cad-venv/lib/python3.12/site-packages/build123d/persistence.pyserialize_shaper6s7 } **,C UC \\^F Mrc|yt}tj|}tj||t |S)zV This does the opposite as serialize, it construct a TopoDS_Shape from bytes. N)r rrrRead_sr)rrrs rdeserialize_shaper Cs9~ NE **V C OOE3 E?rlocationc |y|j}|j}|j}t}t ddD]6}|j t jd|j|8t}|j t jd|j|j t jd|j|j t jd|j|j t jd|jt}|j ||j ||S)zq Serialize a OCP location, this method can be used to provide a custom serialization algo for pickle Nf) TransformationTranslationPart GetRotation bytearrayrangeextendstructpackCoordXYZW)r! transform translationrotationtranslation_bytesirotation_bytesrs rserialize_locationr9Ps" '')I++-K$$&H!  1a[I  S+2C2CA2F!GHI[N&++c8::<89&++c8::<89&++c8::<89&++c8::<89 [F MM#$ MM.! Mrcr|yttjd|}tt |dt |dt |d}t t |dt |dt |dt |d}t }|j||t|S)zY This does the opposite as serialize, it construct a TopLoc_Location from bytes. Nr%r) iterr, iter_unpackrnextrrSetTransformationr)rchunksr4r5r3s rdeserialize_locationr@js~&$$S&1 2F V Q V Q V QK V Q V Q V Q V Q H I +6 9 %%rc&tt|ffS)zNSpecial function used by pickle to serialize or deserialize OCP Shapes objects)r r)rs r reduce_shaperBs  68 99rc&tt|ffS)zPSpecial function used by pickle to serialize or deserialize OCP Location objects)r@r9)r!s rreduce_locationrDs #5h#?"A BBrc\tjtttjtttjt ttjt ttjtttjtttjtttjtttjtttjtty)zg Modify the copyreg so that pickle knows what to look for when it tries to pickle an OCP Shape N)copyregpickler rBrr rr r rr rrrDrrmodify_copyregrIs NN<. NN?L1 NN#\2 NN<. NN<. NN; - NN; - NN; - NN=,/ NN?O4r)"__doc__rFrr, OCP.BinToolsrOCP.gprrr OCP.TopLocr OCP.TopoDSrr r r r r rrrbuild123d.topologyrbytesrr r)r9r@tuplerBrDrIrHrrrRs> !11&   ( < EDL  e  Y5E4&&?&<: :: CoC%C 5r