class RubberBand : public LocalLoop
{
public:
	virtual void MouseMove(Point p, dword keyflags) 	{ points.Add(p); GetMaster().Refresh(); }
	virtual void LeftUp(Point p, dword keyflags)    	{ EndLoop(); }
	virtual void RightUp(Point p, dword keyflags)   	{ EndLoop(); }

	const Vector<Point> &	GetPoints()					{ return points; }
	void 					Clear()						{ points.Clear(); }
private:
	Vector<Point> points;
};

void AWindow::LeftDown(Point p, dword keyflags)
{
	band.Clear();
	band.SetMaster(*this);
	band.Run();
	Refresh();
}

void AWindow::Paint(Draw& w)
{
	const Vector<Point> &p = band.GetPoints();
	
	w.DrawRect(GetSize(), SColorFace);
	
	for (int i = 0; i < p.GetCount()-1; i++)
		w.DrawLine(p[i], p[i+1]);
	if (!band.InLoop() && p.GetCount() > 1)
		w.DrawLine(p[p.GetCount()-1], p[0]);
	w.DrawText(4, 4, AsString(p.GetCount()));
}

class RubberBand : public LocalLoop
{
public:
	virtual void MouseMove(Point p, dword keyflags) 	{ AddPoint(p); GetMaster().Refresh(); }
	virtual void LeftUp(Point p, dword keyflags)    	{ EndLoop(); }
	virtual void RightUp(Point p, dword keyflags)   	{ EndLoop(); }

	const Vector<Point> &	GetPoints()					{ return points; }
	void 					Clear()						{ points.Clear(); }
private:
	Vector<Point> points;
	
	void AddPoint(const Point &newp);
};

void RubberBand::AddPoint(const Point &newp)
{
	if (points.GetCount() < 2)
		return points.Add(newp);
	// Get line vectors
	const Point &p = points[points.GetCount()-2];
	Point p1 = points.Top() - p;
	Point p2 = newp - p;
	// If gradient is different, add the new point
	if (p1.x*p2.y - p2.x*p1.y)
		points.Add(newp);
	// Otherwise update last points
	else
		points.Top() = newp;
}

	RubberBand 	band;
	BackDraw 	back;
	Size	 	backsz;
	
	void LeftDown(Point p, dword keyflags)
	{
		band.Clear();
		band.SetMaster(*this);
		Size sz = GetSize();
		if (sz != backsz) {
		    back.Create(sz);
		 	backsz = sz;   
		}
		Paint(back);
		band.Run();
		Refresh();
	}
	
	void Paint(Draw& w)
	{
		const Vector<Point> &p = band.GetPoints();
		if (!band.InLoop()) {
			// Normal painting
			w.DrawRect(GetSize(), White);	
			if (p.GetCount() >= 2) {
				w.DrawPolyline(p);	
				w.DrawLine(p[p.GetCount()-1], p[0]);
			}
						
		}
		else {
                        // RubberBand painting
			if (p.GetCount() >= 2)
				back.DrawLine(p[p.GetCount()-2], p.Top());
			back.Put(w, 0, 0);
			w.DrawText(4, 4, AsString(p.GetCount()));
		}
	}

void RubberBand::AddPoint(const Point &newp)
{
	if (points.GetCount() < 2)
		return points.Add(newp);
	// Get line vectors
	const Point &p = points[points.GetCount()-2];
	Point p1 = points.Top() - p;
	Point p2 = newp - p;
	// If gradient is different, add the new point
	if (p1.x*p2.y - p2.x*p1.y)
		points.Add(newp);
	// Otherwise update last points
	else
==>HERE		points.Top() = newp;
}

void RubberBandClass::AddPoint(const Point &newp) {
	if (points.GetCount() < 2)
		return points.Add(newp);
	// Get line vectors
	const Point &p_second_last = points[points.GetCount()-2];
	const Point &p_last = points.Top();
	
	Point p1 = p_last - p_second_last;
	Point p2 = newp - p_second_last;
	// If gradient is different, add the new point
	if (p1.x*p2.y - p2.x*p1.y)
		points.Add(newp);
	// Otherwise update last points if the newpoint is continue of the last point from the second last point
	else{
		int pos_from_last = 0;
		
		int pos_1=newp.x, pos_2=p_last.x, pos_3=p_second_last.x;
		
		if (!((p_last.x == p_second_last.x)&&(newp.x == p_last.x)))
			int pos_1=newp.y, pos_2=p_last.y, pos_3=p_second_last.y;
		
		if (pos_2 > pos_3 ){
			if (pos_1 > pos_2) {
				pos_from_last = 1;
			}
			else if ( pos_1 < pos_3) {
				pos_from_last = 2;
			}
		}
		else {
			if (pos_1 < pos_3) {
				pos_from_last = 1;
			}
			else if (pos_1 >pos_2){
				pos_from_last = 2;
			}
		}
		if (pos_from_last) points[points.GetCount()-pos_from_last] = newp;
	}
}