obs/Source/BitmapImageSource.cpp

/********************************************************************************
 Copyright (C) 2012 Hugh Bailey <obs.jim@gmail.com>

 This program is free software; you can redistribute it and/or modify
 it under the terms of the GNU General Public License as published by
 the Free Software Foundation; either version 2 of the License, or
 (at your option) any later version.

 This program is distributed in the hope that it will be useful,
 but WITHOUT ANY WARRANTY; without even the implied warranty of
 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 GNU General Public License for more details.

 You should have received a copy of the GNU General Public License
 along with this program; if not, write to the Free Software
 Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307, USA.
********************************************************************************/


#include "Main.h"
#include "libnsgif.h"


void *def_bitmap_create(int width, int height)          {return Allocate(width * height * 4);}
void def_bitmap_set_opaque(void *bitmap, BOOL opaque)   {}
BOOL def_bitmap_test_opaque(void *bitmap)               {return false;}
unsigned char *def_bitmap_get_buffer(void *bitmap)      {return (unsigned char*)bitmap;}
void def_bitmap_destroy(void *bitmap)                   {Free(bitmap);}
void def_bitmap_modified(void *bitmap)                  {return;}

gif_bitmap_callback_vt bitmap_callbacks =
{
    def_bitmap_create,
    def_bitmap_destroy,
    def_bitmap_get_buffer,
    def_bitmap_set_opaque,
    def_bitmap_test_opaque,
    def_bitmap_modified
};


class BitmapImageSource : public ImageSource
{
    Texture  *texture;

    Vect2    fullSize;
    XElement *data;

    DWORD opacity;
    DWORD color;

    bool bIsAnimatedGif;
    gif_animation gif;
    LPBYTE lpGifData;
    List<float> animationTimes;
    UINT curFrame, curLoop;
    float curTime;

    OSFileChangeData *changeMonitor;

    void CreateErrorTexture()
    {
        LPBYTE textureData = (LPBYTE)Allocate(32*32*4);
        msetd(textureData, 0xFF0000FF, 32*32*4);

        texture = CreateTexture(32, 32, GS_RGB, textureData, FALSE);
        fullSize.Set(32.0f, 32.0f);

        Free(textureData);
    }

public:
    BitmapImageSource(XElement *data)
    {
        this->data = data;
        UpdateSettings();

        Log(TEXT("Using bitmap image"));
    }

    ~BitmapImageSource()
    {
        if(bIsAnimatedGif)
            gif_finalise(&gif);

        if(lpGifData)
            Free(lpGifData);

        if (changeMonitor)
            OSMonitorFileDestroy(changeMonitor);

        delete texture;
    }

    void Tick(float fSeconds)
    {
        if(bIsAnimatedGif)
        {
            UINT totalLoops = (UINT)gif.loop_count;
            if(totalLoops >= 0xFFFF)
                totalLoops = 0;

            if(!totalLoops || curLoop < totalLoops)
            {
                UINT newFrame = curFrame;

                curTime += fSeconds;
                while(curTime > animationTimes[curFrame])
                {
                    curTime -= animationTimes[newFrame];
                    if(++newFrame == animationTimes.Num())
                    {
                        if(!totalLoops || ++curLoop < totalLoops)
                            newFrame = 0;
                        else if (curLoop == totalLoops)
                            break;
                    }
                }

                if(newFrame != curFrame)
                {
                    gif_decode_frame(&gif, newFrame);
                    texture->SetImage(gif.frame_image, GS_IMAGEFORMAT_RGBA, gif.width*4);

                    curFrame = newFrame;
                }
            }
        }

        if (changeMonitor && OSFileHasChanged(changeMonitor))
            UpdateSettings();
    }

    void Render(const Vect2 &pos, const Vect2 &size)
    {
        if(texture)
        {
            DWORD alpha = DWORD(double(opacity)*2.55);
            DWORD outputColor = (alpha << 24) | color&0xFFFFFF;
            DrawSprite(texture, outputColor, pos.x, pos.y, pos.x+size.x, pos.y+size.y);
        }
    }

    void UpdateSettings()
    {
        if(bIsAnimatedGif)
        {
            bIsAnimatedGif = false;
            gif_finalise(&gif);
        }

        if(lpGifData)
        {
            Free(lpGifData);
            lpGifData = NULL;
        }

        animationTimes.Clear();

        delete texture;
        texture = NULL;

        CTSTR lpBitmap = data->GetString(TEXT("path"));
        if(!lpBitmap || !*lpBitmap)
        {
            AppWarning(TEXT("BitmapImageSource::UpdateSettings: Empty path"));
            CreateErrorTexture();
            return;
        }

        //------------------------------------

        if(GetPathExtension(lpBitmap).CompareI(TEXT("gif")))
        {
            bool bFail = false;

            gif_create(&gif, &bitmap_callbacks);

            XFile gifFile;
            if(!gifFile.Open(lpBitmap, XFILE_READ, XFILE_OPENEXISTING))
            {
                AppWarning(TEXT("BitmapImageSource::UpdateSettings: could not open gif file '%s'"), lpBitmap);
                CreateErrorTexture();
                return;
            }


            DWORD fileSize = (DWORD)gifFile.GetFileSize();
            lpGifData = (LPBYTE)Allocate(fileSize);
            gifFile.Read(lpGifData, fileSize);

            gif_result result;
            do
            {
                result = gif_initialise(&gif, fileSize, lpGifData);
                if(result != GIF_OK && result != GIF_WORKING)
                {
                    bFail = true;
                    break;
                }
            }while(result != GIF_OK);

            if(gif.frame_count > 1)
            {
                if(result == GIF_OK || result == GIF_WORKING)
                    bIsAnimatedGif = true;
            }

            if(bIsAnimatedGif)
            {
                gif_decode_frame(&gif, 0);
                texture = CreateTexture(gif.width, gif.height, GS_RGBA, gif.frame_image, FALSE, FALSE);

                for(UINT i=0; i<gif.frame_count; i++)
                {
                    float frameTime = float(gif.frames[i].frame_delay)*0.01f;
                    if (frameTime == 0.0f)
                        frameTime = 0.1f;
                    animationTimes << frameTime;
                }

                fullSize.x = float(gif.width);
                fullSize.y = float(gif.height);

                curTime = 0.0f;
                curFrame = 0;
            }
            else
            {
                gif_finalise(&gif);
                Free(lpGifData);
                lpGifData = NULL;
            }
        }

        if(!bIsAnimatedGif)
        {
            texture = GS->CreateTextureFromFile(lpBitmap, TRUE);
            if(!texture)
            {
                AppWarning(TEXT("BitmapImageSource::UpdateSettings: could not create texture '%s'"), lpBitmap);
                CreateErrorTexture();
                return;
            }

            fullSize.x = float(texture->Width());
            fullSize.y = float(texture->Height());
        }

        //------------------------------------

        opacity = data->GetInt(TEXT("opacity"), 100);
        color = data->GetInt(TEXT("color"), 0xFFFFFFFF);
        if(opacity > 100)
            opacity = 100;
        else if(opacity < 0)
            opacity = 0;

        if (changeMonitor)
        {
            OSMonitorFileDestroy(changeMonitor);
            changeMonitor = NULL;
        }

        int monitor = data->GetInt(TEXT("monitor"), 0);
        if (monitor)
            changeMonitor = OSMonitorFileStart(lpBitmap);
    }

    Vect2 GetSize() const {return fullSize;}
};


ImageSource* STDCALL CreateBitmapSource(XElement *data)
{
    if(!data)
        return NULL;

    return new BitmapImageSource(data);
}

struct ConfigBitmapInfo
{
    XElement *data;
};

INT_PTR CALLBACK ConfigureBitmapProc(HWND hwnd, UINT message, WPARAM wParam, LPARAM lParam)
{
    switch(message)
    {
        case WM_INITDIALOG:
            {
                ConfigBitmapInfo *configInfo = (ConfigBitmapInfo*)lParam;
                SetWindowLongPtr(hwnd, DWLP_USER, (LONG_PTR)configInfo);
                LocalizeWindow(hwnd);

                //--------------------------

                CTSTR lpBitmap = configInfo->data->GetString(TEXT("path"));
                SetWindowText(GetDlgItem(hwnd, IDC_BITMAP), lpBitmap);

                //--------------------------

                int opacity = configInfo->data->GetInt(TEXT("opacity"), 100);
                if(opacity > 100)
                    opacity = 100;
                else if(opacity < 0)
                    opacity = 0;

                SendMessage(GetDlgItem(hwnd, IDC_OPACITY), UDM_SETRANGE32, 0, 100);
                SendMessage(GetDlgItem(hwnd, IDC_OPACITY), UDM_SETPOS32, 0, opacity);

                //--------------------------

                DWORD color = configInfo->data->GetInt(TEXT("color"), 0xFFFFFFFF);
                CCSetColor(GetDlgItem(hwnd, IDC_COLOR), color);

                //--------------------------

                int monitor = configInfo->data->GetInt(TEXT("monitor"), 0);
                SendMessage(GetDlgItem(hwnd, IDC_MONITOR), BM_SETCHECK, monitor ? BST_CHECKED : BST_UNCHECKED, 0);

                return TRUE;
            }

        case WM_COMMAND:
            switch(LOWORD(wParam))
            {
                case IDC_BROWSE:
                    {
                        TCHAR lpFile[MAX_PATH+1];
                        zero(lpFile, sizeof(lpFile));

                        OPENFILENAME ofn;
                        zero(&ofn, sizeof(ofn));
                        ofn.lStructSize = sizeof(ofn);
                        ofn.lpstrFile = lpFile;
                        ofn.hwndOwner = hwnd;
                        ofn.nMaxFile = MAX_PATH;
                        ofn.lpstrFilter = TEXT("All Formats (*.bmp;*.dds;*.jpg;*.png;*.gif)\0*.bmp;*.dds;*.jpg;*.png;*.gif\0");
                        ofn.nFilterIndex = 1;
                        ofn.Flags = OFN_PATHMUSTEXIST | OFN_FILEMUSTEXIST;

                        TCHAR curDirectory[MAX_PATH+1];
                        GetCurrentDirectory(MAX_PATH, curDirectory);

                        BOOL bOpenFile = GetOpenFileName(&ofn);
                        SetCurrentDirectory(curDirectory);

                        if(bOpenFile)
                            SetWindowText(GetDlgItem(hwnd, IDC_BITMAP), lpFile);

                        break;
                    }

                case IDOK:
                    {
                        String strBitmap = GetEditText(GetDlgItem(hwnd, IDC_BITMAP));
                        if(strBitmap.IsEmpty())
                        {
                            MessageBox(hwnd, Str("Sources.BitmapSource.Empty"), NULL, 0);
                            break;
                        }

                        ConfigBitmapInfo *configInfo = (ConfigBitmapInfo*)GetWindowLongPtr(hwnd, DWLP_USER);
                        configInfo->data->SetString(TEXT("path"), strBitmap);

                        BOOL bFailed;
                        int opacity = (int)SendMessage(GetDlgItem(hwnd, IDC_OPACITY), UDM_GETPOS32, 0, (LPARAM)&bFailed);
                        if(opacity > 100)
                            opacity = 100;
                        else if(opacity < 0)
                            opacity = 0;
                        configInfo->data->SetInt(TEXT("opacity"), bFailed ? 100 : opacity);

                        DWORD color = CCGetColor(GetDlgItem(hwnd, IDC_COLOR));
                        configInfo->data->SetInt(TEXT("color"), color);

                        int monitor = (int)SendMessage(GetDlgItem(hwnd, IDC_MONITOR), BM_GETCHECK, 0, 0);
                        if (monitor == BST_CHECKED)
                            configInfo->data->SetInt(TEXT("monitor"), 1);
                        else
                            configInfo->data->SetInt(TEXT("monitor"), 0);
                    }

                case IDCANCEL:
                    EndDialog(hwnd, LOWORD(wParam));
                    break;
            }
            break;
    }

    return 0;
}

bool STDCALL ConfigureBitmapSource(XElement *element, bool bCreating)
{
    if(!element)
    {
        AppWarning(TEXT("ConfigureBitmapSource: NULL element"));
        return false;
    }

    XElement *data = element->GetElement(TEXT("data"));
    if(!data)
        data = element->CreateElement(TEXT("data"));

    ConfigBitmapInfo configInfo;
    configInfo.data = data;

    if(DialogBoxParam(hinstMain, MAKEINTRESOURCE(IDD_CONFIGUREBITMAPSOURCE), hwndMain, ConfigureBitmapProc, (LPARAM)&configInfo) == IDOK)
    {
        if(bCreating)
        {
            CTSTR lpBitmap = data->GetString(TEXT("path"));

            D3DX10_IMAGE_INFO ii;
            if(SUCCEEDED(D3DX10GetImageInfoFromFile(lpBitmap, NULL, &ii, NULL)))
            {
                element->SetInt(TEXT("cx"), ii.Width);
                element->SetInt(TEXT("cy"), ii.Height);
            }
            else
                AppWarning(TEXT("ConfigureBitmapSource: could not get image info for bitmap '%s'"), lpBitmap);
        }

        return true;
    }

    return false;
}