FiberScheduler.cs SpicyPixel::Threading::FiberScheduler 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.Threading; namespaceSpicyPixel.Threading { publicabstractclassFiberScheduler:IFiberScheduler,IDisposable { [ThreadStatic] privatestaticFiberSchedulercurrentScheduler; CancellationTokenSourcecancelSource=newCancellationTokenSource(); publicCancellationTokenCancellationToken { get{returncancelSource.Token;} } publicstaticFiberSchedulerCurrent { get { if(currentScheduler==null) { //Makeanewscheduler SetCurrentScheduler(newSystemFiberScheduler(),true); } returncurrentScheduler; } //Privatefornow.Betternottoletthecallersetthisand //insteadletitbesetduringexecution.Thiswayany //newfibersinsidearunningfiberwillusethesamescheduler, //andthereisnoriskofthecallerchangingitoutfrom //undertheframeworkorwindupwithaschedulertheydidn't //expect.Forexample,acallerinUnitymightthinkthey //couldsetthecurrentschedulerinAwake()andusethe //sameschedulerinUpdate()eventhoughanotherMonoBehaviour //Awake()setittosomethingelsebecausetheyallshare //thesamethread.Thecurrentscheduleronathreadwill //getchangedoutautomaticallyduringTask.Execute() //invokedbyUpdate()invokedbyRun(). /*privateset { SetCurrentScheduler(value,false); }*/ } staticinternalvoidSetCurrentScheduler(FiberSchedulerscheduler,boolinternalInvoke) { //Onlyallowschedulerchangesonathreadwhenafiberisnotexecuting. //Ignorethischeckifthisisaninternalinvocationsuchasfrom //Fiber.Execute()whichneedstoswitchschedulersondemand. if(!internalInvoke&&Fiber.CurrentFiber!=null) thrownewInvalidOperationException("Thecurrentschedulerforthethreadcannotbechangedfrominsideanexecutingfiber."); //Skipworkifnothingtochange if(currentScheduler==scheduler) return; //Assignthescheduler currentScheduler=scheduler; //Updatethesynchronizationcontextifitchanged if(currentScheduler!=null&&SynchronizationContext.Current!=currentScheduler.SynchronizationContext) SynchronizationContext.SetSynchronizationContext(currentScheduler.SynchronizationContext); elseif(currentScheduler==null&&SynchronizationContext.Current!=null) SynchronizationContext.SetSynchronizationContext(null); } privateThreadschedulerThread; internalSynchronizationContextSynchronizationContext{get;privateset;} /* publicboolIsSchedulerThread { get{returnThread.CurrentThread==schedulerThread;} } */ publicThreadSchedulerThread { get{returnschedulerThread;} } publicboolAllowInlining{get;set;} protectedFiberScheduler() { AllowInlining=true; schedulerThread=Thread.CurrentThread; SynchronizationContext=newFiberSchedulerSynchronizationContext(this); } //UsedbyFibertoinvoketheprotectedmethods voidIFiberScheduler.QueueFiber(Fiberfiber) { this.QueueFiber(fiber); } publicvoidRun(Fiberfiber) { Run(fiber,CancellationToken.None,0f); } publicvirtualvoidRun(Fiberfiber,CancellationTokentoken,floatupdatesPerSecond=0f) { thrownewNotImplementedException(); } publicvoidRun() { Run(CancellationToken.None); } publicvirtualvoidRun(CancellationTokentoken,floatupdatesPerSecond=0f) { Run(null,token,updatesPerSecond); } protectedabstractvoidQueueFiber(Fiberfiber); protectedFiberInstructionExecuteFiber(Fiberfiber) { returnfiber.Execute(); } #regionIDisposableimplementation ~FiberScheduler() { Dispose(false); } privateboolisDisposed=false; publicvoidDispose() { Dispose(true); GC.SuppressFinalize(this); } protectedvirtualvoidDispose(booldisposing) { //Donothingifalreadycalled if(isDisposed) return; if(disposing) { //Freeotherstate(managedobjects). cancelSource.Cancel(); } //Freeyourownstate(unmanagedobjects). //Setlargefieldstonull. //Markdisposed isDisposed=true; } #endregion } }