gmFileIOUtils.h

Go to the documentation of this file.

/************************************************************************
**
** Copyright (C) 2014 by Carlos Augusto Teixera Mendes
** All rights reserved.
**
** This file is part of the "GeMA" software. It's use should respect
** the terms in the license agreement that can be found together 
** with this source code.
** It is provided AS IS, with NO WARRANTY OF ANY KIND, 
** INCLUDING THE WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR 
** A PARTICULAR PURPOSE.
**
************************************************************************/
 
#ifndef _GEMA_FILE_IO_UTILS_H_
#define _GEMA_FILE_IO_UTILS_H_
 
#include "gmCoreConfig.h"
#include <functional>
#include "gmFileFilter.h"
 
class GmMesh;
class GmCellMesh;
class GmElementMesh;
class GmValueAccessor;
class GmCellAccessor;
class GmGaussAccessor;
class GmTaskManager;
 
 
 
namespace GmFileIOUtils
{
    using TransformF = std::function<void(const double*, int, double*)>;
/*
* A simple buffer used for batch write data which is not
* contiguous in memory (copy then write)
*/
 
class GmWBuffer {
    const int _maxSize = 8*1024*1024; //8MB
public:
    GmWBuffer(int count, int stride) {
        assert(count > 0 && stride > 0);
        int askedSize = count*stride;
        _size = std::min(askedSize, _maxSize);
        _stride = stride;
        _data = (char*)malloc(_size);
        if(!_data)
            throw std::bad_alloc();
        
        _capacity = _size / stride;
        //Not enough space to hold a single object
        if(!_capacity) {
            if(_data) free(_data);
            throw std::bad_alloc();
        }
    }
 
    void setData(int i, const void* ptr) {
        assert(i < _capacity && ptr);
        memcpy(_data + i*_stride, ptr, _stride);
    }
 
    void * getData(int i) {
        return (void*)(_data + i*_stride);
    }
 
    void* memPtr() const {
        return (void*)_data;
    }
 
    int getCapacity() const { return _capacity;}
    
    ~GmWBuffer() {
        free(_data);
    }
 
    //Make sure it is not being wrongly used
        GmWBuffer(const GmWBuffer&) = delete;
    GmWBuffer& operator=(const GmWBuffer&) = delete;
 
private:
    char* _data; //Pointer to data
    int _capacity; //max number of entries
    int _size; //data size in bytes
    int _stride; //entry size in bytes
};
 
 
 
//Buffer filling utilities
 
GMC_API_EXPORT  void fillNodeAcBuffer(GmTaskManager* tm, const GmMesh* mesh, GmValueAccessor* dataAc, int dimFilter, const TransformF& transformFunc, const QVector<int>& indexer, int nodeStart, int nodeCount, GmWBuffer& bbuf);
GMC_API_EXPORT  void fillCellAcBuffer(GmTaskManager* tm, const GmCellMesh* mesh, const GmCellAccessor* dataAc, const GmValueAccessor* coordAc, int dimFilter, const QVector<int>& indexer, int cellStart, int cellCount, GmWBuffer& bbuf);
GMC_API_EXPORT   int fillGaussAcBuffer(GmTaskManager* tm, const GmElementMesh* mesh, const GmGaussAccessor* dataAc, int rule, int dimFilter, const QVector<GmFileFilter::Indexer>& preIndex, int cellStart, int cellCount, int valueStart, GmWBuffer& bbuf);
 
 
}
#endif