[u/mrichter/AliRoot.git] / TTherminator / Therminator / Hypersurface.cxx

#include <cstdlib>
#include <cstring>
#include "Hypersurface.h"

Hypersurface::Hypersurface(const char *dirname) {
// Read FOHSI.txt file and init parameters
  char *dirfull;
  dirfull = (char *) malloc(sizeof(char)*strlen(dirname) + 5);
  sprintf(dirfull, "%s", dirname);
  if (strlen(dirfull) > 0) 
    if (dirfull[strlen(dirfull)-1] != '/')
      strcat(dirfull, "/");
  FName = (char *) malloc(sizeof(char)* strlen(dirfull)+ 50);
  sprintf(FName,"%sFOHSI.txt", dirfull);
  HSFile = new ifstream;
  HSFile->open(FName);
  if(HSFile->is_open()) {
    HSFile->seekg (0, ios::beg);
// phi
    HSFile->getline(buff,100);
    Np=atoi(buff);
    HSFile->getline(buff,100);
    ip=atof(buff);
    HSFile->getline(buff,100);
    fp=atof(buff);
// zeta
    HSFile->getline(buff,100);
    Nz=atoi(buff);
    HSFile->getline(buff,100);
    iz=atof(buff);
    HSFile->getline(buff,100);
    fz=atof(buff);
// freeze-out temperature
    HSFile->getline(buff,100);
    TFO=atof(buff);
// hydro initial time
    HSFile->getline(buff,100);
    tau0=atof(buff);

    HSFile->close();
    dp = (fp-ip)/(Np-1);
    dz = (fz-iz)/(Nz-1);

// create 2D array
    aArr   = new double* [Np];
    vArr   = new double* [Np];
    dArr   = new double* [Np];
    DpdArr = new double* [Np];
    DzdArr = new double* [Np];
    for(i=0;i<Np;i++) {
      aArr[i]   = new double [Nz];
      vArr[i]   = new double [Nz];
      dArr[i]   = new double [Nz];
      DpdArr[i] = new double [Nz];
      DzdArr[i] = new double [Nz];
    }
  } else {
    Np = 0; ip = 0.0; fp = 0.0; dp = 1.0;
    Nz = 0; iz = 0.0; fz = 0.0; dz = 1.0;
    TFO = 0.0;
    aArr   = NULL;
    vArr   = NULL;
    dArr   = NULL;
    DpdArr = NULL;
    DzdArr = NULL;
  }
  delete HSFile;

  // Read FOHSa.txt file and fill array
  sprintf(FName,"%sFOHSa.txt", dirfull);
  //  FName = "FOHSa.txt";
  HSFile = new ifstream;
  HSFile->open(FName);
  if(HSFile->is_open()) {
    HSFile->seekg (0, ios::beg);
    for(i=0;i<Np;i++) {
      for(j=0;j<Nz;j++) {
        HSFile->getline(buff,100);
        aArr[i][j]=atof(buff);
      }
    }
    HSFile->close();
  }
  delete HSFile;

  // Read FOHSv.txt file and fill array
  sprintf(FName,"%sFOHSv.txt", dirfull);
  //  FName = "FOHSv.txt";
  HSFile = new ifstream;
  HSFile->open(FName);
  if(HSFile->is_open()) {
    HSFile->seekg (0, ios::beg);
    for(i=0;i<Np;i++) {
      for(j=0;j<Nz;j++) {
        HSFile->getline(buff,100);
        vArr[i][j]=atof(buff);
      }
    }
    HSFile->close();
  }
  delete HSFile;

  // Read FOHSd.txt file and fill array
  sprintf(FName,"%sFOHSd.txt", dirfull);
  //  FName = "FOHSd.txt";
  HSFile = new ifstream;
  HSFile->open(FName);
  if(HSFile->is_open()) {
    HSFile->seekg (0, ios::beg);
    for(i=0;i<Np;i++) {
      for(j=0;j<Nz;j++) {
        HSFile->getline(buff,100);
        dArr[i][j]=atof(buff);
      }
    }
    HSFile->close();
  }
  delete HSFile;

  // Read FOHSDpd.txt file and fill array
  sprintf(FName,"%sFOHSDpd.txt", dirfull);
  //  FName = "FOHSDpd.txt";
  HSFile = new ifstream;
  HSFile->open(FName);
  if(HSFile->is_open()) {
    HSFile->seekg (0, ios::beg);
    for(i=0;i<Np;i++) {
      for(j=0;j<Nz;j++) {
        HSFile->getline(buff,100);
        DpdArr[i][j]=atof(buff);
      }
    }
    HSFile->close();
  }
  delete HSFile;

  // Read FOHSDzd.txt file and fill array
  sprintf(FName,"%sFOHSDzd.txt", dirfull);
  //  FName = "FOHSDzd.txt";
  HSFile = new ifstream;
  HSFile->open(FName);
  if(HSFile->is_open()) {
    HSFile->seekg (0, ios::beg);
    for(i=0;i<Np;i++) {
      for(j=0;j<Nz;j++) {
        HSFile->getline(buff,100);
        DzdArr[i][j]=atof(buff);
      }
    }
    HSFile->close();
  }
  delete HSFile;
  free (FName);
  free (dirfull);
}

Hypersurface::Hypersurface(void) {
  Hypersurface("./");
}

Hypersurface::Hypersurface(const Hypersurface &aSurf)
{
  i = aSurf.i;
  Hypersurface("./");
  
}

Hypersurface::~Hypersurface(void) {
  for(i=0;i<Np;i++) {
    delete[] aArr[i];
    delete[] vArr[i];
    delete[] dArr[i];
    delete[] DpdArr[i];
    delete[] DzdArr[i];
  }
  delete[] aArr;
  delete[] vArr;
  delete[] dArr;
  delete[] DpdArr;
  delete[] DzdArr;
}

double Hypersurface::fahs(double p, double z) {
  i=int(p/dp);
  j=int(z/dz);
  if(i>=Np-1) i=Np-2;
  if(j>=Nz-1) j=Nz-2;
  return (aArr[i][j]   * (i+1-p/dp) + aArr[i+1][j]   * (p/dp-i)) * (j+1-z/dz) +
         (aArr[i][j+1] * (i+1-p/dp) + aArr[i+1][j+1] * (p/dp-i)) * (z/dz-j);
}

double Hypersurface::fvhs(double p, double z) {
  i=int(p/dp);
  j=int(z/dz);
  if(i>=Np-1) i=Np-2;
  if(j>=Nz-1) j=Nz-2;
  return (vArr[i][j]   * (i+1-p/dp) + vArr[i+1][j]   * (p/dp-i)) * (j+1-z/dz) +
         (vArr[i][j+1] * (i+1-p/dp) + vArr[i+1][j+1] * (p/dp-i)) * (z/dz-j);
}

double Hypersurface::fdhs(double p, double z) {
  i=int(p/dp);
  j=int(z/dz);
  if(i>=Np-1) i=Np-2;
  if(j>=Nz-1) j=Nz-2;
  return (dArr[i][j]   * (i+1-p/dp) + dArr[i+1][j]   * (p/dp-i)) * (j+1-z/dz) +
         (dArr[i][j+1] * (i+1-p/dp) + dArr[i+1][j+1] * (p/dp-i)) * (z/dz-j);
}

double Hypersurface::fDpdhs(double p, double z) {
  i=int(p/dp);
  j=int(z/dz);
  if(i>=Np-1) i=Np-2;
  if(j>=Nz-1) j=Nz-2;
  return (DpdArr[i][j]   * (i+1-p/dp) + DpdArr[i+1][j]   * (p/dp-i)) * (j+1-z/dz) +
         (DpdArr[i][j+1] * (i+1-p/dp) + DpdArr[i+1][j+1] * (p/dp-i)) * (z/dz-j);
}

double Hypersurface::fDzdhs(double p, double z) {
  i=int(p/dp);
  j=int(z/dz);
  if(i>=Np-1) i=Np-2;
  if(j>=Nz-1) j=Nz-2;
  return (DzdArr[i][j]   * (i+1-p/dp) + DzdArr[i+1][j]   * (p/dp-i)) * (j+1-z/dz) +
         (DzdArr[i][j+1] * (i+1-p/dp) + DzdArr[i+1][j+1] * (p/dp-i)) * (z/dz-j);
}
Commit	Line	Data
b09247a2	1	#include <cstdlib>
b09247a2	2	#include <cstring>
2e967919	3	#include "Hypersurface.h"
2e967919	4
ae89e57a	5	Hypersurface::Hypersurface(const char *dirname) {
2e967919	6	// Read FOHSI.txt file and init parameters
ae89e57a	7	char *dirfull;
	8	dirfull = (char ) malloc(sizeof(char)strlen(dirname) + 5);
	9	sprintf(dirfull, "%s", dirname);
	10	if (strlen(dirfull) > 0)
	11	if (dirfull[strlen(dirfull)-1] != '/')
	12	strcat(dirfull, "/");
	13	FName = (char ) malloc(sizeof(char) strlen(dirfull)+ 50);
	14	sprintf(FName,"%sFOHSI.txt", dirfull);
2e967919	15	HSFile = new ifstream;
	16	HSFile->open(FName);
	17	if(HSFile->is_open()) {
	18	HSFile->seekg (0, ios::beg);
	19	// phi
	20	HSFile->getline(buff,100);
	21	Np=atoi(buff);
	22	HSFile->getline(buff,100);
	23	ip=atof(buff);
	24	HSFile->getline(buff,100);
	25	fp=atof(buff);
	26	// zeta
	27	HSFile->getline(buff,100);
	28	Nz=atoi(buff);
	29	HSFile->getline(buff,100);
	30	iz=atof(buff);
	31	HSFile->getline(buff,100);
	32	fz=atof(buff);
	33	// freeze-out temperature
	34	HSFile->getline(buff,100);
	35	TFO=atof(buff);
9abb118d	36	// hydro initial time
	37	HSFile->getline(buff,100);
	38	tau0=atof(buff);
2e967919	39
	40	HSFile->close();
	41	dp = (fp-ip)/(Np-1);
	42	dz = (fz-iz)/(Nz-1);
	43
	44	// create 2D array
	45	aArr = new double* [Np];
	46	vArr = new double* [Np];
	47	dArr = new double* [Np];
	48	DpdArr = new double* [Np];
	49	DzdArr = new double* [Np];
	50	for(i=0;i<Np;i++) {
	51	aArr[i] = new double [Nz];
	52	vArr[i] = new double [Nz];
	53	dArr[i] = new double [Nz];
	54	DpdArr[i] = new double [Nz];
	55	DzdArr[i] = new double [Nz];
	56	}
	57	} else {
	58	Np = 0; ip = 0.0; fp = 0.0; dp = 1.0;
	59	Nz = 0; iz = 0.0; fz = 0.0; dz = 1.0;
	60	TFO = 0.0;
	61	aArr = NULL;
	62	vArr = NULL;
	63	dArr = NULL;
	64	DpdArr = NULL;
	65	DzdArr = NULL;
	66	}
	67	delete HSFile;
	68
	69	// Read FOHSa.txt file and fill array
ae89e57a	70	sprintf(FName,"%sFOHSa.txt", dirfull);
ae89e57a	71	// FName = "FOHSa.txt";
2e967919	72	HSFile = new ifstream;
	73	HSFile->open(FName);
	74	if(HSFile->is_open()) {
	75	HSFile->seekg (0, ios::beg);
	76	for(i=0;i<Np;i++) {
	77	for(j=0;j<Nz;j++) {
	78	HSFile->getline(buff,100);
	79	aArr[i][j]=atof(buff);
	80	}
	81	}
	82	HSFile->close();
	83	}
	84	delete HSFile;
	85
	86	// Read FOHSv.txt file and fill array
ae89e57a	87	sprintf(FName,"%sFOHSv.txt", dirfull);
ae89e57a	88	// FName = "FOHSv.txt";
2e967919	89	HSFile = new ifstream;
	90	HSFile->open(FName);
	91	if(HSFile->is_open()) {
	92	HSFile->seekg (0, ios::beg);
	93	for(i=0;i<Np;i++) {
	94	for(j=0;j<Nz;j++) {
	95	HSFile->getline(buff,100);
	96	vArr[i][j]=atof(buff);
	97	}
	98	}
	99	HSFile->close();
	100	}
	101	delete HSFile;
	102
	103	// Read FOHSd.txt file and fill array
ae89e57a	104	sprintf(FName,"%sFOHSd.txt", dirfull);
ae89e57a	105	// FName = "FOHSd.txt";
2e967919	106	HSFile = new ifstream;
	107	HSFile->open(FName);
	108	if(HSFile->is_open()) {
	109	HSFile->seekg (0, ios::beg);
	110	for(i=0;i<Np;i++) {
	111	for(j=0;j<Nz;j++) {
	112	HSFile->getline(buff,100);
	113	dArr[i][j]=atof(buff);
	114	}
	115	}
	116	HSFile->close();
	117	}
	118	delete HSFile;
	119
	120	// Read FOHSDpd.txt file and fill array
ae89e57a	121	sprintf(FName,"%sFOHSDpd.txt", dirfull);
ae89e57a	122	// FName = "FOHSDpd.txt";
2e967919	123	HSFile = new ifstream;
	124	HSFile->open(FName);
	125	if(HSFile->is_open()) {
	126	HSFile->seekg (0, ios::beg);
	127	for(i=0;i<Np;i++) {
	128	for(j=0;j<Nz;j++) {
	129	HSFile->getline(buff,100);
	130	DpdArr[i][j]=atof(buff);
	131	}
	132	}
	133	HSFile->close();
	134	}
	135	delete HSFile;
	136
	137	// Read FOHSDzd.txt file and fill array
ae89e57a	138	sprintf(FName,"%sFOHSDzd.txt", dirfull);
ae89e57a	139	// FName = "FOHSDzd.txt";
2e967919	140	HSFile = new ifstream;
	141	HSFile->open(FName);
	142	if(HSFile->is_open()) {
	143	HSFile->seekg (0, ios::beg);
	144	for(i=0;i<Np;i++) {
	145	for(j=0;j<Nz;j++) {
	146	HSFile->getline(buff,100);
	147	DzdArr[i][j]=atof(buff);
	148	}
	149	}
	150	HSFile->close();
	151	}
	152	delete HSFile;
ae89e57a	153	free (FName);
ae89e57a	154	free (dirfull);
2e967919	155	}
2e967919	156
ae89e57a	157	Hypersurface::Hypersurface(void) {
5177ac25	158	Hypersurface("./");
ae89e57a	159	}
ae89e57a	160
095d1adc	161	Hypersurface::Hypersurface(const Hypersurface &aSurf)
	162	{
	163	i = aSurf.i;
	164	Hypersurface("./");
	165
	166	}
ae89e57a	167
2e967919	168	Hypersurface::~Hypersurface(void) {
	169	for(i=0;i<Np;i++) {
	170	delete[] aArr[i];
	171	delete[] vArr[i];
	172	delete[] dArr[i];
	173	delete[] DpdArr[i];
	174	delete[] DzdArr[i];
	175	}
	176	delete[] aArr;
	177	delete[] vArr;
	178	delete[] dArr;
	179	delete[] DpdArr;
	180	delete[] DzdArr;
	181	}
	182
	183	double Hypersurface::fahs(double p, double z) {
	184	i=int(p/dp);
	185	j=int(z/dz);
	186	if(i>=Np-1) i=Np-2;
	187	if(j>=Nz-1) j=Nz-2;
	188	return (aArr[i][j] * (i+1-p/dp) + aArr[i+1][j] * (p/dp-i)) * (j+1-z/dz) +
	189	(aArr[i][j+1] * (i+1-p/dp) + aArr[i+1][j+1] * (p/dp-i)) * (z/dz-j);
	190	}
	191
	192	double Hypersurface::fvhs(double p, double z) {
	193	i=int(p/dp);
	194	j=int(z/dz);
	195	if(i>=Np-1) i=Np-2;
	196	if(j>=Nz-1) j=Nz-2;
	197	return (vArr[i][j] * (i+1-p/dp) + vArr[i+1][j] * (p/dp-i)) * (j+1-z/dz) +
	198	(vArr[i][j+1] * (i+1-p/dp) + vArr[i+1][j+1] * (p/dp-i)) * (z/dz-j);
	199	}
	200
	201	double Hypersurface::fdhs(double p, double z) {
	202	i=int(p/dp);
	203	j=int(z/dz);
	204	if(i>=Np-1) i=Np-2;
	205	if(j>=Nz-1) j=Nz-2;
	206	return (dArr[i][j] * (i+1-p/dp) + dArr[i+1][j] * (p/dp-i)) * (j+1-z/dz) +
	207	(dArr[i][j+1] * (i+1-p/dp) + dArr[i+1][j+1] * (p/dp-i)) * (z/dz-j);
	208	}
	209
	210	double Hypersurface::fDpdhs(double p, double z) {
	211	i=int(p/dp);
	212	j=int(z/dz);
	213	if(i>=Np-1) i=Np-2;
	214	if(j>=Nz-1) j=Nz-2;
	215	return (DpdArr[i][j] * (i+1-p/dp) + DpdArr[i+1][j] * (p/dp-i)) * (j+1-z/dz) +
	216	(DpdArr[i][j+1] * (i+1-p/dp) + DpdArr[i+1][j+1] * (p/dp-i)) * (z/dz-j);
	217	}
	218
	219	double Hypersurface::fDzdhs(double p, double z) {
	220	i=int(p/dp);
	221	j=int(z/dz);
	222	if(i>=Np-1) i=Np-2;
	223	if(j>=Nz-1) j=Nz-2;
	224	return (DzdArr[i][j] * (i+1-p/dp) + DzdArr[i+1][j] * (p/dp-i)) * (j+1-z/dz) +
	225	(DzdArr[i][j+1] * (i+1-p/dp) + DzdArr[i+1][j+1] * (p/dp-i)) * (z/dz-j);
	226	}