Coro/Event/Event.xs

#include "EXTERN.h"
#include "perl.h"
#include "XSUB.h"

#include <assert.h>
#include <string.h>

#include "EventAPI.h"
#include "../Coro/CoroAPI.h"

#define CD_CORO 0
#define CD_TYPE 1
#define CD_OK   2
#define CD_PRIO 3 /* hardcoded in Coro::Event */
#define CD_HITS 4 /* hardcoded in Coro::Event */
#define CD_GOT  5 /* hardcoded in Coro::Event, Coro::Handle */
#define CD_MAX  5

static void
coro_std_cb (pe_event *pe)
{
  AV *priv = (AV *)pe->ext_data;
  IV type = SvIV (AvARRAY (priv)[CD_TYPE]);
  SV **cd_coro;

  SvIV_set (AvARRAY (priv)[CD_PRIO], pe->prio);
  SvIV_set (AvARRAY (priv)[CD_HITS], pe->hits);

  if (type == 1)
    SvIV_set (AvARRAY (priv)[CD_GOT], ((pe_ioevent *)pe)->got);

  GEventAPI->stop (pe->up, 0);

  AvARRAY (priv)[CD_OK] = &PL_sv_yes;

  cd_coro = &AvARRAY(priv)[CD_CORO];
  if (*cd_coro != &PL_sv_undef)
    {
      AvARRAY (priv)[CD_OK] = &PL_sv_yes;
      CORO_READY (*cd_coro);
      SvREFCNT_dec (*cd_coro);
      *cd_coro = &PL_sv_undef;
    }
}

static void
asynccheck_hook (void *data)
{
  /* ceding from C means allocating a stack, but we assume this is a rare case */
  while (CORO_NREADY)
    CORO_CEDE;
}

MODULE = Coro::Event                PACKAGE = Coro::Event

PROTOTYPES: ENABLE

BOOT:
{
        I_EVENT_API ("Coro::Event");
        I_CORO_API ("Coro::Event");

        GEventAPI->add_hook ("asynccheck", (void *)asynccheck_hook, 0);
}

void
_install_std_cb (SV *self, int type)
        CODE:
{
        pe_watcher *w = GEventAPI->sv_2watcher (self);

        if (w->callback)
          croak ("Coro::Event watchers must not have a callback (see Coro::Event), caught");

        {
          AV *priv = newAV ();
          SV *rv = newRV_noinc ((SV *)priv);

          av_extend (priv, CD_MAX);
          AvARRAY (priv)[CD_CORO] = &PL_sv_undef;
          AvARRAY (priv)[CD_TYPE] = newSViv (type);
          AvARRAY (priv)[CD_OK  ] = &PL_sv_no;
          AvARRAY (priv)[CD_PRIO] = newSViv (0);
          AvARRAY (priv)[CD_HITS] = newSViv (0);
          AvARRAY (priv)[CD_GOT ] = newSViv (0);
          SvREADONLY_on (priv);

          w->callback = coro_std_cb;
          w->ext_data = priv;

          /* make sure Event does not use PERL_MAGIC_uvar, which */
          /* we abuse for non-uvar purposes. */
          sv_magicext (SvRV (self), rv, PERL_MAGIC_uvar, 0, 0, 0);
        }
}

void
_next (SV *self)
        CODE:
{
        pe_watcher *w = GEventAPI->sv_2watcher (self);
        AV *priv = (AV *)w->ext_data;

        if (AvARRAY (priv)[CD_OK] == &PL_sv_yes)
          {
            AvARRAY (priv)[CD_OK] = &PL_sv_no;
            XSRETURN_NO; /* got an event */
          }

        if (!w->running)
          GEventAPI->start (w, 1);

        if (AvARRAY (priv)[CD_CORO] == &PL_sv_undef)
          AvARRAY (priv)[CD_CORO] = SvREFCNT_inc (CORO_CURRENT);
        else if (AvARRAY (priv)[CD_CORO] != CORO_CURRENT)
          croak ("Coro::Event::next can only be called from a single coroutine at a time, caught");

        XSRETURN_YES; /* schedule */
}

Revision:	1.13
Committed:	Fri Dec 1 19:41:06 2006 UTC (17 years, 6 months ago) by root
Branch:	MAIN
Changes since 1.12:	+34 -54 lines
Log Message:	* empty log message *
#	User	Rev	Content
1	root	1.1	#include "EXTERN.h"
2			#include "perl.h"
3			#include "XSUB.h"
4
5	root	1.13	#include <assert.h>
6	root	1.3	#include <string.h>
7
8	root	1.2	#include "EventAPI.h"
9			#include "../Coro/CoroAPI.h"
10	root	1.1
11	root	1.3	#define CD_CORO 0
12			#define CD_TYPE 1
13	root	1.4	#define CD_OK 2
14	root	1.6	#define CD_PRIO 3 /* hardcoded in Coro::Event */
15			#define CD_HITS 4 /* hardcoded in Coro::Event */
16			#define CD_GOT 5 /* hardcoded in Coro::Event, Coro::Handle */
17			#define CD_MAX 5
18	root	1.3
19			static void
20	root	1.13	coro_std_cb (pe_event *pe)
21	root	1.3	{
22	root	1.6	AV priv = (AV )pe->ext_data;
23	root	1.13	IV type = SvIV (AvARRAY (priv)[CD_TYPE]);
24	root	1.12	SV **cd_coro;
25	root	1.6
26	root	1.13	SvIV_set (AvARRAY (priv)[CD_PRIO], pe->prio);
27			SvIV_set (AvARRAY (priv)[CD_HITS], pe->hits);
28	root	1.3
29			if (type == 1)
30	root	1.13	SvIV_set (AvARRAY (priv)[CD_GOT], ((pe_ioevent *)pe)->got);
31	root	1.3
32	root	1.12	GEventAPI->stop (pe->up, 0);
33
34	root	1.13	AvARRAY (priv)[CD_OK] = &PL_sv_yes;
35	root	1.12
36			cd_coro = &AvARRAY(priv)[CD_CORO];
37	root	1.6	if (*cd_coro != &PL_sv_undef)
38	root	1.3	{
39	root	1.13	AvARRAY (priv)[CD_OK] = &PL_sv_yes;
40	root	1.6	CORO_READY (*cd_coro);
41			SvREFCNT_dec (*cd_coro);
42			*cd_coro = &PL_sv_undef;
43	root	1.3	}
44			}
45
46	root	1.13	static void
47			asynccheck_hook (void *data)
48	root	1.3	{
49	root	1.13	/* ceding from C means allocating a stack, but we assume this is a rare case */
50	root	1.3	while (CORO_NREADY)
51			CORO_CEDE;
52			}
53
54	root	1.1	MODULE = Coro::Event PACKAGE = Coro::Event
55
56	root	1.3	PROTOTYPES: ENABLE
57
58	root	1.1	BOOT:
59			{
60	root	1.9	I_EVENT_API ("Coro::Event");
61	root	1.1	I_CORO_API ("Coro::Event");
62	root	1.3
63	root	1.13	GEventAPI->add_hook ("asynccheck", (void *)asynccheck_hook, 0);
64	root	1.1	}
65	root	1.3
66			void
67	root	1.13	_install_std_cb (SV *self, int type)
68	root	1.3	CODE:
69	root	1.12	{
70	root	1.3	pe_watcher *w = GEventAPI->sv_2watcher (self);
71
72	root	1.13	if (w->callback)
73			croak ("Coro::Event watchers must not have a callback (see Coro::Event), caught");
74
75	root	1.9	{
76			AV *priv = newAV ();
77			SV rv = newRV_noinc ((SV )priv);
78
79			av_extend (priv, CD_MAX);
80	root	1.13	AvARRAY (priv)[CD_CORO] = &PL_sv_undef;
81			AvARRAY (priv)[CD_TYPE] = newSViv (type);
82			AvARRAY (priv)[CD_OK ] = &PL_sv_no;
83			AvARRAY (priv)[CD_PRIO] = newSViv (0);
84			AvARRAY (priv)[CD_HITS] = newSViv (0);
85			AvARRAY (priv)[CD_GOT ] = newSViv (0);
86	root	1.9	SvREADONLY_on (priv);
87
88			w->callback = coro_std_cb;
89			w->ext_data = priv;
90
91	root	1.13	/* make sure Event does not use PERL_MAGIC_uvar, which */
92			/* we abuse for non-uvar purposes. */
93			sv_magicext (SvRV (self), rv, PERL_MAGIC_uvar, 0, 0, 0);
94	root	1.9	}
95	root	1.12	}
96	root	1.8
97	root	1.3	void
98	root	1.13	_next (SV *self)
99	root	1.3	CODE:
100	root	1.12	{
101	root	1.3	pe_watcher *w = GEventAPI->sv_2watcher (self);
102	root	1.6	AV priv = (AV )w->ext_data;
103	root	1.3
104	root	1.13	if (AvARRAY (priv)[CD_OK] == &PL_sv_yes)
105	root	1.4	{
106	root	1.13	AvARRAY (priv)[CD_OK] = &PL_sv_no;
107			XSRETURN_NO; /* got an event */
108	root	1.4	}
109	root	1.6
110	root	1.12	if (!w->running)
111			GEventAPI->start (w, 1);
112
113	root	1.13	if (AvARRAY (priv)[CD_CORO] == &PL_sv_undef)
114			AvARRAY (priv)[CD_CORO] = SvREFCNT_inc (CORO_CURRENT);
115			else if (AvARRAY (priv)[CD_CORO] != CORO_CURRENT)
116			croak ("Coro::Event::next can only be called from a single coroutine at a time, caught");
117	root	1.12
118	root	1.13	XSRETURN_YES; /* schedule */
119	root	1.12	}
120	root	1.6