## Copyright (C) 2000, 2001, 2003, 2005, 2006, 2007, 2009 Paul Kienzle
## Copyright (C) 2009 Jaroslav Hajek
##
## This file is part of Octave.
##
## Octave is free software; you can redistribute it and/or modify it
## under the terms of the GNU General Public License as published by
## the Free Software Foundation; either version 3 of the License, or (at
## your option) any later version.
##
## Octave is distributed in the hope that it will be useful, but
## WITHOUT ANY WARRANTY; without even the implied warranty of
## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
## General Public License for more details.
##
## You should have received a copy of the GNU General Public License
## along with Octave; see the file COPYING.  If not, see
## <http://www.gnu.org/licenses/>.

## -*- texinfo -*-
## @deftypefn {Function File} {} strjust (@var{s}, ["left"|"right"|"center"])
## Shift the non-blank text of @var{s} to the left, right or center of
## the string.  If @var{s} is a string array, justify each string in the
## array.  Null characters are replaced by blanks.  If no justification
## is specified, then all rows are right-justified.  For example:
##
## @example
## @group
## strjust (["a"; "ab"; "abc"; "abcd"])
##      @result{} ans =
##            a
##           ab
##          abc
##         abcd
## @end group
## @end example
## @end deftypefn

function y = strjust (x, just)

  if (nargin < 1 || nargin > 2)
    print_usage ();
  endif

  if (nargin == 1)
    just = "right";
  else
    just = tolower (just);
  endif

  if (ndims (x) != 2)
    error ("strjust: input must be a string or character matrix");
  endif

  if (isempty (x))
    y = x;
  else
    ## Apparently, Matlab considers nulls to be blanks as well; however, does
    ## not preserve the nulls, but rather converts them to blanks.  That's a
    ## bit unexpected, but it allows simpler processing, because we can move
    ## just the nonblank characters. So we'll do the same here.

    [nr, nc] = size (x);
    ## Find the indices of all nonblanks.
    nonbl = x != " " & x != "\0";
    [idx, jdx] = find (nonbl);

    if (strcmp (just, "right"))
      ## We wish to find the maximum column index for each row. Because jdx is
      ## sorted, we can take advantage of the fact that assignment is processed
      ## sequentially and for duplicate indices the last value will remain.
      maxs = nc * ones (nr, 1);
      maxs(idx) = jdx;
      shift = nc - maxs;
    elseif (strcmp (just, "left"))
      ## See above for explanation.
      mins = ones (nr, 1);
      mins(flipud (idx(:))) = flipud (jdx(:));
      shift = 1 - mins;
    else
      ## Use both of the above.
      mins = ones (nr, 1);
      mins(flipud (idx(:))) = flipud (jdx(:));
      maxs = nc * ones (nr, 1);
      maxs(idx) = jdx;
      shift = floor ((nc + 1 - maxs - mins) / 2); 
    endif

    ## Adjust the column indices.
    jdx += shift (idx);

    ## Create a blank matrix and position the nonblank characters.
    y = " "(ones (1, nr), ones (1, nc));
    y(sub2ind ([nr, nc], idx, jdx)) = x(nonbl);
  endif

endfunction

%!error <Invalid call to strjust> strjust();
%!error <Invalid call to strjust> strjust(["a";"ab"], "center", 1);
%!assert (strjust (["a"; "ab"; "abc"; "abcd"]),
%!        ["   a";"  ab"; " abc"; "abcd"]);
%!assert (strjust (["a"; "ab"; "abc"; "abcd"], "center"),
%!        [" a  "; " ab"; "abc "; "abcd"]);