package studentCode;
import java.awt.Color;
import GridTools.MyGrid;

public class FlagMaker {

	public static void drawFlag(MyGrid grid, int countryCode) {
		
		int width = grid.getWd();
		int height = grid.getHt();
		
		if (countryCode == 1) {
			for (int col = 0; col < height; col++) {
				  drawRows(grid, col, Color.WHITE);
			}
			for (int col = 0; col < height / 2; col++) {
				  drawRows(grid, col, Color.RED);
				  }
		}
		if (countryCode == 2) {
			for (int col = 0; col < height; col++) {
				drawRows(grid, col, Color.RED);
			}
			for (int col = 0; col < (height*2)/3; col++) {
				drawRows(grid,col, Color.GREEN);
			}
			for (int col = 0; col < height/3; col++) {
				drawRows(grid, col, Color.YELLOW);
			}
		}
		if (countryCode == 3) {
			for (int col = 0; col < height; col++) {
				drawRows(grid, col, Color.GREEN);
			}
			for (int col = 0; col < (height*3)/4; col++) {
				drawRows(grid, col, Color.YELLOW);
				}
			for (int col = 0; col < (height/2); col++) {
				drawRows(grid, col, Color.BLUE);
				}
		}
		if (countryCode == 4) {
			for (int col = 0; col < height; col++) {
					drawRows(grid, col, Color.RED);
					}
				
			for (int row = 0; row < width/2; row++){
					drawColumns(grid, row, Color.WHITE);
			}
		}
		if (countryCode == 5) {
			for (int col = 0; col < height; col++) {
				drawRows(grid, col, Color.GREEN);
				}
			
			for (int row = 0; row < (height*2/3); row++) {
				for (int col = 0; col < row; col++) {
					drawRows(grid,row,Color.RED);
		}
			}
		for (int row = 0; row < (height*1/3); row++){
			drawRows(grid, row, Color.BLACK);
	}
		}
		if (countryCode == 6) {
			if (height % 2 == 0) {
				for (int col = 0; col < height; col++) {
					drawRows(grid, col, Color.BLUE);
				}
				for (int col = 0; col < (height/2 - 1); col++) {
					drawRows(grid, col, Color.GREEN);
				}
				for (int col = 1; col <=  width; col ++) {
					for(int row = 0; row < col; row++) {
						drawRows(grid, col, Color.RED);
					}
				}
				for (int col = width - 1; col > 0; col--) {
					for(int row = 0; row < col; row++) {
						drawRows(grid, col, Color.RED);
					}
				}
			}
		}
	
		if (countryCode == 8) {
			for (int col = 0; col < height; col++) {
				  drawRows(grid, col, Color.BLUE);
				}
			for (int col = (height*1/3); col < (height*2/3); col++) {
					  drawRows(grid, col, Color.YELLOW);
					  }
			}
		if (countryCode == 9) {
			for (int col = 0; col < height; col++) {
				drawRows(grid,col,Color.GREEN);
			}
			for (int col = (height*1/7); col < (height*6/7);col++) {
				drawRows(grid, col, Color.YELLOW);
			}
			for (int col = (height*2/7); col < (height*5/7);col++) {
				drawRows(grid, col, Color.RED);
			}
			for (int col = (height*3/7); col < (height*4/7);col++) {
				drawRows(grid, col, Color.BLACK);
			}
			
		}
		}
	
					 
	
	public static void drawRows(MyGrid grid, int rowNum, Color color) {
		int width = grid.getWd();
		int height = grid.getHt();
		for (int col = 0; col < width; col = col + 1) {
			grid.setColor(rowNum, col, color);
		}
	}
	
	public static void drawColumns(MyGrid grid, int colNum, Color color) {
		int width = grid.getWd();
		int height = grid.getHt();
		for (int row = 0; row < height; row++) {
			grid.setColor(colNum, row, color);
		}
	}

	
}