/*
 * Copyright 2017, Adrien Destugues, pulkomandy@pulkomandy.tk
 * Distributed under terms of the MIT license.
 */


#include <Application.h>
#include <Bitmap.h>
#include <File.h>
#include <MediaFile.h>
#include <MediaTrack.h>
#include <Window.h>

#include <stdio.h>


#pragma mark - VideoView


class VideoView: public BView
{
	public:
		VideoView(BMediaTrack* track, int32 width, int32 height);
		void Draw(BRect);
		void KeyDown(const char*, int32);

	private:
		BMediaTrack*	fMediaTrack;
		BBitmap			fBitmap;
};


VideoView::VideoView(BMediaTrack* track, int32 width, int32 height)
	: BView(BRect(0, 0, width, height), "Video", B_FOLLOW_NONE,
		B_WILL_DRAW)
	, fMediaTrack(track)
	, fBitmap(BRect(0, 0, width - 1, height - 1), B_RGB32)
{
}


void
VideoView::Draw(BRect r)
{
	DrawBitmap(&fBitmap);
}
		

void
VideoView::KeyDown(const char*, int32)
{
	puts("Next frame");
	int64 count = 1;
	fMediaTrack->ReadFrames(fBitmap.Bits(), &count);
	Invalidate();
}


#pragma mark - VideoWindow


class VideoWindow: public BWindow
{
public:
	VideoWindow(const char* videoFile);
	~VideoWindow();

private:
	BFile*			fFile;
	BMediaFile*		fMediaFile;
	BMediaTrack*	fMediaTrack;
};


VideoWindow::VideoWindow(const char* path)
	: BWindow(BRect(60, 120, 700, 600), "Video Decoder",
		B_DOCUMENT_WINDOW, B_QUIT_ON_WINDOW_CLOSE)
	, fMediaTrack(NULL)
{
	fFile = new BFile(path, B_READ_ONLY);
	fMediaFile = new BMediaFile(fFile);
	media_format format;

	int i;
	for (i = fMediaFile->CountTracks(); --i >= 0;) {
		BMediaTrack* mediaTrack = fMediaFile->TrackAt(i);

		mediaTrack->EncodedFormat(&format);
		if (format.IsVideo()) {
			fMediaTrack = mediaTrack;
			fMediaTrack->DecodedFormat(&format);
			break;
		}
	}

	if (fMediaTrack) {
		printf("Found video track\n%ld x %ld ; %lld frames, %f seconds\n",
			format.Width(), format.Height(), fMediaTrack->CountFrames(),
			fMediaTrack->Duration() / 1000000.f);
	} else {
		return;
	}

	BView* view = new VideoView(fMediaTrack, format.Width(), format.Height());
	AddChild(view);
	view->MakeFocus();
}


VideoWindow::~VideoWindow()
{
	delete fMediaFile; // Also deletes the track
	delete fFile;
}


#pragma mark - Main


int main(int argc, char* argv[])
{
	if (argc < 2)
		return -1;

	BApplication app("application/x-vnd.Haiku-VideoDecoder");

	BWindow* window = new VideoWindow(argv[1]);
	window->Show();

	app.Run();
}