[已解决]在zemax中 如何將matlab代碼轉換成 python代碼

游家志

各位大神們 想跟大家詢問
小弟我有寫好一組matlab 代碼 是為了zemax 跑公差用的  想問一下 如果想將這組matlab代碼轉成python代碼 可以怎麼做呢 尋求大神們幫忙
以下是小弟的matlab代碼 再麻煩大神們幫忙

感激不盡

~~~
function [ r ] = MATLABStandaloneApplication( args )

if ~exist('args', 'var')
    args = [];
end

% Initialize the OpticStudio connection
TheApplication = InitConnection();
if isempty(TheApplication)
    % failed to initialize a connection
    r = [];
else
    try
        r = BeginApplication(TheApplication, args);
        CleanupConnection(TheApplication);
    catch err
        CleanupConnection(TheApplication);
        rethrow(err);
    end
end
end


function [r] = BeginApplication(TheApplication, args)

import ZOSAPI.*;

TheSystem = TheApplication.PrimarySystem;

    %Figure Setting
    xA = "Two Mirror Gap Change (um)";
    yA = "Contrast";
    yA1 = "Contrast Deformation";
    Leg1 = ['X - Contrast'; 'Y - Contrast'];
    Ptitle = 'Contrast in X/Y vs. 2 Mirror Gap';
    Ptitle1 = 'Contrast Deformation vs. 2 Mirror Gap';

    sqn = 20;
    % Set up primary optical system
%     TheSystem = TheApplication.PrimarySystem;
    sampleDir = TheApplication.SamplesDir;
        % Open file
    testFile = System.String.Concat(sampleDir, '\z_5a.zma');
    if (exist(char(testFile)) == 0)
        fprintf('You need to run Z_5a.zmx before running this example\n');
        r = [];
        return;
    end
    TheSystem.LoadFile(testFile,false);
    TheLDE = TheSystem.LDE;
    Surface_0 = TheLDE.GetSurfaceAt(0);
    Surface_1 = TheLDE.GetSurfaceAt(1);
    Surface_2 = TheLDE.GetSurfaceAt(2);
    Surface_5 = TheLDE.GetSurfaceAt(5);


% Set the calculation cycle number
CN = 41;  Ar_nb = CN-1;
% set max gap variation, unit = micron
Dmzx = -20;   % Gap
Dmzx_Step = -2*Dmzx/(CN-1);

DmN = (1:CN);
DmXX = Dmzx + (DmN - 1)*Dmzx_Step;

Conx = zeros(1, Ar_nb);
Cony = Conx;
for jj = 1:CN
     Surface_2.Thickness = (Dmzx + (jj-1)*Dmzx_Step)/1000; %Wafer gap change
     TheSystem.Save();

     pop = TheSystem.Analyses.Get_AnalysisAtIndex(4);
     pop_setting = pop.GetSettings();
     cfg = System.String.Concat(sampleDir,'\so_lens.cfg');
     pop_setting.SaveTo(cfg);
     pop_setting.LoadFrom(cfg);
     pop_setting.ModifySettings(cfg, 'EXD_FILESIZE', 0.832) %square array size.
     pop_setting.ModifySettings(cfg, 'EXD_DISPLAYSIZE', 0.832) %square array size.
     pop.ApplyAndWaitForCompletion();
     results = pop.GetResults();
     header = results.HeaderData;
     AAA = results.DataGrids(1).Values.double;
     [ax, ay] = size(AAA);
     arxn = round(ax/10);
     aryn = round(ay/10);
     BBB = AAA(arxn:ax-arxn, aryn:ay-aryn);


    Iy0 = max(BBB, [], 1); Ix0 = max(BBB, [], 2);
    Iy = sort(Iy0, 'descend'); Ix = sort(Ix0, 'descend');
    Iymx = mean(Iy(1:sqn)); Iymn = mean(Iy(end-sqn:end));
    Ixmx = mean(Ix(1:sqn)); Ixmn = mean(Ix(end-sqn:end));
    Conx(jj) = (Ixmx - Ixmn)/(Ixmx + Ixmn);
    Cony(jj) = (Iymx - Iymn)/(Iymx + Iymn);   
end

ConDiff = 2*abs(Cony-Conx)./(Conx+Cony);
figure(1)
plot(DmXX, Conx, 'b', DmXX, Cony, 'r');
legend(Leg1);
xlabel(xA);
ylabel(yA);
title(Ptitle)
fontsize(gcf,scale=1.2);

figure(2)
plot(DmXX, ConDiff);
% legend(Leg1);
xlabel(xA);
ylabel(yA1);
title(Ptitle1);
fontsize(gcf,scale=1.2);

    Surface_0.Thickness = 186.13848;
    Surface_2.Thickness = 0;
    TheSystem.Save();



r = [];

end

function app = InitConnection()

import System.Reflection.*;

% Find the installed version of OpticStudio.
zemaxData = winqueryreg('HKEY_CURRENT_USER', 'Software\Zemax', 'ZemaxRoot');
NetHelper = strcat(zemaxData, '\ZOS-API\Libraries\ZOSAPI_NetHelper.dll');
% Note -- uncomment the following line to use a custom NetHelper path
% NetHelper = 'C:\Users\swang250\OneDrive - JNJ\Documents\Zemax\ZOS-API\Libraries\ZOSAPI_NetHelper.dll';
% This is the path to OpticStudio
NET.addAssembly(NetHelper);

success = ZOSAPI_NetHelper.ZOSAPI_Initializer.Initialize();
% Note -- uncomment the following line to use a custom initialization path
% success = ZOSAPI_NetHelper.ZOSAPI_Initializer.Initialize('C:\Program Files\OpticStudio\');
if success == 1
    LogMessage(strcat('Found OpticStudio at: ', char(ZOSAPI_NetHelper.ZOSAPI_Initializer.GetZemaxDirectory())));
else
    app = [];
    return;
end

% Now load the ZOS-API assemblies
NET.addAssembly(AssemblyName('ZOSAPI_Interfaces'));
NET.addAssembly(AssemblyName('ZOSAPI'));

% Create the initial connection class
TheConnection = ZOSAPI.ZOSAPI_Connection();

% Attempt to create a Standalone connection

% NOTE - if this fails with a message like 'Unable to load one or more of
% the requested types', it is usually caused by try to connect to a 32-bit
% version of OpticStudio from a 64-bit version of MATLAB (or vice-versa).
% This is an issue with how MATLAB interfaces with .NET, and the only
% current workaround is to use 32- or 64-bit versions of both applications.
app = TheConnection.CreateNewApplication();
if isempty(app)
   HandleError('An unknown connection error occurred!');
end
if ~app.IsValidLicenseForAPI
    HandleError('License check failed!');
    app = [];
end

end

function LogMessage(msg)
disp(msg);
end

function HandleError(error)
ME = MException('zosapi:HandleError', error);
throw(ME);
end

function  CleanupConnection(TheApplication)
% Note - this will close down the connection.

% If you want to keep the application open, you should skip this step
% and store the instance somewhere instead.
TheApplication.CloseApplication();
end

~~~

最佳答案

月排行榜 / 总排行榜

不如吃茶

2023-6-14 09:34

ChatGPT写的，你试试
又添加了注释

import os
import clr
from System import String
from matplotlib import pyplot as plt

def MATLABStandaloneApplication(args=None):
    if args is None:
        args = []

    # Initialize the OpticStudio connection
    TheApplication = InitConnection()
    if TheApplication is None:
        # failed to initialize a connection
        return None
    else:
        try:
            r = BeginApplication(TheApplication, args)
            CleanupConnection(TheApplication)
            return r
        except Exception as err:
            CleanupConnection(TheApplication)
            raise err

def BeginApplication(TheApplication, args):
    ZOSAPI = clr.Import('ZOSAPI')
    TheSystem = TheApplication.PrimarySystem

    # Figure Setting
    xA = "Two Mirror Gap Change (um)"
    yA = "Contrast"
    yA1 = "Contrast Deformation"
    Leg1 = ['X - Contrast', 'Y - Contrast']
    Ptitle = 'Contrast in X/Y vs. 2 Mirror Gap'
    Ptitle1 = 'Contrast Deformation vs. 2 Mirror Gap'

    sqn = 20
    # Set up primary optical system
    sampleDir = TheApplication.SamplesDir
    # Open file
    testFile = os.path.join(sampleDir, 'z_5a.zma')
    if not os.path.exists(testFile):
        print('You need to run Z_5a.zmx before running this example')
        return None
    TheSystem.LoadFile(testFile, False)
    TheLDE = TheSystem.LDE
    Surface_0 = TheLDE.GetSurfaceAt(0)
    Surface_1 = TheLDE.GetSurfaceAt(1)
    Surface_2 = TheLDE.GetSurfaceAt(2)
    Surface_5 = TheLDE.GetSurfaceAt(5)

    # Set the calculation cycle number
    CN = 41
    Ar_nb = CN - 1
    # set max gap variation, unit = micron
    Dmzx = -20  # Gap
    Dmzx_Step = -2 * Dmzx / (CN - 1)

    DmN = range(1, CN + 1)
    DmXX = [Dmzx + (n - 1) * Dmzx_Step for n in DmN]

    Conx = [0] * Ar_nb
    Cony = [0] * Ar_nb
    for jj in DmN:
        Surface_2.Thickness = (Dmzx + (jj - 1) * Dmzx_Step) / 1000  # Wafer gap change
        TheSystem.Save()

        pop = TheSystem.Analyses.Get_AnalysisAtIndex(4)
        pop_setting = pop.GetSettings()
        cfg = os.path.join(sampleDir, 'so_lens.cfg')
        pop_setting.SaveTo(cfg)
        pop_setting.LoadFrom(cfg)
        pop_setting.ModifySettings(cfg, 'EXD_FILESIZE', 0.832)  # square array size.
        pop_setting.ModifySettings(cfg, 'EXD_DISPLAYSIZE', 0.832)  # square array size.
        pop.ApplyAndWaitForCompletion()
        results = pop.GetResults()
        header = results.HeaderData
        AAA = results.DataGrids(1).Values.double
        ax, ay = AAA.GetLength(0), AAA.GetLength(1)
        arxn = round(ax / 10)
        aryn = round(ay / 10)
        BBB = [[AAA[i, j] for j in range(aryn, ay - aryn)] for i in range(arxn, ax - arxn)]

        Iy0 = [max(row) for row in BBB]
        Ix0 = [max(col) for col in zip(*BBB)]
        Iy = sorted(Iy0, reverse=True)
        Ix = sorted(Ix0, reverse=True)
        Iymx = sum(Iy[:sqn]) / sqn
        Iymn = sum(Iy[-sqn:]) / sqn
        Ixmx = sum(Ix[:sqn]) / sqn
        Ixmn = sum(Ix[-sqn:]) / sqn
        Conx[jj - 1] = (Ixmx - Ixmn) / (Ixmx + Ixmn)
        Cony[jj - 1] = (Iymx - Iymn) / (Iymx + Iymn)

    ConDiff = [2 * abs(Cony - Conx) / (Conx + Cony) for i in range(Ar_nb)]

    plt.figure(1)
    plt.plot(DmXX, Conx, 'b', DmXX, Cony, 'r')
    plt.legend(Leg1)
    plt.xlabel(xA)
    plt.ylabel(yA)
    plt.title(Ptitle)

    plt.figure(2)
    plt.plot(DmXX, ConDiff)
    plt.xlabel(xA)
    plt.ylabel(yA1)
    plt.title(Ptitle1)

    Surface_0.Thickness = 186.13848
    Surface_2.Thickness = 0
    TheSystem.Save()

    return []

def InitConnection():
    NetHelper = r'C:\Program Files\Zemax OpticStudio\ZOS-API\Libraries\ZOSAPI_NetHelper.dll'

    try:
        clr.AddReference(NetHelper)
    except Exception as e:
        print(f"Failed to add reference to {NetHelper}")
        return None

    import ZOSAPI

    success = ZOSAPI.ZOSAPI_Initializer.Initialize()
    if success == 1:
        LogMessage(f"Found OpticStudio at: {ZOSAPI.ZOSAPI_Initializer.GetZemaxDirectory()}")
    else:
        return None

    clr.AddReference("ZOSAPI_Interfaces")
    clr.AddReference("ZOSAPI")

    TheConnection = ZOSAPI.ZOSAPI_Connection()
    app = TheConnection.CreateNewApplication()
    if app is None:
        HandleError("An unknown connection error occurred!")
    if not app.IsValidLicenseForAPI:
        HandleError("License check failed!")
        return None

    return app

def LogMessage(msg):
    print(msg)

def HandleError(error):
    raise Exception(error)

def CleanupConnection(TheApplication):
    TheApplication.CloseApplication()

# 调用示例
MATLABStandaloneApplication()

跳转到最佳答案楼层

不如吃茶

ChatGPT写的，你试试
又添加了注释

import os
import clr
from System import String
from matplotlib import pyplot as plt

def MATLABStandaloneApplication(args=None):
    if args is None:
        args = []

    # Initialize the OpticStudio connection
    TheApplication = InitConnection()
    if TheApplication is None:
        # failed to initialize a connection
        return None
    else:
        try:
            r = BeginApplication(TheApplication, args)
            CleanupConnection(TheApplication)
            return r
        except Exception as err:
            CleanupConnection(TheApplication)
            raise err

def BeginApplication(TheApplication, args):
    ZOSAPI = clr.Import('ZOSAPI')
    TheSystem = TheApplication.PrimarySystem

    # Figure Setting
    xA = "Two Mirror Gap Change (um)"
    yA = "Contrast"
    yA1 = "Contrast Deformation"
    Leg1 = ['X - Contrast', 'Y - Contrast']
    Ptitle = 'Contrast in X/Y vs. 2 Mirror Gap'
    Ptitle1 = 'Contrast Deformation vs. 2 Mirror Gap'

    sqn = 20
    # Set up primary optical system
    sampleDir = TheApplication.SamplesDir
    # Open file
    testFile = os.path.join(sampleDir, 'z_5a.zma')
    if not os.path.exists(testFile):
        print('You need to run Z_5a.zmx before running this example')
        return None
    TheSystem.LoadFile(testFile, False)
    TheLDE = TheSystem.LDE
    Surface_0 = TheLDE.GetSurfaceAt(0)
    Surface_1 = TheLDE.GetSurfaceAt(1)
    Surface_2 = TheLDE.GetSurfaceAt(2)
    Surface_5 = TheLDE.GetSurfaceAt(5)

    # Set the calculation cycle number
    CN = 41
    Ar_nb = CN - 1
    # set max gap variation, unit = micron
    Dmzx = -20  # Gap
    Dmzx_Step = -2 * Dmzx / (CN - 1)

    DmN = range(1, CN + 1)
    DmXX = [Dmzx + (n - 1) * Dmzx_Step for n in DmN]

    Conx = [0] * Ar_nb
    Cony = [0] * Ar_nb
    for jj in DmN:
        Surface_2.Thickness = (Dmzx + (jj - 1) * Dmzx_Step) / 1000  # Wafer gap change
        TheSystem.Save()

        pop = TheSystem.Analyses.Get_AnalysisAtIndex(4)
        pop_setting = pop.GetSettings()
        cfg = os.path.join(sampleDir, 'so_lens.cfg')
        pop_setting.SaveTo(cfg)
        pop_setting.LoadFrom(cfg)
        pop_setting.ModifySettings(cfg, 'EXD_FILESIZE', 0.832)  # square array size.
        pop_setting.ModifySettings(cfg, 'EXD_DISPLAYSIZE', 0.832)  # square array size.
        pop.ApplyAndWaitForCompletion()
        results = pop.GetResults()
        header = results.HeaderData
        AAA = results.DataGrids(1).Values.double
        ax, ay = AAA.GetLength(0), AAA.GetLength(1)
        arxn = round(ax / 10)
        aryn = round(ay / 10)
        BBB = [[AAA[i, j] for j in range(aryn, ay - aryn)] for i in range(arxn, ax - arxn)]

        Iy0 = [max(row) for row in BBB]
        Ix0 = [max(col) for col in zip(*BBB)]
        Iy = sorted(Iy0, reverse=True)
        Ix = sorted(Ix0, reverse=True)
        Iymx = sum(Iy[:sqn]) / sqn
        Iymn = sum(Iy[-sqn:]) / sqn
        Ixmx = sum(Ix[:sqn]) / sqn
        Ixmn = sum(Ix[-sqn:]) / sqn
        Conx[jj - 1] = (Ixmx - Ixmn) / (Ixmx + Ixmn)
        Cony[jj - 1] = (Iymx - Iymn) / (Iymx + Iymn)

    ConDiff = [2 * abs(Cony - Conx) / (Conx + Cony) for i in range(Ar_nb)]

    plt.figure(1)
    plt.plot(DmXX, Conx, 'b', DmXX, Cony, 'r')
    plt.legend(Leg1)
    plt.xlabel(xA)
    plt.ylabel(yA)
    plt.title(Ptitle)

    plt.figure(2)
    plt.plot(DmXX, ConDiff)
    plt.xlabel(xA)
    plt.ylabel(yA1)
    plt.title(Ptitle1)

    Surface_0.Thickness = 186.13848
    Surface_2.Thickness = 0
    TheSystem.Save()

    return []

def InitConnection():
    NetHelper = r'C:\Program Files\Zemax OpticStudio\ZOS-API\Libraries\ZOSAPI_NetHelper.dll'

    try:
        clr.AddReference(NetHelper)
    except Exception as e:
        print(f"Failed to add reference to {NetHelper}")
        return None

    import ZOSAPI

    success = ZOSAPI.ZOSAPI_Initializer.Initialize()
    if success == 1:
        LogMessage(f"Found OpticStudio at: {ZOSAPI.ZOSAPI_Initializer.GetZemaxDirectory()}")
    else:
        return None

    clr.AddReference("ZOSAPI_Interfaces")
    clr.AddReference("ZOSAPI")

    TheConnection = ZOSAPI.ZOSAPI_Connection()
    app = TheConnection.CreateNewApplication()
    if app is None:
        HandleError("An unknown connection error occurred!")
    if not app.IsValidLicenseForAPI:
        HandleError("License check failed!")
        return None

    return app

def LogMessage(msg):
    print(msg)

def HandleError(error):
    raise Exception(error)

def CleanupConnection(TheApplication):
    TheApplication.CloseApplication()

# 调用示例
MATLABStandaloneApplication()