Adding graph display option

bubble_sort.go

@@ -2,15 +2,15 @@ package main
 
 import "time"
 
-func bubbleSortVisualizer(arr []int, delay time.Duration) {
+func bubbleSortVisualizer(arr []int, displayType string, delay time.Duration) {
 	n := len(arr)
 	for i := 0; i < n-1; i++ {
 		for j := 0; j < n-i-1; j++ {
-			visualizeIteration(arr, j, j+1, delay)
+			visualizeIteration(arr, j, j+1, displayType, delay)
 			if arr[j] > arr[j+1] {
-				visualizeIteration(arr, j, j+1, delay)
+				visualizeIteration(arr, j, j+1, displayType, delay)
 				arr[j], arr[j+1] = arr[j+1], arr[j]
-				visualizeIteration(arr, j, j+1, delay)
+				visualizeIteration(arr, j, j+1, displayType, delay)
 			}
 		}
 	}

cocktail_shaker_sort.go

@@ -2,27 +2,27 @@ package main
 
 import "time"
 
-func cocktailShakerSortVisualizer(arr []int, delay time.Duration) {
+func cocktailShakerSortVisualizer(arr []int, displayType string, delay time.Duration) {
 	left := 0
 	right := len(arr) - 1
 
 	for left <= right {
 		for i := left; i < right; i++ {
-			visualizeIteration(arr, i, i+1, delay)
+			visualizeIteration(arr, i, i+1, displayType, delay)
 			if arr[i] > arr[i+1] {
-				visualizeIteration(arr, i, i+1, delay)
+				visualizeIteration(arr, i, i+1, displayType, delay)
 				arr[i], arr[i+1] = arr[i+1], arr[i]
-				visualizeIteration(arr, i, i+1, delay)
+				visualizeIteration(arr, i, i+1, displayType, delay)
 			}
 		}
 		right--
 
 		for i := right; i > left; i-- {
-			visualizeIteration(arr, i-1, i, delay)
+			visualizeIteration(arr, i-1, i, displayType, delay)
 			if arr[i-1] > arr[i] {
-				visualizeIteration(arr, i-1, i, delay)
+				visualizeIteration(arr, i-1, i, displayType, delay)
 				arr[i], arr[i-1] = arr[i-1], arr[i]
-				visualizeIteration(arr, i-1, i, delay)
+				visualizeIteration(arr, i-1, i, displayType, delay)
 			}
 		}
 		left++

comb_sort.go

@@ -2,7 +2,7 @@ package main
 
 import "time"
 
-func combSortVisualizer(arr []int, delay time.Duration) {
+func combSortVisualizer(arr []int, displayType string, delay time.Duration) {
 	n := len(arr)
 	gap := n
 	swapped := true
@@ -12,11 +12,11 @@ func combSortVisualizer(arr []int, delay time.Duration) {
 		swapped = false
 
 		for i := 0; i < n-gap; i++ {
-			visualizeIteration(arr, i, i+gap, delay)
+			visualizeIteration(arr, i, i+gap, displayType, delay)
 			if arr[i] > arr[i+gap] {
-				visualizeIteration(arr, i, i+gap, delay)
+				visualizeIteration(arr, i, i+gap, displayType, delay)
 				arr[i], arr[i+gap] = arr[i+gap], arr[i]
-				visualizeIteration(arr, i, i+gap, delay)
+				visualizeIteration(arr, i, i+gap, displayType, delay)
 				swapped = true
 			}
 		}

gnome_sort.go

@@ -2,17 +2,17 @@ package main
 
 import "time"
 
-func gnomeSortVisualizer(arr []int, delay time.Duration) {
+func gnomeSortVisualizer(arr []int, displayType string, delay time.Duration) {
 	i := 1
 
 	for i < len(arr) {
 		if i == 0 || arr[i] >= arr[i-1] {
-			visualizeIteration(arr, i-1, i, delay)
+			visualizeIteration(arr, i-1, i, displayType, delay)
 			i++
 		} else {
-			visualizeIteration(arr, i-1, i, delay)
+			visualizeIteration(arr, i-1, i, displayType, delay)
 			arr[i], arr[i-1] = arr[i-1], arr[i]
-			visualizeIteration(arr, i-1, i, delay)
+			visualizeIteration(arr, i-1, i, displayType, delay)
 			i--
 			if i == 0 {
 				i = 1

go.mod

@@ -3,9 +3,19 @@ module github.com/go-nerds/cli-algorithms-visualizer
 go 1.20
 
 require (
+	atomicgo.dev/cursor v0.1.3 // indirect
+	atomicgo.dev/keyboard v0.2.9 // indirect
+	atomicgo.dev/schedule v0.0.2 // indirect
 	github.com/chzyer/readline v0.0.0-20180603132655-2972be24d48e // indirect
+	github.com/containerd/console v1.0.3 // indirect
 	github.com/gookit/color v1.5.3 // indirect
+	github.com/lithammer/fuzzysearch v1.1.8 // indirect
 	github.com/manifoldco/promptui v0.9.0 // indirect
+	github.com/mattn/go-runewidth v0.0.14 // indirect
+	github.com/pterm/pterm v0.12.63 // indirect
+	github.com/rivo/uniseg v0.4.4 // indirect
 	github.com/xo/terminfo v0.0.0-20220910002029-abceb7e1c41e // indirect
 	golang.org/x/sys v0.10.0 // indirect
+	golang.org/x/term v0.10.0 // indirect
+	golang.org/x/text v0.11.0 // indirect
 )

go.sum

@@ -1,13 +1,110 @@
+atomicgo.dev/cursor v0.1.3 h1:w8GcylMdZRyFzvDiGm3wy3fhZYYT7BwaqNjUFHxo0NU=
+atomicgo.dev/cursor v0.1.3/go.mod h1:Lr4ZJB3U7DfPPOkbH7/6TOtJ4vFGHlgj1nc+n900IpU=
+atomicgo.dev/keyboard v0.2.9 h1:tOsIid3nlPLZ3lwgG8KZMp/SFmr7P0ssEN5JUsm78K8=
+atomicgo.dev/keyboard v0.2.9/go.mod h1:BC4w9g00XkxH/f1HXhW2sXmJFOCWbKn9xrOunSFtExQ=
+atomicgo.dev/schedule v0.0.2 h1:2e/4KY6t3wokja01Cyty6qgkQM8MotJzjtqCH70oX2Q=
+atomicgo.dev/schedule v0.0.2/go.mod h1:xeUa3oAkiuHYh8bKiQBRojqAMq3PXXbJujjb0hw8pEU=
+github.com/MarvinJWendt/testza v0.1.0/go.mod h1:7AxNvlfeHP7Z/hDQ5JtE3OKYT3XFUeLCDE2DQninSqs=
+github.com/MarvinJWendt/testza v0.2.1/go.mod h1:God7bhG8n6uQxwdScay+gjm9/LnO4D3kkcZX4hv9Rp8=
+github.com/MarvinJWendt/testza v0.2.8/go.mod h1:nwIcjmr0Zz+Rcwfh3/4UhBp7ePKVhuBExvZqnKYWlII=
+github.com/MarvinJWendt/testza v0.2.10/go.mod h1:pd+VWsoGUiFtq+hRKSU1Bktnn+DMCSrDrXDpX2bG66k=
+github.com/MarvinJWendt/testza v0.2.12/go.mod h1:JOIegYyV7rX+7VZ9r77L/eH6CfJHHzXjB69adAhzZkI=
+github.com/MarvinJWendt/testza v0.3.0/go.mod h1:eFcL4I0idjtIx8P9C6KkAuLgATNKpX4/2oUqKc6bF2c=
+github.com/MarvinJWendt/testza v0.4.2/go.mod h1:mSdhXiKH8sg/gQehJ63bINcCKp7RtYewEjXsvsVUPbE=
+github.com/atomicgo/cursor v0.0.1/go.mod h1:cBON2QmmrysudxNBFthvMtN32r3jxVRIvzkUiF/RuIk=
 github.com/chzyer/logex v1.1.10/go.mod h1:+Ywpsq7O8HXn0nuIou7OrIPyXbp3wmkHB+jjWRnGsAI=
 github.com/chzyer/readline v0.0.0-20180603132655-2972be24d48e h1:fY5BOSpyZCqRo5OhCuC+XN+r/bBCmeuuJtjz+bCNIf8=
 github.com/chzyer/readline v0.0.0-20180603132655-2972be24d48e/go.mod h1:nSuG5e5PlCu98SY8svDHJxuZscDgtXS6KTTbou5AhLI=
 github.com/chzyer/test v0.0.0-20180213035817-a1ea475d72b1/go.mod h1:Q3SI9o4m/ZMnBNeIyt5eFwwo7qiLfzFZmjNmxjkiQlU=
+github.com/containerd/console v1.0.3 h1:lIr7SlA5PxZyMV30bDW0MGbiOPXwc63yRuCP0ARubLw=
+github.com/containerd/console v1.0.3/go.mod h1:7LqA/THxQ86k76b8c/EMSiaJ3h1eZkMkXar0TQ1gf3U=
+github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
+github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
+github.com/gookit/color v1.4.2/go.mod h1:fqRyamkC1W8uxl+lxCQxOT09l/vYfZ+QeiX3rKQHCoQ=
+github.com/gookit/color v1.5.0/go.mod h1:43aQb+Zerm/BWh2GnrgOQm7ffz7tvQXEKV6BFMl7wAo=
 github.com/gookit/color v1.5.3 h1:twfIhZs4QLCtimkP7MOxlF3A0U/5cDPseRT9M/+2SCE=
 github.com/gookit/color v1.5.3/go.mod h1:NUzwzeehUfl7GIb36pqId+UGmRfQcU/WiiyTTeNjHtE=
+github.com/klauspost/cpuid/v2 v2.0.9/go.mod h1:FInQzS24/EEf25PyTYn52gqo7WaD8xa0213Md/qVLRg=
+github.com/klauspost/cpuid/v2 v2.0.10/go.mod h1:g2LTdtYhdyuGPqyWyv7qRAmj1WBqxuObKfj5c0PQa7c=
+github.com/klauspost/cpuid/v2 v2.0.12/go.mod h1:g2LTdtYhdyuGPqyWyv7qRAmj1WBqxuObKfj5c0PQa7c=
+github.com/kr/pretty v0.1.0/go.mod h1:dAy3ld7l9f0ibDNOQOHHMYYIIbhfbHSm3C4ZsoJORNo=
+github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ=
+github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI=
+github.com/lithammer/fuzzysearch v1.1.8 h1:/HIuJnjHuXS8bKaiTMeeDlW2/AyIWk2brx1V8LFgLN4=
+github.com/lithammer/fuzzysearch v1.1.8/go.mod h1:IdqeyBClc3FFqSzYq/MXESsS4S0FsZ5ajtkr5xPLts4=
 github.com/manifoldco/promptui v0.9.0 h1:3V4HzJk1TtXW1MTZMP7mdlwbBpIinw3HztaIlYthEiA=
 github.com/manifoldco/promptui v0.9.0/go.mod h1:ka04sppxSGFAtxX0qhlYQjISsg9mR4GWtQEhdbn6Pgg=
+github.com/mattn/go-runewidth v0.0.13/go.mod h1:Jdepj2loyihRzMpdS35Xk/zdY8IAYHsh153qUoGf23w=
+github.com/mattn/go-runewidth v0.0.14 h1:+xnbZSEeDbOIg5/mE6JF0w6n9duR1l3/WmbinWVwUuU=
+github.com/mattn/go-runewidth v0.0.14/go.mod h1:Jdepj2loyihRzMpdS35Xk/zdY8IAYHsh153qUoGf23w=
+github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
+github.com/pterm/pterm v0.12.27/go.mod h1:PhQ89w4i95rhgE+xedAoqous6K9X+r6aSOI2eFF7DZI=
+github.com/pterm/pterm v0.12.29/go.mod h1:WI3qxgvoQFFGKGjGnJR849gU0TsEOvKn5Q8LlY1U7lg=
+github.com/pterm/pterm v0.12.30/go.mod h1:MOqLIyMOgmTDz9yorcYbcw+HsgoZo3BQfg2wtl3HEFE=
+github.com/pterm/pterm v0.12.31/go.mod h1:32ZAWZVXD7ZfG0s8qqHXePte42kdz8ECtRyEejaWgXU=
+github.com/pterm/pterm v0.12.33/go.mod h1:x+h2uL+n7CP/rel9+bImHD5lF3nM9vJj80k9ybiiTTE=
+github.com/pterm/pterm v0.12.36/go.mod h1:NjiL09hFhT/vWjQHSj1athJpx6H8cjpHXNAK5bUw8T8=
+github.com/pterm/pterm v0.12.40/go.mod h1:ffwPLwlbXxP+rxT0GsgDTzS3y3rmpAO1NMjUkGTYf8s=
+github.com/pterm/pterm v0.12.63 h1:fHlrpFiI9qLtEU0TWDWMU+tAt4qKJ/s157BEAPtGm8w=
+github.com/pterm/pterm v0.12.63/go.mod h1:Bq1eoUJ6BhUzzXG8WxA4l7T3s7d3Ogwg7v9VXlsVat0=
+github.com/rivo/uniseg v0.2.0/go.mod h1:J6wj4VEh+S6ZtnVlnTBMWIodfgj8LQOQFoIToxlJtxc=
+github.com/rivo/uniseg v0.4.4 h1:8TfxU8dW6PdqD27gjM8MVNuicgxIjxpm4K7x4jp8sis=
+github.com/rivo/uniseg v0.4.4/go.mod h1:FN3SvrM+Zdj16jyLfmOkMNblXMcoc8DfTHruCPUcx88=
+github.com/sergi/go-diff v1.2.0/go.mod h1:STckp+ISIX8hZLjrqAeVduY0gWCT9IjLuqbuNXdaHfM=
+github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
+github.com/stretchr/testify v1.4.0/go.mod h1:j7eGeouHqKxXV5pUuKE4zz7dFj8WfuZ+81PSLYec5m4=
+github.com/stretchr/testify v1.6.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
+github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
+github.com/xo/terminfo v0.0.0-20210125001918-ca9a967f8778/go.mod h1:2MuV+tbUrU1zIOPMxZ5EncGwgmMJsa+9ucAQZXxsObs=
 github.com/xo/terminfo v0.0.0-20220910002029-abceb7e1c41e h1:JVG44RsyaB9T2KIHavMF/ppJZNG9ZpyihvCd0w101no=
 github.com/xo/terminfo v0.0.0-20220910002029-abceb7e1c41e/go.mod h1:RbqR21r5mrJuqunuUZ/Dhy/avygyECGrLceyNeo4LiM=
+github.com/yuin/goldmark v1.4.13/go.mod h1:6yULJ656Px+3vBD8DxQVa3kxgyrAnzto9xy5taEt/CY=
+golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
+golang.org/x/crypto v0.0.0-20210921155107-089bfa567519/go.mod h1:GvvjBRRGRdwPK5ydBHafDWAxML/pGHZbMvKqRZ5+Abc=
+golang.org/x/mod v0.6.0-dev.0.20220419223038-86c51ed26bb4/go.mod h1:jJ57K6gSWd91VN4djpZkiMVwK6gcyfeH4XE8wZrZaV4=
+golang.org/x/mod v0.8.0/go.mod h1:iBbtSCu2XBx23ZKBPSOrRkjjQPZFPuis4dIYUhu/chs=
+golang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s=
+golang.org/x/net v0.0.0-20210226172049-e18ecbb05110/go.mod h1:m0MpNAwzfU5UDzcl9v0D8zg8gWTRqZa9RBIspLL5mdg=
+golang.org/x/net v0.0.0-20220722155237-a158d28d115b/go.mod h1:XRhObCWvk6IyKnWLug+ECip1KBveYUHfp+8e9klMJ9c=
+golang.org/x/net v0.6.0/go.mod h1:2Tu9+aMcznHK/AK1HMvgo6xiTLG5rD5rZLDS+rp2Bjs=
+golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
+golang.org/x/sync v0.0.0-20220722155255-886fb9371eb4/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
+golang.org/x/sync v0.1.0/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
 golang.org/x/sys v0.0.0-20181122145206-62eef0e2fa9b/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
+golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
+golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
+golang.org/x/sys v0.0.0-20210124154548-22da62e12c0c/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
+golang.org/x/sys v0.0.0-20210330210617-4fbd30eecc44/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
+golang.org/x/sys v0.0.0-20210615035016-665e8c7367d1/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+golang.org/x/sys v0.0.0-20211013075003-97ac67df715c/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+golang.org/x/sys v0.0.0-20220319134239-a9b59b0215f8/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+golang.org/x/sys v0.0.0-20220520151302-bc2c85ada10a/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+golang.org/x/sys v0.0.0-20220722155257-8c9f86f7a55f/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+golang.org/x/sys v0.5.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
 golang.org/x/sys v0.10.0 h1:SqMFp9UcQJZa+pmYuAKjd9xq1f0j5rLcDIk0mj4qAsA=
 golang.org/x/sys v0.10.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
+golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=
+golang.org/x/term v0.0.0-20210220032956-6a3ed077a48d/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=
+golang.org/x/term v0.0.0-20210615171337-6886f2dfbf5b/go.mod h1:jbD1KX2456YbFQfuXm/mYQcufACuNUgVhRMnK/tPxf8=
+golang.org/x/term v0.0.0-20210927222741-03fcf44c2211/go.mod h1:jbD1KX2456YbFQfuXm/mYQcufACuNUgVhRMnK/tPxf8=
+golang.org/x/term v0.5.0/go.mod h1:jMB1sMXY+tzblOD4FWmEbocvup2/aLOaQEp7JmGp78k=
+golang.org/x/term v0.10.0 h1:3R7pNqamzBraeqj/Tj8qt1aQ2HpmlC+Cx/qL/7hn4/c=
+golang.org/x/term v0.10.0/go.mod h1:lpqdcUyK/oCiQxvxVrppt5ggO2KCZ5QblwqPnfZ6d5o=
+golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
+golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
+golang.org/x/text v0.3.7/go.mod h1:u+2+/6zg+i71rQMx5EYifcz6MCKuco9NR6JIITiCfzQ=
+golang.org/x/text v0.7.0/go.mod h1:mrYo+phRRbMaCq/xk9113O4dZlRixOauAjOtrjsXDZ8=
+golang.org/x/text v0.9.0/go.mod h1:e1OnstbJyHTd6l/uOt8jFFHp6TRDWZR/bV3emEE/zU8=
+golang.org/x/text v0.11.0 h1:LAntKIrcmeSKERyiOh0XMV39LXS8IE9UL2yP7+f5ij4=
+golang.org/x/text v0.11.0/go.mod h1:TvPlkZtksWOMsz7fbANvkp4WM8x/WCo/om8BMLbz+aE=
+golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
+golang.org/x/tools v0.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=
+golang.org/x/tools v0.1.12/go.mod h1:hNGJHUnrk76NpqgfD5Aqm5Crs+Hm0VOH/i9J2+nxYbc=
+golang.org/x/tools v0.6.0/go.mod h1:Xwgl3UAJ/d3gWutnCtw505GrjyAbvKui8lOU390QaIU=
+golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
+gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
+gopkg.in/check.v1 v1.0.0-20190902080502-41f04d3bba15/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
+gopkg.in/yaml.v2 v2.2.2/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
+gopkg.in/yaml.v2 v2.2.4/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
+gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
+gopkg.in/yaml.v3 v3.0.0-20210107192922-496545a6307b/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=

insertion_sort.go

@@ -4,14 +4,14 @@ import (
 	"time"
 )
 
-func insertionSortVisualizer(arr []int, delay time.Duration) {
+func insertionSortVisualizer(arr []int, displayType string, delay time.Duration) {
 	for i := 1; i < len(arr); i++ {
 		j := i
-		visualizeIteration(arr, j-1, j, delay)
+		visualizeIteration(arr, j-1, j, displayType, delay)
 		for j > 0 && arr[j] < arr[j-1] {
-			visualizeIteration(arr, j-1, j, delay)
+			visualizeIteration(arr, j-1, j, displayType, delay)
 			arr[j], arr[j-1] = arr[j-1], arr[j]
-			visualizeIteration(arr, j-1, j, delay)
+			visualizeIteration(arr, j-1, j, displayType, delay)
 			j--
 		}
 	}

main.go

@@ -26,6 +26,7 @@ type options struct {
 func main() {
 
 	clearConsole()
+	handleInterrupt()
 
 	arr := generateRandomArray(10)
 	fmt.Println("Initial array:", arr)
@@ -86,6 +87,19 @@ func main() {
 		return
 	}
 
+	clearConsole()
+
+	displayPrompt := promptui.Select{
+		Label: "Select Type of displaying",
+		Items: []string{"Array", "Graph"},
+	}
+
+	_, displayType, err := displayPrompt.Run()
+	if err != nil {
+		fmt.Printf("Prompt failed %v\n", err)
+		return
+	}
+
 	switch action {
 	case "Description":
 		printAlgorithmDescription(algorithms[i].Uuid)
@@ -117,29 +131,29 @@ func main() {
 
 		clearConsole()
 
-		runAlgorithm(algorithms[i].Uuid, arr, delay)
+		runAlgorithm(algorithms[i].Uuid, arr, displayType, delay)
 	default:
 		fmt.Println("Invalid choice")
 		return
 	}
 }
 
-func runAlgorithm(algorithm int, arr []int, delay time.Duration) {
+func runAlgorithm(algorithm int, arr []int, displayType string, delay time.Duration) {
 	switch algorithm {
 	case BubbleSort:
-		bubbleSortVisualizer(arr, delay)
+		bubbleSortVisualizer(arr, displayType, delay)
 	case SelectionSort:
-		selectionSortVisualizer(arr, delay)
+		selectionSortVisualizer(arr, displayType, delay)
 	case InsertionSort:
-		insertionSortVisualizer(arr, delay)
+		insertionSortVisualizer(arr, displayType, delay)
 	case GnomeSort:
-		gnomeSortVisualizer(arr, delay)
+		gnomeSortVisualizer(arr, displayType, delay)
 	case CocktailShakerSort:
-		cocktailShakerSortVisualizer(arr, delay)
+		cocktailShakerSortVisualizer(arr, displayType, delay)
 	case CombSort:
-		combSortVisualizer(arr, delay)
+		combSortVisualizer(arr, displayType, delay)
 	case OddEvenSort:
-		oddEvenSortVisualizer(arr, delay)
+		oddEvenSortVisualizer(arr, displayType, delay)
 	default:
 		fmt.Println("Invalid selection")
 	}

odd_even_sort.go

@@ -2,28 +2,28 @@ package main
 
 import "time"
 
-func oddEvenSortVisualizer(arr []int, delay time.Duration) {
+func oddEvenSortVisualizer(arr []int, displayType string, delay time.Duration) {
 	n := len(arr)
 	sorted := false
 
 	for !sorted {
 		sorted = true
 		for i := 1; i <= n-2; i += 2 {
-			visualizeIteration(arr, i, i+1, delay)
+			visualizeIteration(arr, i, i+1, displayType, delay)
 			if arr[i] > arr[i+1] {
-				visualizeIteration(arr, i, i+1, delay)
+				visualizeIteration(arr, i, i+1, displayType, delay)
 				arr[i], arr[i+1] = arr[i+1], arr[i]
-				visualizeIteration(arr, i, i+1, delay)
+				visualizeIteration(arr, i, i+1, displayType, delay)
 				sorted = false
 			}
 		}
 
 		for i := 0; i <= n-2; i += 2 {
-			visualizeIteration(arr, i, i+1, delay)
+			visualizeIteration(arr, i, i+1, displayType, delay)
 			if arr[i] > arr[i+1] {
-				visualizeIteration(arr, i, i+1, delay)
+				visualizeIteration(arr, i, i+1, displayType, delay)
 				arr[i], arr[i+1] = arr[i+1], arr[i]
-				visualizeIteration(arr, i, i+1, delay)
+				visualizeIteration(arr, i, i+1, displayType, delay)
 				sorted = false
 			}
 		}

selection_sort.go

@@ -4,17 +4,17 @@ import (
 	"time"
 )
 
-func selectionSortVisualizer(arr []int, delay time.Duration) {
+func selectionSortVisualizer(arr []int, displayType string, delay time.Duration) {
 	for i := 0; i < len(arr); i++ {
 		min := i
 		for j := i + 1; j < len(arr); j++ {
-			visualizeIteration(arr, i, j, delay)
+			visualizeIteration(arr, i, j, displayType, delay)
 			if arr[j] < arr[min] {
-				visualizeIteration(arr, i, j, delay)
+				visualizeIteration(arr, i, j, displayType, delay)
 				min = j
 			}
 		}
 		arr[i], arr[min] = arr[min], arr[i]
-		visualizeIteration(arr, i, min, delay)
+		visualizeIteration(arr, i, min, displayType, delay)
 	}
 }