SampleAverager.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
 **
 ** 2005, 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_SAMPLEAVERAGER_H
#define NIDAS_CORE_SAMPLEAVERAGER_H

#include "Resampler.h"
#include "SampleTag.h"

namespace nidas { namespace core {

class Variable;

class SampleAverager : public Resampler {
public:

    SampleAverager();

    SampleAverager(const std::vector<const Variable*>& vars);

    SampleAverager(const std::vector<Variable*>& vars);

    virtual ~SampleAverager();

    void setAveragePeriodSecs(float val) {
        _averagePeriodUsecs = (int)rint((double)val * USECS_PER_SEC);
    _outSample.setRate(val);
    }

    float getAveragePeriodSecs() const { return (double)_averagePeriodUsecs / USECS_PER_SEC; }

    void addVariable(const Variable *var);

    void addVariables(const std::vector<const Variable*>&);

    SampleSource* getRawSampleSource() { return 0; }

    SampleSource* getProcessedSampleSource() { return &_source; }

    std::list<const SampleTag*> getSampleTags() const
    {
        return _source.getSampleTags();
    }

    SampleTagIterator getSampleTagIterator() const
    {
        return _source.getSampleTagIterator();
    }

    void addSampleClient(SampleClient* client) throw()
    {
        _source.addSampleClient(client);
    }

    void removeSampleClient(SampleClient* client) throw()
    {
        _source.removeSampleClient(client);
    }

    void addSampleClientForTag(SampleClient* client,const SampleTag*) throw()
    {
        // I only have one tag, so just call addSampleClient()
        _source.addSampleClient(client);
    }

    void removeSampleClientForTag(SampleClient* client,const SampleTag*) throw()
    {
        _source.removeSampleClient(client);
    }

    int getClientCount() const throw()
    {
        return _source.getClientCount();
    }

    void flush() throw();

    const SampleStats& getSampleStats() const
    {
        return _source.getSampleStats();
    }

    void connect(SampleSource* source) throw(nidas::util::InvalidParameterException);

    void disconnect(SampleSource* source) throw();

    bool receive(const Sample *s) throw();

protected:

    void init() throw();

private:
   
    void addSampleTag(const SampleTag* tag) throw ()
    {
        _source.addSampleTag(tag);
    }

    void removeSampleTag(const SampleTag* tag) throw ()
    {
        _source.removeSampleTag(tag);
    }

    SampleSourceSupport _source;

    SampleTag _outSample;

    int _averagePeriodUsecs;

    dsm_time_t _endTime;

    std::map<Variable*,unsigned int> _outVarIndices;

    std::map<dsm_sample_id_t,std::vector<unsigned int> > _inmap;

    std::map<dsm_sample_id_t,std::vector<unsigned int> > _lenmap;

    std::map<dsm_sample_id_t,std::vector<unsigned int> > _outmap;

    unsigned int _ndataValues;

    double *_sums;

    int *_cnts;

    SampleAverager(const SampleAverager&);

    SampleAverager& operator=(const SampleAverager&);

};

}}  // namespace nidas namespace core

#endif