使用python-cst联合仿真更新参数并导出数据
目的
使用python-cst联合仿真对一个已有的CST文件进行指定参数更新、仿真，并导出反射相位数据。在这里，这里使用的是CST2021，以及一个反射超表面项目文件，其模型如下图所示。对模型中间的Lumped Element的RLC串联电路（即变容二极管的等效电路）的R、C值进行更新、测试。
一，确定python环境
通过查看帮助文档可以知道所用的CST版本支持的python版本，并创建相对应的python环境。同时，文档下面还说明了加入环境变量的办法。
二， 设置环境变量
方法1：添加系统变量
在环境变量中添加下图所示的系统变量，AMD64文件在自己的CST安装路径中。
方法2：代码添加
在python代码文件中添加如下代码。注意：此代码必须添加到代码最开始，要不然后面导入cst.interface时会报错。
# 调用cst接口
import sys
cst_lib_path = r"F:\cst2021\AMD64\python_cst_libraries"
sys.path.append(cst_lib_path)
三，CST项目创建
为了防止DE的重复建立，造成过多的CST进程；以及指定的CST文件的频繁打开。在这里，会先判断DE以及CST文件是否被打开。即：如果DE已被打开，则遍历DE中的所有项目文件，判断指定的CST文件是否已经打开。若已打开，则直接转到此项目；若未打开，则在最后一个DE中打开项目文件。如果DE未被打开，则直接创建一个新的DE，再在其中打开项目文件。
# 打开CST文件
def opencst(cst_path):
# 判断是否存在仿真结果文件，有的话则将其删除
# result_path = cst_path.split('.')[0]
# if os.path.exists(result_path):
# shutil.rmtree(result_path)
allpids = cst.interface.running_design_environments() # 以列表形式返回运行的cst环境名(PIDs)
DE_is_open = False # 判断是否有打开的DE
cst_is_open = False # 判断指定cst文件是否打开
for pid in allpids:
my_DE = cst.interface.DesignEnvironment.connect(pid) # 连接到指定PID的DE
# print(my_DE)
DE_is_open = True
for project_path in my_DE.list_open_projects(): # 返回已打开项目(cst文件)的路径，如果项目中存在未保存的文件、含中文等特殊字符的文件则会报错
# print(project_path)
if cst_path == project_path: # 如果已打开文件存在，直接使用此cst项目
my_project = my_DE.get_open_project(project_path)
cst_is_open = True
break
if not DE_is_open: # DE不存在，则新建一个DE，否则使用上面已打开的最后一个DE
my_DE = cst.interface.DesignEnvironment()
if not cst_is_open: # cst文件未打开，则在最后一个DE中打开指定cst文件
my_project = my_DE.open_project(cst_path)
return my_DE, my_project
四，定义VBA代码执行
sCommand是以列表形式保存的每一行的vba代码，并使用’\n’进行拼接，之后进行宏命令运行（历史书中不显示，等同于在Macro Editor进行编译）。
line_break = '\n' # 用于VBA代码的拼接
# 执行VBA代码
def execute_vba(cst_project, sCommand):
sCommand = line_break.join(sCommand)
res = cst_project.schematic.execute_vba_code(sCommand, timeout=None) # 执行VBA代码
return res
五，参数更新
在这里，对capacitance和resistance两个变量进行参数更新，更新的值位于par_change中。
# 更新参数
sCommand = ['Sub Main()',
f'StoreParameter("capacitance", {par_change[0]})',
f'StoreParameter("resistance", {par_change[1]})',
'RebuildOnParametricChange(False, True)',
'End Sub']
execute_vba(my_project, sCommand)
六，数据导出
由于遍历结果树时，显示的是SZmax(2),Zmax(2)的dB幅度数据曲线，因此，需要先绘制出反射Phase曲线，之后才能将结果导出为txt文件
# 反射相位数据收集
sCommand = ['Sub Main',
'SelectTreeItem("1D Results\S-Parameters\SZmax(2),Zmax(2)")',
'With Plot1D',
'.PlotView "Phase"',
'.Plot',
'End With',
'With ASCIIExport',
'.Reset',
f'.FileName ("D:/workbook/zhenlie/5G_example/data/phase/{str(par_change[2])}V_phase.txt")',
'.Execute',
'End With',
'End Sub']
execute_vba(my_project, sCommand)
由于我目前还不会使用vba代码进行Delete Results操作或者是使得Result Navigator只显示一条曲线，因此，从第二次结果导出过程开始，导出的txt文件中都会包含之前所有的曲线数据，但会以如图所示的列名来进行分隔。同时，最新的仿真结果一定位于最后一个表格处。
因此，需要对收集到的txt文件进行拆分，以保留最后一个结果。（哈，你说直接保存最后一个表不就得到了所有的数据？我也想啊，可是不知道为啥，曲线序号有时会和仿真序号不匹配。当然，如果你的曲线数据保证不会乱掉，也可以直接在最后一次仿真后进行保存，然后进行表格数据拆分。）
# 对一个txt文件的多个表格进行拆分
def txt_split(txt_path, str_split):
txt_path = txt_path.replace('/', '\\')
all_data = pd.read_csv(txt_path, sep='\t', header=None)
up_index = 0 # 记录表格起始行数下标
# down_index = -1 # 记录表格终止行数下标
for index in range(all_data.__len__()):
if str_split in all_data.iloc[index].__str__():
up_index = index
data = all_data.iloc[up_index:]
save_path = os.path.join(r'./data/phase_pro', txt_path.split('\\')[-1])
data.to_csv(save_path, header=None, index=None)
七，完整代码
# 调用cst接口
import sys
cst_lib_path = r"F:\cst2021\AMD64\python_cst_libraries"
sys.path.append(cst_lib_path)
# 打开CST文件并更新其参数值进行仿真计算
import cst.interface
import cst.results
import os
import shutil
# 打开CST文件
def opencst(cst_path):
# 判断是否存在仿真结果文件，有的话则将其删除，如果文件是打开状态则可能会报错
result_path = cst_path.split('.')[0]
if os.path.exists(result_path):
shutil.rmtree(result_path)
allpids = cst.interface.running_design_environments() # 以列表形式返回运行的cst环境名(PIDs)
DE_is_open = False # 判断是否有打开的DE
cst_is_open = False # 判断指定cst文件是否打开
for pid in allpids:
my_DE = cst.interface.DesignEnvironment.connect(pid) # 连接到指定PID的DE
# print(my_DE)
DE_is_open = True
for project_path in my_DE.list_open_projects(): # 返回打开项目(cst文件)的路径
# print(project_path)
if cst_path == project_path: # 如果已打开文件存在，直接使用此cst项目
my_project = my_DE.get_open_project(project_path)
cst_is_open = True
break
if not DE_is_open: # DE不存在，则新建一个DE，否则使用上面已打开的最后一个DE
my_DE = cst.interface.DesignEnvironment()
if not cst_is_open: # cst文件未打开，则打开指定cst文件
my_project = my_DE.open_project(cst_path)
return my_DE, my_project
line_break = '\n' # 用于VBA代码的拼接
# 执行VBA代码
def execute_vba(cst_project, sCommand):
sCommand = line_break.join(sCommand)
res = cst_project.schematic.execute_vba_code(sCommand, timeout=None) # 执行VBA代码
return res
# my_cst_DE, my_project = opencst(r'D:\workbook\zhenlie\5G_example\CST_file\zhenlie.cst')
# 参数改变(C, R)
par_changes = [[1.284e-12, 2.393, -0.5], # -0.5V
[1.085e-12, 2.292, -1], # -1V
[0.873e-12, 2.139, -2], # -2V
[0.682e-12, 1.927, -4], # -4V
[0.528e-12, 1.664, -8], # -8V
[0.423e-12, 1.394, -15], # -15V
[0.367e-12, 1.205, -23]] # -23V
# par_change = [1.284e-12, 2.393]
for par_change in par_changes:
my_cst_DE, my_project = opencst(r'D:\workbook\zhenlie\5G_example\CST_file\zhenlie.cst')
# 更新参数
sCommand = ['Sub Main()',
f'StoreParameter("capacitance", {par_change[0]})',
f'StoreParameter("resistance", {par_change[1]})',
'RebuildOnParametricChange(False, True)',
'End Sub']
execute_vba(my_project, sCommand)
my_project.modeler.run_solver()
my_project.save(r'D:\workbook\zhenlie\5G_example\CST_file\zhenlie.cst')
# 反射相位数据收集
sCommand = ['Sub Main',
'SelectTreeItem("1D Results\S-Parameters\SZmax(2),Zmax(2)")',
'With Plot1D',
'.PlotView "Phase"',
'.Plot',
'End With',
'With ASCIIExport',
'.Reset',
f'.FileName ("D:/workbook/zhenlie/5G_example/data/phase/{str(par_change[2])}V_phase.txt")',
'.Execute',
'End With',
'End Sub']
execute_vba(my_project, sCommand)
my_project.close() # 关闭cst文件
my_cst_DE.close() # 关闭CST
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
import os
import shutil
# 对一个txt文件的多个表格进行拆分
def txt_split(txt_path, str_split):
txt_path = txt_path.replace('/', '\\')
all_data = pd.read_csv(txt_path, sep='\t', header=None)
up_index = 0 # 记录表格起始行数下标
# down_index = -1 # 记录表格终止行数下标
for index in range(all_data.__len__()):
if str_split in all_data.iloc[index].__str__():
up_index = index
data = all_data.iloc[up_index:]
save_path = os.path.join(r'./data/phase_pro', txt_path.split('\\')[-1])
data.to_csv(save_path, header=None, index=None)
if __name__ == '__main__':
path = './data/phase_pro'
str_split = r'Frequency / GHz'
for file_name in os.listdir(path):
file_path = os.path.join(path, file_name)
# print(file_path)
txt_split(file_path, str_split)
5929
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
