FFMPEG 2

/* -LICENSE-START-
** Copyright (c) 2013 Blackmagic Design
**
** Permission is hereby granted, free of charge, to any person or organization
** obtaining a copy of the software and accompanying documentation covered by
** this license (the "Software") to use, reproduce, display, distribute,
** execute, and transmit the Software, and to prepare derivative works of the
** Software, and to permit third-parties to whom the Software is furnished to
** do so, all subject to the following:
**
** The copyright notices in the Software and this entire statement, including
** the above license grant, this restriction and the following disclaimer,
** must be included in all copies of the Software, in whole or in part, and
** all derivative works of the Software, unless such copies or derivative
** works are solely in the form of machine-executable object code generated by
** a source language processor.
**
** 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, TITLE AND NON-INFRINGEMENT. IN NO EVENT
** SHALL THE COPYRIGHT HOLDERS OR ANYONE DISTRIBUTING THE SOFTWARE BE LIABLE
** FOR ANY DAMAGES OR OTHER LIABILITY, WHETHER IN CONTRACT, TORT OR OTHERWISE,
** ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
** DEALINGS IN THE SOFTWARE.
** -LICENSE-END-
*/


#include "stdafx.h"
#include <gl\gl.h>
#include <gl\freeglut.h>
#include "PreviewWindow.h"
#include <stdio.h>
#include <string>
#include <cstring>
#include <sstream>
#include <iostream>

FILE* ffmpeg;
int* buffer;
int viewWidth2;
int viewHeight2;
bool isCapturing = FALSE;
bool valueSet = FALSE;

PreviewWindow::PreviewWindow()
: m_deckLinkScreenPreviewHelper(NULL), m_refCount(1), m_previewBox(NULL), m_previewBoxDC(NULL), m_openGLctx(NULL)
{

}


PreviewWindow::~PreviewWindow()
{
    if (m_deckLinkScreenPreviewHelper != NULL)
    {
        m_deckLinkScreenPreviewHelper->Release();
        m_deckLinkScreenPreviewHelper = NULL;
    }

    if (m_openGLctx != NULL)
    {
        wglDeleteContext(m_openGLctx);
        m_openGLctx = NULL;
    }

    if (m_previewBoxDC != NULL)
    {
        m_previewBox->ReleaseDC(m_previewBoxDC);
        m_previewBoxDC = NULL;
    }
}

bool        PreviewWindow::init(CStatic *previewBox)
{
    m_previewBox = previewBox;

    // Create the DeckLink screen preview helper
    if (CoCreateInstance(CLSID_CDeckLinkGLScreenPreviewHelper, NULL, CLSCTX_ALL, IID_IDeckLinkGLScreenPreviewHelper, (void**)&m_deckLinkScreenPreviewHelper) != S_OK)
        return false;

    // Initialise OpenGL
    return initOpenGL();
}

bool        PreviewWindow::initOpenGL()
{
    PIXELFORMATDESCRIPTOR   pixelFormatDesc;
    int                     pixelFormat;
    bool                    result = false;

    //
    // Here, we create an OpenGL context attached to the screen preview box
    // so we can use it later on when we need to draw preview frames.

    // Get the preview box drawing context
    m_previewBoxDC = m_previewBox->GetDC();
    if (m_previewBoxDC == NULL)
        return false;

    // Ensure the preview box DC uses ARGB pixel format
    ZeroMemory(&pixelFormatDesc, sizeof(pixelFormatDesc));
    pixelFormatDesc.nSize = sizeof(pixelFormatDesc);
    pixelFormatDesc.nVersion = 1;
    pixelFormatDesc.dwFlags = PFD_DRAW_TO_WINDOW | PFD_SUPPORT_OPENGL;
    pixelFormatDesc.iPixelType = PFD_TYPE_RGBA;
    pixelFormatDesc.cColorBits = 32;
    pixelFormatDesc.cDepthBits = 16;
    pixelFormatDesc.cAlphaBits = 8;
    pixelFormatDesc.iLayerType = PFD_MAIN_PLANE;
    pixelFormat = ChoosePixelFormat(m_previewBoxDC->m_hDC, &pixelFormatDesc);
    if (SetPixelFormat(m_previewBoxDC->m_hDC, pixelFormat, &pixelFormatDesc) == false)
        return false;

    // Create OpenGL rendering context
    m_openGLctx = wglCreateContext(m_previewBoxDC->m_hDC);
    if (m_openGLctx == NULL)
        return false;
    // Make the new OpenGL context the current rendering context so
    // we can initialise the DeckLink preview helper
    if (wglMakeCurrent(m_previewBoxDC->m_hDC, m_openGLctx) == FALSE)
        return false;

    if (m_deckLinkScreenPreviewHelper->InitializeGL() == S_OK)
        result = true;

    // Reset the OpenGL rendering context
    wglMakeCurrent(NULL, NULL);

    return result;
}

HRESULT         PreviewWindow::DrawFrame(IDeckLinkVideoFrame* theFrame)
{
    GLfloat width = theFrame->GetWidth();
    GLfloat height = theFrame->GetHeight();
    GLfloat aspect = width / height;
    int viewportY = 0;
    int viewportX = 0;
    CRect rect;

    m_previewBox->GetClientRect(&rect);

    int previewWidth = rect.Width();
    int previewHeight = rect.Height();

    int viewWidth = previewWidth;
    int viewHeight = previewWidth / aspect;

    if (viewHeight > previewHeight) {
        viewHeight = previewHeight;
        viewWidth = previewHeight * aspect;
    }
    viewWidth2 = theFrame->GetWidth();
    viewHeight2 = theFrame->GetHeight();

    viewportY = (previewHeight - viewHeight) / 2;
    viewportX = (previewWidth - viewWidth) / 2;

    // Make sure we are initialised
    if ((m_deckLinkScreenPreviewHelper == NULL) || (m_previewBoxDC == NULL) || (m_openGLctx == NULL))
        return E_FAIL;

    // First, pass the frame to the DeckLink screen preview helper
    m_deckLinkScreenPreviewHelper->SetFrame(theFrame);


    // Then set the OpenGL rendering context to the one we created before
    wglMakeCurrent(m_previewBoxDC->m_hDC, m_openGLctx);

    //set the viewport X/Y coordinates + letter-/pillarbox

    glViewport(viewportX, viewportY, viewWidth, viewHeight);

    if (isCapturing){
        glReadPixels(0, 0, viewWidth2, viewHeight2, GL_RGBA, GL_UNSIGNED_BYTE, buffer);
        fwrite(buffer, sizeof(int)*viewWidth2 * viewHeight2, 1, ffmpeg);
    }
    // and let the helper take care of the drawing
    m_deckLinkScreenPreviewHelper->PaintGL();

    // Last, reset the OpenGL rendering context
    wglMakeCurrent(NULL, NULL);

    return S_OK;
}

void PreviewWindow::startCapture(){
    if (!isCapturing){
        isCapturing = TRUE;

        char cmd[1024];
        sprintf(cmd, "ffmpeg -r 25 -f rawvideo -pix_fmt rgba -s %dx%d -i - "
            "-threads 0 -preset fast -y -pix_fmt yuv420p -crf 21 -vf vflip output.mp4",
            viewWidth2, viewHeight2);


        // open pipe to ffmpeg's stdin in binary write mode
        ffmpeg = _popen(cmd, "wb");

        buffer = new int[viewWidth2 * viewHeight2];
        //glutSwapBuffers();
    }
    else{
        isCapturing = FALSE;
        _pclose(ffmpeg);
    }
}