private void btnRemoveWagons_Click(object sender, EventArgs e)
{
int wagonNumber;
if (lstViewWagons.SelectedItems.Count == 1)
{
if (int.TryParse(lstViewWagons.SelectedItems[0].SubItems[0].Text, out wagonNumber))
{
Wagon remove = administration.FindWagon(wagonNumber);
if (remove != null)
{
administration.Remove(remove);
lstViewWagons.SelectedItems[0].Remove();
foreach (Train t in administration.Trains)
{
t.Wagons.Remove(remove); //remove wagons
remove.RemoveWagonFromTrain();
}
updateListViewTrains();
}
else
{
MessageBox.Show("Could not find wagon");
}
}
}
else
{
MessageBox.Show("Select a wagon to remove");
}
}
private void btnRemove_Click(object sender, EventArgs e) //remove wagon from train
{
int wagonNumber;
int trainUnit;
if (!int.TryParse(cmboBoxWagons.Text, out wagonNumber))
{
MessageBox.Show("Invalid input. Please only enter numbers");
}
else
{
if (lstViewTrains.SelectedItems.Count != 1)
{
MessageBox.Show("Please select a train");
}
else
{
if (int.TryParse(lstViewTrains.SelectedItems[0].SubItems[0].Text, out trainUnit))
{
Train findTrain = administration.FindTrain(trainUnit);
Wagon findWagon = administration.FindWagon(wagonNumber);
if (findTrain != null && findWagon != null)
{
if (findTrain.Wagons.Remove(findWagon))
{
findWagon.RemoveWagonFromTrain(); //remove wagon from train
updateListViewTrains();
}
else
{
MessageBox.Show("Selected train doesn't have that wagon");
}
}
else
{
MessageBox.Show("Could not find wagon.");
}
}
}
}
}
