DayZOfficialSourceCode/Scripts/4_world/classes/weapondebug.c

enum eDebugMode
{
    NORMAL,
	MUZZLE_FIRE,
	CAMERA_MUZZLE_HYBRID,
	COUNT
};



class WeaponDebug
{
	const int BUFFER_SIZE = 1000;
	const float COLLISIONS_DISTANCE_TOLERANCE = 0.01;
	const float MAX_MUZZLE_DISTANCE_TOLERANCE = 20;
	const float CAMERA_BULLET_ORIGIN_OFFSET = 1;
	const float CAMERA_TRACE_MIN_DISTANCE_TOLERANCE = 0.3;
	
	Weapon m_WeaponInHands;
	int m_BufferIndex;
	bool m_IsDrawKeyHeldDown;
	bool m_IsLMBPressed;
	bool m_IsToggleKeyPressed;
	bool m_IsFireKeyPressed;
	float m_TargetDistance;
	eDebugMode m_CurrentMode;
	vector m_AimTrailCyclic[BUFFER_SIZE];	
	vector m_AimTrailOrdered[BUFFER_SIZE];
	
	ref map<int, string> m_DebugModesNames = new map<int, string>;
	
	//ref array<Selection> m_Selections = new array<Selection>();
	
	Shape m_Shape_usti;
	Shape m_Shape_konec;
	Shape m_ShapeFireDirection1;
	Shape m_ShapeFireDirection2;
	Shape m_HitShape;
	Shape m_ShapeEye;
	Shape m_ShapeTrailLines;
	Shape m_ShapeFireDirCamera;
	Shape m_HitShape2;
	Shape m_HitShape3;
	Shape m_HitShape4;
	//Shape temp_shape;
	Shape m_PermanentLine1;
	Shape m_PermanentLine2;
	
	Shape m_PermanentShape1;
	Shape m_PermanentShape2;
	
	Weapon GetWeaponInHands()
	{
		Weapon weapon_in_hands;
		PlayerBase player = PlayerBase.Cast(GetGame().GetPlayer());
		if( player && player.GetItemInHands() ) Class.CastTo(weapon_in_hands,  player.GetItemInHands());
		
		return weapon_in_hands;
	}
	
	void WeaponDebug()
	{
		m_DebugModesNames.Insert(eDebugMode.NORMAL, "Mode "+eDebugMode.NORMAL.ToString()+": shoot along the weapon barrel");
		m_DebugModesNames.Insert(eDebugMode.MUZZLE_FIRE,"Mode "+eDebugMode.MUZZLE_FIRE.ToString()+": shoot from muzzle end to camera trace hit point");
		m_DebugModesNames.Insert(eDebugMode.CAMERA_MUZZLE_HYBRID, "Mode "+eDebugMode.CAMERA_MUZZLE_HYBRID.ToString()+": shoot either from camera lens or from muzzle \n depending on the situation");
	}
	
	void ~WeaponDebug()
	{
		RemoveAllShapes(true);
		
	}

	void RemoveAllShapes(bool is_exit = false)
	{
		Debug.RemoveShape(m_Shape_usti);
		Debug.RemoveShape(m_Shape_konec);
		Debug.RemoveShape(m_ShapeFireDirection1);
		Debug.RemoveShape(m_ShapeFireDirection2);
		Debug.RemoveShape(m_ShapeEye);
		Debug.RemoveShape(m_ShapeTrailLines);
		Debug.RemoveShape(m_ShapeFireDirCamera);
		Debug.RemoveShape(m_HitShape);
		Debug.RemoveShape(m_HitShape2);
		Debug.RemoveShape(m_HitShape3);
		Debug.RemoveShape(m_HitShape4);
		
		if ( is_exit )
		{
			Debug.RemoveShape(m_PermanentLine1);
			Debug.RemoveShape(m_PermanentLine2);
			Debug.RemoveShape(m_PermanentShape1);
			Debug.RemoveShape(m_PermanentShape2);
		}
	}


	void OnCommandHandlerUpdate()
	{
			
		m_IsLMBPressed = false;
		m_IsFireKeyPressed = false;
		
		if (KeyState(KeyCode.KC_LWIN) == 1)
		{
			if (!m_IsDrawKeyHeldDown)
			{
				OnKeyDown(KeyCode.KC_LWIN);
			}
			m_IsDrawKeyHeldDown = true;
		}
		else
		{
			if ( m_IsDrawKeyHeldDown )
			{
				//OnKeyUp();
			}
			m_IsDrawKeyHeldDown = false;
		}
		
		if (KeyState(KeyCode.KC_Z) == 1)
		{
			ClearKey(KeyCode.KC_Z);
			CycleDebugMode();
		}
		
		if (KeyState(KeyCode.KC_X) == 1)
		{
			m_IsFireKeyPressed = true;
			ClearKey(KeyCode.KC_X);
		}

		if (GetMouseState(MouseState.LEFT) & MB_PRESSED_MASK)
		{
			m_IsLMBPressed = true;
		}
	}

	void OnKeyDown(KeyCode key)
	{
		if (key == KeyCode.KC_X)
		{
		}
	}
	
	void CycleDebugMode()
	{
		m_CurrentMode++;
		if ( m_CurrentMode == eDebugMode.COUNT )
		{
			m_CurrentMode = 0;
		}
	
	}
	
	void OnPostFrameUpdate()
	{
		if ( GetWeaponInHands() )
		{
			RemoveAllShapes();
			Weapon weapon = GetWeaponInHands();
			vector cameraDirection = GetGame().GetCurrentCameraDirection();
			vector cameraPosition = GetGame().GetCurrentCameraPosition();
			vector usti_hlavne_position = weapon.GetSelectionPositionMS( "usti hlavne" );//usti hlavne
			vector konec_hlavne_position = weapon.GetSelectionPositionMS( "konec hlavne" );//konec hlavne
			usti_hlavne_position = weapon.ModelToWorld(usti_hlavne_position);
			konec_hlavne_position = weapon.ModelToWorld(konec_hlavne_position);
			
			if ( m_CurrentMode == eDebugMode.NORMAL ) 
			{
				DrawLineOfFire(konec_hlavne_position,usti_hlavne_position );
			}
			
			if ( m_CurrentMode == eDebugMode.MUZZLE_FIRE ) 
			{
				DrawLineOfFireMuzzleToHit(usti_hlavne_position, cameraDirection, cameraPosition);
			}
			
			if ( m_CurrentMode == eDebugMode.CAMERA_MUZZLE_HYBRID ) 
			{
				DrawLineOfFireCameraHybrid(usti_hlavne_position, cameraDirection, cameraPosition, konec_hlavne_position);
			}
			
			if (m_IsDrawKeyHeldDown)
			{
				AddPosToCyclicBuffer(usti_hlavne_position);
				OrderTrailArray();
			}

			m_ShapeTrailLines = Debug.DrawLines( m_AimTrailOrdered,BUFFER_SIZE, COLOR_YELLOW, ShapeFlags.NOZBUFFER );
			
		
			DrawEyePoint(weapon);
			DrawBarrelMemoryPoints(konec_hlavne_position,usti_hlavne_position );
			DisplayGeneralInfo();
			DisplayTargetInfo();
			/*
			vector pos;
			weapon.GetProjectedCursorPos3d(pos);
			Debug.RemoveShape(temp_shape);
			temp_shape = Debug.DrawSphere(pos, 0.1, Colors.GREEN);
			*/
		}
	}

	void DrawBarrelMemoryPoints(vector begin_point, vector end_point)
	{
		if (!m_IsDrawKeyHeldDown)
		{
			m_Shape_usti = Debug.DrawSphere(end_point, 0.011, Colors.GREEN, ShapeFlags.TRANSP|ShapeFlags.NOOUTLINE|ShapeFlags.NOZBUFFER);
			m_Shape_konec = Debug.DrawSphere(begin_point, 0.011, Colors.GREEN, ShapeFlags.TRANSP|ShapeFlags.NOOUTLINE|ShapeFlags.NOZBUFFER);
		}
	}


	void DrawLineOfFire(vector begin_point, vector end_point)
	{
		vector contact_point;
		vector contact_dir;
		vector aim_point = end_point - begin_point;
		int contact_component;
		
		aim_point = aim_point.Normalized() * 100;
		aim_point = aim_point + end_point;
		
		m_ShapeFireDirection1 = Debug.DrawLine(end_point, aim_point);
		m_ShapeFireDirection2 = Debug.DrawLine(begin_point, end_point, ShapeFlags.NOZBUFFER );
		
		if ( DayZPhysics.RaycastRV(end_point, aim_point, contact_point, contact_dir, contact_component, null, null, null, false, false, ObjIntersectFire) )
		{	
			m_HitShape = Debug.DrawSphere(contact_point, 0.04, COLOR_RED);
		}
		
		if ( m_IsFireKeyPressed )
		{
			Debug.RemoveShape(m_PermanentLine1);
			m_PermanentLine1 = Debug.DrawLine(end_point, contact_point, Colors.RED, ShapeFlags.NOZBUFFER );
		}
	}

	void AddPosToCyclicBuffer(vector pos)
	{
		m_AimTrailCyclic[m_BufferIndex] = pos;
		m_BufferIndex++;
		if (m_BufferIndex == BUFFER_SIZE)
		{
			m_BufferIndex = 0;			
		}
	}	
	
	void OrderTrailArray()
	{
		int unordered_index;
		
		for (int i = 0; i < BUFFER_SIZE; i++)
		{
			unordered_index = m_BufferIndex + i;
			if ( unordered_index >= BUFFER_SIZE )
			{
				unordered_index = unordered_index - BUFFER_SIZE;
			}
			m_AimTrailOrdered[i] = m_AimTrailCyclic[unordered_index];
		}
	}	
	
	void DrawEyePoint(Weapon weapon)
	{
		vector position = GetEyePointPosition(weapon);
		m_ShapeEye = Debug.DrawSphere(position, 0.009, COLOR_BLUE, ShapeFlags.TRANSP|ShapeFlags.NOOUTLINE|ShapeFlags.NOZBUFFER);
	}

	vector GetEyePointPosition(Weapon weapon)
	{
		string memory_point_name = weapon.ConfigGetString("memoryPointCamera");
		ItemBase optics = weapon.GetAttachedOptics();
		if (optics)
		{
			memory_point_name = optics.ConfigGetString("memoryPointCamera");
			return optics.ModelToWorld(optics.GetSelectionPositionLS( memory_point_name ));
		}
		return weapon.ModelToWorld(weapon.GetSelectionPositionLS( memory_point_name ));
		
	}

	void DisplayGeneralInfo()
	{
		DbgUI.Begin("sway weight", 50, 50);
		DayZPlayerImplement player = DayZPlayerImplement.Cast(GetGame().GetPlayer());
		float sway_weight = player.GetAimingModel().GetSwayWeight();
		DbgUI.Text("value: " + sway_weight.ToString());
		DbgUI.Text("Hold LWIN to draw debug line");
		DbgUI.Text("Press X to simulate fire");
		DbgUI.Text("Press Z to cycle debug modes");
		DbgUI.Text("Debug " +m_DebugModesNames.Get(m_CurrentMode) );
		DbgUI.Text("");
		DbgUI.End();
	}

	void DisplayTargetInfo()
	{
		DbgUI.Begin("target distance", 50,200);
		DbgUI.Text("value: " + m_TargetDistance.ToString());
		DbgUI.End();
	}

		
	void DrawLineOfFireMuzzleToHit(vector begin_point, vector camera_dir, vector camera_pos)
	{
		vector contact_point_cam_trace;
		vector contact_point_muzzle_trace;
		
		vector contact_dir_muzzle;
		vector contact_dir_camera;
		
		vector contact_point;
		vector contact_dir;
		
		int contact_component;
		int contact_component_muzzle;
		
		vector end_point = camera_pos + camera_dir * 1000;
		Man player = GetGame().GetPlayer();
		Object player_o;
		Class.CastTo(player_o, player);
		
		if ( DayZPhysics.RaycastRV(camera_pos, end_point, contact_point_cam_trace, contact_dir_camera, contact_component,null, null, player_o , false, false, ObjIntersectFire, 0.1) )
		{
			m_ShapeFireDirCamera = Debug.DrawLine(begin_point, contact_point_cam_trace, Colors.RED, ShapeFlags.NOZBUFFER );
			m_HitShape2 = Debug.DrawSphere(contact_point_cam_trace, 0.03, Colors.GREEN);
			m_TargetDistance =  vector.Distance( player.GetPosition(), contact_point_cam_trace);
			
			if ( m_IsFireKeyPressed )
			{
				Debug.RemoveShape(m_PermanentLine1);
				Debug.RemoveShape(m_PermanentLine2);
				m_PermanentLine1 = Debug.DrawLine(begin_point, contact_point_cam_trace, Colors.RED, ShapeFlags.NOZBUFFER );
				m_PermanentLine2 = Debug.DrawLine(camera_pos, contact_point_cam_trace, Colors.GREEN, ShapeFlags.NOZBUFFER );
			}
			
		}
		
		else
		{
			m_ShapeFireDirCamera = Debug.DrawLine(begin_point, end_point, Colors.GREEN, ShapeFlags.NOZBUFFER );
			m_TargetDistance = -1;
		}
		
		if ( DayZPhysics.RaycastRV(begin_point, contact_point_cam_trace, contact_point_muzzle_trace, contact_dir_muzzle, contact_component_muzzle, null, null, null, false, false, ObjIntersectFire, 0.0) )
		{
			m_HitShape3 = Debug.DrawSphere(contact_point_muzzle_trace, 0.03, COLOR_RED);
		}
	}

	void DrawLineOfFireCameraHybrid(vector usti_hlavne_position, vector camera_dir, vector camera_pos, vector konec_hlavne_position)
	{
		bool muzzle_shot;
		vector contact_point_cam_trace;
		vector contact_point_muzzle_trace;
		
		vector aim_at_position;
		
		vector contact_dir_muzzle;
		vector contact_dir;
		
		vector contact_point;
		
		int contact_component;
		int contact_component_muzzle;
		
		float distance_to_aim_at;
		vector end_point = camera_pos + camera_dir * 100;
		vector start_point = camera_pos + (CAMERA_BULLET_ORIGIN_OFFSET * camera_dir);
		vector weapon_aim_direction = usti_hlavne_position - konec_hlavne_position;
		weapon_aim_direction.Normalize();
		
		Man player = GetGame().GetPlayer();
		Object player_o;
		Class.CastTo(player_o, player);

		if ( DayZPhysics.RaycastRV(start_point, end_point, contact_point_cam_trace, contact_dir, contact_component,null, null, player_o , false, false, ObjIntersectFire) )
		{
			m_TargetDistance = vector.Distance(start_point, contact_point_cam_trace);
			aim_at_position = contact_point_cam_trace;
			
			if ( DayZPhysics.RaycastRV(usti_hlavne_position, contact_point_cam_trace, contact_point_muzzle_trace, contact_dir, contact_component,null, null, player_o , false, false, ObjIntersectFire, 0.05) )
			{
				float collision_distance = vector.Distance(contact_point_cam_trace, contact_point_muzzle_trace);
				float muzzle_collision_distance = vector.Distance(usti_hlavne_position, contact_point_muzzle_trace);
				
				if ((collision_distance > 2 && muzzle_collision_distance < MAX_MUZZLE_DISTANCE_TOLERANCE) || m_TargetDistance < CAMERA_TRACE_MIN_DISTANCE_TOLERANCE)
				{
					muzzle_shot = true;
					aim_at_position = contact_point_muzzle_trace;
				}
			}
		}
		
		
		
		distance_to_aim_at = vector.Distance(camera_pos, aim_at_position);
		
		if ( m_IsFireKeyPressed )
		{
			Debug.RemoveShape(m_PermanentLine1);
			Debug.RemoveShape(m_PermanentLine2);
			Debug.RemoveShape(m_PermanentShape1);
			
			vector contact_point_temp;
			
			if (muzzle_shot)
			{
				m_PermanentLine1 = Debug.DrawLine(usti_hlavne_position, usti_hlavne_position + weapon_aim_direction * 5, Colors.RED, ShapeFlags.NOZBUFFER );
				m_PermanentLine2 = Debug.DrawLine(camera_pos, contact_point_cam_trace, Colors.GREEN, ShapeFlags.NOZBUFFER );
				PrintString("muzle shot");
			}
			else
			{
				vector dir = contact_point_cam_trace - camera_pos;
				dir.Normalize();
				dir = dir * 100;
				float dst = vector.Distance(camera_pos, contact_point_cam_trace);
				float dist2 = vector.Distance(camera_pos, contact_point_cam_trace);
				
				//m_PermanentShape1 = Debug.DrawSphere(start_point, 0.015,Colors.GREEN);
				m_PermanentLine1 = Debug.DrawLine(start_point, contact_point_cam_trace + dir, Colors.RED, ShapeFlags.NOZBUFFER );
				PrintString("camera shot");
			}
		}
		
		float clamped_distance = Math.Clamp(distance_to_aim_at, 0 , 100);
		float distance_normalized = Math.InverseLerp(0, 100, clamped_distance);
		float hit_sphere_size = Math.Lerp(0.025, 0.75, distance_normalized);
		
		Debug.RemoveShape(m_HitShape4);
		Debug.RemoveShape(m_HitShape);
		m_HitShape = Debug.DrawSphere(contact_point_cam_trace, 0.20,Colors.GREEN, ShapeFlags.TRANSP);
		m_HitShape4 = Debug.DrawSphere(aim_at_position, hit_sphere_size);
	
	}
	
}