Use spice netlist for chatbot input

This commit is contained in:
Bane Stojkovic
2025-04-05 19:44:02 +02:00
parent c2e8def10e
commit f03a0b1417
6 changed files with 239 additions and 213 deletions
@@ -271,21 +271,21 @@ public abstract class AbstractNetlistAnalyzer {
return res; return res;
} }
protected int find(Node node, List<Group> groups) { public static int find(Node node, List<Group> groups) {
for (int i = 0; i < groups.size(); i++) for (int i = 0; i < groups.size(); i++)
if (groups.get(i).getNodes().contains(node)) if (groups.get(i).getNodes().contains(node))
return i; return i;
return -1; return -1;
} }
protected boolean intersect(List<Group> groups, Node node) { public static boolean intersect(List<Group> groups, Node node) {
for (Group g : groups) for (Group g : groups)
if (g.getNodes().contains(node)) if (g.getNodes().contains(node))
return true; return true;
return false; return false;
} }
protected Node intersect(List<Group> groups, Group group) { public static Node intersect(List<Group> groups, Group group) {
for (Group g : groups) for (Group g : groups)
for (Node node : group.getNodes()) for (Node node : group.getNodes())
if (g.getNodes().contains(node)) if (g.getNodes().contains(node))
@@ -0,0 +1,68 @@
/*
DIY Layout Creator (DIYLC).
Copyright (c) 2009-2018 held jointly by the individual authors.
This file is part of DIYLC.
DIYLC 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.
DIYLC 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 DIYLC. If not, see <http://www.gnu.org/licenses/>.
*/
package org.diylc.netlist;
import java.util.ArrayList;
import java.util.EnumSet;
import java.util.List;
import java.util.Set;
import java.util.stream.Collectors;
import org.diylc.common.INetlistAnalyzer;
import org.diylc.common.NetlistSwitchPreference;
import org.diylc.core.IDIYComponent;
import org.diylc.presenter.ComponentProcessor;
public class SpiceAnalyzer extends AbstractNetlistAnalyzer implements INetlistAnalyzer {
public SpiceAnalyzer() {
}
@Override
public String getName() {
return "Generate Spice Netlist";
}
@Override
public String getShortName() {
return "spice";
}
@Override
public String getIconName() {
return "JarBeanInto";
}
@Override
public String getFontName() {
return "Courier New";
}
@Override
public Set<NetlistSwitchPreference> getSwitchPreference() {
return EnumSet.allOf(NetlistSwitchPreference.class);
}
protected Summary summarize(Netlist netlist) throws TreeException {
return new Summary(netlist, SpiceSumarizer.summarize(netlist, true));
}
}
@@ -0,0 +1,133 @@
package org.diylc.netlist;
import org.diylc.common.ComponentType;
import org.diylc.core.IDIYComponent;
import org.diylc.presenter.ComponentProcessor;
import java.util.ArrayList;
import java.util.List;
import java.util.stream.Collectors;
import static org.diylc.netlist.AbstractNetlistAnalyzer.find;
public class SpiceSumarizer {
public static String summarize(Netlist netlist, boolean useHTML) throws TreeException {
// grab all components that are in the netlist
List<Group> groups = netlist.getSortedGroups();
List<IDIYComponent<?>> allComponents = groups.stream()
.flatMap(x -> x.getNodes().stream()
.map(y -> y.getComponent()))
.distinct()
.collect(Collectors.toList());
int unconnectedIndex = groups.size();
int maxLen = 0;
for (IDIYComponent<?> c : allComponents) {
if (c.getName().length() > maxLen)
maxLen = c.getName().length();
}
List<String> lines = new ArrayList<String>();
for (IDIYComponent<?> c : allComponents) {
StringBuilder sb = new StringBuilder();
// change the prefix to match spice convention if needed
String name = c.getName();
String prefix = null;
if (c instanceof ISpiceMapper)
prefix = ((ISpiceMapper)c).getPrefix();
if (prefix != null && !name.toLowerCase().startsWith(prefix.toLowerCase()))
name = prefix + name;
sb.append(fill(name, (int) (Math.ceil(maxLen / 5.0) * 5), useHTML));
sb.append(" ");
List<Integer> nodeIndices = new ArrayList<Integer>();
// find node indices for each control point
for (int i = 0; i < c.getControlPointCount(); i++) {
// skip non-sticky points
if (c.getControlPointNodeName(i) == null)
continue;
int pointIndex = i;
// remap if needed
if (c instanceof ISpiceMapper)
pointIndex = ((ISpiceMapper)c).mapToSpiceNode(pointIndex);
int nodeIndex = find(new Node(c, pointIndex, 0), groups);
if (nodeIndex < 0)
nodeIndex = unconnectedIndex++;
// 1-based convention
nodeIndex++;
nodeIndices.add(nodeIndex);
}
// output to spice
for (Integer nodeIndex : nodeIndices) {
sb.append(fill(formatSpiceNode(nodeIndex), 5, useHTML));
sb.append(" ");
}
String comment = outputComponentValue(c, sb);
if (c instanceof ISpiceMapper) {
String spiceComment = ((ISpiceMapper)c).getComment();
if (spiceComment != null)
comment += spiceComment;
}
sb.append(" ; ").append(comment);
if (useHTML)
sb.append("<br>");
lines.add(sb.toString());
}
return lines
.stream()
.sorted()
.collect(Collectors.joining("\n"));
}
private static String outputComponentValue(IDIYComponent<?> c, StringBuilder sb) {
ComponentType componentType = ComponentProcessor.getInstance()
.extractComponentTypeFrom((Class<? extends IDIYComponent<?>>) c.getClass());
if (c.getValue() == null || c.getValue().toString().trim().isEmpty()) {
// if (
// Void.class.equals(((ParameterizedType) c.getClass()
// .getGenericSuperclass()).getActualTypeArguments()[0])) {
// return typeName;
// }
String typeName = componentType.getName();
if (typeName.contains(" ")) {
sb.append("\"").append(typeName).append("\"");
} else {
sb.append(typeName);
}
} else {
if (c.getValue().toString().contains(" ")) {
sb.append("\"").append(c.getValue()).append("\"");
} else {
sb.append(c.getValue());
}
}
return componentType.getName() + " [" + componentType.getCategory() + "]";
}
private static String formatSpiceNode(int i) {
return String.format("N%03d" , i);
}
private static String fill(String source, int desiredLength, boolean useHTML) {
String res = source;
for (int i = 0; i < desiredLength - source.length(); i++)
res += useHTML ? "&nbsp;" : " ";
return res;
}
}
@@ -31,10 +31,7 @@ import org.diylc.common.IPlugInPort;
import org.diylc.common.PropertyWrapper; import org.diylc.common.PropertyWrapper;
import org.diylc.core.IDIYComponent; import org.diylc.core.IDIYComponent;
import org.diylc.core.ISwitch; import org.diylc.core.ISwitch;
import org.diylc.netlist.Group; import org.diylc.netlist.*;
import org.diylc.netlist.Netlist;
import org.diylc.netlist.NetlistException;
import org.diylc.netlist.Node;
import org.diylc.plugins.chatbot.model.ChatMessageEntity; import org.diylc.plugins.chatbot.model.ChatMessageEntity;
import org.diylc.plugins.chatbot.model.IChatbotAPI; import org.diylc.plugins.chatbot.model.IChatbotAPI;
import org.diylc.plugins.chatbot.model.SubscriptionEntity; import org.diylc.plugins.chatbot.model.SubscriptionEntity;
@@ -127,7 +124,6 @@ public class ChatbotService {
private static final Set<Class<?>> PROPERTY_TYPES_TO_SKIP = Set.of(Font.class, Color.class); private static final Set<Class<?>> PROPERTY_TYPES_TO_SKIP = Set.of(Font.class, Color.class);
@SuppressWarnings("unchecked")
private String extractNetlists() throws NetlistException { private String extractNetlists() throws NetlistException {
List<Netlist> netlists = plugInPort.extractNetlists(true); List<Netlist> netlists = plugInPort.extractNetlists(true);
@@ -137,26 +133,27 @@ public class ChatbotService {
StringBuilder sb = new StringBuilder(); StringBuilder sb = new StringBuilder();
Set<? extends IDIYComponent<?>> components = netlists.stream() // Set<? extends IDIYComponent<?>> components = netlists.stream()
.flatMap(n -> n.getGroups().stream() // .flatMap(n -> n.getGroups().stream()
.flatMap(g -> g.getNodes().stream(). // .flatMap(g -> g.getNodes().stream().
map(Node::getComponent))) // map(Node::getComponent)))
.collect(Collectors.toSet()); // .collect(Collectors.toSet());
//
// if (!components.isEmpty()) {
// sb.append(
// "Components (each element of the JSON array represents one component with all the relevant properties): ");
// outputComponents(components, sb);
// }
if (!components.isEmpty()) { Set<? extends ISwitch> switches = plugInPort.getCurrentProject().getComponents()
sb.append(
"Components (each element of the JSON array represents one component with all the relevant properties): ");
outputComponents(components, sb);
}
Set<? extends IDIYComponent<?>> switches = plugInPort.getCurrentProject().getComponents()
.stream().filter(x -> x instanceof ISwitch) .stream().filter(x -> x instanceof ISwitch)
.map(x -> (ISwitch)x)
.collect(Collectors.toSet());; .collect(Collectors.toSet());;
if (!switches.isEmpty()) { if (!switches.isEmpty()) {
sb.append( sb.append(
"Switches (each element of the JSON array represents one switch with all the relevant properties): "); "Switches (each row represents one switch, starts with the switch name and lists key properties of a switch):\n\n");
outputComponents(switches, sb); outputSwitches(switches, sb);
} }
for (Netlist netlist : netlists) { for (Netlist netlist : netlists) {
@@ -164,21 +161,30 @@ public class ChatbotService {
if (!netlist.getSwitchSetup().isEmpty()) { if (!netlist.getSwitchSetup().isEmpty()) {
sb.append("Switch configuration: ").append(netlist.getSwitchSetup()).append(". "); sb.append("Switch configuration: ").append(netlist.getSwitchSetup()).append(". ");
} }
sb.append("Each line below represents one JSON array that lists all the nodes that are connected together. Nodes are labeled as (component name).(node name):\n\n"); sb.append("Effective circuit netlist")
for (Group v : netlist.getSortedGroups()) { .append(netlist.getSwitchSetup().isEmpty() ? "" : " in this switch configuration")
// sb.append(v.getSortedNodes()).append("\n"); .append(":\n\n");
List<String> nodeList = v.getSortedNodes().stream().map(x -> x.getComponent().getName() + "." + x.getDisplayName()).toList();
try { try {
String json = MAPPER.writeValueAsString(nodeList); sb.append(SpiceSumarizer.summarize(netlist, false));
sb.append(json).append("\n"); } catch (TreeException e) {
} catch (Exception e) { LOG.error("Error exporting netlist to Spice", e);
LOG.error("Failed to serialize component descriptors to JSON", e);
}
} }
} }
return sb.toString(); return sb.toString();
} }
private static void outputSwitches(Set<? extends ISwitch> switches, StringBuilder sb) {
for (ISwitch sw : switches) {
IDIYComponent<?> component = (IDIYComponent<?>) sw;
ComponentType componentType = ComponentProcessor.getInstance()
.extractComponentTypeFrom((Class<? extends IDIYComponent<?>>) component.getClass());
sb.append(component.getName())
.append(" ").append(component.getValueForDisplay()).append(" ")
.append(componentType.getName())
.append(" [").append(componentType.getCategory()).append("]\n");
}
}
private static void outputComponents(Set<? extends IDIYComponent<?>> components, StringBuilder sb) { private static void outputComponents(Set<? extends IDIYComponent<?>> components, StringBuilder sb) {
List<Map<String, Object>> componentDescriptors = new ArrayList<>(); List<Map<String, Object>> componentDescriptors = new ArrayList<>();
@@ -1,181 +0,0 @@
/*
DIY Layout Creator (DIYLC).
Copyright (c) 2009-2018 held jointly by the individual authors.
This file is part of DIYLC.
DIYLC 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.
DIYLC 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 DIYLC. If not, see <http://www.gnu.org/licenses/>.
*/
package org.diylc.netlist;
import java.lang.reflect.ParameterizedType;
import java.util.ArrayList;
import java.util.EnumSet;
import java.util.List;
import java.util.Set;
import java.util.stream.Collectors;
import org.diylc.common.INetlistAnalyzer;
import org.diylc.common.NetlistSwitchPreference;
import org.diylc.core.IDIYComponent;
import org.diylc.netlist.Group;
import org.diylc.netlist.Netlist;
import org.diylc.netlist.NetlistAnalyzer;
import org.diylc.netlist.Node;
import org.diylc.netlist.Summary;
import org.diylc.netlist.TreeException;
import org.diylc.presenter.ComponentProcessor;
public class SpiceAnalyzer extends AbstractNetlistAnalyzer implements INetlistAnalyzer {
public SpiceAnalyzer() {
}
@Override
public String getName() {
return "Generate Spice Netlist";
}
@Override
public String getShortName() {
return "spice";
}
@Override
public String getIconName() {
return "JarBeanInto";
}
@Override
public String getFontName() {
return "Courier New";
}
@Override
public Set<NetlistSwitchPreference> getSwitchPreference() {
return EnumSet.allOf(NetlistSwitchPreference.class);
}
protected Summary summarize(Netlist netlist) throws TreeException {
// grab all components that are in the netlist
List<Group> groups = netlist.getSortedGroups();
List<IDIYComponent<?>> allComponents = groups.stream()
.flatMap(x -> x.getNodes().stream()
.map(y -> y.getComponent()))
.distinct()
.collect(Collectors.toList());
int unconnectedIndex = groups.size();
int maxLen = 0;
for (IDIYComponent<?> c : allComponents) {
if (c.getName().length() > maxLen)
maxLen = c.getName().length();
}
List<String> lines = new ArrayList<String>();
for (IDIYComponent<?> c : allComponents) {
StringBuilder sb = new StringBuilder();
// change the prefix to match spice convention if needed
String name = c.getName();
String prefix = null;
if (c instanceof ISpiceMapper)
prefix = ((ISpiceMapper)c).getPrefix();
if (prefix != null && !name.toLowerCase().startsWith(prefix.toLowerCase()))
name = prefix + name;
sb.append(fill(name, (int) (Math.ceil(maxLen / 5.0) * 5)));
sb.append(" ");
List<Integer> nodeIndices = new ArrayList<Integer>();
// find node indices for each control point
for (int i = 0; i < c.getControlPointCount(); i++) {
// skip non-sticky points
if (c.getControlPointNodeName(i) == null)
continue;
int pointIndex = i;
// remap if needed
if (c instanceof ISpiceMapper)
pointIndex = ((ISpiceMapper)c).mapToSpiceNode(pointIndex);
int nodeIndex = find(new Node(c, pointIndex, 0), groups);
if (nodeIndex < 0)
nodeIndex = unconnectedIndex++;
// 1-based convention
nodeIndex++;
nodeIndices.add(nodeIndex);
}
// output to spice
for (Integer nodeIndex : nodeIndices) {
sb.append(fill(formatSpiceNode(nodeIndex), 5));
sb.append(" ");
}
sb.append(formatValue(c));
if (c instanceof ISpiceMapper) {
String comment = ((ISpiceMapper)c).getComment();
if (comment != null)
sb.append(" ; ").append(comment);
}
sb.append("<br>");
lines.add(sb.toString());
}
return new Summary(netlist, lines
.stream()
.sorted()
.collect(Collectors.joining("\n")));
}
private static Object formatValue(IDIYComponent<?> c) {
if (c.getValue() == null || c.getValue().toString().trim().isEmpty()) {
String typeName = ComponentProcessor.getInstance()
.extractComponentTypeFrom((Class<? extends IDIYComponent<?>>) c.getClass()).getName();
// if (
// Void.class.equals(((ParameterizedType) c.getClass()
// .getGenericSuperclass()).getActualTypeArguments()[0])) {
// return typeName;
// }
if (typeName.contains(" ")) {
return "\"" + typeName + "\"";
}
return typeName;
}
if (c.getValue().toString().contains(" ")) {
return "\"" + c.getValue() + "\"";
}
return c.getValue();
}
private static String formatSpiceNode(int i) {
return String.format("N%03d" , i);
}
private String fill(String source, int desiredLength) {
String res = source;
for (int i = 0; i < desiredLength - source.length(); i++)
res += "&nbsp;";
return res;
}
}