Created Bar Color (Red, Green, Blue)

Author: musiolmarco

out/production/Java-Algorithm-Visualization/com/marcomsl/algorithmvisualization/AlgorithmFrame.class

@@ -8,7 +8,7 @@ public class AlgorithmFrame extends JFrame {
     int size;
 
 
-    public AlgorithmFrame(int[] array){
+    public AlgorithmFrame(int[] array, BarColor barColor){
 
         size = array.length * 10 + 16;
 
@@ -20,7 +20,7 @@ public AlgorithmFrame(int[] array){
         setBackground(Color.WHITE);
         setSize(size, size);
 
-        ValueBars valueBars = new ValueBars(array, size);
+        ValueBars valueBars = new ValueBars(array, size, barColor);
         valueBars.setBounds(0,0,size, size);
         valueBars.setVisible(true);
         valueBars.setBackground(Color.WHITE);

out/production/Java-Algorithm-Visualization/com/marcomsl/algorithmvisualization/AlgorithmVisualizer.class

@@ -10,11 +10,20 @@ public class AlgorithmVisualizer {
 
     private int[] array;
     private ArrayList<Integer> arrayList = new ArrayList<>();
+    private BarColor barColor;
 
     public AlgorithmVisualizer() {
         array = new int[50];
         fillArray();
-        algorithmFrame = new AlgorithmFrame(array);
+        algorithmFrame = new AlgorithmFrame(array, barColor);
+
+    }
+
+    public AlgorithmVisualizer(BarColor barColor) {
+        this.barColor = barColor;
+        array = new int[50];
+        fillArray();
+        algorithmFrame = new AlgorithmFrame(array, barColor);
 
     }
 
@@ -25,7 +34,21 @@ public AlgorithmVisualizer(int size) {
             array = new int[50];
         }
         fillArray();
-        algorithmFrame = new AlgorithmFrame(array);
+        algorithmFrame = new AlgorithmFrame(array, barColor);
+
+    }
+
+    public AlgorithmVisualizer(int size, BarColor barColor) {
+        if (size >= 25 && size <= 50) {
+            array = new int[size];
+        } else {
+            array = new int[50];
+        }
+
+        this.barColor = barColor;
+
+        fillArray();
+        algorithmFrame = new AlgorithmFrame(array, barColor);
 
     }
 

out/production/Java-Algorithm-Visualization/com/marcomsl/algorithmvisualization/BarColor.class

@@ -0,0 +1,7 @@
+package com.marcomsl.algorithmvisualization;
+
+public enum BarColor {
+
+    RED, GREEN, BLUE;
+
+}

out/production/Java-Algorithm-Visualization/com/marcomsl/algorithmvisualization/Main.class

@@ -6,7 +6,7 @@ public class Main {
     // main function for testing
     public static void main(String[] args) throws InterruptedException {
 
-        AlgorithmVisualizer algorithmVisualizer = new AlgorithmVisualizer(50);
+        AlgorithmVisualizer algorithmVisualizer = new AlgorithmVisualizer(50, BarColor.RED);
         algorithmVisualizer.selectionSort(100);
         algorithmVisualizer.resetArray();
         algorithmVisualizer.bubblesort(15);

out/production/Java-Algorithm-Visualization/com/marcomsl/algorithmvisualization/ValueBars.class

@@ -7,12 +7,14 @@ public class ValueBars extends JPanel {
 
     int[] array;
     int size;
+    BarColor barColor;
 
 
-    public ValueBars(int[] array, int size){
+    public ValueBars(int[] array, int size, BarColor barColor){
 
         this.array = array;
         this.size = size;
+        this.barColor = barColor;
 
     }
 
@@ -28,8 +30,20 @@ protected void paintComponent(Graphics g){
         g.setColor(Color.WHITE);
 
         for(int i = 0; i < array.length; i++){
-            if(array[i] <= 255 && array[i] >= 0){
-                g.setColor(new Color(0,array[i] / 2,0));
+            if(array[i] / 2<= 255 && array[i] / 2 >= 0){
+
+                if(barColor == BarColor.GREEN){
+                    g.setColor(new Color(0,array[i] / 2,0));
+                }else if(barColor == BarColor.RED){
+                    g.setColor(new Color(array[i] / 2,0,0));
+                }else if(barColor == BarColor.BLUE){
+                    g.setColor(new Color(0,0,array[i] / 2));
+                }else{
+                    g.setColor(Color.BLACK);
+                }
+
+            }else{
+                g.setColor(Color.BLACK);
             }
 
             g.fillRect(x , size - 60 - array[i], 10, array[i] + 30);