SCANsat.cs

/* 
 * Scientific Committee on Advanced Navigation S.C.A.N. Satellite
 * SCANsat - SCAN RADAR Altimetry Sensor part
 * 
 * Copyright (c)2013 damny; see LICENSE.txt for licensing details.
 */

using System;
using System.Collections.Generic;
using UnityEngine;

namespace SCANsat
{
	public class SCANsat : PartModule, IScienceDataContainer
	{
		protected SCANcontroller con = SCANcontroller.controller;
		protected bool initialized = false, powerIsProblem;
		protected Animation anim = null;
		protected List<ScienceData> storedData = new List<ScienceData>();
		protected ExperimentsResultDialog expDialog = null;

		public override void OnStart(StartState state) {
			if(state == StartState.Editor) {
				print("[SCANsat] start: in editor");
			} else {
				print("[SCANsat] start: live");
			}
			if(animationName != null) {
				Animation[] a = part.FindModelAnimators(animationName);
				if(a.Length == 0) {
					print("[SCANsat] animation '" + animationName + "' not found");
				} else {
					print("[SCANsat] using animation #1 out of " + a.Length.ToString() + " animations named '" + animationName + "'");
					anim = a[0];
					// maybe use this later for advanced animation...
					Transform modeltransform = part.transform.FindChild("model");
					foreach(Transform t in modeltransform.GetComponentsInChildren<Transform>()) {
						//print("[SCANsat] transform " + t.name + ": " + t);
					}

				}
			}
			print("[SCANsat] sensorType: " + sensorType.ToString() + " fov: " + fov.ToString() + " min_alt: " + min_alt.ToString() + " max_alt: " + max_alt.ToString() + " best_alt: " + best_alt.ToString() + " power: " + power.ToString());
		}

		[KSPField]
		public int sensorType;
		[KSPField]
		public float fov;
		[KSPField]
		public float min_alt;
		[KSPField]
		public float max_alt;
		[KSPField]
		public float best_alt;
		[KSPField]
		public float power;
		[KSPField]
		public string scanName;
		[KSPField]
		public string animationName;

		[KSPField(isPersistant = true)]
		protected bool scanning = false;

		[KSPEvent(guiActive = true, guiName = "Start RADAR Scan", active = true)]
		public void startScan() {
			scanning = true;
			if(sensorType > 0) {
				SCANcontroller.controller.registerSensor(vessel, (SCANdata.SCANtype)sensorType, fov, min_alt, max_alt, best_alt);
			}
			animate(1);
		}

		[KSPEvent(guiActive = true, guiName = "Stop RADAR Scan", active = true)]
		public void stopScan() {
			scanning = false;
			if(sensorType > 0) {
				SCANcontroller.controller.unregisterSensor(vessel, (SCANdata.SCANtype)sensorType);
			}
			animate(-1);
		}

		[KSPEvent(guiActiveEditor = true, guiName = "Extend", active = true)]
		public void editorExtend() {
			Events["editorExtend"].active = false;
			Events["editorRetract"].active = true;
			animate(1);
		}

		[KSPEvent(guiActiveEditor = true, guiName = "Retract", active = false)]
		public void editorRetract() {
			Events["editorExtend"].active = true;
			Events["editorRetract"].active = false;
			animate(-1);
		}

		public void animate(float speed) {
			if(anim != null && anim[animationName] != null) {
				anim[animationName].speed = speed;
				if(anim.IsPlaying(animationName)) {
					if(anim[animationName].normalizedTime <= 0) {
						anim[animationName].normalizedTime = 0;
					} else if(anim[animationName].normalizedTime >= 1 - float.Epsilon) {
						anim[animationName].normalizedTime = 1;
					}
				} else {
					anim[animationName].wrapMode = WrapMode.ClampForever;
					anim.Play(animationName);
				}
			}
		}

		[KSPEvent(guiActive = false, guiName = "Trigger Explosive Charge", active = false)]
		public void explode() {
			part.explode();
		}

		[KSPEvent(guiActive = true, guiName = "Analyze Data", active = false)]
		public void analyze() {
			makeScienceData(true);
			ReviewData();
		}

		[KSPAction("Start RADAR Scan")]
		public void startScanAction(KSPActionParam param) {
			startScan();
		}

		[KSPAction("Stop RADAR Scan")]
		public void stopScanAction(KSPActionParam param) {
			stopScan();
		}

		[KSPAction("Toggle RADAR Scan")]
		public void toggleScanAction(KSPActionParam param) {
			if(scanning) stopScan();
			else startScan();
		}

		[KSPAction("Analyze Data")]
		public void analyzeData(KSPActionParam param) {
			if(scanning) analyze();
		}

		public void addStatic() {
			SCANdata data = SCANcontroller.controller.getData(vessel.mainBody.bodyName);
			Texture2D map = data.map_small;
			if(map != null) {
				for(int i=0; i<1000; ++i) {
					map.SetPixel(UnityEngine.Random.Range(0, 360), UnityEngine.Random.Range(0, 180), Color.Lerp(Color.black, Color.white, UnityEngine.Random.value));
				}
			}
		}

		public override void OnUpdate() {
			if(sensorType < 0) {
				Events["startScan"].active = false;
				Events["stopScan"].active = false;
				Events["analyze"].active = false;
				Events["explode"].active = true;
				Events["explode"].guiActive = true;
				return;
			}
			if(!initialized) {
				if(sensorType == 0) {
					Events["startScan"].guiName = "Open Map";
					Events["stopScan"].guiName = "Close Map";
					Actions["startScanAction"].guiName = "Open Map";
					Actions["stopScanAction"].guiName = "Close Map";
					Actions["toggleScanAction"].guiName = "Toggle Map";
				} else if(scanName != null) {
					Events["startScan"].guiName = "Start " + scanName;
					Events["stopScan"].guiName = "Stop " + scanName;
					Actions["startScanAction"].guiName = "Start " + scanName;
					Actions["stopScanAction"].guiName = "Stop " + scanName;
					Actions["toggleScanAction"].guiName = "Toggle " + scanName;
				}
				if(scanning) {
					startScan();
				}
				initialized = true;
			}
			Events["startScan"].active = !scanning;
			Events["stopScan"].active = scanning;
			Events["analyze"].active = scanning;
			if(scanning && sensorType >= 0) {
				if(sensorType == 0 || SCANcontroller.controller.isVesselKnown(vessel.id, (SCANdata.SCANtype)sensorType)) {
					if(TimeWarp.CurrentRate < 1500) { // would need large buffer batteries, just not very smooth
						float p = power * TimeWarp.deltaTime;
						float e = part.RequestResource("ElectricCharge", p);
						if(e < p) {
							stopScan();
							powerIsProblem = true;
						} else {
							startScan();
							powerIsProblem = false;
						}
					} else if(powerIsProblem) {
						startScan();
						powerIsProblem = false;
					}
				} else {
					stopScan();
				}
			}
			SCANcontroller.controller.scanFromAllVessels();
			if(vessel == FlightGlobals.ActiveVessel) {
				SCANui.gui_ping(powerIsProblem);
				if(powerIsProblem) {
					addStatic();
					startScan();
				} else if(sensorType == 0 && scanning) {
					SCANui.gui_ping_maptraq();
				}
			}
		}

		public override string GetInfo() {
			string str = base.GetInfo();
			str += "Power usage: " + power.ToString("F1") + "/s\n";
			return str;
		}

		public void makeScienceData(bool notZero) {
			if(expDialog != null) DestroyImmediate(expDialog);
			storedData.Clear();
			ScienceData sd = SCANcontroller.controller.getAvailableScience(vessel, (SCANdata.SCANtype)sensorType, notZero);
			if(sd == null) return;
			storedData.Add(sd);
		}

		public ScienceData[] GetData() {
			return storedData.ToArray();
		}

		public void KeepData(ScienceData data) {
			print("[SCANsat] keeping data");
			expDialog = null;
		}
		
		public void TransmitData(ScienceData data) {
			print("[SCANsat] transmitting data");
			expDialog = null;
			if(!storedData.Contains(data)) return;
			foreach(IScienceDataTransmitter t in vessel.FindPartModulesImplementing<IScienceDataTransmitter>()) {
				if(t.CanTransmit()) {
					if(!t.IsBusy()) {
						makeScienceData(false); // just to update values...
						t.TransmitData(storedData);
						storedData = new List<ScienceData>();
						break;
					}
				}
			}
		}

		public void DumpData(ScienceData data) {
			print("[SCANsat] dumping data");
			expDialog = null;
			while(storedData.Contains(data)) {
				storedData.Remove(data);
			}
		}

		public void ReviewDataItem(ScienceData sd) {
			expDialog = ExperimentsResultDialog.DisplayResult(new ExperimentResultDialogPage(part, sd, 1f, 0f, false, "", true, false, DumpData, KeepData, TransmitData, null));
		}

		public void ReviewData() {
			if(storedData.Count < 1) return;
			if(expDialog != null) DestroyImmediate(expDialog);
			ScienceData sd = storedData[0];
			expDialog = ExperimentsResultDialog.DisplayResult(new ExperimentResultDialogPage(part, sd, 1f, 0f, false, "", true, false, DumpData, KeepData, TransmitData, null));
		}

		public bool IsRerunnable() {
			return false;
		}

		public int GetScienceCount() {
			return storedData.Count;
		}
	}
}