---

3dvis.cpp

Go to the documentation of this file.

// 3dvis.cpp : Defines the class behaviors for the application.
//

#include "stdafx.h"
#include "3dvis.h"

#include "MainFrm.h"
#include "3dvisDoc.h"
#include "3dvisView.h"

#include "transfuncfile.h"
#include "FileInfoDialog.h"
#include "VolumeDialog.h"
#include "IniFile.h"


#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

// CMy3dvisApp

BEGIN_MESSAGE_MAP(CMy3dvisApp, CWinApp)
        //{{AFX_MSG_MAP(CMy3dvisApp)
        ON_COMMAND(ID_APP_ABOUT, OnAppAbout)
        ON_COMMAND(ID_FILE_OPEN, OnFileOpen)
        ON_COMMAND(ID_F_CLOSE, OnFClose)
        ON_COMMAND(ID_RENDER_VOLUME, OnRenderVolume)
        ON_COMMAND(ID_OPTIONS_TRANSFER, OnOptionsTransfer)
        ON_COMMAND(ID_PRESET_0, OnPreset0)      
        ON_COMMAND(ID_PRESET_1, OnPreset1)      
        ON_COMMAND(ID_PRESET_2, OnPreset2)      
        ON_COMMAND(ID_PRESET_3, OnPreset3)      
        ON_COMMAND(ID_PRESET_4, OnPreset4)      
        ON_COMMAND(ID_PRESET_5, OnPreset5)      
        ON_COMMAND(ID_PRESET_6, OnPreset6)      
        ON_COMMAND(ID_PRESET_7, OnPreset7)      
        ON_COMMAND(ID_PRESET_8, OnPreset8)      
        ON_COMMAND(ID_PRESET_9, OnPreset9)      
        ON_COMMAND(ID_FILE_INFO, OnFileInfo)
        ON_COMMAND(ID_MENU_VOLUME, OnMenuVolume)
        ON_COMMAND(ID_FILE_INI, OnFileIni)
        //}}AFX_MSG_MAP
        // Standard file based document commands
        ON_COMMAND(ID_FILE_NEW, CWinApp::OnFileNew)
END_MESSAGE_MAP()


// CMy3dvisApp construction

CMy3dvisApp::CMy3dvisApp()
{
        // TODO: add construction code here,
        // Place all significant initialization in InitInstance
        m_bRenderVolume = false;
        m_bClicked = false;
        transferFunction = NULL;

        m_dFileThres = 40;
        m_dPacketSize = 10;

        m_dOperation = 0;

        char path[_MAX_PATH + 1];

        GetModuleFileName(m_hInstance, path, sizeof(path));

        string temp = path;
        int pos = temp.find_last_of('\\', temp.size());
        programPath = temp.substr(0, pos + 1);
}

// The one and only CMy3dvisApp object

CMy3dvisApp theApp;

// CMy3dvisApp initialization

BOOL CMy3dvisApp::InitInstance()
{

        CTransfuncFile funcFile;
        presetTransformFunctions = funcFile.LoadPresets(m_dNumPresets);


        // Standard initialization
        // If you are not using these features and wish to reduce the size
        //  of your final executable, you should remove from the following
        //  the specific initialization routines you do not need.

#ifdef _AFXDLL
        Enable3dControls();                     // Call this when using MFC in a shared DLL
#else
        Enable3dControlsStatic();       // Call this when linking to MFC statically
#endif

        // Change the registry key under which our settings are stored.
        // TODO: You should modify this string to be something appropriate
        // such as the name of your company or organization.
        SetRegistryKey(_T("Local AppWizard-Generated Applications"));

        LoadStdProfileSettings();  // Load standard INI file options (including MRU)

        // Register the application's document templates.  Document templates
        //  serve as the connection between documents, frame windows and views.

        CSingleDocTemplate* pDocTemplate;
        pDocTemplate = new CSingleDocTemplate(
                IDR_MAINFRAME,
                RUNTIME_CLASS(CMy3dvisDoc),
                RUNTIME_CLASS(CMainFrame),       // main SDI frame window
                RUNTIME_CLASS(CMy3dvisView));
        AddDocTemplate(pDocTemplate);

        // Parse command line for standard shell commands, DDE, file open
        CCommandLineInfo cmdInfo;
        ParseCommandLine(cmdInfo);

        // Dispatch commands specified on the command line
        if (!ProcessShellCommand(cmdInfo))
                return FALSE;

        
        m_ProcessIniFile();


        // The one and only window has been initialized, so show and update it.
        m_pMainWnd->ShowWindow(SW_SHOW);
        m_pMainWnd->UpdateWindow();
        return TRUE;
}


// CAboutDlg dialog used for App About

class CAboutDlg : public CDialog
{
public:
        CAboutDlg();

// Dialog Data
        //{{AFX_DATA(CAboutDlg)
        enum { IDD = IDD_ABOUTBOX };
        //}}AFX_DATA

        // ClassWizard generated virtual function overrides
        //{{AFX_VIRTUAL(CAboutDlg)
        protected:
        virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support
        //}}AFX_VIRTUAL

// Implementation
protected:
        //{{AFX_MSG(CAboutDlg)
                // No message handlers
        //}}AFX_MSG
        DECLARE_MESSAGE_MAP()
};

CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
{
        //{{AFX_DATA_INIT(CAboutDlg)
        //}}AFX_DATA_INIT
}

void CAboutDlg::DoDataExchange(CDataExchange* pDX)
{
        CDialog::DoDataExchange(pDX);
        //{{AFX_DATA_MAP(CAboutDlg)
        //}}AFX_DATA_MAP
}

BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
        //{{AFX_MSG_MAP(CAboutDlg)
                // No message handlers
        //}}AFX_MSG_MAP
END_MESSAGE_MAP()

// App command to run the dialog
void CMy3dvisApp::OnAppAbout()
{
        CAboutDlg aboutDlg;
        aboutDlg.DoModal();
}

// CMy3dvisApp message handlers


int CMy3dvisApp::ExitInstance() 
{
        // TODO: Add your specialized code here and/or call the base class
        if (data3D != NULL) {
                delete data3D;
                data3D = NULL;
        }

        if (slice != NULL) {
                delete slice;
                slice = NULL;
        }

        if (raycaster != NULL) {
                delete raycaster;
                raycaster = NULL;
        }

        if (transferFunction != NULL) {
                delete transferFunction;
                transferFunction = NULL;
        }
        
        return CWinApp::ExitInstance();
}

void CMy3dvisApp::OnFileOpen() 
{
        // TODO: Add your command handler code here
        CFileDialog fd(TRUE, NULL, NULL, OFN_FILEMUSTEXIST,
                                        "Dat Files (*.dat)|*.dat| AllFiles (*.*)|* *|", m_pMainWnd);
        

        int returnVal = fd.DoModal();

        if(returnVal == IDOK) {

                m_bOpen = true;
                this->OnFClose();
                m_bOpen = false;
                        
                if (data3D != NULL) {
                        if (MessageBox(m_pMainWnd->m_hWnd, "Already data loaded! Unload currend data?",
                                                                                                        "load new data", MB_YESNO) == IDYES) {
                                delete data3D;
                                data3D = NULL;
                        }
                        else
                                return;
                }


                CString filename = fd.GetFileName();
                char *fn = new char[filename.GetLength() + 1];
                for (int i = 0; i < filename.GetLength(); i++) {
                        fn[i] = filename.GetAt(i);
                }
                fn[i] = '\0';

                
                data3D = new Data();
                m_ProcessIniFile();


                SetWindowText(m_pMainWnd->m_hWnd, "3D Visualisation - loading visualisation data, please wait,...");

                data3D->LoadData(fn);
                delete[] fn;

        
                m_bRenderVolume = false;


                if (slice == NULL)
                        slice = new Slice();

                slice->SetPixels(data3D, 0);
                int num = slice->GetSliceNumber(data3D);


                
                if (!m_bRenderVolume)
                        AfxGetMainWnd()->SendMessage(MYWM_SHOW_BAR, (WPARAM) 0, (LPARAM) 0);
                else
                        AfxGetMainWnd()->SendMessage(MYWM_SHOW_BAR, (WPARAM) 1, (LPARAM) 0);
                

                SetWindowText(m_pMainWnd->m_hWnd, CString("3D Visualisation - loaded \"" + filename + "\""));
   }
}

void CMy3dvisApp::OnFClose() 
{
        // TODO: Add your command handler code here

        if (data3D && !m_bOpen && MessageBox(m_pMainWnd->m_hWnd, "do you really want to close the\ncurrent data-set?",
                                        "close data-set", MB_YESNO) == IDNO)
                return;

        
        AfxGetMainWnd()->SendMessage(MYWM_HIDE_BARS);

        
        if (slice != NULL) {
                delete slice;
                slice = NULL;
        }

        if (data3D != NULL) {
                delete data3D;
                data3D = NULL;
        }

        m_pMainWnd->SetWindowText("3D Visualisation");
}

void CMy3dvisApp::OnOptionsTransfer() 
{
        // TODO: Add your command handler code here
        if (data3D == NULL)
                return;

        int response = m_Transferdiag.DoModal();


        if (response == IDOK) {
                int num = m_Transferdiag.m_vTransferIndices.size() - 1;

                if (num <= 0)
                        return;

                m_BuildTransferFunction(m_Transferdiag.m_vTransferIndices, num);
        }


        m_pMainWnd->Invalidate();
        m_pMainWnd->UpdateWindow();
}


void CMy3dvisApp::OnPreset0() {
        m_BuildTransferFunction(presetTransformFunctions[0],
                                                        presetTransformFunctions[0].size() - 1);
}

void CMy3dvisApp::OnPreset1() {
        m_BuildTransferFunction(presetTransformFunctions[1],
                                                        presetTransformFunctions[1].size() - 1);
}

void CMy3dvisApp::OnPreset2() {
        m_BuildTransferFunction(presetTransformFunctions[2],
                                                        presetTransformFunctions[2].size() - 1);
}

void CMy3dvisApp::OnPreset3() {
        m_BuildTransferFunction(presetTransformFunctions[3],
                                                        presetTransformFunctions[3].size() - 1);
}

void CMy3dvisApp::OnPreset4() {
        m_BuildTransferFunction(presetTransformFunctions[4],
                                                        presetTransformFunctions[4].size() - 1);
}

void CMy3dvisApp::OnPreset5() {
        m_BuildTransferFunction(presetTransformFunctions[5],
                                                        presetTransformFunctions[5].size() - 1);
}

void CMy3dvisApp::OnPreset6() {
        m_BuildTransferFunction(presetTransformFunctions[6],
                                                        presetTransformFunctions[6].size() - 1);
}

void CMy3dvisApp::OnPreset7() {
        m_BuildTransferFunction(presetTransformFunctions[7],
                                                        presetTransformFunctions[7].size() - 1);
}

void CMy3dvisApp::OnPreset8() {
        m_BuildTransferFunction(presetTransformFunctions[8],
                                                        presetTransformFunctions[8].size() - 1);
}

void CMy3dvisApp::OnPreset9() {
        m_BuildTransferFunction(presetTransformFunctions[9],
                                                        presetTransformFunctions[9].size() - 1);
}

void CMy3dvisApp::m_BuildTransferFunction(INDEXLIST &function, int num) {
        if (!data3D)
                return;
        
        if (!transferFunction)
                transferFunction = new Transfunc();

        rgb col;

        float alpha_begin = 0.0;
        float alpha_end = 1.0;


        for (int i = 1; i <= num; i++) {
                COLORREF c = function[i].m_scColor;
                col.r = (unsigned char)(function[i].m_scColor & 0x000000FF);
                col.g = (unsigned char)((function[i].m_scColor & 0x0000FF00) >> 8);
                col.b = (unsigned char)((function[i].m_scColor & 0x00FF0000) >> 16);

                float thres = function[i].m_sdEnd - function[i].m_sdBegin;
                float size = thres * 0.1;
                
                if (i > 1)
                        alpha_begin = (float)function[i - 1].m_ucAlpha / 255;
                else
                        alpha_begin = 0.0;


                if (i == num)
                        alpha_end = 0.0;
                else
                        alpha_end = (float)function[i].m_ucAlpha / 255;

                float alpha = alpha_end - alpha_begin;
                float step = alpha / size;
                if (i > 1)
                        alpha = (float)function[i].m_ucAlpha / 255;


                for (int j = function[i].m_sdBegin;     j < function[i].m_sdEnd; j++) {
                        transferFunction->SetDensityColor(j, col);
                        if ((alpha > 0.0) && ((alpha + step) <= alpha_end))
                                alpha += step;
//                      transferFunction->SetOpacity(j, ((float)function[i].m_ucAlpha / 255));
                        transferFunction->SetOpacity(j, alpha);
                }
        }

        if (slice == NULL)
                slice = new Slice();

        slice->SetTransferfunction(transferFunction);
        slice->SetPixels(data3D, -1);
}


void CMy3dvisApp::OnFileInfo() 
{
        // TODO: Add your command handler code here
        if (!data3D)
                return;
        
        CFileInfoDialog fid;
        fid.SetInfo(data3D->filename, data3D->GetXDim(), data3D->GetYDim(), data3D->GetZDim());
        fid.DoModal();
}

void CMy3dvisApp::OnMenuVolume() 
{
        // TODO: Add your command handler code here
        AfxGetMainWnd()->SendMessage(MYWM_CHANGE_RENDERING, (WPARAM) 1, 1);
}


void CMy3dvisApp::m_ProcessIniFile(bool store) {
        FILE *file;

        string filename = programPath + "config.ini";

        if (!store && (file = fopen(filename.c_str(), "r")) != NULL) {
                if (!data3D)
                        return;
                
                fscanf(file, "%d\n%d", &m_dFileThres, &m_dPacketSize);
                data3D->SetFileThreshold(m_dFileThres);
                data3D->SetDataPackets(m_dPacketSize);

                fclose(file);

                return;
        }


        if ((file = fopen(filename.c_str(), "w")) != NULL) {
                m_dFileThres = 40;
                m_dPacketSize = 10;

                fprintf(file, "%d\n%d", m_dFileThres, m_dPacketSize);
                fclose(file);
        }


}

void CMy3dvisApp::OnFileIni() 
{
        // TODO: Add your command handler code here
        IniFile inifile;
        inifile.SetupData(m_dFileThres, m_dPacketSize);
        int answer = inifile.DoModal();

        if (answer == IDCANCEL)
                return;
        
        
        inifile.GetData(m_dFileThres, m_dPacketSize);   
        m_ProcessIniFile(true);
}



void CMy3dvisApp::OnRenderVolume() 
{
        // TODO: Add your command handler code here
        if (!data3D || !slice)
                return;

        if (!((CMy3dvisApp *)AfxGetApp())->transferFunction) {
                MessageBox(m_pMainWnd->m_hWnd,
                                        "please select or create a transferfunction",
                                        "transferfunction missing",
                                        MB_OK);
                return;
        }
                

        if (((CMy3dvisApp *)AfxGetApp())->m_bRenderVolume) {
                ((CMy3dvisApp *)AfxGetApp())->m_bRenderVolume = false;

                AfxGetMainWnd()->SendMessage(MYWM_CHANGE_RENDERING, (WPARAM) 0, (LPARAM) 0);
        }
        else {
                ((CMy3dvisApp *)AfxGetApp())->m_bRenderVolume = true;

                AfxGetMainWnd()->SendMessage(MYWM_CHANGE_RENDERING, (WPARAM) 1, (LPARAM) 0);

                raycaster = new Raycaster();
                raycaster->SetViewingCondition(VECTOR(300,300,300),Color(0.0f,1.0f,0.0f),0.3f,0.4f,0.2f,8);

                raycaster->Initialize(/*VECTOR(92,-20,85),*/400,400,1.0f,data3D,((CMy3dvisApp *)AfxGetApp())->transferFunction);
                raycaster->Zoom(1.1f); // werte > 1.0 hinein zoomen <1.0 wegzoomen

                raycaster->SetTreshold(1500);
                raycaster->SetRaytype(NN);
                raycaster->SetRendermode(FH);
                raycaster->SetPreCalcGradients(true);
        }
}

---