Added support for EM_REQUESTRESIZE message, EN_REQUESTRESIZE
[wine] / dlls / riched20 / caret.c
1 /*
2  * RichEdit - Caret and selection functions.
3  *
4  * Copyright 2004 by Krzysztof Foltman
5  *
6  * This library is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Lesser General Public
8  * License as published by the Free Software Foundation; either
9  * version 2.1 of the License, or (at your option) any later version.
10  *
11  * This library is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14  * Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with this library; if not, write to the Free Software
18  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
19  */
20
21
22 #include "editor.h"
23
24 WINE_DEFAULT_DEBUG_CHANNEL(richedit);
25
26 void ME_GetSelection(ME_TextEditor *editor, int *from, int *to)
27 {
28   *from = ME_GetCursorOfs(editor, 0);
29   *to =   ME_GetCursorOfs(editor, 1);
30   
31   if (*from > *to)
32   {
33     int tmp = *from;
34     *from = *to;
35     *to = tmp;    
36   }
37 }
38
39 int ME_GetTextLength(ME_TextEditor *editor)
40 {
41   return ME_CharOfsFromRunOfs(editor, ME_FindItemBack(editor->pBuffer->pLast, diRun), 0);   
42 }
43
44
45 int ME_GetTextLengthEx(ME_TextEditor *editor, GETTEXTLENGTHEX *how)
46 {
47   int length;
48   
49   if (how->flags & GTL_PRECISE && how->flags & GTL_CLOSE)
50     return E_INVALIDARG;
51   if (how->flags & GTL_NUMCHARS && how->flags & GTL_NUMBYTES)
52     return E_INVALIDARG;
53   
54   length = ME_GetTextLength(editor);
55   
56   if (how->flags & GTL_USECRLF)
57     length += editor->nParagraphs;
58   
59   if (how->flags & GTL_NUMBYTES)
60   {
61     CPINFO cpinfo;
62     
63     if (how->codepage == 1200)
64       return length * 2;
65     if (how->flags & GTL_PRECISE)
66       FIXME("GTL_PRECISE flag unsupported. Using GTL_CLOSE\n");
67     if (GetCPInfo(how->codepage, &cpinfo))
68       return length * cpinfo.MaxCharSize;
69     ERR("Invalid codepage %u\n", how->codepage);
70     return E_INVALIDARG;
71   }
72   return length; 
73 }
74
75
76 void ME_SetSelection(ME_TextEditor *editor, int from, int to)
77 {
78   if (from == 0 && to == -1)
79   {
80     editor->pCursors[1].pRun = ME_FindItemFwd(editor->pBuffer->pFirst, diRun); 
81     editor->pCursors[1].nOffset = 0; 
82     editor->pCursors[0].pRun = ME_FindItemBack(editor->pBuffer->pLast, diRun); 
83     editor->pCursors[0].nOffset = 0; 
84     ME_Repaint(editor);
85     ME_ClearTempStyle(editor);
86     return;
87   }
88   if (from == -1)
89   {
90     editor->pCursors[1] = editor->pCursors[0]; 
91     ME_Repaint(editor);
92     ME_ClearTempStyle(editor);
93     return;
94   }
95   if (from>to)
96   {
97     int tmp = from;
98     from = to;
99     to = tmp;
100   }
101   ME_RunOfsFromCharOfs(editor, from, &editor->pCursors[1].pRun, &editor->pCursors[1].nOffset);
102   ME_RunOfsFromCharOfs(editor, to, &editor->pCursors[0].pRun, &editor->pCursors[0].nOffset);  
103 }
104
105 void ME_MoveCaret(ME_TextEditor *editor)
106 {
107   HDC hDC = GetDC(editor->hWnd);
108   ME_Context c;
109
110   ME_Cursor *pCursor = &editor->pCursors[0];
111   ME_DisplayItem *pCursorRun = pCursor->pRun;
112   ME_DisplayItem *pSizeRun = pCursor->pRun;
113   
114   ME_InitContext(&c, editor, hDC);
115   assert(!pCursor->nOffset || !editor->bCaretAtEnd);
116   
117   if (pCursorRun->type == diRun) {
118     ME_DisplayItem *row = ME_FindItemBack(pCursorRun, diStartRowOrParagraph);
119     if (row) {
120       ME_DisplayItem *run = pCursorRun;
121       ME_DisplayItem *para;
122       SIZE sz = {0, 0};
123       if (!pCursor->nOffset && !editor->bCaretAtEnd)
124       {
125         ME_DisplayItem *prev = ME_FindItemBack(pCursorRun, diRunOrStartRow);
126         if (prev->type == diRun)
127           pSizeRun = prev;
128       }
129       assert(row->type == diStartRow); /* paragraph -> run without start row ?*/
130       para = ME_FindItemBack(row, diParagraph);
131       if (editor->bCaretAtEnd && !pCursor->nOffset && 
132           run == ME_FindItemFwd(row, diRun))
133       {
134         ME_DisplayItem *tmp = ME_FindItemBack(row, diRunOrParagraph);
135         if (tmp->type == diRun)
136         {
137           row = ME_FindItemBack(tmp, diStartRow);
138           pSizeRun = run = tmp;
139           sz = ME_GetRunSize(&c, &para->member.para, &run->member.run, ME_StrLen(run->member.run.strText));
140         }
141       }
142       if (pCursor->nOffset && !(run->member.run.nFlags & MERF_SKIPPED)) {
143         sz = ME_GetRunSize(&c, &para->member.para, &run->member.run, pCursor->nOffset);
144       }
145       CreateCaret(editor->hWnd, NULL, 0, pSizeRun->member.run.nAscent+pSizeRun->member.run.nDescent);
146       SetCaretPos(run->member.run.pt.x+sz.cx,
147         para->member.para.nYPos+row->member.row.nBaseline+pSizeRun->member.run.pt.y-pSizeRun->member.run.nAscent-ME_GetYScrollPos(editor));
148     } else {
149       assert(0 == "Wrapped paragraph run without a row?");
150       CreateCaret(editor->hWnd, NULL, 0, 10);
151       SetCaretPos(0,0);
152     }
153   }  
154   else {
155     assert(0 == "Cursor not on a run");
156     CreateCaret(editor->hWnd, NULL, 0, 10); /* FIXME use global font */
157     SetCaretPos(0,0);
158   }
159   ME_DestroyContext(&c);
160   ReleaseDC(editor->hWnd, hDC);
161 }
162
163 void ME_ShowCaret(ME_TextEditor *ed)
164 {
165   ME_MoveCaret(ed);
166   ShowCaret(ed->hWnd);
167 }
168
169 void ME_HideCaret(ME_TextEditor *ed)
170 {
171   HideCaret(ed->hWnd);
172   DestroyCaret();
173 }
174
175 void ME_InternalDeleteText(ME_TextEditor *editor, int nOfs, 
176   int nChars)
177 {
178   ME_Cursor c;
179   int shift = 0;
180   
181   while(nChars > 0)
182   {
183     ME_Run *run;
184     ME_CursorFromCharOfs(editor, nOfs, &c);
185     run = &c.pRun->member.run;
186     if (run->nFlags & MERF_ENDPARA) {
187       if (!ME_FindItemFwd(c.pRun, diParagraph))
188       {
189         return;
190       }
191       ME_JoinParagraphs(editor, ME_GetParagraph(c.pRun));
192       /* ME_SkipAndPropagateCharOffset(p->pRun, shift); */
193       ME_CheckCharOffsets(editor);
194       nChars--;
195       if (editor->bEmulateVersion10 && nChars)
196         nChars--;
197       continue;
198     }
199     else
200     {
201       ME_Cursor cursor;
202       int nIntendedChars = nChars;
203       int nCharsToDelete = nChars;
204       int i;
205       int loc = c.nOffset;
206       
207       ME_FindItemBack(c.pRun, diParagraph)->member.para.nFlags |= MEPF_REWRAP;
208       
209       cursor = c;
210       ME_StrRelPos(run->strText, loc, &nChars);
211       /* nChars is the number of characters that should be deleted from the
212          FOLLOWING runs (these AFTER cursor.pRun)
213          nCharsToDelete is a number of chars to delete from THIS run */
214       nCharsToDelete -= nChars;
215       shift -= nCharsToDelete;
216       TRACE("Deleting %d (intended %d-remaning %d) chars at %d in '%s' (%d)\n", 
217         nCharsToDelete, nIntendedChars, nChars, c.nOffset, 
218         debugstr_w(run->strText->szData), run->strText->nLen);
219
220       if (!c.nOffset && ME_StrVLen(run->strText) == nCharsToDelete)
221       {
222         /* undo = reinsert whole run */
223         /* nOfs is a character offset (from the start of the document
224            to the current (deleted) run */
225         ME_UndoItem *pUndo = ME_AddUndoItem(editor, diUndoInsertRun, c.pRun);
226         if (pUndo)
227           pUndo->di.member.run.nCharOfs = nOfs;
228       }
229       else
230       {
231         /* undo = reinsert partial run */
232         ME_UndoItem *pUndo = ME_AddUndoItem(editor, diUndoInsertRun, c.pRun);
233         if (pUndo) {
234           ME_DestroyString(pUndo->di.member.run.strText);
235           pUndo->di.member.run.nCharOfs = nOfs;
236           pUndo->di.member.run.strText = ME_MakeStringN(run->strText->szData+c.nOffset, nCharsToDelete);
237         }
238       }
239       TRACE("Post deletion string: %s (%d)\n", debugstr_w(run->strText->szData), run->strText->nLen);
240       TRACE("Shift value: %d\n", shift);
241       ME_StrDeleteV(run->strText, c.nOffset, nCharsToDelete);
242       
243       /* update cursors (including c) */
244       for (i=-1; i<editor->nCursors; i++) {
245         ME_Cursor *pThisCur = editor->pCursors + i; 
246         if (i == -1) pThisCur = &c;
247         if (pThisCur->pRun == cursor.pRun) {
248           if (pThisCur->nOffset > cursor.nOffset) {
249             if (pThisCur->nOffset-cursor.nOffset < nCharsToDelete)
250               pThisCur->nOffset = cursor.nOffset;
251             else
252               pThisCur->nOffset -= nCharsToDelete;
253             assert(pThisCur->nOffset >= 0);
254             assert(pThisCur->nOffset <= ME_StrVLen(run->strText));
255           }
256           if (pThisCur->nOffset == ME_StrVLen(run->strText))
257           {
258             pThisCur->pRun = ME_FindItemFwd(pThisCur->pRun, diRunOrParagraphOrEnd);
259             assert(pThisCur->pRun->type == diRun);
260             pThisCur->nOffset = 0;
261           }
262         }
263       }
264       
265       /* c = updated data now */
266       
267       if (c.pRun == cursor.pRun)
268         ME_SkipAndPropagateCharOffset(c.pRun, shift);
269       else
270         ME_PropagateCharOffset(c.pRun, shift);
271
272       if (!ME_StrVLen(cursor.pRun->member.run.strText))
273       {
274         TRACE("Removing useless run\n");
275         ME_Remove(cursor.pRun);
276         ME_DestroyDisplayItem(cursor.pRun);
277       }
278       
279       shift = 0;
280       /*
281       ME_CheckCharOffsets(editor);
282       */
283       continue;
284     }
285   }
286 }
287
288 void ME_DeleteTextAtCursor(ME_TextEditor *editor, int nCursor, 
289   int nChars)
290 {  
291   assert(nCursor>=0 && nCursor<editor->nCursors);
292   ME_InternalDeleteText(editor, ME_GetCursorOfs(editor, nCursor), nChars);
293 }
294
295 static WCHAR wszSpace[] = {' ', 0};
296
297 /* FIXME this is temporary, just to have something to test how bad graphics handler is */
298 void ME_InsertGraphicsFromCursor(ME_TextEditor *editor, int nCursor)
299 {
300   ME_Cursor *pCursor = &editor->pCursors[nCursor];
301   ME_DisplayItem *pItem = NULL;
302   ME_DisplayItem *pNewRun = NULL;
303   ME_Style *pStyle = ME_GetInsertStyle(editor, nCursor);
304   ME_UndoItem *pUndo;
305   
306   /* FIXME no no no */
307   if (ME_IsSelection(editor))
308     ME_DeleteSelection(editor);
309
310   pUndo = ME_AddUndoItem(editor, diUndoDeleteRun, NULL);
311   if (pUndo) {
312     pUndo->nStart = pCursor->nOffset + pCursor->pRun->member.run.nCharOfs + ME_GetParagraph(pCursor->pRun)->member.para.nCharOfs;
313     pUndo->nLen = 1;
314   }
315   if (pCursor->nOffset)
316   {
317     ME_SplitRunSimple(editor, pCursor->pRun, pCursor->nOffset);
318   }
319   pItem = pCursor->pRun;
320   pNewRun = ME_MakeRun(pStyle, ME_MakeStringN(wszSpace, 1), MERF_GRAPHICS);
321   pNewRun->member.run.nCharOfs = pCursor->pRun->member.run.nCharOfs;
322   ME_InsertBefore(pCursor->pRun, pNewRun);
323   ME_PropagateCharOffset(pItem, 1);
324   ME_CheckCharOffsets(editor);
325   ME_SendSelChange(editor);
326 }
327
328
329 static void
330 ME_InternalInsertTextFromCursor(ME_TextEditor *editor, int nCursor,
331                                 const WCHAR *str, int len, ME_Style *style,
332                                 int flags)
333 {
334   ME_DisplayItem *pNewRun = NULL;
335   ME_Cursor *p = &editor->pCursors[nCursor];
336
337   editor->bCaretAtEnd = FALSE;
338   
339   assert(p->pRun->type == diRun);
340   
341   ME_AddRefStyle(style);
342   
343   pNewRun = ME_MakeRun(style, ME_MakeStringN(str, len), flags); /* addrefs style */
344   ME_InsertRun(editor, ME_CharOfsFromRunOfs(editor, p->pRun, p->nOffset), pNewRun);
345   ME_DestroyDisplayItem(pNewRun);
346   ME_ReleaseStyle(style);
347 }
348
349
350 void ME_InsertTextFromCursor(ME_TextEditor *editor, int nCursor, 
351   const WCHAR *str, int len, ME_Style *style)
352 {
353   const WCHAR *pos;
354   ME_Cursor *p = NULL;
355
356   assert(style);
357
358   /* FIXME really HERE ? */
359   if (ME_IsSelection(editor))
360     ME_DeleteSelection(editor);
361
362   assert(nCursor>=0 && nCursor<editor->nCursors);
363   if (len == -1)
364     len = lstrlenW(str);
365   while (len)
366   {
367     pos = str;
368     /* FIXME this sucks - no respect for unicode (what else can be a line separator in unicode?) */
369     while(pos-str < len && *pos != '\r' && *pos != '\n' && *pos != '\t')
370       pos++;
371     if (pos-str < len && *pos == '\t') { /* handle tabs */
372       WCHAR tab = '\t';
373
374       if (pos!=str)
375         ME_InternalInsertTextFromCursor(editor, nCursor, str, pos-str, style, 0);
376     
377       ME_InternalInsertTextFromCursor(editor, nCursor, &tab, 1, style, MERF_TAB);
378  
379       pos++;
380       if(pos-str <= len) {
381         len -= pos - str;
382         str = pos;
383         continue;
384       }
385     }
386     if (pos-str < len) {   /* handle EOLs */
387       ME_DisplayItem *tp, *end_run;
388       ME_Paragraph *para;
389       ME_Style *tmp_style;
390       if (pos!=str)
391         ME_InternalInsertTextFromCursor(editor, nCursor, str, pos-str, style, 0);
392       p = &editor->pCursors[nCursor];
393       tp = ME_FindItemBack(p->pRun, diParagraph);
394       para = &tp->member.para;
395       assert(tp);
396       if (p->nOffset) {
397         ME_SplitRunSimple(editor, p->pRun, p->nOffset);
398         p = &editor->pCursors[nCursor];
399       }
400       tmp_style = ME_GetInsertStyle(editor, nCursor);
401       /* ME_SplitParagraph increases style refcount */
402       tp = ME_SplitParagraph(editor, p->pRun, p->pRun->member.run.style);
403       p->pRun = ME_FindItemFwd(tp, diRun);
404       end_run = ME_FindItemBack(tp, diRun);
405       ME_ReleaseStyle(end_run->member.run.style);
406       end_run->member.run.style = tmp_style;
407       p->nOffset = 0;
408       if(pos-str < len && *pos =='\r')
409         pos++;
410       if(pos-str < len && *pos =='\n')
411         pos++;
412       if(pos-str <= len) {
413         len -= pos - str;
414         str = pos;
415         continue;
416       }
417     }
418     ME_InternalInsertTextFromCursor(editor, nCursor, str, len, style, 0);
419     len = 0;
420   }
421 }
422
423 static BOOL ME_ArrowLeft(ME_TextEditor *editor, ME_Cursor *p)
424 {
425   if (p->nOffset) {
426     p->nOffset = ME_StrRelPos2(p->pRun->member.run.strText, p->nOffset, -1);
427     return TRUE;
428   }
429   else
430   {
431     ME_DisplayItem *pRun = ME_FindItemBack(p->pRun, diRunOrParagraph);
432     assert(pRun);
433     if (pRun->type == diRun) {
434       p->pRun = pRun;
435       assert(p->pRun->type == diRun);
436       assert(pRun->member.run.strText->nLen);
437       p->nOffset = pRun->member.run.strText->nLen;
438       if (p->nOffset) {
439         p->nOffset = ME_StrRelPos2(pRun->member.run.strText, p->nOffset, -1);
440         return TRUE;
441       }
442       else
443         assert(0);
444     }
445     if (pRun->type == diParagraph)
446     {
447       if (pRun->member.para.prev_para->type == diTextStart)
448         return FALSE;
449       assert(pRun->member.para.prev_para->type == diParagraph);
450       pRun = ME_FindItemBack(pRun, diRunOrParagraph);
451       /* every paragraph ought to have at least one run */
452       assert(pRun && pRun->type == diRun);
453       assert(pRun->member.run.nFlags & MERF_ENDPARA);
454       p->pRun = pRun;
455       p->nOffset = 0;
456       return TRUE;
457     }
458     assert(0);
459   }
460   return FALSE;
461 }
462
463 static BOOL ME_ArrowRight(ME_TextEditor *editor, ME_Cursor *p)
464 {
465   ME_DisplayItem *pRun;
466   
467   if (!(p->pRun->member.run.nFlags & MERF_ENDPARA))
468   {
469     int new_ofs = ME_StrRelPos2(p->pRun->member.run.strText, p->nOffset, 1);
470     
471     if (new_ofs<p->pRun->member.run.strText->nLen)
472     {
473       p->nOffset = new_ofs;
474       return TRUE;
475     }
476   }
477   pRun = ME_FindItemFwd(p->pRun, diRun);
478   if (pRun) {
479     p->pRun = pRun;
480     assert(p->pRun->type == diRun);
481     p->nOffset = 0;
482   }
483   return TRUE;
484 }
485
486 int ME_GetCursorOfs(ME_TextEditor *editor, int nCursor)
487 {
488   ME_Cursor *pCursor = &editor->pCursors[nCursor];
489   
490   return ME_GetParagraph(pCursor->pRun)->member.para.nCharOfs
491     + pCursor->pRun->member.run.nCharOfs + pCursor->nOffset;
492 }
493
494 int ME_FindPixelPos(ME_TextEditor *editor, int x, int y, ME_Cursor *result, BOOL *is_eol)
495 {
496   ME_DisplayItem *p = editor->pBuffer->pFirst->member.para.next_para;
497   int rx = 0;
498   
499   if (is_eol)
500     *is_eol = 0;
501
502   while(p != editor->pBuffer->pLast)
503   {
504     if (p->type == diParagraph)
505     {
506       int ry = y - p->member.para.nYPos;
507       if (ry < 0)
508       {
509         result->pRun = ME_FindItemFwd(p, diRun);
510         result->nOffset = 0;
511         return 0;
512       }
513       if (ry >= p->member.para.nHeight)
514       {
515         p = p->member.para.next_para;
516         continue;
517       }
518       p = ME_FindItemFwd(p, diStartRow);
519       y = ry;
520       continue;
521     }
522     if (p->type == diStartRow)
523     {
524       int ry = y - p->member.row.nYPos;
525       if (ry < 0)
526         return 0;
527       if (ry >= p->member.row.nHeight)
528       {
529         p = ME_FindItemFwd(p, diStartRowOrParagraphOrEnd);
530         if (p->type != diStartRow)
531           return 0;
532         continue;
533       }
534       p = ME_FindItemFwd(p, diRun);
535       continue;
536     }
537     if (p->type == diRun)
538     {
539       ME_DisplayItem *pp;
540       rx = x - p->member.run.pt.x;
541       if (rx < 0)
542         rx = 0;
543       if (rx >= p->member.run.nWidth) /* not this run yet... find next item */
544       {
545         pp = p;
546         do {
547           p = p->next;
548           if (p->type == diRun)
549           {
550             rx = x - p->member.run.pt.x;
551             goto continue_search;
552           }
553           if (p->type == diStartRow)
554           {
555             p = ME_FindItemFwd(p, diRun);
556             if (is_eol)
557               *is_eol = 1;
558             rx = 0; /* FIXME not sure */
559             goto found_here;
560           }
561           if (p->type == diParagraph || p->type == diTextEnd)
562           {
563             rx = 0; /* FIXME not sure */
564             p = pp;
565             goto found_here;
566           }
567         } while(1);
568         continue;
569       }
570     found_here:
571       if (p->member.run.nFlags & MERF_ENDPARA)
572         rx = 0;
573       result->pRun = p;
574       result->nOffset = ME_CharFromPointCursor(editor, rx, &p->member.run);
575       if (editor->pCursors[0].nOffset == p->member.run.strText->nLen && rx)
576       {
577         result->pRun = ME_FindItemFwd(editor->pCursors[0].pRun, diRun);
578         result->nOffset = 0;
579       }
580       return 1;
581     }
582     assert(0);
583   continue_search:
584     ;
585   }
586   result->pRun = ME_FindItemBack(p, diRun);
587   result->nOffset = 0;
588   assert(result->pRun->member.run.nFlags & MERF_ENDPARA);
589   return 0;
590 }
591
592
593 int
594 ME_CharFromPos(ME_TextEditor *editor, int x, int y)
595 {
596   ME_Cursor cursor;
597   RECT rc;
598
599   GetClientRect(editor->hWnd, &rc);
600   if (x < 0 || y < 0 || x >= rc.right || y >= rc.bottom)
601     return -1;
602   ME_FindPixelPos(editor, x, y, &cursor, NULL);
603   return (ME_GetParagraph(cursor.pRun)->member.para.nCharOfs
604           + cursor.pRun->member.run.nCharOfs + cursor.nOffset);
605 }
606
607
608 void ME_LButtonDown(ME_TextEditor *editor, int x, int y)
609 {
610   ME_Cursor tmp_cursor;
611   int is_selection = 0;
612   
613   editor->nUDArrowX = -1;
614   
615   y += ME_GetYScrollPos(editor);
616
617   tmp_cursor = editor->pCursors[0];
618   is_selection = ME_IsSelection(editor);
619
620   ME_FindPixelPos(editor, x, y, &editor->pCursors[0], &editor->bCaretAtEnd);
621   
622   if (GetKeyState(VK_SHIFT)>=0)
623   {
624     editor->pCursors[1] = editor->pCursors[0];
625   }
626   else
627   {
628     if (!is_selection) {
629       editor->pCursors[1] = tmp_cursor;
630       is_selection = 1;
631     }
632   }
633   HideCaret(editor->hWnd);
634   ME_MoveCaret(editor);
635   if (is_selection)
636     ME_Repaint(editor);
637   ShowCaret(editor->hWnd);
638   ME_ClearTempStyle(editor);
639   ME_SendSelChange(editor);
640 }
641
642 void ME_MouseMove(ME_TextEditor *editor, int x, int y)
643 {
644   ME_Cursor tmp_cursor;
645   
646   y += ME_GetYScrollPos(editor);
647
648   tmp_cursor = editor->pCursors[0];
649   if (!ME_FindPixelPos(editor, x, y, &editor->pCursors[0], &editor->bCaretAtEnd))
650     /* return */;
651   
652   if (tmp_cursor.pRun == editor->pCursors[0].pRun && 
653       tmp_cursor.nOffset == editor->pCursors[0].nOffset)
654     return;
655   
656   HideCaret(editor->hWnd);
657   ME_MoveCaret(editor);
658   ME_Repaint(editor);
659   ShowCaret(editor->hWnd);
660   ME_SendSelChange(editor);
661 }
662
663 static ME_DisplayItem *ME_FindRunInRow(ME_TextEditor *editor, ME_DisplayItem *pRow, 
664                                 int x, int *pOffset, int *pbCaretAtEnd)
665 {
666   ME_DisplayItem *pNext, *pLastRun;
667   pNext = ME_FindItemFwd(pRow, diRunOrStartRow);
668   assert(pNext->type == diRun);
669   pLastRun = pNext;
670   *pbCaretAtEnd = FALSE;
671   do {
672     int run_x = pNext->member.run.pt.x;
673     int width = pNext->member.run.nWidth;
674     if (x < run_x)
675     {
676       if (pOffset) *pOffset = 0;
677       return pNext;
678     }
679     if (x >= run_x && x < run_x+width)
680     {
681       int ch = ME_CharFromPointCursor(editor, x-run_x, &pNext->member.run);
682       ME_String *s = pNext->member.run.strText;
683       if (ch < s->nLen) {
684         if (pOffset)
685           *pOffset = ch;
686         return pNext;          
687       }
688     }
689     pLastRun = pNext;
690     pNext = ME_FindItemFwd(pNext, diRunOrStartRow);
691   } while(pNext && pNext->type == diRun);
692   
693   if ((pLastRun->member.run.nFlags & MERF_ENDPARA) == 0)
694   {
695     pNext = ME_FindItemFwd(pNext, diRun);
696     if (pbCaretAtEnd) *pbCaretAtEnd = 1;
697     if (pOffset) *pOffset = 0;
698     return pNext;
699   } else {
700     if (pbCaretAtEnd) *pbCaretAtEnd = 0;
701     if (pOffset) *pOffset = 0;
702     return pLastRun;
703   }
704 }
705
706 static int ME_GetXForArrow(ME_TextEditor *editor, ME_Cursor *pCursor)
707 {
708   ME_DisplayItem *pRun = pCursor->pRun;
709   int x;
710
711   if (editor->nUDArrowX != -1)
712     x = editor->nUDArrowX;
713   else {
714     if (editor->bCaretAtEnd)
715     {
716       pRun = ME_FindItemBack(pRun, diRun);
717       assert(pRun);
718       x = pRun->member.run.pt.x + pRun->member.run.nWidth;
719     }
720     else {
721       x = pRun->member.run.pt.x;
722       x += ME_PointFromChar(editor, &pRun->member.run, pCursor->nOffset);
723     }
724     editor->nUDArrowX = x;
725   }
726   return x;
727 }
728
729 static void ME_ArrowUp(ME_TextEditor *editor, ME_Cursor *pCursor)
730 {
731   ME_DisplayItem *pRun = pCursor->pRun;
732   ME_DisplayItem *pItem, *pItem2;
733   int x = ME_GetXForArrow(editor, pCursor);
734   
735   if (editor->bCaretAtEnd && !pCursor->nOffset)
736   {
737     pRun = ME_FindItemBack(pRun, diRun);
738     if (!pRun)
739       return;
740   }
741   
742   /* start of this row */
743   pItem = ME_FindItemBack(pRun, diStartRow);
744   assert(pItem);
745   /* start of the previous row */
746   pItem2 = ME_FindItemBack(pItem, diStartRow);
747   /* no previous row = the first line of the first paragraph */
748   if (!pItem2) /* can't go up - don't go BOL (as in MS richedit) */
749     return;
750   /* FIXME
751   ME_WrapTextParagraph(editor, ME_FindItemBack(pItem2, diParagraph));
752   */
753   pCursor->pRun = ME_FindRunInRow(editor, pItem2, x, &pCursor->nOffset, &editor->bCaretAtEnd);
754 }
755
756 static void ME_ArrowDown(ME_TextEditor *editor, ME_Cursor *pCursor)
757 {
758   ME_DisplayItem *pRun = pCursor->pRun;
759   ME_DisplayItem *pItem;
760   int x = ME_GetXForArrow(editor, pCursor);
761   if (!pCursor->nOffset && editor->bCaretAtEnd)
762   {
763     pRun = ME_FindItemBack(pRun, diRun);
764 /*    x = pRun->member.run.pt.x + pRun->member.run.nWidth; */
765   }
766   /* start of the next row */
767   pItem = ME_FindItemFwd(pRun, diStartRow);
768   /* FIXME If diParagraph is before diStartRow, wrap the next paragraph?
769   */
770   if (!pItem)
771   {
772     /* next row not found - ignore */
773     return;
774   }
775   pCursor->pRun = ME_FindRunInRow(editor, pItem, x, &pCursor->nOffset, &editor->bCaretAtEnd);
776   assert(pCursor->pRun);
777   assert(pCursor->pRun->type == diRun);
778 }
779
780 static void ME_ArrowPageUp(ME_TextEditor *editor, ME_Cursor *pCursor)
781 {
782   ME_DisplayItem *pRun = pCursor->pRun;
783   ME_DisplayItem *pLast, *p;
784   int x, y, ys, yd, yp, yprev;
785   ME_Cursor tmp_curs = *pCursor;
786   
787   x = ME_GetXForArrow(editor, pCursor);
788   if (!pCursor->nOffset && editor->bCaretAtEnd)
789     pRun = ME_FindItemBack(pRun, diRun);
790   
791   p = ME_FindItemBack(pRun, diStartRowOrParagraph);
792   assert(p->type == diStartRow);
793   yp = ME_FindItemBack(p, diParagraph)->member.para.nYPos;
794   yprev = ys = y = yp + p->member.row.nYPos;
795   yd = y - editor->sizeWindow.cy;
796   pLast = p;
797   
798   do {
799     p = ME_FindItemBack(p, diStartRowOrParagraph);
800     if (!p)
801       break;
802     if (p->type == diParagraph) { /* crossing paragraphs */
803       if (p->member.para.prev_para == NULL)
804         break;
805       yp = p->member.para.prev_para->member.para.nYPos;
806       continue;
807     }
808     y = yp + p->member.row.nYPos;
809     if (y < yd)
810       break;
811     pLast = p;
812     yprev = y;
813   } while(1);
814   
815   pCursor->pRun = ME_FindRunInRow(editor, pLast, x, &pCursor->nOffset, &editor->bCaretAtEnd);
816   ME_UpdateSelection(editor, &tmp_curs);
817   if (yprev < editor->sizeWindow.cy)
818   {
819     ME_EnsureVisible(editor, ME_FindItemFwd(editor->pBuffer->pFirst, diRun));
820     ME_Repaint(editor);
821   }
822   else {
823     ME_Scroll(editor, 0, ys-yprev);
824     ME_Repaint(editor);
825   }
826   assert(pCursor->pRun);
827   assert(pCursor->pRun->type == diRun);
828 }
829
830 /* FIXME: in the original RICHEDIT, PageDown always scrolls by the same amount 
831    of pixels, even if it makes the scroll bar position exceed its normal maximum.
832    In such a situation, clicking the scrollbar restores its position back to the
833    normal range (ie. sets it to (doclength-screenheight)). */
834
835 static void ME_ArrowPageDown(ME_TextEditor *editor, ME_Cursor *pCursor)
836 {
837   ME_DisplayItem *pRun = pCursor->pRun;
838   ME_DisplayItem *pLast, *p;
839   int x, y, ys, yd, yp, yprev;
840   ME_Cursor tmp_curs = *pCursor;
841   
842   x = ME_GetXForArrow(editor, pCursor);
843   if (!pCursor->nOffset && editor->bCaretAtEnd)
844     pRun = ME_FindItemBack(pRun, diRun);
845   
846   p = ME_FindItemBack(pRun, diStartRowOrParagraph);
847   assert(p->type == diStartRow);
848   yp = ME_FindItemBack(p, diParagraph)->member.para.nYPos;
849   yprev = ys = y = yp + p->member.row.nYPos;
850   yd = y + editor->sizeWindow.cy;
851   pLast = p;
852   
853   do {
854     p = ME_FindItemFwd(p, diStartRowOrParagraph);
855     if (!p)
856       break;
857     if (p->type == diParagraph) {
858       yp = p->member.para.nYPos;
859       continue;
860     }
861     y = yp + p->member.row.nYPos;
862     if (y >= yd)
863       break;
864     pLast = p;
865     yprev = y;
866   } while(1);
867   
868   pCursor->pRun = ME_FindRunInRow(editor, pLast, x, &pCursor->nOffset, &editor->bCaretAtEnd);
869   ME_UpdateSelection(editor, &tmp_curs);
870   if (yprev >= editor->nTotalLength-editor->sizeWindow.cy)
871   {
872     ME_EnsureVisible(editor, ME_FindItemBack(editor->pBuffer->pLast, diRun));
873     ME_Repaint(editor);
874   }
875   else {
876     ME_Scroll(editor, 0, ys-yprev);
877     ME_Repaint(editor);
878   }
879   assert(pCursor->pRun);
880   assert(pCursor->pRun->type == diRun);
881 }
882
883 static void ME_ArrowHome(ME_TextEditor *editor, ME_Cursor *pCursor)
884 {
885   ME_DisplayItem *pRow = ME_FindItemBack(pCursor->pRun, diStartRow);
886   if (pRow) {
887     ME_DisplayItem *pRun;
888     if (editor->bCaretAtEnd && !pCursor->nOffset) {
889       pRow = ME_FindItemBack(pRow, diStartRow);
890       if (!pRow)
891         return;
892     }
893     pRun = ME_FindItemFwd(pRow, diRun);
894     if (pRun) {
895       pCursor->pRun = pRun;
896       pCursor->nOffset = 0;
897     }
898   }
899   editor->bCaretAtEnd = FALSE;
900 }
901
902 static void ME_ArrowCtrlHome(ME_TextEditor *editor, ME_Cursor *pCursor)
903 {
904   ME_DisplayItem *pRow = ME_FindItemBack(pCursor->pRun, diTextStart);
905   if (pRow) {
906     ME_DisplayItem *pRun = ME_FindItemFwd(pRow, diRun);
907     if (pRun) {
908       pCursor->pRun = pRun;
909       pCursor->nOffset = 0;
910     }
911   }
912 }
913
914 static void ME_ArrowEnd(ME_TextEditor *editor, ME_Cursor *pCursor)
915 {
916   ME_DisplayItem *pRow;
917   
918   if (editor->bCaretAtEnd && !pCursor->nOffset)
919     return;
920   
921   pRow = ME_FindItemFwd(pCursor->pRun, diStartRowOrParagraphOrEnd);
922   assert(pRow);
923   if (pRow->type == diStartRow) {
924     /* FIXME WTF was I thinking about here ? */
925     ME_DisplayItem *pRun = ME_FindItemFwd(pRow, diRun);
926     assert(pRun);
927     pCursor->pRun = pRun;
928     pCursor->nOffset = 0;
929     editor->bCaretAtEnd = 1;
930     return;
931   }
932   pCursor->pRun = ME_FindItemBack(pRow, diRun);
933   assert(pCursor->pRun && pCursor->pRun->member.run.nFlags & MERF_ENDPARA);
934   pCursor->nOffset = 0;
935   editor->bCaretAtEnd = FALSE;
936 }
937       
938 static void ME_ArrowCtrlEnd(ME_TextEditor *editor, ME_Cursor *pCursor)
939 {
940   ME_DisplayItem *p = ME_FindItemFwd(pCursor->pRun, diTextEnd);
941   assert(p);
942   p = ME_FindItemBack(p, diRun);
943   assert(p);
944   assert(p->member.run.nFlags & MERF_ENDPARA);
945   pCursor->pRun = p;
946   pCursor->nOffset = 0;
947   editor->bCaretAtEnd = FALSE;
948 }
949
950 BOOL ME_IsSelection(ME_TextEditor *editor)
951 {
952   return memcmp(&editor->pCursors[0], &editor->pCursors[1], sizeof(ME_Cursor))!=0;
953 }
954
955 static int ME_GetSelCursor(ME_TextEditor *editor, int dir)
956 {
957   int cdir = ME_GetCursorOfs(editor, 0) - ME_GetCursorOfs(editor, 1);
958   
959   if (cdir*dir>0)
960     return 0;
961   else
962     return 1;
963 }
964       
965 static BOOL ME_CancelSelection(ME_TextEditor *editor, int dir)
966 {
967   int cdir;
968   
969   if (GetKeyState(VK_SHIFT)<0)
970     return FALSE;
971   if (!memcmp(&editor->pCursors[0], &editor->pCursors[1], sizeof(ME_Cursor)))
972     return FALSE;
973   
974   cdir = ME_GetCursorOfs(editor, 0) - ME_GetCursorOfs(editor, 1);
975   
976   if (cdir*dir>0)
977     editor->pCursors[1] = editor->pCursors[0];
978   else
979     editor->pCursors[0] = editor->pCursors[1];
980   ME_Repaint(editor);
981   return TRUE;
982 }
983
984 BOOL ME_UpdateSelection(ME_TextEditor *editor, ME_Cursor *pTempCursor)
985 {
986   ME_Cursor old_anchor = editor->pCursors[1];
987   
988   if (GetKeyState(VK_SHIFT)>=0) /* cancelling selection */
989   {
990     /* any selection was present ? if so, it's no more, repaint ! */
991     editor->pCursors[1] = editor->pCursors[0];
992     if (memcmp(pTempCursor, &old_anchor, sizeof(ME_Cursor))) {
993       return TRUE;
994     }
995     return FALSE;
996   }
997   else
998   {
999     if (!memcmp(pTempCursor, &editor->pCursors[1], sizeof(ME_Cursor))) /* starting selection */
1000     {
1001       editor->pCursors[1] = *pTempCursor;
1002       return TRUE;
1003     }
1004   }
1005
1006   ME_Repaint(editor);
1007   return TRUE;
1008 }
1009
1010 static void ME_RepaintSelection(ME_TextEditor *editor, ME_Cursor *pTempCursor)
1011 {
1012   if (ME_UpdateSelection(editor, pTempCursor)) {
1013     ME_EnsureVisible(editor, editor->pCursors[0].pRun); 
1014     ME_Repaint(editor);
1015   }
1016 }
1017
1018 void ME_DeleteSelection(ME_TextEditor *editor)
1019 {
1020   int from, to;
1021   ME_GetSelection(editor, &from, &to);
1022   ME_DeleteTextAtCursor(editor, ME_GetSelCursor(editor,-1), to-from);
1023 }
1024
1025 ME_Style *ME_GetSelectionInsertStyle(ME_TextEditor *editor)
1026 {
1027   ME_Style *style;
1028   int from, to;
1029   ME_Cursor c;
1030   
1031   ME_GetSelection(editor, &from, &to);
1032   ME_CursorFromCharOfs(editor, from, &c);
1033   if (from != to) {
1034     style = c.pRun->member.run.style;
1035     ME_AddRefStyle(style); /* ME_GetInsertStyle has already done that */
1036   }
1037   else
1038     style = ME_GetInsertStyle(editor, 0);
1039   return style;
1040 }
1041
1042 void ME_SendSelChange(ME_TextEditor *editor)
1043 {
1044   SELCHANGE sc;
1045   if (!(editor->nEventMask & ENM_SELCHANGE))
1046     return;
1047   sc.nmhdr.hwndFrom = editor->hWnd;
1048   sc.nmhdr.idFrom = GetWindowLongW(editor->hWnd, GWLP_ID);
1049   sc.nmhdr.code = EN_SELCHANGE;
1050   SendMessageW(editor->hWnd, EM_EXGETSEL, 0, (LPARAM)&sc.chrg);
1051   sc.seltyp = SEL_EMPTY;
1052   if (sc.chrg.cpMin != sc.chrg.cpMax)
1053     sc.seltyp |= SEL_TEXT;
1054   if (sc.chrg.cpMin < sc.chrg.cpMax+1) /* wth were RICHEDIT authors thinking ? */
1055     sc.seltyp |= SEL_MULTICHAR;
1056   SendMessageW(GetParent(editor->hWnd), WM_NOTIFY, sc.nmhdr.idFrom, (LPARAM)&sc);
1057 }
1058
1059 BOOL ME_ArrowKey(ME_TextEditor *editor, int nVKey, int nCtrl)
1060 {
1061   int nCursor = 0;
1062   ME_Cursor *p = &editor->pCursors[nCursor];
1063   ME_Cursor tmp_curs = *p;
1064   
1065   switch(nVKey) {
1066     case VK_UP:
1067       ME_ArrowUp(editor, p);
1068       ME_ClearTempStyle(editor);
1069       ME_RepaintSelection(editor, &tmp_curs);
1070       ME_SendSelChange(editor);
1071       return TRUE;
1072     case VK_DOWN:
1073       ME_ArrowDown(editor, p);
1074       ME_ClearTempStyle(editor);
1075       ME_RepaintSelection(editor, &tmp_curs);
1076       ME_SendSelChange(editor);
1077       return TRUE;
1078     case VK_PRIOR:
1079       ME_ArrowPageUp(editor, p);
1080       ME_ClearTempStyle(editor);
1081       ME_SendSelChange(editor);
1082       return TRUE;
1083     case VK_NEXT:
1084       ME_ArrowPageDown(editor, p);
1085       ME_ClearTempStyle(editor);
1086       ME_SendSelChange(editor);
1087       return TRUE;
1088   }
1089   
1090   editor->nUDArrowX = -1;
1091   switch(nVKey) {
1092     case VK_BACK: { /* FIXME backspace and delete aren't the same, they act different wrt paragraph style of the merged paragraph */
1093       if (GetWindowLongW(editor->hWnd, GWL_STYLE) & ES_READONLY)
1094         return FALSE;
1095       if (ME_IsSelection(editor))
1096       {
1097         editor->bCaretAtEnd = FALSE; /* FIXME or maybe not */
1098         ME_DeleteSelection(editor);
1099         ME_UpdateRepaint(editor);
1100         ME_SendRequestResize(editor, FALSE);
1101         return TRUE;
1102       }
1103       if (ME_ArrowLeft(editor, p)) {
1104         editor->bCaretAtEnd = FALSE; /* FIXME or maybe not */
1105         ME_ClearTempStyle(editor);
1106         ME_MoveCaret(editor);
1107         ME_DeleteTextAtCursor(editor, nCursor, 1);
1108         ME_UpdateRepaint(editor);
1109         ME_SendRequestResize(editor, FALSE);
1110       }
1111       return TRUE;
1112     }
1113     case VK_DELETE: {
1114       if (GetWindowLongW(editor->hWnd, GWL_STYLE) & ES_READONLY)
1115         return FALSE;
1116       /* editor->bCaretAtEnd = 0; FIXME or maybe not */
1117       if (ME_IsSelection(editor))
1118       {
1119         ME_DeleteSelection(editor);
1120         ME_ClearTempStyle(editor);
1121         ME_UpdateRepaint(editor);
1122         ME_SendRequestResize(editor, FALSE);
1123         return TRUE;
1124       }
1125       ME_DeleteTextAtCursor(editor, nCursor, 1);
1126       ME_ClearTempStyle(editor);
1127       ME_UpdateRepaint(editor);
1128       ME_SendRequestResize(editor, FALSE);
1129       return TRUE;
1130     }
1131     case VK_HOME: {
1132       if (GetKeyState(VK_CONTROL)<0)
1133         ME_ArrowCtrlHome(editor, p);
1134       else
1135         ME_ArrowHome(editor, p);
1136       editor->bCaretAtEnd = 0;
1137       ME_ClearTempStyle(editor);
1138       ME_RepaintSelection(editor, &tmp_curs);
1139       ME_SendSelChange(editor);
1140       return TRUE;
1141     }
1142     case VK_END: 
1143       if (GetKeyState(VK_CONTROL)<0)
1144         ME_ArrowCtrlEnd(editor, p);
1145       else
1146         ME_ArrowEnd(editor, p);
1147       ME_ClearTempStyle(editor);
1148       ME_RepaintSelection(editor, &tmp_curs);
1149       ME_SendSelChange(editor);
1150       return TRUE;
1151     case VK_LEFT:
1152       editor->bCaretAtEnd = 0;
1153       if (ME_CancelSelection(editor, -1))
1154         return TRUE;
1155       ME_ArrowLeft(editor, p);
1156       ME_RepaintSelection(editor, &tmp_curs);
1157       ME_ClearTempStyle(editor);
1158       ME_SendSelChange(editor);
1159       return TRUE;
1160     case VK_RIGHT:
1161       editor->bCaretAtEnd = 0;
1162       if (ME_CancelSelection(editor, +1))
1163         return TRUE;
1164       ME_ArrowRight(editor, p);
1165       ME_RepaintSelection(editor, &tmp_curs);
1166       ME_ClearTempStyle(editor);
1167       ME_SendSelChange(editor);
1168       return TRUE;
1169   }
1170   return FALSE;
1171 }