EnumerablePartitioner.cs System::Collections::Concurrent::Partitioners::EnumerablePartitioner System::Collections::Concurrent::Partitioners::EnumerablePartitioner::PartitionerState System System::Collections System::Collections::Concurrent System::Collections::Concurrent::Partitioners // //EnumerablePartitioner.cs // //Author: //Jérémie"Garuma"Laval<jeremie.laval@gmail.com> // //Copyright(c)2009Jérémie"Garuma"Laval // //Permissionisherebygranted,freeofcharge,toanypersonobtainingacopy //ofthissoftwareandassociateddocumentationfiles(the"Software"),todeal //intheSoftwarewithoutrestriction,includingwithoutlimitationtherights //touse,copy,modify,merge,publish,distribute,sublicense,and/orsell //copiesoftheSoftware,andtopermitpersonstowhomtheSoftwareis //furnishedtodoso,subjecttothefollowingconditions: // //Theabovecopyrightnoticeandthispermissionnoticeshallbeincludedin //allcopiesorsubstantialportionsoftheSoftware. // //THESOFTWAREISPROVIDED"ASIS",WITHOUTWARRANTYOFANYKIND,EXPRESSOR //IMPLIED,INCLUDINGBUTNOTLIMITEDTOTHEWARRANTIESOFMERCHANTABILITY, //FITNESSFORAPARTICULARPURPOSEANDNONINFRINGEMENT.INNOEVENTSHALLTHE //AUTHORSORCOPYRIGHTHOLDERSBELIABLEFORANYCLAIM,DAMAGESOROTHER //LIABILITY,WHETHERINANACTIONOFCONTRACT,TORTOROTHERWISE,ARISINGFROM, //OUTOFORINCONNECTIONWITHTHESOFTWAREORTHEUSEOROTHERDEALINGSIN //THESOFTWARE. #ifNET_4_0||UNITY_5_3_OR_NEWER usingSystem; usingSystem.Threading; usingSystem.Threading.Tasks; usingSystem.Collections.Generic; namespaceSystem.Collections.Concurrent.Partitioners { //Representachunkpartitioner internalclassEnumerablePartitioner<T>:OrderablePartitioner<T> { IEnumerable<T>source; constintInitialPartitionSize=1; constintPartitionMultiplier=2; intinitialPartitionSize; intpartitionMultiplier; publicEnumerablePartitioner(IEnumerable<T>source) :this(source,InitialPartitionSize,PartitionMultiplier) { } //Thisisusedtogetstripedpartitionning(forTakeandSkipforinstance publicEnumerablePartitioner(IEnumerable<T>source,intinitialPartitionSize,intpartitionMultiplier) :base(true,false,true) { this.source=source; this.initialPartitionSize=initialPartitionSize; this.partitionMultiplier=partitionMultiplier; } publicoverrideIList<IEnumerator<KeyValuePair<long,T>>>GetOrderablePartitions(intpartitionCount) { if(partitionCount<=0) thrownewArgumentOutOfRangeException("partitionCount"); IEnumerator<KeyValuePair<long,T>>[]enumerators =newIEnumerator<KeyValuePair<long,T>>[partitionCount]; PartitionerStatestate=newPartitionerState(); IEnumerator<T>src=source.GetEnumerator(); boolisSimple=initialPartitionSize==1&&partitionMultiplier==1; for(inti=0;i<enumerators.Length;i++){ enumerators[i]=isSimple?GetPartitionEnumeratorSimple(src,state,i==enumerators.Length-1):GetPartitionEnumerator(src,state); } returnenumerators; } //Thispartitionerthatissimplerthanthegeneralcase(don'tusealist)iscalledinthecase //ofinitialPartitionSize==partitionMultiplier==1 IEnumerator<KeyValuePair<long,T>>GetPartitionEnumeratorSimple(IEnumerator<T>src, PartitionerStatestate, boollast) { longindex=-1; varvalue=default(T); try{ do{ lock(state.SyncLock){ if(state.Finished) break; if(state.Finished=!src.MoveNext()) break; index=state.Index++; value=src.Current; } yieldreturnnewKeyValuePair<long,T>(index,value); }while(!state.Finished); }finally{ if(last) src.Dispose(); } } IEnumerator<KeyValuePair<long,T>>GetPartitionEnumerator(IEnumerator<T>src,PartitionerStatestate) { intcount=initialPartitionSize; List<T>list=newList<T>(); while(!state.Finished){ list.Clear(); longind=-1; lock(state.SyncLock){ if(state.Finished) break; ind=state.Index; for(inti=0;i<count;i++){ if(state.Finished=!src.MoveNext()){ if(list.Count==0) yieldbreak; else break; } list.Add(src.Current); state.Index++; } } for(inti=0;i<list.Count;i++) yieldreturnnewKeyValuePair<long,T>(ind+i,list[i]); count*=partitionMultiplier; } } classPartitionerState { publicboolFinished; publiclongIndex=0; publicreadonlyobjectSyncLock=newobject(); } } } #endif