[PATCH 5/5] riched20: Return a run ptr from the run insert function.

[Huw Davies]

Signed-off-by: Huw Davies <huw(a)codeweavers.com>
---
 dlls/riched20/caret.c  |  2 +-
 dlls/riched20/editor.h |  5 +++--
 dlls/riched20/run.c    | 49 ++++++++++++++++++++++++------------------
 dlls/riched20/undo.c   |  7 +++---
 4 files changed, 35 insertions(+), 28 deletions(-)

diff --git a/dlls/riched20/caret.c b/dlls/riched20/caret.c
index c56e5961f73..c5fa51894e8 100644
--- a/dlls/riched20/caret.c
+++ b/dlls/riched20/caret.c
@@ -471,7 +471,7 @@ ME_InternalInsertTextFromCursor(ME_TextEditor *editor, int nCursor,
   
   assert(p->pRun->type == diRun);
   
-  return ME_InsertRunAtCursor(editor, p, style, str, len, flags);
+  return run_get_di( run_insert( editor, p, style, str, len, flags ) );
 }
 
 static struct re_object* create_re_object(const REOBJECT *reo)
diff --git a/dlls/riched20/editor.h b/dlls/riched20/editor.h
index b41d0f7db68..1a9f3fd6fb0 100644
--- a/dlls/riched20/editor.h
+++ b/dlls/riched20/editor.h
@@ -119,8 +119,8 @@ int ME_RowNumberFromCharOfs(ME_TextEditor *editor, int nOfs) DECLSPEC_HIDDEN;
 
 /* run.c */
 ME_Run *run_create( ME_Style *s, int nFlags ) DECLSPEC_HIDDEN;
-ME_DisplayItem *ME_InsertRunAtCursor(ME_TextEditor *editor, ME_Cursor *cursor,
-                                     ME_Style *style, const WCHAR *str, int len, int flags) DECLSPEC_HIDDEN;
+ME_Run *run_insert( ME_TextEditor *editor, ME_Cursor *cursor,
+                    ME_Style *style, const WCHAR *str, int len, int flags ) DECLSPEC_HIDDEN;
 void ME_CheckCharOffsets(ME_TextEditor *editor) DECLSPEC_HIDDEN;
 void ME_PropagateCharOffset(ME_DisplayItem *p, int shift) DECLSPEC_HIDDEN;
 /* this one accounts for 1/2 char tolerance */
@@ -133,6 +133,7 @@ void run_join( ME_TextEditor *editor, ME_Run *run ) DECLSPEC_HIDDEN;
 ME_Run *run_next( ME_Run *run ) DECLSPEC_HIDDEN;
 ME_Run *run_next_all_paras( ME_Run *run ) DECLSPEC_HIDDEN;
 ME_Run *run_prev( ME_Run *run ) DECLSPEC_HIDDEN;
+ME_Run *run_prev_all_paras( ME_Run *run ) DECLSPEC_HIDDEN;
 ME_Run *run_split( ME_TextEditor *editor, ME_Cursor *cursor ) DECLSPEC_HIDDEN;
 void ME_UpdateRunFlags(ME_TextEditor *editor, ME_Run *run) DECLSPEC_HIDDEN;
 SIZE ME_GetRunSizeCommon(ME_Context *c, const ME_Paragraph *para, ME_Run *run, int nLen,
diff --git a/dlls/riched20/run.c b/dlls/riched20/run.c
index ed0273395c3..a5888960f38 100644
--- a/dlls/riched20/run.c
+++ b/dlls/riched20/run.c
@@ -57,6 +57,16 @@ ME_Run *run_next_all_paras( ME_Run *run )
     return NULL;
 }
 
+ME_Run *run_prev_all_paras( ME_Run *run )
+{
+    ME_DisplayItem *item = run_get_di( run ), *dummy = para_get_di( run->para );
+
+    if (ME_PrevRun( &dummy, &item, TRUE ))
+        return &item->member.run;
+
+    return NULL;
+}
+
 /******************************************************************************
  * ME_CanJoinRuns
  *
@@ -347,56 +357,53 @@ ME_Run *run_create( ME_Style *s, int flags )
 }
 
 /******************************************************************************
- * ME_InsertRunAtCursor
+ * run_insert
  *
  * Inserts a new run with given style, flags and content at a given position,
  * which is passed as a cursor structure (which consists of a run and 
  * a run-relative character offset).
  */
-ME_DisplayItem *
-ME_InsertRunAtCursor(ME_TextEditor *editor, ME_Cursor *cursor, ME_Style *style,
-                     const WCHAR *str, int len, int flags)
+ME_Run *run_insert( ME_TextEditor *editor, ME_Cursor *cursor, ME_Style *style,
+                    const WCHAR *str, int len, int flags )
 {
-  ME_DisplayItem *insert_before = cursor->pRun, *prev;
-  ME_Run *run;
+  ME_Run *insert_before = &cursor->pRun->member.run, *run, *prev;
 
   if (cursor->nOffset)
   {
-    if (cursor->nOffset == cursor->pRun->member.run.len)
+    if (cursor->nOffset == insert_before->len)
     {
-      insert_before = ME_FindItemFwd( cursor->pRun, diRun );
-      if (!insert_before) insert_before = cursor->pRun; /* Always insert before the final eop run */
+      insert_before = run_next_all_paras( insert_before );
+      if (!insert_before) insert_before = &cursor->pRun->member.run; /* Always insert before the final eop run */
     }
     else
     {
       run_split( editor, cursor );
-      insert_before = cursor->pRun;
+      insert_before = &cursor->pRun->member.run;
     }
   }
 
-  add_undo_delete_run( editor, insert_before->member.run.para->nCharOfs +
-                       insert_before->member.run.nCharOfs, len );
+  add_undo_delete_run( editor, insert_before->para->nCharOfs + insert_before->nCharOfs, len );
 
   run = run_create( style, flags );
-  run->nCharOfs = insert_before->member.run.nCharOfs;
+  run->nCharOfs = insert_before->nCharOfs;
   run->len = len;
-  run->para = insert_before->member.run.para;
+  run->para = insert_before->para;
   ME_InsertString( run->para->text, run->nCharOfs, str, len );
-  ME_InsertBefore( insert_before, run_get_di( run ) );
+  ME_InsertBefore( run_get_di( insert_before ), run_get_di( run ) );
   TRACE("Shift length:%d\n", len);
-  ME_PropagateCharOffset( insert_before, len );
-  para_mark_rewrap( editor, insert_before->member.run.para );
+  ME_PropagateCharOffset( run_get_di( insert_before ), len );
+  para_mark_rewrap( editor, insert_before->para );
 
   /* Move any cursors that were at the end of the previous run to the end of the inserted run */
-  prev = ME_FindItemBack( run_get_di( run ), diRun );
+  prev = run_prev_all_paras( run );
   if (prev)
   {
     int i;
 
     for (i = 0; i < editor->nCursors; i++)
     {
-      if (editor->pCursors[i].pRun == prev &&
-          editor->pCursors[i].nOffset == prev->member.run.len)
+      if (editor->pCursors[i].pRun == run_get_di( prev ) &&
+          editor->pCursors[i].nOffset == prev->len)
       {
         editor->pCursors[i].pRun = run_get_di( run );
         editor->pCursors[i].nOffset = len;
@@ -404,7 +411,7 @@ ME_InsertRunAtCursor(ME_TextEditor *editor, ME_Cursor *cursor, ME_Style *style,
     }
   }
 
-  return run_get_di( run );
+  return run;
 }
 
 static BOOL run_is_splittable( const ME_Run *run )
diff --git a/dlls/riched20/undo.c b/dlls/riched20/undo.c
index 55574dc412b..1f5c7ae4cd6 100644
--- a/dlls/riched20/undo.c
+++ b/dlls/riched20/undo.c
@@ -357,10 +357,9 @@ static void ME_PlayUndoItem(ME_TextEditor *editor, struct undo_item *undo)
   {
     ME_Cursor tmp;
     ME_CursorFromCharOfs(editor, undo->u.insert_run.pos, &tmp);
-    ME_InsertRunAtCursor(editor, &tmp, undo->u.insert_run.style,
-                         undo->u.insert_run.str,
-                         undo->u.insert_run.len,
-                         undo->u.insert_run.flags);
+    run_insert( editor, &tmp, undo->u.insert_run.style,
+                undo->u.insert_run.str, undo->u.insert_run.len,
+                undo->u.insert_run.flags );
     break;
   }
   case undo_delete_run:
-- 
2.23.0