writer.h

Go to the documentation of this file.

//******************************************************************************
//*       ██╗  ██╗██╗██████╗  ██████╗     ██╗  ██╗    ██████╗                  *
//*       ██║  ██║██║██╔══██╗██╔═══██╗    ██║  ██║   ██╔═████╗                 *
//*       ███████║██║██████╔╝██║   ██║    ███████║   ██║██╔██║                 *
//*       ██╔══██║██║██╔═══╝ ██║   ██║    ╚════██║   ████╔╝██║                 *
//*       ██║  ██║██║██║     ╚██████╔╝         ██║██╗╚██████╔╝                 *
//*       ╚═╝  ╚═╝╚═╝╚═╝      ╚═════╝          ╚═╝╚═╝ ╚═════╝                  *
//************************ Jefferson National Lab (2017) ***********************
/*
 *   Copyright (c) 2017.  Jefferson Lab (JLab). All rights reserved. Permission
 *   to use, copy, modify, and distribute  this software and its documentation
 *   for educational, research, and not-for-profit purposes, without fee and
 *   without a signed licensing agreement.
 *
 *   IN NO EVENT SHALL JLAB BE LIABLE TO ANY PARTY FOR DIRECT, INDIRECT, SPECIAL
 *   INCIDENTAL, OR CONSEQUENTIAL DAMAGES, INCLUDING LOST PROFITS, ARISING
 *   OUT OF THE USE OF THIS SOFTWARE AND ITS DOCUMENTATION, EVEN IF JLAB HAS
 *   BEEN ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 *
 *   JLAB SPECIFICALLY DISCLAIMS ANY WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
 *   THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
 *   PURPOSE. THE HIPO DATA FORMAT SOFTWARE AND ACCOMPANYING DOCUMENTATION, IF
 *   ANY, PROVIDED HEREUNDER IS PROVIDED "AS IS". JLAB HAS NO OBLIGATION TO
 *   PROVIDE MAINTENANCE, SUPPORT, UPDATES, ENHANCEMENTS, OR MODIFICATIONS.
 *
 *   This software was developed under the United States Government license.
 *   For more information contact author at gavalian@jlab.org
 *   Department of Experimental Nuclear Physics, Jefferson Lab.
 */
/*
 *This sowftware was developed at Jefferson National Laboratory.
 */
/*
 * File:   writer.h
 * Author: gavalian
 *
 * Created on April 11, 2017, 2:07 PM
 */
 
 
#ifndef HIPOWRITER_H
#define HIPOWRITER_H
 
 
#include <iostream>
#include <vector>
#include <fstream>
#include <stdio.h>
#include <stdlib.h>
#include <memory>
#include <climits>
#include "recordbuilder.h"
#include "reader.h"
 
namespace hipo {
 
  typedef struct {
    int  uniqueid;            
    int  filenumber;          
    int  headerLength;        
    int  recordCount;         
    int  indexArrayLength;    
    int  bitInfoVersion;      
    int  userHeaderLength;    
    int  magicNumber;         
    long userRegister;        
    long trailerPosition;     
    int  userIntegerOne;      
    int  userIntegerTwo;      
  } hipoFileHeader_t;
 
class writer {
  
  private:
    std::ofstream         outputStream;
    hipo::recordbuilder   recordBuilder;
    hipo::dictionary      writerDictionary;
    std::vector<hipo::recordInfo_t>   writerRecordInfo;
    std::map<std::string,std::string> userConfig;
    
    std::map<int,hipo::recordbuilder> extendedBuilder;
 
    void writeIndexTable();
    int  verbose = 0;
  
  public:
 
     writer(){};
     virtual ~writer(){};
 
     void addEvent(hipo::event &hevent);
     void addEvent(std::vector<char> &vec, int size = -1);
     void addUserConfig(std::string key, std::string value){ userConfig[key] = value;}
     void addUserConfig(const char *key, const char *value){ userConfig[std::string(key)] = std::string(value);}
     void writeRecord(recordbuilder &builder);
     void open(const char *filename);
     void close();
     void showSummary();
     void addDictionary(hipo::dictionary &dict);
     hipo::dictionary &getDictionary(){ return writerDictionary;}
     void setUserIntegerOne(long userIntOne);
     void setUserIntegerTwo(long userIntTwo);
     void flush();
     void setVerbose(int level){ verbose = level;}
};
 
};
#endif /* HIPOFILE_H */