moonlight-android/app/src/main/java/com/limelight/nvstream/StreamConfiguration.java

package com.limelight.nvstream;

import com.limelight.nvstream.http.NvApp;
import com.limelight.nvstream.jni.MoonBridge;

public class StreamConfiguration {
	public static final int INVALID_APP_ID = 0;

	public static final int STREAM_CFG_LOCAL = 0;
	public static final int STREAM_CFG_REMOTE = 1;
	public static final int STREAM_CFG_AUTO = 2;

	private static final int CHANNEL_COUNT_STEREO = 2;
	private static final int CHANNEL_COUNT_5_1 = 6;
	
	private static final int CHANNEL_MASK_STEREO = 0x3;
	private static final int CHANNEL_MASK_5_1 = 0xFC;
	
	private NvApp app;
	private int width, height;
	private int refreshRate;
	private int clientRefreshRateX100;
	private int bitrate;
	private boolean sops;
	private boolean enableAdaptiveResolution;
	private boolean playLocalAudio;
	private int maxPacketSize;
	private int remote;
	private int audioChannelMask;
	private int audioChannelCount;
	private int audioConfiguration;
	private boolean supportsHevc;
	private int hevcBitratePercentageMultiplier;
	private boolean enableHdr;
	private int attachedGamepadMask;

	public static class Builder {
		private StreamConfiguration config = new StreamConfiguration();
		
		public StreamConfiguration.Builder setApp(NvApp app) {
			config.app = app;
			return this;
		}
		
		public StreamConfiguration.Builder setRemoteConfiguration(int remote) {
			config.remote = remote;
			return this;
		}
		
		public StreamConfiguration.Builder setResolution(int width, int height) {
			config.width = width;
			config.height = height;
			return this;
		}
		
		public StreamConfiguration.Builder setRefreshRate(int refreshRate) {
			config.refreshRate = refreshRate;
			return this;
		}
		
		public StreamConfiguration.Builder setBitrate(int bitrate) {
			config.bitrate = bitrate;
			return this;
		}
		
		public StreamConfiguration.Builder setEnableSops(boolean enable) {
			config.sops = enable;
			return this;
		}
		
		public StreamConfiguration.Builder enableAdaptiveResolution(boolean enable) {
			config.enableAdaptiveResolution = enable;
			return this;
		}
		
		public StreamConfiguration.Builder enableLocalAudioPlayback(boolean enable) {
			config.playLocalAudio = enable;
			return this;
		}
		
		public StreamConfiguration.Builder setMaxPacketSize(int maxPacketSize) {
			config.maxPacketSize = maxPacketSize;
			return this;
		}

		public StreamConfiguration.Builder setHevcBitratePercentageMultiplier(int multiplier) {
			config.hevcBitratePercentageMultiplier = multiplier;
			return this;
		}

		public StreamConfiguration.Builder setEnableHdr(boolean enableHdr) {
			config.enableHdr = enableHdr;
			return this;
		}

		public StreamConfiguration.Builder setAttachedGamepadMask(int attachedGamepadMask) {
			config.attachedGamepadMask = attachedGamepadMask;
			return this;
		}

		public StreamConfiguration.Builder setAttachedGamepadMaskByCount(int gamepadCount) {
			config.attachedGamepadMask = 0;
			for (int i = 0; i < 4; i++) {
				if (gamepadCount > i) {
					config.attachedGamepadMask |= 1 << i;
				}
			}
			return this;
		}

		public StreamConfiguration.Builder setClientRefreshRateX100(int refreshRateX100) {
			config.clientRefreshRateX100 = refreshRateX100;
			return this;
		}
		
		public StreamConfiguration.Builder setAudioConfiguration(int audioConfig) {
			if (audioConfig == MoonBridge.AUDIO_CONFIGURATION_STEREO) {
				config.audioChannelCount = CHANNEL_COUNT_STEREO;
				config.audioChannelMask = CHANNEL_MASK_STEREO;
			}
			else if (audioConfig == MoonBridge.AUDIO_CONFIGURATION_51_SURROUND) {
				config.audioChannelCount = CHANNEL_COUNT_5_1;
				config.audioChannelMask = CHANNEL_MASK_5_1;
			}
			else {
				throw new IllegalArgumentException("Invalid audio configuration");
			}

			config.audioConfiguration = audioConfig;

			return this;
		}
		
		public StreamConfiguration.Builder setHevcSupported(boolean supportsHevc) {
			config.supportsHevc = supportsHevc;
			return this;
		}
		
		public StreamConfiguration build() {
			return config;
		}
	}
	
	private StreamConfiguration() {
		// Set default attributes
		this.app = new NvApp("Steam");
		this.width = 1280;
		this.height = 720;
		this.refreshRate = 60;
		this.bitrate = 10000;
		this.maxPacketSize = 1024;
		this.remote = STREAM_CFG_AUTO;
		this.sops = true;
		this.enableAdaptiveResolution = false;
		this.audioChannelCount = CHANNEL_COUNT_STEREO;
		this.audioChannelMask = CHANNEL_MASK_STEREO;
		this.supportsHevc = false;
		this.enableHdr = false;
		this.attachedGamepadMask = 0;
	}
	
	public int getWidth() {
		return width;
	}
	
	public int getHeight() {
		return height;
	}
	
	public int getRefreshRate() {
		return refreshRate;
	}
	
	public int getBitrate() {
		return bitrate;
	}
	
	public int getMaxPacketSize() {
		return maxPacketSize;
	}

	public NvApp getApp() {
		return app;
	}
	
	public boolean getSops() {
		return sops;
	}
	
	public boolean getAdaptiveResolutionEnabled() {
		return enableAdaptiveResolution;
	}
	
	public boolean getPlayLocalAudio() {
		return playLocalAudio;
	}
	
	public int getRemote() {
		return remote;
	}
	
	public int getAudioChannelCount() {
		return audioChannelCount;
	}
	
	public int getAudioChannelMask() {
		return audioChannelMask;
	}

	public int getAudioConfiguration() {
		return audioConfiguration;
	}
	
	public boolean getHevcSupported() {
		return supportsHevc;
	}

	public int getHevcBitratePercentageMultiplier() {
		return hevcBitratePercentageMultiplier;
	}

	public boolean getEnableHdr() {
		return enableHdr;
	}

	public int getAttachedGamepadMask() {
		return attachedGamepadMask;
	}

	public int getClientRefreshRateX100() {
		return clientRefreshRateX100;
	}
}