io_progenitors.h

Go to the documentation of this file.

/*******************************************************************************
 * \copyright   This file is part of the GADGET4 N-body/SPH code developed
 * \copyright   by Volker Springel. Copyright (C) 2014-2020 by Volker Springel
 * \copyright   (vspringel@mpa-garching.mpg.de) and all contributing authors.
 *******************************************************************************/
 
#ifndef PROGENITORS_IO_H
#define PROGENITORS_IO_H
 
#include "gadgetconfig.h"
 
#ifdef MERGERTREE
 
#include "../data/allvars.h"
#include "../data/dtypes.h"
#include "../data/mymalloc.h"
#include "../fof/fof.h"
#include "../io/hdf5_util.h"
#include "../io/io.h"
#include "../logs/timer.h"
#include "../main/simulation.h"
#include "../mergertree/mergertree.h"
#include "../mpi_utils/mpi_utils.h"
#include "../sort/parallel_sort.h"
#include "../subfind/subfind.h"
#include "../system/system.h"
 
class progenitors_io : public IO_Def
{
 private:
  mergertree *MergerTree;
 
 public:
  progenitors_io(mergertree *MergerTree_ptr, MPI_Comm comm, int format);
 
  void mergertree_read_progenitors(int num);
  void mergertree_save_progenitors(int num);
 
  /* supplied virtual functions */
  void fill_file_header(int writeTask, int lastTask, long long *nloc_part, long long *npart);
  void read_file_header(const char *fname, int filenr, int readTask, int lastTask, long long *nloc_part, long long *npart,
                        int *nstart);
  void get_datagroup_name(int grnr, char *gname);
  void write_header_fields(hid_t);
  void read_header_fields(const char *fname);
  void read_increase_numbers(int type, int n_for_this_task);
  int get_filenr_from_header(void);
  void set_filenr_in_header(int);
  void *get_base_address_of_structure(enum arrays array, int index);
  int get_type_of_element(int index);
  void set_type_of_element(int index, int type);
 
  struct io_header
  {
    long long Nsubhalos;
    long long TotNsubhalos;
    int num_files;
  };
  io_header header;
 
  int Nsubhalos;
  long long TotNsubhalos;
 
 private:
  static void io_func_progsubhalonr(IO_Def *ptr, int particle, int components, void *buffer, int mode)
  {
    progenitors_io *thisobj = (progenitors_io *)ptr;
 
    if(mode == 0)
      {
        long long *out_buffer = (long long *)buffer;
        out_buffer[0]         = thisobj->MergerTree->Progenitors[particle].ProgSubhaloNr;
        if(out_buffer[0] == HALONR_MAX)
          out_buffer[0] = -1;
      }
    else
      {
        long long *in_buffer                                     = (long long *)buffer;
        thisobj->MergerTree->Progenitors[particle].ProgSubhaloNr = in_buffer[0];
        if(thisobj->MergerTree->Progenitors[particle].ProgSubhaloNr == -1)
          thisobj->MergerTree->Progenitors[particle].ProgSubhaloNr = HALONR_MAX;
      }
  }
 
  static void io_func_firstprogsubhalonr(IO_Def *ptr, int particle, int components, void *buffer, int mode)
  {
    progenitors_io *thisobj = (progenitors_io *)ptr;
 
    if(mode == 0)
      {
        long long *out_buffer = (long long *)buffer;
        out_buffer[0]         = thisobj->MergerTree->Progenitors[particle].FirstProgSubhaloNr;
        if(out_buffer[0] == HALONR_MAX)
          out_buffer[0] = -1;
      }
    else
      {
        long long *in_buffer                                          = (long long *)buffer;
        thisobj->MergerTree->Progenitors[particle].FirstProgSubhaloNr = in_buffer[0];
        if(thisobj->MergerTree->Progenitors[particle].FirstProgSubhaloNr == -1)
          thisobj->MergerTree->Progenitors[particle].FirstProgSubhaloNr = HALONR_MAX;
      }
  }
 
  static void io_func_nextdescsubhalonr(IO_Def *ptr, int particle, int components, void *buffer, int mode)
  {
    progenitors_io *thisobj = (progenitors_io *)ptr;
 
    if(mode == 0)
      {
        long long *out_buffer = (long long *)buffer;
        out_buffer[0]         = thisobj->MergerTree->Progenitors[particle].NextDescSubhaloNr;
        if(out_buffer[0] == HALONR_MAX)
          out_buffer[0] = -1;
      }
    else
      {
        long long *in_buffer                                         = (long long *)buffer;
        thisobj->MergerTree->Progenitors[particle].NextDescSubhaloNr = in_buffer[0];
        if(thisobj->MergerTree->Progenitors[particle].NextDescSubhaloNr == -1)
          thisobj->MergerTree->Progenitors[particle].NextDescSubhaloNr = HALONR_MAX;
      }
  }
};
 
#endif
 
#endif /* DESCENDANT_IO_H */