Subversion Repositories f9daq

#ifdef _CVI_

#  include <ansi_c.h>

# else _CVI_

#  include <stdlib.h>

#  include <stdio.h>

#  include <string.h>

#endif _CVI_

#include "H3D.h"

#define H3DMAX 500

H3D *h3[H3DMAX];

//int Printf(char *format, ...);

int H3DClear(int h3d) {

  if (!h3[h3d]) return -1;

  memset(h3[h3d]->data, 0,h3[h3d]->size );

  h3[h3d]->min=0;

  h3[h3d]->max=0;

  h3[h3d]->nentries=0;

  return  0;

}

int H3DPrint(int h3d) {

  if (!h3[h3d]) return -1;

//Printf("PrintH3D nx=%d minx=%f stepx=%f ny=%d miny=%f stepy=%f size=%d\n", h3[h3d]->nx, h3[h3d]->minx, h3[h3d]->stepx, h3[h3d]->ny, h3[h3d]->miny, h3[h3d]->stepy, h3[h3d]->size ) ;

  return  0;

}

int H3DExist(int h) {

  if (h3[h]) return 1;

  else return 0;

}

int H3DCalculateBin(int h, int axis, double value) {

  int nx=1,xmin=0,dx=0;

  int bin;

  switch (axis) {

    case 0:

      nx = H3DGetNbinsX(h);

      xmin= H3DGetMinX(h);

      dx = H3DGetStepX(h);

      break;

    case 1:

      nx = H3DGetNbinsY(h);

      xmin= H3DGetMinY(h);

      dx = H3DGetStepY(h);

      break;

    case 2:

      nx = H3DGetNbinsZ(h);

      xmin= H3DGetMinZ(h);

      dx = H3DGetStepZ(h);

      break;

  }

  if (dx==0) return -1;

  if (value<xmin) return -1;

  bin = (int)((value-xmin)/dx);

  if (bin>=nx) return -1;

  else return bin;

}

int H3DFill(int h3d,double x, double y, double z, double val) {

  int idx, ix,iy, iz;

  if (!h3[h3d]) return -1;

  ix = H3DCalculateBin(h3d,0,x);

  if (ix<0) return ix;

  iy = H3DCalculateBin(h3d,1,y);

  if (iy<0) return iy;

  iz = H3DCalculateBin(h3d,2,z);

  if (iz<0) return iz;

  H3DFillBin(h3d, ix, iy, iz, val);

  return 0;

}

int H3DGetBin(int h3d,int x, int y, int z) {

  return x+h3[h3d]->nx *(y+z*h3[h3d]->ny);

}

int H3DFillBin(int h3d,int x, int y, int z, double val) {

  int idx;

  if (!h3[h3d]) {

//Printf("FillH3D error h3d is not initialized\n");

    return -1;

  }

  idx = H3DGetBin(h3d,x,y,z);

  h3[h3d]->data[idx]+=val;

//Printf("%d %d data %f %f\n",x,y,val, h3[h3d]->data[idx]);

  if (h3[h3d]->data[idx]>h3[h3d]->max) h3[h3d]->max= h3[h3d]->data[idx];

  if (h3[h3d]->data[idx]<h3[h3d]->min) h3[h3d]->min= h3[h3d]->data[idx];

  h3[h3d]->nentries++;

  return 0;

}

double H3DGetBinContent(int h3d,int atx, int aty, int atz) {

  int idx;

  if (!h3[h3d]) return 0;

  if (h3[h3d]->nx <= atx)  return 0;

  if (h3[h3d]->ny <= aty)  return 0;

  if (h3[h3d]->nz <= atz)  return 0;

  idx = H3DGetBin(h3d,atx,aty,atz);

  if (idx*sizeof(double) < h3[h3d]->size ) return h3[h3d]->data[idx];

  return 0;

}

int H3DInit(int h3d,char *name, char *title,int nx, double minx, double stepx, int ny, double miny, double stepy, int nz, double minz, double stepz) {

  if (h3[h3d]) {

    free(h3[h3d]->data);

    free(h3[h3d]);

    h3[h3d] = NULL;

  }

  //printf("InitH3D hID=%d\n",h3d);

  h3[h3d] = (H3D *) malloc(sizeof(H3D));

//h2  =h3d;

  H3DSetTitle(h3d,title);

  H3DSetName(h3d,name);

  h3[h3d]->id=H3D_ID;

  h3[h3d]->nx = nx;

  h3[h3d]->ny = ny;

  h3[h3d]->nz = nz;

  h3[h3d]->minx = minx;

  h3[h3d]->miny = miny;

  h3[h3d]->minz = minz;

  h3[h3d]->stepx = stepx;

  h3[h3d]->stepy = stepy;

  h3[h3d]->stepz = stepz;

  h3[h3d]->size = h3[h3d]->nx*h3[h3d]->ny*h3[h3d]->nz*sizeof(double);

  h3[h3d]->data = (double *) malloc(h3[h3d]->size);

  h3[h3d]->len=sizeof(H3D)-sizeof(double *)+h3[h3d]->size;

  H3DClear(h3d);

  H3DPrint(h3d);

//Printf("InitH3D 0x%x\n", h3d );

  return  0;

}

double H3DGetXBinCenter(int h3d,int xbin) {

  return h3[h3d]->minx+xbin*h3[h3d]->stepx;

}

double H3DGetYBinCenter(int h3d,int ybin) {

  return h3[h3d]->miny+ybin*h3[h3d]->stepy;

}

double H3DGetZBinCenter(int h3d,int zbin) {

  return h3[h3d]->minz+zbin*h3[h3d]->stepz;

}

int H3DWrite2File(int h3d,FILE *fp) {

  if (!fp) return -1;

//printf("H3D sizeof(H3D)=%lu len-datasize=%d len=%lu datasize=%d\t",sizeof(H3D)-sizeof(double *),h3[h3d]->len-h3[h3d]->size,h3[h3d]->len,h3[h3d]->size);

//printf("H3D sz=%d %d\n",sizeof(double),sizeof(double *));

  fwrite (h3[h3d], 1, sizeof(H3D)-sizeof(double *), fp);

  fwrite (h3[h3d]->data, 1, h3[h3d]->size, fp);

  return  0;

}

int H3DWrite(int h3d,const char *fname,const char *opt) {

  FILE *fp=fopen(fname,opt);

  H3DWrite2File(h3d,fp);

  fclose(fp);

  return  0;

}

int H3DSetTitle(int h3d,char *title) {

  sprintf(h3[h3d]->title,"%s",title);

  return 0;

}

int H3DSetTitleX(int h3d,char *title) {

  sprintf(h3[h3d]->titlex,"%s",title);

  return 0;

}

int H3DSetTitleY(int h3d,char *title) {

  sprintf(h3[h3d]->titley,"%s",title);

  return 0;

}

int H3DSetTitleZ(int h3d,char *title) {

  sprintf(h3[h3d]->titlez,"%s",title);

  return 0;

}

int H3DSetName(int h3d,char *title) {

  sprintf(h3[h3d]->name,"%s",title);

  return 0;

}

int H3DGetNbinsY(int h) {

  if (h3[h]) return h3[h]->ny;

  else return 0;

}

int H3DGetNbinsX(int h) {

  if (h3[h]) return h3[h]->nx;

  else return 0;

}

int H3DGetNbinsZ(int h) {

  if (h3[h]) return h3[h]->nz;

  else return 0;

}

double H3DGetMinX(int h) {

  if (h3[h]) return h3[h]->minx;

  else return 0;

}

double H3DGetMinY(int h) {

  if (h3[h]) return h3[h]->miny;

  else return 0;

}

double H3DGetMinZ(int h) {

  if (h3[h]) return h3[h]->minz;

  else return 0;

}

double H3DGetStepX(int h) {

  if (h3[h]) return h3[h]->stepx;

  else return 0;

}

double H3DGetStepY(int h) {

  if (h3[h]) return h3[h]->stepy;

  else return 0;

}

double H3DGetStepZ(int h) {

  if (h3[h]) return h3[h]->stepz;

  else return 0;

}

double H3DGetMin(int h) {

  if (h3[h]) return h3[h]->min;

  else return 0;

}

double H3DGetMax(int h) {

  if (h3[h]) return h3[h]->max;

  else return 0;

}

double * H3DGetSliceXYData(int h, int slice) {

  if (h3[h]) return (h3[h]->data+slice*h3[h]->nx*h3[h]->ny);

  else return NULL;

}

double * H3DGetData(int h) {

  if (h3[h]) return h3[h]->data;

  else return NULL;

}