File: CompoundsManager.cpp

package info (click to toggle)
dyssol 1.5.0-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 18,184 kB
  • sloc: cpp: 53,870; sh: 85; python: 59; makefile: 11
file content (140 lines) | stat: -rw-r--r-- 4,289 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
/* Copyright (c) 2020, Dyssol Development Team.
 * Copyright (c) 2023, DyssolTEC GmbH.
 * All rights reserved. This file is part of Dyssol. See LICENSE file for license information. */

#include "CompoundsManager.h"
#include "MaterialsDatabase.h"
#include "Flowsheet.h"

CCompoundsManager::CCompoundsManager(CFlowsheet* _pFlowsheet, CMaterialsDatabase* _pDatabase, QWidget* _parent)
	: CQtDialog{ _parent }
{
	ui.setupUi(this);
	m_pFlowsheet = _pFlowsheet;
	m_pMaterialsDatabase = _pDatabase;

	SetHelpLink("001_ui/gui.html#sec-gui-menu-setup-compounds");
}

void CCompoundsManager::InitializeConnections()
{
	// signals from buttons
	connect(ui.pushButtonOk,			&QPushButton::clicked,				this, &CCompoundsManager::accept);
	connect(ui.pushButtonCancel,		&QPushButton::clicked,				this, &CCompoundsManager::close);
	connect(ui.addSelectedCompound,		&QPushButton::clicked,				this, &CCompoundsManager::AddSelectedCompound);
	connect(ui.removeSelectedCompound,	&QPushButton::clicked,				this, &CCompoundsManager::RemoveSelectedCompound);
	connect(ui.availableCompounds,		&QTableWidget::itemDoubleClicked,	this, &CCompoundsManager::AddSelectedCompound);
	connect(ui.usedCompounds,			&QTableWidget::itemDoubleClicked,	this, &CCompoundsManager::RemoveSelectedCompound);
}

void CCompoundsManager::UpdateWholeView()
{
	m_vAddedCompKeys = m_pFlowsheet->GetCompounds();
	UpdateAvailableCompounds();
	UpdateUsedCompounds();
}

void CCompoundsManager::UpdateAvailableCompounds()
{
	ui.availableCompounds->clear();
	ui.availableCompounds->setColumnCount(1);
	ui.availableCompounds->setRowCount((int)m_pMaterialsDatabase->CompoundsNumber());
	for (int i = 0; i < (int)m_pMaterialsDatabase->CompoundsNumber(); ++i)
		ui.availableCompounds->setItem(i, 0, new QTableWidgetItem(QString::fromStdString(m_pMaterialsDatabase->GetCompound(i)->GetName())));
}

void CCompoundsManager::UpdateUsedCompounds()
{
	ui.usedCompounds->clear();
	ui.usedCompounds->setColumnCount(1);
	ui.usedCompounds->setRowCount((int)m_vAddedCompKeys.size());
	for (int i = 0; i < (int)m_vAddedCompKeys.size(); ++i)
	{
		const CCompound* pCompound = m_pMaterialsDatabase->GetCompound( m_vAddedCompKeys[i] );
		if (pCompound)
			ui.usedCompounds->setItem(i, 0, new QTableWidgetItem(QString::fromStdString(pCompound->GetName())));
		else
			ui.usedCompounds->setItem(i, 0, new QTableWidgetItem(QString::fromStdString(m_vAddedCompKeys[i])));
	}
}

void CCompoundsManager::setVisible( bool _bVisible )
{
	if ( _bVisible )
		UpdateWholeView();
	CQtDialog::setVisible( _bVisible );
}

void CCompoundsManager::UpdateFromFlowsheet()
{
	if (isVisible())
		UpdateWholeView();
}

void CCompoundsManager::AddSelectedCompound()
{
	if ( ui.availableCompounds->currentRow() < 0 ) return;

	const CCompound* pCompound = m_pMaterialsDatabase->GetCompound(ui.availableCompounds->currentRow());
	if (!pCompound) return;

	int nIndex = GetAddedCompIndex( pCompound->GetKey() );
	if( nIndex != -1 ) return; // already exists

	m_vAddedCompKeys.push_back( pCompound->GetKey() );
	UpdateUsedCompounds();
}

void CCompoundsManager::RemoveSelectedCompound()
{
	if ((ui.usedCompounds->currentRow() < 0) || (ui.usedCompounds->currentRow() >= (int)m_vAddedCompKeys.size())) return;

	m_vAddedCompKeys.erase( m_vAddedCompKeys.begin() + ui.usedCompounds->currentRow() );
	UpdateUsedCompounds();
}

void CCompoundsManager::accept()
{
	ApplyChanges();
	QDialog::accept();
}

void CCompoundsManager::close()
{
	m_vAddedCompKeys.clear();
	QDialog::close();
}

void CCompoundsManager::ApplyChanges()
{
	size_t nOldNumber = m_pFlowsheet->GetCompoundsNumber();
	size_t nNewNumber = m_vAddedCompKeys.size();

	std::vector<std::string> vOldCompounds = m_pFlowsheet->GetCompounds();
	std::vector<std::string> vNewCompounds = m_vAddedCompKeys;

	// remove compounds
	for (size_t i = 0; i < nOldNumber; ++i)
		if (GetAddedCompIndex(vOldCompounds[i]) == -1)
			m_pFlowsheet->RemoveCompound(vOldCompounds[i]);

	// add compounds
	for (size_t i = 0; i < nNewNumber; ++i)
		m_pFlowsheet->AddCompound(vNewCompounds[i]);

	emit DataChanged();
}

int CCompoundsManager::GetAddedCompIndex( const std::string& _sKey ) const
{
	int nRes = -1;
	for( unsigned i=0; i<m_vAddedCompKeys.size(); ++i )
	{
		if( _sKey == m_vAddedCompKeys[i] )
		{
			nRes = i;
			break;
		}
	}
	return nRes;
}