963448026

Build Type

github

Committed by GitHub

Commit Message

Merge 05343418f into 42d106c9c

Pull Request Pull Request #1205: Feature 1164 two handed weapon mechanics

Run Details

1968 of 3760 branches covered (52.34%)

Branch coverage included in aggregate %.

50 of 67 new or added lines in 3 files covered. (74.63%)

1 existing line in 1 file now uncovered.

5045 of 7958 relevant lines covered (63.4%)

37040.69 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

77.78

/Zilon.Core/Zilon.Core/Tactics/Behaviour/EquipTask.cs

﻿using System;
using System.Collections.Generic;
using System.Diagnostics;
using System.Linq;

using Zilon.Core.Components;
using Zilon.Core.PersonModules;
using Zilon.Core.Persons;
using Zilon.Core.Props;
using Zilon.Core.Schemes;

namespace Zilon.Core.Tactics.Behaviour
{
    /// <summary>
    /// Задача на назначение экипировки в указанный слот.
    /// </summary>
    public class EquipTask : OneTurnActorTaskBase
    {
        private readonly Equipment? _equipment;
        private readonly int _slotIndex;

        public EquipTask(IActor actor,
            IActorTaskContext context,
            Equipment? equipment,
            int slotIndex) :
            base(actor, context)
        {
            _equipment = equipment;
            _slotIndex = slotIndex;
        }

        protected override void ExecuteTask()
        {
            // Первоначально процедура экипировки предмета в слот инвентаря
            // делится на перемещение предмета в слот и обнуление слота.
            if (_equipment != null)
            {
                EquipPropToSlot();
            }
            else
            {
                Actor.Person.UnequipProp(_slotIndex);
            }
        }

        private void EquipOneSlotEquipmentFromInventory(
            IEquipmentModule equipmentModule,
            Equipment targetEquipment,
            Equipment? currentEquipment)
        {
            if (currentEquipment != null)
            {
                Actor.Person.GetModule<IInventoryModule>().Add(currentEquipment);
            }

            Actor.Person.GetModule<IInventoryModule>().Remove(targetEquipment);
            equipmentModule[_slotIndex] = targetEquipment;
        }

        private void EquipPropToSlot()
        {
            var equipmentCarrier = Actor.Person.GetModule<IEquipmentModule>();

            // Предмет может быть экипирован из инвентаря (и) или из другого слота (с).
            // Предмет может быть экипирован в пустой слот (0) и слот, в котором уже есть другой предмет (1).
            //    (и)                                           (с)
            // (0) изымаем предмет из инвентаря                 меняем предметы в слотах местами
            // (1) изымаем из инвентаря, а текущий в инвентярь  меняем предметы в слотах местами
            //TODO Ещё в этой схеме учитывает однослотовые и двуслотовые предметы.

            // проверяем, есть ли в текущем слоте предмет (0)/(1).
            var currentEquipment = equipmentCarrier[_slotIndex];

            // проверяем, из инвентаря или из слота экипируем (и)/(с)
            if (_equipment is null)
            {
                // There is no reason to check equipemnt is null.
                // But analyser can't understand condition in parent method ExecuteTask.
                throw new InvalidOperationException();
            }

            var currentEquipedSlotIndex = FindPropInEquiped(_equipment, equipmentCarrier);

            var equipScheme = _equipment.Scheme.Equip;
            if (equipScheme is null)
            {
                throw new InvalidOperationException($"{_equipment.Scheme.Sid} is not equipment.");
            }

            if (currentEquipedSlotIndex is null)
            {
                // (и)

                // текущий предмет возвращаем в инвентарь (1)
                // при (0) ничего не делаем

                if (IsTargetSlotBeHand(equipmentCarrier, _slotIndex))
                {
                    // Follow to logic of one/two hand equipment
                    var equipRestrictions = equipScheme.EquipRestrictions;
                    if (equipRestrictions is null || equipRestrictions.PropHandUsage is null)
                    {
                        // Equip one-handed item in a specified hand.

                        EquipOneSlotEquipmentFromInventory(equipmentCarrier, _equipment, currentEquipment);
                    }
                    else if (equipRestrictions.PropHandUsage.GetValueOrDefault().HasFlag(PropHandUsage.TwoHanded))
                    {
                        // Equip two handed weapon/tool in 2 of all slots.

                        EquipTwoSlotEquipmentFromInventory(equipmentCarrier, _equipment, currentEquipment);
                    }
                    else
                    {
                        //Do nothings beacause this state is unknown.
                        Debug.Fail("Unknown state");
                    }
                }
                else
                {
                    // Follow to logic of other slots.

                    EquipOneSlotEquipmentFromInventory(equipmentCarrier, _equipment, currentEquipment);
                }
            }
            else
            {
                // (с)

                if (currentEquipment != null)
                {
                    if (IsTargetSlotBeHand(equipmentCarrier, _slotIndex))
                    {
                        var equipRestrictions = equipScheme.EquipRestrictions;
                        if (equipRestrictions is null || equipRestrictions.PropHandUsage is null)
                        {
                            // (1) Ставим существующий в данном слоте предмет в слот, в котором был выбранный предмет
                            equipmentCarrier[currentEquipedSlotIndex.Value] = currentEquipment;
                        }
                        else if (equipRestrictions.PropHandUsage.GetValueOrDefault().HasFlag(PropHandUsage.TwoHanded))
                        {
                            //TODO this must be rewriten to meet issue requirements.
                            // Considering a two-handed item must takes 2 hand slots.
                            // So target item can take 2 slots, no hand slots will be free, current equipment must be place in inventory.

                            // (1) Ставим существующий в данном слоте предмет в слот, в котором был выбранный предмет
                            equipmentCarrier[currentEquipedSlotIndex.Value] = currentEquipment;
                        }
                    }
                    else
                    {
                        // (1) Ставим существующий в данном слоте предмет в слот, в котором был выбранный предмет
                        equipmentCarrier[currentEquipedSlotIndex.Value] = currentEquipment;
                    }
                }
                else
                {
                    if (IsTargetSlotBeHand(equipmentCarrier, _slotIndex))
                    {
                        var equipRestrictions = equipScheme.EquipRestrictions;
                        if (equipRestrictions is null || equipRestrictions.PropHandUsage is null)
                        {
                            // В старый слот выбранного предмета записываем пустоту.
                            // Потому что предмет перенесён из этого слота в другой.
                            equipmentCarrier[currentEquipedSlotIndex.Value] = null;
                        }
                        else if (equipRestrictions.PropHandUsage.GetValueOrDefault().HasFlag(PropHandUsage.TwoHanded))
                        {
                            //TODO this must be rewriten to meet issue requirements.
                            // Considering a two-handed item must takes 2 hand slots.
                            // So target item can take 2 slots, no hand slots will be free, current equipment must be place in inventory.

                            // В старый слот выбранного предмета записываем пустоту.
                            // Потому что предмет перенесён из этого слота в другой.
                            equipmentCarrier[currentEquipedSlotIndex.Value] = null;
                        }
                    }
                    else
                    {
                        // В старый слот выбранного предмета записываем пустоту.
                        // Потому что предмет перенесён из этого слота в другой.
                        equipmentCarrier[currentEquipedSlotIndex.Value] = null;
                    }
                }

                equipmentCarrier[_slotIndex] = _equipment;
            }
        }

        private void EquipTwoSlotEquipmentFromInventory(
            IEquipmentModule equipmentModule,
            Equipment targetEquipment,
            Equipment? currentEquipment)
        {
            var handSlotIndexes = GetHandSlotIndexes(equipmentModule);
            var otherHandSlotIndex = handSlotIndexes.First(x => x != _slotIndex);

            var equipmentInHandSlots = new List<Equipment>();
            var equipmentInTargetSlot = equipmentModule[_slotIndex];
            Debug.Assert(equipmentInTargetSlot == currentEquipment, "This is same equipment.");
            if (equipmentInTargetSlot is not null)
            {
                equipmentInHandSlots.Add(equipmentInTargetSlot);
            }

            var equipmentInOtherSlot = equipmentModule[otherHandSlotIndex];
            if (equipmentInOtherSlot is not null)
            {
                equipmentInHandSlots.Add(equipmentInOtherSlot);
            }

            foreach (var equipment in equipmentInHandSlots)
            {
                Actor.Person.GetModule<IInventoryModule>().Add(equipment);
            }

            Actor.Person.GetModule<IInventoryModule>().Remove(targetEquipment);
            equipmentModule[_slotIndex] = targetEquipment;
            equipmentModule[otherHandSlotIndex] = null;
        }

        /// <summary>
        /// Ищем предмет в уже экипированных.
        /// </summary>
        /// <param name="equipment"> Целевой предмет. </param>
        /// <param name="equipmentModule"> Объект для хранения экипировки. </param>
        /// <returns> Возвращает индекс слота, в котором находится указанный предмет. Или null, если предмет не найден. </returns>
        private static int? FindPropInEquiped(Equipment equipment, IEquipmentModule equipmentModule)
        {
            for (var i = 0; i < equipmentModule.Count(); i++)
            {
                if (equipmentModule[i] == equipment)
                {
                    return i;
                }
            }

            return null;
        }

        private static IEnumerable<int> GetHandSlotIndexes(IEquipmentModule equipmentModule)
        {
            var slots = equipmentModule.Slots;
            for (var slotIndex = 0; slotIndex < slots.Length; slotIndex++)
            {
                if (slots[slotIndex].Types.HasFlag(EquipmentSlotTypes.Hand))
                {
                    yield return slotIndex;
                }
            }
        }

        private static bool IsTargetSlotBeHand(IEquipmentModule equipmentModule, int slotIndex)
        {
            var slots = equipmentModule.Slots;
            if (!slots.Any())
            {
                // Module has no slots.
                // So it is not hand slot.
                // Used for tests.
                return false;
            }

            return slots[slotIndex].Types.HasFlag(EquipmentSlotTypes.Hand);
        }
    }
}

1	using System;
2	using System.Collections.Generic;
3	using System.Diagnostics;
4	using System.Linq;
5
6	using Zilon.Core.Components;
7	using Zilon.Core.PersonModules;
8	using Zilon.Core.Persons;
9	using Zilon.Core.Props;
10	using Zilon.Core.Schemes;
11
12	namespace Zilon.Core.Tactics.Behaviour
13	{
14	/// <summary>
15	/// Задача на назначение экипировки в указанный слот.
16	/// </summary>
17	public class EquipTask : OneTurnActorTaskBase
18	{
19	private readonly Equipment? _equipment;
20	private readonly int _slotIndex;
21
22	public EquipTask(IActor actor,
23	IActorTaskContext context,
24	Equipment? equipment,
25	int slotIndex) :
26	base(actor, context)	36✔
27	{
28	_equipment = equipment;	36✔
29	_slotIndex = slotIndex;	36✔
30	}	36✔
31
32	protected override void ExecuteTask()
33	{
34	// Первоначально процедура экипировки предмета в слот инвентаря
35	// делится на перемещение предмета в слот и обнуление слота.
36	if (_equipment != null)	35✔
37	{
38	EquipPropToSlot();	31✔
39	}
40	else
41	{
42	Actor.Person.UnequipProp(_slotIndex);	4✔
43	}
44	}	4✔
45
46	private void EquipOneSlotEquipmentFromInventory(
47	IEquipmentModule equipmentModule,
48	Equipment targetEquipment,
49	Equipment? currentEquipment)
50	{
51	if (currentEquipment != null)	25✔
52	{
53	Actor.Person.GetModule<IInventoryModule>().Add(currentEquipment);	6✔
54	}
55
56	Actor.Person.GetModule<IInventoryModule>().Remove(targetEquipment);	25✔
57	equipmentModule[_slotIndex] = targetEquipment;	25✔
58	}	25✔
59
60	private void EquipPropToSlot()
61	{
62	var equipmentCarrier = Actor.Person.GetModule<IEquipmentModule>();	31✔
63
64	// Предмет может быть экипирован из инвентаря (и) или из другого слота (с).
65	// Предмет может быть экипирован в пустой слот (0) и слот, в котором уже есть другой предмет (1).
66	// (и) (с)
67	// (0) изымаем предмет из инвентаря меняем предметы в слотах местами
68	// (1) изымаем из инвентаря, а текущий в инвентярь меняем предметы в слотах местами
69	//TODO Ещё в этой схеме учитывает однослотовые и двуслотовые предметы.
70
71	// проверяем, есть ли в текущем слоте предмет (0)/(1).
72	var currentEquipment = equipmentCarrier[_slotIndex];	31✔
73
74	// проверяем, из инвентаря или из слота экипируем (и)/(с)
75	if (_equipment is null)	31!
76	{
77	// There is no reason to check equipemnt is null.
78	// But analyser can't understand condition in parent method ExecuteTask.
79	throw new InvalidOperationException();	×
80	}
81
82	var currentEquipedSlotIndex = FindPropInEquiped(_equipment, equipmentCarrier);	31✔
83
84	var equipScheme = _equipment.Scheme.Equip;	31✔
85	if (equipScheme is null)	31!
86	{
NEW 87	throw new InvalidOperationException($"{_equipment.Scheme.Sid} is not equipment.");	×
88	}
89
90	if (currentEquipedSlotIndex is null)	31✔
91	{
92	// (и)
93
94	// текущий предмет возвращаем в инвентарь (1)
95	// при (0) ничего не делаем
96
97	if (IsTargetSlotBeHand(equipmentCarrier, _slotIndex))	29✔
98	{
99	// Follow to logic of one/two hand equipment
100	var equipRestrictions = equipScheme.EquipRestrictions;	26✔
101	if (equipRestrictions is null \|\| equipRestrictions.PropHandUsage is null)	26✔
102	{
103	// Equip one-handed item in a specified hand.
104
105	EquipOneSlotEquipmentFromInventory(equipmentCarrier, _equipment, currentEquipment);	22✔
106	}
107	else if (equipRestrictions.PropHandUsage.GetValueOrDefault().HasFlag(PropHandUsage.TwoHanded))	4✔
108	{
109	// Equip two handed weapon/tool in 2 of all slots.
110
111	EquipTwoSlotEquipmentFromInventory(equipmentCarrier, _equipment, currentEquipment);	4✔
112	}
113	else
114	{
115	//Do nothings beacause this state is unknown.
116	Debug.Fail("Unknown state");
117	}
118	}
119	else
120	{
121	// Follow to logic of other slots.
122
123	EquipOneSlotEquipmentFromInventory(equipmentCarrier, _equipment, currentEquipment);	3✔
124	}
125	}
126	else
127	{
128	// (с)
129
130	if (currentEquipment != null)	2✔
131	{
132	if (IsTargetSlotBeHand(equipmentCarrier, _slotIndex))	1!
133	{
NEW 134	var equipRestrictions = equipScheme.EquipRestrictions;	×
NEW 135	if (equipRestrictions is null \|\| equipRestrictions.PropHandUsage is null)	×
136	{
137	// (1) Ставим существующий в данном слоте предмет в слот, в котором был выбранный предмет
NEW 138	equipmentCarrier[currentEquipedSlotIndex.Value] = currentEquipment;	×
139	}
NEW 140	else if (equipRestrictions.PropHandUsage.GetValueOrDefault().HasFlag(PropHandUsage.TwoHanded))	×
141	{
142	//TODO this must be rewriten to meet issue requirements.
143	// Considering a two-handed item must takes 2 hand slots.
144	// So target item can take 2 slots, no hand slots will be free, current equipment must be place in inventory.
145
146	// (1) Ставим существующий в данном слоте предмет в слот, в котором был выбранный предмет
NEW 147	equipmentCarrier[currentEquipedSlotIndex.Value] = currentEquipment;	×
148	}
149	}
150	else
151	{
152	// (1) Ставим существующий в данном слоте предмет в слот, в котором был выбранный предмет
153	equipmentCarrier[currentEquipedSlotIndex.Value] = currentEquipment;	1✔
154	}
155	}
156	else
157	{
158	if (IsTargetSlotBeHand(equipmentCarrier, _slotIndex))	1!
159	{
NEW 160	var equipRestrictions = equipScheme.EquipRestrictions;	×
NEW 161	if (equipRestrictions is null \|\| equipRestrictions.PropHandUsage is null)	×
162	{
163	// В старый слот выбранного предмета записываем пустоту.
164	// Потому что предмет перенесён из этого слота в другой.
NEW 165	equipmentCarrier[currentEquipedSlotIndex.Value] = null;	×
166	}
NEW 167	else if (equipRestrictions.PropHandUsage.GetValueOrDefault().HasFlag(PropHandUsage.TwoHanded))	×
168	{
169	//TODO this must be rewriten to meet issue requirements.
170	// Considering a two-handed item must takes 2 hand slots.
171	// So target item can take 2 slots, no hand slots will be free, current equipment must be place in inventory.
172
173	// В старый слот выбранного предмета записываем пустоту.
174	// Потому что предмет перенесён из этого слота в другой.
NEW 175	equipmentCarrier[currentEquipedSlotIndex.Value] = null;	×
176	}
177	}
178	else
179	{
180	// В старый слот выбранного предмета записываем пустоту.
181	// Потому что предмет перенесён из этого слота в другой.
182	equipmentCarrier[currentEquipedSlotIndex.Value] = null;	1✔
183	}
184	}
185
186	equipmentCarrier[_slotIndex] = _equipment;	2✔
187	}
188	}	2✔
189
190	private void EquipTwoSlotEquipmentFromInventory(
191	IEquipmentModule equipmentModule,
192	Equipment targetEquipment,
193	Equipment? currentEquipment)
194	{
195	var handSlotIndexes = GetHandSlotIndexes(equipmentModule);	4✔
196	var otherHandSlotIndex = handSlotIndexes.First(x => x != _slotIndex);	4✔
197
198	var equipmentInHandSlots = new List<Equipment>();	4✔
199	var equipmentInTargetSlot = equipmentModule[_slotIndex];	4✔
200	Debug.Assert(equipmentInTargetSlot == currentEquipment, "This is same equipment.");
201	if (equipmentInTargetSlot is not null)	4✔
202	{
203	equipmentInHandSlots.Add(equipmentInTargetSlot);	3✔
204	}
205
206	var equipmentInOtherSlot = equipmentModule[otherHandSlotIndex];	4✔
207	if (equipmentInOtherSlot is not null)	4✔
208	{
209	equipmentInHandSlots.Add(equipmentInOtherSlot);	2✔
210	}
211
212	foreach (var equipment in equipmentInHandSlots)	18✔
213	{
214	Actor.Person.GetModule<IInventoryModule>().Add(equipment);	5✔
215	}
216
217	Actor.Person.GetModule<IInventoryModule>().Remove(targetEquipment);	4✔
218	equipmentModule[_slotIndex] = targetEquipment;	4✔
219	equipmentModule[otherHandSlotIndex] = null;	4✔
220	}	4✔
221
222	/// <summary>
223	/// Ищем предмет в уже экипированных.
224	/// </summary>
225	/// <param name="equipment"> Целевой предмет. </param>
226	/// <param name="equipmentModule"> Объект для хранения экипировки. </param>
227	/// <returns> Возвращает индекс слота, в котором находится указанный предмет. Или null, если предмет не найден. </returns>
228	private static int? FindPropInEquiped(Equipment equipment, IEquipmentModule equipmentModule)
229	{
230	for (var i = 0; i < equipmentModule.Count(); i++)	448✔
231	{
232	if (equipmentModule[i] == equipment)	195✔
233	{
234	return i;	2✔
235	}
236	}
237
238	return null;	29✔
239	}
240
241	private static IEnumerable<int> GetHandSlotIndexes(IEquipmentModule equipmentModule)
242	{
243	var slots = equipmentModule.Slots;
244	for (var slotIndex = 0; slotIndex < slots.Length; slotIndex++)
245	{
246	if (slots[slotIndex].Types.HasFlag(EquipmentSlotTypes.Hand))
247	{
248	yield return slotIndex;
249	}
250	}
251	}
252
253	private static bool IsTargetSlotBeHand(IEquipmentModule equipmentModule, int slotIndex)
254	{
255	var slots = equipmentModule.Slots;	31✔
256	if (!slots.Any())	31✔
257	{
258	// Module has no slots.
259	// So it is not hand slot.
260	// Used for tests.
261	return false;	4✔
262	}
263
264	return slots[slotIndex].Types.HasFlag(EquipmentSlotTypes.Hand);	27✔
265	}
266	}
267	}

kreghek / Zilon_Roguelike / 963448026

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous