//----------------------------------------------------------------------------
//
//  ise_test_15.cpp
//
//  Authors: Joe Jarchow, Geoffrey Griffith, Joseph Kadhim, Shinya Daigaku
//           Andrew Pouzeshi
//
//  Sponsor: Tom Lookabaugh, Assistant Professor of Computer Science
//           University of Colorado
//
//  Senior Project: Team ISE (Image Selective Encryption)
//----------------------------------------------------------------------------


// This file is designed to test the ISE production code with test15.sh

#include <stdlib.h>     
#include <iostream>
#include <fstream> 
#include <string>
#include "rijndael-api-fst.h"
#include "ise.h"

using namespace std;

int main(int argc, char * argv[])
{
    /* Encrypt_File Function Normal Use */
    size_t length = 0;
    int err = 0;
    char * MyKey;
    char * MyJpeg;
    char * MyISE;	

    // Set the key
    if (argc > 0)
    {
        length = strlen(argv[1]);
        MyKey = new char [length + 1];
        strcpy (MyKey,argv[1]);
    }
    else
    {
        length = strlen("Bad_Key");
        MyKey = new char [length + 1];
        strcpy (MyKey,"Bad_Key");
    }

    // Set JPEG file name
    if (argc > 1)
    {
        length = strlen(argv[2]);
        MyJpeg = new char [length + 1];
        strcpy (MyJpeg,argv[2]);
    }
    else
    {
        length = strlen("images/ralphie.jpg");
        MyJpeg = new char [length + 1];
        strcpy (MyJpeg,"images/ralphie.jpg");
    }

    // Set ISE name
    if (argc > 2)
    {
        length = strlen(argv[3]);
        MyISE = new char [length + 1];
        strcpy (MyISE,argv[3]);
    }
    else
    {
      length = strlen("images/ralphie.ise");
      MyISE = new char [length + 1];
      strcpy (MyISE, "images/ralphie.ise");
    }
    /* Call the constructor with key, jpeg file, and ise file */
    jpeg_ise encrypt_test(MyKey, MyJpeg, MyISE );


    /* Encryption */
    err = encrypt_test.encrypt_file();
    if (err != 0) cout << "Encryption failed on file: " << MyJpeg << "\n" ;
	
    delete [] MyKey;
    delete [] MyISE;
    delete [] MyJpeg;

    return 0;
}
