portcommand.cpp

来自「一个语言识别引擎」· C++ 代码 · 共 74 行

// -*- mode:C++; tab-width:4; c-basic-offset:4; indent-tabs-mode:nil -*-

/*
 * Copyright (C) 2006 Paul Fitzpatrick
 * CopyPolicy: Released under the terms of the GNU GPL v2.0.
 *
 */

#include <yarp/PortCommand.h>
#include <yarp/Logger.h>
#include <yarp/NetType.h>

using namespace yarp;

void PortCommand::readBlock(ConnectionReader& reader) {
    //ACE_DEBUG((LM_DEBUG,"PortCommand::readBlock"));
    ch = '\0';
    str = "";
    if (!reader.isTextMode()) {
        reader.expectBlock(header.get(),header.length());
        char *base = header.get();
        if (base[4] == '~') {
            ch = base[5];
            if (ch=='\0') {
                //str = reader.expectString(reader.getSize());
                str = reader.expectText('\0').c_str();
                if (str.length()>0) {
                    ch = str[0];
                }
            }
        } else {
            throw IOException("bad header in PortCommand::readBlock");
        }
    } else {
        //ACE_OS::printf("PortCommand::readBlock pre read\n");
        str = reader.expectText().c_str();
        //ACE_OS::printf("PortCommand::readBlock post read\n");
        if (str.length()>0) {
            ch = str[0];
        }
    }
}

void PortCommand::writeBlock(ConnectionWriter& writer) {
    ACE_DEBUG((LM_DEBUG,"PortCommand::writeBlock"));
    //ACE_OS::printf("Writing port command, text mode %d\n", writer.isTextMode());
    if (!writer.isTextMode()) {
        int len = 0;
        if (ch=='\0') {
            len = str.length()+1;
        }
        YARP_ASSERT(header.length()==8);
        char *base = header.get();
        Bytes b(base,4);
        NetType::netInt(len,b);
        base[4] = '~';
        base[5] = ch;
        base[6] = 0;
        base[7] = 1;
        writer.appendBlock(header.bytes().get(),header.bytes().length());
        if (ch=='\0') {
            writer.appendBlock(str.c_str(),str.length()+1);
        }
    } else {
        if (ch!='\0') {
            writer.appendString(String(ch).c_str(),'\n');
        } else {
            writer.appendString(str.c_str(),'\n');
        }
    }
}