Return to jretile_maxmem.c CVS log

Up to [Development] / JSOC / proj / globalhs / apps

get rid of compiler warnings

char *cvsinfo_jretile_maxmem = "cvsinfo: $Header: /home/cvsuser/cvsroot/JSOC/proj/globalhs/apps/jretile_maxmem.c,v 1.6 2017/03/31 20:02:28 tplarson Exp $";

int jretile_maxmem(void)
{
  int newstat = 0;
  int status = DRMS_SUCCESS;

  char *inrecquery = NULL;
  char *outseries = NULL;
  char *segnamein = NULL;
  char *segnameout = NULL;
  DRMS_RecordSet_t *inrecset = NULL;
  DRMS_RecordSet_t *outrecset = NULL;
  int irecin, irecout, nrecsin=0, nrecsout=0, nlchunks;
  DRMS_Record_t *inrec = NULL;
  DRMS_Record_t *outrec = NULL;
  DRMS_Segment_t *segin = NULL;
  DRMS_Segment_t *segout = NULL;
  DRMS_Array_t *inarr = NULL;
  DRMS_Array_t *outarr = NULL;
  DRMS_RecLifetime_t lifetime;
  DRMS_Type_t usetype = DRMS_TYPE_FLOAT;
  int length[2], startind[2], endind[2];
  float *inptr, *outptr;
  long long histrecnum=-1;
  int quality;
  int mapmmax=-1;
  int sinbdivs=-1;
  double cadence=0;
  int i;

  TIME tnow, UNIX_epoch = -220924792.000; /* 1970.01.01_00:00:00_UTC */
  char tstartstr[100];

  double tstart, tepoch, tstep, tround, tstop, tstartin, tstopin, tstepin, tstartuse, tstopuse, nseconds, chunksecs;
  char *ttotal, *tchunk;
  int ndt, ndtin;
  int lmin, lmax, lminin, lmaxin, lminuse, lmaxuse, lchunk, lchunksize, lchunkfirst, lchunklast;
  int ntimechunks, nmodes, npts, imode, itime;
  int out_time_offset, out_modes_offset, out_offset, in_time_offset, in_modes_offset, in_offset, out_index, in_index;
  int iset, lminout, lmaxout;
  double tstartout, tstopout;
  float *arrptr;
  int nrecords, nsegments, nskip;

  int errbufstat=setvbuf(stderr, NULL, _IONBF, BUFSIZ);
  int outbufstat=setvbuf(stdout, NULL, _IONBF, BUFSIZ);

  double wt0, wt1, wt2, wt3, wt;
  double ut0, ut1, ut2, ut3, ut;
  double st0, st1, st2, st3, st;
  double ct0, ct1, ct2, ct3, ct;

  wt0=getwalltime();
  ct0=getcputime(&ut0, &st0);

  inrecquery = (char *)cmdparams_save_str(&cmdparams, "in", &newstat);
  outseries = (char *)cmdparams_save_str(&cmdparams, "out", &newstat);
  segnamein = (char *)cmdparams_save_str(&cmdparams, "segin", &newstat);
  segnameout = (char *)cmdparams_save_str(&cmdparams, "segout", &newstat);
  int seginflag = strcmp(kNOTSPECIFIED, segnamein);
  int segoutflag = strcmp(kNOTSPECIFIED, segnameout);
  int verbflag = cmdparams_save_int(&cmdparams, "VERB", &newstat);
  int permflag = cmdparams_save_int(&cmdparams, "PERM", &newstat);
  if (permflag)
    lifetime = DRMS_PERMANENT;
  else
    lifetime = DRMS_TRANSIENT;
  unsigned short calverkey = (unsigned short)cmdparams_save_int(&cmdparams, "CALVERKEY", &newstat);

  char *histlinkname = (char *)cmdparams_save_str(&cmdparams, "histlink", &newstat);

  tstart=cmdparams_save_time(&cmdparams, "TSTART", &newstat);
  ttotal=(char *)cmdparams_save_str(&cmdparams, "TTOTAL", &newstat);
  status=drms_names_parseduration(&ttotal, &nseconds, 1);
  if (status != DRMS_SUCCESS)
  {
//    newstat = newstat | CPSAVE_UNKNOWN_ERROR;
    fprintf(stderr, "ERROR: problem parsing TTOTAL, = %s\n", ttotal);
    return 1; 
  }
  tchunk=(char *)cmdparams_save_str(&cmdparams, "TCHUNK", &newstat);
  if (strcmp(kNOTSPECIFIED, tchunk))
  {
    status=drms_names_parseduration(&tchunk, &chunksecs, 1);
    if (status != DRMS_SUCCESS)
      newstat = newstat | CPSAVE_UNKNOWN_ERROR;
  }
  else
    chunksecs=0;

  lmin=cmdparams_save_int(&cmdparams, "LMIN", &newstat);
  lmax=cmdparams_save_int(&cmdparams, "LMAX", &newstat);
  lchunksize=cmdparams_save_int(&cmdparams, "LCHUNK", &newstat);
  if (lchunksize == 0)
    lchunksize=lmax+1;

  if (newstat) 
  {
    fprintf(stderr, "ERROR: problem with input arguments, status = %d, diagnosis follows\n", newstat);
    cpsave_decode_error(newstat);
    return 1;
  }  
  else if (savestrlen != strlen(savestr)) 
  {
    fprintf(stderr, "ERROR: problem with savestr, savestrlen = %d, strlen(savestr) = %d\n", savestrlen, (int)strlen(savestr));
    return 1;
  }

  DRMS_Record_t *tempoutrec = drms_create_record(drms_env, 
                                                 outseries,
                                                 DRMS_TRANSIENT, 
                                                 &status);

  if (status != DRMS_SUCCESS) 
  {
   fprintf(stderr,"ERROR: couldn't open a record in output dataseries %s, status = %d\n", outseries, status);
   return 1;
  }

// set up ancillary dataseries for processing metadata
  DRMS_Link_t *histlink = hcon_lookup_lower(&tempoutrec->links, histlinkname);
  if (histlink != NULL) 
  {
    histrecnum=set_history(histlink);
    if (histrecnum < 0)
    {
      drms_close_record(tempoutrec, DRMS_FREE_RECORD);
      return 1;
    }
  }
  else
  {
    fprintf(stderr,"WARNING: could not find history link in output dataseries\n");
  }

// these must be present in the output dataseries and variable, not links or constants
// now done in DoIt() that calls this function
/*
  char *outchecklist[] = {"T_START", "QUALITY", "LMIN", "LMAX", "NDT"};
  int itest;
  for (itest=0; itest < ARRLENGTH(outchecklist); itest++)
  {
    DRMS_Keyword_t *outkeytest = hcon_lookup_lower(&tempoutrec->keywords, outchecklist[itest]);
    if (outkeytest == NULL || outkeytest->info->islink || outkeytest->info->recscope == 1)
    {
      fprintf(stderr, "ERROR: output keyword %s is either missing, constant, or a link\n", outchecklist[itest]);
      drms_close_record(tempoutrec, DRMS_FREE_RECORD);
      return 1;
    }
  }
*/

  tepoch=drms_getkey_time(tempoutrec, "T_START_epoch", &status);
  tstep=drms_getkey_float(tempoutrec, "T_START_step", &status);
  tround=drms_getkey_float(tempoutrec, "T_START_round", &status);
  cadence=drms_getkey_float(tempoutrec, "T_STEP", &status);
  if (fmod(tstart-tepoch,tstep) > tround/2)
  {
    fprintf(stderr, "ERROR: output dataseries seems incompatible with input parameters (tstep must divide tstart-tepoch): tstart = %f, tepoch = %f, tstep = %f\n", tstart, tepoch, tstep);
    drms_close_record(tempoutrec, DRMS_FREE_RECORD);
    return 1;
  }
  if (chunksecs == 0.0)
    chunksecs = tstep;
  else if (fmod(chunksecs,tstep))
  {
    fprintf(stderr, "ERROR: output dataseries seems incompatible with input parameters (tstep must divide chunksecs): chunksecs = %f, tstep = %f\n", chunksecs, tstep);
    drms_close_record(tempoutrec, DRMS_FREE_RECORD);
    return 1;
  }
  if (fmod(nseconds,chunksecs) != 0.0)
  {
    fprintf(stderr, "ERROR: input parameters seem incompatible (chunksecs must divide nseconds): nseconds = %f, chunksecs = %f\n", nseconds, chunksecs);
    drms_close_record(tempoutrec, DRMS_FREE_RECORD);
    return 1;
  }
  ntimechunks=nseconds/chunksecs;
  ndt=chunksecs/cadence;
  if (verbflag)
  {
    printf("%d timechunks, %.1f seconds per chunk\n", ntimechunks, chunksecs);
  }

  int mapmmaxout=-1;
  int sinbdivsout=-1;
  DRMS_Keyword_t *outkeytest = hcon_lookup_lower(&tempoutrec->keywords, "MAPMMAX");
  if (outkeytest != NULL && outkeytest->info->recscope == 1)
    mapmmaxout=drms_getkey_int(tempoutrec, "MAPMMAX", &status);
  outkeytest = hcon_lookup_lower(&tempoutrec->keywords, "SINBDIVS");
  if (outkeytest != NULL && outkeytest->info->recscope == 1)
    sinbdivsout=drms_getkey_int(tempoutrec, "SINBDIVS", &status);

  float apinnerout=0.0;
  float apwidthout=0.0;
  float apinner=0.0;
  float apwidth=0.0;
  outkeytest = hcon_lookup_lower(&tempoutrec->keywords, "APINNER");
  if (outkeytest != NULL && outkeytest->info->recscope == 1)
    apinnerout=drms_getkey_float(tempoutrec, "APINNER", &status);
  outkeytest = hcon_lookup_lower(&tempoutrec->keywords, "APWIDTH");
  if (outkeytest != NULL && outkeytest->info->recscope == 1)
    apwidthout=drms_getkey_float(tempoutrec, "APWIDTH", &status);

  drms_close_record(tempoutrec, DRMS_FREE_RECORD);


  char *inchecklist[] = {"T_START", "QUALITY", "LMIN", "LMAX", "T_STEP"};
  DRMS_Keyword_t *inkeytest;
  int itest;
  inrecset = drms_open_records(drms_env, inrecquery, &status);
//  inrecset = drms_open_recordset(drms_env, inrecquery, &status);

  if (status != DRMS_SUCCESS || inrecset == NULL)
  {
    fprintf(stderr, "ERROR: problem opening input recordset: status = %d\n", status);
    return 1;
  }
  nrecsin = inrecset->n;

  if (verbflag) 
    printf("input recordset opened, nrecs = %d\n", nrecsin);

  int noinput=0;
  if (nrecsin == 0)
  {
    printf("WARNING: input recordset contains no records\n");
    noinput=1;
    goto skip1;
//    return 1;
  }

  inrec = inrecset->records[0];

  for (itest=0; itest < ARRLENGTH(inchecklist); itest++)
  {
    inkeytest = hcon_lookup_lower(&inrec->keywords, inchecklist[itest]);
    if (inkeytest == NULL)
    {
      fprintf(stderr, "ERROR: required input keyword %s is missing\n", inchecklist[itest]);
      drms_close_records(inrecset, DRMS_FREE_RECORD);
      return 1;
    }
  }

  if (cadence != drms_getkey_float(inrec, "T_STEP", &status))
  {
    fprintf(stderr, "ERROR: input T_STEP does not equal output T_STEP\n");
    drms_close_records(inrecset, DRMS_FREE_RECORD);
    return 1;
  }

  inkeytest = hcon_lookup_lower(&inrec->keywords, "MAPMMAX");
  if (inkeytest != NULL)
    mapmmax=drms_getkey_int(inrec, "MAPMMAX", &status);
  if (mapmmaxout != -1 && mapmmaxout != mapmmax)
  {
    fprintf(stderr, "ERROR: input MAPMMAX does not equal output MAPMMAX, in=%d, out=%d\n", mapmmax, mapmmaxout);
    drms_close_records(inrecset, DRMS_FREE_RECORD);
    return 1;
  }

  inkeytest = hcon_lookup_lower(&inrec->keywords, "SINBDIVS");
  if (inkeytest != NULL)
    sinbdivs=drms_getkey_int(inrec, "SINBDIVS", &status);
  if (sinbdivsout != -1 && sinbdivsout != sinbdivs)
  {
    fprintf(stderr, "ERROR: input SINBDIVS does not equal output SINBDIVS, in=%d, out=%d\n", sinbdivs, sinbdivsout);
    drms_close_records(inrecset, DRMS_FREE_RECORD);
    return 1;
  }

  inkeytest = hcon_lookup_lower(&inrec->keywords, "APINNER");
  if (inkeytest != NULL)
    apinner=drms_getkey_float(inrec, "APINNER", &status);
  if (apinnerout != 0.0 && (int)(10000*apinnerout) != (int)(10000*apinner))
  {
    fprintf(stderr, "ERROR: input APINNER does not equal output APINNER, in=%f, out=%f\n",apinner, apinnerout);
    drms_close_records(inrecset, DRMS_FREE_RECORD);
    return 1;
  }

  inkeytest = hcon_lookup_lower(&inrec->keywords, "APWIDTH");
  if (inkeytest != NULL)
    apwidth=drms_getkey_float(inrec, "APWIDTH", &status);
  if (apinnerout != 0.0 && (int)(10000*apwidthout) != (int)(10000*apwidth))
  {
    fprintf(stderr, "ERROR: input APWIDTH does not equal output APWIDTH, in=%f, out=%f\n", apwidth, apwidth);
    drms_close_records(inrecset, DRMS_FREE_RECORD);
    return 1;
  }


  status=drms_stage_records(inrecset, 1, 0);
  if (status != DRMS_SUCCESS)
  {
    fprintf(stderr, "ERROR: drms_stage_records returned status = %d\n", status);
    return 1;
  }

  skip1:

  nrecords=0;
  nsegments=0;
  nskip=0;

  lchunkfirst = lmin/lchunksize;
  lchunklast = lmax/lchunksize;

  nlchunks = (lchunklast - lchunkfirst) + 1;
  nrecsout = nlchunks*ntimechunks;
  outrecset = drms_create_records(drms_env, nrecsout, outseries, lifetime, &status);
  if (status != DRMS_SUCCESS || outrecset == NULL)
  {
    fprintf(stderr,"ERROR: unable to create records record in output dataseries %s, status = %d\n", outseries, status);
    drms_close_records(inrecset, DRMS_FREE_RECORD);
    return 1;
  }

  unsigned long long calversout, calvers;
  int calversunset=1;
  int mixflag=0;
  unsigned int nybblearrout[16] = {0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0};
  int fixflagarr[16]            = {0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0};
  for (i=0;i<16;i++)
  {
    if (getbits(calverkey,i,1))
      fixflagarr[i]=1;
  }

  int *firstflagarr=(int *)malloc(nrecsin*sizeof(int));
  DRMS_Array_t **inarrarr=(DRMS_Array_t **)malloc(nrecsin*sizeof(DRMS_Array_t *));
  for (irecin=0; irecin < nrecsin; irecin++)
  {
    firstflagarr[irecin]=1;
    inarrarr[irecin]=NULL;
  }

  irecout=0;
  for (iset=0; iset < ntimechunks; iset++)
  {
    tstartout=tstart + iset * chunksecs;
    tstopout=tstartout+chunksecs;
    sprint_time(tstartstr, tstartout, "TAI", 0);

    if (verbflag)
    {
      wt1=getwalltime();
      ct1=getcputime(&ut1, &st1);
      printf("processing timechunk %d, tstart = %s\n", iset, tstartstr);
    }

    for (lchunk = lchunkfirst; lchunk <= lchunklast; lchunk++)
    {
      lminout = lchunk * lchunksize; 
      lmaxout = lminout + lchunksize - 1;
      lminout = MAXIMUM(lminout,lmin);
      lmaxout = MINIMUM(lmaxout,lmax);

      outrec = outrecset->records[irecout++];
      if (histlink != NULL)
        drms_setlink_static(outrec, histlinkname,  histrecnum);

      if (verbflag > 1)
      {
        wt2=getwalltime();
        ct2=getcputime(&ut2, &st2);
        printf("  processing lchunk %d, min = %d, max = %d\n", lchunk-lchunkfirst, lminout, lmaxout);
      }

      if (noinput)
      {
        goto skip2;
      }

      if (segoutflag)
        segout = drms_segment_lookup(outrec, segnameout);
      else
        segout = drms_segment_lookupnum(outrec, 0);

      length[0]=2*ndt;
      length[1]=lmaxout*(lmaxout+1)/2+lmaxout - lminout*(lminout+1)/2 + 1;
      arrptr=(float *)(calloc(length[0]*length[1],sizeof(float)));
      outarr = drms_array_create(usetype, 2, length, arrptr, &status);
      if (status != DRMS_SUCCESS || outarr == NULL || segout == NULL)
      {
        fprintf(stderr,"ERROR: problem creating output array or segment: length = [%d, %d], status = %d, histrecnum = %lld", length[0], length[1], status, histrecnum);
        drms_close_records(inrecset, DRMS_FREE_RECORD);
        drms_close_records(outrecset, DRMS_FREE_RECORD);
        return 0; 
      }
      outptr = (float *)(outarr->data);

      nskip=0;
      mixflag=0;
      calversunset=1;
      for (irecin=0; irecin < nrecsin; irecin++)
      {
        inrec = inrecset->records[irecin];
        quality=drms_getkey_int(inrec, "QUALITY", &status);
        if (status != DRMS_SUCCESS || (quality & QUAL_NODATA)) //may want stricter test on quality here
        {
          if (verbflag>2) printf("SKIP: input data not used due to quality, irecin = %d, status = %d, quality = %08x\n", irecin, status, quality);
          nskip++;
          continue;
        }

        tstartin=drms_getkey_time(inrec, "T_START", &status);
        tstopin=drms_getkey_time(inrec, "T_STOP", &status);
        lminin=drms_getkey_int(inrec, "LMIN", &status);
        lmaxin=drms_getkey_int(inrec, "LMAX", &status);
        ndtin=(tstopin-tstartin)/cadence;

        if (tstartin >= tstopout || tstopin <= tstartout || lminin > lmaxout || lmaxin < lminout)
        {
          nskip++;
          continue;
        }

        if (calversunset)
        {
          calversout=drms_getkey_longlong(inrec, "CALVER64", &status);
          if (status != DRMS_SUCCESS)
            calversout = 0;
          calversunset=0;
          for (i=0;i<16;i++)
            nybblearrout[i]=getbits(calversout,4*i+3,4);
        }

        calvers=drms_getkey_longlong(inrec, "CALVER64", &status);
        if (status != DRMS_SUCCESS)
          calvers = 0;

        for (i=0;i<16;i++)
        {
          int nybble=getbits(calvers,4*i+3,4);
          if (fixflagarr[i])
          {
            if (nybble != nybblearrout[i])
            {
              fprintf(stderr, "ERROR: input data has mixed values for field %d of CALVER64: %d and %d, recnum = %lld, histrecnum = %lld\n", i, nybblearrout[i], nybble, inrec->recnum, histrecnum);
              return 0;
            }
          }
          else
          {
            if (nybble < nybblearrout[i])
              nybblearrout[i]=nybble;
          }
        }

        if (!mixflag && (quality & QUAL_MIXEDCALVER || calvers != calversout))
          mixflag=1;

        if (firstflagarr[irecin])
        {
          if (seginflag)
            segin = drms_segment_lookup(inrec, segnamein);
          else
            segin = drms_segment_lookupnum(inrec, 0);
          if (segin != NULL)
            inarr = drms_segment_read(segin, usetype, &status);
          if (status != DRMS_SUCCESS || inarr == NULL || segin == NULL)
          {
            fprintf(stderr, "ERROR: problem reading input segment: tstart = %f, lmin = %d, lmax = %d, irecin = %d, status = %d, histrecnum = %lld\n", tstartin, lminin, lmaxin, irecin, status, histrecnum);
            drms_close_records(inrecset, DRMS_FREE_RECORD);
            drms_close_records(outrecset, DRMS_FREE_RECORD);
            return 0;
          }
          else
          {
            inarrarr[irecin]=inarr;
          }
          firstflagarr[irecin]=0;
        }

        inptr = (float *)(inarrarr[irecin])->data;

        tstartuse=MAXIMUM(tstartout, tstartin);
        tstopuse= MINIMUM(tstopout, tstopin);
        lminuse=MAXIMUM(lminout, lminin);
        lmaxuse=MINIMUM(lmaxout, lmaxin);
        nmodes=MODES(lmaxuse+1)-MODES(lminuse);
        npts=(tstopuse - tstartuse)/cadence;

        out_time_offset = (tstartuse - tstartout)/cadence;
        out_modes_offset = MODES(lminuse) - MODES(lminout);
        out_offset = 2 * (out_modes_offset * ndt + out_time_offset);
        in_time_offset = (tstartuse - tstartin)/cadence;
        in_modes_offset = MODES(lminuse) - MODES(lminin);
        in_offset =  2 * (in_modes_offset * ndtin + in_time_offset);

        for (imode=0; imode<nmodes; imode++)
        {
          for (itime=0; itime<npts; itime++)
          {
            in_index=in_offset + 2*itime;
            out_index=out_offset + 2*itime;
            outptr[out_index] = inptr[in_index];
            outptr[out_index+1] = inptr[in_index+1];
          }
          out_offset+=2*ndt;
          in_offset+=2*ndtin;
        }

      }

      if (nskip == nrecsin)
      {
        fprintf(stderr, "WARNING: no inputs for this output, no segment written\n");
      }
      else
      {
        outarr->bzero=segout->bzero;
        outarr->bscale=segout->bscale;
        status=drms_segment_write(segout, outarr, 0);
        nsegments++;
        if (status != DRMS_SUCCESS)
        {
          fprintf(stderr, "ERROR: problem writing output segment: status = %d, T_START = %s, LMIN = %d, LMAX = %d, histrecnum = %lld\n", status, tstartstr, lminout, lmaxout, histrecnum);
          drms_close_records(inrecset, DRMS_FREE_RECORD);
          drms_close_records(outrecset, DRMS_FREE_RECORD);
          return 0;
        }
      }
      drms_free_array(outarr);

      drms_copykeys(outrec, inrec, 0, kDRMS_KeyClass_Explicit);
// copykeys takes care of MAPMMAX, SINBDIVS, etc. that should be constant across the input
      skip2:

      drms_setkey_int(outrec, "LMIN", lminout);
      drms_setkey_int(outrec, "LMAX", lmaxout);
      drms_setkey_time(outrec, "T_START", tstartout);
      drms_setkey_time(outrec, "T_STOP", tstopout);
      drms_setkey_time(outrec, "T_OBS", tstartout+chunksecs/2);
      drms_setkey_int(outrec, "NDT", ndt);

      for (i=0;i<16;i++)
        calversout=setbits(calversout,4*i+3,4,nybblearrout[i]);
      drms_setkey_longlong(outrec, "CALVER64", calversout);

      if (noinput || nskip == nrecsin)
        drms_setkey_int(outrec, "QUALITY", QUAL_NODATA);
      else if (mixflag)
        drms_setkey_int(outrec, "QUALITY", QUAL_MIXEDCALVER);
      else
        drms_setkey_int(outrec, "QUALITY", 0);

      tnow = (double)time(NULL);
      tnow += UNIX_epoch;
      drms_setkey_time(outrec, "DATE", tnow);

      nrecords++;

      if (verbflag > 1)
      {
        wt=getwalltime();
        ct=getcputime(&ut, &st);
        fprintf(stdout, "  lchunk %d done: %.2f ms wall time, %.2f ms cpu time\n", lchunk-lchunkfirst,
                wt-wt2,
                ct-ct2);
      }

    }

    if (verbflag)
    {
      wt=getwalltime();
      ct=getcputime(&ut, &st);
      fprintf(stdout, "timechunk %d done: %.2f ms wall time, %.2f ms cpu time\n", iset,
              wt-wt1,
              ct-ct1);

    }

  }

  for (irecin=0; irecin < nrecsin; irecin++)
    if (inarrarr[irecin] != NULL) drms_free_array(inarrarr[irecin]);
  free(inarrarr);
  free(firstflagarr);

  drms_close_records(inrecset, DRMS_FREE_RECORD);
  drms_close_records(outrecset, DRMS_INSERT_RECORD);

  wt=getwalltime();
  ct=getcputime(&ut, &st);
  if (verbflag) 
  {
    printf("number of records created  = %d\n", nrecords);
    printf("number of segments created = %d\n", nsegments);
    fprintf(stdout, "total time spent: %.2f ms wall time, %.2f ms cpu time\n", 
            wt-wt0, ct-ct0);
  }

  printf("module %s successful completion\n", cmdparams.argv[0]);

  return 0;

}