Пример #1
0
    def getScriptCode(self, obj, **kwargs):
        if isinstance(obj, MeshSceneObject):
            configSection = kwargs.get('configSection', False)
            namemap = kwargs.get('namemap', {})
            convertpath = kwargs['convertPath']
            script = ''
            args = {
                'varname': namemap[obj],
                'objname': obj.name,
                'timesteps': obj.getTimestepList()
            }

            if not configSection:
                if 'filename' in obj.kwargs:
                    args['filename'] = convertpath(obj.kwargs['filename'])
                    script += '%(varname)s=VTK.loadObject(%(filename)s,%(objname)r)'
                elif 'filenames' in obj.kwargs:
                    args['filenames'] = ('[' + ','.join(
                        map(convertpath, obj.kwargs['filenames'])) + ']')
                    script += '%(varname)s=VTK.loadSequence(%(filenames)s,%(objname)r)'
            else:
                script += '%(varname)s.setTimestepList(%(timesteps)r)\n'

            return eidolon.setStrIndent(script % args).strip() + '\n'
        else:
            return MeshScenePlugin.getScriptCode(self, obj, **kwargs)
Пример #2
0
    def getScriptCode(self, obj, **kwargs):
        configSection = kwargs.get('configSection', False)
        namemap = kwargs.get('namemap', {})
        convertpath = kwargs['convertPath']
        script = ''
        args = {'varname': namemap[obj], 'objname': obj.name}

        if not configSection and isinstance(obj, PlotSceneObject):
            args['filename'] = convertpath(obj.filename)
            script += '%(varname)s = Plot.loadObject(%(filename)s,%(objname)r)\n'

        return eidolon.setStrIndent(script % args).strip() + '\n'
Пример #3
0
    def getScriptCode(self, obj, **kwargs):
        if isinstance(obj, MeshSceneObject):
            configSection = kwargs.get('configSection', False)
            namemap = kwargs.get('namemap', {})
            convertpath = kwargs['convertPath']
            script = ''
            args = {
                'varname': namemap[obj],
                'objname': obj.name,
                'loadorder': obj.source['loadorder']
            }

            if not configSection:
                args['filename'] = convertpath(obj.kwargs['filename'])
                script += '%(varname)s=X4DF.loadObject(%(filename)s,%(objname)r)[%(loadorder)]'

            return eidolon.setStrIndent(script % args).strip() + '\n'
        else:
            return eidolon.MeshScenePlugin.getScriptCode(self, obj, **kwargs)
Пример #4
0
    def getScriptCode(self, obj, **kwargs):
        configSection = kwargs.get('configSection', False)
        namemap = kwargs.get('namemap', {})
        convertpath = kwargs['convertPath']
        varname = namemap[obj]
        script = ''
        args = {}

        if not configSection and isinstance(obj, ImageSceneObject):
            filename = convertpath(obj.source['filename'])

            args = {
                'varname': varname,
                'objname': obj.name,
                'filename': filename,
                'loadorder': obj.source[
                    'loadorder'],  # load order is used to load the same Parrec file multiple times once for each image
                'scalemethod': obj.source['scalemethod'],
            }

            script += '%(varname)s = ParRec.loadObject(%(filename)s,%(objname)r,%(scalemethod)r)[%(loadorder)i]\n'

        elif isinstance(obj, ImageSceneObjectRepr):
            args = {
                'varname': varname,
                'pname': namemap[obj.parent],
                'reprtype': obj.reprtype,
                'matname': namemap.get(obj.getMaterialName(), 'Greyscale')
            }

            if configSection:
                script = ImageScenePlugin.getScriptCode(self,
                                                        obj,
                                                        setMaterial=False,
                                                        **kwargs)
            else:
                script = "%(varname)s=%(pname)s.createRepr(ReprType._%(reprtype)s,imgmat=%(matname)s)\n"

        return setStrIndent(script % args).strip() + '\n'
Пример #5
0
    def getScriptCode(self, obj, **kwargs):
        configSection = kwargs.get('configSection', False)
        namemap = kwargs.get('namemap', {})
        convertpath = kwargs['convertPath']
        varname = namemap[obj]
        script = ''
        args = {}

        if not configSection and isinstance(obj, ImageSceneObject):
            filename = convertpath(obj.source['filename'])

            args = {
                'varname': varname,
                'objname': obj.name,
                'filename': filename
            }

            script += '%(varname)s = Nifti.loadObject(%(filename)s,%(objname)r)\n'

        elif isinstance(obj, ImageSceneObjectRepr):
            args = {
                'varname': varname,
                'pname': namemap[obj.parent],
                'reprtype': obj.reprtype,
                'matname': namemap.get(obj.getMaterialName(), 'Greyscale')
            }

            if configSection:
                script = ImageScenePlugin.getScriptCode(self,
                                                        obj,
                                                        setMaterial=False,
                                                        **kwargs)
            else:
                script = "%(varname)s=%(pname)s.createRepr(ReprType._%(reprtype)s,imgmat=%(matname)s)\n"

        return eidolon.setStrIndent(script % args).strip() + '\n'