Beispiel #1
0
    def __init__(self):
        self.socket = VRaySocket()

        self.params = []

        self.verboseLevel = '1'
        self.showProgress = '2'
Beispiel #2
0
	def execute(self, context):
		s = VRaySocket()
		s.connect()
		s.send("stop", result=False)
		s.send("quit", result=False)
		s.disconnect()

		return {'FINISHED'}
Beispiel #3
0
    def __init__(self):
        self.socket = VRaySocket()

        self.params = []

        self.verboseLevel = '1'
        self.showProgress = '2'
Beispiel #4
0
	def execute(self, context):
		s = VRaySocket()
		s.connect()
		s.send("stop", result=False)
		s.send("quit", result=False)
		s.disconnect()

		return {'FINISHED'}
Beispiel #5
0
class VRayProcess():
    # V-Ray process
    process    = None
    exit_ready = None
    params     = None

    # V-Ray command socket
    socket = None

    # Executable parameters
    sceneFile     = None
    imgFile       = None
    showProgress  = None
    progressUseCR = None
    verboseLevel  = None
    cmdMode       = None

    bus = None
    scene = None

    VRayScene    = None
    VRayExporter = None
    VRayDR       = None


    def __init__(self):
        self.socket = VRaySocket()

        self.params = []

        self.verboseLevel = '1'
        self.showProgress = '2'


    def __del__(self):
        pass


    def set_params(self, bus=None, params=None):
        self.bus = bus

        self.VRayScene    = self.scene.vray
        self.VRayExporter = self.VRayScene.exporter
        self.VRayDR       = self.VRayScene.VRayDR

        self.params = []
        self.params.append(vb25.utils.get_vray_standalone_path(self.scene))

        self.params.append('-sceneFile=')
        self.params.append(self.sceneFile)
        self.params.append('-imgFile=')
        self.params.append(self.imgFile)

        if self.VRayExporter.use_progress:
            # We need only progress info
            self.params.append('-verboseLevel=')
            self.params.append('3')

            # Always show progress
            self.params.append('-showProgress=')
            self.params.append('2')

            # Use log line breaks
            self.params.append('-progressUseCR=')
            self.params.append('0')

        else:
            self.params.append('-verboseLevel=')
            self.params.append(self.verboseLevel)
            self.params.append('-showProgress=')
            self.params.append(self.showProgress)

        if self.VRayDR.on:
            if len(self.VRayDR.nodes):
                self.params.append('-distributed=1')
                self.params.append('-portNumber=%i' % self.VRayDR.port)
                self.params.append('-renderhost=%s' % Quotes(';'.join([n.address for n in self.VRayDR.nodes])))
                self.params.append('-include=%s' % Quotes(self.bus['filenames']['DR']['shared_dir'] + os.sep))

        # Setup command mode
        # Disable VFB
        self.params.append('-display=')
        self.params.append('0')

        # Enable command socket
        self.params.append('-cmdMode=')
        self.params.append('1')

        if not self.VRayExporter.autorun:
            vb25.utils.debug(self.scene, "Enable \"Autorun\" option to start V-Ray automatically after export.")
            vb25.utils.debug(self.scene, "Command: %s" % ' '.join(self.params))


    def run(self):
        if not self.VRayExporter.autorun:
            return

        if self.VRayExporter.use_progress:
            self.process = subprocess.Popen(self.params, bufsize=256, stdout=subprocess.PIPE)

            if vb25.utils.PLATFORM != 'win32':
                fd = self.process.stdout.fileno()
                fl = fcntl.fcntl(fd, fcntl.F_GETFL)
                fcntl.fcntl(fd, fcntl.F_SETFL, fl | os.O_NONBLOCK)
        else:
            self.process = subprocess.Popen(self.params)

        self.exit_ready = False

        time.sleep(0.25)
        self.socket.connect()


    def is_running(self):
        if self.process is None:
            return False
        if self.process.poll() is None:
            return True
        return False


    def kill(self):
        self.quit()

        if self.is_running():
            self.process.terminate()

        self.process = None


    def get_progress(self):
        msg  = None
        prog = None

        if not self.exit_ready:
            if self.process and self.is_running():
                stdout_lines = None
                try:
                    self.process.stdout.flush()
                    stdout_lines = self.process.stdout.readlines(256)
                except:
                    pass

                if stdout_lines:
                    for stdout_line in stdout_lines:
                        line = stdout_line.decode('ascii').strip()

                        if self.VRayExporter.debug:
                            print(line)

                        if line.find("Building light cache") != -1:
                            msg = "Light cache"
                        elif line.find("Prepass") != -1:
                            prepass_num = line[line.find("Prepass")+7:line.find("of")].strip()
                            msg = "Irradiance map (prepass %s)" % (prepass_num)
                        elif line.find("Rendering image") != -1:
                            msg = "Rendering"
                        elif line.find("Building caustics") != -1:
                            msg = "Caustics"
                        elif line.find("Frame took") != -1:
                            self.exit_ready = True

                        if msg is None:
                            continue

                        p_start = line.find("...: ") + 5
                        p_end   = line.find("%")

                        if p_start != -1 and p_end != -1 and p_end > p_start:
                            p_str = line[p_start:p_end].strip()
                            if len(p_str):
                                prog = float(p_str) / 100.0
                                break

        return msg, prog


    def load_scene(self):
        if not self.sceneFile:
            vb25.utils.vb25.utils.debug(None, "Scene file is not set", error=True)
            return 'Scene file is not set'

        self.socket.send("load %s" % self.sceneFile)

        return None


    def unload_scene(self):
        self.socket.send("unload")
        return None


    def reload_scene(self):
        self.unload_scene()
        self.load_scene()
        return None


    def render(self):
        self.socket.send("render", result=False)
        return None


    def quit(self):
        self.socket.send("stop")
        self.socket.send("quit")
        self.socket.disconnect()
        return None


    def recieve_image(self, progressFile):
        jpeg_image = None
        jpeg_size  = 0
        buff  = []

        if not self.is_running():
            self.exit_ready = True
            return 'V-Ray is not running'

        # Request image
        self.socket.send("getImage 90 1", result=False)

        # Read image stream size
        jpeg_size_bytes = self.socket.recv(4)

        # Check if 'fail' recieved
        if jpeg_size_bytes == b'fail':
            self.socket.recv(3) # Read 'e', 'd', '\0'
            self.exit_ready = True
            return 'getImage failed'

        try:
            # Get stream size in bytes
            jpeg_size = struct.unpack("<L", jpeg_size_bytes)[0]

            # print("JPEG stream size = %i"%(jpeg_size))

            # Read JPEG stream
            jpeg_image = self.socket.recv(jpeg_size)

            # Write stream to file
            open(progressFile, 'wb').write(jpeg_image)
        except:
            return 'JPEG stream recieve fail'

        return None
Beispiel #6
0
class VRayProcess():
    # V-Ray process
    process = None
    exit_ready = None
    params = None

    # V-Ray command socket
    socket = None

    # Executable parameters
    sceneFile = None
    imgFile = None
    showProgress = None
    progressUseCR = None
    verboseLevel = None
    cmdMode = None

    scene = None
    VRayScene = None
    VRayExporter = None
    VRayDR = None

    def __init__(self):
        self.socket = VRaySocket()

        self.params = []

        self.verboseLevel = '1'
        self.showProgress = '2'

    def __del__(self):
        pass

    def set_params(self, params=None):
        self.VRayScene = self.scene.vray
        self.VRayExporter = self.VRayScene.exporter
        self.VRayDR = self.VRayScene.VRayDR

        self.params = []
        self.params.append(vb25.utils.get_vray_standalone_path(self.scene))

        self.params.append('-sceneFile=')
        self.params.append(self.sceneFile)
        self.params.append('-imgFile=')
        self.params.append(self.imgFile)

        if self.VRayExporter.use_progress:
            # We need only progress info
            self.params.append('-verboseLevel=')
            self.params.append('3')

            # Always show progress
            self.params.append('-showProgress=')
            self.params.append('2')

            # Use log line breaks
            self.params.append('-progressUseCR=')
            self.params.append('0')

        else:
            self.params.append('-verboseLevel=')
            self.params.append(self.verboseLevel)
            self.params.append('-showProgress=')
            self.params.append(self.showProgress)

        # Setup command mode
        # Disable VFB
        self.params.append('-display=')
        self.params.append('0')

        # Enable command socket
        self.params.append('-cmdMode=')
        self.params.append('1')

        if not self.VRayExporter.autorun:
            vb25.utils.debug(
                self.scene,
                "Enable \"Autorun\" option to start V-Ray automatically after export."
            )
            vb25.utils.debug(self.scene, "Command: %s" % ' '.join(self.params))

    def run(self):
        if not self.VRayExporter.autorun:
            return

        if self.VRayExporter.use_progress:
            self.process = subprocess.Popen(self.params,
                                            bufsize=256,
                                            stdout=subprocess.PIPE)

            if vb25.utils.PLATFORM != 'win32':
                fd = self.process.stdout.fileno()
                fl = fcntl.fcntl(fd, fcntl.F_GETFL)
                fcntl.fcntl(fd, fcntl.F_SETFL, fl | os.O_NONBLOCK)
        else:
            self.process = subprocess.Popen(self.params)

        self.exit_ready = False

        time.sleep(0.25)
        self.socket.connect()

    def is_running(self):
        if self.process is None:
            return False
        if self.process.poll() is None:
            return True
        return False

    def kill(self):
        self.quit()

        if self.is_running():
            self.process.terminate()

        self.process = None

    def get_progress(self):
        msg = None
        prog = None

        if not self.exit_ready:
            if self.process and self.is_running():
                stdout_lines = None
                try:
                    self.process.stdout.flush()
                    stdout_lines = self.process.stdout.readlines(256)
                except:
                    pass

                if stdout_lines:
                    for stdout_line in stdout_lines:
                        line = stdout_line.decode('ascii').strip()

                        if self.VRayExporter.debug:
                            print(line)

                        if line.find("Building light cache") != -1:
                            msg = "Light cache"
                        elif line.find("Prepass") != -1:
                            prepass_num = line[line.find("Prepass") +
                                               7:line.find("of")].strip()
                            msg = "Irradiance map (prepass %s)" % (prepass_num)
                        elif line.find("Rendering image") != -1:
                            msg = "Rendering"
                        elif line.find("Building caustics") != -1:
                            msg = "Caustics"
                        elif line.find("Frame took") != -1:
                            self.exit_ready = True

                        if msg is None:
                            continue

                        p_start = line.find("...: ") + 5
                        p_end = line.find("%")

                        if p_start != -1 and p_end != -1 and p_end > p_start:
                            p_str = line[p_start:p_end].strip()
                            if len(p_str):
                                prog = float(p_str) / 100.0
                                break

        return msg, prog

    def load_scene(self):
        if not self.sceneFile:
            vb25.utils.vb25.utils.debug(None,
                                        "Scene file is not set",
                                        error=True)
            return 'Scene file is not set'

        self.socket.send("load %s" % self.sceneFile)

        return None

    def unload_scene(self):
        self.socket.send("unload")
        return None

    def reload_scene(self):
        self.unload_scene()
        self.load_scene()
        return None

    def render(self):
        self.socket.send("render", result=False)
        return None

    def quit(self):
        self.socket.send("stop")
        self.socket.send("quit")
        self.socket.disconnect()
        return None

    def recieve_image(self, progressFile):
        jpeg_image = None
        jpeg_size = 0
        buff = []

        if not self.is_running():
            self.exit_ready = True
            return 'V-Ray is not running'

        # Request image
        self.socket.send("getImage 90 1", result=False)

        # Read image stream size
        jpeg_size_bytes = self.socket.recv(4)

        # Check if 'fail' recieved
        if jpeg_size_bytes == b'fail':
            self.socket.recv(3)  # Read 'e', 'd', '\0'
            self.exit_ready = True
            return 'getImage failed'

        try:
            # Get stream size in bytes
            jpeg_size = struct.unpack("<L", jpeg_size_bytes)[0]

            # print("JPEG stream size = %i"%(jpeg_size))

            # Read JPEG stream
            jpeg_image = self.socket.recv(jpeg_size)

            # Write stream to file
            open(progressFile, 'wb').write(jpeg_image)
        except:
            return 'JPEG stream recieve fail'

        return None