283 lines
10 KiB
C#
283 lines
10 KiB
C#
|
using System;
|
||
|
|
using System.Diagnostics;
|
||
|
|
using System.IO;
|
||
|
|
using System.Net;
|
||
|
|
using System.Text;
|
||
|
|
using System.Collections;
|
||
|
|
using System.Collections.Generic;
|
||
|
|
using System.Text.RegularExpressions;
|
||
|
|
using UnityEngine;
|
||
|
|
|
||
|
|
namespace OpenSee {
|
||
|
|
|
||
|
|
public class OpenSeeLauncher : MonoBehaviour {
|
||
|
|
[Header("Settings")]
|
||
|
|
[Tooltip("If this is left empty, it will default to the OpenSee component on the same game object.")]
|
||
|
|
public OpenSee openSeeTarget;
|
||
|
|
[Tooltip("The numeric index of the camera as in the list returned by the ListCameras function.")]
|
||
|
|
public int cameraIndex = -1;
|
||
|
|
[Tooltip("The path to a video file that should be used as input for the tracker instead of a camera")]
|
||
|
|
public string videoPath = "";
|
||
|
|
[Tooltip("Should facetracker.exe be started automatically with the scene?")]
|
||
|
|
public bool autoStart = false;
|
||
|
|
[Tooltip("If this is enabled, available cameras are automatically printed to the debug console.")]
|
||
|
|
public bool autoListCameras = false;
|
||
|
|
[Tooltip("If this is enabled, no output from the tracker will be printed to the debug console. This cannot be changed while the tracker is running.")]
|
||
|
|
public bool dontPrint = true;
|
||
|
|
[Tooltip("The path to \"facetracker.exe\".")]
|
||
|
|
public string exePath = "facetracker.exe";
|
||
|
|
[Tooltip("The path, where the .onnx model files can be found")]
|
||
|
|
public string modelPath = "models\\";
|
||
|
|
[Tooltip("Additional options that should be passed to the face tracker.")]
|
||
|
|
public string[] commandlineOptions = new string[] { "--silent", "1", "--max-threads", "2", "--model", "1" };
|
||
|
|
[Header("Runtime information")]
|
||
|
|
[Tooltip("This field shows if the tracker is currently alive.")]
|
||
|
|
public bool trackerAlive = false;
|
||
|
|
|
||
|
|
private string ip;
|
||
|
|
private Process trackerProcess = null;
|
||
|
|
private StringBuilder trackerSB = null;
|
||
|
|
private bool dontPrintNow = false;
|
||
|
|
private Job job = null;
|
||
|
|
|
||
|
|
private bool CheckSetup(bool requireTarget) {
|
||
|
|
if (openSeeTarget == null)
|
||
|
|
openSeeTarget = GetComponent<OpenSee>();
|
||
|
|
if (openSeeTarget == null) {
|
||
|
|
UnityEngine.Debug.LogError("No openSeeTarget is set.");
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
IPAddress address;
|
||
|
|
if (!File.Exists(exePath)) {
|
||
|
|
UnityEngine.Debug.LogError("Facetracker executable cannot be found.");
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
if (!Directory.Exists(modelPath)) {
|
||
|
|
UnityEngine.Debug.LogError("Model directory cannot be found.");
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
if (requireTarget) {
|
||
|
|
ip = openSeeTarget.listenAddress;
|
||
|
|
if (!IPAddress.TryParse(ip, out address)) {
|
||
|
|
UnityEngine.Debug.LogError("No valid IP address was given in the OpenSee component.");
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
if (openSeeTarget.listenPort < 1 || openSeeTarget.listenPort > 65535) {
|
||
|
|
UnityEngine.Debug.LogError("No valid port was given in the OpenSee component.");
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
if (cameraIndex >= 0 && videoPath != "") {
|
||
|
|
UnityEngine.Debug.LogError("Entering both a camera index and a video filename is not valid.");
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
if (cameraIndex < 0 && !File.Exists(videoPath)) {
|
||
|
|
if (videoPath != "") {
|
||
|
|
UnityEngine.Debug.LogError("The given video file does not exist.");
|
||
|
|
} else {
|
||
|
|
UnityEngine.Debug.LogError("No camera or video file was given.");
|
||
|
|
}
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
if (job == null) {
|
||
|
|
job = new Job();
|
||
|
|
}
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
|
||
|
|
public string[] ListCameras() {
|
||
|
|
if (!CheckSetup(false))
|
||
|
|
return null;
|
||
|
|
|
||
|
|
StringBuilder stringBuilder;
|
||
|
|
ProcessStartInfo processStartInfo;
|
||
|
|
Process process;
|
||
|
|
|
||
|
|
stringBuilder = new StringBuilder();
|
||
|
|
|
||
|
|
processStartInfo = new ProcessStartInfo();
|
||
|
|
processStartInfo.CreateNoWindow = true;
|
||
|
|
processStartInfo.RedirectStandardOutput = true;
|
||
|
|
processStartInfo.RedirectStandardInput = true;
|
||
|
|
processStartInfo.RedirectStandardError = true;
|
||
|
|
processStartInfo.UseShellExecute = false;
|
||
|
|
processStartInfo.FileName = exePath;
|
||
|
|
processStartInfo.Arguments = "--list-cameras 2";
|
||
|
|
|
||
|
|
process = new Process();
|
||
|
|
process.StartInfo = processStartInfo;
|
||
|
|
process.EnableRaisingEvents = true;
|
||
|
|
process.OutputDataReceived += new DataReceivedEventHandler(delegate(object sender, DataReceivedEventArgs e) {
|
||
|
|
stringBuilder.Append(e.Data);
|
||
|
|
stringBuilder.Append("\n");
|
||
|
|
});
|
||
|
|
process.Start();
|
||
|
|
job.AddProcess(process.Handle);
|
||
|
|
process.BeginOutputReadLine();
|
||
|
|
process.WaitForExit();
|
||
|
|
process.CancelOutputRead();
|
||
|
|
|
||
|
|
string cameraInfo = stringBuilder.ToString();
|
||
|
|
string[] cameras = cameraInfo.Split(new[] { "\r\n", "\r", "\n" }, StringSplitOptions.RemoveEmptyEntries);
|
||
|
|
return cameras;
|
||
|
|
}
|
||
|
|
|
||
|
|
// From: http://csharptest.net/529/how-to-correctly-escape-command-line-arguments-in-c/index.html
|
||
|
|
private static string EscapeArguments(params string[] args)
|
||
|
|
{
|
||
|
|
StringBuilder arguments = new StringBuilder();
|
||
|
|
Regex invalidChar = new Regex("[\x00\x0a\x0d]");// these can not be escaped
|
||
|
|
Regex needsQuotes = new Regex(@"\s|""");// contains whitespace or two quote characters
|
||
|
|
Regex escapeQuote = new Regex(@"(\\*)(""|$)");// one or more '\' followed with a quote or end of string
|
||
|
|
for (int carg = 0; args != null && carg < args.Length; carg++)
|
||
|
|
{
|
||
|
|
if (args[carg] == null) { throw new ArgumentNullException("args[" + carg + "]"); }
|
||
|
|
if (invalidChar.IsMatch(args[carg])) { throw new ArgumentOutOfRangeException("args[" + carg + "]"); }
|
||
|
|
if (args[carg] == String.Empty) { arguments.Append("\"\""); }
|
||
|
|
else if (!needsQuotes.IsMatch(args[carg])) { arguments.Append(args[carg]); }
|
||
|
|
else
|
||
|
|
{
|
||
|
|
arguments.Append('"');
|
||
|
|
arguments.Append(escapeQuote.Replace(args[carg], m => m.Groups[1].Value + m.Groups[1].Value + (m.Groups[2].Value == "\"" ? "\\\"" : "")));
|
||
|
|
arguments.Append('"');
|
||
|
|
}
|
||
|
|
if (carg + 1 < args.Length)
|
||
|
|
arguments.Append(' ');
|
||
|
|
}
|
||
|
|
return arguments.ToString();
|
||
|
|
}
|
||
|
|
|
||
|
|
public bool StartTracker() {
|
||
|
|
if (!CheckSetup(true))
|
||
|
|
return false;
|
||
|
|
|
||
|
|
ProcessStartInfo processStartInfo;
|
||
|
|
|
||
|
|
processStartInfo = new ProcessStartInfo();
|
||
|
|
processStartInfo.CreateNoWindow = true;
|
||
|
|
processStartInfo.RedirectStandardOutput = true;
|
||
|
|
processStartInfo.RedirectStandardInput = true;
|
||
|
|
processStartInfo.RedirectStandardError = true;
|
||
|
|
processStartInfo.UseShellExecute = false;
|
||
|
|
processStartInfo.FileName = exePath;
|
||
|
|
|
||
|
|
List<string> arguments = new List<string>();
|
||
|
|
arguments.Add("--ip");
|
||
|
|
arguments.Add(ip);
|
||
|
|
arguments.Add("--port");
|
||
|
|
arguments.Add(openSeeTarget.listenPort.ToString());
|
||
|
|
arguments.Add("--model-dir");
|
||
|
|
arguments.Add(modelPath);
|
||
|
|
arguments.Add("--capture");
|
||
|
|
if (videoPath != "" && File.Exists(videoPath))
|
||
|
|
arguments.Add(videoPath);
|
||
|
|
else
|
||
|
|
arguments.Add(cameraIndex.ToString());
|
||
|
|
foreach (string argument in commandlineOptions) {
|
||
|
|
arguments.Add(argument);
|
||
|
|
}
|
||
|
|
string argumentString = EscapeArguments(arguments.ToArray());
|
||
|
|
processStartInfo.Arguments = argumentString;
|
||
|
|
|
||
|
|
if (trackerProcess != null && !trackerProcess.HasExited) {
|
||
|
|
trackerProcess.CloseMainWindow();
|
||
|
|
trackerProcess.Close();
|
||
|
|
if (!dontPrintNow) {
|
||
|
|
try {
|
||
|
|
trackerProcess.CancelOutputRead();
|
||
|
|
trackerProcess.CancelErrorRead();
|
||
|
|
} catch {}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
trackerSB = new StringBuilder();
|
||
|
|
trackerProcess = new Process();
|
||
|
|
trackerProcess.StartInfo = processStartInfo;
|
||
|
|
dontPrintNow = dontPrint;
|
||
|
|
if (!dontPrintNow) {
|
||
|
|
trackerProcess.EnableRaisingEvents = true;
|
||
|
|
trackerProcess.OutputDataReceived += new DataReceivedEventHandler(delegate(object sender, DataReceivedEventArgs e) {
|
||
|
|
trackerSB.Append(e.Data);
|
||
|
|
trackerSB.Append("\n");
|
||
|
|
});
|
||
|
|
trackerProcess.ErrorDataReceived += new DataReceivedEventHandler(delegate(object sender, DataReceivedEventArgs e) {
|
||
|
|
trackerSB.Append(e.Data);
|
||
|
|
trackerSB.Append("\n");
|
||
|
|
});
|
||
|
|
}
|
||
|
|
trackerProcess.Start();
|
||
|
|
job.AddProcess(trackerProcess.Handle);
|
||
|
|
if (!dontPrintNow) {
|
||
|
|
trackerProcess.BeginOutputReadLine();
|
||
|
|
trackerProcess.BeginErrorReadLine();
|
||
|
|
}
|
||
|
|
|
||
|
|
return !trackerProcess.HasExited;
|
||
|
|
}
|
||
|
|
|
||
|
|
public void StopTracker() {
|
||
|
|
if (trackerProcess != null && !trackerProcess.HasExited) {
|
||
|
|
trackerProcess.CloseMainWindow();
|
||
|
|
trackerProcess.Close();
|
||
|
|
if (!dontPrintNow) {
|
||
|
|
try {
|
||
|
|
trackerProcess.CancelOutputRead();
|
||
|
|
trackerProcess.CancelErrorRead();
|
||
|
|
} catch {}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
trackerProcess = null;
|
||
|
|
}
|
||
|
|
|
||
|
|
public void Start() {
|
||
|
|
if (autoStart)
|
||
|
|
StartTracker();
|
||
|
|
if (autoListCameras) {
|
||
|
|
string[] cameras = ListCameras();
|
||
|
|
if (cameras == null || cameras.Length < 1) {
|
||
|
|
UnityEngine.Debug.Log("No cameras found.");
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
UnityEngine.Debug.Log("Cameras:");
|
||
|
|
for (int i = 0; i < cameras.Length; i++) {
|
||
|
|
UnityEngine.Debug.Log(i + " = " + cameras[i]);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
public void Update() {
|
||
|
|
if (trackerProcess != null)
|
||
|
|
trackerAlive = !trackerProcess.HasExited;
|
||
|
|
else {
|
||
|
|
trackerAlive = false;
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
if (dontPrintNow || trackerSB == null)
|
||
|
|
return;
|
||
|
|
int len = trackerSB.Length;
|
||
|
|
if (len > 0) {
|
||
|
|
string output = trackerSB.ToString();
|
||
|
|
UnityEngine.Debug.Log("Tracker: " + output);
|
||
|
|
trackerSB.Clear();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
private void CleanJob() {
|
||
|
|
if (job != null) {
|
||
|
|
job.Dispose();
|
||
|
|
job = null;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
public void OnDestroy() {
|
||
|
|
StopTracker();
|
||
|
|
CleanJob();
|
||
|
|
}
|
||
|
|
|
||
|
|
public void OnApplicationQuit() {
|
||
|
|
StopTracker();
|
||
|
|
CleanJob();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|