#include <cppunit/TestFixture.h>
#include <cppunit/TestSuite.h>
#include <cppunit/extensions/HelperMacros.h>

#include <map>
#include <vector>

#include "mp_CERT.h"
#include "PluginController.h"
#include "SVMPlugin.h"

class mpt_RutgersData;

class RutgersTest : public CppUnit::TestFixture
{
    CPPUNIT_TEST_SUITE( RutgersTest );
    CPPUNIT_TEST( testAU1 );
    CPPUNIT_TEST( testAU2 );
    CPPUNIT_TEST( testAU4 );
    CPPUNIT_TEST( testAU5 );
    CPPUNIT_TEST( testAU10 );
    CPPUNIT_TEST( testAU12 );
    CPPUNIT_TEST( testAU14 );
    CPPUNIT_TEST( testAU20 );
    CPPUNIT_TEST_SUITE_END();

public:
    RutgersTest();

	void setUp();
	void tearDown();

    void testAU1();
    void testAU2();
    void testAU4();
    void testAU5();
    void testAU10();
    void testAU12();
    void testAU14();
    void testAU20();

private:
    bool calcAUs( mpt_RutgersData &data, unsigned int frame, double &auValue, bool present );
    void calcAUsForData( mpt_RutgersData &data, std::vector<double> &auPresent, std::vector<double> &auNotPresent );
    bool dprimeForData( std::vector<double> &auPresent, std::vector<double> &auNotPresent, double &dPrime );
    bool areaUnderROC( std::vector<double> &auPresent, std::vector<double> &auNotPresent, double &dPrime );
    double AROCerr( double AUR, size_t numPresent, size_t numNotPresent );

    void initTestData();
    void printTestData();
    void testAU( unsigned int au );

    CERT::PluginController *pc;
    CERT::SVMPlugin *plugin;
    MP_CERT cert;

    std::map<unsigned int, std::vector<mpt_RutgersData> > auTestData;
};