using UnityEngine;
using System.Collections;
   
public class SwitchWeapon : MonoBehaviour {
     public GameObject gun1;
     public GameObject gun2;
     private int curWeapon;
  SmgController pp = GameObject.Find("1st Person Smg").GetComponent <SmgController>();
  PistolController tt = GameObject.Find("Pistol 1_st person").GetComponent <PistolController>();
     // Use this for initialization
     void Start ()  
     {
       
     }
   
       
     // Update is called once per frame
     void Update ()  
     {
         if(Input.GetAxis("Mouse ScrollWheel")<0)
         {
             curWeapon -= 1;
             StartCoroutine(Switch());
         }
         if(Input.GetAxis("Mouse ScrollWheel")>0)
         {
             curWeapon += 1;
             StartCoroutine(Switch());
         }
         if(curWeapon>2)
         {
             curWeapon = 1;
             StartCoroutine(Switch());
         }
         if(curWeapon<1)
         {
             curWeapon = 2;
             StartCoroutine(Switch());
         }
         if(Input.GetKeyDown("1"))
         {
             curWeapon = 1;
             StartCoroutine(Switch());
         }
         if(Input.GetKeyDown("2"))
         {
             curWeapon = 2;
             StartCoroutine(Switch());
         }
     }
     void Null()
     {
         gun1.SetActive(false);
         gun2.SetActive(false);
     }
     IEnumerator Switch()
     {
         if(curWeapon == 1)
         {
    SmgController pp = GameObject.Find("1st Person Smg").GetComponent <SmgController>();
    pp.enabled = false;
       gun2.animation.Play ("UnArm");
             yield return new WaitForSeconds(1f);
             Null();
             gun1.SetActive(true);
    tt.enabled = true;
         }
         if(curWeapon == 2)
         {
    PistolController tt = GameObject.Find("Pistol 1_st person").GetComponent <PistolController>();
    tt.enabled = false;
    gun1.animation.Play ("UnArm");
             yield return new WaitForSeconds(0.7f);
             Null();
             gun2.SetActive(true);
    pp.enabled = true;
         }
  }
         }