UnityEditor上でStartContinueっぽいのを動かす

UnityはEditor上でStartContinueでコルーチンが使えないのが結構不便だったので、作ってみた。

yield return 0;　//1フレーム待つ
とか
yield return new EditorCoroutine.WaitForSeconds(1f); //1秒待つ
とか

WWW www = new WWW ("http://　〜〜〜〜");
yield return www     //ダウンロード完了まで待つ;

といった、StartContinueでできることはだいたいできる。

使い方は
1, 以下のコードを書いたc#スクリプトをEditorフォルダにいれる
2, StartContinue(hoge());の代わりに、EditorCoroutine.Start(hoge());を呼ぶ（ hoge()はIEnumrator ）

要するにStartCorutineとほとんど同じです。

using UnityEngine;
using UnityEditor;
using System.Collections;
using System.Collections.Generic;
using System.Linq;

/// <summary>
/// エディター上でStartCorutineのような処理を使用可能にするクラスです 
/// </summary>
[UnityEditor.InitializeOnLoad]
public sealed class EditorCoroutine {

	static EditorCoroutine () {
		EditorApplication.update += Update;
		Debug.Log("EditorCoroutine SetUp");
	}

	static Dictionary<IEnumerator, EditorCoroutine.Coroutine> asyncList = new Dictionary<IEnumerator, Coroutine>();
	static List<EditorCoroutine.WaitForSeconds> waitForSecondsList = new List<EditorCoroutine.WaitForSeconds>();

	static void Update () {

		CheackIEnumerator ();
		CheackWaitForSeconds();
	}

	static void CheackIEnumerator () {
		List<IEnumerator> removeList = new List<IEnumerator>();
		foreach(KeyValuePair<IEnumerator, EditorCoroutine.Coroutine> pair in asyncList){
			if(pair.Key != null){

				//IEnumratorのCurrentがCoroutineを返しているかどうか 
				EditorCoroutine.Coroutine c = pair.Key.Current as EditorCoroutine.Coroutine;
				if(c != null){
					if(c.isActive) continue;
				}
				//wwwクラスのダウンロードが終わっていなければ進まない 
				WWW www = pair.Key.Current as WWW;
				if(www != null){
					if(!www.isDone) continue; 
				}
				//これ以上MoveNextできなければ終了 
				if(!pair.Key.MoveNext()){
					if(pair.Value != null){
						pair.Value.isActive = false;
					}
					removeList.Add(pair.Key);
				}
			}else {
				removeList.Add(pair.Key);
			}
		}

		foreach(IEnumerator async in removeList) {
			asyncList.Remove(async);
		}
	}

	static void CheackWaitForSeconds () {
		for(int i=0; i<waitForSecondsList.Count; i++){
			if(waitForSecondsList[i] != null){
				if(EditorApplication.timeSinceStartup - waitForSecondsList[i].InitTime > waitForSecondsList[i].Time){
					waitForSecondsList[i].isActive = false;
					waitForSecondsList.RemoveAt(i);
				}
			}else {
				Debug.LogError("rem");
				waitForSecondsList.RemoveAt(i);
			}
		}
	}

	//=====================================================================================
	//関数 

	/// <summary>
	/// コルーチンを起動します 
	/// </summary>
	static public EditorCoroutine.Coroutine Start (IEnumerator iEnumerator) {
		if(Application.isEditor && !Application.isPlaying){
			EditorCoroutine.Coroutine c = new Coroutine();
			if(!asyncList.Keys.Contains(iEnumerator)) asyncList.Add(iEnumerator, c);
			iEnumerator.MoveNext();
			return c;
		}else {
			Debug.LogError("EditorCoroutine.Startはゲーム起動中に使うことはできません");
			return null;
		}
	}

	/// <summary>
	/// コルーチンを停止します 
	/// </summary>
	static public void Stop (IEnumerator iEnumerator) {
		if(Application.isEditor){
			if(asyncList.Keys.Contains(iEnumerator)){
				asyncList.Remove(iEnumerator);
			}
		}else {
			Debug.LogError("EditorCoroutine.Startはゲーム中に使うことはできません");
		}
	}

	/// <summary>
	/// 使用不可
	/// WaitForSecondsのインスタンスを登録します 
	/// </summary>
	static public void AddWaitForSecondsList (EditorCoroutine.WaitForSeconds coroutine){
		if(waitForSecondsList.Contains(coroutine) == false){
			waitForSecondsList.Add(coroutine);
		}
	}


	//=====================================================================================
	//待機処理用クラス 

	public class Coroutine {
		//trueなら待機させる 
		public bool isActive;

		public Coroutine() {
			isActive = true;
		}
	}

	public sealed class WaitForSeconds : EditorCoroutine.Coroutine {
		private float time;
		private double initTime;

		public float Time {
			get{return time;}
		}
		public double InitTime {
			get{return initTime;}
		}

		public WaitForSeconds(float time) : base(){
			this.time = time;
			this.initTime = EditorApplication.timeSinceStartup;
			EditorCoroutine.AddWaitForSecondsList(this);
		}
	}
}

仕組みとしては、Start関数でIEnumratorの関数を登録しておいて、EditorApplication.updateのデリゲート登録したUpdate関数のループのなかで、IEnumratorのMoveNext関数を呼ぶかどうかをチェックしている感じです。

コルーチンが動いている最中ににスクリプトのコンパイルが入ったり、予期せぬ割り込みが入ってしまうとよくないかも…

注) AddWaitForSecondsListは使わないこと C#にはfriendとかないので仕方なくpublicになってます