cOSHandleSet.h

Go to the documentation of this file.

//
//
 
#ifndef _INC_cOSHandleSet_H
#define _INC_cOSHandleSet_H
#ifndef NO_PRAGMA_ONCE
#pragma once
#endif
 
#include "cOSHandle.h"
#include "cArray.h"
 
namespace Gray
{
        class GRAYCORE_LINK cOSHandleSet
        {
 
        public:
                static const int k_nHandeMax =  
#ifdef _WIN32
                MAXIMUM_WAIT_OBJECTS
#else
                FD_SETSIZE
#endif
                ;
 
        private:
#ifdef _WIN32
                cArrayVal<HANDLE> m_fds;        
#elif defined(__linux__)
                HANDLE m_hHandleMax;            
                fd_set m_fds;                           
#else
#error NOOS
#endif
 
        private:
                void InitHandles() noexcept
                {
#ifdef __linux__
                        m_hHandleMax = 0;
                        FD_ZERO(&m_fds);
#endif
                }
 
        public:
                cOSHandleSet(void) noexcept
                {
                        InitHandles();
                }
                cOSHandleSet(HANDLE h)
                {
                        // a handle set with a single handle.
#ifdef __linux__                         
                        m_hHandleMax = h;
                        FD_ZERO(&fds);
                        FD_SET(h, &m_fds);
#else
                        InitHandles();
                        AddHandle(h);
#endif
                }
                cOSHandleSet(const cOSHandleSet& nss)
                {
                        InitHandles();
                        SetCopy(nss);
                }
                ~cOSHandleSet()
                {
                }
 
                void operator = (const cOSHandleSet& nss)
                {
                        SetCopy(nss);
                }
                void SetCopy(const cOSHandleSet& nss)
                {
#ifdef __linux__
                        m_hHandleMax = nss.m_hHandleMax;
                        // FD_COPY(pfds,&m_fds);
                        cMem::Copy(&m_fds,&nss.m_fds,sizeof(m_fds));
#else
                        m_fds = nss.m_fds;
#endif
                }
 
                bool AddHandle(HANDLE h)
                {
 
                        if (h == INVALID_HANDLE_VALUE)
                                return false;
#ifdef __linux__
                        if ( h > m_hHandleMax )
                                m_hHandleMax = h;
                        FD_SET(h,&m_fds);
#else
                        // Only add up to
                        if (h == HANDLE_NULL)   // cOSHandle
                                return false;
                        if (m_fds.GetSize() >= MAXIMUM_WAIT_OBJECTS)
                                return false;
                        m_fds.Add(h);
#endif
                        return true;
                }
                void RemoveHandle(HANDLE h)
                {
                        if (h == INVALID_HANDLE_VALUE)
                                return;
#ifdef __linux__
                        FD_CLR(h,&m_fds);
#else
                        m_fds.RemoveArg(h);
#endif
                }
                void ClearHandles()
                {
#ifdef __linux__
                        InitHandles();
#else
                        m_fds.RemoveAll();
#endif
                }
 
                HRESULT WaitForObjects(TIMESYSD_t dwMilliseconds, bool bWaitForAll = false);
        };
};
#endif