From 04c90470556a5d3a66fb592ff7b6fee1228be95e Mon Sep 17 00:00:00 2001
From: rodri <rgl@antares-labs.eu>
Date: Tue, 15 Aug 2023 02:41:43 +0000
Subject: added ship drawing and placement procedures.

---
 bts.c | 105 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++--------
 1 file changed, 93 insertions(+), 12 deletions(-)

(limited to 'bts.c')

diff --git a/bts.c b/bts.c
index 4f636a6..b6c02f9 100644
--- a/bts.c
+++ b/bts.c
@@ -19,6 +19,7 @@ Image *screenb;
 Image *tiletab[NTILES];
 Board alienboard;
 Board localboard;
+Ship *carrier;
 
 struct {
 	int state;
@@ -83,6 +84,25 @@ drawtile(Image *dst, Board *b, Point2 cell, int type)
 	draw(dst, Rpt(p, addpt(p, Pt(TW,TH))), ti, nil, ZP);
 }
 
+void
+drawship(Image *dst, Ship *s)
+{
+	Point2 p, sv;
+	int i;
+
+	p = s->p;
+	switch(s->orient){
+	case OH: sv = Vec2(1,0); break;
+	case OV: sv = Vec2(0,1); break;
+	default: return;
+	}
+
+	for(i = 0; i < s->ncells; i++){
+		drawtile(dst, &localboard, p, s->hit[i]? Thit: Tship);
+		p = addpt2(p, sv);
+	}
+}
+
 void
 drawboard(Image *dst, Board *b)
 {
@@ -101,6 +121,7 @@ redraw(void)
 	draw(screenb, screenb->r, display->black, nil, ZP);
 	drawboard(screenb, &alienboard);
 	drawboard(screenb, &localboard);
+	drawship(screenb, carrier);
 
 	draw(screen, screen->r, screenb, nil, ZP);
 
@@ -190,6 +211,44 @@ initboards(void)
 	localboard.bbox = Rpt(fromworld(localboard.p), fromworld(addpt2(localboard.p, Pt2(TW*MAPW,TH*MAPH,1))));
 }
 
+void
+initships(void)
+{
+	Point2 sv;
+
+	carrier = emalloc(sizeof *carrier);
+	carrier->p = Pt2(2,4,1);
+	carrier->orient = OV;
+	carrier->ncells = 5;
+	carrier->hit = emalloc(carrier->ncells*sizeof(int));
+	memset(carrier->hit, 0, carrier->ncells*sizeof(int));
+	carrier->sunk = 0;
+	switch(carrier->orient){
+	case OH: sv = Vec2(1,0); break;
+	case OV: sv = Vec2(0,1); break;
+	default: sysfatal("initships: wrong ship orientation");
+	}
+	carrier->bbox = Rpt(
+		fromboard(&localboard, carrier->p),
+		fromboard(&localboard, addpt2(addpt2(carrier->p, mulpt2(sv, carrier->ncells)), Vec2(sv.y,sv.x)))
+	);
+}
+
+void
+placeship(Mousectl *mc, Ship *s)
+{
+	for(;;){
+		if(readmouse(mc) < 0)
+			break;
+		mc->xy = subpt(mc->xy, screen->r.min);
+		if(ptinrect(mc->xy, localboard.bbox) && mc->buttons == 1){
+			s->p = toboard(&localboard, mc->xy);
+			send(drawchan, nil);
+			break;
+		}
+	}
+}
+
 void
 lmb(Mousectl *mc)
 {
@@ -202,20 +261,41 @@ lmb(Mousectl *mc)
 	else if(ptinrect(mc->xy, localboard.bbox))
 		b = &localboard;
 
-	if(b != nil){
-		cell = toboard(b, mc->xy);
-		switch(game.state){
-		case Outlaying:
-			settile(b, cell, Tship);
-		case Playing:
-			settile(b, cell, Tmiss);
-			chanprint(egress, "shoot %d-%d", (int)cell.x, (int)cell.y);
-			break;
-		}
+	if(b == nil)
+		return;
+
+	cell = toboard(b, mc->xy);
+	switch(game.state){
+	case Outlaying:
+		settile(b, cell, Tship);
+		break;
+	case Playing:
+		settile(b, cell, Tmiss);
+		chanprint(egress, "shoot %d-%d", (int)cell.x, (int)cell.y);
+		break;
 	}
 	send(drawchan, nil);
 }
 
+void
+rmb(Mousectl *mc)
+{
+	enum {
+		PLACESHIP,
+	};
+	static char *items[] = {
+	 [PLACESHIP]	"place ship",
+		nil
+	};
+	static Menu menu = { .item = items };
+
+	switch(menuhit(3, mc, &menu, _screen)){
+	case PLACESHIP:
+		placeship(mc, carrier);
+		break;
+	}
+}
+
 void
 mouse(Mousectl *mc)
 {
@@ -226,10 +306,10 @@ mouse(Mousectl *mc)
 		lmb(mc);
 		break;
 	case 2:
-		//mmb(mc);
+//		mmb(mc);
 		break;
 	case 4:
-		//rmb(mc);
+		rmb(mc);
 		break;
 	}
 }
@@ -387,6 +467,7 @@ threadmain(int argc, char *argv[])
 
 	inittiles();
 	initboards();
+	initships();
 	game.state = Waiting0;
 
 	drawchan = chancreate(sizeof(void*), 0);
-- 
cgit v1.2.3