#include <iostream>

#include "StreamManager.hpp"
#include "ImageManager.hpp"
#include "FaultManager.hpp"

using namespace std;

int main(int argc, char **argv)
{
    StreamManager stream_mgr;
    ImageManager image_mgr;
    FaultManager fault_mgr;
    PvString lConnectionID;

    // If running in test mode
    if (argc > 1 && !strcmp(argv[1], "test"))
    {
        while(1)
        {
            if (!image_mgr.ProcessImage(0, 0, 0, fault_mgr, true))
                break;
        }
        return 0;
    }
    
    // If no devies are found, exit.
    if (stream_mgr.FindDevice(&lConnectionID).IsFailure())
    {
        return -1;
    }

    if (stream_mgr.ConnectToDevice(lConnectionID).IsFailure())
    {
        return -1;
    }

    stream_mgr.OpenStream(lConnectionID);
    
    stream_mgr.AcquireImages(image_mgr, fault_mgr);

    stream_mgr.Disconnect();

    return 0;
}