forked from lijiext/lammps
85 lines
2.8 KiB
C++
85 lines
2.8 KiB
C++
|
#include "ATC_Transfer.h"
|
||
|
#include "PhysicsModelThermal.h"
|
||
|
#include "Material.h"
|
||
|
#include <string>
|
||
|
#include <iostream>
|
||
|
#include <fstream>
|
||
|
|
||
|
namespace ATC {
|
||
|
|
||
|
PhysicsModelThermal::PhysicsModelThermal(string matFileName,
|
||
|
ATC_Transfer * atcTransfer)
|
||
|
: PhysicsModel(matFileName,atcTransfer)
|
||
|
{
|
||
|
initialize();
|
||
|
}
|
||
|
|
||
|
PhysicsModelThermal::~PhysicsModelThermal(void)
|
||
|
{
|
||
|
}
|
||
|
|
||
|
//--------------------------------------------------------------
|
||
|
// initialize
|
||
|
//--------------------------------------------------------------
|
||
|
void PhysicsModelThermal::initialize(void)
|
||
|
{
|
||
|
string list[2] = {"heat_capacity",
|
||
|
"heat_flux"};
|
||
|
set<string> needs(list,list+2);
|
||
|
vector< Material* >::iterator iter;
|
||
|
for (iter = materials_.begin(); iter != materials_.end(); iter++) {
|
||
|
Material * mat = *iter;
|
||
|
if (! (mat->check_registry(needs)) ) {
|
||
|
throw ATC_Error(0,"material does not provide all interfaces for physics");
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
//---------------------------------------------------------------------
|
||
|
// compute heat capacity
|
||
|
//---------------------------------------------------------------------
|
||
|
void PhysicsModelThermal::M_integrand(const Array<FieldName> &mask,
|
||
|
const FIELDS &fields,
|
||
|
FIELDS &capacity,
|
||
|
const int matIndex) const
|
||
|
{
|
||
|
Material* material = materials_[matIndex];
|
||
|
for (int n = 0; n < mask.get_length(); n++) {
|
||
|
if (mask(n) == TEMPERATURE) {
|
||
|
material->heat_capacity(fields, capacity[TEMPERATURE]);
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
//---------------------------------------------------------------------
|
||
|
// compute total energy
|
||
|
//---------------------------------------------------------------------
|
||
|
void PhysicsModelThermal::E_integrand(const Array<FieldName> &mask,
|
||
|
const FIELDS &fields,
|
||
|
const GRAD_FIELDS &grad_fields,
|
||
|
FIELDS &energy,
|
||
|
const int matIndex) const
|
||
|
{
|
||
|
Material* material = materials_[matIndex];
|
||
|
for (int n = 0; n < mask.get_length(); n++) {
|
||
|
if (mask(n) == TEMPERATURE) {
|
||
|
material->thermal_energy(fields, energy[TEMPERATURE]);
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
//--------------------------------------------------------------
|
||
|
// compute heat flux
|
||
|
//--------------------------------------------------------------
|
||
|
void PhysicsModelThermal::B_integrand(const Array2D<bool> & mask,
|
||
|
const FIELDS & fields,
|
||
|
const GRAD_FIELDS & gradFields,
|
||
|
GRAD_FIELDS & flux,
|
||
|
const int matIndex) const
|
||
|
{
|
||
|
Material *material = materials_[matIndex];
|
||
|
material->heat_flux(fields, gradFields, flux[TEMPERATURE]);
|
||
|
}
|
||
|
};
|
||
|
|
||
|
|