C++ Model::GetInteractions方法代码示例

std::string GenerateNbInteractionTypesFunction(Model &model) {
	std::stringstream stream;

	stream << "InteractionType NbInteractionTypes() {\n"
		   << "\treturn " << model.GetInteractions().size() << ";\n"
		   << "}\n";

	return stream.str();
}

std::string GenerateAttributesStruct(Model &model) {
	std::stringstream stream;

	for (const auto &agent : model.GetAgents()) {
		stream << agent.second.AttributesStruct(agent.first);
		stream << agent.second.PublicAttributesStruct(agent.first);
		stream << agent.second.CriticalAttributesStruct(agent.first);
		stream << agent.second.MessageStruct(agent.first);
	}

	for (const auto &interaction : model.GetInteractions()) {
		stream << interaction.second.AttributesStruct(interaction.first);
		stream << interaction.second.MessageStruct(interaction.first);
	}

	return stream.str();
}

std::string GenerateInteractionsMPIDatatypesFunction(Model &model, clang::ASTContext *context) {
	std::stringstream stream;

	// Add prototype
	stream << "size_t CreateInteractionsMPIDatatypes(std::unordered_map<InteractionType, MPI_Datatype> &interactions_MPI_types) {\n"
		   << "\tstd::vector<int> lengths; "
		   << "std::vector<MPI_Aint> offsets; "
		   << "std::vector<MPI_Datatype> mpi_types;\n"
		   << "\tMPI_Datatype t;\n"
		   << "\tsize_t max_size = 0;\n";

	std::unordered_set<std::string> temp_database;
	temp_database.insert("t");

	for (const auto &interaction : model.GetInteractions()) {
		int n_fields = interaction.second.GetFields().size();
		std::map<int64_t, std::string> type_temporaries;
		int i = 0;
		// Construct the data types of the fields
		for (const auto &field : interaction.second.GetFields()) {
			std::string temp = "t" + std::to_string(i);
			if (!temp_database.count(temp)) {
				stream << "\tMPI_Datatype " << temp << ";\n";
				temp_database.insert(temp);
			}

			std::string code_field = GenerateCodeMPIDatatype(field.second.GetType(), context, temp, temp_database);
			if (code_field.substr(0,6) != "MPI_Da" && code_field.substr(0,3) == "MPI") // No temporary to use
				type_temporaries[field.second.GetId()] = code_field;
			else {
				type_temporaries[field.second.GetId()] = temp;
				stream << code_field;
			}
			i++;
		}
		// Now construct the corresponding struct MPI_Datatype
		stream << "\tlengths = {";
		for (int j = 0; j < n_fields; j++)
			stream << "1,";
		if (n_fields > 0)
			stream.seekp(-1,std::ios_base::cur);
		stream << "};\n";

		stream << "\toffsets = {";
		for (const auto &field : interaction.second.GetFields()) {
			stream << "offsetof(" << interaction.first << "Attrs"
				   << "," << field.first
				   << "),";
		}
		if (n_fields > 0)
			stream.seekp(-1,std::ios_base::cur);
		stream << "};\n";

		stream << "\tmpi_types = {";
		for (const auto &field : interaction.second.GetFields()) {
			stream << type_temporaries[field.second.GetId()] <<",";
		}
		if (n_fields > 0)
			stream.seekp(-1,std::ios_base::cur);
		stream << "};\n";

		stream << "\tMPI_Type_create_struct(" << n_fields
			   << ", lengths.data(), offsets.data(), mpi_types.data(), &t);\n"
			   << "\tMPI_Type_commit(&t);\n";
		// Now the MPI_Datatype of the message structure
		// i.e. we add the type and the id and types of the sender and recipient
		stream << "\tlengths = {1,1,1,1,1,1};\n"
			   << "\toffsets = {offsetof(" << interaction.first << "MessageStruct,type),"
			   << "offsetof(" << interaction.first << "MessageStruct,sender_id),\n"
			   << "\t           offsetof(" << interaction.first << "MessageStruct,sender_type),"
			   << "offsetof(" << interaction.first << "MessageStruct,recipient_id),\n"
			   << "\t           offsetof(" << interaction.first << "MessageStruct,recipient_type),"
			   << "offsetof(" << interaction.first << "MessageStruct,data)}"
			   << ";\n";
		stream << "\tmpi_types = {MPI_UINT64_T,MPI_UINT64_T,MPI_UINT64_T,MPI_UINT64_T,MPI_UINT64_T,t};\n";
		stream << "\tMPI_Type_create_struct(" << "6"
			   << ", lengths.data(), offsets.data(), mpi_types.data(), &t);\n"
			   << "\tMPI_Type_commit(&t);\n";
		// Free the intermediary generated MPI_Datatypes
		for (const auto &temporary : type_temporaries) {
			if (temporary.second.length() > 0 && temporary.second.substr(0,1) == "t") // if it represents a constructed MPI_Datatype, free it
				stream << "\tMPI_Type_free(&" << temporary.second <<");\n";
		}
		// Store the MPI_Datatype
		stream << "\tinteractions_MPI_types[" << interaction.second.GetId()
			   << "] = t;\n";
		// Update the maximum size
		stream << "\tif (sizeof(" << interaction.first << "MessageStruct) > max_size)"
			   << " {max_size = sizeof(" << interaction.first << "MessageStruct);}\n";
	}

	stream << "\treturn max_size;\n"
		   << "}\n";

	return stream.str();
}