public void SetPrevBeltElement(IBeltElement beltElement)
{
UndergroundBelt belt = beltElement as UndergroundBelt;
if (transitionGround == TransitionGround.output && belt != null && belt.transitionGround == TransitionGround.input)
{
input = belt;
}
else if (transitionGround == TransitionGround.input && beltElement.GetOutputOrientation() == outputOrientation)
{
input = beltElement;
}
}
public void SetNextBeltElement(IBeltElement beltElement)
{
UndergroundBelt belt = beltElement as UndergroundBelt;
if (transitionGround == TransitionGround.input && belt != null && belt.transitionGround == TransitionGround.output)
{
output = belt;
}
else if (transitionGround == TransitionGround.output)
{
output = beltElement;
}
}
public bool CanConnectToBeltElement(IBeltElement prevElement)
{
UndergroundBelt belt = prevElement as UndergroundBelt;
if (transitionGround == TransitionGround.output && belt != null)
{
return(true);
}
else if (transitionGround == TransitionGround.input && prevElement.GetOutputOrientation() == this.outputOrientation)
{
return(true);
}
return(false);
}
