UnityFiberScheduler.cs SpicyPixel::Threading::UnityFiberScheduler SpicyPixel SpicyPixel::Threading /* Author:AaronOneal,http://aarononeal.info Copyright(c)2012SpicyPixel,http://spicypixel.com Permissionisherebygranted,freeofcharge,toanypersonobtaining acopyofthissoftwareandassociateddocumentationfiles(the "Software"),todealintheSoftwarewithoutrestriction,including withoutlimitationtherightstouse,copy,modify,merge,publish, distribute,sublicense,and/orsellcopiesoftheSoftware,andto permitpersonstowhomtheSoftwareisfurnishedtodoso,subjectto thefollowingconditions: Theabovecopyrightnoticeandthispermissionnoticeshallbe includedinallcopiesorsubstantialportionsoftheSoftware. THESOFTWAREISPROVIDED"ASIS",WITHOUTWARRANTYOFANYKIND, EXPRESSORIMPLIED,INCLUDINGBUTNOTLIMITEDTOTHEWARRANTIESOF MERCHANTABILITY,FITNESSFORAPARTICULARPURPOSEAND NONINFRINGEMENT.INNOEVENTSHALLTHEAUTHORSORCOPYRIGHTHOLDERSBE LIABLEFORANYCLAIM,DAMAGESOROTHERLIABILITY,WHETHERINANACTION OFCONTRACT,TORTOROTHERWISE,ARISINGFROM,OUTOFORINCONNECTION WITHTHESOFTWAREORTHEUSEOROTHERDEALINGSINTHESOFTWARE. */ usingSystem; usingSystem.Collections; usingSystem.Collections.Generic; usingSystem.Collections.Concurrent; usingSystem.Threading; usingUnityEngine; namespaceSpicyPixel.Threading { publicsealedclassUnityFiberScheduler:FiberScheduler { internalconststringUnityCoroutineKey="spicypixel.threading.unity.coroutine"; privatestaticreadonlyUnityFiberSchedulerinstance=newUnityFiberScheduler(ConcurrentBehaviour.SharedInstance); publicstaticUnityFiberSchedulerDefault{ get{ returninstance; } } privateMonoBehaviourbehaviour; privateConcurrentQueue<Fiber>fiberQueue=newConcurrentQueue<Fiber>(); privateCancellationTokenSourcefiberQueueCancelSource=newCancellationTokenSource(); publicUnityFiberScheduler(MonoBehaviourbehaviour) { this.behaviour=behaviour; behaviour.StartCoroutine(ProcessFiberQueue()); } protectedoverridevoidQueueFiber(Fiberfiber) { if(isDisposed) thrownewObjectDisposedException(GetType().FullName); //Coroutinesarealwaysinlineduptotheir //firstyield,soenqueuingherelikelydoesn'tdo //anythingotherthandelayaframe.But,itdoes //ensurethecorrectthread. if(AllowInlining&&SchedulerThread==Thread.CurrentThread) StartUnityFiber(fiber); else fiberQueue.Enqueue(fiber); } privateIEnumeratorProcessFiberQueue() { Fiberfiber; while(!fiberQueueCancelSource.IsCancellationRequested) { while(!fiberQueueCancelSource.IsCancellationRequested&&fiberQueue.TryDequeue(outfiber)) StartUnityFiber(fiber); yieldreturnnull; } } privatevoidStartUnityFiber(Fiberfiber) { Coroutinecoroutine=behaviour.StartCoroutine(ExecuteFiberInternal(fiber)); fiber.Properties[UnityCoroutineKey]=coroutine; } privateIEnumeratorExecuteFiberInternal(Fiberfiber,boolsingleStep=false,intfiberSwitchCount=0) { FiberInstructionfiberInstruction=null; boolranOnce=false; while(!fiber.IsCompleted) { //Ifwearesettoonlyadvanceoneinstructionthen //abortifwehavealreadydonethat if(singleStep&&ranOnce) yieldbreak; ranOnce=true; //Executethefiber fiberInstruction=ExecuteFiber(fiber); //Nothingmoretodoifstopped if(fiberInstructionisStopInstruction) yieldbreak; //NotsupportedinUnity if(fiberInstructionisYieldToFiber) thrownewInvalidOperationException("YieldToFiberisnotsupportedbytheUnityscheduler."); //YieldtoanyfibermeanssendnulltotheUnityscheduler if(fiberInstructionisYieldToAnyFiber) { yieldreturnnull; continue; } //PassbackanyobjectsdirectlytotheUnityschedulersince //thesecouldbeUnityschedulercommands if(fiberInstructionisObjectInstruction) { yieldreturn((ObjectInstruction)fiberInstruction).Value; continue; } //ConvertframeworkwaitinstructiontoUnityinstruction if(fiberInstructionisYieldForSeconds) { yieldreturnnewWaitForSeconds(((YieldForSeconds)fiberInstruction).Seconds); continue; } //ConvertframeworkwaitinstructiontoUnityinstruction if(fiberInstructionisYieldUntilComplete) { //Yieldthecoroutinethatwasstoredwhentheinstructionwasstarted. yieldreturn((YieldUntilComplete)fiberInstruction).Fiber.Properties[UnityCoroutineKey]; continue; } } } publicoverridestringToString() { if(SchedulerThread==Thread.CurrentThread) returnstring.Format("[UnityFiberScheduler][{0}]",behaviour.ToString()); else returnbase.ToString(); } privateboolisDisposed=false; protectedoverridevoidDispose(booldisposing) { if(isDisposed) return; if(disposing) { fiberQueueCancelSource.Cancel(); Fiberfiber; while(fiberQueue.TryDequeue(outfiber)); } isDisposed=true; base.Dispose(disposing); } } }