TestData.h

Go to the documentation of this file.

/*
 * Copyright (C) 2015 Niek Linnenbank
 *
 * 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 3 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.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
 
#ifndef __LIBTEST_TESTDATA_H
#define __LIBTEST_TESTDATA_H
 
#include <Types.h>
#include <Macros.h>
#include <Vector.h>
#include <stdlib.h>
#include <unistd.h>
 
#ifdef __HOST__
#include <sys/time.h>
#else
#include <FreeNOS/System.h>
#endif /* __HOST__ */
 
template <class T> class TestData
{
  public:
 
    TestData()
    {
        seed();
    }
 
    virtual ~TestData()
    {
    }
 
    void seed()
    {
        // Collect seed from process id.
        pid_t pid = getpid();
        unsigned int seed = pid;
 
        // Collect seed from the current time
#ifdef __HOST__
        struct timeval tv;
        gettimeofday(&tv, (struct timezone *)NULL);
        seed += tv.tv_sec + tv.tv_usec;
#else
        seed += timestamp();
#endif
        // Seed the random generator
        srandom(seed);
    }
 
    Size count() const
    {
        return m_values.count();
    }
 
    T & get(Size index)
    {
        return m_values[index];
    }
 
    T & operator[](Size index)
    {
        return m_values[index];
    }
 
    virtual T random(Size count = 1) = 0;
 
    virtual T unique(Size count = 1) = 0;
 
  protected:
 
    Vector<T> m_values;
};
 
#endif /* __LIBTEST_TESTDATA_H */