main.go

package main

import (
	"flag"
	"fmt"
	"strings"
	"time"

	"github.com/manifoldco/promptui"
	"github.com/tcluri/go-life/gameoflife"
)

var cellPatternsDir string = "./patterns"
var cellfileNameExtension string = ".cells"

func main() {
	// Use flag to set the length of grid
	size := flag.Int("grid_size", 15, "Enter the length of the grid(square) of board for the game of life")
	// Use flag to get different types of grid and cell layout
	sim := flag.String("sim", "select", "Enter the kind of simulation you want to run and see: Example - 'glider', 'gliders'(for two!)")
	// Parse flags
	flag.Parse()

	var gridSize int
	var populateFunc func(gameoflife.Grid) gameoflife.Grid

	// Create grid and populate cells based on sim value
	switch *sim {
	case "select":
		files := gameoflife.ListPatterns(cellPatternsDir, cellfileNameExtension)
		prompt := promptui.Select{
			Label: "Select GOL simulation you want to run:",
			Items: files,
			Searcher: func(input string, index int) bool {
				option := files[index]
				name := fmt.Sprintf("%s", option)
				return strings.Contains(name, input)
			},
			StartInSearchMode: true,
		}

		_, selected, err := prompt.Run()
		if err != nil {
			fmt.Printf("Prompt failed %v\n", err)
			return
		}

		fmt.Printf("You choose %q\n", selected)
		shape := gameoflife.LoadCellPatternFromFile(selected, cellPatternsDir, cellfileNameExtension)
		gameoflife.AssignShape(shape)
		shapeHeight, shapeWidth := shape.FindShapeHeightAndWidth()
		fmt.Println("Shape height:", shapeHeight, "Shape width:", shapeWidth)
		if shapeHeight >= shapeWidth {
			gridSize = shapeHeight * 3
		} else {
			gridSize = shapeWidth * 3
		}
		fmt.Println("Grid size:", gridSize)
		populateFunc = gameoflife.PopulateAndPlaceShapeAtCenter
	case "random":
		gridSize = *size
		// Populate the grid cells with random cells
		populateFunc = gameoflife.PopulateRandomCells
	case "glider":
		// Create an empty grid for the glider to go across smoothly
		gridSize = 20
		populateFunc = gameoflife.PopulateGlider
	case "gliders":
		gridSize = 20
		populateFunc = gameoflife.PopulateTwoGliders
	}

	// Create an empty grid for the cells
	initialGrid := gameoflife.CreateEmptyGrid(gridSize)
	currentGeneration := populateFunc(initialGrid)

	// Evolution!
	generationNum := 1
	for {
		fmt.Printf("Generation #%d\n", generationNum)
		aliveCells := gameoflife.CurrentAlive(currentGeneration)
		if aliveCells == 0 {
			fmt.Println("All cells have died")
			return
		} else {
			fmt.Printf("Alive: %d\n", aliveCells)
		}
		gameoflife.PrintCellsInGrid(currentGeneration)

		// Evolve the cell grid
		currentGeneration = gameoflife.Evolution(currentGeneration)
		time.Sleep(80 * time.Millisecond)
		fmt.Print("\033[H\033[2J") // Clear screen
		generationNum++
	}
}