1

/*****************************************************************************

     2

 *

     3

 *  $Id$

     4

 *

     5

 *  Copyright (C) 2006-2012  Florian Pose, Ingenieurgemeinschaft IgH

     6

 *

     7

 *  This file is part of the IgH EtherCAT Master.

     8

 *

     9

 *  The IgH EtherCAT Master is free software; you can redistribute it and/or

    10

 *  modify it under the terms of the GNU General Public License version 2, as

    11

 *  published by the Free Software Foundation.

    12

 *

    13

 *  The IgH EtherCAT Master is distributed in the hope that it will be useful,

    14

 *  but WITHOUT ANY WARRANTY; without even the implied warranty of

    15

 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General

    16

 *  Public License for more details.

    17

 *

    18

 *  You should have received a copy of the GNU General Public License along

    19

 *  with the IgH EtherCAT Master; if not, write to the Free Software

    20

 *  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA

    21

 *

    22

 *  ---

    23

 *

    24

 *  The license mentioned above concerns the source code only. Using the

    25

 *  EtherCAT technology and brand is only permitted in compliance with the

    26

 *  industrial property and similar rights of Beckhoff Automation GmbH.

    27

 *

    28

 ****************************************************************************/

    29

    30

#include <iostream>

    31

#include <iomanip>

    32

#include <fstream>

    33

using namespace std;

    34

    35

#include "CommandRegWrite.h"

    36

#include "sii_crc.h"

    37

#include "MasterDevice.h"

    38

    39

/*****************************************************************************/

    40

    41

CommandRegWrite::CommandRegWrite():

    42

    Command("reg_write", "Write data to a slave's registers.")

    43

{

    44

}

    45

    46

/*****************************************************************************/

    47

    48

string CommandRegWrite::helpString(const string &binaryBaseName) const

    49

{

    50

    stringstream str;

    51

    52

    str << binaryBaseName << " " << getName()

    53

        << " [OPTIONS] <OFFSET> <DATA>" << endl

    54

        << endl

    55

        << getBriefDescription() << endl

    56

        << endl

    57

        << "This command requires a single slave to be selected." << endl

    58

        << endl

    59

        << "Arguments:" << endl

    60

        << "  ADDRESS is the register address to write to." << endl

    61

        << "  DATA    depends on whether a datatype was specified" << endl

    62

        << "          with the --type option: If not, DATA must be" << endl

    63

        << "          either a path to a file with data to write," << endl

    64

        << "          or '-', which means, that data are read from" << endl

    65

        << "          stdin. If a datatype was specified, VALUE is" << endl

    66

        << "          interpreted respective to the given type." << endl

    67

        << endl

    68

        << typeInfo()

    69

        << endl

    70

        << "Command-specific options:" << endl

    71

        << "  --alias    -a <alias>" << endl

    72

        << "  --position -p <pos>    Slave selection. See the help of" << endl

    73

        << "                         the 'slaves' command." << endl

    74

        << "  --type     -t <type>   Data type (see above)." << endl

    75

        << endl

    76

        << numericInfo();

    77

    78

    return str.str();

    79

}

    80

    81

/****************************************************************************/

    82

    83

void CommandRegWrite::execute(const StringVector &args)

    84

{

    85

    stringstream strOffset, err;

    86

    ec_ioctl_slave_reg_t io;

    87

    ifstream file;

    88

    SlaveList slaves;

    89

    90

    if (args.size() != 2) {

    91

        err << "'" << getName() << "' takes exactly two arguments!";

    92

        throwInvalidUsageException(err);

    93

    }

    94

    95

    strOffset << args[0];

    96

    strOffset

    97

        >> resetiosflags(ios::basefield) // guess base from prefix

    98

        >> io.address;

    99

    if (strOffset.fail()) {

   100

        err << "Invalid address '" << args[0] << "'!";

   101

        throwInvalidUsageException(err);

   102

    }

   103

   104

    if (getDataType().empty()) {

   105

        if (args[1] == "-") {

   106

            loadRegData(&io, cin);

   107

        } else {

   108

            file.open(args[1].c_str(), ifstream::in | ifstream::binary);

   109

            if (file.fail()) {

   110

                err << "Failed to open '" << args[1] << "'!";

   111

                throwCommandException(err);

   112

            }

   113

            loadRegData(&io, file);

   114

            file.close();

   115

        }

   116

    } else {

   117

        stringstream strValue;

   118

        const DataType *dataType = findDataType(getDataType());

   119

   120

        if (!dataType) {

   121

            err << "Invalid data type '" << getDataType() << "'!";

   122

            throwInvalidUsageException(err);

   123

        }

   124

   125

        if (dataType->byteSize) {

   126

            io.size = dataType->byteSize;

   127

        } else {

   128

            io.size = 1024; // FIXME

   129

        }

   130

   131

        io.data = new uint8_t[io.size];

   132

   133

        try {

   134

            io.size = interpretAsType(

   135

                    dataType, args[1], io.data, io.size);

   136

        } catch (SizeException &e) {

   137

            delete [] io.data;

   138

            throwCommandException(e.what());

   139

        } catch (ios::failure &e) {

   140

            delete [] io.data;

   141

            err << "Invalid value argument '" << args[1]

   142

                << "' for type '" << dataType->name << "'!";

   143

            throwInvalidUsageException(err);

   144

        }

   145

    }

   146

   147

    if ((uint32_t) io.address + io.size > 0xffff) {

   148

        err << "Address and size exceeding 64k!";

   149

        delete [] io.data;

   150

        throwInvalidUsageException(err);

   151

    }

   152

   153

    MasterDevice m(getSingleMasterIndex());

   154

    try {

   155

        m.open(MasterDevice::ReadWrite);

   156

    } catch (MasterDeviceException &e) {

   157

        delete [] io.data;

   158

        throw e;

   159

    }

   160

   161

    slaves = selectedSlaves(m);

   162

    if (slaves.size() != 1) {

   163

        delete [] io.data;

   164

        throwSingleSlaveRequired(slaves.size());

   165

    }

   166

    io.slave_position = slaves.front().position;

   167

   168

    // send data to master

   169

    try {

   170

        m.writeReg(&io);

   171

    } catch (MasterDeviceException &e) {

   172

        delete [] io.data;

   173

        throw e;

   174

    }

   175

   176

    if (getVerbosity() == Verbose) {

   177

        cerr << "Register writing finished." << endl;

   178

    }

   179

   180

    delete [] io.data;

   181

}

   182

   183

/*****************************************************************************/

   184

   185

void CommandRegWrite::loadRegData(

   186

        ec_ioctl_slave_reg_t *io,

   187

        const istream &in

   188

        )

   189

{

   190

    stringstream err;

   191

    ostringstream tmp;

   192

   193

    tmp << in.rdbuf();

   194

    string const &contents = tmp.str();

   195

   196

    if (getVerbosity() == Verbose) {

   197

        cerr << "Read " << contents.size() << " bytes of data." << endl;

   198

    }

   199

   200

    if (contents.size() > 0xffff) {

   201

        err << "Invalid data size " << contents.size() << "!";

   202

        throwInvalidUsageException(err);

   203

    }

   204

    io->size = contents.size();

   205

   206

    // allocate buffer and read file into buffer

   207

    io->data = new uint8_t[io->size];

   208

    contents.copy((char *) io->data, contents.size());

   209

}

   210

   211

/*****************************************************************************/