source: GTP/trunk/Lib/Illum/GPUObscurancesGT/src/MyFrame.cpp @ 953

/* -----------------------------------------------------------------------
  Main Window
-------------------------------------------------------------------------*/
#include "MyFrame.h"

//Icones
#include "UDG.xpm"

#include <wx/progdlg.h>
#include <wx/file.h>
#include <wx/string.h>
#include <wx/strconv.h>
#include <wx/bitmap.h>
#include <wx/image.h>

#include "glew.h"
#include "vcObscuranceMap.h"
#include "CMesh.h"

BEGIN_EVENT_TABLE(MyFrame,wxFrame)
        EVT_BUTTON(MyFrame::MAIN_MODEL, MyFrame::OnSelectModel)
        EVT_BUTTON(MyFrame::MAIN_IMAGE, MyFrame::OnSelectImage)
        EVT_BUTTON(MyFrame::MAIN_START, MyFrame::OnStart)
        EVT_COMBOBOX(MyFrame::MAIN_RES, MyFrame::OnSelectResolution) 
END_EVENT_TABLE()


// My frame constructor
MyFrame::MyFrame (const wxChar * title, int xpos, int ypos, int width, int height)
        : wxFrame ( (wxFrame *) NULL, -1, title, wxPoint(xpos, ypos), wxSize(width,height), wxCLOSE_BOX|wxCAPTION|wxSYSTEM_MENU )
{
        //Init Parameters
        modelString = new wxString("");
        imageString = new wxString("");
        res = 0;

        wxString choices_res_text[3];
        choices_res_text[0] = " 256 x  256";
        choices_res_text[1] = " 512 x  512";
        choices_res_text[2] = "1024 x 1024";

        choices_res[0] = 256;
        choices_res[1] = 512;
        choices_res[2] = 1024;

        //Create Fonts
        wxFont * font10 = new wxFont(10, wxFONTFAMILY_SWISS, wxFONTSTYLE_NORMAL, wxFONTWEIGHT_NORMAL);
        wxFont * font10bold = new wxFont(10, wxFONTFAMILY_SWISS, wxFONTSTYLE_NORMAL, wxFONTWEIGHT_BOLD);

        //Set Frame Icon
        SetIcon(wxIcon(UDG_xpm));

        //Set BackGround
        SetBackgroundColour(wxColour(200,200,200));

        //Create Main Sizer
        wxBoxSizer * mainSizer = new wxBoxSizer(wxVERTICAL);

        //Create GridSizer
        wxGridSizer * gridSizer = new wxGridSizer(3,1,5,5);
        mainSizer->Add(gridSizer,1,wxALL|wxEXPAND,10);

        //Model
        wxStaticBoxSizer * modelBoxSizer = new wxStaticBoxSizer(wxVERTICAL,this);
        gridSizer->Add(modelBoxSizer,2,wxALL|wxEXPAND,5);

        wxButton * modelButton = new wxButton( this, MAIN_MODEL, "Select Model");
        modelButton->SetFont(*font10bold);
        modelBoxSizer->Add(modelButton,1,wxALL|wxEXPAND,5);
        
        modelText = new wxStaticText(this, wxID_ANY, "Load a Model...");
        modelText->SetFont(*font10);
        modelBoxSizer->Add(modelText,1,wxALL|wxEXPAND,5);

        //Image
        wxStaticBoxSizer * imageBoxSizer = new wxStaticBoxSizer(wxVERTICAL,this);
        gridSizer->Add(imageBoxSizer,2,wxALL|wxEXPAND,5);

        wxButton * imageButton = new wxButton( this, MAIN_IMAGE, "Select Image");
        imageButton->SetFont(*font10bold);
        imageBoxSizer->Add(imageButton,1,wxALL|wxEXPAND,5);
        
        imageText = new wxStaticText(this, wxID_ANY, "Save to...");
        imageText->SetFont(*font10);
        imageBoxSizer->Add(imageText,1,wxALL|wxEXPAND,5);

        //Group
        wxGridSizer * groupSizer = new wxGridSizer(1,2,5,5);
        gridSizer->Add(groupSizer,1,wxALL|wxEXPAND,2);

        //Resolution
        wxStaticBoxSizer * resBoxSizer = new wxStaticBoxSizer(wxVERTICAL,this);
        groupSizer->Add(resBoxSizer,1,wxALL|wxEXPAND);

        wxStaticText * resLabel = new wxStaticText(this, wxID_ANY, "Resolution");
        resLabel->SetFont(*font10bold);
        resBoxSizer->Add(resLabel,1,wxALL|wxEXPAND,2);

        wxComboBox * combo_res = new wxComboBox(this, MAIN_RES, "", wxDefaultPosition, wxDefaultSize, 3, choices_res_text, wxCB_READONLY);
        combo_res->Select(0);
        res = choices_res[0];
        resBoxSizer->Add(combo_res,1,wxALL|wxEXPAND,2);

        //Start
        startButton = new wxButton( this, MAIN_START, "Start",wxDefaultPosition, wxDefaultSize, wxBU_EXACTFIT);
        startButton->SetFont(*font10bold);
        startButton->Enable(false);
        mainSizer->Add(startButton,0,wxALL|wxCENTER,5);

        //mainSizer
        SetSizer(mainSizer);
        mainSizer->SetSizeHints(this);
}

//Destructor
MyFrame::~MyFrame()
{
}

void MyFrame::CheckStart()
{

        if  (  ( *modelString != "" )
                && ( *imageString != "" )
                && ( res != 0))
        {
                //Toggle Button
                startButton->Enable(true);
        }
        else
                startButton->Enable(false);

}


void MyFrame::OnSelectModel(wxCommandEvent & event)
{
        const wxChar * FILETYPES = wxT 
                (
                        //"Ogre Mesh files|*.mesh|"
                        "Ogre XML files|*.xml|"
                        "All files|*.*"
                );

        wxFileDialog * openFileDialog = new wxFileDialog(this,"Select Mesh",
                /*Default folder*/"",/*Default Filename*/"",
                FILETYPES,wxOPEN|wxFILE_MUST_EXIST,wxDefaultPosition);

        if (openFileDialog->ShowModal() == wxID_OK )
        {
                modelString = new wxString("");
                modelString->Append(openFileDialog->GetPath());
                modelText->SetLabel(openFileDialog->GetFilename());
        }

        CheckStart();
}

//Image saving location
void MyFrame::OnSelectImage(wxCommandEvent & event)
{
                const wxChar * FILETYPES = wxT 
                (
                        "Bitmap|*.bmp|"
                        "All files|*.*"
                );

        wxFileDialog * saveFileDialog = new wxFileDialog(this,"Select where to Save Image",
                /*Default folder*/"",/*Default Filename*/"",
                FILETYPES,wxSAVE,wxDefaultPosition);

        if (saveFileDialog->ShowModal() == wxID_OK )
        {
                imageString = new wxString("");
                imageString->Append(saveFileDialog->GetPath());
                imageText->SetLabel(saveFileDialog->GetFilename());
        }

        CheckStart();
}

void MyFrame::OnStart(wxCommandEvent & event)
{
        float *ObscuranceMap;
        int i, j;
        
        
        CMesh g_pMesh;

        //if image.bmp exists...
        if ( wxFile::Exists(imageString->GetData()) )
        {
                int response = ::wxMessageBox("Image already exists?\nOverwrite?", "Attention!", wxYES_NO);
                
                if ( response == wxNO )
                        return;
        }


        ObscuranceMap = new float[res * res * 4];
        
        char *name = new char[modelString->Length() + 100], character;
        
        
        for (i = 0, j = 0; i < modelString->Length(); i++, j++) 
        {               
                character = modelString->GetChar(i);

                if (character == '\\')
                {
                        name[j] = modelString->GetChar(i);
                        name[j + 1] = modelString->GetChar(i);
                        j++;
                }
                else 
                        name[j] = modelString->GetChar(i);
        }

        name[j] = '\0';

        //start Obscurances generation
        vcObscurerGenerateImage(g_pMesh, name, int(res/256), &res, &res, &ObscuranceMap);

        typedef unsigned char uchar;
        uchar *bmp = new unsigned char[res * res * 3];
        float r_float;

        for (i = 0, j = 0; i < (res * res * 4); i += 4, j += 3)
        {
                r_float = ObscuranceMap[i] * 255;
                bmp[j] = uchar(r_float);
                r_float = ObscuranceMap[i+1] * 255;
                bmp[j + 1] = uchar(r_float);
                r_float = ObscuranceMap[i+2] * 255;
                bmp[j + 2] = uchar(r_float);
        }


        wxImage img(res,res);

        img.SetData(bmp,res,res);
        
        img.SaveFile(*imageString, wxBITMAP_TYPE_BMP);

        //freeing memory
        g_pMesh.FreeMem();
        delete name;
        

}

//Seleccionar Resolucio
void MyFrame::OnSelectResolution(wxCommandEvent & event)
{
        int id = event.GetSelection();
        res = choices_res[id];
        CheckStart();
}

//Seleccionar Qualitat
/*->void MyFrame::OnSelectQuality(wxCommandEvent & event)
{
        int id = event.GetSelection();
        qua = choices_qua[id];
        ComprovaStart();
}*/