[wxWidgets.git] / utils / ogl / samples / ogledit / doc.cpp

/////////////////////////////////////////////////////////////////////////////
// Name:        doc.cpp
// Purpose:     Implements document functionality in OGLEdit
// Author:      Julian Smart
// Modified by:
// Created:     12/07/98
// RCS-ID:      $Id$
// Copyright:   (c) Julian Smart
// Licence:   	wxWindows licence
/////////////////////////////////////////////////////////////////////////////

#ifdef __GNUG__
// #pragma implementation
#endif

// For compilers that support precompilation, includes "wx.h".
#include <wx/wxprec.h>

#ifdef __BORLANDC__
#pragma hdrstop
#endif

#ifndef WX_PRECOMP
#include <wx/wx.h>
#endif

#if !USE_DOC_VIEW_ARCHITECTURE
#error You must set USE_DOC_VIEW_ARCHITECTURE to 1 in wx_setup.h!
#endif

#include "ogledit.h"
#include "doc.h"
#include <wx/wxexpr.h>
#include "view.h"

IMPLEMENT_DYNAMIC_CLASS(DiagramDocument, wxDocument)

DiagramDocument::DiagramDocument(void)
{
}

DiagramDocument::~DiagramDocument(void)
{
}

bool DiagramDocument::OnCloseDocument(void)
{
  diagram.DeleteAllShapes();
  return TRUE;
}

ostream& DiagramDocument::SaveObject(ostream& stream)
{
  wxDocument::SaveObject(stream);
  
  char buf[400];
  (void) wxGetTempFileName("diag", buf);

  diagram.SaveFile(buf);
  wxTransferFileToStream(buf, stream);

  wxRemoveFile(buf);
  
  return stream;
}

istream& DiagramDocument::LoadObject(istream& stream)
{
  wxDocument::LoadObject(stream);

  char buf[400];
  (void) wxGetTempFileName("diag", buf);

  wxTransferStreamToFile(stream, buf);

  diagram.DeleteAllShapes();
  diagram.LoadFile(buf);
  wxRemoveFile(buf);

  return stream;
}

/*
 * Implementation of drawing command
 */

DiagramCommand::DiagramCommand(char *name, int command, DiagramDocument *ddoc, wxClassInfo *info, double xx, double yy,
  bool sel, wxShape *theShape, wxShape *fs, wxShape *ts):
  wxCommand(TRUE, name)
{
  doc = ddoc;
  cmd = command;
  shape = theShape;
  fromShape = fs;
  toShape = ts;
  shapeInfo = info;
  shapeBrush = NULL;
  shapePen = NULL;
  x = xx;
  y = yy;
  selected = sel;
  deleteShape = FALSE;
}

DiagramCommand::DiagramCommand(char *name, int command, DiagramDocument *ddoc, wxBrush *backgroundColour, wxShape *theShape):
  wxCommand(TRUE, name)
{
  doc = ddoc;
  cmd = command;
  shape = theShape;
  fromShape = NULL;
  toShape = NULL;
  shapeInfo = NULL;
  x = 0.0;
  y = 0.0;
  selected = FALSE;
  deleteShape = FALSE;
  shapeBrush = backgroundColour;
  shapePen = NULL;
}

DiagramCommand::DiagramCommand(char *name, int command, DiagramDocument *ddoc, const wxString& lab, wxShape *theShape):
  wxCommand(TRUE, name)
{
  doc = ddoc;
  cmd = command;
  shape = theShape;
  fromShape = NULL;
  toShape = NULL;
  shapeInfo = NULL;
  x = 0.0;
  y = 0.0;
  selected = FALSE;
  deleteShape = FALSE;
  shapeBrush = NULL;
  shapePen = NULL;
  shapeLabel = lab;
}

DiagramCommand::~DiagramCommand(void)
{
  if (shape && deleteShape)
  {
    shape->SetCanvas(NULL);
    delete shape;
  }
}

bool DiagramCommand::Do(void)
{
  switch (cmd)
  {
    case OGLEDIT_CUT:
    {
      if (shape)
      {
        deleteShape = TRUE;
        
        shape->Select(FALSE);
        
        // Generate commands to explicitly remove each connected line.
        RemoveLines(shape);
        
        doc->GetDiagram()->RemoveShape(shape);
        if (shape->IsKindOf(CLASSINFO(wxLineShape)))
        {
          wxLineShape *lineShape = (wxLineShape *)shape;
          fromShape = lineShape->GetFrom();
          toShape = lineShape->GetTo();
        }
        shape->Unlink();
        
        doc->Modify(TRUE);
        doc->UpdateAllViews();
      }

      break;
    }
    case OGLEDIT_ADD_SHAPE:
    {
      wxShape *theShape = NULL;
      if (shape)
        theShape = shape; // Saved from undoing the shape
      else
      {
        theShape = (wxShape *)shapeInfo->CreateObject();
        theShape->AssignNewIds();
        theShape->SetEventHandler(new MyEvtHandler(theShape, theShape, wxString("")));
        theShape->SetCentreResize(FALSE);
        theShape->SetPen(wxBLACK_PEN);
        theShape->SetBrush(wxCYAN_BRUSH);
      
        theShape->SetSize(60, 60);
      }
      doc->GetDiagram()->AddShape(theShape);
      theShape->Show(TRUE);

      wxClientDC dc(theShape->GetCanvas());
      theShape->GetCanvas()->PrepareDC(dc);

      theShape->Move(dc, x, y);
      
      shape = theShape;
      deleteShape = FALSE;

      doc->Modify(TRUE);
      doc->UpdateAllViews();
      break;
    }
    case OGLEDIT_ADD_LINE:
    {
      wxShape *theShape = NULL;
      if (shape)
        theShape = shape; // Saved from undoing the line
      else
      {
        theShape = (wxShape *)shapeInfo->CreateObject();
        theShape->AssignNewIds();
        theShape->SetEventHandler(new MyEvtHandler(theShape, theShape, wxString("")));
        theShape->SetPen(wxBLACK_PEN);
        theShape->SetBrush(wxRED_BRUSH);

        wxLineShape *lineShape = (wxLineShape *)theShape;

        // Yes, you can have more than 2 control points, in which case
        // it becomes a multi-segment line.
        lineShape->MakeLineControlPoints(2);
        lineShape->AddArrow(ARROW_ARROW, ARROW_POSITION_END, 10.0, 0.0, "Normal arrowhead");
      }
      
      doc->GetDiagram()->AddShape(theShape);
      
      fromShape->AddLine((wxLineShape *)theShape, toShape);
      
      theShape->Show(TRUE);

      wxClientDC dc(theShape->GetCanvas());
      theShape->GetCanvas()->PrepareDC(dc);

      // It won't get drawn properly unless you move both
      // connected images
      fromShape->Move(dc, fromShape->GetX(), fromShape->GetY());
      toShape->Move(dc, toShape->GetX(), toShape->GetY());
      
      shape = theShape;
      deleteShape = FALSE;

      doc->Modify(TRUE);
      doc->UpdateAllViews();
      break;
    }
    case OGLEDIT_CHANGE_BACKGROUND_COLOUR:
    {
      if (shape)
      {
        wxClientDC dc(shape->GetCanvas());
        shape->GetCanvas()->PrepareDC(dc);

        wxBrush *oldBrush = shape->GetBrush();
        shape->SetBrush(shapeBrush);
        shapeBrush = oldBrush;
        shape->Draw(dc);
        
        doc->Modify(TRUE);
        doc->UpdateAllViews();
      }

      break;
    }
    case OGLEDIT_EDIT_LABEL:
    {
      if (shape)
      {
        MyEvtHandler *myHandler = (MyEvtHandler *)shape->GetEventHandler();
        wxString oldLabel(myHandler->label);
        myHandler->label = shapeLabel;
        shapeLabel = oldLabel;

        wxClientDC dc(shape->GetCanvas());
        shape->GetCanvas()->PrepareDC(dc);

        shape->FormatText(dc, (char*) (const char*) myHandler->label);
        shape->Draw(dc);
        
        doc->Modify(TRUE);
        doc->UpdateAllViews();
      }

      break;
    }
  }
  return TRUE;
}

bool DiagramCommand::Undo(void)
{
  switch (cmd)
  {
    case OGLEDIT_CUT:
    {
      if (shape)
      {
        doc->GetDiagram()->AddShape(shape);
        shape->Show(TRUE);

        if (shape->IsKindOf(CLASSINFO(wxLineShape)))
        {
          wxLineShape *lineShape = (wxLineShape *)shape;

          fromShape->AddLine(lineShape, toShape);
        }
        if (selected)
          shape->Select(TRUE);

        deleteShape = FALSE;
      }
      doc->Modify(TRUE);
      doc->UpdateAllViews();
      break;
    }
    case OGLEDIT_ADD_SHAPE:
    case OGLEDIT_ADD_LINE:
    {
      if (shape)
      {
        wxClientDC dc(shape->GetCanvas());
        shape->GetCanvas()->PrepareDC(dc);

        shape->Select(FALSE, &dc);
        doc->GetDiagram()->RemoveShape(shape);
        shape->Unlink();
        deleteShape = TRUE;
      }
      doc->Modify(TRUE);
      doc->UpdateAllViews();
      break;
    }
    case OGLEDIT_CHANGE_BACKGROUND_COLOUR:
    {
      if (shape)
      {
        wxClientDC dc(shape->GetCanvas());
        shape->GetCanvas()->PrepareDC(dc);

        wxBrush *oldBrush = shape->GetBrush();
        shape->SetBrush(shapeBrush);
        shapeBrush = oldBrush;
        shape->Draw(dc);
        
        doc->Modify(TRUE);
        doc->UpdateAllViews();
      }
      break;
    }
    case OGLEDIT_EDIT_LABEL:
    {
      if (shape)
      {
        MyEvtHandler *myHandler = (MyEvtHandler *)shape->GetEventHandler();
        wxString oldLabel(myHandler->label);
        myHandler->label = shapeLabel;
        shapeLabel = oldLabel;

        wxClientDC dc(shape->GetCanvas());
        shape->GetCanvas()->PrepareDC(dc);

        shape->FormatText(dc, (char*) (const char*) myHandler->label);
        shape->Draw(dc);
        
        doc->Modify(TRUE);
        doc->UpdateAllViews();
      }

      break;
    }
  }
  return TRUE;
}

// Remove each individual line connected to a shape by sending a command.
void DiagramCommand::RemoveLines(wxShape *shape)
{
  wxNode *node = shape->GetLines().First();
  while (node)
  {
    wxLineShape *line = (wxLineShape *)node->Data();
    doc->GetCommandProcessor()->Submit(new DiagramCommand("Cut", OGLEDIT_CUT, doc, NULL, 0.0, 0.0, line->Selected(), line));
    
    node = shape->GetLines().First();
  }
}

/*
 * MyEvtHandler: an event handler class for all shapes
 */
 
void MyEvtHandler::OnLeftClick(double x, double y, int keys, int attachment)
{
  wxClientDC dc(GetShape()->GetCanvas());
  GetShape()->GetCanvas()->PrepareDC(dc);

  if (keys == 0)
  {
    // Selection is a concept the library knows about
    if (GetShape()->Selected())
    {
      GetShape()->Select(FALSE, &dc);
      GetShape()->GetCanvas()->Redraw(dc); // Redraw because bits of objects will be are missing
    }
    else
    {
      // Ensure no other shape is selected, to simplify Undo/Redo code
      bool redraw = FALSE;
      wxNode *node = GetShape()->GetCanvas()->GetDiagram()->GetShapeList()->First();
      while (node)
      {
        wxShape *eachShape = (wxShape *)node->Data();
        if (eachShape->GetParent() == NULL)
        {
          if (eachShape->Selected())
          {
            eachShape->Select(FALSE, &dc);
            redraw = TRUE;
          }
        }
        node = node->Next();
      }
      GetShape()->Select(TRUE, &dc);
      if (redraw)
        GetShape()->GetCanvas()->Redraw(dc);
    }
  }
  else if (keys & KEY_CTRL)
  {
    // Do something for CONTROL
  }
  else
  {
    wxGetApp().frame->SetStatusText(label);
  }
}

/*
 * Implement connection of two shapes by right-dragging between them.
 */

void MyEvtHandler::OnBeginDragRight(double x, double y, int keys, int attachment)
{
  // Force attachment to be zero for now. Eventually we can deal with
  // the actual attachment point, e.g. a rectangle side if attachment mode is on.
  attachment = 0;
  
  wxClientDC dc(GetShape()->GetCanvas());
  GetShape()->GetCanvas()->PrepareDC(dc);

  wxPen dottedPen(wxColour(0, 0, 0), 1, wxDOT);
  dc.SetLogicalFunction(wxXOR);
  dc.SetPen(dottedPen);
  double xp, yp;
  GetShape()->GetAttachmentPosition(attachment, &xp, &yp);
  dc.DrawLine(xp, yp, x, y);
  GetShape()->GetCanvas()->CaptureMouse();
}

void MyEvtHandler::OnDragRight(bool draw, double x, double y, int keys, int attachment)
{
  // Force attachment to be zero for now
  attachment = 0;

  wxClientDC dc(GetShape()->GetCanvas());
  GetShape()->GetCanvas()->PrepareDC(dc);

  wxPen dottedPen(wxColour(0, 0, 0), 1, wxDOT);
  dc.SetLogicalFunction(wxXOR);
  dc.SetPen(dottedPen);
  double xp, yp;
  GetShape()->GetAttachmentPosition(attachment, &xp, &yp);
  dc.DrawLine(xp, yp, x, y);
}

void MyEvtHandler::OnEndDragRight(double x, double y, int keys, int attachment)
{
  GetShape()->GetCanvas()->ReleaseMouse();
  MyCanvas *canvas = (MyCanvas *)GetShape()->GetCanvas();

  // Check if we're on an object
  int new_attachment;
  wxShape *otherShape = canvas->FindFirstSensitiveShape(x, y, &new_attachment, OP_DRAG_RIGHT);
  
  if (otherShape && !otherShape->IsKindOf(CLASSINFO(wxLineShape)))
  {
    canvas->view->GetDocument()->GetCommandProcessor()->Submit(
      new DiagramCommand("wxLineShape", OGLEDIT_ADD_LINE, (DiagramDocument *)canvas->view->GetDocument(), CLASSINFO(wxLineShape),
      0.0, 0.0, FALSE, NULL, GetShape(), otherShape));
  }
}

void MyEvtHandler::OnEndSize(double x, double y)
{
  wxClientDC dc(GetShape()->GetCanvas());
  GetShape()->GetCanvas()->PrepareDC(dc);

  GetShape()->FormatText(dc, (char*) (const char*) label);
}

/*
 * Diagram
 */
 
bool MyDiagram::OnShapeSave(wxExprDatabase& db, wxShape& shape, wxExpr& expr)
{
  wxDiagram::OnShapeSave(db, shape, expr);
  MyEvtHandler *handler = (MyEvtHandler *)shape.GetEventHandler();
  expr.AddAttributeValueString("label", handler->label);
  return TRUE;
}

bool MyDiagram::OnShapeLoad(wxExprDatabase& db, wxShape& shape, wxExpr& expr)
{
  wxDiagram::OnShapeLoad(db, shape, expr);
  char *label = NULL;
  expr.AssignAttributeValue("label", &label);
  MyEvtHandler *handler = new MyEvtHandler(&shape, &shape, wxString(label));
  shape.SetEventHandler(handler);
  
  if (label)
    delete[] label;
  return TRUE;
}

/*
 * New shapes
 */

IMPLEMENT_DYNAMIC_CLASS(wxRoundedRectangleShape, wxRectangleShape)

wxRoundedRectangleShape::wxRoundedRectangleShape(double w, double h):
 wxRectangleShape(w, h)
{
  // 0.3 of the smaller rectangle dimension
  SetCornerRadius((double) -0.3);
}

IMPLEMENT_DYNAMIC_CLASS(wxDiamondShape, wxPolygonShape)

wxDiamondShape::wxDiamondShape(double w, double h):
  wxPolygonShape()
{
  // wxPolygonShape::SetSize relies on the shape having non-zero
  // size initially.
  if (w == 0.0)
    w = 60.0;
  if (h == 0.0)
    h = 60.0;
    
  wxList *thePoints = new wxList;
  wxRealPoint *point = new wxRealPoint(0.0, (-h/2.0));
  thePoints->Append((wxObject*) point);

  point = new wxRealPoint((w/2.0), 0.0);
  thePoints->Append((wxObject*) point);

  point = new wxRealPoint(0.0, (h/2.0));
  thePoints->Append((wxObject*) point);

  point = new wxRealPoint((-w/2.0), 0.0);
  thePoints->Append((wxObject*) point);

  Create(thePoints);
}
Commit	Line	Data
f449ef69 JS	1	/////////////////////////////////////////////////////////////////////////////
	2	// Name: doc.cpp
	3	// Purpose: Implements document functionality in OGLEdit
	4	// Author: Julian Smart
	5	// Modified by:
	6	// Created: 12/07/98
	7	// RCS-ID: $Id$
	8	// Copyright: (c) Julian Smart
	9	// Licence: wxWindows licence
	10	/////////////////////////////////////////////////////////////////////////////
	11
	12	#ifdef __GNUG__
	13	// #pragma implementation
	14	#endif
	15
	16	// For compilers that support precompilation, includes "wx.h".
	17	#include <wx/wxprec.h>
	18
	19	#ifdef __BORLANDC__
	20	#pragma hdrstop
	21	#endif
	22
	23	#ifndef WX_PRECOMP
	24	#include <wx/wx.h>
	25	#endif
	26
	27	#if !USE_DOC_VIEW_ARCHITECTURE
	28	#error You must set USE_DOC_VIEW_ARCHITECTURE to 1 in wx_setup.h!
	29	#endif
	30
	31	#include "ogledit.h"
	32	#include "doc.h"
	33	#include <wx/wxexpr.h>
	34	#include "view.h"
	35
	36	IMPLEMENT_DYNAMIC_CLASS(DiagramDocument, wxDocument)
	37
	38	DiagramDocument::DiagramDocument(void)
	39	{
	40	}
	41
	42	DiagramDocument::~DiagramDocument(void)
	43	{
	44	}
	45
	46	bool DiagramDocument::OnCloseDocument(void)
	47	{
	48	diagram.DeleteAllShapes();
	49	return TRUE;
	50	}
	51
	52	ostream& DiagramDocument::SaveObject(ostream& stream)
	53	{
	54	wxDocument::SaveObject(stream);
	55
	56	char buf[400];
	57	(void) wxGetTempFileName("diag", buf);
	58
	59	diagram.SaveFile(buf);
	60	wxTransferFileToStream(buf, stream);
	61
	62	wxRemoveFile(buf);
	63
	64	return stream;
65	}
66
67	istream& DiagramDocument::LoadObject(istream& stream)
68	{
69	wxDocument::LoadObject(stream);
70
71	char buf[400];
72	(void) wxGetTempFileName("diag", buf);
73
74	wxTransferStreamToFile(stream, buf);
75
76	diagram.DeleteAllShapes();
77	diagram.LoadFile(buf);
78	wxRemoveFile(buf);
79
80	return stream;
81	}
82
83	/*
84	* Implementation of drawing command
85	*/
86
42cfaf8c	87	DiagramCommand::DiagramCommand(char name, int command, DiagramDocument ddoc, wxClassInfo *info, double xx, double yy,
f449ef69 JS	88	bool sel, wxShape theShape, wxShape fs, wxShape *ts):
	89	wxCommand(TRUE, name)
	90	{
	91	doc = ddoc;
	92	cmd = command;
	93	shape = theShape;
	94	fromShape = fs;
	95	toShape = ts;
	96	shapeInfo = info;
	97	shapeBrush = NULL;
	98	shapePen = NULL;
	99	x = xx;
	100	y = yy;
	101	selected = sel;
	102	deleteShape = FALSE;
	103	}
	104
	105	DiagramCommand::DiagramCommand(char name, int command, DiagramDocument ddoc, wxBrush backgroundColour, wxShape theShape):
	106	wxCommand(TRUE, name)
	107	{
	108	doc = ddoc;
	109	cmd = command;
	110	shape = theShape;
	111	fromShape = NULL;
	112	toShape = NULL;
	113	shapeInfo = NULL;
	114	x = 0.0;
	115	y = 0.0;
	116	selected = FALSE;
	117	deleteShape = FALSE;
	118	shapeBrush = backgroundColour;
	119	shapePen = NULL;
	120	}
	121
	122	DiagramCommand::DiagramCommand(char name, int command, DiagramDocument ddoc, const wxString& lab, wxShape *theShape):
	123	wxCommand(TRUE, name)
	124	{
	125	doc = ddoc;
	126	cmd = command;
	127	shape = theShape;
	128	fromShape = NULL;
	129	toShape = NULL;
	130	shapeInfo = NULL;
	131	x = 0.0;
	132	y = 0.0;
	133	selected = FALSE;
	134	deleteShape = FALSE;
	135	shapeBrush = NULL;
	136	shapePen = NULL;
	137	shapeLabel = lab;
	138	}
	139
	140	DiagramCommand::~DiagramCommand(void)
	141	{
	142	if (shape && deleteShape)
	143	{
	144	shape->SetCanvas(NULL);
	145	delete shape;
	146	}
	147	}
	148
	149	bool DiagramCommand::Do(void)
	150	{
	151	switch (cmd)
152	{
153	case OGLEDIT_CUT:
154	{
155	if (shape)
156	{
157	deleteShape = TRUE;
158
159	shape->Select(FALSE);
160
161	// Generate commands to explicitly remove each connected line.
162	RemoveLines(shape);
163
164	doc->GetDiagram()->RemoveShape(shape);
165	if (shape->IsKindOf(CLASSINFO(wxLineShape)))
166	{
167	wxLineShape lineShape = (wxLineShape )shape;
168	fromShape = lineShape->GetFrom();
169	toShape = lineShape->GetTo();
170	}
171	shape->Unlink();
172
173	doc->Modify(TRUE);
174	doc->UpdateAllViews();
175	}
176
177	break;
178	}
179	case OGLEDIT_ADD_SHAPE:
180	{
181	wxShape *theShape = NULL;
182	if (shape)
183	theShape = shape; // Saved from undoing the shape
184	else
185	{
186	theShape = (wxShape *)shapeInfo->CreateObject();
187	theShape->AssignNewIds();
188	theShape->SetEventHandler(new MyEvtHandler(theShape, theShape, wxString("")));
189	theShape->SetCentreResize(FALSE);
190	theShape->SetPen(wxBLACK_PEN);
191	theShape->SetBrush(wxCYAN_BRUSH);
192
193	theShape->SetSize(60, 60);
194	}
195	doc->GetDiagram()->AddShape(theShape);
196	theShape->Show(TRUE);
197
198	wxClientDC dc(theShape->GetCanvas());
199	theShape->GetCanvas()->PrepareDC(dc);
200
201	theShape->Move(dc, x, y);
202
203	shape = theShape;
204	deleteShape = FALSE;
205
206	doc->Modify(TRUE);
207	doc->UpdateAllViews();
208	break;
209	}
210	case OGLEDIT_ADD_LINE:
211	{
212	wxShape *theShape = NULL;
213	if (shape)
214	theShape = shape; // Saved from undoing the line
215	else
216	{
217	theShape = (wxShape *)shapeInfo->CreateObject();
218	theShape->AssignNewIds();
219	theShape->SetEventHandler(new MyEvtHandler(theShape, theShape, wxString("")));
220	theShape->SetPen(wxBLACK_PEN);
221	theShape->SetBrush(wxRED_BRUSH);
222
223	wxLineShape lineShape = (wxLineShape )theShape;
224
225	// Yes, you can have more than 2 control points, in which case
226	// it becomes a multi-segment line.
227	lineShape->MakeLineControlPoints(2);
228	lineShape->AddArrow(ARROW_ARROW, ARROW_POSITION_END, 10.0, 0.0, "Normal arrowhead");
229	}
230
231	doc->GetDiagram()->AddShape(theShape);
232
233	fromShape->AddLine((wxLineShape *)theShape, toShape);
234
235	theShape->Show(TRUE);
236
237	wxClientDC dc(theShape->GetCanvas());
238	theShape->GetCanvas()->PrepareDC(dc);
239
240	// It won't get drawn properly unless you move both
241	// connected images
242	fromShape->Move(dc, fromShape->GetX(), fromShape->GetY());
243	toShape->Move(dc, toShape->GetX(), toShape->GetY());
244
245	shape = theShape;
246	deleteShape = FALSE;
247
248	doc->Modify(TRUE);
249	doc->UpdateAllViews();
250	break;
251	}
252	case OGLEDIT_CHANGE_BACKGROUND_COLOUR:
253	{
254	if (shape)
255	{
256	wxClientDC dc(shape->GetCanvas());
257	shape->GetCanvas()->PrepareDC(dc);
258
259	wxBrush *oldBrush = shape->GetBrush();
260	shape->SetBrush(shapeBrush);
261	shapeBrush = oldBrush;
262	shape->Draw(dc);
263
264	doc->Modify(TRUE);
265	doc->UpdateAllViews();
266	}
267
268	break;
269	}
270	case OGLEDIT_EDIT_LABEL:
271	{
272	if (shape)
273	{
274	MyEvtHandler myHandler = (MyEvtHandler )shape->GetEventHandler();
275	wxString oldLabel(myHandler->label);
276	myHandler->label = shapeLabel;
277	shapeLabel = oldLabel;
278
279	wxClientDC dc(shape->GetCanvas());
280	shape->GetCanvas()->PrepareDC(dc);
281
282	shape->FormatText(dc, (char) (const char) myHandler->label);
283	shape->Draw(dc);
284
285	doc->Modify(TRUE);
286	doc->UpdateAllViews();
287	}
288
289	break;
290	}
291	}
292	return TRUE;
293	}
294
295	bool DiagramCommand::Undo(void)
296	{
297	switch (cmd)
298	{
299	case OGLEDIT_CUT:
300	{
301	if (shape)
302	{
303	doc->GetDiagram()->AddShape(shape);
304	shape->Show(TRUE);
305
306	if (shape->IsKindOf(CLASSINFO(wxLineShape)))
307	{
308	wxLineShape lineShape = (wxLineShape )shape;
309
310	fromShape->AddLine(lineShape, toShape);
311	}
312	if (selected)
313	shape->Select(TRUE);
314
315	deleteShape = FALSE;
316	}
317	doc->Modify(TRUE);
318	doc->UpdateAllViews();
319	break;
320	}
321	case OGLEDIT_ADD_SHAPE:
322	case OGLEDIT_ADD_LINE:
323	{
324	if (shape)
325	{
326	wxClientDC dc(shape->GetCanvas());
327	shape->GetCanvas()->PrepareDC(dc);
328
329	shape->Select(FALSE, &dc);
330	doc->GetDiagram()->RemoveShape(shape);
331	shape->Unlink();
332	deleteShape = TRUE;
333	}
334	doc->Modify(TRUE);
335	doc->UpdateAllViews();
336	break;
337	}
338	case OGLEDIT_CHANGE_BACKGROUND_COLOUR:
339	{
340	if (shape)
341	{
342	wxClientDC dc(shape->GetCanvas());
343	shape->GetCanvas()->PrepareDC(dc);
344
345	wxBrush *oldBrush = shape->GetBrush();
346	shape->SetBrush(shapeBrush);
347	shapeBrush = oldBrush;
348	shape->Draw(dc);
349
350	doc->Modify(TRUE);
351	doc->UpdateAllViews();
352	}
353	break;
354	}
355	case OGLEDIT_EDIT_LABEL:
356	{
357	if (shape)
358	{
359	MyEvtHandler myHandler = (MyEvtHandler )shape->GetEventHandler();
360	wxString oldLabel(myHandler->label);
361	myHandler->label = shapeLabel;
362	shapeLabel = oldLabel;
363
364	wxClientDC dc(shape->GetCanvas());
365	shape->GetCanvas()->PrepareDC(dc);
366
367	shape->FormatText(dc, (char) (const char) myHandler->label);
368	shape->Draw(dc);
369
370	doc->Modify(TRUE);
371	doc->UpdateAllViews();
372	}
373
374	break;
375	}
376	}
377	return TRUE;
378	}
379
380	// Remove each individual line connected to a shape by sending a command.
381	void DiagramCommand::RemoveLines(wxShape *shape)
382	{
383	wxNode *node = shape->GetLines().First();
384	while (node)
385	{
386	wxLineShape line = (wxLineShape )node->Data();
387	doc->GetCommandProcessor()->Submit(new DiagramCommand("Cut", OGLEDIT_CUT, doc, NULL, 0.0, 0.0, line->Selected(), line));
388
389	node = shape->GetLines().First();
390	}
391	}
392
393	/*
394	* MyEvtHandler: an event handler class for all shapes
395	*/
396
42cfaf8c	397	void MyEvtHandler::OnLeftClick(double x, double y, int keys, int attachment)
f449ef69 JS	398	{
	399	wxClientDC dc(GetShape()->GetCanvas());
	400	GetShape()->GetCanvas()->PrepareDC(dc);
	401
	402	if (keys == 0)
	403	{
	404	// Selection is a concept the library knows about
	405	if (GetShape()->Selected())
	406	{
	407	GetShape()->Select(FALSE, &dc);
	408	GetShape()->GetCanvas()->Redraw(dc); // Redraw because bits of objects will be are missing
	409	}
	410	else
	411	{
	412	// Ensure no other shape is selected, to simplify Undo/Redo code
	413	bool redraw = FALSE;
	414	wxNode *node = GetShape()->GetCanvas()->GetDiagram()->GetShapeList()->First();
	415	while (node)
	416	{
	417	wxShape eachShape = (wxShape )node->Data();
	418	if (eachShape->GetParent() == NULL)
	419	{
	420	if (eachShape->Selected())
	421	{
	422	eachShape->Select(FALSE, &dc);
	423	redraw = TRUE;
	424	}
	425	}
	426	node = node->Next();
	427	}
	428	GetShape()->Select(TRUE, &dc);
	429	if (redraw)
	430	GetShape()->GetCanvas()->Redraw(dc);
	431	}
	432	}
	433	else if (keys & KEY_CTRL)
	434	{
	435	// Do something for CONTROL
	436	}
	437	else
	438	{
	439	wxGetApp().frame->SetStatusText(label);
	440	}
	441	}
	442
	443	/*
	444	* Implement connection of two shapes by right-dragging between them.
	445	*/
	446
42cfaf8c	447	void MyEvtHandler::OnBeginDragRight(double x, double y, int keys, int attachment)
f449ef69 JS	448	{
	449	// Force attachment to be zero for now. Eventually we can deal with
	450	// the actual attachment point, e.g. a rectangle side if attachment mode is on.
	451	attachment = 0;
	452
	453	wxClientDC dc(GetShape()->GetCanvas());
	454	GetShape()->GetCanvas()->PrepareDC(dc);
	455
	456	wxPen dottedPen(wxColour(0, 0, 0), 1, wxDOT);
	457	dc.SetLogicalFunction(wxXOR);
	458	dc.SetPen(dottedPen);
42cfaf8c	459	double xp, yp;
f449ef69 JS	460	GetShape()->GetAttachmentPosition(attachment, &xp, &yp);
	461	dc.DrawLine(xp, yp, x, y);
	462	GetShape()->GetCanvas()->CaptureMouse();
	463	}
	464
42cfaf8c	465	void MyEvtHandler::OnDragRight(bool draw, double x, double y, int keys, int attachment)
f449ef69 JS	466	{
	467	// Force attachment to be zero for now
	468	attachment = 0;
	469
	470	wxClientDC dc(GetShape()->GetCanvas());
	471	GetShape()->GetCanvas()->PrepareDC(dc);
	472
	473	wxPen dottedPen(wxColour(0, 0, 0), 1, wxDOT);
	474	dc.SetLogicalFunction(wxXOR);
	475	dc.SetPen(dottedPen);
42cfaf8c	476	double xp, yp;
f449ef69 JS	477	GetShape()->GetAttachmentPosition(attachment, &xp, &yp);
	478	dc.DrawLine(xp, yp, x, y);
	479	}
	480
42cfaf8c	481	void MyEvtHandler::OnEndDragRight(double x, double y, int keys, int attachment)
f449ef69 JS	482	{
	483	GetShape()->GetCanvas()->ReleaseMouse();
	484	MyCanvas canvas = (MyCanvas )GetShape()->GetCanvas();
	485
	486	// Check if we're on an object
	487	int new_attachment;
	488	wxShape *otherShape = canvas->FindFirstSensitiveShape(x, y, &new_attachment, OP_DRAG_RIGHT);
	489
	490	if (otherShape && !otherShape->IsKindOf(CLASSINFO(wxLineShape)))
	491	{
	492	canvas->view->GetDocument()->GetCommandProcessor()->Submit(
	493	new DiagramCommand("wxLineShape", OGLEDIT_ADD_LINE, (DiagramDocument *)canvas->view->GetDocument(), CLASSINFO(wxLineShape),
	494	0.0, 0.0, FALSE, NULL, GetShape(), otherShape));
	495	}
	496	}
	497
42cfaf8c	498	void MyEvtHandler::OnEndSize(double x, double y)
f449ef69 JS	499	{
	500	wxClientDC dc(GetShape()->GetCanvas());
	501	GetShape()->GetCanvas()->PrepareDC(dc);
	502
	503	GetShape()->FormatText(dc, (char) (const char) label);
	504	}
	505
	506	/*
	507	* Diagram
	508	*/
	509
	510	bool MyDiagram::OnShapeSave(wxExprDatabase& db, wxShape& shape, wxExpr& expr)
	511	{
	512	wxDiagram::OnShapeSave(db, shape, expr);
	513	MyEvtHandler handler = (MyEvtHandler )shape.GetEventHandler();
	514	expr.AddAttributeValueString("label", handler->label);
	515	return TRUE;
	516	}
	517
	518	bool MyDiagram::OnShapeLoad(wxExprDatabase& db, wxShape& shape, wxExpr& expr)
	519	{
	520	wxDiagram::OnShapeLoad(db, shape, expr);
	521	char *label = NULL;
	522	expr.AssignAttributeValue("label", &label);
	523	MyEvtHandler *handler = new MyEvtHandler(&shape, &shape, wxString(label));
	524	shape.SetEventHandler(handler);
	525
	526	if (label)
	527	delete[] label;
	528	return TRUE;
	529	}
	530
	531	/*
	532	* New shapes
	533	*/
	534
	535	IMPLEMENT_DYNAMIC_CLASS(wxRoundedRectangleShape, wxRectangleShape)
	536
42cfaf8c	537	wxRoundedRectangleShape::wxRoundedRectangleShape(double w, double h):
f449ef69 JS	538	wxRectangleShape(w, h)
	539	{
	540	// 0.3 of the smaller rectangle dimension
42cfaf8c	541	SetCornerRadius((double) -0.3);
f449ef69 JS	542	}
	543
	544	IMPLEMENT_DYNAMIC_CLASS(wxDiamondShape, wxPolygonShape)
	545
42cfaf8c	546	wxDiamondShape::wxDiamondShape(double w, double h):
f449ef69 JS	547	wxPolygonShape()
	548	{
	549	// wxPolygonShape::SetSize relies on the shape having non-zero
	550	// size initially.
	551	if (w == 0.0)
	552	w = 60.0;
	553	if (h == 0.0)
	554	h = 60.0;
	555
	556	wxList *thePoints = new wxList;
	557	wxRealPoint *point = new wxRealPoint(0.0, (-h/2.0));
	558	thePoints->Append((wxObject*) point);
	559
	560	point = new wxRealPoint((w/2.0), 0.0);
	561	thePoints->Append((wxObject*) point);
	562
	563	point = new wxRealPoint(0.0, (h/2.0));
	564	thePoints->Append((wxObject*) point);
	565
	566	point = new wxRealPoint((-w/2.0), 0.0);
	567	thePoints->Append((wxObject*) point);
	568
	569	Create(thePoints);
	570	}