Ignition/ModuleIgnitionRCSController.cs at main · CharleRoger/Ignition · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
using System.Collections.Generic;
using System.Linq;

namespace Ignition
{
    class ModuleIgnitionRCSController : ModuleIgnitionThrusterController
    {
        private ModuleRCS _moduleRCS;
        private ModuleRCS ModuleRCS
        {
            get
            {
                if (_moduleRCS is null)
                {
                    var rcsModules = part.FindModulesImplementing<ModuleRCS>();
                    if (rcsModules.Count > 0) _moduleRCS = rcsModules.FirstOrDefault();
                }
                return _moduleRCS;
            }
        }

        protected override bool ModuleIsNull()
        {
            return ModuleRCS is null;
        }

        protected override string GetGuiGroupName()
        {
            return "RCS";
        }

        protected override List<Propellant> GetModulePropellants()
        {
            if (ModuleIsNull()) return new List<Propellant>();

            return ModuleRCS.propellants;
        }

        protected override void SetupOriginalData()
        {
            if (ModuleIsNull()) return;

            if (!(ModuleRCS.resourceName is null))
            {
                int propellantIndex = -1;
                for (int i = 0; i < ModuleRCS.propellants.Count; i++)
                {
                    if (ModuleRCS.propellants[i].name == ModuleRCS.resourceName)
                    {
                        propellantIndex = i;
                        break;
                    }
                }
                if (propellantIndex > -1) ModuleRCS.propellants.RemoveAt(propellantIndex);
            }

            if (ModuleRCS.atmosphereCurve.Curve.keys.Length == 0) return;
            if (MaxThrustOriginal == 0) MaxThrustOriginal = ModuleRCS.thrusterPower;
            if (IspVacuumOriginal == 0) IspVacuumOriginal = GetKeyframeValue(ModuleRCS.atmosphereCurve.Curve.keys, 0);
            if (IspSeaLevelOriginal == 0) IspSeaLevelOriginal = GetKeyframeValue(ModuleRCS.atmosphereCurve.Curve.keys, 1);
        }

        protected override void ApplyPropellantCombinationToModule()
        {
            if (ModuleIsNull()) return;

            ModuleRCS.thrusterPower = (float)MaxThrustCurrent;
            ModuleRCS.maxFuelFlow = MaxFuelFlowCurrent;
            ModuleRCS.atmosphereCurve.Curve.keys = GetIspKeys();
            ModuleRCS.propellants = GetAllCurrentPropellants(ModuleRCS.propellants);
        }

        protected override void RecompilePartInfo()
        {
            if (part.partInfo is null || part.partInfo.moduleInfos is null) return;

            var rcsModules = part.FindModulesImplementing<ModuleRCS>();
            var rcsIndex = 0;
            for (int i = 0; i < part.partInfo.moduleInfos.Count; i++)
            {
                if (part.partInfo.moduleInfos[i].moduleName != "RCS" || part.partInfo.moduleInfos[i].moduleName != "RCSFX") continue;
                part.partInfo.moduleInfos[i].info = rcsModules[rcsIndex].GetInfo();
                rcsIndex++;
            }
        }

        protected override double GetG()
        {
            if (ModuleIsNull()) return 9.80665;

            return (double)ModuleRCS.G;
        }

        protected override bool UseIspSeaLevel()
        {
            return true;
        }

        protected override double GetScaledMaxThrustOriginal()
        {
            return GetScale(RCSThrustScaleExponent) * MaxThrustOriginal;
        }
    }
}