home *** CD-ROM | disk | FTP | other *** search
/ Geek Gadgets 1 / ADE-1.bin / ade-bin / lib / octave / 1.1.1 / m / plot / polar_int.m < prev    next >
Encoding:
Text File  |  1996-10-12  |  3.3 KB  |  125 lines
  1. # Copyright (C) 1993, 1994, 1995 John W. Eaton
  2. # 
  3. # This file is part of Octave.
  4. # 
  5. # Octave is free software; you can redistribute it and/or modify it
  6. # under the terms of the GNU General Public License as published by the
  7. # Free Software Foundation; either version 2, or (at your option) any
  8. # later version.
  9. # 
  10. # Octave is distributed in the hope that it will be useful, but WITHOUT
  11. # ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  12. # FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  13. # for more details.
  14. # 
  15. # You should have received a copy of the GNU General Public License
  16. # along with Octave; see the file COPYING.  If not, write to the Free
  17. # Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA.
  18.  
  19. function polar_int (theta, rho, fmt)
  20.  
  21.   if (nargin == 1)
  22.     [nr, nc] = size (theta);
  23.     if (nr == 1)
  24.       theta = theta';
  25.       tmp = nr;
  26.       nr = nc;
  27.       nc = tmp;
  28.     endif
  29.     theta_i = imag (theta);
  30.     if (any (theta_i))
  31.       rho = theta_i;
  32.       theta = real (theta);
  33.     else
  34.       rho = theta;
  35.       theta = (1:nr)';
  36.     endif
  37.   endif
  38.  
  39.   if (nargin <= 2)
  40.     if (any (imag (theta)))
  41.       theta = real (theta);
  42.     endif
  43.     if (any (imag (rho)))
  44.       rho = real (rho);
  45.     endif
  46.     if (is_scalar (theta))
  47.       if (is_scalar (rho))
  48.         x = rho * cos (theta);
  49.         y = rho * sin (theta);
  50.         plot_2_s_s (x, y, fmt);
  51.       endif
  52.     elseif (is_vector (theta))
  53.       if (is_vector (rho))
  54.         if (length (theta) != length (rho))
  55.           error ("polar: vector lengths must match");
  56.         endif
  57.         if (rows (rho) == 1)
  58.           rho = rho';
  59.         endif
  60.         if (rows (theta) == 1)
  61.           theta = theta';
  62.         endif
  63.         x = rho .* cos (theta);
  64.         y = rho .* sin (theta);
  65.         plot_2_v_v (x, y, fmt);
  66.       elseif (is_matrix (rho))
  67.         [t_nr, t_nc] = size (theta);
  68.         if (t_nr == 1)
  69.           theta = theta';
  70.           tmp = t_nr;
  71.           t_nr = t_nc;
  72.           t_nc = tmp;
  73.         endif
  74.         [r_nr, r_nc] = size (rho);
  75.         if (t_nr != r_nr)
  76.           rho = rho'
  77.           tmp = r_nr;
  78.           r_nr = r_nc;
  79.           r_nc = tmp;
  80.         endif
  81.         if (t_nr != r_nr)
  82.           error ("polar: vector and matrix sizes must match");
  83.         endif
  84.         x = diag (cos (theta)) * rho;
  85.         y = diag (sin (theta)) * rho;
  86.         plot_2_v_m (x, y, fmt);
  87.       endif
  88.     elseif (is_matrix (theta))
  89.       if (is_vector (rho))
  90.         [r_nr, r_nc] = size (rho);
  91.         if (r_nr == 1)
  92.           rho = rho';
  93.           tmp = r_nr;
  94.           r_nr = r_nc;
  95.           r_nc = tmp;
  96.         endif
  97.         [t_nr, t_nc] = size (theta);
  98.         if (r_nr != t_nr)
  99.           theta = rho'
  100.           tmp = t_nr;
  101.           t_nr = t_nc;
  102.           t_nc = tmp;
  103.         endif
  104.         if (r_nr != t_nr)
  105.           error ("polar: vector and matrix sizes must match");
  106.         endif
  107.         diag_r = diag (r);
  108.         x = diag_r * cos (theta);
  109.         y = diag_r * sin (theta);
  110.         plot_2_m_v (x, y, fmt);
  111.       elseif (is_matrix (rho))
  112.         if (size (rho) != size (theta))
  113.           error ("polar: matrix dimensions must match");
  114.         endif
  115.         x = rho .* cos (theta);
  116.         y = rho .* sin (theta);
  117.         plot_2_m_m (x, y, fmt);
  118.       endif
  119.     endif
  120.   else
  121.     usage ("polar_int (x [, y])");
  122.   endif
  123.  
  124. endfunction
  125.