matlab-simulink-mcp 0.1.1__py3-none-any.whl → 0.1.2__py3-none-any.whl

matlab_simulink_mcp/data/helpers/describe_system.m

@@ -0,0 +1,64 @@
+function system_data = describe_system(system_path, main_system, open)
+
+% This function returns the information about a system layout as a JSON. 
+% It's still problematic with Simscape because some blocks such as
+% SolverConfiguration and PS-Simulink Convertor are defined internally as
+% SubSystems instead of Simscape blocks.
+
+load_system(main_system)
+
+if open
+    open_system(system_path);
+end
+
+blocks = find_system(system_path, 'SearchDepth', 1, 'Type', 'Block');
+
+    elements = {};
+    connections = {};
+
+    for i = 1:length(blocks)
+
+        blk = blocks{i};
+
+        if strcmp(blk, system_path) % Skips the subsystem itself if it is being asked for
+            continue;
+        end 
+
+        blk_source = get_param(blk, 'ReferenceBlock');
+
+        element = struct();
+        element.Name = get_param(blk, 'Name');
+        element.Type = get_param(blk, 'BlockType');
+        
+        if blk_source ~= ""
+            element.Source = blk_source;
+        end 
+
+        if ~strcmp(element.Type, "SimscapeBlock") && ~strcmp(element.Type, "SubSystem") && ~contains("Port", "port", 'IgnoreCase', true)
+            % It's a built-in simulink block
+            if blk_source ~= ""
+                element.Source = ['built-in/' element.Type];
+            end 
+            element.Type = "Block";
+                     
+        end
+
+        [inports, outports, simscapeports, connects] = get_ports_connections(blk, element.Name, element.Type);
+        
+        if ~isempty(inports), element.Inports = inports; end
+        if ~isempty(outports), element.Outports = outports; end
+        if ~isempty(simscapeports), element.SimscapePorts = simscapeports; end
+        if ~isempty(connects), connections = [connections, connects]; end
+
+        elements{end+1} = element;
+
+    end
+
+    system = struct();
+    system.Elements = elements;
+    system.Connections = connections; 
+
+    %system_data = jsonencode(system);
+    system_data = system;
+
+end

matlab_simulink_mcp/data/helpers/format_system.m

@@ -0,0 +1,21 @@
+function format_system()
+%function result = format_system(system_path, main_system, arrange)
+%load_system(main_system)
+
+%if arrange
+%    Simulink.BlockDiagram.arrangeSystem(system_path);
+%end 
+
+systems = find_system('type','block_diagram');
+
+for i = 1:numel(systems)
+    sys = systems{i};
+    hLines = find_system(sys, 'FindAll', 'on', 'Type', 'line', 'Connected', 'off');
+    delete_line(hLines);
+end
+
+%hLines = find_system(system_path, 'SearchDepth', 1, 'FindAll', 'on', 'Type', 'Line', 'Connected', 'off');
+%delete_line(hLines);
+%result = "Success";
+
+

matlab_simulink_mcp/data/helpers/get_images.m

@@ -0,0 +1,14 @@
+function cwds = get_images()
+    figs = findall(0, 'type', 'figure');  
+    num = numel(figs);
+    if num == 0 
+        cwds = {};     
+    else
+        cwds = cell(1, num);
+        for k = 1:num
+            filename = sprintf('temp_plot_%d.png', k);
+            saveas(figs(k), filename);
+            cwds{k} = fullfile(pwd, filename);
+        end 
+    end
+end

matlab_simulink_mcp/data/helpers/get_ports_connections.m

@@ -0,0 +1,124 @@
+function [inports, outports, simscapeports, connections] = get_ports_connections(blk, blk_name, blk_type)
+
+% This function gets, for a given block, its name and type, the names (if
+% it's a subsystem) and tags of its ports and connections.
+
+inports = {};
+outports = {};
+simscapeports = {};
+connections = {};
+
+is_subsystem = strcmp(blk_type, 'SubSystem');
+PC = get_param(blk, 'PortConnectivity');
+
+% First create a list of the port names. Unfortunately, this is only
+% possible for subsystems since only port blocks can have names, not ports themselves
+
+% For non-Simscape port blocks, the order through find_system is the
+% same as the ports in PortConnectivity matrix
+
+% For Simscape port blocks, the order through find_system is based on the
+% number of the port (displayed visually at the center of PortConnection block)
+% But in PortConnectivity matrix, the ports on Left are listed first and then right
+% Within each type, the order follows the visual number of the port
+% So we first get port blocks on left, and then right, and then concenate all of them
+
+if is_subsystem
+    
+    SimulinkPortBlks = find_system(blk, 'Regexp', 'on', 'BlockType', ...
+        'Inport|Outport|EnablePort|TriggerPort|ResetPort|ActionPort');
+    
+    PhyPortBlksLefts = find_system(blk, 'Regexp', 'on', 'BlockType', 'PMIOPort', 'Side', 'Left');
+    PhyPortBlksRights = find_system(blk, 'Regexp', 'on', 'BlockType', 'PMIOPort', 'Side', 'Right');
+
+    PortBlks = [SimulinkPortBlks(:); PhyPortBlksLefts(:); PhyPortBlksRights(:)];
+    PortNames = get_param(PortBlks, 'Name');
+
+    if isempty(PortNames)
+        is_subsystem = 0;
+    end 
+
+end 
+
+% Some Simscape blocks e.g. PS-Simulink Convertor are defined internally as
+% SubSystems instead of Simscape blocks. So the if condition above ensures
+% that if such blocks are counted as subsystems and port names gathered
+% (which would be empty), we flag the block manually as not a subsystem
+
+% Loop through the ports. For each port, create a strcuture with tag and
+% name (if the element is a subsystem) fields
+% For connections, first get the Simscape ports (i.e. LConnX and % RConnX).
+% For these ports, DstPort contains the handle of the attached port. So we
+% get the DstBlock's PortHandles through its handle, and then get its tag
+% For non-simscape ports, the DstPort contains -1 the field index of the
+% attached port in the DstBlock's PortConnectivity matrix.
+% Also, only consider outports for these ports as they are easily
+% distinguishable through an empty SrcBlock. For inports, the connections
+% will be gathered when the script is ran for the connected block. This
+% also prevents repitition. Inports also include ports such as trigger etc
+
+for i = 1:length(PC)
+    
+    Port = PC(i).Type;  
+    PortStr = struct('tag', Port);
+    if is_subsystem
+        try
+            PortStr.name = PortNames{i};
+        catch 
+        end 
+    end
+
+    if contains(Port, {'LConn', 'RConn'})
+        
+        simscapeports{end+1} = PortStr;
+
+        for j = 1:length(PC(i).DstBlock)
+            
+            DstBlockHandle = PC(i).DstBlock(j);
+            DstBlockName = get_param(DstBlockHandle, 'Name');
+            
+            DstPortHandle = PC(i).DstPort(j);
+            DstPH = get_param(DstBlockHandle, 'PortHandles');
+               
+            idxL = find(DstPH.LConn == DstPortHandle, 1);
+            idxR = find(DstPH.RConn == DstPortHandle, 1);
+                
+            if ~isempty(idxL)
+                DstPort = sprintf('LConn%d', idxL);
+            else
+                DstPort = sprintf('RConn%d', idxR);
+            end
+                
+            connections{end+1} = struct('from', sprintf("%s/%s", blk_name, Port), ...
+                'to', sprintf("%s/%s", DstBlockName, DstPort));
+        
+        end 
+    
+    else 
+        
+        if isempty(PC(i).SrcBlock) 
+            
+            outports{end+1} = PortStr;
+            
+            for j = 1:length(PC(i).DstBlock)
+   
+                DstBlockHandle = PC(i).DstBlock(j);
+                DstBlockName = get_param(DstBlockHandle, 'Name');
+
+                DstPortField = PC(i).DstPort(j) + 1;
+                DstPC = get_param(DstBlockHandle, 'PortConnectivity');
+                DstPort = DstPC(DstPortField).Type;
+                
+                connections{end+1} = struct('from', sprintf("%s/%s", blk_name, Port), ...
+                    'to', sprintf("%s/%s", DstBlockName, DstPort));
+            
+            end 
+       
+        else
+
+            inports{end+1} = PortStr;
+
+        end
+    end
+end
+end 

matlab_simulink_mcp/data/helpers/snapshot_system.m

@@ -0,0 +1,18 @@
+function cwd = snapshot_system(system_path, main_system, open)
+
+load_system(main_system)
+
+if open
+    open_system(system_path);
+end
+
+dpi = '150'; 
+path = "-s" + system_path;
+quality = "-r" + dpi;
+file = "snapshot.png";
+
+print(path, "-dpng", quality, file);
+
+cwd = fullfile(pwd, 'snapshot.png');
+
+end

matlab_simulink_mcp/data/helpers/validate_code.m

@@ -0,0 +1,2 @@
+function issues = validate_script(file_name)
+issues = {checkcode(file_name).message};

{matlab_simulink_mcp-0.1.1.dist-info → matlab_simulink_mcp-0.1.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: matlab-simulink-mcp
-Version: 0.1.1
+Version: 0.1.2
 Summary: An MCP server for allowing LLMs to interact with MATLAB and Simulink
 Author: Rohail
 License: MIT

{matlab_simulink_mcp-0.1.1.dist-info → matlab_simulink_mcp-0.1.2.dist-info}/RECORD

@@ -8,13 +8,19 @@ matlab_simulink_mcp/state.py,sha256=bT8Ud2E3W2eWxcqgSuRcuvj43O9BYERWciBYh82ey24,
 matlab_simulink_mcp/data/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 matlab_simulink_mcp/data/blacklist.txt,sha256=YXWTjcjFM-uTb3gD_pblEpPPGoIKjSHMFgZJdWfmuMw,656
 matlab_simulink_mcp/data/simlib_db.json,sha256=bni7DcGG6SofIcCMX5KSlaHOHOYsELT833reNhz4S5Q,538658
+matlab_simulink_mcp/data/helpers/describe_system.m,sha256=H80_L2YQ189_TI8qDWFA3xdvtPIHqhCm4llNoGQBeDw,2004
+matlab_simulink_mcp/data/helpers/format_system.m,sha256=f0TynV7VxZPDYHjXe0t56ZnFQoVb6NXvuriJHqFcWtM,576
+matlab_simulink_mcp/data/helpers/get_images.m,sha256=5n_gkJqDYxhLLje6V0sRT3U-zjoO4D-ub3k6EbAKRjE,376
+matlab_simulink_mcp/data/helpers/get_ports_connections.m,sha256=nw4iiW_jKg9R9cvZXCbPMoKOO-Ha0rbas_5we266nfA,4653
+matlab_simulink_mcp/data/helpers/snapshot_system.m,sha256=bitsd65cnPecIJpfe-OFH1Ia2oWTYFrWVMvVB3ZJN0I,316
+matlab_simulink_mcp/data/helpers/validate_code.m,sha256=cIKh1Fnl486UOPGUc4QYYWOn1MOkWcZNXuPclSkPg64,86
 matlab_simulink_mcp/installer/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 matlab_simulink_mcp/installer/installer.py,sha256=0SHMTaqZjSvb9zKx4in67bS5xT4JcnsO2U2vYvmb968,5949
 matlab_simulink_mcp/installer/launcher.py,sha256=aGuxkXnLcb1OZ4P6rHvXFi8KZsDUZVQlo0rNvjdmFqA,1438
 matlab_simulink_mcp/installer/win_elevate.py,sha256=LyHHUh4vWmIlIXxrM9tOhQUvVeMxcVdXMzkC3vbGm3A,514
-matlab_simulink_mcp-0.1.1.dist-info/licenses/LICENSE,sha256=qxK38PIeAkpnIu0mX7Re5nGKlHEx1Ez4p_4Yg4-3-Hw,1090
-matlab_simulink_mcp-0.1.1.dist-info/METADATA,sha256=fA8SHhzZyrCMsgm86TKMcHtMSezn14B9DP-IduVpaUc,4125
-matlab_simulink_mcp-0.1.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-matlab_simulink_mcp-0.1.1.dist-info/entry_points.txt,sha256=0ba8PFlQcaWyAqKXRIVDfDvrifeD5LY60lV7i6ak9oE,74
-matlab_simulink_mcp-0.1.1.dist-info/top_level.txt,sha256=90gv-RNCWWS7E-JMiqwUf2TjB_B8Zb2B-crAeXaABqU,20
-matlab_simulink_mcp-0.1.1.dist-info/RECORD,,
+matlab_simulink_mcp-0.1.2.dist-info/licenses/LICENSE,sha256=qxK38PIeAkpnIu0mX7Re5nGKlHEx1Ez4p_4Yg4-3-Hw,1090
+matlab_simulink_mcp-0.1.2.dist-info/METADATA,sha256=0dgAZ1Gw7NGleIOELyYHcDAdYhkaH2dRjzIGZhZwiJU,4125
+matlab_simulink_mcp-0.1.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+matlab_simulink_mcp-0.1.2.dist-info/entry_points.txt,sha256=0ba8PFlQcaWyAqKXRIVDfDvrifeD5LY60lV7i6ak9oE,74
+matlab_simulink_mcp-0.1.2.dist-info/top_level.txt,sha256=90gv-RNCWWS7E-JMiqwUf2TjB_B8Zb2B-crAeXaABqU,20
+matlab_simulink_mcp-0.1.2.dist-info/RECORD,,