libmv-capi-sys 0.1.3

/****************************************************************************
**
** Copyright (c) 2011 libmv authors.
**
** 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.
**
** THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
** IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
** FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
** AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
** LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
** FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
** IN THE SOFTWARE.
**
****************************************************************************/

#include "ui/tracker/clip.h"
#include <libmv/image/convolve.h> //LaplaceFilter

#ifdef USE_FFMPEG
extern "C" {
#include <stdint.h>
typedef uint64_t UINT64_C;
#include <libavformat/avformat.h>
#include <libavcodec/avcodec.h>
}
#endif

#include <QFileInfo>
#include <QDir>
#include <QVBoxLayout>
#include <QPushButton>
#include <QProgressDialog>

#if (QT_VERSION < QT_VERSION_CHECK(4, 7, 0))
#define constBits bits
#endif

FileDialog::FileDialog(QWidget* parent,QString caption) : QDialog(parent) {
    setWindowTitle(caption);
    view.setModel(&fileSystem);
    view.setHeaderHidden(true);
    view.setSelectionMode(QAbstractItemView::ExtendedSelection);
    for(int i=1;i<fileSystem.columnCount();i++) view.setColumnHidden(i,true);
    QVBoxLayout* layout = new QVBoxLayout(this);
    layout->addWidget(&view);
    view.setRootIndex(fileSystem.setRootPath(QDir::rootPath()));
    view.resizeColumnToContents(0); setMinimumHeight(600); setMinimumWidth(view.columnWidth(0));
    QPushButton* openButton = new QPushButton(QIcon::fromTheme("document-open"),"Open");
    connect(openButton,SIGNAL(clicked()),this,SLOT(accept()));
    layout->addWidget(openButton);
}

QStringList FileDialog::selectedFiles() {
    QStringList files;
    foreach(QModelIndex index, view.selectionModel()->selectedRows(0) )
        files << index.data(QFileSystemModel::FilePathRole).toString();
    return files;
}

Clip::Clip(QStringList files) {
  images_.clear();
  if ( files.count() == 1 ) {
    if (QFileInfo(files.first()).isDir()) {
      QString folder = files.takeFirst();
      cache_.setFileName(folder+"/cache");
      foreach (QString file, QDir(folder,"*.jpg *.png")
               .entryList(QDir::Files, QDir::Name)) {
        files << QDir(folder).filePath(file);
      }
    } else {
      cache_.setFileName(files.first()+".cache");
    }
  } else {
    cache_.setFileName(QFileInfo(files.first()).dir().path()+"/cache");
  }
  QFile meta(cache_.fileName()+".meta");
  if(!cache_.exists() || !meta.exists()) {
    // Decode to raw on-disk video cache
    cache_.open(QFile::WriteOnly|QFile::Truncate);
    if (files.count() == 1) {
      DecodeVideo(files.first());
    } else {
      DecodeSequence(files);
    }
    cache_.close();
    meta.open(QFile::WriteOnly|QFile::Truncate);
    meta.write((char*)&size_,sizeof(size_));
    meta.close();
  }
  meta.open(QFile::ReadOnly);
  meta.read((char*)&size_, sizeof(size_));
  cache_.open(QFile::ReadOnly);
  int image_size = size_.width()*size_.height();
  images_.resize( cache_.size()/image_size );
  uchar* data = cache_.map(0, cache_.size());
  for(int i = 0; i < images_.count(); i++, data+=image_size) {
    images_[i] = QImage(data,size_.width(),size_.height(),QImage::Format_Indexed8);
  }
}

void Clip::DecodeSequence(QStringList files) {
  QProgressDialog progress("Caching "+QFileInfo(files.first()).dir().path(),
                           "Abort",0,files.count());
  progress.setWindowModality(Qt::WindowModal);
  int i=0;
  foreach (QString file, files) {
    QImage image(file);
    if(i==0) size_ = image.size();
    if(image.size() != size_) image.scaled(size_,Qt::IgnoreAspectRatio,Qt::SmoothTransformation);
    if(image.depth() != 8) {
      QImage grayscale(image.width(),image.height(),QImage::Format_Indexed8);
      QRgb *src = (QRgb*)image.constBits();
      uchar *dst = grayscale.bits();
      for(int i = 0; i < grayscale.byteCount(); i++) dst[i] = qGray(src[i]);
      image = grayscale;
    }
    cache_.write((const char*)image.constBits(),image.byteCount());
    progress.setValue(i++);
    if (progress.wasCanceled()) {
      break;
    }
  }
}

void Clip::DecodeVideo(QString path) {
#ifdef USE_FFMPEG
  av_register_all();
  AVFormatContext* file = 0;
#if LIBAVFORMAT_VERSION_MAJOR >= 53
  if(avformat_open_input(&file, path.toUtf8(), 0, 0)) return;
#else
  if(av_open_input_file(&file, path.toUtf8(), 0, 0, 0)) return;
#endif
  avformat_find_stream_info(file, NULL);
  int video_stream = 0;
  AVCodecContext* video = 0;
  for (int i = 0; i < (int)file->nb_streams; i++ ) {
    if( file->streams[i]->codec->codec_type == AVMEDIA_TYPE_VIDEO) {
      video_stream = i;
      video = file->streams[i]->codec;
      AVCodec* codec = avcodec_find_decoder(video->codec_id);
      if( codec ) avcodec_open2(video, codec, NULL);
      break;
    }
  }
  QProgressDialog progress("Caching "+path+"("+QString::number(video->frame_number)+")","Abort",0,video->frame_number);
  progress.setWindowModality(Qt::WindowModal);
  int i=0;
  for (AVPacket packet; av_read_frame(file, &packet) >= 0; ) {
    if ( packet.stream_index == video_stream ) {
      AVFrame* frame = av_frame_alloc();
      int complete_frame = 0;
      avcodec_decode_video2(video, frame, &complete_frame, &packet);
      if (complete_frame) {
        size_ = QSize(video->width,video->height);
        // FIXME: Assume planar format
        const uchar* data = frame->data[0];
        if( frame->linesize[0] != video->width ) {
          for(int y = 0; y < video->height; y++) {
            cache_.write((const char*)data,video->width);
            data += frame->linesize[0];
          }
        } else {
          cache_.write((const char*)data,video->width*video->height);
        }
        av_free(frame);
      }
    }
    av_free_packet(&packet);
    progress.setValue(i++);
    if (progress.wasCanceled()) {
        break;
    }
  }
  avcodec_close(video);
  avformat_close_input(&file);
#endif
}

int Clip::Count() {
  return images_.count();
}

QSize Clip::Size() {
  return size_;
}

QImage Clip::Image(int i) {
  QImage src=images_[i];
  QImage dst(src.width(), src.height(), QImage::Format_Indexed8);
  //TODO: UI settings for Laplace filter.
  libmv::LaplaceFilter((unsigned char*)src.constBits(), dst.bits(), src.width(), src.height(), 0);
  return dst;
}