X-Git-Url: http://git.shadowcat.co.uk/gitweb/gitweb.cgi?a=blobdiff_plain;f=Parameters.xs;h=1a4e58ec372efa51d3eb64fb8763f0c1438ae953;hb=0e5931eaf95baf2580df7bb8cb584dd2377a8763;hp=5372f04024cbb6c9d7cd1f5c7f9e73d919bc2ab5;hpb=abccbe865a63e6620c5f24d8cbd7872247e68b1b;p=p5sagit%2FFunction-Parameters.git

diff --git a/Parameters.xs b/Parameters.xs
index 5372f04..1a4e58e 100644
--- a/Parameters.xs
+++ b/Parameters.xs
@@ -221,6 +221,7 @@ static int parse_fun(pTHX_ OP **pop, const char *keyword_ptr, STRLEN keyword_len
 	I32 floor_ix;
 	int save_ix;
 	SV *saw_name;
+	OP **prelude_sentinel;
 	AV *params;
 	DefaultParamSpec *defaults;
 	int args_min, args_max;
@@ -276,12 +277,27 @@ static int parse_fun(pTHX_ OP **pop, const char *keyword_ptr, STRLEN keyword_len
 	/* create outer block: '{' */
 	save_ix = S_block_start(aTHX_ TRUE);
 
+	/* initialize synthetic optree */
+	Newx(prelude_sentinel, 1, OP *);
+	*prelude_sentinel = NULL;
+	SAVEDESTRUCTOR_X(free_ptr_op, prelude_sentinel);
+
 	/* parameters */
 	params = NULL;
 	defaults = NULL;
 	args_min = 0;
 	args_max = -1;
 
+	/* my $self; */
+	if (SvTRUE(spec->shift)) {
+		OP *var;
+
+		var = newOP(OP_PADSV, OPf_MOD | (OPpLVAL_INTRO << 8));
+		var->op_targ = pad_add_name_sv(spec->shift, 0, NULL, NULL);
+
+		*prelude_sentinel = op_append_list(OP_LINESEQ, *prelude_sentinel, newSTATEOP(0, NULL, var));
+	}
+
 	c = lex_peek_unichar(0);
 	if (c == '(') {
 		DefaultParamSpec **pdefaults_tail = &defaults;
@@ -356,6 +372,16 @@ static int parse_fun(pTHX_ OP **pop, const char *keyword_ptr, STRLEN keyword_len
 					c = lex_peek_unichar(0);
 				}
 
+				/* my $param; */
+				{
+					OP *var;
+
+					var = newOP(OP_PADSV, OPf_MOD | (OPpLVAL_INTRO << 8));
+					var->op_targ = pad_add_name_sv(param, 0, NULL, NULL);
+
+					*prelude_sentinel = op_append_list(OP_LINESEQ, *prelude_sentinel, newSTATEOP(0, NULL, var));
+				}
+
 				if (c == ',') {
 					lex_read_unichar(0);
 					lex_read_space(0);
@@ -485,7 +511,6 @@ static int parse_fun(pTHX_ OP **pop, const char *keyword_ptr, STRLEN keyword_len
 		);
 	}
 
-
 	if (builtin_attrs & MY_ATTR_LVALUE) {
 		CvLVALUE_on(PL_compcv);
 	}
@@ -496,136 +521,132 @@ static int parse_fun(pTHX_ OP **pop, const char *keyword_ptr, STRLEN keyword_len
 		CvSPECIAL_on(PL_compcv);
 	}
 
-	/* munge */
-	{
-		OP *prelude = NULL;
-
-		/* min/max argument count checks */
-		if (spec->flags & FLAG_CHECK_NARGS) {
-			if (SvTRUE(spec->shift)) {
-				args_min++;
-				if (args_max != -1) {
-					args_max++;
-				}
+	/* min/max argument count checks */
+	if (spec->flags & FLAG_CHECK_NARGS) {
+		if (SvTRUE(spec->shift)) {
+			args_min++;
+			if (args_max != -1) {
+				args_max++;
 			}
+		}
 
-			if (args_min > 0) {
-				OP *chk, *cond, *err, *croak;
+		if (args_min > 0) {
+			OP *chk, *cond, *err, *croak;
 
-				err = newSVOP(OP_CONST, 0,
-				              newSVpvf("Not enough arguments for %"SVf, SVfARG(declarator)));
+			err = newSVOP(OP_CONST, 0,
+				          newSVpvf("Not enough arguments for %"SVf, SVfARG(declarator)));
 
-				croak = newCVREF(OPf_WANT_SCALAR,
-				                 newGVOP(OP_GV, 0, gv_fetchpvs("Carp::croak", 0, SVt_PVCV)));
-				err = newUNOP(OP_ENTERSUB, OPf_STACKED,
-				              op_append_elem(OP_LIST, err, croak));
+			croak = newCVREF(OPf_WANT_SCALAR,
+				             newGVOP(OP_GV, 0, gv_fetchpvs("Carp::croak", 0, SVt_PVCV)));
+			err = newUNOP(OP_ENTERSUB, OPf_STACKED,
+				          op_append_elem(OP_LIST, err, croak));
 
-				cond = newBINOP(OP_LT, 0,
-				                newAVREF(newGVOP(OP_GV, 0, PL_defgv)),
-				                newSVOP(OP_CONST, 0, newSViv(args_min)));
-				chk = newLOGOP(OP_AND, 0, cond, err);
+			cond = newBINOP(OP_LT, 0,
+				            newAVREF(newGVOP(OP_GV, 0, PL_defgv)),
+				            newSVOP(OP_CONST, 0, newSViv(args_min)));
+			chk = newLOGOP(OP_AND, 0, cond, err);
 
-				prelude = op_append_list(OP_LINESEQ, prelude, newSTATEOP(0, NULL, chk));
-			}
-			if (args_max != -1) {
-				OP *chk, *cond, *err, *croak;
+			*prelude_sentinel = op_append_list(OP_LINESEQ, *prelude_sentinel, newSTATEOP(0, NULL, chk));
+		}
+		if (args_max != -1) {
+			OP *chk, *cond, *err, *croak;
 
-				err = newSVOP(OP_CONST, 0,
-				              newSVpvf("Too many arguments for %"SVf, SVfARG(declarator)));
+			err = newSVOP(OP_CONST, 0,
+				          newSVpvf("Too many arguments for %"SVf, SVfARG(declarator)));
 
-				croak = newCVREF(OPf_WANT_SCALAR,
-				                 newGVOP(OP_GV, 0, gv_fetchpvs("Carp::croak", 0, SVt_PVCV)));
-				err = newUNOP(OP_ENTERSUB, OPf_STACKED,
-				              op_append_elem(OP_LIST, err, croak));
+			croak = newCVREF(OPf_WANT_SCALAR,
+				             newGVOP(OP_GV, 0, gv_fetchpvs("Carp::croak", 0, SVt_PVCV)));
+			err = newUNOP(OP_ENTERSUB, OPf_STACKED,
+				          op_append_elem(OP_LIST, err, croak));
 
-				cond = newBINOP(OP_GT, 0,
-				                newAVREF(newGVOP(OP_GV, 0, PL_defgv)),
-				                newSVOP(OP_CONST, 0, newSViv(args_max)));
-				chk = newLOGOP(OP_AND, 0, cond, err);
+			cond = newBINOP(OP_GT, 0,
+				            newAVREF(newGVOP(OP_GV, 0, PL_defgv)),
+				            newSVOP(OP_CONST, 0, newSViv(args_max)));
+			chk = newLOGOP(OP_AND, 0, cond, err);
 
-				prelude = op_append_list(OP_LINESEQ, prelude, newSTATEOP(0, NULL, chk));
-			}
+			*prelude_sentinel = op_append_list(OP_LINESEQ, *prelude_sentinel, newSTATEOP(0, NULL, chk));
 		}
+	}
 
-		/* my $self = shift; */
-		if (SvTRUE(spec->shift)) {
-			OP *var, *shift;
-
-			var = newOP(OP_PADSV, OPf_WANT_SCALAR | (OPpLVAL_INTRO << 8));
-			var->op_targ = pad_add_name_sv(spec->shift, 0, NULL, NULL);
-
-			shift = newASSIGNOP(OPf_STACKED, var, 0, newOP(OP_SHIFT, 0));
-			prelude = op_append_list(OP_LINESEQ, prelude, newSTATEOP(0, NULL, shift));
-		}
+	/* $self = shift; */
+	if (SvTRUE(spec->shift)) {
+		OP *var, *shift;
 
-		/* my (PARAMS) = @_; */
-		if (params && av_len(params) > -1) {
-			SV *param;
-			OP *init_param, *left, *right;
-
-			left = NULL;
-			while ((param = av_shift(params)) != &PL_sv_undef) {
-				OP *const var = newOP(OP_PADSV, OPf_WANT_LIST | (OPpLVAL_INTRO << 8));
-				var->op_targ = pad_add_name_sv(param, 0, NULL, NULL);
-				SvREFCNT_dec(param);
-				left = op_append_elem(OP_LIST, left, var);
-			}
+		var = newOP(OP_PADSV, OPf_WANT_SCALAR);
+		var->op_targ = pad_findmy_sv(spec->shift, 0);
 
-			left->op_flags |= OPf_PARENS;
-			right = newAVREF(newGVOP(OP_GV, 0, PL_defgv));
-			init_param = newASSIGNOP(OPf_STACKED, left, 0, right);
-			init_param = newSTATEOP(0, NULL, init_param);
+		shift = newASSIGNOP(OPf_STACKED, var, 0, newOP(OP_SHIFT, 0));
+		*prelude_sentinel = op_append_list(OP_LINESEQ, *prelude_sentinel, newSTATEOP(0, NULL, shift));
+	}
 
-			prelude = op_append_list(OP_LINESEQ, prelude, init_param);
+	/* (PARAMS) = @_; */
+	if (params && av_len(params) > -1) {
+		SV *param;
+		OP *init_param, *left, *right;
+
+		left = NULL;
+		while ((param = av_shift(params)) != &PL_sv_undef) {
+			OP *const var = newOP(OP_PADSV, OPf_WANT_LIST);
+			var->op_targ = pad_findmy_sv(param, 0);
+			SvREFCNT_dec(param);
+			left = op_append_elem(OP_LIST, left, var);
 		}
 
-		/* defaults */
-		{
-			OP *gen = NULL;
-			DefaultParamSpec *dp;
+		left->op_flags |= OPf_PARENS;
+		right = newAVREF(newGVOP(OP_GV, 0, PL_defgv));
+		init_param = newASSIGNOP(OPf_STACKED, left, 0, right);
+		init_param = newSTATEOP(0, NULL, init_param);
 
-			for (dp = defaults; dp; dp = dp->next) {
-				OP *init = dp->init;
-				OP *var, *args, *cond;
+		*prelude_sentinel = op_append_list(OP_LINESEQ, *prelude_sentinel, init_param);
+	}
 
-				/* var = `$,name */
-				var = newOP(OP_PADSV, 0);
-				var->op_targ = pad_findmy_sv(dp->name, 0);
+	/* defaults */
+	{
+		OP *gen = NULL;
+		DefaultParamSpec *dp;
 
-				/* init = `,var = ,init */
-				init = newASSIGNOP(OPf_STACKED, var, 0, init);
+		for (dp = defaults; dp; dp = dp->next) {
+			OP *init = dp->init;
+			OP *var, *args, *cond;
 
-				/* args = `@_ */
-				args = newAVREF(newGVOP(OP_GV, 0, PL_defgv));
+			/* var = `$,name */
+			var = newOP(OP_PADSV, 0);
+			var->op_targ = pad_findmy_sv(dp->name, 0);
 
-				/* cond = `,args < ,index */
-				cond = newBINOP(OP_LT, 0, args, newSVOP(OP_CONST, 0, newSViv(dp->limit)));
+			/* init = `,var = ,init */
+			init = newASSIGNOP(OPf_STACKED, var, 0, init);
 
-				/* init = `,init if ,cond */
-				init = newLOGOP(OP_AND, 0, cond, init);
+			/* args = `@_ */
+			args = newAVREF(newGVOP(OP_GV, 0, PL_defgv));
 
-				/* gen = `,gen ; ,init */
-				gen = op_append_list(OP_LINESEQ, gen, newSTATEOP(0, NULL, init));
+			/* cond = `,args < ,index */
+			cond = newBINOP(OP_LT, 0, args, newSVOP(OP_CONST, 0, newSViv(dp->limit)));
 
-				dp->init = NULL;
-			}
+			/* init = `,init if ,cond */
+			init = newLOGOP(OP_AND, 0, cond, init);
 
-			prelude = op_append_list(OP_LINESEQ, prelude, gen);
+			/* gen = `,gen ; ,init */
+			gen = op_append_list(OP_LINESEQ, gen, newSTATEOP(0, NULL, init));
+
+			dp->init = NULL;
 		}
 
-		/* finally let perl parse the actual subroutine body */
-		body = parse_block(0);
+		*prelude_sentinel = op_append_list(OP_LINESEQ, *prelude_sentinel, gen);
+	}
 
-		/* add '();' to make function return nothing by default */
-		/* (otherwise the invisible parameter initialization can "leak" into
-		   the return value: fun ($x) {}->("asdf", 0) == 2) */
-		if (prelude) {
-			body = newSTATEOP(0, NULL, body);
-		}
+	/* finally let perl parse the actual subroutine body */
+	body = parse_block(0);
 
-		body = op_append_list(OP_LINESEQ, prelude, body);
+	/* add '();' to make function return nothing by default */
+	/* (otherwise the invisible parameter initialization can "leak" into
+	   the return value: fun ($x) {}->("asdf", 0) == 2) */
+	if (*prelude_sentinel) {
+		body = newSTATEOP(0, NULL, body);
 	}
 
+	body = op_append_list(OP_LINESEQ, *prelude_sentinel, body);
+	*prelude_sentinel = NULL;
+
 	/* it's go time. */
 	{
 		OP *const attrs = *attrs_sentinel;