ViewVC Help
View File | Revision Log | Show Annotations | Download File
/cvs/Coro/Coro/State.xs
(Generate patch)

Comparing Coro/Coro/State.xs (file contents):
Revision 1.82 by root, Mon Nov 20 22:26:16 2006 UTC vs.
Revision 1.87 by root, Fri Nov 24 13:40:36 2006 UTC

145 I32 retstack_max; 145 I32 retstack_max;
146 PMOP *curpm; 146 PMOP *curpm;
147 COP *curcop; 147 COP *curcop;
148 JMPENV *top_env; 148 JMPENV *top_env;
149 149
150 /* coro process data */
151 int prio;
152
150 /* data associated with this coroutine (initial args) */ 153 /* data associated with this coroutine (initial args) */
151 AV *args; 154 AV *args;
155 int refcnt;
152}; 156};
153 157
154typedef struct coro *Coro__State; 158typedef struct coro *Coro__State;
155typedef struct coro *Coro__State_or_hashref; 159typedef struct coro *Coro__State_or_hashref;
156 160
667 CALLRUNOPS(aTHX); 671 CALLRUNOPS(aTHX);
668 672
669 abort (); 673 abort ();
670} 674}
671 675
676/* never call directly, always through the coro_state_transfer global variable */
672static void 677static void
673transfer (pTHX_ struct coro *prev, struct coro *next, int flags) 678transfer_impl (pTHX_ struct coro *prev, struct coro *next, int flags)
674{ 679{
675 dSTACKLEVEL; 680 dSTACKLEVEL;
676 681
677 if (prev != next) 682 if (prev != next)
678 { 683 {
771 coro_mortal = 0; 776 coro_mortal = 0;
772 } 777 }
773 UNLOCK; 778 UNLOCK;
774} 779}
775 780
781/* use this function pointer to call the above function */
782/* this is done to increase chances of the compiler not inlining the call */
783void (*coro_state_transfer)(pTHX_ struct coro *prev, struct coro *next, int flags) = transfer_impl;
784
776#define SV_CORO(sv,func) \ 785#define SV_CORO(sv,func) \
777 do { \ 786 do { \
778 if (SvROK (sv)) \ 787 if (SvROK (sv)) \
779 sv = SvRV (sv); \ 788 sv = SvRV (sv); \
780 \ 789 \
781 if (SvTYPE (sv) == SVt_PVHV) \ 790 if (SvTYPE (sv) == SVt_PVHV && SvSTASH (sv) != coro_state_stash) \
782 { \ 791 { \
783 HE *he = hv_fetch_ent ((HV *)sv, ucoro_state_sv, 0, ucoro_state_hash); \ 792 HE *he = hv_fetch_ent ((HV *)sv, ucoro_state_sv, 0, ucoro_state_hash); \
784 \ 793 \
785 if (!he) \ 794 if (!he) \
786 croak ("%s() -- %s is a hashref but lacks the " UCORO_STATE " key", func, # sv); \ 795 croak ("%s() -- %s is a hashref but lacks the " UCORO_STATE " key", func, # sv); \
792 if (!SvOBJECT (sv) || SvSTASH (sv) != coro_state_stash) \ 801 if (!SvOBJECT (sv) || SvSTASH (sv) != coro_state_stash) \
793 croak ("%s() -- %s is not (and contains not) a Coro::State object", func, # sv); \ 802 croak ("%s() -- %s is not (and contains not) a Coro::State object", func, # sv); \
794 \ 803 \
795 } while(0) 804 } while(0)
796 805
797#define SvSTATE(sv) INT2PTR (struct coro *, SvIV (sv)) 806static void
807coro_state_destroy (struct coro *coro)
808{
809 if (coro->refcnt--)
810 return;
798 811
812 if (coro->mainstack && coro->mainstack != main_mainstack)
813 {
814 struct coro temp;
815
816 SAVE (aTHX_ (&temp), TRANSFER_SAVE_ALL);
817 LOAD (aTHX_ coro);
818
819 destroy_stacks (aTHX);
820
821 LOAD ((&temp)); /* this will get rid of defsv etc.. */
822
823 coro->mainstack = 0;
824 }
825
826 deallocate_stack (coro);
827 SvREFCNT_dec (coro->args);
828 Safefree (coro);
829}
830
799static void 831static int
832coro_state_clear (SV *sv, MAGIC *mg)
833{
834 struct coro *coro = (struct coro *)mg->mg_ptr;
835 mg->mg_ptr = 0;
836
837 coro_state_destroy (coro);
838
839 return 0;
840}
841
842static int
843coro_state_dup (MAGIC *mg, CLONE_PARAMS *params)
844{
845 struct coro *coro = (struct coro *)mg->mg_ptr;
846
847 ++coro->refcnt;
848
849 return 0;
850}
851
852static MGVTBL coro_state_vtbl = { 0, 0, 0, 0, coro_state_clear, 0, coro_state_dup, 0 };
853
854static struct coro *
855SvSTATE (SV *coro)
856{
857 MAGIC *mg = SvMAGIC (SvROK (coro) ? SvRV (coro) : coro);
858 assert (mg->mg_type == PERL_MAGIC_ext);
859 return (struct coro *)mg->mg_ptr;
860}
861
862static void
800api_transfer(pTHX_ SV *prev, SV *next, int flags) 863api_transfer (pTHX_ SV *prev, SV *next, int flags)
801{ 864{
802 SV_CORO (prev, "Coro::transfer"); 865 SV_CORO (prev, "Coro::transfer");
803 SV_CORO (next, "Coro::transfer"); 866 SV_CORO (next, "Coro::transfer");
804 867
805 transfer (aTHX_ SvSTATE (prev), SvSTATE (next), flags); 868 coro_state_transfer (aTHX_ SvSTATE (prev), SvSTATE (next), flags);
806} 869}
807 870
808/** Coro ********************************************************************/ 871/** Coro ********************************************************************/
809 872
810#define PRIO_MAX 3 873#define PRIO_MAX 3
814#define PRIO_IDLE -3 877#define PRIO_IDLE -3
815#define PRIO_MIN -4 878#define PRIO_MIN -4
816 879
817/* for Coro.pm */ 880/* for Coro.pm */
818static GV *coro_current, *coro_idle; 881static GV *coro_current, *coro_idle;
819static AV *coro_ready[PRIO_MAX-PRIO_MIN+1]; 882static AV *coro_ready [PRIO_MAX-PRIO_MIN+1];
820static int coro_nready; 883static int coro_nready;
821 884
822static void 885static void
823coro_enq (pTHX_ SV *sv) 886coro_enq (pTHX_ SV *sv)
824{ 887{
825 SV **xprio;
826 int prio; 888 int prio;
827 889
828 if (SvTYPE (sv) != SVt_PVHV) 890 if (SvTYPE (sv) != SVt_PVHV)
829 croak ("Coro::ready tried to enqueue something that is not a coroutine"); 891 croak ("Coro::ready tried to enqueue something that is not a coroutine");
830 892
831 xprio = hv_fetch ((HV *)sv, "prio", 4, 0); 893 {
832 prio = xprio ? SvIV (*xprio) : PRIO_NORMAL; 894 SV *coro = sv;
833 895 SV_CORO (coro, "omg");
834 prio = prio > PRIO_MAX ? PRIO_MAX 896 prio = SvSTATE (coro)->prio;
835 : prio < PRIO_MIN ? PRIO_MIN 897 }
836 : prio;
837 898
838 av_push (coro_ready [prio - PRIO_MIN], sv); 899 av_push (coro_ready [prio - PRIO_MIN], sv);
839 coro_nready++; 900 coro_nready++;
840} 901}
841 902
847 min_prio -= PRIO_MIN; 908 min_prio -= PRIO_MIN;
848 if (min_prio < 0) 909 if (min_prio < 0)
849 min_prio = 0; 910 min_prio = 0;
850 911
851 for (prio = PRIO_MAX - PRIO_MIN + 1; --prio >= min_prio; ) 912 for (prio = PRIO_MAX - PRIO_MIN + 1; --prio >= min_prio; )
852 if (AvFILLp (coro_ready[prio]) >= 0) 913 if (AvFILLp (coro_ready [prio]) >= 0)
853 { 914 {
854 coro_nready--; 915 coro_nready--;
855 return av_shift (coro_ready[prio]); 916 return av_shift (coro_ready [prio]);
856 } 917 }
857 918
858 return 0; 919 return 0;
859} 920}
860 921
894 955
895 SV_CORO (next, "Coro::schedule"); 956 SV_CORO (next, "Coro::schedule");
896 957
897 UNLOCK; 958 UNLOCK;
898 959
899 transfer (aTHX_ SvSTATE (prev), SvSTATE (next), 960 coro_state_transfer (aTHX_ SvSTATE (prev), SvSTATE (next),
900 TRANSFER_SAVE_ALL | TRANSFER_LAZY_STACK); 961 TRANSFER_SAVE_ALL | TRANSFER_LAZY_STACK);
901} 962}
902 963
903static void 964static void
904api_cede (void) 965api_cede (void)
905{ 966{
912 api_schedule (); 973 api_schedule ();
913} 974}
914 975
915MODULE = Coro::State PACKAGE = Coro::State 976MODULE = Coro::State PACKAGE = Coro::State
916 977
917PROTOTYPES: ENABLE 978PROTOTYPES: DISABLE
918 979
919BOOT: 980BOOT:
920{ /* {} necessary for stoopid perl-5.6.x */ 981{ /* {} necessary for stoopid perl-5.6.x */
921#ifdef USE_ITHREADS 982#ifdef USE_ITHREADS
922 MUTEX_INIT (&coro_mutex); 983 MUTEX_INIT (&coro_mutex);
934 995
935 main_mainstack = PL_mainstack; 996 main_mainstack = PL_mainstack;
936 997
937 coroapi.ver = CORO_API_VERSION; 998 coroapi.ver = CORO_API_VERSION;
938 coroapi.transfer = api_transfer; 999 coroapi.transfer = api_transfer;
939}
940 1000
941Coro::State 1001 assert (("PRIO_NORMAL must be 0", !PRIO_NORMAL));
942_newprocess(args) 1002}
943 SV * args 1003
944 PROTOTYPE: $ 1004SV *
1005new (char *klass, ...)
945 CODE: 1006 CODE:
946 Coro__State coro; 1007{
1008 struct coro *coro;
1009 HV *hv;
1010 int i;
947 1011
948 if (!SvROK (args) || SvTYPE (SvRV (args)) != SVt_PVAV)
949 croak ("Coro::State::_newprocess expects an arrayref");
950
951 Newz (0, coro, 1, struct coro); 1012 Newz (0, coro, 1, struct coro);
1013 coro->args = newAV ();
952 1014
953 coro->args = (AV *)SvREFCNT_inc (SvRV (args)); 1015 for (i = 1; i < items; i++)
1016 av_push (coro->args, newSVsv (ST (i)));
1017
954 /*coro->mainstack = 0; *//*actual work is done inside transfer */ 1018 /*coro->mainstack = 0; *//*actual work is done inside transfer */
955 /*coro->stack = 0;*/ 1019 /*coro->stack = 0;*/
956 1020
957 RETVAL = coro; 1021 hv = newHV ();
1022 sv_magicext ((SV *)hv, 0, PERL_MAGIC_ext, &coro_state_vtbl, (char *)coro, 0)->mg_flags |= MGf_DUP;
1023 RETVAL = sv_bless (newRV_noinc ((SV *)hv), gv_stashpv (klass, 1));
1024}
958 OUTPUT: 1025 OUTPUT:
959 RETVAL 1026 RETVAL
960 1027
961void 1028void
962transfer(prev, next, flags) 1029transfer(prev, next, flags)
963 SV *prev 1030 SV *prev
964 SV *next 1031 SV *next
965 int flags 1032 int flags
966 PROTOTYPE: @
967 CODE: 1033 CODE:
968 PUTBACK; 1034 PUTBACK;
969 SV_CORO (next, "Coro::transfer"); 1035 SV_CORO (next, "Coro::transfer");
970 SV_CORO (prev, "Coro::transfer"); 1036 SV_CORO (prev, "Coro::transfer");
971 transfer (aTHX_ SvSTATE (prev), SvSTATE (next), flags); 1037 coro_state_transfer (aTHX_ SvSTATE (prev), SvSTATE (next), flags);
972 SPAGAIN; 1038 SPAGAIN;
973 1039
1040int
1041prio (Coro::State coro, int newprio = 0)
1042 ALIAS:
1043 nice = 1
1044 CODE:
1045{
1046 RETVAL = coro->prio;
1047
1048 if (items > 1)
1049 {
1050 if (ix)
1051 newprio += coro->prio;
1052
1053 if (newprio < PRIO_MIN) newprio = PRIO_MIN;
1054 if (newprio > PRIO_MAX) newprio = PRIO_MAX;
1055
1056 coro->prio = newprio;
1057 }
1058}
1059
974void 1060void
975DESTROY(coro) 1061_clone_state_from (SV *dst, SV *src)
976 Coro::State coro 1062 CODE:
977 CODE: 1063{
1064 struct coro *coro_src = SvSTATE (src);
978 1065
979 if (coro->mainstack && coro->mainstack != main_mainstack) 1066 sv_unmagic (SvRV (dst), PERL_MAGIC_ext);
980 {
981 struct coro temp;
982 1067
983 PUTBACK; 1068 ++coro_src->refcnt;
984 SAVE (aTHX_ (&temp), TRANSFER_SAVE_ALL); 1069 sv_magicext (SvRV (dst), 0, PERL_MAGIC_ext, &coro_state_vtbl, (char *)coro_src, 0)->mg_flags |= MGf_DUP;
985 LOAD (aTHX_ coro); 1070}
986 SPAGAIN;
987
988 destroy_stacks (aTHX);
989
990 LOAD ((&temp)); /* this will get rid of defsv etc.. */
991 SPAGAIN;
992
993 coro->mainstack = 0;
994 }
995
996 deallocate_stack (coro);
997 SvREFCNT_dec (coro->args);
998 Safefree (coro);
999 1071
1000void 1072void
1001_exit(code) 1073_exit (code)
1002 int code 1074 int code
1003 PROTOTYPE: $ 1075 PROTOTYPE: $
1004 CODE: 1076 CODE:
1005 _exit (code); 1077 _exit (code);
1006 1078
1007MODULE = Coro::State PACKAGE = Coro::Cont 1079MODULE = Coro::State PACKAGE = Coro::Cont
1008 1080
1009void 1081void
1010yield(...) 1082yield (...)
1011 PROTOTYPE: @ 1083 PROTOTYPE: @
1012 CODE: 1084 CODE:
1013 SV *yieldstack; 1085 SV *yieldstack;
1014 SV *sv; 1086 SV *sv;
1015 AV *defav = GvAV (PL_defgv); 1087 AV *defav = GvAV (PL_defgv);
1026 av_fill (defav, items - 1); 1098 av_fill (defav, items - 1);
1027 while (items--) 1099 while (items--)
1028 av_store (defav, items, SvREFCNT_inc (ST(items))); 1100 av_store (defav, items, SvREFCNT_inc (ST(items)));
1029 1101
1030 sv = av_pop ((AV *)SvRV (yieldstack)); 1102 sv = av_pop ((AV *)SvRV (yieldstack));
1031 prev = INT2PTR (struct coro *, SvIV ((SV*)SvRV (*av_fetch ((AV *)SvRV (sv), 0, 0)))); 1103 prev = SvSTATE ((SV*)SvRV (*av_fetch ((AV *)SvRV (sv), 0, 0)));
1032 next = INT2PTR (struct coro *, SvIV ((SV*)SvRV (*av_fetch ((AV *)SvRV (sv), 1, 0)))); 1104 next = SvSTATE ((SV*)SvRV (*av_fetch ((AV *)SvRV (sv), 1, 0)));
1033 SvREFCNT_dec (sv); 1105 SvREFCNT_dec (sv);
1034 1106
1035 transfer (aTHX_ prev, next, 0); 1107 coro_state_transfer (aTHX_ prev, next, 0);
1036 1108
1037MODULE = Coro::State PACKAGE = Coro 1109MODULE = Coro::State PACKAGE = Coro
1038 1110
1039BOOT: 1111BOOT:
1040{ 1112{
1070} 1142}
1071 1143
1072#if !PERL_MICRO 1144#if !PERL_MICRO
1073 1145
1074void 1146void
1075ready(self) 1147ready (self)
1076 SV * self 1148 SV * self
1077 PROTOTYPE: $ 1149 PROTOTYPE: $
1078 CODE: 1150 CODE:
1079 api_ready (self); 1151 api_ready (self);
1080 1152
1081#endif 1153#endif
1082 1154
1083int 1155int
1084nready(...) 1156nready (...)
1085 PROTOTYPE: 1157 PROTOTYPE:
1086 CODE: 1158 CODE:
1087 RETVAL = coro_nready; 1159 RETVAL = coro_nready;
1088 OUTPUT: 1160 OUTPUT:
1089 RETVAL 1161 RETVAL
1090 1162
1091void 1163void
1092schedule(...) 1164schedule (...)
1093 PROTOTYPE: 1165 PROTOTYPE:
1094 CODE: 1166 CODE:
1095 api_schedule (); 1167 api_schedule ();
1096 1168
1097void 1169void
1098cede(...) 1170cede (...)
1099 PROTOTYPE: 1171 PROTOTYPE:
1100 CODE: 1172 CODE:
1101 api_cede (); 1173 api_cede ();
1102 1174
1103# and these are hacks 1175# and these are hacks

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines