iphreeqc/PPassemblage.cxx
2021-04-19 17:17:02 -06:00

377 lines
10 KiB
C++

// PPassemblage.cxx: implementation of the cxxPPassemblage class.
//
//////////////////////////////////////////////////////////////////////
#ifdef _DEBUG
#pragma warning(disable : 4786) // disable truncation warning (Only used by debugger)
#endif
#include <cassert> // assert
#include <algorithm> // std::sort
#include "Utils.h" // define first
#include "Phreeqc.h"
#include "PPassemblage.h"
#include "cxxMix.h"
#include "phqalloc.h"
#if defined(PHREEQCI_GUI)
#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif
#endif
//////////////////////////////////////////////////////////////////////
// Construction/Destruction
//////////////////////////////////////////////////////////////////////
cxxPPassemblage::cxxPPassemblage(PHRQ_io * io)
//
// default constructor for cxxPPassemblage
//
: cxxNumKeyword(io)
{
new_def = false;
eltList.type = cxxNameDouble::ND_ELT_MOLES;
}
cxxPPassemblage::cxxPPassemblage(const std::map < int,
cxxPPassemblage > &entities, cxxMix & mix,
int l_n_user, PHRQ_io * io):
cxxNumKeyword(io)
{
this->n_user = this->n_user_end = l_n_user;
eltList.type = cxxNameDouble::ND_ELT_MOLES;
this->new_def = false;
//
// Mix
//
const std::map < int, LDBLE >&mixcomps = mix.Get_mixComps();
std::map < int, LDBLE >::const_iterator it;
for (it = mixcomps.begin(); it != mixcomps.end(); it++)
{
if (entities.find(it->first) != entities.end())
{
const cxxPPassemblage *entity_ptr =
&(entities.find(it->first)->second);
this->add(*entity_ptr, it->second);
}
}
}
cxxPPassemblage::~cxxPPassemblage()
{
}
void
cxxPPassemblage::dump_xml(std::ostream & s_oss, unsigned int indent) const
{
unsigned int i;
s_oss.precision(DBL_DIG - 1);
std::string indent0(""), indent1(""), indent2("");
for (i = 0; i < indent; ++i)
indent0.append(Utilities::INDENT);
for (i = 0; i < indent + 1; ++i)
indent1.append(Utilities::INDENT);
for (i = 0; i < indent + 2; ++i)
indent2.append(Utilities::INDENT);
// PPassemblage element and attributes
s_oss << indent0;
s_oss << "<EQUILIBRIUM_PHASES " << "\n";
// eltList
this->eltList.dump_xml(s_oss, indent + 1);
// ppAssemblageComps
s_oss << indent1;
s_oss << "<pure_phases " << "\n";
for (std::map < std::string, cxxPPassemblageComp >::const_iterator it =
pp_assemblage_comps.begin(); it != pp_assemblage_comps.end(); ++it)
{
(*it).second.dump_xml(s_oss, indent + 2);
}
}
void
cxxPPassemblage::dump_raw(std::ostream & s_oss, unsigned int indent, int *n_out) const
{
unsigned int i;
s_oss.precision(DBL_DIG - 1);
std::string indent0(""), indent1(""), indent2("");
for (i = 0; i < indent; ++i)
indent0.append(Utilities::INDENT);
for (i = 0; i < indent + 1; ++i)
indent1.append(Utilities::INDENT);
for (i = 0; i < indent + 2; ++i)
indent2.append(Utilities::INDENT);
// PPassemblage element and attributes
s_oss << indent0;
int n_user_local = (n_out != NULL) ? *n_out : this->n_user;
s_oss << "EQUILIBRIUM_PHASES_RAW " << n_user_local << " " << this->
description << "\n";
s_oss << indent1 << "# EXCHANGE_MODIFY candidates; use new_def=true #\n";
s_oss << indent1 << "-new_def " << 0 << "\n";
for (std::map < std::string, cxxPPassemblageComp >::const_iterator it =
pp_assemblage_comps.begin(); it != pp_assemblage_comps.end(); ++it)
{
s_oss << indent1;
s_oss << "-component " << (*it).second.Get_name() << "\n";
(*it).second.dump_raw(s_oss, indent + 2);
}
s_oss << indent1;
s_oss << "-eltList # List of all elements in phases and alternate reactions\n";
this->eltList.dump_raw(s_oss, indent + 2);
s_oss << indent1 << "# PPassemblage workspace variables #\n";
s_oss << indent1 << "-assemblage_totals" << "\n";
this->assemblage_totals.dump_raw(s_oss, indent + 1);
}
void
cxxPPassemblage::read_raw(CParser & parser, bool check)
{
std::istream::pos_type ptr;
std::istream::pos_type next_char;
std::string token;
int opt_save;
bool useLastLine(false);
// Read PPassemblage number and description
this->read_number_description(parser);
this->Set_new_def(false);
opt_save = CParser::OPT_ERROR;
for (;;)
{
int opt;
if (useLastLine == false)
{
opt = parser.get_option(vopts, next_char);
}
else
{
opt = parser.getOptionFromLastLine(vopts, next_char, true);
}
if (opt == CParser::OPT_DEFAULT)
{
opt = opt_save;
}
useLastLine = false;
switch (opt)
{
case CParser::OPT_EOF:
break;
case CParser::OPT_KEYWORD:
break;
case CParser::OPT_DEFAULT:
case CParser::OPT_ERROR:
opt = CParser::OPT_EOF;
parser.
error_msg("Unknown input in EQUILIBRIUM_PHASES_RAW keyword.",
PHRQ_io::OT_CONTINUE);
parser.error_msg(parser.line().c_str(), PHRQ_io::OT_CONTINUE);
useLastLine = false;
break;
case 0: // eltList
if (this->eltList.read_raw(parser, next_char) !=
CParser::PARSER_OK)
{
parser.incr_input_error();
parser.
error_msg("Expected element name and moles for totals.",
PHRQ_io::OT_CONTINUE);
}
opt_save = 0;
break;
case 1: // component
{
std::string str;
if (!(parser.get_iss() >> str))
{
parser.incr_input_error();
parser.error_msg("Expected string value for component name.",
PHRQ_io::OT_CONTINUE);
}
else
{
cxxPPassemblageComp temp_comp(this->io);
temp_comp.Set_name(str.c_str());
cxxPPassemblageComp *comp_ptr = this->Find(str);
if (comp_ptr)
{
temp_comp = *comp_ptr;
}
temp_comp.read_raw(parser, check);
this->pp_assemblage_comps[str] = temp_comp;
useLastLine = true;
}
}
break;
case 2: // new_def
if (!(parser.get_iss() >> this->new_def))
{
this->new_def = false;
parser.incr_input_error();
parser.
error_msg
("Expected boolean value for new_def in PPassemblage.",
PHRQ_io::OT_CONTINUE);
}
break;
case 3: // assemblage_totals
if (this->assemblage_totals.read_raw(parser, next_char) !=
CParser::PARSER_OK)
{
parser.incr_input_error();
parser.
error_msg
("Expected element name and molality for PPassemblage totals.",
PHRQ_io::OT_CONTINUE);
}
opt_save = 3;
break;
}
if (opt == CParser::OPT_EOF || opt == CParser::OPT_KEYWORD)
break;
}
}
void
cxxPPassemblage::totalize(Phreeqc * phreeqc_ptr)
{
this->assemblage_totals.clear();
// component structures
for (std::map < std::string, cxxPPassemblageComp >::iterator it =
pp_assemblage_comps.begin(); it != pp_assemblage_comps.end(); ++it)
{
(*it).second.totalize(phreeqc_ptr);
this->assemblage_totals.add_extensive((*it).second.Get_totals(), 1.0);
}
return;
}
std::set<std::string>
cxxPPassemblage::GetPhases(Phreeqc * phreeqc_ptr)
{
std::set<std::string> phase_list;
// component structures
for (std::map < std::string, cxxPPassemblageComp >::iterator it =
pp_assemblage_comps.begin(); it != pp_assemblage_comps.end(); ++it)
{
int l;
phase * phase_ptr = phreeqc_ptr->phase_bsearch((*it).second.Get_name().c_str(), &l, FALSE);;
if (phase_ptr != NULL)
{
phase_list.insert(phase_ptr->name);
}
}
return phase_list;
}
void
cxxPPassemblage::add(const cxxPPassemblage & addee, LDBLE extensive)
//
// Add to existing ppassemblage to "this" ppassemblage
//
{
if (extensive == 0.0)
return;
for (std::map < std::string, cxxPPassemblageComp >::const_iterator itadd = addee.pp_assemblage_comps.begin();
itadd != addee.pp_assemblage_comps.end(); ++itadd)
{
bool found = false;
for (std::map < std::string, cxxPPassemblageComp >::iterator it =
this->pp_assemblage_comps.begin();
it != this->pp_assemblage_comps.end(); ++it)
{
if ((*it).second.Get_name() == itadd->second.Get_name())
{
(*it).second.add((*itadd).second, extensive);
found = true;
break;
}
}
if (!found)
{
cxxPPassemblageComp entity = (*itadd).second;
entity.multiply(extensive);
std::string str(entity.Get_name());
this->pp_assemblage_comps[str] = entity;
}
}
//cxxNameDouble eltList;
this->eltList.add_extensive(addee.eltList, extensive);
}
cxxPPassemblageComp * cxxPPassemblage::
Find(const std::string name_in)
{
cxxPPassemblageComp * comp = NULL;
std::map<std::string, cxxPPassemblageComp>::iterator it;
it = this->pp_assemblage_comps.begin();
for ( ; it != this->pp_assemblage_comps.end(); it++)
{
if (Utilities::strcmp_nocase(name_in.c_str(), it->first.c_str()) == 0)
{
comp = &it->second;
break;
}
}
return comp;
}
/* ---------------------------------------------------------------------- */
void
cxxPPassemblage::Serialize(Dictionary & dictionary, std::vector < int >&ints, std::vector < double >&doubles)
/* ---------------------------------------------------------------------- */
{
/* int n_user; */
ints.push_back(this->n_user);
ints.push_back(this->new_def ? 1 : 0);
ints.push_back((int) this->pp_assemblage_comps.size());
for (std::map < std::string, cxxPPassemblageComp >::iterator it =
this->pp_assemblage_comps.begin(); it != this->pp_assemblage_comps.end();
it++)
{
(*it).second.Serialize(dictionary, ints, doubles);
}
this->eltList.Serialize(dictionary, ints, doubles);
this->assemblage_totals.Serialize(dictionary, ints, doubles);
}
/* ---------------------------------------------------------------------- */
void
cxxPPassemblage::Deserialize(Dictionary & dictionary, std::vector < int >&ints,
std::vector < double >&doubles, int &ii, int &dd)
/* ---------------------------------------------------------------------- */
{
/* int n_user; */
this->n_user = ints[ii++];
this->n_user_end = this->n_user;
this->description = " ";
this->new_def = (ints[ii++] != 0);
int count = ints[ii++];
this->pp_assemblage_comps.clear();
for (int n = 0; n < count; n++)
{
cxxPPassemblageComp ppc(this->io);
ppc.Deserialize(dictionary, ints, doubles, ii, dd);
std::string str(ppc.Get_name());
this->pp_assemblage_comps[str] = ppc;
}
this->eltList.Deserialize(dictionary, ints, doubles, ii, dd);
this->assemblage_totals.Deserialize(dictionary, ints, doubles, ii, dd);
}
const std::vector< std::string >::value_type temp_vopts[] = {
std::vector< std::string >::value_type("eltlist"), // 0
std::vector< std::string >::value_type("component"), // 1
std::vector< std::string >::value_type("new_def"), // 2
std::vector< std::string >::value_type("assemblage_totals") // 3
};
const std::vector< std::string > cxxPPassemblage::vopts(temp_vopts, temp_vopts + sizeof temp_vopts / sizeof temp_vopts[0]);