SampleSourceSupport.h

Go to the documentation of this file.

// -*- mode: C++; indent-tabs-mode: nil; c-basic-offset: 4; tab-width: 4; -*-
// vim: set shiftwidth=4 softtabstop=4 expandtab:
/*
 ********************************************************************
 ** NIDAS: NCAR In-situ Data Acquistion Software
 **
 ** 2009, Copyright University Corporation for Atmospheric Research
 **
 ** 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; either version 2 of the License, or
 ** (at your option) any later version.
 **
 ** 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.
 **
 ** The LICENSE.txt file accompanying this software contains
 ** a copy of the GNU General Public License. If it is not found,
 ** write to the Free Software Foundation, Inc.,
 ** 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 **
 ********************************************************************
*/

#ifndef NIDAS_CORE_SAMPLESOURCESUPPORT_H
#define NIDAS_CORE_SAMPLESOURCESUPPORT_H

#include "SampleSource.h"
#include "SampleClientList.h"

#include <set>
#include <map>

namespace nidas { namespace core {

class SampleTag;

class SampleSourceSupport: public SampleSource {
public:

    SampleSourceSupport(bool raw);

    SampleSourceSupport(const SampleSourceSupport& x);

    virtual ~SampleSourceSupport() {}

    SampleSource* getRawSampleSource()
    {
        if (!_raw) return 0;
        return this;
    }

    SampleSource* getProcessedSampleSource()
    {
        if (_raw) return 0;
        return this;
    }

    void addSampleTag(const SampleTag* tag) throw();

    void removeSampleTag(const SampleTag* tag) throw();

    std::list<const SampleTag*> getSampleTags() const;

    SampleTagIterator getSampleTagIterator() const;

    void addSampleClient(SampleClient* c) throw();

    void removeSampleClient(SampleClient* c) throw();

    void addSampleClientForTag(SampleClient* c,const SampleTag*) throw();

    void removeSampleClientForTag(SampleClient* c,const SampleTag*) throw();

    int getClientCount() const throw();

    void removeAllSampleClients() throw()
    {
        _clients.removeAll();
        _clientMapLock.lock();
        _clientSet.clear();
        _clientMapLock.unlock();
    }

    void distribute(const Sample* s) throw();

    void distribute(const std::list<const Sample*>& samps) throw();

    void flush() throw() {}

    const SampleStats& getSampleStats() const
    {
        return _stats;
    }

    void setKeepStats(bool val)
    {
        _keepStats = val;
    }

    bool getKeepStats() const
    {
        return _keepStats;
    }

private:

    mutable nidas::util::Mutex _tagsMutex;

    std::list<const SampleTag*> _sampleTags;

    SampleClientList _clients;

    std::map<dsm_sample_id_t,SampleClientList> _clientsBySampleId;

    std::set<SampleClient*> _clientSet;

    nidas::util::Mutex _clientMapLock;

    SampleStats _stats;

    bool _raw;

    bool _keepStats;

    SampleSourceSupport& operator = (const SampleSourceSupport& x);

};

}}  // namespace nidas namespace core


#endif