VisionPlus_V3.0/VisionPlusTools/CstSpotAnalysis/Tools/blobselectshape.cpp

#pragma execution_character_set("utf-8")
#include "blobselectshape.h"
#include "ui_blobselectshape.h"


#include "../ToolDialog.h"
class ToolDialogImpl;


BlobSelectShape::BlobSelectShape(QWidget *parent , ToolDialogImpl* p) 
    : ToolInterface(parent), ui(new Ui::BlobSelectShape)
{
	ui->setupUi(this);
	m_pToolDialogImpl = p;

    ui->tableWidget->verticalHeader()->setVisible(false);
    ui->tableWidget->setHorizontalScrollBarPolicy(Qt::ScrollBarAlwaysOff);
    ui->tableWidget->setVerticalScrollBarPolicy(Qt::ScrollBarAlwaysOff);
    ui->tableWidget->horizontalHeader()->setSectionResizeMode(QHeaderView::Stretch);
    ui->tableWidget->setFocusPolicy(Qt::NoFocus);
    ui->tableWidget->setShowGrid(false);
    ui->tableWidget->setSelectionBehavior(QAbstractItemView::SelectRows);
    QMenu *menu = new QMenu();
    QMenu* shuxing = menu->addMenu("属性");
    shuxing->addAction("面积", this, SLOT(ActionSlot()));
    shuxing->addAction("X", this, SLOT(ActionSlot()));
    shuxing->addAction("Y", this, SLOT(ActionSlot()));
    shuxing->addAction("宽度", this, SLOT(ActionSlot()));
    shuxing->addAction("高度", this, SLOT(ActionSlot()));
    shuxing->addAction("角度", this, SLOT(ActionSlot()));
    QMenu *xingtai = menu->addMenu("形态");
    xingtai->addAction("圆度", this, SLOT(ActionSlot()));
    xingtai->addAction("紧密度", this, SLOT(ActionSlot()));
    xingtai->addAction("凸度", this, SLOT(ActionSlot()));
    xingtai->addAction("矩形度", this, SLOT(ActionSlot()));
    xingtai->addAction("最大内接", this, SLOT(ActionSlot()));
    ui->BTN_Add->setMenu(menu);
    ui->comboBox->setView(new QListView());
    ui->comboBox_index->setView(new QListView());
    if(m_pToolDialogImpl->m_LinkProcess.size() > 0)
    {
        ui->comboBox_index->setVisible(true);
        ui->label_index->setVisible(true);
    }
    else
    {
        ui->comboBox_index->setVisible(false);
        ui->label_index->setVisible(false);
    }
    for(int i = 0; i < m_pToolDialogImpl->m_LinkProcess.size(); i++)
    {
        ui->comboBox_index->addItem(QString::number(i));
    }



	m_NamesMap.insert("面积", "area");
	m_NamesMap.insert("X", "column");
	m_NamesMap.insert("Y", "row");
	m_NamesMap.insert("宽度", "width");
	m_NamesMap.insert("高度", "height");
	m_NamesMap.insert("角度", "rect2_phi");
	m_NamesMap.insert("紧密度", "compactness");
	m_NamesMap.insert("凸度", "convexity");
	m_NamesMap.insert("矩形度", "rectangularity");
	m_NamesMap.insert("最大内接", "max_diameter");


}

void BlobSelectShape::ActionSlot()
{
    QAction* action = (QAction*)sender();
    QString strName = action->text();

    AddTool(strName, 0 , 999999 );
}

BlobSelectShape::~BlobSelectShape()
{
    delete ui;
}

bool BlobSelectShape::Serialized(QDataStream& ar, bool bIsOut)
{
	int paranum;//参数数量
	if (bIsOut)//保存参数流程
	{
		paranum = 3;//以下参数的数量
		ar << paranum;//先保存参数数量
		ar << (UINT)m_AllScreening.size();
		for (UINT i = 0; i < m_AllScreening.size(); i++)
		{
			Screening nScreening = m_AllScreening[i];

			ar << (int)1 << nScreening.strName;
			ar << (int)2 << nScreening.nMin;
            ar << (int)3 << nScreening.nMax;
		}

	}
	else//加载参数流程，参数加载顺序一定要跟保存顺序一致
	{
		m_AllScreening.clear();
		ui->tableWidget->clear();

        UINT nToolSize = 0;

		ar >> paranum;//读取参数数量
		ar >> nToolSize;

		for (UINT i = 0; i < nToolSize; i++)
		{
			int nToolID = 0;
			QString strName;
            int  nMin;
            int  nMax;

			int nPpara;
			for (int j = 0; j < paranum; j++)
			{
				ar >> nPpara;
				switch (nPpara)
				{
				case 1:			  ar >> strName;				break;
				case 2:			  ar >> nMin;					break;
                case 3:			  ar >> nMax;					break;
				default:

					return false;
				}
			}

			AddTool(strName, nMin, nMax);
     
		}

        emit updateParameter(ShowParameter());

	}
	return true;
}


QString BlobSelectShape::ShowParameter()
{

    m_Key.Clear();
    m_MinValue.Clear();
    m_MaxValue.Clear();

    for (int i = 0; i < m_AllScreening.size(); i++)
    {
        Screening nScreening = m_AllScreening[i];

        QString strName = m_NamesMap.value(nScreening.strName);
        int nMin = nScreening.nMin;
        int nMax = nScreening.nMax;

        m_Key[i] = HTuple(strName.toStdString().c_str());
        m_MinValue[i] = HTuple(nMin);
        m_MaxValue[i] = HTuple(nMax);

//         if (i != 0)
//         {
//             m_Key.Append();
//         
//             m_MinValue.Append(HTuple(nMin));
//             m_MaxValue.Append(HTuple(nMax));
//         }
//         else
// 		{
//             m_Key = HTuple(strName.toStdString().c_str());
//             m_MinValue = HTuple(nMin);
//             m_MaxValue = HTuple(nMax);
//         }  
    }

	//  SelectShape(ho_ConnectedRegions, &ho_SelectedRegions, (HTuple("width").Append("height")), "and", (HTuple(1000).Append(200)), (HTuple(2000).Append(1000)));

    return ui->comboBox->currentText();
}

int BlobSelectShape::Execute()
{
    int result = 1;
    try
    {  
		if (m_AllScreening.size() == 0)
		{
			m_OutImage = m_InImage;
        }
        else
        {
			if (ui->comboBox_index->currentIndex() == 0)
			{
				SelectShape(m_InImage, &m_OutImage, m_Key, "and", m_MinValue, m_MaxValue);
			}
			else
			{
                SelectShape(m_InImage, &m_OutImage, m_Key, "or", m_MinValue, m_MaxValue);
			}
        }
    }
    catch(...)
    {
        vWarning() << "Error";
    }
    return result;
}


void BlobSelectShape::AddTool(QString strName, int nMin, int nMax)
{

    int nIndex = m_AllScreening.size();
	
    Screening nScreening;
	nScreening.nIndex = nIndex;
	nScreening.strName = strName;
	nScreening.nMin = nMin;
	nScreening.nMax = nMax;

    m_AllScreening.push_back(nScreening);

	ui->tableWidget->setRowCount(nIndex + 1);
	ui->tableWidget->setItem(nIndex, 0, new QTableWidgetItem(strName));
	ui->tableWidget->item(nIndex, 0)->setTextAlignment(Qt::AlignHCenter | Qt::AlignVCenter);
	ui->tableWidget->setItem(nIndex, 1, new QTableWidgetItem(QString::number(nScreening.nMin)));
	ui->tableWidget->item(nIndex, 1)->setTextAlignment(Qt::AlignHCenter | Qt::AlignVCenter);
	ui->tableWidget->setItem(nIndex, 2, new QTableWidgetItem(QString::number(nScreening.nMax)));
	ui->tableWidget->item(nIndex, 2)->setTextAlignment(Qt::AlignHCenter | Qt::AlignVCenter);




}

void BlobSelectShape::on_comboBox_currentIndexChanged(int index)
{
    emit updateParameter(ShowParameter());
}

void BlobSelectShape::on_tableWidget_cellChanged(int row, int column)
{

	int rowIndex = ui->tableWidget->currentRow();
    if (rowIndex != -1)
    { // 更新数据
        QTableWidgetItem* item1 = ui->tableWidget->item(row, 1);
        if (item1 != nullptr)
        {
			int nMinValue = item1->text().toInt();
			m_AllScreening[row].nMin = nMinValue;
		
			emit updateParameter(ShowParameter());
        }

		QTableWidgetItem* item2 = ui->tableWidget->item(row, 2);
		if (item2 != nullptr)
		{
			int nMaxValue = item2->text().toInt();
			m_AllScreening[row].nMax = nMaxValue;

			emit updateParameter(ShowParameter());
		}
    }
}

void BlobSelectShape::on_BTN_Del_clicked()
{
    int rowIndex = ui->tableWidget->currentRow();
    if (rowIndex != -1)
    {
        ui->tableWidget->removeRow(rowIndex);
        m_AllScreening.remove(rowIndex);
    
    }
}

void BlobSelectShape::on_comboBox_index_currentIndexChanged(int index)
{
    if(index > 0)
    {
         emit updateParameterIndex(QString::number(index - 1));
    }
    emit updateParameter(ShowParameter());
}