vector animation plots

Author: itchono

postprocessing/plotting/imseq_vecex.m

@@ -0,0 +1,79 @@
+function imseq_vecex(y, cfg, nfmax, view_dim)
+% plot for vector export, based on maximum # of vector lines and frames
+% Expect y to be (6, N)
+% no need to interpolate vectors unless you have VERY big skips in L
+% Reference: https://www.mathworks.com/help/matlab/ref/getframe.html
+% Splits by orbit number
+
+[~, ~, ~, ~, ~, L] = unpack_mee(y);
+
+if nargin < 3
+    nfmax = 20;
+    view_dim = 3;
+end
+
+%% Data processing
+ind_orbits = find(diff(mod(L, 2*pi)) < 0);
+num_orbits = length(ind_orbits);
+
+%% Initial plot stuff
+L_sample = linspace(0, 2*pi, 60);
+plot_sphere(0, 0, 0, 6378e3, [0.3010, 0.7450, 0.9330]);
+hold on
+
+cm = colormap("turbo");
+th = title(sprintf("Orbit 1 of %d", num_orbits));
+
+% plot initial trace
+cart_sample = mee2cartesian([repmat(y(1:5, 1), 1, numel(L_sample)); L_sample]);
+plot3(cart_sample(1, :), cart_sample(2, :), cart_sample(3, :), "Color", cm(1, :), "DisplayName", "Initial Orbit", "LineWidth", 1, "LineStyle", "--");
+
+% plot main orbit
+main_orbit = plot3(cart_sample(1, :), cart_sample(2, :), cart_sample(3, :), "Color", "black", "DisplayName", "Current Orbit", "LineWidth", 1);
+
+% plot final trace
+cart_sample = mee2cartesian([repmat(y(1:5, end), 1, numel(L_sample)); L_sample]);
+plot3(cart_sample(1, :), cart_sample(2, :), cart_sample(3, :), "Color", cm(end, :), "DisplayName", "Final Orbit", "LineWidth", 1, "LineStyle", "--");
+
+% colorbar
+colorbar;
+clim([1,num_orbits]); 
+
+axis equal
+view(view_dim)
+
+%% Initialize path
+mkdir("anim")
+mkdir("anim", cfg.casename)
+
+%% Animation
+% spacing; never plot more than 20 frames
+stride = max(1, ceil(num_orbits/nfmax));
+
+% ensure we always plot the last orbit
+ii = 1; % counter
+for j = [1:stride:num_orbits, num_orbits]
+    % Update plots
+    idx = ind_orbits(j);
+    cart_sample = mee2cartesian([repmat(y(1:5, idx), 1, numel(L_sample)); L_sample]);
+    main_orbit.XData = cart_sample(1, :);
+    main_orbit.YData = cart_sample(2, :);
+    main_orbit.ZData = cart_sample(3, :);
+    
+    % Add shadow of previous orbits
+    colour_idx = ceil(j/num_orbits*length(cm));
+    plot3(cart_sample(1, :), cart_sample(2, :), cart_sample(3, :), "Color", cm(colour_idx, :), "LineWidth", 1);
+    
+    th.String = sprintf("Orbit %d of %d", j, num_orbits);
+    
+    % Draw and get frame
+    drawnow
+
+    % https://www.mathworks.com/matlabcentral/answers/1615265-physical-size-of-exportgraphics
+    hf = gcf;
+    set(hf, 'Units', 'centimeters',...
+          'OuterPosition', [4 4 12 12])
+    exportgraphics(hf, fullfile("anim", cfg.casename, sprintf("anim-%d.pdf", ii)), 'ContentType', 'vector')
+    ii = ii + 1; % JANK but works
+end
+end