I have program of following format. I need the plot to be of different colors based on the if statement in program. How do I do it?

iter=1500;

for I=2:iter

function stage1

body //solves 2nd order ODE and gives displacement and velocity//

end function

function stage2

body //solves 2nd order ODE and gives displacement and velocity//

end function

if displacement >0 and velocity<0

execute stage1

else if (condition)

execute stage2

//The condition goes on//

endif

endfor

plot(tim(:),displ(:,1));

Now I want the time vs displ plot to be green for stage1 values and other colors for subsequent stage values.

[Jan]

There is no "end function" and "endfor" in Matlab. Please mention, if you use Octave. The pseudo-code does not allow to suggest a modification. Please post the actual code and ask a specific question.

[Rik]

closed as superseded by later questions