Game1.cs

namespace Pong_Clone
{
    using System;
    using Microsoft.Xna.Framework;
    using Microsoft.Xna.Framework.Graphics;
    using Microsoft.Xna.Framework.Input;

    public class Game1 : Microsoft.Xna.Framework.Game
    {
        public static GameStates gamestate;
        private GraphicsDeviceManager graphics;
        private SpriteBatch spriteBatch;
        private Bat rightBat;
        private Bat leftBat;
        private Ball ball;
        private Menu menu;
        private SpriteFont arial;
        private int resetTimer;
        private bool resetTimerInUse;
        private bool lastScored;

        private Input input;
        private int screenWidth;
        private int screenHeight;

        public enum GameStates
        {
            Menu,
            Running,
            End
        }

        public Game1()
        {
            graphics = new GraphicsDeviceManager(this);
            Content.RootDirectory = "Content";
        }

        protected override void Initialize()
        {
            screenHeight = 600;
            screenWidth = 800;
            menu = new Menu();
            gamestate = GameStates.Menu;
            resetTimer = 0;
            resetTimerInUse = true;
            lastScored = false;
            graphics.PreferredBackBufferWidth = screenWidth;
            graphics.PreferredBackBufferHeight = screenHeight;
            graphics.IsFullScreen = false;
            graphics.ApplyChanges();
            ball = new Ball(Content, new Vector2(screenWidth, screenHeight));
            SetUpMulti();
            input = new Input();
            base.Initialize();
        }

        protected override void LoadContent()
        {
            arial = Content.Load<SpriteFont>("Arial");
            spriteBatch = new SpriteBatch(GraphicsDevice);


        }

        protected override void UnloadContent()
        {
            // TODO: Unload any non ContentManager content here
        }

        private void SetUpSingle()
        {
            rightBat = new AIBat(Content, new Vector2(screenWidth, screenHeight), false);
            leftBat = new Bat(Content, new Vector2(screenWidth, screenHeight), true);
        }

        private void SetUpMulti()
        {
            rightBat = new Bat(Content, new Vector2(screenWidth, screenHeight), false);
            leftBat = new Bat(Content, new Vector2(screenWidth, screenHeight), true);
        }

        private void IncreaseSpeed()
        {
            ball.IncreaseSpeed();
            leftBat.IncreaseSpeed();
            rightBat.IncreaseSpeed();
        }

        protected override void Update(GameTime gameTime)
        {
            input.Update();

            if (gamestate == GameStates.Running)
            {
                if (leftBat.GetPoints() > 9)
                {
                    menu.InfoText = "Left Player Wins";
                    gamestate = GameStates.End;
                }
                else if (rightBat.GetPoints() > 9)
                {
                    menu.InfoText = "Right Player Wins";
                    gamestate = GameStates.End;
                }
                if (resetTimerInUse)
                {
                    resetTimer++;
                    ball.Stop();
                }

                if (resetTimer == 120)
                {
                    resetTimerInUse = false;
                    ball.Reset(lastScored);
                    resetTimer = 0;
                }

                if (rightBat.GetType() != typeof(Pong_Clone.AIBat))
                {
                    if (input.LeftDown) leftBat.MoveDown();
                    else if ((input.LeftUp)) leftBat.MoveUp();
                    if (input.RightDown) rightBat.MoveDown();
                    else if (input.RightUp) rightBat.MoveUp();
                }
                else if (rightBat.GetType() == typeof(Pong_Clone.AIBat))
                {
                    if (input.LeftDown) leftBat.MoveDown();
                    else if ((input.LeftUp)) leftBat.MoveUp();
                    if (input.RightDown) leftBat.MoveDown();
                    else if (input.RightUp) leftBat.MoveUp();
                }

                leftBat.UpdatePosition(ball);
                rightBat.UpdatePosition(ball);
                ball.UpdatePosition();
                if (ball.GetDirection() > 1.5f * Math.PI || ball.GetDirection() < 0.5f * Math.PI)
                {
                    if (rightBat.GetSize().Intersects(ball.GetSize()))
                    {
                        ball.BatHit(CheckHitLocation(rightBat));
                    }
                }
                else if (leftBat.GetSize().Intersects(ball.GetSize()))
                {
                    ball.BatHit(CheckHitLocation(leftBat));
                }



                if (!resetTimerInUse)
                {
                    if (ball.GetPosition().X > screenWidth)
                    {
                        resetTimerInUse = true;
                        lastScored = true;
                        leftBat.IncrementPoints();
                        IncreaseSpeed();
                    }
                    else if (ball.GetPosition().X < 0)
                    {

                        resetTimerInUse = true;
                        lastScored = false;

                        rightBat.IncrementPoints();
                        IncreaseSpeed();
                    }
                }
            }
            else if (gamestate == GameStates.Menu)
            {
                if (input.RightDown || input.LeftDown)
                {
                    menu.Iterator++;
                }
                else if (input.RightUp || input.LeftUp)
                {
                    menu.Iterator--;
                }

                if (input.MenuSelect)
                {
                    if (menu.Iterator == 0)
                    {
                        gamestate = GameStates.Running;
                        SetUpSingle();
                    }
                    else if (menu.Iterator == 1)
                    {
                        gamestate = GameStates.Running;
                        SetUpMulti();
                    }
                    else if (menu.Iterator == 2)
                    {
                        this.Exit();
                    }
                    menu.Iterator = 0;
                }
            }
            else if (gamestate == GameStates.End)
            {
                if (input.MenuSelect)
                {
                    gamestate = GameStates.Menu;
                }
            }

            // TODO: Add your update logic here
            base.Update(gameTime);
        }

        private int CheckHitLocation(Bat bat)
        {
            int block = 0;
            if (ball.GetPosition().Y < bat.GetPosition().Y + bat.GetSize().Height / 20) block = 1;
            else if (ball.GetPosition().Y < bat.GetPosition().Y + bat.GetSize().Height / 10 * 2) block = 2;
            else if (ball.GetPosition().Y < bat.GetPosition().Y + bat.GetSize().Height / 10 * 3) block = 3;
            else if (ball.GetPosition().Y < bat.GetPosition().Y + bat.GetSize().Height / 10 * 4) block = 4;
            else if (ball.GetPosition().Y < bat.GetPosition().Y + bat.GetSize().Height / 10 * 5) block = 5;
            else if (ball.GetPosition().Y < bat.GetPosition().Y + bat.GetSize().Height / 10 * 6) block = 6;
            else if (ball.GetPosition().Y < bat.GetPosition().Y + bat.GetSize().Height / 10 * 7) block = 7;
            else if (ball.GetPosition().Y < bat.GetPosition().Y + bat.GetSize().Height / 10 * 8) block = 8;
            else if (ball.GetPosition().Y < bat.GetPosition().Y + bat.GetSize().Height / 20 * 19) block = 9;
            else block = 10;
            return block;
        }

        protected override void Draw(GameTime gameTime)
        {
            GraphicsDevice.Clear(Color.Black);

            // TODO: Add your drawing code here
            spriteBatch.Begin();
            if (gamestate == GameStates.Running)
            {
                leftBat.Draw(spriteBatch);
                rightBat.Draw(spriteBatch);
                ball.Draw(spriteBatch);
                spriteBatch.DrawString(arial, leftBat.GetPoints().ToString(), new Vector2(screenWidth / 4 - arial.MeasureString(rightBat.GetPoints().ToString()).X, 20), Color.White);
                spriteBatch.DrawString(arial, rightBat.GetPoints().ToString(), new Vector2(screenWidth / 4 * 3 - arial.MeasureString(rightBat.GetPoints().ToString()).X, 20), Color.White);
            }
            else if (gamestate == GameStates.Menu)
            {
                menu.DrawMenu(spriteBatch, screenWidth, arial);
            }
            else if (gamestate == GameStates.End)
            {
                menu.DrawEndScreen(spriteBatch, screenWidth, arial);
            }
            spriteBatch.End();

            base.Draw(gameTime);
        }
    }
}