def loadSMPX(cls, smpx): iarch = IArchive(str(smpx)) # because alembic hates unicode try: top = iarch.getTop() par = top.children[0] par = IXform(top, par.getName()) abcMesh = par.children[0] abcMesh = IPolyMesh(par, abcMesh.getName()) systemSchema = par.getSchema() props = systemSchema.getUserProperties() prop = props.getProperty('simplex') jsString = prop.getValue() js = json.loads(jsString) system = cls.loadJSON(js) system._smpx = smpx meshSchema = abcMesh.getSchema() rawFaces = meshSchema.getFaceIndicesProperty().samples[0] rawCounts = meshSchema.getFaceCountsProperty().samples[0] system._faces = [i for i in rawFaces] system._counts = [i for i in rawCounts] for shape, sample in zip( system.shapes, meshSchema.getPositionsProperty().samples): shape.loadSMPX(sample) finally: del iarch return system
def buildFromAbc(self, thing, abcPath, pBar=None): """ Load a system from an exported abc file onto the current system """ if pBar is not None: pBar.show() pBar.setMaximum(100) pBar.setValue(0) pBar.setLabelText("Loading Smpx") QApplication.processEvents() iarch = IArchive(str(abcPath)) # because alembic hates unicode try: top = iarch.getTop() par = top.children[0] par = IXform(top, par.getName()) abcMesh = par.children[0] abcMesh = IPolyMesh(par, abcMesh.getName()) systemSchema = par.getSchema() props = systemSchema.getUserProperties() prop = props.getProperty("simplex") jsString = prop.getValue() js = json.loads(jsString) system = self.buildFromDict(thing, js, True, pBar) self.DCC.loadABC(abcMesh, js, pBar) finally: del iarch return system
def getMesh(iarch): '''Load the static mesh data from an alembic archive Parameters ---------- iarch : IArchive The input alembic archive Returns ------- : imath.IntArray The Faces array : imath.IntArray The Counts array ''' top = iarch.getTop() par = top.children[0] par = IXform(top, par.getName()) abcMesh = par.children[0] abcMesh = IPolyMesh(par, abcMesh.getName()) sch = abcMesh.getSchema() faces = sch.getFaceIndicesProperty().samples[0] counts = sch.getFaceCountsProperty().samples[0] return faces, counts
def loadSmpx(iarch): '''Load the json and shape data from a .smpx file Parameters ---------- iarch : IArchive The input alembic archive Returns ------- : np.array The array of point positions for all the shapes ''' top = iarch.getTop() par = top.children[0] par = IXform(top, par.getName()) abcMesh = par.children[0] abcMesh = IPolyMesh(par, abcMesh.getName()) shapes = getSampleArray(abcMesh) print "Done Loading" return shapes
def exportUnsub(inPath, outPath, newFaces, kept, shapePrefix=None, pBar=None): ''' Export the unsubdivided simplex ''' iarch = IArchive(str(inPath)) # because alembic hates unicode top = iarch.getTop() ixfo = IXform(top, top.children[0].getName()) iprops = ixfo.getSchema().getUserProperties() iprop = iprops.getProperty("simplex") jsString = iprop.getValue() if shapePrefix is not None: d = json.loads(jsString) if d['encodingVersion'] > 1: for shape in d['shapes']: shape['name'] = shapePrefix + shape['name'] else: d['shapes'] = [shapePrefix + i for i in d['shapes']] jsString = json.dumps(d) imesh = IPolyMesh(ixfo, ixfo.children[0].getName()) verts = getSampleArray(imesh) verts = verts[:, kept] indices = [] counts = [] for f in newFaces: indices.extend(f) counts.append(len(f)) abcCounts = mkArray(IntArray, counts) abcIndices = mkArray(IntArray, indices) # `False` for HDF5 `True` for Ogawa oarch = OArchive(str(outPath), False) oxfo = OXform(oarch.getTop(), ixfo.getName()) oprops = oxfo.getSchema().getUserProperties() oprop = OStringProperty(oprops, "simplex") oprop.setValue(str(jsString)) omesh = OPolyMesh(oxfo, imesh.getName()) osch = omesh.getSchema() if pBar is not None: pBar.setValue(0) pBar.setMaximum(len(verts)) pBar.setLabelText("Exporting Unsubdivided Shapes") QApplication.processEvents() for i, v in enumerate(verts): if pBar is not None: pBar.setValue(i) QApplication.processEvents() else: print "Exporting Unsubdivided Shape {0: <4}\r".format(i + 1), sample = OPolyMeshSchemaSample(mkSampleVertexPoints(v), abcIndices, abcCounts) osch.set(sample) if pBar is None: print "Exporting Unsubdivided Shape {0: <4}".format(len(verts))
def _openSmpx(inPath): iarch = IArchive(str(inPath)) # because alembic hates unicode top = iarch.getTop() ixfo = IXform(top, top.children[0].getName()) iprops = ixfo.getSchema().getUserProperties() iprop = iprops.getProperty("simplex") jsString = iprop.getValue() imesh = IPolyMesh(ixfo, ixfo.children[0].getName()) return iarch, imesh, jsString, ixfo.getName(), imesh.getName()
def loadJSString(iarch): top = iarch.getTop() par = top.children[0] par = IXform(top, par.getName()) systemSchema = par.getSchema() props = systemSchema.getUserProperties() prop = props.getProperty("simplex") jsString = prop.getValue() return jsString
def loadSmpx(iarch): ''' Load the json and shape data from a .smpx file ''' top = iarch.getTop() par = top.children[0] par = IXform(top, par.getName()) abcMesh = par.children[0] abcMesh = IPolyMesh(par, abcMesh.getName()) shapes = getSampleArray(abcMesh) print "Done Loading" return shapes
def loadMesh(iarch): top = iarch.getTop() par = top.children[0] par = IXform(top, par.getName()) abcMesh = par.children[0] abcMesh = IPolyMesh(par, abcMesh.getName()) sch = abcMesh.getSchema() faces = sch.getFaceIndicesProperty().samples[0] counts = sch.getFaceCountsProperty().samples[0] return faces, counts
def getMesh(iarch): ''' Load the static mesh data from an alembic archive ''' top = iarch.getTop() par = top.children[0] par = IXform(top, par.getName()) abcMesh = par.children[0] abcMesh = IPolyMesh(par, abcMesh.getName()) sch = abcMesh.getSchema() faces = sch.getFaceIndicesProperty().samples[0] counts = sch.getFaceCountsProperty().samples[0] return faces, counts
def loadSmpx(iarch): top = iarch.getTop() par = top.children[0] par = IXform(top, par.getName()) abcMesh = par.children[0] abcMesh = IPolyMesh(par, abcMesh.getName()) meshSchema = abcMesh.getSchema() posProp = meshSchema.getPositionsProperty() shapes = [] lpps = len(posProp.samples) for i, s in enumerate(posProp.samples): print "Reading {0: 3d} of {1}\r".format(i, lpps), shapes.append(s) print "Reading {0: 3d} of {1}".format(lpps, lpps) return shapes
def buildBaseAbc(self, abcPath): iarch = IArchive(str(abcPath)) # because alembic hates unicode try: top = iarch.getTop() par = top.children[0] par = IXform(top, par.getName()) abcMesh = par.children[0] abcMesh = IPolyMesh(par, abcMesh.getName()) systemSchema = par.getSchema() props = systemSchema.getUserProperties() prop = props.getProperty("simplex") jsString = prop.getValue() js = json.loads(jsString) obj = self.DCC.buildRestABC(abcMesh, js) finally: del iarch return obj
def loadSmpx(iarch): ''' Load the json and shape data from a .smpx file ''' top = iarch.getTop() par = top.children[0] par = IXform(top, par.getName()) abcMesh = par.children[0] abcMesh = IPolyMesh(par, abcMesh.getName()) meshSchema = abcMesh.getSchema() posProp = meshSchema.getPositionsProperty() print "Loading Verts" shapes = np.empty((len(posProp.samples), len(posProp.samples[0]), 3)) for i, s in enumerate(posProp.samples): shapes[i] = arrayToNumpy(s) print "Done Loading" return shapes
def getSmpxArchiveData(abcPath): '''Read and return the low level relevant data from a simplex alembic Parameters ---------- abcPath : str The path to the .smpx file Returns ------- : IArchive An opened Alembic IArchive object handle : IPolyMesh An Alembic Mesh handle : str The json definition string ''' if not os.path.isfile(str(abcPath)): raise IOError("File does not exist: " + str(abcPath)) iarch = IArchive(str(abcPath)) # because alembic hates unicode top, par, abcMesh = [None] * 3 try: top = iarch.getTop() par = top.children[0] par = IXform(top, par.getName()) abcMesh = par.children[0] abcMesh = IPolyMesh(par, abcMesh.getName()) # I *could* come up with a generic property reader # but it's useless for me at this time sch = par.getSchema() props = sch.getUserProperties() jsString = readStringProperty(props, "simplex") except Exception: #pylint: disable=broad-except # ensure that the .smpx file is released iarch, top, par, abcMesh = [None] * 4 raise # Must return the archive, otherwise it gets GC'd return iarch, abcMesh, jsString
def loadJSString(iarch): '''Get the json string out of a .smpx file Parameters ---------- iarch : IArchive The input alembic archive Returns ------- : str The simplex json string ''' top = iarch.getTop() par = top.children[0] par = IXform(top, par.getName()) systemSchema = par.getSchema() props = systemSchema.getUserProperties() prop = props.getProperty("simplex") jsString = prop.getValue() return jsString
def readFalloffData(abcPath): '''Load the relevant data from a simplex alembic Parameters ---------- abcPath : str Path to the .smpx file ''' if not os.path.isfile(str(abcPath)): raise IOError("File does not exist: " + str(abcPath)) iarch = IArchive(str(abcPath)) # because alembic hates unicode top, par, systemSchema, foPropPar, foProp = [None] * 5 try: top = iarch.getTop() par = top.children[0] par = IXform(top, par.getName()) systemSchema = par.getSchema() props = systemSchema.getUserProperties() foDict = {} try: foPropPar = props.getProperty("falloffs") except KeyError: pass else: nps = foPropPar.getNumProperties() for i in range(nps): foProp = foPropPar.getProperty(i) fon = foProp.getName() fov = foProp.getValue() # imath.FloatArray fov = list(fov) if np is None else np.array(fov) foDict[fon] = fov finally: iarch, top, par, systemSchema, foPropPar, foProp = [None] * 6 return foDict