xref: /original-bsd/usr.bin/pascal/src/pccaseop.c (revision 0b685140)
1 /* Copyright (c) 1980 Regents of the University of California */
2 
3 static	char sccsid[] = "@(#)pccaseop.c 1.8 02/02/82";
4 
5 #include "whoami.h"
6 #ifdef PC
7     /*
8      *	and the rest of the file
9      */
10 #include "0.h"
11 #include "tree.h"
12 #include "objfmt.h"
13 #include "pcops.h"
14 #include "pc.h"
15 
16     /*
17      *	structure for a case:
18      *	    its constant label, line number (for errors), and location label.
19      */
20 struct ct {
21     long	cconst;
22     int		cline;
23     int		clabel;
24 };
25 
26     /*
27      *	the P2FORCE operator puts its operand into a register.
28      *	these to keep from thinking of it as r0 all over.
29      */
30 #define	FORCENAME	"r0"
31 
32     /*
33      *	given a tree for a case statement, generate code for it.
34      *	this computes the expression into a register,
35      *	puts down the code for each of the cases,
36      *	and then decides how to do the case switching.
37      *	tcase	[0]	T_CASE
38      *		[1]	lineof "case"
39      *		[2]	expression
40      *		[3]	list of cased statements:
41      *			cstat	[0]	T_CSTAT
42      *				[1]	lineof ":"
43      *				[2]	list of constant labels
44      *				[3]	statement
45      */
46 pccaseop( tcase )
47     int	*tcase;
48 {
49     struct nl	*exprtype;
50     struct nl	*exprnlp;
51     struct nl	*rangetype;
52     long	low;
53     long	high;
54     long	exprctype;
55     long	swlabel;
56     long	endlabel;
57     long	label;
58     long	count;
59     long	*cstatlp;
60     long	*cstatp;
61     long	*casep;
62     struct ct	*ctab;
63     struct ct	*ctp;
64     long	i;
65     long	nr;
66     long	goc;
67     int		casecmp();
68     bool	dupcases;
69 
70     goc = gocnt;
71 	/*
72 	 *  find out the type of the case expression
73 	 *  even if the expression has errors (exprtype == NIL), continue.
74 	 */
75     line = tcase[1];
76     codeoff();
77     exprtype = rvalue( (int *) tcase[2] , NIL  , RREQ );
78     codeon();
79     if ( exprtype != NIL ) {
80 	if ( isnta( exprtype , "bcsi" ) ) {
81 	    error("Case selectors cannot be %ss" , nameof( exprtype ) );
82 	    exprtype = NIL;
83 	} else {
84 	    if ( exprtype -> class != RANGE ) {
85 		rangetype = exprtype -> type;
86 	    } else {
87 		rangetype = exprtype;
88 	    }
89 	    if ( rangetype == NIL ) {
90 		exprtype = NIL;
91 	    } else {
92 		low = rangetype -> range[0];
93 		high = rangetype -> range[1];
94 	    }
95 	}
96     }
97     if ( exprtype != NIL ) {
98 	    /*
99 	     *	compute and save the case expression.
100 	     *	also, put expression into a register
101 	     *	save its c-type and jump to the code to do the switch.
102 	     */
103 	exprctype = p2type( exprtype );
104 	exprnlp = tmpalloc( sizeof (long) , nl + T4INT , NOREG );
105 	putRV( 0 , cbn , exprnlp -> value[ NL_OFFS ] ,
106 			exprnlp -> extra_flags , P2INT );
107 	(void) rvalue( (int *) tcase[2] , NIL , RREQ );
108 	putop( P2ASSIGN , P2INT );
109 	putop( P2FORCE , P2INT );
110 	putdot( filename , line );
111 	swlabel = getlab();
112 	putjbr( swlabel );
113     }
114 	/*
115 	 *  count the number of cases
116 	 *  and allocate table for cases, lines, and labels
117 	 *  default case goes in ctab[0].
118 	 */
119     count = 1;
120     for ( cstatlp = tcase[3] ; cstatlp != NIL ; cstatlp = cstatlp[2] ) {
121 	cstatp = cstatlp[1];
122 	if ( cstatp == NIL ) {
123 	    continue;
124 	}
125 	for ( casep = cstatp[2] ; casep != NIL ; casep = casep[2] ) {
126 	    count++;
127 	}
128     }
129 	/*
130 	 */
131     ctab = (struct ct *) malloc( count * sizeof( struct ct ) );
132     if ( ctab == (struct ct *) 0 ) {
133 	error("Ran out of memory (case)");
134 	pexit( DIED );
135     }
136 	/*
137 	 *  pick up default label and label for after case statement.
138 	 */
139     ctab[0].clabel = getlab();
140     endlabel = getlab();
141 	/*
142 	 *  generate code for each case
143 	 *  filling in ctab for each.
144 	 *  nr is for error if no case falls out bottom.
145 	 */
146     nr = 1;
147     count = 0;
148     for ( cstatlp = tcase[3] ; cstatlp != NIL ; cstatlp = cstatlp[2] ) {
149 	cstatp = cstatlp[1];
150 	if ( cstatp == NIL ) {
151 	    continue;
152 	}
153 	line = cstatp[1];
154 	label = getlab();
155 	for ( casep = cstatp[2] ; casep != NIL ; casep = casep[2] ) {
156 	    gconst( casep[1] );
157 	    if( exprtype == NIL || con.ctype == NIL ) {
158 		continue;
159 	    }
160 	    if ( incompat( con.ctype , exprtype , NIL ) ) {
161 		cerror("Case label type clashed with case selector expression type");
162 		continue;
163 	    }
164 	    if ( con.crval < low || con.crval > high ) {
165 		error("Case label out of range");
166 		continue;
167 	    }
168 	    count++;
169 	    ctab[ count ].cconst = con.crval;
170 	    ctab[ count ].cline = line;
171 	    ctab[ count ].clabel = label;
172 	}
173 	    /*
174 	     *	put out the statement
175 	     */
176 	putlab( label );
177 	putcnt();
178 	level++;
179 	statement( cstatp[3] );
180 	nr = (nr && noreach);
181 	noreach = 0;
182 	level--;
183 	if (gotos[cbn]) {
184 		ungoto();
185 	}
186 	putjbr( endlabel );
187     }
188     noreach = nr;
189 	/*
190 	 *	default action is to call error
191 	 */
192     putlab( ctab[0].clabel );
193     putleaf( P2ICON , 0 , 0 , ADDTYPE( P2FTN | P2INT , P2PTR ) , "_CASERNG" );
194     putRV( 0 , cbn , exprnlp -> value[ NL_OFFS ] ,
195 		    exprnlp -> extra_flags , P2INT );
196     putop( P2CALL , P2INT );
197     putdot( filename , line );
198 	/*
199 	 *  sort the cases
200 	 */
201     qsort( &ctab[1] , count , sizeof (struct ct) , casecmp );
202 	/*
203 	 *  check for duplicates
204 	 */
205     dupcases = FALSE;
206     for ( ctp = &ctab[1] ; ctp < &ctab[ count ] ; ctp++ ) {
207 	if ( ctp[0].cconst == ctp[1].cconst ) {
208 	    error("Multiply defined label in case, lines %d and %d" ,
209 		    ctp[0].cline , ctp[1].cline );
210 	    dupcases = TRUE;
211 	}
212     }
213     if ( dupcases ) {
214 	return;
215     }
216 	/*
217 	 *  choose a switch algorithm and implement it:
218 	 *	direct switch	>= 1/3 full and >= 4 cases.
219 	 *	binary switch	not direct switch and > 8 cases.
220 	 *	ifthenelse	not direct or binary switch.
221 	 */
222     putlab( swlabel );
223     if ( ctab[ count ].cconst - ctab[1].cconst < 3 * count && count >= 4 ) {
224 	directsw( ctab , count );
225     } else if ( count > 8 ) {
226 	binarysw( ctab , count );
227     } else {
228 	itesw( ctab , count );
229     }
230     putlab( endlabel );
231     if ( goc != gocnt ) {
232 	    putcnt();
233     }
234 }
235 
236     /*
237      *	direct switch
238      */
239 directsw( ctab , count )
240     struct ct	*ctab;
241     int		count;
242 {
243     int		fromlabel = getlab();
244     long	i;
245     long	j;
246 
247     putprintf( "	casel	%s,$%d,$%d" , 0 , FORCENAME ,
248 	    ctab[1].cconst , ctab[ count ].cconst - ctab[1].cconst );
249     putlab( fromlabel );
250     i = 1;
251     j = ctab[1].cconst;
252     while ( i <= count ) {
253 	if ( j == ctab[ i ].cconst ) {
254 	    putprintf( "	.word	" , 1 );
255 	    putprintf( PREFIXFORMAT , 1 , LABELPREFIX , ctab[ i ].clabel );
256 	    putprintf( "-" , 1 );
257 	    putprintf( PREFIXFORMAT , 0 , LABELPREFIX , fromlabel );
258 	    i++;
259 	} else {
260 	    putprintf( "	.word	" , 1 );
261 	    putprintf( PREFIXFORMAT , 1 , LABELPREFIX , ctab[ 0 ].clabel );
262 	    putprintf( "-" , 1 );
263 	    putprintf( PREFIXFORMAT , 0 , LABELPREFIX , fromlabel );
264 	}
265 	j++;
266     }
267     putjbr( ctab[0].clabel );
268 }
269 
270     /*
271      *	binary switch
272      *	special case out default label and start recursion.
273      */
274 binarysw( ctab , count )
275     struct ct	*ctab;
276     int		count;
277 {
278 
279     bsrecur( ctab[0].clabel , &ctab[0] , count );
280 }
281 
282     /*
283      *	recursive log( count ) search.
284      */
285 bsrecur( deflabel , ctab , count )
286     int		deflabel;
287     struct ct	*ctab;
288     int		count;
289 {
290 
291     if ( count <= 0 ) {
292 	putprintf( "	jbr	L%d" , 0 , deflabel );
293 	return;
294     } else if ( count == 1 ) {
295 	putprintf( "	cmpl	%s,$%d" , 0 , FORCENAME , ctab[1].cconst );
296 	putprintf( "	jeql	L%d" , 0 , ctab[1].clabel );
297 	putprintf( "	jbr	L%d" , 0 , deflabel );
298 	return;
299     } else {
300 	int	half = ( count + 1 ) / 2;
301 	int	gtrlabel = getlab();
302 
303 	putprintf( "	cmpl	%s,$%d" , 0 , FORCENAME , ctab[ half ].cconst );
304 	putprintf( "	jgtr	L%d" , 0 , gtrlabel );
305 	putprintf( "	jeql	L%d" , 0 , ctab[ half ].clabel );
306 	bsrecur( deflabel , &ctab[0] , half - 1 );
307 	putprintf( "L%d:" , 0 , gtrlabel );
308 	bsrecur( deflabel , &ctab[ half ] , count - half );
309 	return;
310     }
311 }
312 
313 itesw( ctab , count )
314     struct ct	*ctab;
315     int		count;
316 {
317     int	i;
318 
319     for ( i = 1 ; i <= count ; i++ ) {
320 	putprintf( "	cmpl	%s,$%d" , 0 , FORCENAME , ctab[ i ].cconst );
321 	putprintf( "	jeql	L%d" , 0 , ctab[ i ].clabel );
322     }
323     putprintf( "	jbr	L%d" , 0 , ctab[0].clabel );
324     return;
325 }
326 int
327 casecmp( this , that )
328     struct ct 	*this;
329     struct ct 	*that;
330 {
331     if ( this -> cconst < that -> cconst ) {
332 	return -1;
333     } else if ( this -> cconst > that -> cconst ) {
334 	return 1;
335     } else {
336 	return 0;
337     }
338 }
339 #endif PC
340