samples/docview/doc.cpp

   1 /////////////////////////////////////////////////////////////////////////////
   2 // Name:        doc.cpp
   3 // Purpose:     Implements document functionality
   4 // Author:      Julian Smart
   5 // Modified by:
   6 // Created:     04/01/98
   7 // RCS-ID:      $Id$
   8 // Copyright:   (c) Julian Smart and Markus Holzem
   9 // Licence:     wxWindows license
  10 /////////////////////////////////////////////////////////////////////////////
  11
  12 #ifdef __GNUG__
  13 // #pragma implementation
  14 #endif
  15
  16 // For compilers that support precompilation, includes "wx/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 #include "wx/txtstrm.h"
  27
  28 #if !wxUSE_DOC_VIEW_ARCHITECTURE
  29 #error You must set wxUSE_DOC_VIEW_ARCHITECTURE to 1 in setup.h!
  30 #endif
  31
  32 #include "doc.h"
  33 #include "view.h"
  34
  35 IMPLEMENT_DYNAMIC_CLASS(DrawingDocument, wxDocument)
  36
  37 DrawingDocument::DrawingDocument(void)
  38 {
  39 }
  40
  41 DrawingDocument::~DrawingDocument(void)
  42 {
  43     doodleSegments.DeleteContents(TRUE);
  44 }
  45
  46 #if wxUSE_STD_IOSTREAM
  47 ostream& DrawingDocument::SaveObject(ostream& stream)
  48 {
  49     wxDocument::SaveObject(stream);
  50
  51     wxInt32 n = doodleSegments.Number();
  52     stream << n << '\n';
  53
  54     wxNode *node = doodleSegments.First();
  55     while (node)
  56     {
  57         DoodleSegment *segment = (DoodleSegment *)node->Data();
  58         segment->SaveObject(stream);
  59         stream << '\n';
  60
  61         node = node->Next();
  62     }
  63
  64     return stream;
  65 }
  66 #else
  67 wxOutputStream& DrawingDocument::SaveObject(wxOutputStream& stream)
  68 {
  69     wxDocument::SaveObject(stream);
  70
  71     wxTextOutputStream text_stream( stream );
  72
  73     wxInt32 n = doodleSegments.Number();
  74     text_stream << n << '\n';
  75
  76     wxNode *node = doodleSegments.First();
  77     while (node)
  78     {
  79         DoodleSegment *segment = (DoodleSegment *)node->Data();
  80         segment->SaveObject(stream);
  81         text_stream << '\n';
  82
  83         node = node->Next();
  84     }
  85
  86     return stream;
  87 }
  88 #endif
  89
  90 #if wxUSE_STD_IOSTREAM
  91 istream& DrawingDocument::LoadObject(istream& stream)
  92 {
  93     wxDocument::LoadObject(stream);
  94
  95     wxInt32 n = 0;
  96     stream >> n;
  97
  98     for (int i = 0; i < n; i++)
  99     {
 100         DoodleSegment *segment = new DoodleSegment;
 101         segment->LoadObject(stream);
 102         doodleSegments.Append(segment);
 103     }
 104
 105     return stream;
 106 }
 107 #else
 108 wxInputStream& DrawingDocument::LoadObject(wxInputStream& stream)
 109 {
 110     wxDocument::LoadObject(stream);
 111
 112     wxTextInputStream text_stream( stream );
 113
 114     wxInt32 n = 0;
 115     text_stream >> n;
 116
 117     for (int i = 0; i < n; i++)
 118     {
 119         DoodleSegment *segment = new DoodleSegment;
 120         segment->LoadObject(stream);
 121         doodleSegments.Append(segment);
 122     }
 123
 124     return stream;
 125 }
 126 #endif
 127
 128 DoodleSegment::DoodleSegment(void)
 129 {
 130 }
 131
 132 DoodleSegment::DoodleSegment(DoodleSegment& seg)
 133 {
 134     wxNode *node = seg.lines.First();
 135     while (node)
 136     {
 137         DoodleLine *line = (DoodleLine *)node->Data();
 138         DoodleLine *newLine = new DoodleLine;
 139         newLine->x1 = line->x1;
 140         newLine->y1 = line->y1;
 141         newLine->x2 = line->x2;
 142         newLine->y2 = line->y2;
 143
 144         lines.Append(newLine);
 145
 146         node = node->Next();
 147     }
 148 }
 149
 150 DoodleSegment::~DoodleSegment(void)
 151 {
 152     lines.DeleteContents(TRUE);
 153 }
 154
 155 #if wxUSE_STD_IOSTREAM
 156 ostream& DoodleSegment::SaveObject(ostream& stream)
 157 {
 158     wxInt32 n = lines.Number();
 159     stream << n << '\n';
 160
 161     wxNode *node = lines.First();
 162     while (node)
 163     {
 164         DoodleLine *line = (DoodleLine *)node->Data();
 165         stream << line->x1 << " " <<
 166             line->y1 << " " <<
 167             line->x2 << " " <<
 168             line->y2 << "\n";
 169         node = node->Next();
 170     }
 171
 172     return stream;
 173 }
 174 #else
 175 wxOutputStream &DoodleSegment::SaveObject(wxOutputStream& stream)
 176 {
 177     wxTextOutputStream text_stream( stream );
 178
 179     wxInt32 n = lines.Number();
 180     text_stream << n << '\n';
 181
 182     wxNode *node = lines.First();
 183     while (node)
 184     {
 185         DoodleLine *line = (DoodleLine *)node->Data();
 186         text_stream << line->x1 << " " <<
 187             line->y1 << " " <<
 188             line->x2 << " " <<
 189             line->y2 << "\n";
 190         node = node->Next();
 191     }
 192
 193     return stream;
 194 }
 195 #endif
 196
 197 #if wxUSE_STD_IOSTREAM
 198 istream& DoodleSegment::LoadObject(istream& stream)
 199 {
 200     wxInt32 n = 0;
 201     stream >> n;
 202
 203     for (int i = 0; i < n; i++)
 204     {
 205         DoodleLine *line = new DoodleLine;
 206         stream >> line->x1 >>
 207             line->y1 >>
 208             line->x2 >>
 209             line->y2;
 210         lines.Append(line);
 211     }
 212
 213     return stream;
 214 }
 215 #else
 216 wxInputStream &DoodleSegment::LoadObject(wxInputStream& stream)
 217 {
 218     wxTextInputStream text_stream( stream );
 219
 220     wxInt32 n = 0;
 221     text_stream >> n;
 222
 223     for (int i = 0; i < n; i++)
 224     {
 225         DoodleLine *line = new DoodleLine;
 226         text_stream >> line->x1 >>
 227             line->y1 >>
 228             line->x2 >>
 229             line->y2;
 230         lines.Append(line);
 231     }
 232
 233     return stream;
 234 }
 235 #endif
 236
 237 void DoodleSegment::Draw(wxDC *dc)
 238 {
 239     wxNode *node = lines.First();
 240     while (node)
 241     {
 242         DoodleLine *line = (DoodleLine *)node->Data();
 243         dc->DrawLine(line->x1, line->y1, line->x2, line->y2);
 244         node = node->Next();
 245     }
 246 }
 247
 248 /*
 249 * Implementation of drawing command
 250 */
 251
 252 DrawingCommand::DrawingCommand(const wxString& name, int command, DrawingDocument *ddoc, DoodleSegment *seg):
 253 wxCommand(TRUE, name)
 254 {
 255     doc = ddoc;
 256     segment = seg;
 257     cmd = command;
 258 }
 259
 260 DrawingCommand::~DrawingCommand(void)
 261 {
 262     if (segment)
 263         delete segment;
 264 }
 265
 266 bool DrawingCommand::Do(void)
 267 {
 268     switch (cmd)
 269     {
 270     case DOODLE_CUT:
 271         {
 272             // Cut the last segment
 273             if (doc->GetDoodleSegments().Number() > 0)
 274             {
 275                 wxNode *node = doc->GetDoodleSegments().Last();
 276                 if (segment)
 277                     delete segment;
 278
 279                 segment = (DoodleSegment *)node->Data();
 280                 delete node;
 281
 282                 doc->Modify(TRUE);
 283                 doc->UpdateAllViews();
 284             }
 285             break;
 286         }
 287     case DOODLE_ADD:
 288         {
 289             doc->GetDoodleSegments().Append(new DoodleSegment(*segment));
 290             doc->Modify(TRUE);
 291             doc->UpdateAllViews();
 292             break;
 293         }
 294     }
 295     return TRUE;
 296 }
 297
 298 bool DrawingCommand::Undo(void)
 299 {
 300     switch (cmd)
 301     {
 302     case DOODLE_CUT:
 303         {
 304             // Paste the segment
 305             if (segment)
 306             {
 307                 doc->GetDoodleSegments().Append(segment);
 308                 doc->Modify(TRUE);
 309                 doc->UpdateAllViews();
 310                 segment = (DoodleSegment *) NULL;
 311             }
 312             doc->Modify(TRUE);
 313             doc->UpdateAllViews();
 314             break;
 315         }
 316     case DOODLE_ADD:
 317         {
 318             // Cut the last segment
 319             if (doc->GetDoodleSegments().Number() > 0)
 320             {
 321                 wxNode *node = doc->GetDoodleSegments().Last();
 322                 DoodleSegment *seg = (DoodleSegment *)node->Data();
 323                 delete seg;
 324                 delete node;
 325
 326                 doc->Modify(TRUE);
 327                 doc->UpdateAllViews();
 328             }
 329         }
 330     }
 331     return TRUE;
 332 }
 333
 334 IMPLEMENT_DYNAMIC_CLASS(TextEditDocument, wxDocument)
 335
 336 // Since text windows have their own method for saving to/loading from files,
 337 // we override OnSave/OpenDocument instead of Save/LoadObject
 338 bool TextEditDocument::OnSaveDocument(const wxString& filename)
 339 {
 340     TextEditView *view = (TextEditView *)GetFirstView();
 341
 342     if (!view->textsw->SaveFile(filename))
 343         return FALSE;
 344     Modify(FALSE);
 345     return TRUE;
 346 }
 347
 348 bool TextEditDocument::OnOpenDocument(const wxString& filename)
 349 {
 350     TextEditView *view = (TextEditView *)GetFirstView();
 351     if (!view->textsw->LoadFile(filename))
 352         return FALSE;
 353
 354     SetFilename(filename, TRUE);
 355     Modify(FALSE);
 356     UpdateAllViews();
 357     return TRUE;
 358 }
 359
 360 bool TextEditDocument::IsModified(void) const
 361 {
 362     TextEditView *view = (TextEditView *)GetFirstView();
 363     if (view)
 364     {
 365         return (wxDocument::IsModified() || view->textsw->IsModified());
 366     }
 367     else
 368         return wxDocument::IsModified();
 369 }
 370
 371 void TextEditDocument::Modify(bool mod)
 372 {
 373     TextEditView *view = (TextEditView *)GetFirstView();
 374
 375     wxDocument::Modify(mod);
 376
 377     if (!mod && view && view->textsw)
 378         view->textsw->DiscardEdits();
 379 }