void OnEnable()
{
setNames = SolutionServer.Instance().GetSetNames();
threeBodySoln = (ThreeBodySolution)target;
// find index of existing soln
setIndex = 0;
for (int i = 0; i < setNames.Length; i++)
{
if (setNames[i] == threeBodySoln.solutionSet)
{
setIndex = i;
break;
}
}
solutionNames = SolutionServer.Instance().GetSolutionNamesForSet(setNames[setIndex]);
solutionIndex = 0;
for (int i = 0; i < solutionNames.Length; i++)
{
if (solutionNames[i] == threeBodySoln.solutionName)
{
solutionIndex = i;
break;
}
}
}
private void SetBodies(NBody[] nbodies)
{
double[,] x = new double[3, 3];
double[,] v = new double[3, 3];
GravityEngine.Algorithm algorithm = GravityEngine.Algorithm.AZTRIPLE;
// Scale is a suggested visual scale based on how solution evolves.
// Currently not used.
float scale = 1.0f; // visual scale to display solution
if (!SolutionServer.Instance().GetData(solutionSet, solutionName, ref x, ref v, ref algorithm, ref scale))
{
Debug.LogError("Could not find set " + solutionSet + " soln=" + solutionName);
return;
}
for (int i = 0; i < 3; i++)
{
nbodies[i].vel = new Vector3((float)v[i, 0], (float)v[i, 1], (float)v[i, 2]);
// Awkward: Inflate positions by the NBE scale factor so that when NBE adds them (and removes the scale
// factor) it comes out as a wash. This is because the literature uses [-1..1] and this awkward approach
// is easier than scaling the velocities and masses.
Vector3 position = new Vector3((float)x[i, 0], (float)x[i, 1], (float)x[i, 2]) * gravityEngine.physToWorldFactor;
nbodies[i].initialPos = position;
nbodies[i].transform.position = position;
}
// solution server provides a per-solution algorithm
GravityEngine.instance.algorithm = algorithm;
}
public static SolutionServer Instance()
{
if (instance == null)
{
instance = new SolutionServer();
}
return(instance);
}
public override void OnInspectorGUI()
{
GUI.changed = false;
int oldSetIndex = setIndex;
setIndex = EditorGUILayout.Popup("Solution Set", setIndex, setNames);
if (setIndex != oldSetIndex)
{
// need to reset on set change to ensure index is in bounds
solutionIndex = 0;
}
string solutionSet = null;
string solutionName = null;
GameObject body1 = null;
GameObject body2 = null;
GameObject body3 = null;
solutionNames = SolutionServer.Instance().GetSolutionNamesForSet(setNames[setIndex]);
solutionIndex = EditorGUILayout.Popup("Solution", solutionIndex, solutionNames);
solutionSet = setNames[setIndex];
solutionName = solutionNames[solutionIndex];
// bodies
body1 = (GameObject)EditorGUILayout.ObjectField(
new GUIContent("Body 1", "Game object with NBody"), threeBodySoln.body1, typeof(GameObject), true);
body2 = (GameObject)EditorGUILayout.ObjectField(
new GUIContent("Body 2", "Game object with NBody"), threeBodySoln.body2, typeof(GameObject), true);
body3 = (GameObject)EditorGUILayout.ObjectField(
new GUIContent("Body 3", "Game object with NBody"), threeBodySoln.body3, typeof(GameObject), true);
EditorGUILayout.LabelField("Solutions are in positions [-1..1].");
EditorGUILayout.LabelField("To enlarge use GravityEngine Phy To World scaling.");
if (GUI.changed)
{
Undo.RecordObject(threeBodySoln, "ThreeBodySolution Change");
threeBodySoln.solutionSet = solutionSet;
threeBodySoln.solutionName = solutionName;
threeBodySoln.body1 = body1;
threeBodySoln.body2 = body2;
threeBodySoln.body3 = body3;
EditorUtility.SetDirty(threeBodySoln);
}
}
