package dongfang.mkt.main;
import java.io.FileInputStream;
import java.io.FileWriter;
import java.io.IOException;
import java.io.InputStream;
import javax.xml.parsers.DocumentBuilder;
import javax.xml.parsers.DocumentBuilderFactory;
import javax.xml.parsers.ParserConfigurationException;
import javax.xml.xpath.XPath;
import javax.xml.xpath.XPathConstants;
import javax.xml.xpath.XPathExpressionException;
import javax.xml.xpath.XPathFactory;
import org.w3c.dom.Document;
import org.w3c.dom.Element;
import org.w3c.dom.NodeList;
import org.xml.sax.SAXException;
import dongfang.mkt.comm.FrameQueue;
import dongfang.mkt.comm.MKConnection;
import dongfang.mkt.comm.serial.RXTXSerialPort;
import dongfang.mkt.comm.tcp.MKTCPConnection;
import dongfang.mkt.configuration.ConfigSet;
import dongfang.mkt.frames.UniversalReadParamSetRequestFrame;
import dongfang.mkt.frames.UniversalReadParamSetResponseFrame;
import dongfang.mkt.frames.UniversalWriteParamSetRequestFrame;
import dongfang.mkt.frames.UniversalWriteParamSetResponseFrame;
public class UniversalConfigurator
{
private static void configure
(String portIdentifier, UniversalWriteParamSetRequestFrame frame,
int parameterSetNumber
) throws IOException {
MKConnection port =
new RXTXSerialPort
();
port.
init(portIdentifier
);
FrameQueue q =
new FrameQueue
(port
);
frame.
setConfigurationSetNumber(parameterSetNumber
);
q.
sendRequest(frame
);
UniversalWriteParamSetResponseFrame r =
(UniversalWriteParamSetResponseFrame
) q
.
getResponseFor(frame,
5000);
if (r ==
null) {
System.
err.
println("ERROR. Timeout waiting for response.");
} else if (!r.
wasAccepted()) {
System.
err
.
println("ERROR. Parameter set not accepted. Check version against MK firmware EEPROM configuration version and checkparameter set number");
} else {
System.
out.
println("Saved parameter set #"
+ r.
getParameterSetNumber() +
".");
}
q.
kill();
}
private static void writeConfiguration
(String portIdentifier,
String s_parameterSetNumber,
String fileName
) throws IOException {
System.
out.
println("Writing parameter set #" + s_parameterSetNumber +
" from file: " + fileName
);
int parameterSetNumber =
Integer.
parseInt(s_parameterSetNumber
);
InputStream inputStream =
new FileInputStream(fileName
);
UniversalWriteParamSetRequestFrame frame = parseXMLParameterSet
(inputStream
);
configure
(portIdentifier, frame, parameterSetNumber
);
}
private static ConfigSet readConfiguration
(String portIdentifier,
int parameterSetNumber
) throws IOException {
MKConnection port =
new RXTXSerialPort
();
port.
init(portIdentifier
);
FrameQueue q =
new FrameQueue
(port
);
ConfigSet cs =
null;
UniversalReadParamSetRequestFrame frame =
new UniversalReadParamSetRequestFrame
();
frame.
setConfigurationSetNumber(parameterSetNumber
);
q.
sendRequest(frame
);
UniversalReadParamSetResponseFrame r =
(UniversalReadParamSetResponseFrame
) q.
getResponseFor(frame,
5000);
if (r ==
null) {
System.
err.
println("ERROR. Timeout waiting for response.");
} else {
int paramSetVersion = r.
getConfigurationVersion();
int[] data = r.
getData();
cs = ConfigSet.
parseXMLConfigSet(paramSetVersion
);
cs.
setData(data
);
System.
out.
println(cs.
toXML());
}
q.
kill();
return cs
;
}
/*
private static ConfigSet simpleReadConfiguration(int parameterSetNumber)
throws IOException {
MKCommPort port = new RXTXSerialPort();
port.init(null);
FrameQueue q = new FrameQueue(port);
ConfigSet cs = null;
UniversalReadParamSetRequestFrame frame = new UniversalReadParamSetRequestFrame();
frame.setConfigurationSetNumber(parameterSetNumber);
q.sendRequest(frame);
UniversalReadParamSetResponseFrame r = (UniversalReadParamSetResponseFrame) q
.getResponseFor(frame, 3000);
if (r == null) {
System.err.println("ERROR. Timeout waiting for response.");
} else {
int paramSetVersion = r.getConfigurationVersion();
int[] data = r.getData();
for (int i=0; i<data.length; i++)
System.out.println(data[i]);
}
q.kill();
return cs;
}
*/
private static void readConfiguration
(String portIdentifier,
String s_parameterSetNumber,
String fileName
) throws IOException {
int parameterSetNumber =
Integer.
parseInt(s_parameterSetNumber
);
ConfigSet cs = readConfiguration
(portIdentifier, parameterSetNumber
);
if (cs
!=
null) {
FileWriter fw =
new FileWriter(fileName
);
fw.
write(cs.
toXML());
fw.
close();
}
}
private static int substituteVariables
(String s_value
) {
int numberOfHighestVar =
255;
int numberOfLowestVar = numberOfHighestVar
- ConfigSet.
NUMBER_OF_VARIABLES;
for (int i =
0; i
< ConfigSet.
NUMBER_OF_VARIABLES; i++
) {
String varName =
"var" + i
;
if (varName.
equals(s_value
)) {
System.
out.
println("Substing: " + s_value +
"-->"
+
(numberOfLowestVar + i
));
return numberOfLowestVar + i
;
}
}
return -
1;
}
private static UniversalWriteParamSetRequestFrame parseXMLParameterSet
(
InputStream input
) throws IOException {
DocumentBuilderFactory saxfac =
DocumentBuilderFactory.
newInstance();
saxfac.
setValidating(false);
try {
DocumentBuilder bldr = saxfac.
newDocumentBuilder();
Document doc = bldr.
parse(input
);
XPath xpath =
XPathFactory.
newInstance().
newXPath();
String s_eepromVersion = xpath.
evaluate(
"/parameterset/@eepromVersion", doc
);
String ss_length = xpath.
evaluate("/parameterset/@length", doc
);
int s_length =
0;
try {
s_length =
Integer.
parseInt(ss_length
);
} catch (NumberFormatException ex
) {
// ignore.
}
int eepromVersion = -
1;
try {
eepromVersion =
Integer.
parseInt(s_eepromVersion
);
} catch (NumberFormatException ex
) {
System.
err
.
println("The parameterset element must have an 'eepromVersion' attribute with a numerical value (eg.<parameterset eepromVersion='74'>)");
System.
exit(-
1);
}
// if (s_length != null)
// try {
// length = Integer.parseInt(s_length);
// } catch (NumberFormatException ex) {
// System.err.println("The length attribute of the parameterset must have a numerical value");
// System.exit(-1);
// }
NodeList parameterNodes =
(NodeList) xpath.
evaluate(
"/parameterset/parameter", doc,
XPathConstants.
NODESET);
int[] parameters =
new int[parameterNodes.
getLength()];
if (s_length
>=
0 && s_length
!= parameterNodes.
getLength()) {
System.
err.
println("The number of parameters ("+parameterNodes.
getLength()+
") does not match the number in the length attribute of the parameterset element ("+s_length+
").");
// System.exit(-1);
}
for (int i =
0; i
< parameterNodes.
getLength(); i++
) {
Element e =
(Element) parameterNodes.
item(i
);
int value =
0;
if (e.
hasAttribute("value")) {
String s_value = e.
getAttribute("value");
value = substituteVariables
(s_value
);
if (value
< 0)
try {
value =
Integer.
parseInt(s_value
);
} catch (NumberFormatException ex
) {
throw new NumberFormatException(
"Value is not a number and not a variable name.");
}
} else {
NodeList bitNodes =
(NodeList) xpath.
evaluate("bit", e,
XPathConstants.
NODESET);
for (int j =
0; j
< 8; j++
) {
Element bitElement =
(Element) bitNodes.
item(j
);
if (bitElement
!=
null) {
String s_bitValue = bitElement
.
getAttribute("value");
if ("1".
equals(s_bitValue
))
value |=
(1 << j
);
else if (!"0".
equals(s_bitValue
))
throw new NumberFormatException(
"Bit value was not 0 or 1.");
}
}
}
parameters
[i
] = value
;
}
input.
close();
return new UniversalWriteParamSetRequestFrame
(eepromVersion,
parameters
);
} catch (ParserConfigurationException ex
) {
// Should never happen.
throw new RuntimeException(ex
);
} catch (SAXException ex
) {
System.
err.
println(ex
);
System.
err
.
println("There is something screwed with your XML document. It is not well-formed and won't parse.");
throw new RuntimeException("Parse error.");
} catch (XPathExpressionException ex
) {
// Should never happen.
throw new RuntimeException(ex
);
}
}
static void help
() {
System.
err
.
println("Usage: UniversalConfigurator r [parameter set number to read from] [filename to write to]");
System.
err
.
println("Usage: UniversalConfigurator w [parameter set number to write to] [filename to read from]");
System.
exit(-
1);
}
public static void main
(String[] args
) throws IOException {
if (!"r".
equals(args
[0]) && !"w".
equals(args
[0])) help
();
if ("w".
equals(args
[0]) && (args.
length!=
3 && args.
length!=
4)) help
();
if ("r".
equals(args
[0]) && (args.
length!=
3 && args.
length!=
4)) help
();
String portIdentifier =
null;
if ("r".
equals(args
[0])) {
readConfiguration
(portIdentifier,
Integer.
parseInt(args
[1]));
} else {
writeConfiguration
(portIdentifier, args
[1], args
[2]);
}
System.
exit(0);
}
}