#include "hdf5.h"
#include <cstdlib>
#include <dirent.h>
#include <stdio.h>
#include <string.h>
#include <math.h>
#define MAX_NAME 1024
#ifdef NAN
/* NAN is supported */
#endif
using namespace std;
struct alias {
double prob[4];
double alias[4];
};
struct road_network {
int gid;
double length;
double tt;
};
bool isvalueinarray(int val, int *arr, int size) {
int i;
for (i = 0; i < size; i++) {
if (arr[i] == val)
return true;
}
return false;
}
double *probab(double arr [], double ffs, int len) {
double *pr = (double *) malloc(4 * sizeof (double));
for (int i = 0; i < 4; i++)
pr[i] = 0;
for (int i = 0; i < len; i++) {
// printf("ar %f\n",arr[i]);
if (arr[i] > ffs * 0.95)
pr[0] += 1;
else if (arr[i] > ffs * 0.75)
pr[1] += 1;
else if (arr[i] > ffs * 0.45)
pr[2] += 1;
else
pr[3] += 1;
}
//printf("%f\n",pr[0]);
for (int i = 0; i < 4; i++) {
pr[i] = pr[i] / (double) len;
// printf("pr %f\n", pr[i]);
// printf("len %d\n", len);
}
return pr;
}
struct alias compalias(double *probb) {
struct alias a;
int n = 5;
double prob[5];
double probbm[5];
double alias[5];
int small[5];
int large [5];
for (int i = 0; i < 4; i++) {
prob[i + 1] = NAN;
alias[i + 1] = NAN;
probbm[i + 1] = probb[i] * ((n - 1) / 1);
small[i + 1] = NAN;
large[i + 1] = NAN;
}
int ns = 0;
int nl = 0;
for (int j = 1; j < n; j++) {
//printf("pr %f\n", probbm[j]);
if (probbm[j] > 1) {
nl = nl + 1;
large[nl] = j;
} else {
ns = ns + 1;
small[ns] = j;
}
}
// printf("ns %d\n",ns);
// printf("nl %d\n",nl);
// for (int j=1;j<5;j++){
// printf("large %d\n",large[j]);
// printf("small %d\n",small[j]);
//
// }
while (ns != 0 && nl != 0) {
int j = small[ns];
ns = ns - 1;
int k = large[nl];
nl = nl - 1;
prob[j] = probbm[j];
alias[j] = k;
probb[k] = probbm[k] + probbm[j] - 1;
if (probbm[k] > 1) {
nl = nl + 1;
large[nl] = k;
} else {
ns = ns + 1;
small[ns] = k;
}
}
while (ns != 0) {
prob[small[ns]] = 1;
ns = ns - 1;
}
while (nl != 0) {
prob[large[nl]] = 1;
nl = nl - 1;
}
for (int i = 1; i < 5; i++) {
alias[i] = alias[i] - 1;
}
for (int i = 1; i < 5; i++) {
if (isnan(alias[i]) == 1) {
//printf("fdsf");
alias[i] = -1;
}
}
double aliasm[4];
double probm[4];
for (int i = 1; i < 5; i++) {
aliasm[i - 1] = alias[i];
probm[i - 1] = prob[i];
}
memcpy(&a.alias, &aliasm, sizeof a.alias);
memcpy(&a.prob, &probm, sizeof a.prob);
return a;
}
int main(int argc, char *argv[]) {
struct dirent *de; // Pointer for directory entry
struct dirent *de2; // Pointer for directory entry
int num_per = atoi(argv[2]);
int sim_length = atoi(argv[3]);
int dim1, dim2, dim3;
int *prof = (int *) malloc(100000 * sizeof (int));
int *len = (int *) malloc(100000 * sizeof (int));
int numprof = 4;
int timestep = atoi(argv[5]);
int size = 0;
double *ffs = (double *) malloc(100000 * sizeof (double));
dim1 = num_per;
dim2 = sim_length + 1;
dim3 = 100000;
int siz, flag;
int i, j, k;
char ver[] = "1.0";
char mystring1 [100];
char *str1;
double f;
FILE* my_file1 = fopen(argv[6], "r");
if (my_file1 == NULL) {
perror(argv[1]);
exit(1);
}
struct road_network *cz = (road_network*) malloc(sizeof (struct road_network)*5000000);
size_t count1 = 0;
while (fgets(mystring1, 100, my_file1) != NULL) {
str1 = strtok(mystring1, ",");
cz[count1].gid = atoi(str1);
str1 = strtok(NULL, ",");
cz[count1].length = atof(str1);
str1 = strtok(NULL, ",");
cz[count1].tt = atof(str1);
str1 = strtok(NULL, ",");
++count1;
}
hid_t file_id, dataset_id, dataspace_id, group_id, plist, aid2, attr2, ret, atype, aid3, attr3; /* identifiers */
hsize_t dims[2];
hsize_t cdims[2];
herr_t status;
file_id = H5Fcreate(argv[4], H5F_ACC_TRUNC, H5P_DEFAULT, H5P_DEFAULT);
aid2 = H5Screate(H5S_SCALAR);
attr2 = H5Acreate1(file_id, "IntervalsPerSegment", H5T_NATIVE_INT, aid2, H5P_DEFAULT);
ret = H5Awrite(attr2, H5T_NATIVE_INT, &sim_length);
ret = H5Sclose(aid2);
aid2 = H5Screate(H5S_SCALAR);
attr2 = H5Acreate1(file_id, "TimeStep", H5T_NATIVE_INT, aid2, H5P_DEFAULT);
ret = H5Awrite(attr2, H5T_NATIVE_INT, ×tep);
ret = H5Sclose(aid2);
aid2 = H5Screate(H5S_SCALAR);
attr2 = H5Acreate1(file_id, "NumberOfProfiles", H5T_NATIVE_INT, aid2, H5P_DEFAULT);
ret = H5Awrite(attr2, H5T_NATIVE_INT, &numprof);
ret = H5Sclose(aid2);
aid3 = H5Screate(H5S_SCALAR);
atype = H5Tcopy(H5T_C_S1);
H5Tset_size(atype, 4);
attr3 = H5Acreate1(file_id, "Version", atype, aid3, H5P_DEFAULT);
ret = H5Awrite(attr3, atype, ver);
ret = H5Sclose(aid3);
double *** array = (double ***) malloc(dim1 * sizeof (double**));
for (i = 0; i < dim1; i++) {
array[i] = (double **) malloc(dim2 * sizeof (double *));
for (j = 0; j < dim2; j++) {
array[i][j] = (double *) malloc(dim3 * sizeof (double));
}
}
char *record = (char *) malloc(1000 * sizeof (char));
char *line = (char *) malloc(1000 * sizeof (char));
char *file2 = (char *) malloc(1000 * sizeof (char));
char *file = (char *) malloc(1000 * sizeof (char));
DIR *dr = opendir(argv[1]);
if (dr == NULL) // opendir returns NULL if couldn't open directory
{
printf("Could not open current directory");
return 0;
}
i = 0;
while ((de = readdir(dr)) != NULL) {
if (i > 1) {
strcpy(file, argv[1]);
printf("%s\n", de->d_name);
//printf("%s\n",strcat(strcat(argv[1],"\\"),de->d_name));
DIR *dr2 = opendir(strcat(strcat(file, "\\"), de->d_name));
if (dr2 == NULL) // opendir returns NULL if couldn't open directory
{
printf("Could not open current directory");
return 0;
}
j = 0;
siz = 0;
while ((de2 = readdir(dr2)) != NULL) {
if (j > 1) {
printf("%s\n", de2->d_name);
strcpy(file2, file);
strcat(file2, "\\");
strcat(file2, de2->d_name);
FILE* my_file1 = fopen(file2, "r");
if (my_file1 == NULL) {
perror(argv[1]);
exit(1);
}
while (fgets(record, 100000, my_file1) != NULL) {
line = strtok(record, ";");
if (siz == 0) {
// printf("id:%f\n",atof(line));
array[i - 2][0][siz] = atof(line);
if (isvalueinarray(atoi(line), prof, size) == false) {
prof[size] = atoi(line);
for (int y = 0; y < count1; y++) {
if (prof[size] == cz[y].gid) {
ffs[size] = (cz[y].length / (cz[y].tt / 1000))*3.6;
//printf("ffs %f\n", ffs[size]);
break;
}
}
size++;
}
for (int z = 0; z < size + 1; z++) {
if (array[i - 2][0][siz] == prof[z]) {
f = ffs[z];
//printf("fs %f\n", f);
}
}
for (int l = 1; l < dim2; l++) {
array[i - 2][l][siz] = f;
}
line = strtok(NULL, ";");
// printf("speed:%f\n",atof(line));
//printf("j:%d\n",j-1);
// printf("siz:%d\n",siz);
array[i - 2][j - 1][siz] = atof(line);
siz++;
} else {
flag = 0;
for (int l = 0; l < siz; l++) {
// printf("1:%f\n",array[i - 2][0][l]);
// printf("2:%f\n",atof(line));
if (array[i - 2][0][l] == atof(line)) {
line = strtok(NULL, ";");
// printf("speed:%f\n",atof(line));
// printf("j:%d\n",j-1);
// printf("siz:%d\n",siz);
array[i - 2][j - 1][l] = atof(line);
flag = 1;
break;
}
}
if (flag == 0) {
// printf("id:%f\n",atof(line));
array[i - 2][0][siz] = atof(line);
if (isvalueinarray(atoi(line), prof, size) == false) {
prof[size] = atoi(line);
for (int y = 0; y < count1; y++) {
if (prof[size] == cz[y].gid) {
ffs[size] = (cz[y].length / (cz[y].tt / 1000))*3.6;
break;
}
}
size++;
}
for (int z = 0; z < size + 1; z++) {
if (array[i - 2][0][siz] == prof[z]) {
f = ffs[z];
}
}
for (int l = 1; l < dim2; l++) {
array[i - 2][l][siz] = f;
}
line = strtok(NULL, ";");
// printf("speed:%f\n",atof(line));
// printf("j:%d\n",j-1);
// printf("siz:%d\n",siz);
array[i - 2][j - 1][siz] = atof(line);
siz++;
}
}
}
fclose(my_file1);
}
j++;
}
len[i - 2] = siz;
closedir(dr2);
}
i++;
}
closedir(dr);
free(cz);
free(file);
free(record);
free(file2);
for (i = 0; i < size; i++) {
printf("%d\n",i);
double *sp_p[dim2 - 1];
for (int i = 0; i < dim2 - 1; i++)
sp_p[i] = (double *) malloc(10000 * sizeof (double));
int pr = 0;
for (j = 0; j < dim1; j++) {
for (int k = 0; k < len[j]; k++) {
if (array[j][0][k] == prof[i]) {
for (int l = 1; l < dim2; l++) {
sp_p[pr][l-1] = array[j][l][k];
//printf("ar %f\n",sp_p[pr][l-1]);
}
pr++;
break;
}
}
}
// for (int qq=0;qq<dim2-1;qq++){
// for (int rr=0;rr<pr;rr++){
// //printf("ar %f\n",sp_p[rr][qq]);
// }
// }
double *sppr = (double *) malloc((dim2 - 1) * sizeof (double));
struct alias a;
double avg = 0;
double probb[pr];
double *probb2 = (double *) malloc(4 * sizeof (double));
double *sp_pro[8];
for (int ii = 0; ii < 8; ii++)
sp_pro[ii] = (double *) malloc((dim2 - 1) * sizeof (double));
double *al_alias[4];
for (int ii = 0; ii < 4; ii++)
al_alias[ii] = (double *) malloc((dim2 - 1) * sizeof (double));
double *al_pro[4];
for (int ii = 0; ii < 4; ii++)
al_pro[ii] = (double *) malloc((dim2 - 1) * sizeof (double));
for (int w = 0; w < dim2 - 1; w++) {
avg = 0;
for (int j = 0; j < pr; j++) {
avg += sp_p[j][w];
probb[j] = sp_p[j][w];
//printf("probs %f\n",probb[j]);
}
probb2 = probab(probb, ffs[i], pr);
for (int q = 0; q < 4; q++) {
//printf("prob %f\n",probb2[q]);
sp_pro[(2 * q) + 1][w] = probb2[q];
switch (q) {
case 0:
sp_pro[2 * q][w] = ffs[i];
break;
case 1:
sp_pro[q * 2][w] = 0.75 * ffs[i];
break;
case 2:
sp_pro[2 * q][w] = 0.5 * ffs[i];
break;
case 3:
sp_pro[2 * q][w] = 0.1 * ffs[i];
break;
}
}
a = compalias(probb2);
for (int r = 0; r < 4; r++) {
al_alias[r][w] = a.alias[r];
//printf("alias %f\n",a.alias[r]);
al_pro[r][w] = a.prob[r];
//printf("prob %f\n",a.prob[r]);
// printf("\n");
}
sppr[w] = avg / pr;
}
double *sp_prol = (double *) malloc((8*sim_length) * sizeof (double));
double *al_aliasl = (double *) malloc((4*sim_length) * sizeof (double));
double *al_prol = (double *) malloc((4*sim_length) * sizeof (double));
int counter=0;
for (int q=0;q<sim_length;q++){
for (int s=0;s<8;s++){
//printf("prof %f\n",sp_pro[s][q]);
sp_prol[counter++]=sp_pro[s][q];
//printf("lin %f\n",sp_prol[(q*sim_length)+s]);
}
}
counter=0;
for (int q=0;q<sim_length;q++){
for (int s=0;s<4;s++){
//printf("prof %f\n",sp_pro[s][q]);
al_aliasl[counter]=al_alias[s][q];
al_prol[counter++]=al_pro[s][q];
//printf("lin %f\n",sp_prol[(q*sim_length)+s]);
}
}
//for (int qq=0;qq<8*sim_length;qq++)
//printf("prof %f\n",sp_prol[qq]);
//printf("\n");
char str[20];
sprintf(str, "/%d", prof[i]);
group_id = H5Gcreate1(file_id, str, 0);
dims[0] = sim_length;
dims[1] = 4;
cdims[0] = 4;
cdims[1] = 4;
dataspace_id = H5Screate_simple(2, dims, NULL);
plist = H5Pcreate(H5P_DATASET_CREATE);
H5Pset_chunk(plist, 2, cdims);
H5Pset_deflate(plist, 6);
char str1[80];
strcpy(str1, str);
strcat(str1, "/alias");
dataset_id = H5Dcreate1(file_id, str1, H5T_NATIVE_DOUBLE, dataspace_id, plist);
dataset_id = H5Dopen(file_id, str1, H5P_DEFAULT);
status = H5Dwrite(dataset_id, H5T_NATIVE_DOUBLE, H5S_ALL, H5S_ALL, H5P_DEFAULT, al_aliasl);
status = H5Dclose(dataset_id);
strcpy(str1, str);
strcat(str1, "/probability");
dataset_id = H5Dcreate1(file_id, str1, H5T_NATIVE_DOUBLE, dataspace_id, plist);
dataset_id = H5Dopen(file_id, str1, H5P_DEFAULT);
status = H5Dwrite(dataset_id, H5T_NATIVE_DOUBLE, H5S_ALL, H5S_ALL, H5P_DEFAULT, al_prol);
status = H5Dclose(dataset_id);
dims[0] = sim_length;
dims[1] = 8;
cdims[0] = 4;
cdims[1] = 4;
dataspace_id = H5Screate_simple(2, dims, NULL);
plist = H5Pcreate(H5P_DATASET_CREATE);
H5Pset_chunk(plist, 2, cdims);
H5Pset_deflate(plist, 0);
strcpy(str1, str);
strcat(str1, "/prob_profile");
dataset_id = H5Dcreate1(file_id, str1, H5T_NATIVE_DOUBLE, dataspace_id, plist);
dataset_id = H5Dopen(file_id, str1, H5P_DEFAULT);
status = H5Dwrite(dataset_id, H5T_NATIVE_DOUBLE, H5S_ALL, H5S_ALL, H5P_DEFAULT, sp_prol);
status = H5Dclose(dataset_id);
dims[0] = 1;
dims[1] = sim_length;
cdims[0] = 1;
cdims[1] = 4;
dataspace_id = H5Screate_simple(2, dims, NULL);
plist = H5Pcreate(H5P_DATASET_CREATE);
H5Pset_chunk(plist, 2, cdims);
H5Pset_deflate(plist, 6);
strcpy(str1, str);
strcat(str1, "/speed_profile");
dataset_id = H5Dcreate1(file_id, str1, H5T_NATIVE_DOUBLE, dataspace_id, plist);
dataset_id = H5Dopen(file_id, str1, H5P_DEFAULT);
status = H5Dwrite(dataset_id, H5T_NATIVE_DOUBLE, H5S_ALL, H5S_ALL, H5P_DEFAULT, sppr);
status = H5Dclose(dataset_id);
}
status = H5Fclose(file_id);
return 0;
}