zed-ar-headfree-passthrough/Assets/ZED/SDK/NativeInterface/ZEDCameraSettings.cs

414 lines
15 KiB
C#

//======= Copyright (c) Stereolabs Corporation, All rights reserved. ===============
using System.Runtime.InteropServices;
/// <summary>
/// Stores the camera settings (brightness/contrast, gain/exposure, etc.) and interfaces with the ZED
/// when they need to be loaded or changed.
/// Created by ZEDCamera and referenced by ZEDCameraSettingsEditor.
/// </summary><remarks>
/// The actual settings themselves are stored in an instance of CameraSettings, for easier manipulation.
/// But this class provides accessors for each value within it.
/// </remarks>
public class ZEDCameraSettings
{
#region DLL Calls
const string nameDll = "sl_unitywrapper";
[DllImport(nameDll, EntryPoint = "dllz_set_camera_settings")]
private static extern void dllz_set_camera_settings(int id, int mode, int value, int usedefault);
[DllImport(nameDll, EntryPoint = "dllz_get_camera_settings")]
private static extern int dllz_get_camera_settings(int id, int mode);
#endregion
/// <summary>
/// Container for ZED camera settings, with constructors for easily creating default or specific values
/// or making duplicate instances.
/// </summary>
public class CameraSettings
{
/// <summary>
/// Holds an int for each setting, with indexes corresponding to sl.CAMERA_SETTINGS.
/// </summary>
public int[] settings = new int[System.Enum.GetNames(typeof(sl.CAMERA_SETTINGS)).Length];
/// <summary>
/// Constructor. Call without arguments to set all values to default.
/// </summary>
/// <param name="brightness">Camera's brightness setting.</param>
/// <param name="contrast">Camera's contrast setting.</param>
/// <param name="hue">Camera's hue setting.</param>
/// <param name="saturation">Camera's saturation setting.</param>
/// <param name="whiteBalance">Camera's white balance setting. -1 means automatic.</param>
/// <param name="gain">Camera's gain setting. -1 means automatic.</param>
/// <param name="exposure">Camera's exposure setting. -1 means automatic.</param>
public CameraSettings(int brightness = 4, int contrast = 4, int hue = 0, int saturation = 4, int whiteBalance = -1, int gain = -1, int exposure = -1,int ledStatus = 1)
{
settings = new int[System.Enum.GetNames(typeof(sl.CAMERA_SETTINGS)).Length];
settings[(int)sl.CAMERA_SETTINGS.BRIGHTNESS] = brightness;
settings[(int)sl.CAMERA_SETTINGS.CONTRAST] = contrast;
settings[(int)sl.CAMERA_SETTINGS.SATURATION] = saturation;
settings[(int)sl.CAMERA_SETTINGS.HUE] = hue;
settings[(int)sl.CAMERA_SETTINGS.WHITEBALANCE] = whiteBalance;
settings[(int)sl.CAMERA_SETTINGS.GAIN] = gain;
settings[(int)sl.CAMERA_SETTINGS.EXPOSURE] = exposure;
settings[(int)sl.CAMERA_SETTINGS.LED_STATUS] = ledStatus;
}
/// <summary>
/// Constructor. Sets settings to match another CameraSettings passed in the argument.
/// </summary>
/// <param name="other"></param>
public CameraSettings(CameraSettings other)
{
settings = new int[System.Enum.GetNames(typeof(sl.CAMERA_SETTINGS)).Length];
settings[(int)sl.CAMERA_SETTINGS.BRIGHTNESS] = other.settings[(int)sl.CAMERA_SETTINGS.BRIGHTNESS];
settings[(int)sl.CAMERA_SETTINGS.CONTRAST] = other.settings[(int)sl.CAMERA_SETTINGS.CONTRAST];
settings[(int)sl.CAMERA_SETTINGS.SATURATION] = other.settings[(int)sl.CAMERA_SETTINGS.SATURATION];
settings[(int)sl.CAMERA_SETTINGS.HUE] = other.settings[(int)sl.CAMERA_SETTINGS.HUE];
settings[(int)sl.CAMERA_SETTINGS.WHITEBALANCE] = other.settings[(int)sl.CAMERA_SETTINGS.WHITEBALANCE];
settings[(int)sl.CAMERA_SETTINGS.GAIN] = other.settings[(int)sl.CAMERA_SETTINGS.GAIN];
settings[(int)sl.CAMERA_SETTINGS.EXPOSURE] = other.settings[(int)sl.CAMERA_SETTINGS.EXPOSURE];
settings[(int)sl.CAMERA_SETTINGS.LED_STATUS] = other.settings[(int)sl.CAMERA_SETTINGS.LED_STATUS];
}
/// <summary>
/// Returns a new instance of CameraSettings with the same settings as the instance this function was called with.
/// </summary>
/// <returns>New instance of CameraSettings.</returns>
public CameraSettings Clone()
{
return new CameraSettings(this);
}
/// <summary>
/// ZED camera's brightness setting.
/// </summary>
public int Brightness
{
get
{
return settings[(int)sl.CAMERA_SETTINGS.BRIGHTNESS];
}
set
{
settings[(int)sl.CAMERA_SETTINGS.BRIGHTNESS] = value;
}
}
/// <summary>
/// ZED camera's saturation setting.
/// </summary>
public int Saturation
{
get
{
return settings[(int)sl.CAMERA_SETTINGS.SATURATION];
}
set
{
settings[(int)sl.CAMERA_SETTINGS.SATURATION] = value;
}
}
/// <summary>
/// ZED camera's hue setting.
/// </summary>
public int Hue
{
get
{
return settings[(int)sl.CAMERA_SETTINGS.HUE];
}
set
{
settings[(int)sl.CAMERA_SETTINGS.HUE] = value;
}
}
/// <summary>
/// ZED camera's contrast setting.
/// </summary>
public int Contrast
{
get
{
return settings[(int)sl.CAMERA_SETTINGS.CONTRAST];
}
set
{
settings[(int)sl.CAMERA_SETTINGS.CONTRAST] = value;
}
}
/// <summary>
/// ZED camera's gain setting. -1 means automatic.
/// </summary>
public int Gain
{
get
{
return settings[(int)sl.CAMERA_SETTINGS.GAIN];
}
set
{
settings[(int)sl.CAMERA_SETTINGS.GAIN] = value;
}
}
/// <summary>
/// ZED camera's exposure setting. -1 means automatic.
/// </summary>
public int Exposure
{
get
{
return settings[(int)sl.CAMERA_SETTINGS.EXPOSURE];
}
set
{
settings[(int)sl.CAMERA_SETTINGS.EXPOSURE] = value;
}
}
/// <summary>
/// ZED camera's white balance setting. -1 means automatic.
/// </summary>
public int WhiteBalance
{
get
{
return settings[(int)sl.CAMERA_SETTINGS.WHITEBALANCE];
}
set
{
settings[(int)sl.CAMERA_SETTINGS.WHITEBALANCE] = value;
}
}
/// <summary>
/// ZED camera's LED status
/// </summary>
public int LEDStatus
{
get
{
return settings[(int)sl.CAMERA_SETTINGS.LED_STATUS];
}
set
{
settings[(int)sl.CAMERA_SETTINGS.LED_STATUS] = value;
}
}
}
/// <summary>
/// Reference to the settings container object.
/// </summary>
private CameraSettings settings_;
/// <summary>
/// Reference to the settings container object.
/// </summary>
public CameraSettings Settings
{
get { return settings_.Clone(); }
}
/// <summary>
/// Whether exposure is set to automatic.
/// </summary>
public bool auto = true;
/// <summary>
/// Whether white balance is set to automatic.
/// </summary>
public bool whiteBalanceAuto = true;
/// <summary>
/// Constructor. Creates a new instance of CameraSettings to contain all settings values.
/// </summary>
public ZEDCameraSettings()
{
settings_ = new CameraSettings();
}
/// <summary>
/// Applies all settings from the container to the actual ZED camera.
/// </summary>
/// <param name="zedCamera">Current instance of the ZEDCamera wrapper.</param>
public void SetSettings(sl.ZEDCamera zedCamera)
{
if (zedCamera != null)
{
zedCamera.SetCameraSettings(sl.CAMERA_SETTINGS.BRIGHTNESS, settings_.Brightness, false);
zedCamera.SetCameraSettings(sl.CAMERA_SETTINGS.CONTRAST, settings_.Contrast, false);
zedCamera.SetCameraSettings(sl.CAMERA_SETTINGS.HUE, settings_.Hue, false);
zedCamera.SetCameraSettings(sl.CAMERA_SETTINGS.SATURATION, settings_.Saturation, false);
zedCamera.SetCameraSettings(sl.CAMERA_SETTINGS.GAIN, settings_.Gain, false);
zedCamera.SetCameraSettings(sl.CAMERA_SETTINGS.EXPOSURE, settings_.Exposure, false);
zedCamera.SetCameraSettings(sl.CAMERA_SETTINGS.LED_STATUS, settings_.LEDStatus, false);
if (settings_.WhiteBalance != -1)
{
zedCamera.SetCameraSettings(sl.CAMERA_SETTINGS.WHITEBALANCE, settings_.WhiteBalance, false);
}
}
}
/// <summary>
/// Applies all settings from the container to the actual ZED camera.
/// </summary>
/// <param name="zedCamera">Current instance of the ZEDCamera wrapper.</param>
public void ResetCameraSettings(sl.ZEDCamera zedCamera)
{
if (zedCamera != null)
{
zedCamera.SetCameraSettings(sl.CAMERA_SETTINGS.BRIGHTNESS, 0, true);
zedCamera.SetCameraSettings(sl.CAMERA_SETTINGS.CONTRAST, 0 , true);
zedCamera.SetCameraSettings(sl.CAMERA_SETTINGS.HUE, 0 , true);
zedCamera.SetCameraSettings(sl.CAMERA_SETTINGS.SATURATION, 0 , true);
zedCamera.SetCameraSettings(sl.CAMERA_SETTINGS.WHITEBALANCE, 0, true);
zedCamera.SetCameraSettings(sl.CAMERA_SETTINGS.EXPOSURE, 0, true);
zedCamera.SetCameraSettings(sl.CAMERA_SETTINGS.GAIN, 0, true);
zedCamera.SetCameraSettings(sl.CAMERA_SETTINGS.LED_STATUS, 1, true);
}
}
/// <summary>
/// Loads camera settings from a file, and sets them to the container and camera.
/// File is loaded from the root project folder (one above Assets).
/// </summary>
/// <param name="zedCamera"></param>
/// <param name="path"></param>
public void LoadCameraSettings(sl.ZEDCamera zedCamera, string path = "ZED_Settings.conf")
{
string[] lines = null;
try
{
lines = System.IO.File.ReadAllLines(path);
}
catch (System.Exception)
{
}
if (lines == null) return;
foreach (string line in lines)
{
string[] splittedLine = line.Split('=');
if (splittedLine.Length == 2)
{
string key = splittedLine[0];
string field = splittedLine[1];
if (key == "brightness")
{
settings_.Brightness = int.Parse(field);
}
else if (key == "contrast")
{
settings_.Contrast = int.Parse(field);
}
else if (key == "hue")
{
settings_.Hue = int.Parse(field);
}
else if (key == "saturation")
{
settings_.Saturation = int.Parse(field);
}
else if (key == "whiteBalance")
{
settings_.WhiteBalance = int.Parse(field);
}
else if (key == "gain")
{
settings_.Gain = int.Parse(field);
}
else if (key == "exposure")
{
settings_.Exposure = int.Parse(field);
}
else if (key == "LED")
{
settings_.LEDStatus = int.Parse(field);
}
}
}
SetSettings(zedCamera);
auto = (settings_.Exposure == -1);
whiteBalanceAuto = (settings_.WhiteBalance == -1);
}
/// <summary>
/// Retrieves current settings from the ZED camera.
/// </summary>
/// <param name="zedCamera"></param>
public void RetrieveSettingsCamera(sl.ZEDCamera zedCamera)
{
if (zedCamera != null)
{
settings_.Brightness = zedCamera.GetCameraSettings(sl.CAMERA_SETTINGS.BRIGHTNESS);
settings_.Contrast = zedCamera.GetCameraSettings(sl.CAMERA_SETTINGS.CONTRAST);
settings_.Hue = zedCamera.GetCameraSettings(sl.CAMERA_SETTINGS.HUE);
settings_.Saturation = zedCamera.GetCameraSettings(sl.CAMERA_SETTINGS.SATURATION);
settings_.Gain = zedCamera.GetCameraSettings(sl.CAMERA_SETTINGS.GAIN);
settings_.Exposure = zedCamera.GetCameraSettings(sl.CAMERA_SETTINGS.EXPOSURE);
settings_.WhiteBalance = zedCamera.GetCameraSettings(sl.CAMERA_SETTINGS.WHITEBALANCE);
settings_.LEDStatus = zedCamera.GetCameraSettings(sl.CAMERA_SETTINGS.LED_STATUS);
}
}
/// <summary>
/// Applies an individual setting to the ZED camera.
/// </summary>
/// <param name="settings">Setting to be changed (brightness, contrast, gain, exposure, etc.)</param>
/// <param name="value">New value for the setting.</param>
/// <param name="usedefault">If true, ignores the value and applies the default setting.</param>
public void SetCameraSettings(int cid, sl.CAMERA_SETTINGS settings, int value, bool usedefault = false)
{
settings_.settings[(int)settings] = !usedefault && value != -1 ? value : -1;
dllz_set_camera_settings(cid, (int)settings, value, System.Convert.ToInt32(usedefault));
}
/// <summary>
/// Gets the value from an individual ZED camera setting (brightness, contrast, gain, exposure, etc.)
/// </summary>
/// <param name="settings">Setting to be retrieved.</param>
/// <returns>Current value.</returns>
public int GetCameraSettings(int cid, sl.CAMERA_SETTINGS settings)
{
settings_.settings[(int)settings] = dllz_get_camera_settings(cid, (int)settings);
return settings_.settings[(int)settings];
}
/// <summary>
/// Saves all camera settings into a file into the specified path/name.
/// </summary>
/// <param name="path">Path and filename to save the file (ex. /Assets/ZED_Settings.conf)</param>
public void SaveCameraSettings(string path)
{
using (System.IO.StreamWriter file = new System.IO.StreamWriter(path))
{
file.WriteLine("brightness=" + settings_.Brightness.ToString());
file.WriteLine("contrast=" + settings_.Contrast.ToString());
file.WriteLine("hue=" + settings_.Hue.ToString());
file.WriteLine("saturation=" + settings_.Saturation.ToString());
file.WriteLine("whiteBalance=" + settings_.WhiteBalance.ToString());
file.WriteLine("gain=" + settings_.Gain.ToString());
file.WriteLine("exposure=" + settings_.Exposure.ToString());
file.WriteLine("LED=" + settings_.LEDStatus.ToString());
file.Close();
}
}
}