FunWithList/PlanetSystem.cs at master · Mee7891/FunWithList · 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
105
106
107
108
109
110
111
112
using System;
using System.Collections;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;

namespace FunWithList
{
    class PlanetSystem: IEnumerable<Planet>
    {
        public string StarName { get; private set; }

        List<Planet> planets;

        SortState sortState = SortState.byDistance;

        public PlanetSystem(string starName)
        {
            StarName = starName;
            planets = new List<Planet>();
        }

        public void AddPlanet(Planet newPlanet)
        {
            planets.Add(newPlanet);
            newPlanet.plSystem = this;

            switch(sortState)
            {
                case SortState.byMass:
                    {
                        SortByMass();
                        break;
                    }
                case SortState.byRadius:
                    {
                        SortByRadius();
                        break;
                    }
                case SortState.byDistance:
                    {
                        SortByDistance();
                        break;
                    }
            }
        }

        public void SortByMass()
        {
            if (planets.Count > 0)
            {
                List<Planet> planetsOrdered  =  planets.OrderBy(p => p.Mass).ToList();

                planets = planetsOrdered;

                for (int i = planets.Count; --i > 0;)
                {
                    planets[i - 1].nextMassPl = planets[i];
                }
            }

            sortState = SortState.byMass;
        }

        public void SortByRadius()
        {
            if (planets.Count > 0)
            {
                List<Planet> planetsOrdered = planets.OrderBy(p => p.Radius).ToList();

                planets = planetsOrdered;

                for (int i = planets.Count; --i > 0;)
                {
                    planets[i - 1].nextRadiusPl = planets[i];
                }
            }

            sortState = SortState.byRadius;
        }

        public void SortByDistance()
        {
            if (planets.Count > 0)
            {
                List<Planet> planetsOrdered = planets.OrderBy(p => p.Distance).ToList();

                planets = planetsOrdered;

                for (int i = planets.Count; --i > 0;)
                {
                    planets[i - 1].nextDistancePl = planets[i];
                }
            }

            sortState = SortState.byDistance;
        }

        IEnumerator<Planet> IEnumerable<Planet>.GetEnumerator()
        {
            return ((IEnumerable<Planet>)planets).GetEnumerator();
        }

        IEnumerator IEnumerable.GetEnumerator()
        {
            return ((IEnumerable<Planet>)this).GetEnumerator();
        }

        enum SortState { byMass, byRadius, byDistance };
    }
}