MagicYModule.h

Go to the documentation of this file.

/* ========================================================================
 * Copyright (c) 2006,
 * Institute for Computer Graphics and Vision
 * Graz University of Technology
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are
 * met:
 *
 * Redistributions of source code must retain the above copyright notice,
 * this list of conditions and the following disclaimer.
 *
 * Redistributions in binary form must reproduce the above copyright
 * notice, this list of conditions and the following disclaimer in the
 * documentation and/or other materials provided with the distribution.
 *
 * Neither the name of the Graz University of Technology nor the names of
 * its contributors may be used to endorse or promote products derived from
 * this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS
 * IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A
 * PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER
 * OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
 * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
 * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 * ========================================================================
 * PROJECT: OpenTracker
 * ======================================================================== */
/* ======================================================================== */

#ifndef _MAGICYMODULE_H
#define _MAGICYMODULE_H

#include "../OpenTracker.h"
#include "MagicYSource.h"

#include <ace/SOCK_Dgram.h>
#include <ace/SOCK_Connector.h>
#include <ace/INET_Addr.h>
#include <ace/Handle_Set.h>

#include <vector>


#ifndef OT_NO_MAGICY_SUPPORT


namespace ot {

    class MagicY
    {
    public:
        int number;    
        bool average;
        Event event;
        int modified;
        MagicYSource * source;

        MagicY( const int number_, const bool average_, MagicYSource * source_ ) :
            number( number_ ), average( average_ ), modified( 0 ), source( source_ )
        {
            event.getPosition()[0] = 0;
            event.getPosition()[1] = 0;
            event.getPosition()[2] = 0;

            event.getOrientation()[0] = 0;
            event.getOrientation()[1] = 0;
            event.getOrientation()[2] = 0;
            event.getOrientation()[3] = 0;

            event.getButton() = 0;
            event.getConfidence() = 0.0f;
        };
    };

    // struct to store points for intermediate processing (prediction, etc.)
    struct MagicPoint
    {
        int x, y;
        bool trigger;

        MagicPoint(const int x_, const int y_) : x(x_), y(y_), trigger(0) {};
        MagicPoint(const int x_, const int y_, const bool trigger_) : x(x_), y(y_), trigger(trigger_) {};
        MagicPoint(const MagicPoint &src) : x(src.x), y(src.y), trigger(src.trigger) {};
    };

    // struct to store ports and offsets for data streams for extra screens
    struct Screen
    {
        ACE_SOCK_Stream socket;
        ACE_INET_Addr address;
        int x_offset, y_offset;
        bool connected;

        Screen(const int port_, const std::string hostname_, const int x_offset_, const int y_offset_)
        {
            address.set((u_short)port_, hostname_.c_str()); 
            x_offset = x_offset_; 
            y_offset = y_offset_;
            connected = false;
        };
    }; 

    typedef std::vector<MagicY *> MagicYVector;
    typedef std::vector<Screen *> ScreenVector;
    typedef std::vector<MagicPoint> PointVector;

    const int magicYMaxUnits = 1000;

    class OPENTRACKER_API MagicYModule : public ThreadModule, public NodeFactory
    {
        // members
    protected:    
    
        MagicYVector magicYs;
        PointVector points;
        ScreenVector screens;
        std::string hostname;

        ACE_SOCK_Connector connector;
        ACE_Handle_Set readHandles;

        int stop;
        int positionMapping[3];
        int invertPosition[3];
        float z_value;
        float orientation[4];
    
        // methods
    protected:

        int connect();
        void setSelect();
        int receive();
        int stillConnected(); 
        void disconnect();
        void run();
        int parseVector(const std::string & line, int * val);
        int parseVector(const std::string & line, float * val);
        int parseScreens(const std::string & line);
        void initMappping(int *mapping);
        void initInversion(int *inversion);
        void initOrientation(float *orientation);
        void calcInversion(int *inversion);
        void calcMapping(int *mapping);
        void correctData(std::vector<float> &d, int *mapping, int *inversion);

    public:    
        MagicYModule();

        virtual ~MagicYModule();
        virtual void init(StringTable& attributes, ConfigNode * localTree);
        virtual Node * createNode( const std::string& name,  StringTable& attributes);        
        virtual void start();
        virtual void close();    
        virtual void pushEvent();             
    };
        OT_MODULE(MagicYModule);
        

} // namespace ot


#endif // OT_NO_MAGICY_SUPPORT


#endif

/* 
 * ------------------------------------------------------------
 *   End of MagicYModule.h
 * ------------------------------------------------------------
 *   Automatic Emacs configuration follows.
 *   Local Variables:
 *   mode:c++
 *   c-basic-offset: 4
 *   eval: (c-set-offset 'substatement-open 0)
 *   eval: (c-set-offset 'case-label '+)
 *   eval: (c-set-offset 'statement 'c-lineup-runin-statements)
 *   eval: (setq indent-tabs-mode nil)
 *   End:
 * ------------------------------------------------------------ 
 */

---