tape/recorder.c

#include <stdlib.h>
#include <stdio.h>
#include <errno.h>
#include <ctype.h>
#include "gridlabd.h"
#include "object.h"
#include "aggregate.h"

#include "tape.h"
#include "file.h"
#include "odbc.h"

#ifndef WIN32
#define strtok_s strtok_r
#endif

CLASS *recorder_class = NULL;
static OBJECT *last_recorder = NULL;

EXPORT int create_recorder(OBJECT **obj, OBJECT *parent)
{
    *obj = gl_create_object(recorder_class);
    if (*obj!=NULL)
    {
        struct recorder *my = OBJECTDATA(*obj,struct recorder);
        last_recorder = *obj;
        gl_set_parent(*obj,parent);
        strcpy(my->file,"");
        strcpy(my->multifile,"");
        strcpy(my->filetype,"txt");
        strcpy(my->delim,",");
        strcpy(my->property,"(undefined)");
        my->interval = -1; /* transients only */
        my->dInterval = -1.0;
        my->last.ts = -1;
        strcpy(my->last.value,"");
        my->limit = 0;
        my->samples = 0;
        my->status = TS_INIT;
        my->trigger[0]='\0';
        my->format = 0;
        strcpy(my->plotcommands,"");
        my->target = gl_get_property(*obj,my->property);
        return 1;
    }
    return 0;
}

static int recorder_open(OBJECT *obj)
{
    char32 type="file";
    char1024 fname="";
    char32 flags="w";
    struct recorder *my = OBJECTDATA(obj,struct recorder);
    
    my->interval = (int64)(my->dInterval/TS_SECOND);
    /* if prefix is omitted (no colons found) */
    if (sscanf(my->file,"%32[^:]:%1024[^:]:%[^:]",type,fname,flags)==1)
    {
        /* filename is file by default */
        strcpy(fname,my->file);
        strcpy(type,"file");
    }

    /* if no filename given */
    if (strcmp(fname,"")==0)

        /* use object name-id as default file name */
        sprintf(fname,"%s-%d.%s",obj->parent->oclass->name,obj->parent->id, my->filetype);

    /* open multiple-run input file & temp output file */
    if(my->type == FT_FILE && my->multifile[0] != 0){
        if(my->interval < 1){
            gl_error("transient recorders cannot use multi-run output files");
            return 0;
        }
        sprintf(my->multitempfile, "temp_%s", my->file);
        my->multifp = fopen(my->multitempfile, "w");
        if(my->multifp == NULL){
            gl_error("unable to open \'%s\' for multi-run output", my->multitempfile);
        } else {
            time_t now=time(NULL);

            my->inputfp = fopen(my->multifile, "r");

            // write header into temp file
            fprintf(my->multifp,"# file...... %s\n", my->file);
            fprintf(my->multifp,"# date...... %s", asctime(localtime(&now)));
#ifdef WIN32
            fprintf(my->multifp,"# user...... %s\n", getenv("USERNAME"));
            fprintf(my->multifp,"# host...... %s\n", getenv("MACHINENAME"));
#else
            fprintf(my->multifp,"# user...... %s\n", getenv("USER"));
            fprintf(my->multifp,"# host...... %s\n", getenv("HOST"));
#endif
            fprintf(my->multifp,"# target.... %s %d\n", obj->parent->oclass->name, obj->parent->id);
            fprintf(my->multifp,"# trigger... %s\n", my->trigger[0]=='\0'?"(none)":my->trigger);
            fprintf(my->multifp,"# interval.. %d\n", my->interval);
            fprintf(my->multifp,"# limit..... %d\n", my->limit);
            fprintf(my->multifp,"# property.. %s\n", my->property);
            //fprintf(my->multifp,"# timestamp,%s\n", my->property);
        }
        if(my->inputfp != NULL){
            char1024 inbuffer;
            char *data;
            int get_col = 0;
            do{
                if(0 != fgets(inbuffer, 1024, my->inputfp)){
                    char *end = strchr(inbuffer, '\n');
                    data = inbuffer+strlen("# file...... ");
                    if(end != 0){
                        *end = 0; // trim the trailing newline
                    }
                    //                     "# columns... "
                    if(strncmp(inbuffer, "# file", strlen("# file")) == 0){
                        ; // ignore
                    } else if(strncmp(inbuffer, "# date", strlen("# date")) == 0){
                        ; // ignore
                    } else if(strncmp(inbuffer, "# user", strlen("# user")) == 0){
                        ; // ignore
                    } else if(strncmp(inbuffer, "# host", strlen("# host")) == 0){
                        ; // ignore
                    } else if(strncmp(inbuffer, "# target", strlen("# target")) == 0){
                        // verify same target
                        char256 target;
                        sprintf(target, "%s %d", obj->parent->oclass->name, obj->parent->id);
                        if(0 != strncmp(target, data, strlen(data))){
                            gl_error("recorder:%i: re-recording target mismatch: was %s, now %s", obj->id, data, target);
                        }
                    } else if(strncmp(inbuffer, "# trigger", strlen("# trigger")) == 0){
                        // verify same trigger, or absence thereof
                        ;
                    } else if(strncmp(inbuffer, "# interval", strlen("# interval")) == 0){
                        // verify same interval
                        int interval = atoi(data);
                        if(interval != my->interval){
                            gl_error("recorder:%i: re-recording interval mismatch: was %i, now %i", obj->id, interval, my->interval);
                        }
                    } else if(strncmp(inbuffer, "# limit", strlen("# limit")) == 0){
                        // verify same limit
                        int limit = atoi(data);
                        if(limit != my->limit){
                            gl_error("recorder:%i: re-recording limit mismatch: was %i, now %i", obj->id, limit, my->limit);
                        }
                    } else if(strncmp(inbuffer, "# property", strlen("# property")) == 0){
                        // verify same columns
                        if(0 != strncmp(my->property, data, strlen(my->property))){
                            gl_error("recorder:%i: re-recording property mismatch: was %s, now %s", obj->id, data, my->property);
                        }
                        // next line is full header column list
                        get_col = 1;
                    }
                } else {
                    gl_error("error reading multi-read input file \'%s\'", my->multifile);
                    break;
                }
            } while(inbuffer[0] == '#' && get_col == 0);
            // get full column list
            if(0 != fgets(inbuffer, 1024, my->inputfp)){
                int rep=0;
                int replen = strlen("# repetition");
                int len, lenmax = 1024, i = 0;
                char1024 propstr, shortstr;
                PROPERTY *tprop = my->target;
                gl_verbose("read last buffer");
                if(strncmp(inbuffer, "# repetition", replen) == 0){
                    char *trim;
                    rep = atoi(inbuffer + replen + 1); // skip intermediate space
                    ++rep;
                    fprintf(my->multifp, "# repetition %i\n", rep);
                    fgets(inbuffer, 1024, my->inputfp);
                    trim = strchr(inbuffer, '\n');
                    if(trim) *trim = 0;
                } else { // invalid input file or somesuch, we could error ... or we can trample onwards with our output file.
                    rep = 0;
                    fprintf(my->multifp, "# repetition %i\n", rep);
                }
                // following block matches below
                while(tprop != NULL){
                    sprintf(shortstr, ",%s(%i)", tprop->name, rep);
                    len = strlen(shortstr);
                    if(len > lenmax){
                        gl_error("multi-run recorder output full property list is larger than the buffer, please start a new file!");
                        break; // will still print everything up to this one
                    }
                    strncpy(propstr+i, shortstr, len+1);
                    i += len;
                    tprop = tprop->next;
                }
                fprintf(my->multifp, "%s%s\n", inbuffer, propstr);
            }
        } else { /* new file, so write repetition & properties with (0) */
            char1024 propstr, shortstr;
            int len, lenmax = 1024, i = 0;
            PROPERTY *tprop = my->target;
            fprintf(my->multifp, "# repetition 0\n");
            // no string from previous runs to append new props to
            sprintf(propstr, "# timestamp");
            len = strlen(propstr);
            lenmax-=len;
            i = len;
            // following block matches above
            while(tprop != NULL){
                sprintf(shortstr, ",%s(0)", tprop->name);
                len = strlen(shortstr);
                if(len > lenmax){
                    gl_error("multi-run recorder output full property list is larger than the buffer, please start a new file!");
                    break; // will still print everything up to this one
                }
                strncpy(propstr+i, shortstr, len+1);
                i += len;
                tprop = tprop->next;
            }
            fprintf(my->multifp, "%s\n", propstr);
        }
    }

    /* if type is file or file is stdin */
    my->ops = get_ftable(type)->recorder;
    if(my->ops == NULL)
        return 0;
    return my->ops->open(my, fname, flags);
}

static int write_recorder(struct recorder *my, char *ts, char *value)
{
    return my->ops->write(my, ts, value);
}

static void close_recorder(struct recorder *my)
{
    if (my->ops){
        my->ops->close(my);
    }
    if(my->multifp){
        if(0 != fclose(my->multifp)){
            gl_error("unable to close multi-run temp file \'%s\'", my->multitempfile);
            perror("fclose(): ");
        }

        my->multifp = 0; // since it's closed

        if(my->inputfp != NULL){
            fclose(my->inputfp);
            if(0 != remove(my->multifile)){ // old file
                gl_error("unable to remove out-of-data multi-run file \'%s\'", my->multifile);
                perror("remove(): ");
            }
        }
        if(0 != rename(my->multitempfile, my->multifile)){
            gl_error("unable to rename multi-run file \'%s\' to \'%s\'", my->multitempfile, my->multifile);
            perror("rename(): ");
        }
        
    }
}

static TIMESTAMP recorder_write(OBJECT *obj)
{
    struct recorder *my = OBJECTDATA(obj,struct recorder);
    char ts[64]="0"; /* 0 = INIT */
    if (my->format==0)
    {
        if (my->last.ts>TS_ZERO)
        {
            DATETIME dt;
            gl_localtime(my->last.ts,&dt);
            gl_strtime(&dt,ts,sizeof(ts));
        }
        /* else leave INIT in the buffer */
    }
    else
        sprintf(ts,"%" FMT_INT64 "d", my->last.ts);
    if ((my->limit>0 && my->samples > my->limit) /* limit reached */
        || write_recorder(my, ts, my->last.value)==0) /* write failed */
    {
        close_recorder(my);
        my->status = TS_DONE;
    }
    else
        my->samples++;

    /* at this point we've written the sample to the normal recorder output */

    // if file based
    if(my->multifp != NULL){
        char1024 inbuffer;
        char1024 outbuffer;
        char *lasts = 0;
        char *in_ts = 0;
        char *in_tok = 0;

        memset(inbuffer, 0, sizeof(inbuffer));
        
        if(my->inputfp != NULL){
            // read line
            do{
                if(0 == fgets(inbuffer, 1024, my->inputfp)){
                    if(feof(my->inputfp)){
                        // if there is no more data to append rows to, we're done with the aggregate 
                        //fclose(my->multifp); // happens in close()
                        //fclose(my->inputfp); // one-over read never happens
                        return TS_NEVER;
                    } else {
                        gl_error("error reading past recordings file");
                        my->status = TS_ERROR;
                        return TS_NEVER;
                    }
                }
                // if first char == '#', re-read
            } while(inbuffer[0] == '#');
            
            // NOTE: this is not thread safe!
            // split on first comma
            in_ts = strtok_s(inbuffer, ",\n", &lasts);
            in_tok = strtok_s(NULL, "\n", &lasts);

            if(in_ts == NULL){
                gl_error("unable to indentify a timestamp within the line read from ");
            }

            // compare timestamps if my->format == 0
                //  warn if timestamps mismatch
            if(strcmp(in_ts, ts) != 0){
                gl_warning("timestamp mismatch between current input line and simulation time");
            }
            sprintf(outbuffer, "%s,%s", in_tok, my->last.value);
        } else { // no input file ~ write normal output
            strcpy(outbuffer, my->last.value);
        }
        // fprintf 
        fprintf(my->multifp, "%s,%s\n", ts, outbuffer);
    }
    return TS_NEVER;
}

PROPERTY *link_properties(OBJECT *obj, char *property_list)
{
    char *item;
    PROPERTY *first=NULL, *last=NULL;
    UNIT *unit = NULL;
    char1024 list;
    complex oblig;
    strcpy(list,property_list); /* avoid destroying orginal list */
    for (item=strtok(list,","); item!=NULL; item=strtok(NULL,","))
    {
        char256 pstr, ustr;
        char *cpart = 0;
        int64 cid = -1;
        PROPERTY *prop = NULL;
        PROPERTY *target = NULL;
        double scale = 1.0;

        // everything that looks like a property name, then read units up to ]
        while (isspace(*item)) item++;
        if(2 == sscanf(item,"%[A-Za-z0-9_.][%[^]\n,\0]", pstr, ustr)){
            unit = gl_find_unit(ustr);
            if(unit == NULL){
                gl_error("recorder:%d: unable to find unit '%s' for property '%s'",obj->id, ustr,pstr);
                return NULL;
            }
            item = pstr;
        }
        prop = (PROPERTY*)malloc(sizeof(PROPERTY));
        
        /* branch: test to see if we're trying to split up a complex property */
        /* must occur w/ *cpart=0 before gl_get_property in order to properly reformat the property name string */
        cpart = strchr(item, '.');
        if(cpart != NULL){
            if(strcmp("imag", cpart+1) == 0){
                cid = (int)((int64)&(oblig.i) - (int64)&oblig);
                *cpart = 0;
            } else if(strcmp("real", cpart+1) == 0){
                cid = (int)((int64)&(oblig.r) - (int64)&oblig);
                *cpart = 0;
            } else {
                ;
            }
        }

        target = gl_get_property(obj,item);

        if (prop!=NULL && target!=NULL)
        {
            if(unit != NULL && target->unit == NULL){
                gl_error("recorder:%d: property '%s' is unitless, ignoring unit conversion", obj->id, item);
            }
            else if(unit != NULL && 0 == gl_convert_ex(target->unit, unit, &scale))
            {
                gl_error("recorder:%d: unable to convert property '%s' units to '%s'", obj->id, item, ustr);
                return NULL;
            }
            if (first==NULL) first=prop; else last->next=prop;
            last=prop;
            memcpy(prop,target,sizeof(PROPERTY));
            if(unit) prop->unit = unit;
            prop->next = NULL;
        }
        else
        {
            gl_error("recorder: property '%s' not found", item);
            return NULL;
        }
        if(cid >= 0){ /* doing the complex part thing */
            prop->ptype = PT_double;
            (prop->addr) = (int64)(prop->addr) + cid;
        }
    }
    return first;
}

int read_properties(OBJECT *obj, PROPERTY *prop, char *buffer, int size)
{
    PROPERTY *p;
    int offset=0;
    int count=0;
    for (p=prop; p!=NULL && offset<size-33; p=p->next)
    {
        if (offset>0) strcpy(buffer+offset++,",");
        offset+=gl_get_value(obj,GETADDR(obj,p),buffer+offset,size-offset-1,p); /* pointer => int64 */
        buffer[offset]='\0';
        count++;
    }
    return count;
}

EXPORT TIMESTAMP sync_recorder(OBJECT *obj, TIMESTAMP t0, PASSCONFIG pass)
{
    struct recorder *my = OBJECTDATA(obj,struct recorder);
    typedef enum {NONE='\0', LT='<', EQ='=', GT='>'} COMPAREOP;
    COMPAREOP comparison;
    char1024 buffer = "";
    
    if (my->status==TS_DONE)
    {
        close_recorder(my); /* note: potentially called every sync pass for multiple timesteps, catch fp==NULL in tape ops */
        return TS_NEVER;
    }

    if(obj->parent == NULL){
        char tb[32];
        sprintf(buffer, "'%s' lacks a parent object", obj->name ? obj->name : tb, sprintf(tb, "recorder:%i", obj->id));
        close_recorder(my);
        my->status = TS_ERROR;
        goto Error;
    }

    if(my->last.ts < 1 && my->interval != -1)
        my->last.ts = t0;

    /* connect to property */
    if (my->target==NULL){
        my->target = link_properties(obj->parent,my->property);
    }
    if (my->target==NULL)
    {
        sprintf(buffer,"'%s' contains a property of %s %d that is not found", my->property, obj->parent->oclass->name, obj->parent->id);
        close_recorder(my);
        my->status = TS_ERROR;
        goto Error;
    }

    // update clock
    if ((my->status==TS_OPEN) && (t0 > obj->clock)) 
    {   
        obj->clock = t0;
        // if the recorder is clock-based, write the value
        if((my->interval > 0) && (my->last.ts < t0) && (my->last.value[0] != 0)){
            recorder_write(obj);
            my->last.value[0] = 0; // once it's been finalized, dump it
        }
    }

    /* update property value */
    if ((my->target != NULL) && (my->interval == 0 || my->interval == -1)){ 
        if(read_properties(obj->parent,my->target,buffer,sizeof(buffer))==0)
        {
            sprintf(buffer,"unable to read property '%s' of %s %d", my->property, obj->parent->oclass->name, obj->parent->id);
            close_recorder(my);
            my->status = TS_ERROR;
        }
    }
    if ((my->target != NULL) && (my->interval > 0)){
        if((t0 >=my->last.ts + my->interval) || (t0 == my->last.ts)){
            if(read_properties(obj->parent,my->target,buffer,sizeof(buffer))==0)
            {
                sprintf(buffer,"unable to read property '%s' of %s %d", my->property, obj->parent->oclass->name, obj->parent->id);
                close_recorder(my);
                my->status = TS_ERROR;
            }
            my->last.ts = t0;
        }
    }

    /* check trigger, if any */
    comparison = (COMPAREOP)my->trigger[0];
    if (comparison!=NONE)
    {
        int desired = comparison==LT ? -1 : (comparison==EQ ? 0 : (comparison==GT ? 1 : -2));

        /* if not trigger or can't get access */
        int actual = strcmp(buffer,my->trigger+1);
        if (actual!=desired || (my->status==TS_INIT && !recorder_open(obj)))
        {
            /* better luck next time */
            return (my->interval==0 || my->interval==-1) ? TS_NEVER : t0+my->interval;
        }
    }
    else if (my->status==TS_INIT && !recorder_open(obj))
    {
        close_recorder(my);
        return TS_NEVER;
    }

    if(my->last.ts < 1 && my->interval != -1)
        my->last.ts = t0;

    /* write tape */
    if (my->status==TS_OPEN)
    {   
        if (my->interval==0 /* sample on every pass */
            || ((my->interval==-1) && my->last.ts!=t0 && strcmp(buffer,my->last.value)!=0) /* sample only when value changes */
            )

        {
            strncpy(my->last.value,buffer,sizeof(my->last.value));
            my->last.ts = t0;
            recorder_write(obj);
        } else if (my->interval > 0 && my->last.ts == t0){
            strncpy(my->last.value,buffer,sizeof(my->last.value));
        }
    }
Error:
    if (my->status==TS_ERROR)
    {
        gl_error("recorder %d %s\n",obj->id, buffer);
        close_recorder(my);
        my->status=TS_DONE;
        return TS_NEVER;
    }

    if (my->interval==0 || my->interval==-1) 
    {
        return TS_NEVER;
    }
    else
        return my->last.ts+my->interval;
}

---

GridLAB-D^TM Version 2.0

An open-source project initiated by the US Department of Energy