% Generate transfer functions for loop shaping performance requirements
% from parameters specified in uav_params.m
%
% Copyright (C) 2024, Naoki Sean Pross, ETH Zürich
% This work is distributed under a permissive license, see LICENSE.txt
%
% Arguments:
%   PARAMS  Struct of design parameters and constants generated by uav_params
%   PLOT    When set to 'true' it plots the inverse magnitude of the
%           performance transfer function
%
% Return value:
%   PERF    Struct performance transfer functions


function [perf] = uav_performance_musyn(params, do_plots)

% Laplace variable
s = tf('s');

% TODO: set proper constraints
G = 1e2 / (s + 1e2);

W_Palpha = G; 
W_Pomega = G;

W_Pxy = G;
W_Pz = G;

W_Pxydot = tf(0);
W_Pzdot = tf(0);
 
W_Pphitheta = tf(0);
W_Ppsi = tf(0);

W_PTheta = tf(0);

% Construct performance vector by combining xy and z
W_PP = blkdiag(W_Pxy * eye(2), W_Pz);
W_PPdot = blkdiag(W_Pxydot * eye(2), W_Pzdot);
W_PTheta = blkdiag(W_Pphitheta * eye(2), W_Ppsi);

perf = struct(...
  'FlapAngle', W_Palpha * eye(4), ...
  'Thrust', W_Pomega, ...
  'Position', W_PP, ...
  'Velocity', W_PPdot, ...
  'Angles', W_PTheta);

if do_plots
  % Bode plots of performance requirements
  figure; hold on;

  bodemag(W_Palpha);
  bodemag(W_Pomega);
  bodemag(W_Pxy);
  bodemag(W_Pz);
  bodemag(W_Pxydot);
  bodemag(W_Pzdot);
  bodemag(W_Pphitheta);
  bodemag(W_Ppsi);

  grid on;
  legend('$W_{P,\alpha}$', '$W_{P,\omega}$', ...
    '$W_{P,xy}$', '$W_{P,z}$', ...
    '$W_{P,\dot{x}\dot{y}}$', '$W_{P,\dot{z}}$', ...
    '$W_{P,\phi\theta}$', '$W_{P,\psi}$', ...
    'interpreter', 'latex', 'fontSize', 8);
  title('Performance Requirements');

  % Step response of position requirements
  figure; hold on;
  step(W_Pxy); step(W_Pz);
  step(W_Pxydot); step(W_Pzdot);
  step(W_Palpha);
  step(W_Pomega);
  grid on;
  legend('$W_{P,xy}$', '$W_{P,z}$', ...
    '$W_{P,\dot{x}\dot{y}}$', '$W_{P,\dot{z}}$', ...
    '$W_{P,\alpha}$', '$W_{P,\omega}$', ...
    'interpreter', 'latex', 'fontSize', 8);
  title('Step responses of performance requirements');
end

end

% Make a n-order performance weight function
%
% Arguments:
%   OMEGA  Cutting frequency (-3dB)
%   A      Magnitude at DC, i.e. |Wp(0)|
%   M      Magnitude at infinity, i.e. |Wp(inf)|
%   ORD    Order
function [Wp] = make_weight(omega, A, M, ord)

if nargin > 3
  n = ord;
else
  n = 1;
end

s = tf('s');
Wp = (s / (M^(1/n)) + omega)^n / (s + omega * A^(1/n))^n;

end
% vim: ts=2 sw=2 et: