mockgallib/src

1
2
3
4
5
6
7
8
9
10
11
12
13
#ifndef COLA_FILE_H
#define COLA_FILE_H 1

#include <vector>
#include "halo.h"

void cola_halo_file_open(const char filename[], float* const boxsize);
int  cola_halo_file_read_one(Halo* const h);
void cola_halo_file_close();

class ColaFileError {};

#endif

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
#include <cstdio>
#include <cassert>

#include "msg.h"
#include "cola_file.h"

using namespace std;

FILE* fp;
static int nhalo;

void cola_halo_file_open(const char filename[],
			 float* const boxsize)
{
  fp= fopen(filename, "r");
  if(fp == 0) {
    msg_printf(msg_fatal, "Error: unable to open cola fof file, %s", filename);
    throw ColaFileError();
  }

  float params[3];
  int ret= fread(params, sizeof(float), 3, fp); assert(ret == 3);

  *boxsize= params[0];
  
  nhalo= 0;
}

int cola_halo_file_read_one(Halo* const h)
{
  int ret= fread(&h->nfof, sizeof(int), 1, fp); assert(ret == 1);
  if(h->nfof == 0) return 0;

  float f[3];
  
  ret= fread(h->x, sizeof(float), 3, fp); assert(ret == 3);
  ret= fread(h->v, sizeof(float), 3, fp); assert(ret == 3);
  ret= fread(f, sizeof(float), 3, fp);    assert(ret == 3);

  nhalo++;

  return 1;
}

void cola_halo_file_close()
{
  int nhalo_check= 0;
  int ret= fread(&nhalo_check, sizeof(int), 1, fp); assert(ret == 1);
  assert(nhalo == nhalo_check);

  ret= fclose(fp); assert(ret == 0);
}