#include <mpi.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <math.h>

#include "allvars.h"
#include "proto.h"
#include "treewalk.h"
#include "mymalloc.h"
#include "endrun.h"

/* This routine allocates memory for
 * particle storage, both the collisionless and the SPH particles.
 * The memory for the ordered binary tree of the timeline
 * is also allocated.
 */
void allocate_memory(void)
{
    size_t bytes;
    message(0, "Allocating memory MByte for particle storage.\n");
    //TreeWalk_allocate_memory();

    //NextActiveParticle = (int *) mymalloc("NextActiveParticle", bytes = All.MaxPart * sizeof(int));

    //NextInTimeBin = (int *) mymalloc("NextInTimeBin", bytes = All.MaxPart * sizeof(int));

    //PrevInTimeBin = (int *) mymalloc("PrevInTimeBin", bytes = All.MaxPart * sizeof(int));
    //
    P = (struct particle1id_data *) mymalloc("P", bytes = All.MaxPart * sizeof(struct particle1id_data));
    //P = (struct particle1Id_data *) mymalloc("P", bytes = 1 * sizeof(struct particle1Id_data));
    /* clear the memory to avoid valgrind errors;
     *
     * note that I tried to set each component in P to zero but
     * valgrind still complains in PFFT
     * seems to be to do with how the struct is padded and
     * the missing holes being accessed by __kmp_atomic fucntions. 
     * (memory lock etc?)
     * */
    memset(P, 0, sizeof(struct particle1id_data) * All.MaxPart);

    message(0, "Allocated %g MByte for particle storage.\n", bytes / (1024.0 * 1024.0));

    //SphP = (struct sph_particle_data *) mymalloc("SphP", bytes =
    //                 All.MaxPartSph * sizeof(struct sph_particle_data));
    //message(0, "Allocated %g MByte for storage of SPH data.\n", bytes / (1024.0 * 1024.0));

    BhP = (struct bh_particle1Id_data *) mymalloc("BhP", bytes =
                     All.MaxPartBh * sizeof(struct bh_particle1Id_data));
    message(0, "Allocated %g MByte for storage of BH data.\n", bytes / (1024.0 * 1024.0));
}