test_fof.c File Reference

#include <stdarg.h>
#include <stddef.h>
#include <setjmp.h>
#include <cmocka.h>
#include <math.h>
#include <mpi.h>
#include <stdio.h>
#include <string.h>
#include <time.h>
#include <gsl/gsl_rng.h>
#include <libgadget/fof.h>
#include <libgadget/walltime.h>
#include <libgadget/domain.h>
#include <libgadget/forcetree.h>
#include <libgadget/partmanager.h>
#include "stub.h"

Include dependency graph for test_fof.c:

Go to the source code of this file.

Macros
#define	qsort_openmp   qsort
#define	NUMPART1   8

Functions
static int	setup_particles (int NumPart, double BoxSize)
static int	teardown_particles (void **state)
static void	test_fof (void **state)
int	main (void)

Variables
static struct ClockTable	CT

Macro Definition Documentation

NUMPART1

#define NUMPART1   8

Definition at line 25 of file test_fof.c.

qsort_openmp

#define qsort_openmp   qsort

Definition at line 14 of file test_fof.c.

Function Documentation

main()

int main

(

void

)

Definition at line 114 of file test_fof.c.

               {
    const struct CMUnitTest tests[] = {
        cmocka_unit_test(test_fof),
    };
    return cmocka_run_group_tests_mpi(tests, NULL, NULL);
}

References test_fof().

Here is the call graph for this function:

setup_particles()

static int setup_particles ( int  NumPart, double  BoxSize  )

static

Definition at line 27 of file test_fof.c.

{
 
    int ThisTask;
    MPI_Comm_rank(MPI_COMM_WORLD, &ThisTask);
 
    gsl_rng * r = gsl_rng_alloc(gsl_rng_mt19937);
    gsl_rng_set(r, 0);
 
    particle_alloc_memory(PartManager, BoxSize, 1.5 * NumPart);
    PartManager->NumPart = NumPart;
 
    int i;
    #pragma omp parallel for
    for(i = 0; i < PartManager->NumPart; i ++) {
        P[i].ID = i + PartManager->NumPart * ThisTask;
        /* DM only*/
        P[i].Type = 1;
        P[i].Mass = 1;
        P[i].TimeBin = 0;
        P[i].IsGarbage = 0;
        int j;
        for(j=0; j<3; j++)
            P[i].Pos[j] = BoxSize * gsl_rng_uniform(r);
        P[i].Key = PEANO(P[i].Pos, BoxSize);
 
    }
 
    gsl_rng_free(r);
    /* TODO: Here create particles in some halo-like configuration*/
 
    return 0;
}

References part_manager_type::NumPart, P, particle_alloc_memory(), PartManager, PEANO(), and ThisTask.

Referenced by test_fof().

Here is the call graph for this function:

Here is the caller graph for this function:

teardown_particles()

static int teardown_particles ( void **  state )

static

Definition at line 62 of file test_fof.c.

{
    myfree(P);
    MPI_Barrier(MPI_COMM_WORLD);
    return 0;
}

References myfree, and P.

Referenced by test_fof().

Here is the caller graph for this function:

test_fof()

static void test_fof ( void **  state )

static

Definition at line 71 of file test_fof.c.

{
    walltime_init(&CT);
 
    struct DomainParams dp = {0};
    dp.DomainOverDecompositionFactor = 1;
    dp.DomainUseGlobalSorting = 0;
    dp.TopNodeAllocFactor = 1.;
    dp.SetAsideFactor = 1;
    set_domain_par(dp);
    init_forcetree_params(2);
 
    int NumPart = 1024;
    /* 20000 kpc*/
    double BoxSize = 20000;
    setup_particles(NumPart, BoxSize);
 
    int i, NTask, ThisTask;
    /* Build a tree and domain decomposition*/
    DomainDecomp ddecomp = {0};
    domain_decompose_full(&ddecomp);
 
    FOFGroups fof = fof_fof(&ddecomp, 1, MPI_COMM_WORLD);
 
    /* Example assertion: this checks that the address of a struct is not NULL,
     * which will always pass! */
    assert_all_true(&fof);
 
    MPI_Comm_size(MPI_COMM_WORLD, &NTask);
    MPI_Comm_rank(MPI_COMM_WORLD, &ThisTask);
 
    /* Assert some more things about the particles,
     * maybe checking the halo properties*/
    for(i = 0; i < PartManager->NumPart; i ++) {
        assert_true(P[i].ID % NTask == ThisTask);
    }
 
    fof_finish(&fof);
 
    teardown_particles(state);
    return;
}

References CT, domain_decompose_full(), DomainParams::DomainOverDecompositionFactor, DomainParams::DomainUseGlobalSorting, fof_finish(), fof_fof(), init_forcetree_params(), NTask, part_manager_type::NumPart, P, PartManager, set_domain_par(), DomainParams::SetAsideFactor, setup_particles(), teardown_particles(), ThisTask, DomainParams::TopNodeAllocFactor, and walltime_init().

Referenced by main().

Here is the call graph for this function:

Here is the caller graph for this function:

Variable Documentation

CT

struct ClockTable CT

static

Definition at line 1 of file test_fof.c.

Referenced by test_fof().