mucap_expand.cpp

Go to the documentation of this file.

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <endian.h>
#include <byteswap.h>
#include "midas.h"

#include "mucap_structures.h"
#include "mucap_compress.h"

HNDLE hDB;

int main(int argc, char *argv[])
{
    char host_name[256], exp_name[32];

    cm_get_environment(host_name, sizeof(host_name), exp_name,
                       sizeof(exp_name));
    cm_connect_experiment(host_name, exp_name, "Compression", NULL);
    cm_get_experiment_database(&hDB, NULL);

    /* Not really necessary */
    // Determine which compression to use
    char key_name[80];
    int runnumber=-1;
    int size=sizeof(int);
    sprintf(key_name, "Runinfo/Run number");
    db_get_value(hDB,0,key_name, &runnumber, &size, TID_INT,1);
    bool USE_OLD_COMPRESSION=false;
    if (runnumber > 30000 && runnumber < 50000)
      {
        USE_OLD_COMPRESSION=true;
      }
    printf("Determined that we should use old compression %d\n", USE_OLD_COMPRESSION);
    // Done determining which decompression to use
    
    compress_load_all(USE_OLD_COMPRESSION);
    /* Done with unnecessary code */
    compress_load_all();
    if(argc != 3) {
      printf("Usage: mucap_expand uncompressed_input.mid uncompressed_output.mid\n");
      exit(1);
    }

    FILE *fp = fopen(argv[1], "r");
    if (!fp) {
      printf("Unable to open %s\n", argv[1]);
      exit(1);
    }

    FILE *output_fp = fopen(argv[2], "w");
    if (!fp) {
      printf("Unable to open %s\n", argv[2]);
      exit(1);
    }

    char *input_event = new char[MAX_EVENT_SIZE];
    char *output_event = new char[MAX_EVENT_SIZE];

    while (read_midas_event(fp, input_event) == SUCCESS) {
        EVENT_HEADER header = *((EVENT_HEADER *) input_event);

        int event_id = header.event_id;

        if (event_id == EVENTID_BOR || event_id == EVENTID_EOR ||
            event_id == EVENTID_MESSAGE) {

           // For non-ordinary events, just write out the raw data
           fwrite(input_event, 1, sizeof(EVENT_HEADER) + header.data_size,
                  output_fp);
        } else {
    
          bk_init32(output_event);
          expand_event(input_event + sizeof(EVENT_HEADER), output_event,USE_OLD_COMPRESSION);
          header.data_size = bk_size(output_event);

          fwrite(&header, sizeof(EVENT_HEADER), 1, output_fp);
          fwrite(output_event, 1, header.data_size, output_fp);
       }
    }

    fclose(fp);
    fclose(output_fp);

    cm_disconnect_experiment();
}