~ubuntu-branches/ubuntu/natty/qtpfsgui/natty

 */

 

#include <getopt.h>

#include "options.h"

#include "config.h"

#include "../HdrCreation/createhdr.h"

#include "commandline.h"

#include "../Threads/loadHdrThread.h"

#include "../Threads/tonemapperThread.h"

#include "../Exif/exif_operations.h"

void writeRGBEfile (pfs::Frame* inputpfsframe, const char* outfilename);

void writeEXRfile  (pfs::Frame* inputpfsframe, const char* outfilename);

 

#define error(Z) { fprintf(stderr,Z); exit(1); }

#else

#include <error.h>

#define error(Z) error(1,0,Z);

#endif

 

 

static struct option cmdLineOptions[] = {

        { "verbose", no_argument, NULL, 'v' },

CommandLineInterfaceManager::CommandLineInterfaceManager(const int argc, char **argv) : argc(argc), argv(argv) {

        hdrCreationManager=NULL;

        align_mode=NO_ALIGN;

        connect(this,SIGNAL(startParsing()),this,SLOT(parseArgs()),Qt::QueuedConnection);

        emit startParsing();

}

        }

        for (int index = optind; index < argc; index++) {

                inputFiles << QString(argv[index]);

        }

 

        if (!ev.isEmpty() && ev.count()!=inputFiles.count())

                error(qPrintable(tr("Error: The number of EV values specified is different from the number of input files.")));

 

        //now validate operation mode.

                operation_mode=CREATE_HDR_MODE;

                operation_mode=LOAD_HDR_MODE;

                printHelp(argv[0]);

                error("Wrong combination of parameters.");

        }

 

        if (operation_mode==CREATE_HDR_MODE) {

                connect(hdrCreationManager,SIGNAL(finishedLoadingInputFiles(QStringList)),this, SLOT(finishedLoadingInputFiles(QStringList)));

                connect(hdrCreationManager,SIGNAL(errorWhileLoading(QString)),this, SLOT(errorWhileLoading(QString)));

                connect(hdrCreationManager, SIGNAL(finishedAligning()), this, SLOT(createHDR()));

                hdrCreationManager->loadInputFiles();

        }

        else {

                connect(loadthread, SIGNAL(finished()), loadthread, SLOT(deleteLater()));

                connect(loadthread, SIGNAL(hdr_ready(pfs::Frame*,QString)), this, SLOT(loadFinished(pfs::Frame*,QString)));

                connect(loadthread, SIGNAL(load_failed(QString)), this, SLOT(errorWhileLoading(QString)));

                loadthread->start();

        }

}

 

        HDR=hdr;

        saveHDR();

}

 

void CommandLineInterfaceManager::finishedLoadingInputFiles(QStringList filesLackingExif) {

        if (filesLackingExif.size()!=0) {

                        for(int i=0; i<ev.size(); i++)

                                hdrCreationManager->setEV(ev.at(i),i);

                        error(qPrintable(tr("Error: Exif data missing in images and EV values not specifed on the commandline, bailing out.")));

        }

        hdrCreationManager->checkEVvalues();

        exit(1);

}

void CommandLineInterfaceManager::ais_failed(QProcess::ProcessError) {

}

 

void CommandLineInterfaceManager::createHDR() {

        hdrCreationManager->removeTempFiles();

        HDR=hdrCreationManager->createHdr(false,1);

        saveHDR();

}

 

void CommandLineInterfaceManager::saveHDR() {

        if (!saveHdrFilename.isEmpty()) {

                QFileInfo qfi(saveHdrFilename);

                if (qfi.suffix().toUpper()=="EXR") {

                        writeEXRfile(HDR,qfi.filePath().toUtf8().constData());

                        pfsio.writeFrame(HDR,qfi.filePath());

                        HDR->convertXYZChannelsToRGB();

                } else {

                }

        }

 

        startTonemap();

 

void  CommandLineInterfaceManager::startTonemap() {

        if (!saveLdrFilename.isEmpty()) {

                //now check if user wants to resize (create thread with either -2 or true original size as first argument in ctor, see options.cpp).

                int origxsize= (tmopts->xsize==-2) ? -2 : HDR->getWidth();

                TonemapperThread *thread = new TonemapperThread(origxsize, *tmopts);

                connect(thread, SIGNAL(ImageComputed(const QImage&,tonemapping_options*)), this, SLOT(tonemapTerminated(const QImage&,tonemapping_options*)));

        } else {

                emit finishedParsing();

        }

}

 

void CommandLineInterfaceManager::tonemapTerminated(const QImage& newimage,tonemapping_options*) {

        QFileInfo qfi(saveLdrFilename);

        if (!newimage.save(saveLdrFilename, qfi.suffix().toAscii().constData(), 100)) {

                error(qPrintable(tr("ERROR: Cannot save to file: %1").arg(saveLdrFilename)));

        } else {

                TMOptionsOperations operations(tmopts);

        }

        emit finishedParsing();

}

        -e --ev EV1,EV2,...    Specify numerical EV values (as many as INPUTFILES).\n \

        -c --config            HDR creation config. Possible values: \n\

                weight=triangular|gaussian|plateau:response_curve=from_file|linear|gamma|log|robertson:model=robertson|debevec:curve_filename=your_file_here.m \n\

        -l --load HDR_FILE     Load an HDR instead of creating a new one. \n \

                ashikhmin|drago|durand|fattal|pattanaik|reinhard02|reinhard05|mantiuk\n \

                (Default is mantiuk)\n \

                alpha=VALUE:beta=VALUE:color=VALUE:noise=VALUE:new=true|false (for fattal)\n\

                contrast=VALUE:saturation=VALUE:equalization=true|false (for mantiuk)\n\

                localcontrast=VALUE:eq=2|4:simple=true|false (for ashikhmin)\n\

                local=true|false:autolum=true|false:cone=VALUE:rod=VALUE:multiplier=VALUE (for pattanaik)\n\

                scales=true|false:key=VALUE:phi=VALUE:num=VALUE:low=VALUE:high=VALUE (for reinhard02)\n\

                brightness=VALUE:chroma=VALUE:lightness=VALUE (for reinhard05)\n\

        -o --output LDR_FILE   File name you want to save your tone mapped LDR to.\n \

        \n\

        fprintf(stderr,qPrintable(help));

}