/***************************************************************************

 * Copyright (c) 2009 Enrico Ros                                           *

 *         2009 Enrico Ros <enrico.ros@email.it>                           *

 *         2009 Alberto Scarpa <skaal.sl@gmail.com>                        *

 *                                                                         *

 * Permission is hereby granted, free of charge, to any person             *

 * obtaining a copy of this software and associated documentation          *

 * files (the "Software"), to deal in the Software without                 *

 * restriction, including without limitation the rights to use,            *

 * copy, modify, merge, publish, distribute, sublicense, and/or sell       *

 * copies of the Software, and to permit persons to whom the               *

 * Software is furnished to do so, subject to the following                *

 * conditions:                                                             *

 *                                                                         *

 * The above copyright notice and this permission notice shall be          *

 * included in all copies or substantial portions of the Software.         *

 *                                                                         *

 ***************************************************************************/

#include "WCGame.h"

#include "Scrambler.h"

#include "ScreenCapture.h"

#include "InputUtils.h"

#include "ocr/Ocr.h"

#include "ui_AppWidget.h"

#include <QTimer>

#include <QDebug>

#include <QPainter>

//BEGIN portableMSleep

#ifdef Q_OS_UNIX

#include <unistd.h>

void portableMSleep( int ms ) {

    ::usleep( ms * 1000 );

}

#else

#ifdef Q_OS_WIN32

#include <windows.h>

void portableMSleep( int ms ) {

    Sleep( ms );

}

#endif

#endif

//END portableMSleep

//static const int WC_CAP_L = 114;

//static const int WC_CAP_T = 245;

static const int WC_BASELINE_H[] = { 7, 61, 116, 170, 224, 279 };

static const int WC_BASELINE_V[] = { 3,  3,   3,   3,   3,   3 };

static const int WC_LETTER_W = 35;

static const int WC_LETTER_H = 37;

WCGame::WCGame( QObject * parent )

    : AbstractGame( parent )

    , m_wordsDictionary( new Scrambler() )

    , m_namesDictionary( new Scrambler() )

{

    // load the dictionaries

    m_wordsDictionary->addDictionary( "dic-it.txt" );

    m_namesDictionary->addDictionary( "dic-enrico.txt" );

}

WCGame::~WCGame()

{

    delete m_wordsDictionary;

    delete m_namesDictionary;

}

QPixmap WCGame::highlightPixmap( const QPixmap & pixmap ) const

{

    QPixmap pix = pixmap;

    QPainter pp( &pix );

    pp.setPen( Qt::red );

    for ( int i = 0; i < 6; i++ )

        pp.drawRect( WC_BASELINE_H[ i ], WC_BASELINE_V[ i ], WC_LETTER_W - 1, WC_LETTER_H - 1 );

    pp.end();

    return pix;

}

void WCGame::train( Ocr * ocr, QString lettersText, const QImage & gamePixmap ) const

{

    for ( int i = 0; i < 6; i++ ) {

        QImage letterImage = gamePixmap.copy( WC_BASELINE_H[ i ], WC_BASELINE_V[ i ], WC_LETTER_W, WC_LETTER_H );

        QChar character = lettersText.at( i );

        letterImage.save( QString( "wc-glyphs/glyph_%1.png" ).arg( character.toLatin1() ), "PNG" );

        ocr->trainGlyph( letterImage, character );

    }

}

void WCGame::run( Ui::AppWidgetClass * ui, const ScreenCapture * capture, const Ocr * ocr )

{

    if ( m_time.isNull() )

        m_time.start();

    if ( m_time.elapsed() > (60 * 60 * 1000) )

        return;

    // OCR screen -> 6Chars + mean color identification

    QString letters;

    QImage gamePixmap = capture->lastPixmap().toImage();

    int meanR = 0, meanG = 0, meanB = 0;

    for ( int i = 0; i < 6; i++ ) {

        // accumulate intermediate color (yellow for normal, pink for name)

        if ( i > 1 ) {

            int sampleX = (WC_BASELINE_H[ i ] + WC_BASELINE_H[ i - 1 ] + WC_LETTER_W) >> 1;

            int sampleY = (WC_BASELINE_V[ i ] + WC_LETTER_H) >> 1;

            QRgb sampleColor = gamePixmap.pixel( sampleX, sampleY );

            meanR += qRed( sampleColor );

            meanG += qGreen( sampleColor );

            meanB += qBlue( sampleColor );

        }

        QImage letterImage = gamePixmap.copy( WC_BASELINE_H[ i ], WC_BASELINE_V[ i ], WC_LETTER_W, WC_LETTER_H );

        OcrResult res = ocr->recognizeGlyph( letterImage );

        if ( res.confidence > 0.1 )

            letters.append( res.character.toLower() );

        //qWarning() << res.character << res.confidence;

    }

    ui->currentLetters->setText( letters );

    if ( letters.length() != 6 )

        return;

    // select the dictionary based on the game type

    qreal meanHue = QColor( meanR / 5, meanG / 5, meanB / 5 ).hueF();

    Scrambler * scrambler = (meanHue > 0.89 && meanHue < 0.99) ? m_namesDictionary : m_wordsDictionary;

    // Do Anagram

    QStringList words = scrambler->words( letters, ui->minLetters->value(), ui->maxLetters->value() );

    ui->anagramWords->setText(QString::number(words.size()));

    // Write Words

    InputUtils::mouseMove( capture->geometry().center() );

    InputUtils::mouseLeftClick();

    portableMSleep( 200 );

    int count = words.size();

    for ( int i = 0; i < count; i++ ) {

        InputUtils::keyWrite( words.at( i ) + '\n' );

        portableMSleep( 10 );

    }

    InputUtils::keyClickSpecial( Qt::Key_Control );

}