---

transferfuncform.cpp

Go to the documentation of this file.

// transferfuncform.cpp : implementation file
//

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


#include <string>


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

#include "data.h"
extern Data *data3D;


// transferfuncform dialog


transferfuncform::transferfuncform(CWnd* pParent /*=NULL*/)
        : CDialog(transferfuncform::IDD, pParent)
{
        //{{AFX_DATA_INIT(transferfuncform)
        m_dTransferTo = 0;
        m_sStaticRight = _T("4095");
        m_sStaticLeft = _T("0");
        m_sStaticFrom = _T("0");
        m_dTransferAlpha = 255;
        m_dSliderAlpha = 255;
        m_dStaticMaxDenstity = _T("");
        //}}AFX_DATA_INIT

        //0x00bbggrr
        INDICES i = { 0, 4095, 255, 0x00FFFFFF };
        m_vTransferIndices.push_back(i);
        m_cTemp = 0x00FFFFFF;
}


void transferfuncform::DoDataExchange(CDataExchange* pDX)
{
        CDialog::DoDataExchange(pDX);
        //{{AFX_DATA_MAP(transferfuncform)
        DDX_Text(pDX, IDC_EDIT_TRANSFER_TO, m_dTransferTo);
        DDV_MinMaxInt(pDX, m_dTransferTo, 0, 4095);
        DDX_Text(pDX, IDC_STATIC_RIGHT, m_sStaticRight);
        DDV_MaxChars(pDX, m_sStaticRight, 4);
        DDX_Text(pDX, IDC_STATIC_LEFT, m_sStaticLeft);
        DDV_MaxChars(pDX, m_sStaticLeft, 4);
        DDX_Text(pDX, IDC_STATIC_TRANSFER_FROM, m_sStaticFrom);
        DDV_MaxChars(pDX, m_sStaticFrom, 4);
        DDX_Text(pDX, IDC_EDIT_TRANSFER_ALPHA, m_dTransferAlpha);
        DDV_MinMaxInt(pDX, m_dTransferAlpha, 0, 255);
        DDX_Slider(pDX, IDC_SLIDER_TRANSFER_ALPHA, m_dSliderAlpha);
        DDX_Text(pDX, IDC_STATIC_TRANSFER_MAXDENS, m_dStaticMaxDenstity);
        //}}AFX_DATA_MAP
}


void transferfuncform::m_ResetComboBox(bool clear) {
        ((CButton *)GetDlgItem(IDC_BUTTON_TRANSFER_SAVE))->EnableWindow(FALSE);

        CComboBox *comboBox = (CComboBox *)GetDlgItem(IDC_COMBO_TRANSFER);
        comboBox->ResetContent();
        comboBox->AddString("HISTOGRAM");
        comboBox->SetCurSel(0);

        if (clear)
                return;


        comboBox->ResetContent();
        comboBox->AddString("HISTOGRAM");

        std::string text = "TRANSFORM_";
        for (int i = 0; i < m_vTransferIndices.size() - 1; i++) {
                char s[5];
                _itoa(i, s, 10);
                text += s;
                comboBox->AddString(text.c_str());
                text.erase(10, strlen(s));
        }

        comboBox->SetCurSel(0);
}


void transferfuncform::m_DrawHistogram(CDC *dc,
                                                                           int densityBegin, 
                                                                           int densityEnd) {

        
        //draw the background
        dc->Rectangle(9, 30, 418, 280); 
        int spanX = 418 - 9;
        int spanY = 280 - 30;
        int beginX = 9;
        int beginY = 280 - 1;
        int beginY2 = 30;

        
        int dLineWidth = 1;
        int dXRatio;
        int *hist = new int[spanX];


        if ((densityEnd - densityBegin) >= spanX)
                dXRatio = (densityEnd - densityBegin) / spanX;
        else {
                if ((densityEnd - densityBegin) == 0)
                        dLineWidth = spanX;
                else {
                        dXRatio = spanX / (densityEnd - densityBegin);
                        dLineWidth = dXRatio;
                }
        }


        float dLocalMax = 0;

        for (int i = 0; i < spanX; i++) {
                hist[i] = data3D->GetHistogram(densityBegin + i * dXRatio);
                if (hist[i] > dLocalMax)
                        dLocalMax = hist[i];
        }
        
        for (i = 0; i < spanX; i++)
                hist[i] = beginY2 + (spanY - (int)(((float)hist[i] / dLocalMax) * (float)spanY));


        CPen *pPen = (CPen *)dc->SelectStockObject(BLACK_PEN);

        for (i = 0; i < spanX; i++) {
                dc->MoveTo(beginX + i, beginY);
                dc->LineTo(beginX + i, hist[i]);
        }

        char temp[10];
        _itoa(dLocalMax, temp, 10);
        m_dStaticMaxDenstity = temp;


        if (hist)
                delete[] hist;

        UpdateData(FALSE);
}


BEGIN_MESSAGE_MAP(transferfuncform, CDialog)
        //{{AFX_MSG_MAP(transferfuncform)
        ON_WM_SHOWWINDOW()
        ON_WM_PAINT()
        ON_BN_CLICKED(IDC_BUTTON_TRANSFER_COLOR, OnButtonTransferColor)
        ON_BN_CLICKED(IDC_BUTTON_TRANSFER_SAVE, OnButtonTransferSave)
        ON_BN_CLICKED(IDC_BUTTON_TRANSFER_CLEAR, OnButtonTransferClear)
        ON_BN_CLICKED(IDC_BUTTON_TRANSFER_NEW, OnButtonTransferNew)
        ON_CBN_SELCHANGE(IDC_COMBO_TRANSFER, OnSelchangeComboTransfer)
        ON_BN_CLICKED(IDC_BUTTON_TRANSFER_DELETE, OnButtonTransferDelete)
        ON_EN_CHANGE(IDC_EDIT_TRANSFER_ALPHA, OnChangeEditTransferAlpha)
        ON_BN_CLICKED(IDC_BUTTON_TRANSFER_EXPORT, OnButtonTransferExport)
        ON_BN_CLICKED(IDC_BUTTON_TRANSFER_IMPORT, OnButtonTransferImport)
        ON_WM_HSCROLL()
        ON_EN_CHANGE(IDC_EDIT_TRANSFER_TO, OnChangeEditTransferTo)
        //}}AFX_MSG_MAP
END_MESSAGE_MAP()


// transferfuncform message handlers
void transferfuncform::OnOK() 
{
        // TODO: Add extra validation here
//      UpdateData(TRUE);

        INDICES i;
        
/*
        if (m_vTransferIndices.size() == 1) {
                INDICES j = { 0, 4095, 255, 0x00FFFFFF };
                m_vTransferIndices.push_back(j);
        }
        else 
*/
        if ((i = m_vTransferIndices.back()).m_sdEnd != 4095) {
                INDICES j = { i.m_sdEnd + 1, 4095, 255, 0x00000000 };
                m_vTransferIndices.push_back(j);
        }

        CDialog::OnOK();
}

void transferfuncform::OnShowWindow(BOOL bShow, UINT nStatus) 
{
        CDialog::OnShowWindow(bShow, nStatus);

        // TODO: Add your message handler code here
        if (!bShow)
                return;

        ((CButton *)GetDlgItem(IDC_BUTTON_TRANSFER_SAVE))->EnableWindow(FALSE);
        ((CSliderCtrl *)GetDlgItem(IDC_SLIDER_TRANSFER_ALPHA))->SetRange(0, 255, TRUE);

//      UpdateData(FALSE);

        CComboBox *comboBox = (CComboBox *)GetDlgItem(IDC_COMBO_TRANSFER);

        std::string text = "TRANSFORM_";
        for (int i = 0; i < m_vTransferIndices.size() - 1; i++) {
                char s[5];
                _itoa(i, s, 10);
                text += s;
                comboBox->AddString(text.c_str());
                text.erase(10, strlen(s));
        }

        comboBox->SetCurSel(0);

        UpdateData(FALSE);
}


void transferfuncform::OnPaint() 
{
        CPaintDC dc(this); // device context for painting

        CComboBox *comboBox = (CComboBox *)this->GetDlgItem(IDC_COMBO_TRANSFER);
        int sel = comboBox->GetCurSel();
        dc.FillSolidRect(560, 159, 61, 23, m_vTransferIndices[sel].m_scColor);


        m_DrawHistogram(&dc,
                                        m_vTransferIndices[sel].m_sdBegin,
                                        m_vTransferIndices[sel].m_sdEnd);
        



        // TODO: Add your message handler code here
        
        // Do not call CDialog::OnPaint() for painting messages
}


void transferfuncform::OnButtonTransferColor() 
{
        CComboBox *comboBox = (CComboBox *)this->GetDlgItem(IDC_COMBO_TRANSFER);
        int sel = comboBox->GetCurSel();

        if (sel == CB_ERR)
                return;

        // TODO: Add your control notification handler code here
        CColorDialog colDiag;
        if (colDiag.DoModal() != IDOK)
                return;

        ((CButton *)GetDlgItem(IDC_BUTTON_TRANSFER_SAVE))->EnableWindow(TRUE);
        COLORREF col = colDiag.GetColor();

        m_cTemp = col;

        CDC *cdc = (CDC *)this->GetDC();
        cdc->FillSolidRect(560, 159, 61, 23, col);
}


void transferfuncform::OnButtonTransferSave() 
{
        // TODO: Add your control notification handler code here
        if (((CComboBox *)GetDlgItem(IDC_COMBO_TRANSFER))->GetCurSel() == 0)
                return;


        UpdateData(TRUE);
        ((CButton *)GetDlgItem(IDC_BUTTON_TRANSFER_SAVE))->EnableWindow(FALSE);

        if (m_dTransferTo < atoi(m_sStaticFrom)) {
                MessageBox("to must be greater or equal than from");
                ((CEdit *)GetDlgItem(IDC_EDIT_TRANSFER_TO))->SetFocus();
        }

        char text[5];
        m_sStaticLeft = m_sStaticFrom;
        _itoa(m_dTransferTo, text, 10);
        m_sStaticRight = text;

        int index;

        if ((index = ((CComboBox *)GetDlgItem(IDC_COMBO_TRANSFER))->GetCurSel()) < m_vTransferIndices.size()) {
                m_vTransferIndices[index].m_scColor = m_cTemp;
                m_vTransferIndices[index].m_sdBegin = atoi(m_sStaticFrom);
                m_vTransferIndices[index].m_sdEnd = m_dTransferTo;
                m_vTransferIndices[index].m_ucAlpha = m_dTransferAlpha;
        }
        else {
                INDICES i = { atoi(m_sStaticFrom), m_dTransferTo, m_dTransferAlpha, m_cTemp };
                m_vTransferIndices.push_back(i);
        }


        if (index != m_vTransferIndices.size() - 1) {
                if (m_vTransferIndices[index].m_sdEnd != m_vTransferIndices[index + 1].m_sdBegin - 1)
                        m_vTransferIndices[index + 1].m_sdBegin = m_vTransferIndices[index].m_sdEnd + 1;
        }


        m_bActive = false;


        m_DrawHistogram((CDC *)this->GetDC(),
                                        m_vTransferIndices[index].m_sdBegin,
                                        m_vTransferIndices[index].m_sdEnd);

        ((CComboBox *)GetDlgItem(IDC_COMBO_TRANSFER))->SetFocus();
        UpdateData(FALSE);
}

void transferfuncform::OnButtonTransferClear() 
{
        // TODO: Add your control notification handler code here
        if (MessageBox("Clear the whole transfer-function?", "clear trans-func", MB_YESNO) == IDNO)
                return;

        ((CButton *)GetDlgItem(IDC_BUTTON_TRANSFER_SAVE))->EnableWindow(FALSE);

        m_ResetComboBox(TRUE);

        
        m_vTransferIndices.erase(m_vTransferIndices.begin() + 1, m_vTransferIndices.end());
        m_sStaticLeft = "0";
        m_sStaticRight = "4095";
        m_vTransferIndices[0].m_sdBegin = 0;
        m_vTransferIndices[0].m_sdEnd = 4095;
        m_vTransferIndices[0].m_scColor = 0x00FFFFFFFF;
        m_sStaticFrom = "0";
        m_dTransferTo = 4095;

        CDC *cdc = (CDC *)this->GetDC();
        cdc->FillSolidRect(560, 159, 61, 23, 0x00FFFFFF);
        
        UpdateData(FALSE);
}


void transferfuncform::OnButtonTransferNew() 
{
        // TODO: Add your control notification handler code here
        if (m_bActive)
                return;

        m_bActive = true;

        
        ((CButton *)GetDlgItem(IDC_BUTTON_TRANSFER_SAVE))->EnableWindow(TRUE);

        CComboBox *comboBox = (CComboBox *)GetDlgItem(IDC_COMBO_TRANSFER);
        int size = m_vTransferIndices.size() - 1;
        std::string text = "TRANSFORM_";
        char s[5];
        _itoa(size, s, 10);
        text += s;

        comboBox->SetCurSel(comboBox->AddString(text.c_str()));
        INDICES i;
        if (!size) {
                _itoa(0, s, 10);
                m_sStaticFrom = s;
        }
        else {
                i = m_vTransferIndices.back();
                _itoa(i.m_sdEnd + 1, s, 10);
                m_sStaticFrom = s;
        }

        m_dTransferTo = 4095;
        
        UpdateData(FALSE);
}

void transferfuncform::OnSelchangeComboTransfer() 
{
        // TODO: Add your control notification handler code here
        ((CButton *)GetDlgItem(IDC_BUTTON_TRANSFER_SAVE))->EnableWindow(FALSE);
                
        CComboBox *comboBox = (CComboBox *)GetDlgItem(IDC_COMBO_TRANSFER);
        int i = comboBox->GetCount();
        if ((m_vTransferIndices.size() != i) && (comboBox->GetCurSel() != (i - 1)))
                comboBox->DeleteString(i - 1);

        i = comboBox->GetCurSel();
        char s[5];
        _itoa(m_vTransferIndices[i].m_sdBegin, s, 10);
        m_sStaticFrom = s;
        m_sStaticLeft = s;
        _itoa(m_vTransferIndices[i].m_sdEnd, s, 10);
        m_sStaticRight = s;
        m_dTransferTo = m_vTransferIndices[i].m_sdEnd;

        m_dSliderAlpha = m_vTransferIndices[i].m_ucAlpha;
        m_dTransferAlpha = m_vTransferIndices[i].m_ucAlpha;

        CDC *cdc = (CDC *)this->GetDC();
        cdc->FillSolidRect(560, 159, 61, 23, m_vTransferIndices[i].m_scColor);

        m_DrawHistogram(cdc,
                                        m_vTransferIndices[i].m_sdBegin,
                                        m_vTransferIndices[i].m_sdEnd);

        UpdateData(FALSE);

}

void transferfuncform::OnButtonTransferDelete() 
{
        // TODO: Add your control notification handler code here
        ((CButton *)GetDlgItem(IDC_BUTTON_TRANSFER_SAVE))->EnableWindow(FALSE);



        CComboBox *comboBox = (CComboBox *)GetDlgItem(IDC_COMBO_TRANSFER);
        int index;
        if ((index = comboBox->GetCurSel()) == 0)
                return;


        if (m_vTransferIndices.size() == 2) {
                m_ResetComboBox(TRUE);
                return;
        }


        INDEXLIST::iterator it = m_vTransferIndices.erase(m_vTransferIndices.begin() + index);
        INDEXLIST::iterator it2 = m_vTransferIndices.end();

        if ((it - 1) == m_vTransferIndices.begin())
                it->m_sdBegin = 0;
        else if (it != m_vTransferIndices.end())
                it->m_sdBegin = (it - 1)->m_sdEnd + 1;
        
        
        m_sStaticFrom = "0";
        m_sStaticLeft = "0";
        m_sStaticRight = "4095";
        m_dTransferTo = 4095;


        m_ResetComboBox();      


        CDC *cdc = (CDC *)this->GetDC();
        cdc->FillSolidRect(560, 159, 61, 23, 0x00FFFFFF);

        UpdateData(FALSE);
}

void transferfuncform::OnHScroll(UINT nSBCode, UINT nPos, CScrollBar* pScrollBar) 
{
        // TODO: Add your message handler code here and/or call default
        UpdateData(TRUE);
        m_dTransferAlpha = m_dSliderAlpha;
        UpdateData(FALSE);
        ((CButton *)GetDlgItem(IDC_BUTTON_TRANSFER_SAVE))->EnableWindow(TRUE);
        
        CDialog::OnHScroll(nSBCode, nPos, pScrollBar);
}

void transferfuncform::OnChangeEditTransferAlpha() 
{
        // TODO: If this is a RICHEDIT control, the control will not
        // send this notification unless you override the CDialog::OnInitDialog()
        // function and call CRichEditCtrl().SetEventMask()
        // with the ENM_CHANGE flag ORed into the mask.
        
        // TODO: Add your control notification handler code here
        UpdateData(TRUE);
        m_dSliderAlpha = m_dTransferAlpha;
        UpdateData(FALSE);
        ((CButton *)GetDlgItem(IDC_BUTTON_TRANSFER_SAVE))->EnableWindow(TRUE);
}


void transferfuncform::OnButtonTransferExport() 
{
        // TODO: Add your control notification handler code here
        CFileDialog fd(FALSE, NULL, NULL, OFN_HIDEREADONLY | OFN_OVERWRITEPROMPT,
                                        "Dat Files (*.tff)|*.tff| AllFiles (*.*)|* *|", this);
        

        int returnVal = fd.DoModal();

        if (returnVal != IDOK)
                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';

        CTransfuncFile funcFile;
        funcFile.SaveTransferFunc(m_vTransferIndices, fn);
}

void transferfuncform::OnButtonTransferImport() 
{
        // TODO: Add your control notification handler code here
        CFileDialog fd(TRUE, NULL, NULL, OFN_FILEMUSTEXIST,
                                        "Dat Files (*.tff)|*.tff| AllFiles (*.*)|* *|", this);
        

        int returnVal = fd.DoModal();

        if (returnVal != IDOK)
                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';

        CTransfuncFile funcFile;
        m_vTransferIndices.clear();
        bool success;
        INDEXLIST temp = funcFile.LoadTransferFunc(fn, success);
        if (success)
                m_vTransferIndices = temp;
        else
                return;

        m_cTemp = 0x00FFFFFF;
        m_sStaticFrom = "0";
        m_sStaticLeft = "0";
        m_sStaticRight = "4095";
        m_dTransferAlpha = 255;
        m_dSliderAlpha = 255;
        m_dTransferTo = 4095;

        m_ResetComboBox();

        
        UpdateData(FALSE);
}

void transferfuncform::OnChangeEditTransferTo() 
{
        // TODO: If this is a RICHEDIT control, the control will not
        // send this notification unless you override the CDialog::OnInitDialog()
        // function and call CRichEditCtrl().SetEventMask()
        // with the ENM_CHANGE flag ORed into the mask.
        
        // TODO: Add your control notification handler code here
        UpdateData(TRUE);
        char s[5];
        _itoa(m_dTransferTo, s, 10);
        m_sStaticRight = s;
        UpdateData(FALSE);
        ((CButton *)GetDlgItem(IDC_BUTTON_TRANSFER_SAVE))->EnableWindow(TRUE);
}

---