Merge remote-tracking branch 'origin/develop-teammanager-cubewars' into develop-teammanager-cubewars

Project/Assets/ML-Agents/Examples/CubeWars/Prefabs/CubeWarArea.prefab

       propertyPath: largeAgent
       value: 
       objectReference: {fileID: 5707177387575911509}
+    - target: {fileID: 5940168890073970201, guid: 4dd35c7381e7445c99133a34a83e21db,
+        type: 3}
+      propertyPath: role
+      value: 2
+      objectReference: {fileID: 0}
     - target: {fileID: 8083776131842293165, guid: 4dd35c7381e7445c99133a34a83e21db,
         type: 3}
       propertyPath: m_Name
       propertyPath: area
       value: 
       objectReference: {fileID: 1819751139121548}
+    - target: {fileID: 5940168890073970201, guid: 4dd35c7381e7445c99133a34a83e21db,
+        type: 3}
+      propertyPath: role
+      value: 1
+      objectReference: {fileID: 0}
     - target: {fileID: 8083776131842293165, guid: 4dd35c7381e7445c99133a34a83e21db,
         type: 3}
       propertyPath: m_Name

Project/Assets/ML-Agents/Examples/CubeWars/Scripts/SmallCubeAgent.cs

     float m_ShootTime;
     Rigidbody m_AgentRb;
     //float m_LaserLength;
-    float m_HitPoints;
-    public float moveSpeed;
     public Material normalMaterial;
     public Material weakMaterial;
     public Material deadMaterial;
+    [HideInInspector]
+    float m_HitPoints;
+    [HideInInspector]
+    float m_Damage;
+    [HideInInspector]
+    float m_Heal;
+    [HideInInspector]
+    float m_MoveSpeed;
+    [HideInInspector]
+    float m_Cooldown;
+
+    public enum Role
+    {
+        Healer,
+        Tank,
+        DPS
+    }
+
+    public Role role;
+        if (role == Role.Healer)
+        {
+            m_HitPoints = .7f;
+            m_Damage = 0f;
+            m_Heal = .7f;
+            m_MoveSpeed = 10f;
+            m_Cooldown = .2f;
+        }
+        else if (role == Role.DPS)
+        {
+            m_HitPoints = .5f;
+            m_Damage = .05f;
+            m_Heal = 0f;
+            m_MoveSpeed = 10f;
+            m_Cooldown = .25f;
+        }
+        else if (role == Role.Tank)
+        {
+            m_HitPoints = 1f;
+            m_Damage = .02f;
+            m_Heal = .2f;
+            m_MoveSpeed = 6f;
+            m_Cooldown = .4f;
+        }
+
         m_AgentRb = GetComponent<Rigidbody>();
         m_MyArea = area.GetComponent<CubeWarArea>();
         m_LargeAgent = largeAgent.GetComponent<LargeCubeAgent>();
                 }
             }
             transform.Rotate(rotateDir, Time.fixedDeltaTime * turnSpeed);
-            m_AgentRb.AddForce(dirToGo * moveSpeed, ForceMode.VelocityChange);
+            m_AgentRb.AddForce(dirToGo * m_MoveSpeed, ForceMode.VelocityChange);
         }
 
         if (m_Shoot)
             {
                 if (hit.collider.gameObject.CompareTag("StrongSmallAgent") || hit.collider.gameObject.CompareTag("WeakSmallAgent"))
                 {
-                    hit.collider.gameObject.GetComponent<SmallCubeAgent>().HealAgent();
+                    hit.collider.gameObject.GetComponent<SmallCubeAgent>().HealAgent(m_Heal);
-                    hit.collider.gameObject.GetComponent<LargeCubeAgent>().HitAgent(.02f);
+                    hit.collider.gameObject.GetComponent<LargeCubeAgent>().HitAgent(m_Damage);
 
                     AddReward(.1f + .4f * m_Bonus);
                 }
-        else if (Time.time > m_ShootTime + .25f)
+        else if (Time.time > m_ShootTime + m_Cooldown)
         {
             myLaser.isFired = false;
         }
         }
     }
 
-    public void HealAgent()
+    public void HealAgent(float heal)
-            m_HitPoints = Mathf.Min(m_HitPoints + .25f, 1f);
+            m_HitPoints = Mathf.Min(m_HitPoints + heal, 1f);
             HealthStatus();
         }
     }