socketstreamdrive.h

Go to the documentation of this file.

/* $Header: /var/cvs/mbdyn/mbdyn/mbdyn-1.0/mbdyn/base/socketstreamdrive.h,v 1.37 2017/09/09 09:18:21 masarati Exp $ */
/*
 * MBDyn (C) is a multibody analysis code.
 * http://www.mbdyn.org
 *
 * Copyright (C) 1996-2017
 *
 * Pierangelo Masarati  <masarati@aero.polimi.it>
 * Paolo Mantegazza     <mantegazza@aero.polimi.it>
 *
 * Dipartimento di Ingegneria Aerospaziale - Politecnico di Milano
 * via La Masa, 34 - 20156 Milano, Italy
 * http://www.aero.polimi.it
 *
 * Changing this copyright notice is forbidden.
 *
 * This program 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 (version 2 of the License).
 *
 *
 * This program 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 this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 */

/*
 * Author: Michele Attolico <attolico@aero.polimi.it>
 */

/* socket driver */

#ifndef SOCKETSTREAMDRIVE_H
#define SOCKETSTREAMDRIVE_H

#include "streamdrive.h"

#ifdef USE_SOCKET

#include "usesock.h"

/* SocketStreamDrive - begin */

class SocketStreamDrive : public StreamDrive {
protected:
        unsigned int InputEvery;
        bool bReceiveFirst;
        unsigned int InputCounter;

        UseSocket *pUS;
        int recv_flags;
        struct timeval SocketTimeout;

        StreamDriveEcho *pSDE;

public:
        SocketStreamDrive(unsigned int uL,
                const DriveHandler* pDH,
                UseSocket *pUS, bool c,
                const std::string& sFileName,
                integer nd, const std::vector<doublereal>& v0,
                StreamDrive::Modifier *pMod,
                unsigned int ie, bool bReceiveFirst,
                int flags,
                const struct timeval& st,
                StreamDriveEcho *pSDE);

        virtual ~SocketStreamDrive(void);

        /* Scrive il contributo del DriveCaller al file di restart */
        virtual std::ostream& Restart(std::ostream& out) const;

        virtual void ServePending(const doublereal& t);
};

/* SocketStreamDrive - end */

#endif // USE_SOCKET

class DataManager;
class MBDynParser;

struct StreamDR : public DriveRead {
private:
        std::string s;

public:
        StreamDR(void) { NO_OP; };
        StreamDR(const std::string &s) : s(s) { NO_OP; };

        virtual Drive *
        Read(unsigned uLabel, const DataManager *pDM, MBDynParser& HP);
};

/* Luca Conti edits - GSOC 2017 */

/* file drive content type reader: every content type must inherit
from this struct and implement its own Read method */
struct FileDriveContentTypeReader {
        virtual StreamDrive::Modifier * Read(std::vector<doublereal> &v0, MBDynParser& HP, int &idrives) = 0;
        virtual ~FileDriveContentTypeReader(void) { NO_OP; };
};
/* bag of content-type readers - every content type is registered inside
of it by using SetFileDriveContentType(...) */
typedef std::map<std::string,FileDriveContentTypeReader*> FileDriveContentTypeMap;
extern FileDriveContentTypeMap fileDriveContentTypeMap;

struct FileDriveContentTypeWordSetType : public HighParser::WordSet {
        virtual bool IsWord(const std::string& s) const;
};
extern FileDriveContentTypeWordSetType fileDriveContentTypeWordSet;

/* registration function: call it to register a new content type */
bool SetFileDriveContentType(const char *name, FileDriveContentTypeReader *rf);

/* deallocation of all content types in fileDriveContentTypeMap, if any was added */
void DestroyFileDriveContentTypes(void);

#endif /* SOCKETSTREAMDRIVE_H */