Compare commits
5 Commits
Author | SHA1 | Date | |
---|---|---|---|
83051e10c3 | |||
08d76d7953 | |||
1f02001502 | |||
2f2678546d | |||
285fab7587 |
2
Makefile
2
Makefile
@ -11,7 +11,7 @@ REQUIRED+=asynMotor
|
|||||||
REQUIRED+=sinqMotor
|
REQUIRED+=sinqMotor
|
||||||
|
|
||||||
# Specify the version of sinqMotor we want to build against
|
# Specify the version of sinqMotor we want to build against
|
||||||
sinqMotor_VERSION=0.4.0
|
sinqMotor_VERSION=0.6.3
|
||||||
|
|
||||||
# These headers allow to depend on this library for derived drivers.
|
# These headers allow to depend on this library for derived drivers.
|
||||||
HEADERS += src/pmacv3Axis.h
|
HEADERS += src/pmacv3Axis.h
|
||||||
|
@ -8,6 +8,11 @@ This is a driver for the pmacV3 motion controller with the SINQ communication pr
|
|||||||
|
|
||||||
This driver is a standard sinqMotor-derived driver and does not need any specific configuration. For the general configuration, please see https://git.psi.ch/sinq-epics-modules/sinqmotor/-/blob/main/README.md.
|
This driver is a standard sinqMotor-derived driver and does not need any specific configuration. For the general configuration, please see https://git.psi.ch/sinq-epics-modules/sinqmotor/-/blob/main/README.md.
|
||||||
|
|
||||||
|
The folder "utils" contains utility scripts for working with pmac motor controllers. To read their manual, run the scripts without any arguments.
|
||||||
|
- writeRead.py: Allows sending commands to and receiving commands from a pmac controller over an ethernet connection.
|
||||||
|
- analyzeTcpDump.py: Parse the TCP communication between an IOC and a MCU and format it into a dictionary. "demo.py" shows how this data can be easily visualized for analysis.
|
||||||
|
|
||||||
|
|
||||||
## Developer guide
|
## Developer guide
|
||||||
|
|
||||||
### Usage in IOC shell
|
### Usage in IOC shell
|
||||||
|
15
db/pmacv3.db
15
db/pmacv3.db
@ -1,18 +1,7 @@
|
|||||||
# Read out the encoder type in human-readable form. The output numbers can be
|
|
||||||
# interpreted as ASCII.
|
|
||||||
# This record is coupled to the parameter library via encoderType -> ENCODER_TYPE.
|
|
||||||
record(waveform, "$(INSTR)$(M):Encoder_Type") {
|
|
||||||
field(DTYP, "asynOctetRead")
|
|
||||||
field(INP, "@asyn($(CONTROLLER),$(AXIS),1) ENCODER_TYPE")
|
|
||||||
field(FTVL, "CHAR")
|
|
||||||
field(NELM, "80")
|
|
||||||
field(SCAN, "I/O Intr")
|
|
||||||
}
|
|
||||||
|
|
||||||
# Trigger a rereading of the encoder. This action is sometimes necessary for
|
# Trigger a rereading of the encoder. This action is sometimes necessary for
|
||||||
# absolute encoders after enabling them. For incremental encoders, this is a no-op.
|
# absolute encoders after enabling them. For incremental encoders, this is a no-op.
|
||||||
# This record is coupled to the parameter library via rereadEncoderPosition_ -> REREAD_ENCODER_POSITION.
|
# This record is coupled to the parameter library via rereadEncoderPosition_ -> REREAD_ENCODER_POSITION.
|
||||||
record(longout, "$(INSTR)$(M):Reread_Encoder") {
|
record(longout, "$(INSTR)$(M):RereadEncoder") {
|
||||||
field(DTYP, "asynInt32")
|
field(DTYP, "asynInt32")
|
||||||
field(OUT, "@asyn($(CONTROLLER),$(AXIS),1) REREAD_ENCODER_POSITION")
|
field(OUT, "@asyn($(CONTROLLER),$(AXIS),1) REREAD_ENCODER_POSITION")
|
||||||
field(PINI, "NO")
|
field(PINI, "NO")
|
||||||
@ -23,7 +12,7 @@ record(longout, "$(INSTR)$(M):Reread_Encoder") {
|
|||||||
# once at IOC startup during atFirstPoll. However, it can be triggered manually
|
# once at IOC startup during atFirstPoll. However, it can be triggered manually
|
||||||
# by setting this record value to 1.
|
# by setting this record value to 1.
|
||||||
# This record is coupled to the parameter library via readConfig_ -> READ_CONFIG.
|
# This record is coupled to the parameter library via readConfig_ -> READ_CONFIG.
|
||||||
record(longout, "$(INSTR)$(M):Read_Config") {
|
record(longout, "$(INSTR)$(M):ReadConfig") {
|
||||||
field(DTYP, "asynInt32")
|
field(DTYP, "asynInt32")
|
||||||
field(OUT, "@asyn($(CONTROLLER),$(AXIS),1) READ_CONFIG")
|
field(OUT, "@asyn($(CONTROLLER),$(AXIS),1) READ_CONFIG")
|
||||||
field(PINI, "NO")
|
field(PINI, "NO")
|
||||||
|
@ -50,7 +50,8 @@ pmacv3Axis::pmacv3Axis(pmacv3Controller *pC, int axisNo)
|
|||||||
exit(-1);
|
exit(-1);
|
||||||
}
|
}
|
||||||
|
|
||||||
status = pC_->setDoubleParam(axisNo_, pC_->motorPosition_, 0.0);
|
// pmacv3 motors can always be disabled
|
||||||
|
status = pC_->setIntegerParam(axisNo_, pC_->motorCanDisable_, 1);
|
||||||
if (status != asynSuccess) {
|
if (status != asynSuccess) {
|
||||||
asynPrint(
|
asynPrint(
|
||||||
pC_->pasynUserSelf, ASYN_TRACE_ERROR,
|
pC_->pasynUserSelf, ASYN_TRACE_ERROR,
|
||||||
@ -73,12 +74,7 @@ pmacv3Axis::~pmacv3Axis(void) {
|
|||||||
/**
|
/**
|
||||||
Read the configuration at the first poll
|
Read the configuration at the first poll
|
||||||
*/
|
*/
|
||||||
asynStatus pmacv3Axis::atFirstPoll() { return readConfig(); }
|
asynStatus pmacv3Axis::atFirstPoll() {
|
||||||
|
|
||||||
/*
|
|
||||||
Read the configuration from the motor control unit and the parameter library.
|
|
||||||
*/
|
|
||||||
asynStatus pmacv3Axis::readConfig() {
|
|
||||||
|
|
||||||
// Local variable declaration
|
// Local variable declaration
|
||||||
asynStatus status = asynSuccess;
|
asynStatus status = asynSuccess;
|
||||||
@ -102,7 +98,10 @@ asynStatus pmacv3Axis::readConfig() {
|
|||||||
__PRETTY_FUNCTION__, __LINE__);
|
__PRETTY_FUNCTION__, __LINE__);
|
||||||
}
|
}
|
||||||
|
|
||||||
// Software limits and current position
|
/*
|
||||||
|
Read out the axis status, the current position, current and maximum speed,
|
||||||
|
acceleration and the air cushion delay.
|
||||||
|
*/
|
||||||
snprintf(command, sizeof(command),
|
snprintf(command, sizeof(command),
|
||||||
"P%2.2d00 Q%2.2d10 Q%2.2d03 Q%2.2d04 Q%2.2d06 P%2.2d22", axisNo_,
|
"P%2.2d00 Q%2.2d10 Q%2.2d03 Q%2.2d04 Q%2.2d06 P%2.2d22", axisNo_,
|
||||||
axisNo_, axisNo_, axisNo_, axisNo_, axisNo_);
|
axisNo_, axisNo_, axisNo_, axisNo_, axisNo_);
|
||||||
@ -241,7 +240,7 @@ asynStatus pmacv3Axis::doPoll(bool *moving) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Transform from EPICS to motor coordinates (see comment in
|
// Transform from EPICS to motor coordinates (see comment in
|
||||||
// pmacv3Axis::readConfig())
|
// pmacv3Axis::atFirstPoll)
|
||||||
previousPosition = previousPosition * motorRecResolution;
|
previousPosition = previousPosition * motorRecResolution;
|
||||||
|
|
||||||
// Query the axis status
|
// Query the axis status
|
||||||
@ -264,13 +263,13 @@ asynStatus pmacv3Axis::doPoll(bool *moving) {
|
|||||||
The axis limits are set as: ({[]})
|
The axis limits are set as: ({[]})
|
||||||
where [] are the positive and negative limits set in EPICS/NICOS, {} are the
|
where [] are the positive and negative limits set in EPICS/NICOS, {} are the
|
||||||
software limits set on the MCU and () are the hardware limit switches. In
|
software limits set on the MCU and () are the hardware limit switches. In
|
||||||
other words, the EPICS/NICOS limits must be stricter than the software
|
other words, the EPICS/NICOS limits should be stricter than the software
|
||||||
limits on the MCU which in turn should be stricter than the hardware limit
|
limits on the MCU which in turn should be stricter than the hardware limit
|
||||||
switches. For example, if the hardware limit switches are at [-10, 10], the
|
switches. For example, if the hardware limit switches are at [-10, 10], the
|
||||||
software limits could be at [-9, 9] and the EPICS / NICOS limits could be at
|
software limits could be at [-9, 9] and the EPICS / NICOS limits could be at
|
||||||
[-8, 8]. Therefore, we cannot use the software limits read from the MCU
|
[-8, 8]. Therefore, we cannot use the software limits read from the MCU
|
||||||
directly, but need to shrink them a bit. In this case, we're shrinking them
|
directly, but need to shrink them a bit. In this case, we're shrinking them
|
||||||
by 0.1 mm or 0.1 degree (depending on the axis type) on both sides.
|
by limitsOffset on both sides.
|
||||||
*/
|
*/
|
||||||
pl_status =
|
pl_status =
|
||||||
pC_->getDoubleParam(axisNo_, pC_->motorLimitsOffset_, &limitsOffset);
|
pC_->getDoubleParam(axisNo_, pC_->motorLimitsOffset_, &limitsOffset);
|
||||||
@ -284,6 +283,14 @@ asynStatus pmacv3Axis::doPoll(bool *moving) {
|
|||||||
// Store the axis status
|
// Store the axis status
|
||||||
axisStatus_ = axStatus;
|
axisStatus_ = axStatus;
|
||||||
|
|
||||||
|
// Update the enablement PV
|
||||||
|
pl_status = setIntegerParam(pC_->motorEnableRBV_,
|
||||||
|
(axStatus != -3 && axStatus != -5));
|
||||||
|
if (pl_status != asynSuccess) {
|
||||||
|
return pC_->paramLibAccessFailed(pl_status, "motorEnableRBV_",
|
||||||
|
__PRETTY_FUNCTION__, __LINE__);
|
||||||
|
}
|
||||||
|
|
||||||
// Intepret the status
|
// Intepret the status
|
||||||
switch (axStatus) {
|
switch (axStatus) {
|
||||||
case -6:
|
case -6:
|
||||||
@ -514,7 +521,7 @@ asynStatus pmacv3Axis::doPoll(bool *moving) {
|
|||||||
|
|
||||||
snprintf(command, sizeof(command),
|
snprintf(command, sizeof(command),
|
||||||
"Maximum allowed following error exceeded (P%2.2d01 = %d). "
|
"Maximum allowed following error exceeded (P%2.2d01 = %d). "
|
||||||
"Check if movement range is blocked."
|
"Check if movement range is blocked. "
|
||||||
"Otherwise please call the support.",
|
"Otherwise please call the support.",
|
||||||
axisNo_, error);
|
axisNo_, error);
|
||||||
pl_status = setStringParam(pC_->motorMessageText_, command);
|
pl_status = setStringParam(pC_->motorMessageText_, command);
|
||||||
@ -608,13 +615,6 @@ asynStatus pmacv3Axis::doPoll(bool *moving) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
pl_status = setIntegerParam(pC_->motorEnableRBV_,
|
|
||||||
(axStatus != -3 && axStatus != -5));
|
|
||||||
if (pl_status != asynSuccess) {
|
|
||||||
return pC_->paramLibAccessFailed(pl_status, "motorEnableRBV_",
|
|
||||||
__PRETTY_FUNCTION__, __LINE__);
|
|
||||||
}
|
|
||||||
|
|
||||||
pl_status = setIntegerParam(pC_->motorStatusMoving_, *moving);
|
pl_status = setIntegerParam(pC_->motorStatusMoving_, *moving);
|
||||||
if (pl_status != asynSuccess) {
|
if (pl_status != asynSuccess) {
|
||||||
return pC_->paramLibAccessFailed(pl_status, "motorStatusMoving_",
|
return pC_->paramLibAccessFailed(pl_status, "motorStatusMoving_",
|
||||||
@ -648,7 +648,7 @@ asynStatus pmacv3Axis::doPoll(bool *moving) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Transform from motor to EPICS coordinates (see comment in
|
// Transform from motor to EPICS coordinates (see comment in
|
||||||
// pmacv3Axis::readConfig())
|
// pmacv3Axis::atFirstPoll())
|
||||||
currentPosition = currentPosition / motorRecResolution;
|
currentPosition = currentPosition / motorRecResolution;
|
||||||
|
|
||||||
pl_status = setDoubleParam(pC_->motorPosition_, currentPosition);
|
pl_status = setDoubleParam(pC_->motorPosition_, currentPosition);
|
||||||
@ -950,7 +950,11 @@ asynStatus pmacv3Axis::rereadEncoder() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// Abort if the axis is incremental
|
// Abort if the axis is incremental
|
||||||
if (strcmp(encoderType, IncrementalEncoder) == 1) {
|
if (strcmp(encoderType, IncrementalEncoder) == 0) {
|
||||||
|
asynPrint(pC_->pasynUserSelf, ASYN_TRACE_FLOW,
|
||||||
|
"%s => line %d:\nEncoder of axis %d is not reread because it "
|
||||||
|
"is incremental.\n",
|
||||||
|
__PRETTY_FUNCTION__, __LINE__, axisNo_);
|
||||||
return asynSuccess;
|
return asynSuccess;
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -1123,8 +1127,3 @@ asynStatus pmacv3Axis::enable(bool on) {
|
|||||||
}
|
}
|
||||||
return asynError;
|
return asynError;
|
||||||
}
|
}
|
||||||
|
|
||||||
asynStatus pmacv3Axis::isEnabled(bool *on) {
|
|
||||||
*on = (axisStatus_ != -3 && axisStatus_ != -5);
|
|
||||||
return asynSuccess;
|
|
||||||
}
|
|
||||||
|
@ -88,15 +88,6 @@ class pmacv3Axis : public sinqAxis {
|
|||||||
*/
|
*/
|
||||||
asynStatus enable(bool on);
|
asynStatus enable(bool on);
|
||||||
|
|
||||||
/**
|
|
||||||
* @brief EThis function sets "on" to true, if the motor is enabled, and to
|
|
||||||
* false otherwise
|
|
||||||
*
|
|
||||||
* @param on
|
|
||||||
* @return asynStatus
|
|
||||||
*/
|
|
||||||
asynStatus isEnabled(bool *on);
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* @brief Read the encoder type (incremental or absolute) for this axis from
|
* @brief Read the encoder type (incremental or absolute) for this axis from
|
||||||
* the MCU and store the information in the PV ENCODER_TYPE.
|
* the MCU and store the information in the PV ENCODER_TYPE.
|
||||||
@ -115,7 +106,6 @@ class pmacv3Axis : public sinqAxis {
|
|||||||
protected:
|
protected:
|
||||||
pmacv3Controller *pC_;
|
pmacv3Controller *pC_;
|
||||||
|
|
||||||
asynStatus readConfig();
|
|
||||||
bool initial_poll_;
|
bool initial_poll_;
|
||||||
bool waitForHandshake_;
|
bool waitForHandshake_;
|
||||||
time_t timeAtHandshake_;
|
time_t timeAtHandshake_;
|
||||||
|
@ -1,6 +1,3 @@
|
|||||||
// Needed to use strcpy_s from string.h
|
|
||||||
#define __STDC_WANT_LIB_EXT1__ 1
|
|
||||||
|
|
||||||
#include "pmacv3Controller.h"
|
#include "pmacv3Controller.h"
|
||||||
#include "asynMotorController.h"
|
#include "asynMotorController.h"
|
||||||
#include "asynOctetSyncIO.h"
|
#include "asynOctetSyncIO.h"
|
||||||
@ -14,21 +11,21 @@
|
|||||||
#include <unistd.h>
|
#include <unistd.h>
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* @brief Copy src into dst and replace all carriage returns with spaces
|
* @brief Copy src into dst and replace all carriage returns with spaces. This
|
||||||
|
* allows to print *dst with asynPrint.
|
||||||
|
*
|
||||||
*
|
*
|
||||||
* @param dst Buffer for the modified string
|
* @param dst Buffer for the modified string
|
||||||
* @param src Original string
|
* @param src Original string
|
||||||
*/
|
*/
|
||||||
void adjustResponseForPrint(char *dst, const char *src) {
|
void adjustResponseForPrint(char *dst, const char *src, size_t buf_length) {
|
||||||
// Needed to use strcpy_s from string.h
|
for (size_t i = 0; i < buf_length; i++) {
|
||||||
#ifdef __STDC_LIB_EXT1__
|
if (src[i] == '\r') {
|
||||||
strcpy_s(dst, src);
|
dst[i] = ' ';
|
||||||
for (size_t i = 0; i < strlen(dst); i++) {
|
} else {
|
||||||
if (dst[i] == '\r') {
|
dst[i] = src[i];
|
||||||
dst[i] = '_';
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
#endif
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@ -51,12 +48,10 @@ pmacv3Controller::pmacv3Controller(const char *portName,
|
|||||||
portName, ipPortConfigName, numAxes, movingPollPeriod, idlePollPeriod,
|
portName, ipPortConfigName, numAxes, movingPollPeriod, idlePollPeriod,
|
||||||
/*
|
/*
|
||||||
The following parameter library entries are added in this driver:
|
The following parameter library entries are added in this driver:
|
||||||
- ENCODER_TYPE
|
|
||||||
- REREAD_ENCODER_POSITION
|
- REREAD_ENCODER_POSITION
|
||||||
- READ_CONFIG
|
- READ_CONFIG
|
||||||
- ACCEL_FROM_DRIVER
|
|
||||||
*/
|
*/
|
||||||
4)
|
NUM_PMACV3_DRIVER_PARAMS)
|
||||||
|
|
||||||
{
|
{
|
||||||
|
|
||||||
@ -85,15 +80,6 @@ pmacv3Controller::pmacv3Controller(const char *portName,
|
|||||||
// =========================================================================
|
// =========================================================================
|
||||||
// Create additional parameter library entries
|
// Create additional parameter library entries
|
||||||
|
|
||||||
status = createParam("ENCODER_TYPE", asynParamOctet, &encoderType_);
|
|
||||||
if (status != asynSuccess) {
|
|
||||||
asynPrint(this->pasynUserSelf, ASYN_TRACE_ERROR,
|
|
||||||
"%s => line %d:\nFATAL ERROR (creating a parameter failed "
|
|
||||||
"with %s).\nTerminating IOC",
|
|
||||||
__PRETTY_FUNCTION__, __LINE__, stringifyAsynStatus(status));
|
|
||||||
exit(-1);
|
|
||||||
}
|
|
||||||
|
|
||||||
status = createParam("REREAD_ENCODER_POSITION", asynParamInt32,
|
status = createParam("REREAD_ENCODER_POSITION", asynParamInt32,
|
||||||
&rereadEncoderPosition_);
|
&rereadEncoderPosition_);
|
||||||
if (status != asynSuccess) {
|
if (status != asynSuccess) {
|
||||||
@ -307,14 +293,14 @@ asynStatus pmacv3Controller::writeRead(int axisNo, const char *command,
|
|||||||
// Second check: If this fails, give up and propagate the error.
|
// Second check: If this fails, give up and propagate the error.
|
||||||
if (numExpectedResponses != numReceivedResponses) {
|
if (numExpectedResponses != numReceivedResponses) {
|
||||||
|
|
||||||
adjustResponseForPrint(modResponse, response);
|
adjustResponseForPrint(modResponse, response, MAXBUF_);
|
||||||
asynPrint(this->pasynUserSelf, ASYN_TRACE_ERROR,
|
asynPrint(this->pasynUserSelf, ASYN_TRACE_ERROR,
|
||||||
"%s => line %d:\nUnexpected response %s (_ are "
|
"%s => line %d:\nUnexpected response '%s' (carriage "
|
||||||
"carriage returns) for command %s\n",
|
"returns are replaced with spaces) for command %s\n",
|
||||||
__PRETTY_FUNCTION__, __LINE__, modResponse, command);
|
__PRETTY_FUNCTION__, __LINE__, modResponse, command);
|
||||||
snprintf(drvMessageText, sizeof(drvMessageText),
|
snprintf(drvMessageText, sizeof(drvMessageText),
|
||||||
"Received unexpected response %s (_ are "
|
"Received unexpected response '%s' (carriage returns "
|
||||||
"carriage returns) for command %s. "
|
"are replaced with spaces) for command %s. "
|
||||||
"Please call the support",
|
"Please call the support",
|
||||||
modResponse, command);
|
modResponse, command);
|
||||||
pl_status = setStringParam(motorMessageText_, drvMessageText);
|
pl_status = setStringParam(motorMessageText_, drvMessageText);
|
||||||
@ -356,29 +342,6 @@ asynStatus pmacv3Controller::writeRead(int axisNo, const char *command,
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if (status != asynSuccess) {
|
|
||||||
// Check if the axis already is in an error communication mode. If it is
|
|
||||||
// not, upstream the error. This is done to avoid "flooding" the user
|
|
||||||
// with different error messages if more than one error ocurred before
|
|
||||||
// an error-free communication
|
|
||||||
|
|
||||||
pl_status =
|
|
||||||
getIntegerParam(axisNo, motorStatusProblem_, &motorStatusProblem);
|
|
||||||
if (pl_status != asynSuccess) {
|
|
||||||
return paramLibAccessFailed(pl_status, "motorStatusProblem_",
|
|
||||||
__PRETTY_FUNCTION__, __LINE__);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (motorStatusProblem == 0) {
|
|
||||||
pl_status =
|
|
||||||
axis->setStringParam(this->motorMessageText_, drvMessageText);
|
|
||||||
if (pl_status != asynSuccess) {
|
|
||||||
return paramLibAccessFailed(pl_status, "motorMessageText_",
|
|
||||||
__PRETTY_FUNCTION__, __LINE__);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Log the overall status (communication successfull or not)
|
// Log the overall status (communication successfull or not)
|
||||||
if (status == asynSuccess) {
|
if (status == asynSuccess) {
|
||||||
|
|
||||||
@ -388,21 +351,43 @@ asynStatus pmacv3Controller::writeRead(int axisNo, const char *command,
|
|||||||
__PRETTY_FUNCTION__, __LINE__, modResponse);
|
__PRETTY_FUNCTION__, __LINE__, modResponse);
|
||||||
pl_status = axis->setIntegerParam(this->motorStatusCommsError_, 0);
|
pl_status = axis->setIntegerParam(this->motorStatusCommsError_, 0);
|
||||||
} else {
|
} else {
|
||||||
if (status == asynSuccess) {
|
asynPrint(lowLevelPortUser_, ASYN_TRACE_ERROR,
|
||||||
asynPrint(
|
|
||||||
lowLevelPortUser_, ASYN_TRACE_ERROR,
|
|
||||||
"%s => line %d:\nCommunication failed for command %s (%s)\n",
|
"%s => line %d:\nCommunication failed for command %s (%s)\n",
|
||||||
__PRETTY_FUNCTION__, __LINE__, fullCommand,
|
__PRETTY_FUNCTION__, __LINE__, fullCommand,
|
||||||
stringifyAsynStatus(status));
|
stringifyAsynStatus(status));
|
||||||
pl_status = axis->setIntegerParam(this->motorStatusCommsError_, 1);
|
|
||||||
|
// Check if the axis already is in an error communication mode. If it is
|
||||||
|
// not, upstream the error. This is done to avoid "flooding" the user
|
||||||
|
// with different error messages if more than one error ocurred before
|
||||||
|
// an error-free communication
|
||||||
|
pl_status =
|
||||||
|
getIntegerParam(axisNo, motorStatusProblem_, &motorStatusProblem);
|
||||||
|
if (pl_status != asynSuccess) {
|
||||||
|
return paramLibAccessFailed(pl_status, "motorStatusProblem_",
|
||||||
|
__PRETTY_FUNCTION__, __LINE__);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if (motorStatusProblem == 0) {
|
||||||
|
pl_status = axis->setStringParam(motorMessageText_, drvMessageText);
|
||||||
|
if (pl_status != asynSuccess) {
|
||||||
|
return paramLibAccessFailed(pl_status, "motorMessageText_",
|
||||||
|
__PRETTY_FUNCTION__, __LINE__);
|
||||||
|
}
|
||||||
|
|
||||||
|
pl_status = axis->setIntegerParam(motorStatusProblem_, 1);
|
||||||
|
if (pl_status != asynSuccess) {
|
||||||
|
return paramLibAccessFailed(pl_status, "motorStatusProblem",
|
||||||
|
__PRETTY_FUNCTION__, __LINE__);
|
||||||
|
}
|
||||||
|
|
||||||
|
pl_status = axis->setIntegerParam(motorStatusProblem_, 1);
|
||||||
if (pl_status != asynSuccess) {
|
if (pl_status != asynSuccess) {
|
||||||
return paramLibAccessFailed(pl_status, "motorStatusCommsError_",
|
return paramLibAccessFailed(pl_status, "motorStatusCommsError_",
|
||||||
__PRETTY_FUNCTION__, __LINE__);
|
__PRETTY_FUNCTION__, __LINE__);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return asynSuccess;
|
}
|
||||||
|
return status;
|
||||||
}
|
}
|
||||||
|
|
||||||
asynStatus pmacv3Controller::writeInt32(asynUser *pasynUser, epicsInt32 value) {
|
asynStatus pmacv3Controller::writeInt32(asynUser *pasynUser, epicsInt32 value) {
|
||||||
@ -420,17 +405,27 @@ asynStatus pmacv3Controller::writeInt32(asynUser *pasynUser, epicsInt32 value) {
|
|||||||
if (function == rereadEncoderPosition_) {
|
if (function == rereadEncoderPosition_) {
|
||||||
return axis->rereadEncoder();
|
return axis->rereadEncoder();
|
||||||
} else if (function == readConfig_) {
|
} else if (function == readConfig_) {
|
||||||
return axis->readConfig();
|
return axis->atFirstPoll();
|
||||||
} else {
|
} else {
|
||||||
return sinqController::writeInt32(pasynUser, value);
|
return sinqController::writeInt32(pasynUser, value);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
asynStatus sinqController::readInt32(asynUser *pasynUser, epicsInt32 *value) {
|
||||||
|
// PMACs can be disabled
|
||||||
|
if (pasynUser->reason == motorCanDisable_) {
|
||||||
|
*value = 1;
|
||||||
|
return asynSuccess;
|
||||||
|
} else {
|
||||||
|
return asynMotorController::readInt32(pasynUser, value);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
asynStatus pmacv3Controller::errMsgCouldNotParseResponse(
|
asynStatus pmacv3Controller::errMsgCouldNotParseResponse(
|
||||||
const char *command, const char *response, int axisNo,
|
const char *command, const char *response, int axisNo,
|
||||||
const char *functionName, int lineNumber) {
|
const char *functionName, int lineNumber) {
|
||||||
char modifiedResponse[MAXBUF_] = {0};
|
char modifiedResponse[MAXBUF_] = {0};
|
||||||
adjustResponseForPrint(modifiedResponse, response);
|
adjustResponseForPrint(modifiedResponse, response, MAXBUF_);
|
||||||
return sinqController::errMsgCouldNotParseResponse(
|
return sinqController::errMsgCouldNotParseResponse(
|
||||||
command, modifiedResponse, axisNo, functionName, lineNumber);
|
command, modifiedResponse, axisNo, functionName, lineNumber);
|
||||||
}
|
}
|
||||||
@ -446,7 +441,7 @@ C wrapper for the controller constructor. Please refer to the pmacv3Controller
|
|||||||
constructor documentation.
|
constructor documentation.
|
||||||
*/
|
*/
|
||||||
asynStatus pmacv3CreateController(const char *portName,
|
asynStatus pmacv3CreateController(const char *portName,
|
||||||
const char *lowLevelPortName, int numAxes,
|
const char *ipPortConfigName, int numAxes,
|
||||||
double movingPollPeriod,
|
double movingPollPeriod,
|
||||||
double idlePollPeriod, double comTimeout) {
|
double idlePollPeriod, double comTimeout) {
|
||||||
/*
|
/*
|
||||||
@ -461,7 +456,7 @@ asynStatus pmacv3CreateController(const char *portName,
|
|||||||
#pragma GCC diagnostic ignored "-Wunused-but-set-variable"
|
#pragma GCC diagnostic ignored "-Wunused-but-set-variable"
|
||||||
#pragma GCC diagnostic ignored "-Wunused-variable"
|
#pragma GCC diagnostic ignored "-Wunused-variable"
|
||||||
pmacv3Controller *pController =
|
pmacv3Controller *pController =
|
||||||
new pmacv3Controller(portName, lowLevelPortName, numAxes,
|
new pmacv3Controller(portName, ipPortConfigName, numAxes,
|
||||||
movingPollPeriod, idlePollPeriod, comTimeout);
|
movingPollPeriod, idlePollPeriod, comTimeout);
|
||||||
|
|
||||||
return asynSuccess;
|
return asynSuccess;
|
||||||
|
@ -12,9 +12,6 @@
|
|||||||
#include "sinqAxis.h"
|
#include "sinqAxis.h"
|
||||||
#include "sinqController.h"
|
#include "sinqController.h"
|
||||||
|
|
||||||
#define IncrementalEncoder "Incremental encoder"
|
|
||||||
#define AbsoluteEncoder "Absolute encoder"
|
|
||||||
|
|
||||||
class pmacv3Controller : public sinqController {
|
class pmacv3Controller : public sinqController {
|
||||||
|
|
||||||
public:
|
public:
|
||||||
@ -127,11 +124,13 @@ class pmacv3Controller : public sinqController {
|
|||||||
double comTimeout_;
|
double comTimeout_;
|
||||||
|
|
||||||
// Indices of additional PVs
|
// Indices of additional PVs
|
||||||
|
#define FIRST_PMACV3_PARAM rereadEncoderPosition_
|
||||||
int rereadEncoderPosition_;
|
int rereadEncoderPosition_;
|
||||||
int readConfig_;
|
int readConfig_;
|
||||||
int encoderType_;
|
#define LAST_PMACV3_PARAM readConfig_
|
||||||
|
|
||||||
friend class pmacv3Axis;
|
friend class pmacv3Axis;
|
||||||
};
|
};
|
||||||
|
#define NUM_PMACV3_DRIVER_PARAMS (&LAST_PMACV3_PARAM - &FIRST_PMACV3_PARAM + 1)
|
||||||
|
|
||||||
#endif /* pmacv3Controller_H */
|
#endif /* pmacv3Controller_H */
|
||||||
|
200
utils/analyzeTcpDump/analyzeTcpDump.py
Executable file
200
utils/analyzeTcpDump/analyzeTcpDump.py
Executable file
@ -0,0 +1,200 @@
|
|||||||
|
#! venv/bin/python3
|
||||||
|
"""
|
||||||
|
This script can be used to format the communication between an EPICS IOC and a
|
||||||
|
PMAC MCU into JSON files. It does this by parsing PCAP files created by tcpdump
|
||||||
|
and rearranging the information in a more structured manner.
|
||||||
|
|
||||||
|
To read the manual, simply run this script without any arguments.
|
||||||
|
|
||||||
|
Stefan Mathis, January 2025
|
||||||
|
"""
|
||||||
|
|
||||||
|
from scapy.all import *
|
||||||
|
import json
|
||||||
|
import re
|
||||||
|
import codecs
|
||||||
|
import os
|
||||||
|
from datetime import datetime
|
||||||
|
|
||||||
|
def parse(fileAndPath):
|
||||||
|
|
||||||
|
try:
|
||||||
|
scapyCap = PcapReader(fileAndPath)
|
||||||
|
except:
|
||||||
|
print(f"Could not read file {fileAndPath} as PCAP file")
|
||||||
|
|
||||||
|
requests = []
|
||||||
|
sent = []
|
||||||
|
|
||||||
|
jsonDict = dict()
|
||||||
|
lastLayer = None
|
||||||
|
for packet in scapyCap:
|
||||||
|
|
||||||
|
layer = packet.getlayer(Raw)
|
||||||
|
|
||||||
|
if layer is None:
|
||||||
|
continue
|
||||||
|
|
||||||
|
# Skip the package if it is not a command or a response. A command ends
|
||||||
|
# with a carriage return (x0d), a response ends with ACKNOWLEDGE (x06)
|
||||||
|
last = layer.load[-1]
|
||||||
|
if last == 6:
|
||||||
|
isResponse = True
|
||||||
|
elif last == 13:
|
||||||
|
isResponse = False
|
||||||
|
else:
|
||||||
|
continue
|
||||||
|
|
||||||
|
# Store the info by the IP adress of the MCU
|
||||||
|
if isResponse:
|
||||||
|
ip = packet[IP].src
|
||||||
|
else:
|
||||||
|
ip = packet[IP].dst
|
||||||
|
|
||||||
|
if ip not in jsonDict:
|
||||||
|
jsonDict[ip] = dict()
|
||||||
|
|
||||||
|
# Convert to ASCII
|
||||||
|
ascii = layer.load.decode("unicode_escape")
|
||||||
|
|
||||||
|
# Convert the time to a float
|
||||||
|
time = float(packet.time)
|
||||||
|
|
||||||
|
if isResponse:
|
||||||
|
|
||||||
|
# A response is always a number followed by a carriage return
|
||||||
|
responses = re.findall("-?\d+\.\d+\r|-?\d+\r", ascii)
|
||||||
|
|
||||||
|
# Check if the number of responses matches the number of requests
|
||||||
|
valid = len(responses) == len(requests)
|
||||||
|
|
||||||
|
# Pair up the request-response pairs
|
||||||
|
for (request, response) in zip(requests, responses):
|
||||||
|
if request not in jsonDict[ip]:
|
||||||
|
jsonDict[ip][request] = dict()
|
||||||
|
|
||||||
|
if "." in response:
|
||||||
|
value = float(response)
|
||||||
|
else:
|
||||||
|
value = int(response)
|
||||||
|
|
||||||
|
lastLayer = lastPacket.getlayer(Raw)
|
||||||
|
lastTime = int(lastPacket.time)
|
||||||
|
data = {
|
||||||
|
'command': {
|
||||||
|
'hex': [format(value, '02x') for value in lastLayer.load],
|
||||||
|
'ascii': lastLayer.load.decode("unicode_escape"),
|
||||||
|
'timestamp': lastTime,
|
||||||
|
'timeiso': str(datetime.fromtimestamp(lastTime).isoformat()),
|
||||||
|
},
|
||||||
|
'response': {
|
||||||
|
'hex': [format(value, '02x') for value in layer.load],
|
||||||
|
'ascii': ascii,
|
||||||
|
'value': value,
|
||||||
|
'timestamp': time,
|
||||||
|
'timeiso': str(datetime.fromtimestamp(time).isoformat()),
|
||||||
|
'valid': valid
|
||||||
|
}
|
||||||
|
}
|
||||||
|
jsonDict[ip][request][time] = data
|
||||||
|
else:
|
||||||
|
requests.clear()
|
||||||
|
sent.clear()
|
||||||
|
|
||||||
|
# Store the packet for use in the response iteration
|
||||||
|
lastPacket = packet
|
||||||
|
|
||||||
|
# Parse the ASCII text via regex. A PMAC command usually has the
|
||||||
|
# format LDDDD(=<Number>), where L is a capital letter, the first
|
||||||
|
# two digits D are the axis number and the last two digits together
|
||||||
|
# with the letter form the command.
|
||||||
|
|
||||||
|
# Separate the commands into sent data (e.g. setting a position)
|
||||||
|
# and data requests (e.g. reading the axis status). Sent data always
|
||||||
|
# has an equal sign.
|
||||||
|
for command in re.findall("[A-Z]\d+=-?\d+|[A-Z]\d+", ascii):
|
||||||
|
if "=" in command:
|
||||||
|
sent.append(command)
|
||||||
|
else:
|
||||||
|
requests.append(command)
|
||||||
|
|
||||||
|
# Store the sent. The requests yfd stored together with the responses later.
|
||||||
|
for command in sent:
|
||||||
|
splitted = command.split("=")
|
||||||
|
key = splitted[0]
|
||||||
|
key = key + "="
|
||||||
|
if key not in jsonDict[ip]:
|
||||||
|
jsonDict[ip][key] = dict()
|
||||||
|
|
||||||
|
if "." in splitted[1]:
|
||||||
|
value = float(splitted[1])
|
||||||
|
else:
|
||||||
|
value = int(splitted[1])
|
||||||
|
|
||||||
|
data = {
|
||||||
|
'command': {
|
||||||
|
'hex': [format(value, '02x') for value in layer.load],
|
||||||
|
'ascii': ascii,
|
||||||
|
'value': value,
|
||||||
|
'timestamp': time,
|
||||||
|
'timeiso': str(datetime.fromtimestamp(time).isoformat()),
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
jsonDict[ip][key][time] = data
|
||||||
|
|
||||||
|
return jsonDict
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
|
||||||
|
isInstalled = False
|
||||||
|
try:
|
||||||
|
from scapy.all import *
|
||||||
|
isInstalled = True
|
||||||
|
|
||||||
|
except ImportError:
|
||||||
|
print("This script needs the Scapy package to run. In order to install a "
|
||||||
|
"suitable virtual environment, use the 'makevenv' script.")
|
||||||
|
|
||||||
|
if isInstalled:
|
||||||
|
from sys import argv
|
||||||
|
|
||||||
|
if len(argv) < 2:
|
||||||
|
print("""
|
||||||
|
This script can be used to format the communication between an EPICS IOC and a
|
||||||
|
PMAC MCU into JSON files. It does this by parsing PCAP files created by tcpdump
|
||||||
|
and rearranging the information in a more structured manner.
|
||||||
|
|
||||||
|
After a successfull parse run, the resulting JSON data looks like this:
|
||||||
|
<IP Adress MCU1>
|
||||||
|
<Request command type> (e.g. Q0100 to request the position of axis 1)
|
||||||
|
<Event timestamp>
|
||||||
|
Command
|
||||||
|
<Raw ASCII string>
|
||||||
|
<Actual command> (e.g. P0100)
|
||||||
|
<Timestamp in Epoch>
|
||||||
|
Response
|
||||||
|
<Raw ASCII string>
|
||||||
|
<Actual response (e.g. -3)
|
||||||
|
<Timestamp in Epoch>
|
||||||
|
<Set command type> (e.g. Q0100= to set the position of axis 1)
|
||||||
|
<Event timestamp>
|
||||||
|
<Raw ASCII string>
|
||||||
|
<Actual command (e.g. P0100)
|
||||||
|
<Set value>
|
||||||
|
<Timestamp in Epoch>
|
||||||
|
<IP Adress MCU2>
|
||||||
|
""")
|
||||||
|
|
||||||
|
else:
|
||||||
|
for fileAndPath in argv[1:]:
|
||||||
|
jsonDict = parse(fileAndPath)
|
||||||
|
|
||||||
|
# Save the dict into a JSON
|
||||||
|
fileName = os.path.basename(fileAndPath)
|
||||||
|
jsonfileAndPath = f"{fileName}.json"
|
||||||
|
with open(jsonfileAndPath, 'w') as fp:
|
||||||
|
json.dump(jsonDict, fp, indent=4)
|
||||||
|
|
||||||
|
print(f"Stored parse result of {fileAndPath} in {fileName}")
|
BIN
utils/analyzeTcpDump/demo.pcap
Normal file
BIN
utils/analyzeTcpDump/demo.pcap
Normal file
Binary file not shown.
83
utils/analyzeTcpDump/demo.py
Executable file
83
utils/analyzeTcpDump/demo.py
Executable file
@ -0,0 +1,83 @@
|
|||||||
|
#! demovenv/bin/python3
|
||||||
|
"""
|
||||||
|
This demo script shows how the "parse" function of "analyzeTcpDump.py" can be
|
||||||
|
used to easily visualize data from a PCAP file created by the tcpdump tool /
|
||||||
|
wireshark. A suitable virtual environment can be created with the "makedemovenv"
|
||||||
|
script.
|
||||||
|
|
||||||
|
Stefan Mathis, January 2025
|
||||||
|
"""
|
||||||
|
|
||||||
|
import matplotlib.pyplot as plt
|
||||||
|
import matplotlib.dates as mdates
|
||||||
|
from datetime import datetime, timedelta
|
||||||
|
|
||||||
|
from analyzeTcpDump import parse
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
|
||||||
|
data = parse("demo.pcap")
|
||||||
|
|
||||||
|
plt.figure(figsize=(12, 6))
|
||||||
|
|
||||||
|
# Plot the position of axis 5 over time
|
||||||
|
|
||||||
|
# Actual position
|
||||||
|
position_valid = []
|
||||||
|
dates_valid = []
|
||||||
|
position_all = []
|
||||||
|
dates_all = []
|
||||||
|
|
||||||
|
for (timestamp, item) in data["172.28.101.24"]["Q0510"].items():
|
||||||
|
date = datetime.fromtimestamp(timestamp)
|
||||||
|
value = item["response"]["value"]
|
||||||
|
|
||||||
|
dates_all.append(date)
|
||||||
|
position_all.append(value)
|
||||||
|
|
||||||
|
if item["response"]["valid"]:
|
||||||
|
dates_valid.append(date)
|
||||||
|
position_valid.append(value)
|
||||||
|
else:
|
||||||
|
command = item["command"]["ascii"]
|
||||||
|
response = item["response"]["ascii"]
|
||||||
|
|
||||||
|
# Replace non-renderable characters
|
||||||
|
command = command.replace("\0", "\\x00")
|
||||||
|
command = command.replace("\r", "\\x0d")
|
||||||
|
command = command.replace("\x12", "\\x12")
|
||||||
|
response = response.replace("\r", "\\x0d")
|
||||||
|
response = response.replace("\06", "\\x06")
|
||||||
|
|
||||||
|
# Shift the text a bit to the right
|
||||||
|
plt.text(date, value, f"Command: {command}\nResponse: {response}", horizontalalignment="right", verticalalignment="top")
|
||||||
|
|
||||||
|
# Target position
|
||||||
|
position_target = [position_valid[0]]
|
||||||
|
dates_target = [dates_valid[0]]
|
||||||
|
|
||||||
|
for (timestamp, item) in data["172.28.101.24"]["Q0501="].items():
|
||||||
|
date = datetime.fromtimestamp(timestamp)
|
||||||
|
value = item["command"]["value"]
|
||||||
|
|
||||||
|
dates_target.append(date)
|
||||||
|
position_target.append(position_target[-1])
|
||||||
|
|
||||||
|
dates_target.append(date)
|
||||||
|
position_target.append(value)
|
||||||
|
|
||||||
|
dates_target.append(dates_valid[-1])
|
||||||
|
position_target.append(position_target[-1])
|
||||||
|
|
||||||
|
plt.plot(dates_target, position_target, "k--", label="Target position")
|
||||||
|
plt.plot(dates_all, position_all, "r", label="All responses")
|
||||||
|
plt.plot(dates_valid, position_valid, "b", label="Valid responses")
|
||||||
|
plt.xlabel("Time (ISO 8601)")
|
||||||
|
plt.ylabel("Axis position in degree")
|
||||||
|
plt.gca().xaxis.set_major_formatter(mdates.DateFormatter("%Y-%m-%dT%H:%M:%S"))
|
||||||
|
plt.xticks(rotation=45)
|
||||||
|
plt.grid(True)
|
||||||
|
plt.legend(loc="lower left")
|
||||||
|
plt.title("Position of axis 5")
|
||||||
|
plt.tight_layout()
|
||||||
|
plt.show()
|
22
utils/analyzeTcpDump/makedemovenv
Executable file
22
utils/analyzeTcpDump/makedemovenv
Executable file
@ -0,0 +1,22 @@
|
|||||||
|
#!/bin/bash
|
||||||
|
#-------------------------------------------------------------------------
|
||||||
|
# Script which installs a virtual environment for PCAP file parsing
|
||||||
|
#
|
||||||
|
# Stefan Mathis, September 2024
|
||||||
|
#-------------------------------------------------------------------------
|
||||||
|
|
||||||
|
# Remove any previous testing environment
|
||||||
|
if [ -d "demovenv" ]; then
|
||||||
|
rm -r demovenv
|
||||||
|
fi
|
||||||
|
|
||||||
|
/usr/bin/python3.11 -m venv demovenv
|
||||||
|
|
||||||
|
source demovenv/bin/activate
|
||||||
|
|
||||||
|
pip install --upgrade pip
|
||||||
|
pip install "scapy>=2.5,<3.0"
|
||||||
|
pip install matplotlib
|
||||||
|
|
||||||
|
# Exit the virtual environment
|
||||||
|
exit
|
21
utils/analyzeTcpDump/makevenv
Executable file
21
utils/analyzeTcpDump/makevenv
Executable file
@ -0,0 +1,21 @@
|
|||||||
|
#!/bin/bash
|
||||||
|
#-------------------------------------------------------------------------
|
||||||
|
# Script which installs a virtual environment for PCAP file parsing
|
||||||
|
#
|
||||||
|
# Stefan Mathis, September 2024
|
||||||
|
#-------------------------------------------------------------------------
|
||||||
|
|
||||||
|
# Remove any previous testing environment
|
||||||
|
if [ -d "venv" ]; then
|
||||||
|
rm -r venv
|
||||||
|
fi
|
||||||
|
|
||||||
|
/usr/bin/python3.11 -m venv venv
|
||||||
|
|
||||||
|
source venv/bin/activate
|
||||||
|
|
||||||
|
pip install --upgrade pip
|
||||||
|
pip install "scapy>=2.5,<3.0"
|
||||||
|
|
||||||
|
# Exit the virtual environment
|
||||||
|
exit
|
172
utils/writeRead.py
Normal file
172
utils/writeRead.py
Normal file
@ -0,0 +1,172 @@
|
|||||||
|
#!/usr/bin/env python3
|
||||||
|
"""
|
||||||
|
This script allows direct interaction with a pmac-Controller over an ethernet connection.
|
||||||
|
To read the manual, simply run this script without any arguments.
|
||||||
|
|
||||||
|
Stefan Mathis, December 2024
|
||||||
|
"""
|
||||||
|
|
||||||
|
import struct
|
||||||
|
import socket
|
||||||
|
import curses
|
||||||
|
|
||||||
|
def packPmacCommand(command):
|
||||||
|
# 0x40 = VR_DOWNLOAD
|
||||||
|
# 0xBF = VR_PMAC_GETRESPONSE
|
||||||
|
buf = struct.pack('BBHHH',0x40,0xBF,0,0,socket.htons(len(command)))
|
||||||
|
buf = buf + bytes(command,'utf-8')
|
||||||
|
return buf
|
||||||
|
|
||||||
|
def readPmacReply(input):
|
||||||
|
msg = bytearray()
|
||||||
|
expectAck = True
|
||||||
|
while True:
|
||||||
|
b = input.recv(1)
|
||||||
|
bint = int.from_bytes(b,byteorder='little')
|
||||||
|
if bint == 2 or bint == 7: #STX or BELL
|
||||||
|
expectAck = False
|
||||||
|
continue
|
||||||
|
if expectAck and bint == 6: # ACK
|
||||||
|
return bytes(msg)
|
||||||
|
else:
|
||||||
|
if bint == 13 and not expectAck: # CR
|
||||||
|
return bytes(msg)
|
||||||
|
else:
|
||||||
|
msg.append(bint)
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
from sys import argv
|
||||||
|
|
||||||
|
try:
|
||||||
|
|
||||||
|
addr = argv[1].split(':')
|
||||||
|
s = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
|
||||||
|
s.connect((addr[0],int(addr[1])))
|
||||||
|
|
||||||
|
if len(argv) == 3:
|
||||||
|
buf = packPmacCommand(argv[2])
|
||||||
|
s.send(buf)
|
||||||
|
reply = readPmacReply(s)
|
||||||
|
print(reply.decode('utf-8') + '\n')
|
||||||
|
|
||||||
|
else:
|
||||||
|
|
||||||
|
try:
|
||||||
|
|
||||||
|
stdscr = curses.initscr()
|
||||||
|
curses.noecho()
|
||||||
|
curses.cbreak()
|
||||||
|
stdscr.keypad(True)
|
||||||
|
stdscr.scrollok(True)
|
||||||
|
|
||||||
|
stdscr.addstr(">> ")
|
||||||
|
stdscr.refresh()
|
||||||
|
|
||||||
|
history = [""]
|
||||||
|
ptr = len(history) - 1
|
||||||
|
|
||||||
|
while True:
|
||||||
|
c = stdscr.getch()
|
||||||
|
if c == curses.KEY_RIGHT:
|
||||||
|
(y, x) = stdscr.getyx()
|
||||||
|
if x < len(history[ptr]) + 3:
|
||||||
|
stdscr.move(y, x+1)
|
||||||
|
stdscr.refresh()
|
||||||
|
elif c == curses.KEY_LEFT:
|
||||||
|
(y, x) = stdscr.getyx()
|
||||||
|
if x > 3:
|
||||||
|
stdscr.move(y, x-1)
|
||||||
|
stdscr.refresh()
|
||||||
|
elif c == curses.KEY_UP:
|
||||||
|
if ptr > 0:
|
||||||
|
ptr -= 1
|
||||||
|
stdscr.addch("\r")
|
||||||
|
stdscr.clrtoeol()
|
||||||
|
stdscr.addstr(">> " + history[ptr])
|
||||||
|
elif c == curses.KEY_DOWN:
|
||||||
|
if ptr < len(history) - 1:
|
||||||
|
ptr += 1
|
||||||
|
stdscr.addch("\r")
|
||||||
|
stdscr.clrtoeol()
|
||||||
|
stdscr.addstr(">> " + history[ptr])
|
||||||
|
elif c == curses.KEY_ENTER or c == ord('\n') or c == ord('\r'):
|
||||||
|
if history[ptr] == 'quit':
|
||||||
|
break
|
||||||
|
|
||||||
|
# because of arrow keys move back to the end of the line
|
||||||
|
(y, x) = stdscr.getyx()
|
||||||
|
stdscr.move(y, 3+len(history[ptr]))
|
||||||
|
|
||||||
|
if history[ptr]:
|
||||||
|
buf = packPmacCommand(history[ptr])
|
||||||
|
s.send(buf)
|
||||||
|
reply = readPmacReply(s)
|
||||||
|
stdscr.addstr("\n" + reply.decode('utf-8')[0:-1])
|
||||||
|
|
||||||
|
if ptr == len(history) - 1 and history[ptr] != "":
|
||||||
|
history += [""]
|
||||||
|
else:
|
||||||
|
history[-1] = ""
|
||||||
|
ptr = len(history) - 1
|
||||||
|
|
||||||
|
stdscr.addstr("\n>> ")
|
||||||
|
stdscr.refresh()
|
||||||
|
|
||||||
|
else:
|
||||||
|
if ptr < len(history) - 1: # Modifying previous input
|
||||||
|
if len(history[-1]) == 0:
|
||||||
|
history[-1] = history[ptr]
|
||||||
|
ptr = len(history) - 1
|
||||||
|
|
||||||
|
else:
|
||||||
|
history += [history[ptr]]
|
||||||
|
ptr = len(history) - 1
|
||||||
|
|
||||||
|
if c == curses.KEY_BACKSPACE:
|
||||||
|
if len(history[ptr]) == 0:
|
||||||
|
continue
|
||||||
|
(y, x) = stdscr.getyx()
|
||||||
|
history[ptr] = history[ptr][0:x-4] + history[ptr][x-3:]
|
||||||
|
stdscr.addch("\r")
|
||||||
|
stdscr.clrtoeol()
|
||||||
|
stdscr.addstr(">> " + history[ptr])
|
||||||
|
stdscr.move(y, x-1)
|
||||||
|
stdscr.refresh()
|
||||||
|
|
||||||
|
else:
|
||||||
|
(y, x) = stdscr.getyx()
|
||||||
|
history[ptr] = history[ptr][0:x-3] + chr(c) + history[ptr][x-3:]
|
||||||
|
stdscr.addch("\r")
|
||||||
|
stdscr.clrtoeol()
|
||||||
|
stdscr.addstr(">> " + history[ptr])
|
||||||
|
stdscr.move(y, x+1)
|
||||||
|
stdscr.refresh()
|
||||||
|
|
||||||
|
finally:
|
||||||
|
|
||||||
|
# to quit
|
||||||
|
curses.nocbreak()
|
||||||
|
stdscr.keypad(False)
|
||||||
|
curses.echo()
|
||||||
|
curses.endwin()
|
||||||
|
|
||||||
|
except:
|
||||||
|
print("""
|
||||||
|
Invalid Arguments
|
||||||
|
|
||||||
|
Option 1: Single Command
|
||||||
|
------------------------
|
||||||
|
|
||||||
|
Usage: writeRead.py pmachost:port command
|
||||||
|
This then returns the response for command.
|
||||||
|
|
||||||
|
Option 2: CLI Mode
|
||||||
|
------------------
|
||||||
|
|
||||||
|
Usage: writeRead.py pmachost:port
|
||||||
|
|
||||||
|
You can then type in a command, hit enter, and the response will see
|
||||||
|
the reponse, before being prompted to again enter a command. Type
|
||||||
|
'quit' to close prompt.
|
||||||
|
""")
|
||||||
|
|
Reference in New Issue
Block a user