Fixed plotting scripts to match new data structure

Author: fabriciocravo

.DS_Store

@@ -137,10 +137,10 @@ function calculate_power(varargin)
             eval([method ' = PowerRes;']);  % creates a variable named after the method
 
             save(file_name, method, '-append');
-
-            fprintf('Finished power calculation for file %s \n', file_name);
 
         end
+
+        fprintf('Finished power calculation for file %s \n', file_name);
 
     end
 

calculate_power.m

@@ -28,15 +28,16 @@
 % Author: Fabricio Cravo | Date: March 2025
 
 % Datasets - Commented for easy use
-Params.data_dir = './data/s_abcd_fc_rosenblatt.mat';
+% Params.data_dir = './data/s_abcd_fc_rosenblatt.mat';
 % Params.data_dir = './data/s_hcp_fc_noble_tasks.mat';
-% Params.data_dir = './data/s_hcp_act_noble_1.mat';
+Params.data_dir = './data/s_hcp_act_noble_1.mat';
+Params.output = NaN;
 
 % Save specifications - if NaN output becomes dataset file name
 Params.save_directory = './power_calculator_results/';
 Params.gt_data_dir = './power_calculator_results/ground_truth/';
 
-Params.output = 'abcd_100_reps';
+
 
 % Options - full_file, compact_file;
 Params.subsample_file_type = 'compact_file';

config_files/setparams.m

@@ -6,11 +6,11 @@
 
     % Default Weibull function
     % default_func = @(params, x) 100*(1 - exp(-(x./params(1)).^params(2)));
-    default_func = @(params, x) 100 ./ (1 + (params(1) ./ x).^params(2));
+    default_func = @(params, x) params(3) ./ (1 + (params(1) ./ x).^params(2));
 
     addParameter(p, 'fit_function', default_func, @(x) isa(x, 'function_handle'));
-    addParameter(p, 'lower_bounds', [1, 0.1], @isnumeric);  % Allow custom initial params
-    addParameter(p, 'upper_bounds', [100000, 5], @isnumeric);  % Allow custom initial params
+    addParameter(p, 'lower_bounds', [1, 0.1, 0], @isnumeric);  % Allow custom initial params
+    addParameter(p, 'upper_bounds', [100000, 5, 100], @isnumeric);  % Allow custom initial params
 
     parse(p, varargin{:});
 
@@ -24,9 +24,10 @@
     % b: start with exponential shape (b=1)
     a_init = median(results_x);
     b_init = 1;
+    c_init = 50;
 
     % Initial guess: [scale, shape]
-    initial_params = [a_init, b_init];
+    initial_params = [a_init, b_init, c_init];
 
     % Define cost function (sum of squared residuals)
     cost_func = @(params) sum((results_y - power_func(params, results_x)).^2);

data_fit_scripts/fit_power_curve.m

@@ -0,0 +1,13 @@
+function edge_groups = get_edge_groups_from_meta_data(meta_data)
+
+    if isfield(meta_data, 'edge_groups')
+        edge_groups = meta_data.edge_groups;
+        return;
+    end
+
+    if isfield(meta_data, 'rep_parameters')
+        edge_groups = meta_data.rep_parameters.edge_groups;
+        return;
+    end
+
+end

file_handlers/get_edge_groups_from_meta_data.m

@@ -0,0 +1,13 @@
+function mask = get_mask_from_meta_data(meta_data)
+
+    if isfield(meta_data, 'mask')
+        mask = meta_data.mask;
+        return;
+    end 
+
+    if isfield(meta_data, 'rep_parameters')
+        mask = meta_data.rep_parameters.mask;
+        return
+    end
+
+end

file_handlers/get_mask_from_meta_data.m

@@ -0,0 +1,13 @@
+function study_name = get_study_name_from_meta_data(meta_data)
+  
+    if isfield(meta_data, 'study_name')
+        study_name = meta_data.study_name;
+        return;
+    end
+
+    if isfield(meta_data, 'rep_parameters')
+        study_name = meta_data.rep_parameters.test_name;
+        return;
+    end
+
+end

file_handlers/get_study_name_from_meta_data.m

@@ -1,9 +1,10 @@
-function average_array = average_heat_map_gen(method, n_subs, directory, varargin)   
+function [average_array, c_above_80] = ...
+    average_heat_map_gen(method, n_subs, directory, varargin)   
 %
 % For the paper: Shitf - down to select for paste
-% - average_heat_map_gen('Parametric_FDR',40,'/Users/f.cravogomes/Desktop/Pc_Res_Updated/power_calculation/hcp_fc')
-% - average_heat_map_gen('Constrained_cpp_FWER',40,'/Users/f.cravogomes/Desktop/Pc_Res_Updated/power_calculation/abcd_fc_test2')
-    
+% - average_heat_map_gen('Constrained_FDR',80,'/Users/f.cravogomes/Desktop/Pc_Res_Updated/SHOCK Paper Results/power_calculation/hcp_fc')
+% - average_heat_map_gen('Size_cpp',500,'/Users/f.cravogomes/Desktop/Pc_Res_Updated/SHOCK Paper Results/power_calculation/abcd_100_sex/')
+
     % Create parser object
     p = inputParser;
 
@@ -45,23 +46,28 @@
         end
 
         data = load(fullfile(files(i_f).folder, files(i_f).name));
-
-        if data.meta_data.subject_number ~= n_subs
+ 
+        if get_sub_number_from_meta_data(data.meta_data) ~= n_subs
             continue
         end
-        
+
         data_acrs_tasks{end + 1} = data.(method).tpr;
     end
 
     % Calculate average accross tasks 
     data_acrs_tasks = cat(3, data_acrs_tasks{:});
     average_array = mean(data_acrs_tasks, 3);
+
+    c_above_80 = sum(average_array > 80, 'all');
 
     % Get unflattening function based on experiment mask
-    unflat_matrix_fun = unflatten_matrix(data.meta_data.rep_parameters.mask);
+    mask = get_mask_from_meta_data(data.meta_data);
+    unflat_matrix_fun = unflatten_matrix(mask);
 
     % Unflat
-    if ~strcmp(data.(method).meta_data.level, 'network')
+    if ~strcmp( ...
+            get_stats_level_from_method_data(method, data.(method), data.meta_data), ...
+            'network')
 
         try
             power_data = unflat_matrix_fun(average_array);
@@ -89,8 +95,9 @@
         end
 
     end
-
-    edge_groups_line = data.meta_data.rep_parameters.edge_groups(1, :);
+    
+    edge_groups = get_edge_groups_from_meta_data(data.meta_data);
+    edge_groups_line = edge_groups(1, :);
 
 
     previous_number = edge_groups_line(1);
@@ -114,21 +121,43 @@
         end
 
     end
-
-    % Assuming your 2D matrix is called 'data'
-    figure;
+    
+    % Figure config
+    figure('Position', [100, 100, 800, 700]);
+    
     imagesc(power_data);
-    colorbar;
-    title(heatmap_title);
-    xlabel(x_label);
-    ylabel(y_label);
+    
+    c = colorbar;
+    c.FontSize = 12;
+    c.FontWeight = 'bold';
+    c.Label.String = 'Power (%)';
+    c.Label.FontSize = 14;
+    c.Label.FontWeight = 'bold';
+    
+    title(heatmap_title, 'FontSize', 16, 'FontWeight', 'bold');
+    
+    xlabel(x_label, 'FontSize', 14, 'FontWeight', 'bold');
+    ylabel(y_label, 'FontSize', 14, 'FontWeight', 'bold');
+    
+    set(gca, 'FontSize', 11, 'FontWeight', 'normal');
+    
+    set(gca, 'LineWidth', 1.5);
+    
+    set(gca, 'Layer', 'top');
+    set(gca, 'GridLineStyle', '-');
+    set(gca, 'GridColor', [0.5, 0.5, 0.5]);
+    set(gca, 'GridAlpha', 0.1);
+    grid on;
 
     add_network_lines_from_starts(network_boundaries);
 
     heat_map_color = custom_colors('sci_blu');
-
-    % Convert to 0-1 range
     colormap(heat_map_color);
+    
+    axis square;  
+    
+    set(gca, 'TickDir', 'out');
+    set(gca, 'Box', 'on');  
 
 end
 

plot_scripts/.DS_Store

@@ -19,7 +19,7 @@
     parse(p, varargin{:});
 
     decimal_places = p.Results.format_numbers;
-    
+
     if size(cell_data, 2) ~= numel(headers)
         error('Header data must be provided with correct number of headers')
     end