#include "fix_rigid_npt.h"
#include <cstring>
#include "modify.h"
#include "error.h"
using namespace LAMMPS_NS;
FixRigidNPT::FixRigidNPT(LAMMPS *lmp, int narg, char **arg) :
FixRigidNH(lmp, narg, arg)
{
scalar_flag = 1;
restart_global = 1;
box_change_size = 1;
extscalar = 1;
if (tstat_flag == 0 || pstat_flag == 0)
error->all(FLERR,"Did not set temperature or pressure for fix rigid/npt");
if (t_start <= 0.0 || t_stop <= 0.0)
error->all(FLERR,"Target temperature for fix rigid/npt cannot be 0.0");
if (t_period <= 0.0) error->all(FLERR,"Fix rigid/npt period must be > 0.0");
if (t_chain < 1) error->all(FLERR,"Illegal fix rigid/npt command");
if (t_iter < 1) error->all(FLERR,"Illegal fix rigid/npt command");
if (t_order != 3 && t_order != 5)
error->all(FLERR,"Fix rigid/npt temperature order must be 3 or 5");
t_freq = 0.0;
p_freq[0] = p_freq[1] = p_freq[2] = 0.0;
t_freq = 1.0 / t_period;
if (p_flag[0]) p_freq[0] = 1.0 / p_period[0];
if (p_flag[1]) p_freq[1] = 1.0 / p_period[1];
if (p_flag[2]) p_freq[2] = 1.0 / p_period[2];
int n = strlen(id) + 6;
id_temp = new char[n];
strcpy(id_temp,id);
strcat(id_temp,"_temp");
char **newarg = new char*[3];
newarg[0] = id_temp;
newarg[1] = (char *) "all";
newarg[2] = (char *) "temp";
modify->add_compute(3,newarg);
delete [] newarg;
tcomputeflag = 1;
n = strlen(id) + 7;
id_press = new char[n];
strcpy(id_press,id);
strcat(id_press,"_press");
newarg = new char*[4];
newarg[0] = id_press;
newarg[1] = (char *) "all";
newarg[2] = (char *) "pressure";
newarg[3] = id_temp;
modify->add_compute(4,newarg);
delete [] newarg;
pcomputeflag = 1;
}