# This code is part of KQCircuits
# Copyright (C) 2021 IQM Finland Oy
#
# This program is free software: you can redistribute it and/or modify it under the terms of the GNU General Public
# License as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later
# version.
#
# This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied
# warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License along with this program. If not, see
# https://www.gnu.org/licenses/gpl-3.0.html.
#
# The software distribution should follow IQM trademark policy for open-source software
# (meetiqm.com/developers/osstmpolicy). IQM welcomes contributions to the code. Please see our contribution agreements
# for individuals (meetiqm.com/developers/clas/individual) and organizations (meetiqm.com/developers/clas/organization).
import os
import stat
import json
import logging
from typing import Optional, Union, Sequence, Tuple
from pathlib import Path
from kqcircuits.simulations.export.ansys.ansys_solution import AnsysSolution
from kqcircuits.simulations.export.simulation_export import (
copy_content_into_directory,
get_post_process_command_lines,
get_combined_parameters,
)
from kqcircuits.util.export_helper import write_commit_reference_file
from kqcircuits.util.geometry_json_encoder import GeometryJsonEncoder
from kqcircuits.simulations.export.util import export_layers
from kqcircuits.defaults import ANSYS_EXECUTABLE, ANSYS_SCRIPT_PATHS
from kqcircuits.simulations.simulation import Simulation
from kqcircuits.simulations.post_process import PostProcess
[docs]
def export_ansys_json(simulation: Simulation, solution: AnsysSolution, path: Path):
"""
Export Ansys simulation into json and gds files.
Arguments:
simulation: The simulation to be exported.
solution: The solution to be exported.
path: Location where to write json and gds files.
Returns:
Path to exported json file.
"""
if simulation is None or not isinstance(simulation, Simulation):
raise ValueError("Cannot export without simulation")
# write .gds file
gds_file = simulation.name + ".gds"
gds_file_path = str(path.joinpath(gds_file))
if not Path(gds_file_path).exists():
export_layers(
gds_file_path, simulation.layout, [simulation.cell], output_format="GDS2", layers=simulation.get_layers()
)
full_name = simulation.name + solution.name
# collect data for .json file
json_data = {
"name": full_name,
**solution.get_solution_data(),
**simulation.get_simulation_data(),
"gds_file": gds_file,
"parameters": get_combined_parameters(simulation, solution),
}
# write .json file
json_file_path = str(path.joinpath(full_name + ".json"))
with open(json_file_path, "w") as fp:
json.dump(json_data, fp, cls=GeometryJsonEncoder, indent=4)
return json_file_path
[docs]
def export_ansys_bat(
json_filenames,
path: Path,
file_prefix="simulation",
exit_after_run=False,
execution_script="scripts/import_and_simulate.py",
post_process=None,
use_rel_path=True,
):
"""
Create a batch file for running one or more already exported simulations.
Arguments:
json_filenames: List of paths to json files to be included into the batch.
path: Location where to write the bat file.
file_prefix: Name of the batch file to be created.
exit_after_run: Defines if the Ansys Electronics Desktop is automatically closed after running the script.
execution_script: The script file to be executed.
post_process: List of PostProcess objects, a single PostProcess object, or None to be executed after simulations
use_rel_path: Determines if to use relative paths.
Returns:
Path to exported bat file.
"""
run_cmd = "RunScriptAndExit" if exit_after_run else "RunScript"
bat_filename = str(path.joinpath(file_prefix + ".bat"))
with open(bat_filename, "w") as file:
file.write(
"@echo off\n"
r'powershell -Command "Get-Process | Where-Object {$_.MainWindowTitle -like \"Run Simulations*\"} '
'| Select -ExpandProperty Id | Export-Clixml -path blocking_pids.xml"\n'
"title Run Simulations\n"
'powershell -Command "$sim_pids = Import-Clixml -Path blocking_pids.xml; if ($sim_pids) '
r"{ echo \"Waiting for $sim_pids\"; Wait-Process $sim_pids -ErrorAction SilentlyContinue }; "
'Remove-Item blocking_pids.xml"\n'
)
# Commands for each simulation
for i, json_filename in enumerate(json_filenames):
printing = "echo Simulation {}/{} - {}\n".format(
i + 1, len(json_filenames), str(Path(json_filename).relative_to(path))
)
file.write(printing)
command = '"{}" -scriptargs "{}" -{} "{}"\n'.format(
ANSYS_EXECUTABLE,
str(Path(json_filename).relative_to(path) if use_rel_path else json_filename),
run_cmd,
str(execution_script),
)
file.write(command)
file.write(get_post_process_command_lines(post_process, path, json_filenames))
# Make the bat file executable in linux
os.chmod(bat_filename, os.stat(bat_filename).st_mode | stat.S_IEXEC)
return bat_filename
[docs]
def export_ansys(
simulations: Sequence[Union[Simulation, Tuple[Simulation, AnsysSolution]]],
path: Path,
script_folder: str = "scripts",
file_prefix: str = "simulation",
exit_after_run: bool = False,
import_script: str = "import_and_simulate.py",
post_process: Optional[Union[PostProcess, Sequence[PostProcess]]] = None,
use_rel_path: bool = True,
skip_errors: bool = False,
**solution_params,
) -> Path:
"""
Export Ansys simulations by writing necessary scripts and json, gds, and bat files.
Arguments:
simulations: List of Simulation objects or tuples containing Simulation and Solution objects.
path: Location where to write export files.
script_folder: Path to the Ansys-scripts folder.
file_prefix: Name of the batch file to be created.
exit_after_run: Defines if the Ansys Electronics Desktop is automatically closed after running the script.
import_script: Name of import script file.
post_process: List of PostProcess objects, a single PostProcess object, or None to be executed after simulations
use_rel_path: Determines if to use relative paths.
skip_errors: Skip simulations that cause errors. Default is False.
.. warning::
**Use this carefully**, some of your simulations might not make sense physically and
you might end up wasting time on bad simulations.
solution_params: AnsysSolution parameters if simulations is a list of Simulation objects.
Returns:
Path to exported bat file.
"""
write_commit_reference_file(path)
copy_content_into_directory(ANSYS_SCRIPT_PATHS, path, script_folder)
json_filenames = []
for sim_sol in simulations:
if isinstance(sim_sol, Sequence):
simulation, solution = sim_sol
else:
simulation = sim_sol
solution = AnsysSolution(**solution_params)
try:
json_filenames.append(export_ansys_json(simulation, solution, path))
except (IndexError, ValueError, Exception) as e: # pylint: disable=broad-except
if skip_errors:
logging.warning(
f"Simulation {simulation.name} skipped due to {e.args}. "
"Some of your other simulations might not make sense geometrically. "
"Disable `skip_errors` to see the full traceback."
)
else:
raise UserWarning(
"Generating simulation failed. You can discard the errors using `skip_errors` in `export_ansys`. "
"Moreover, `skip_errors` enables visual inspection of failed and successful simulation "
"geometry files."
) from e
return export_ansys_bat(
json_filenames,
path,
file_prefix=file_prefix,
exit_after_run=exit_after_run,
execution_script=Path(script_folder).joinpath(import_script),
post_process=post_process,
use_rel_path=use_rel_path,
)