runlen.hpp

Go to the documentation of this file.

#ifndef OPM_VERTEQ_RUNLEN_HPP_INCLUDED
#define OPM_VERTEQ_RUNLEN_HPP_INCLUDED

// Copyright (C) 2013 Uni Research AS
// This file is licensed under the GNU General Public License v3.0

// forward declaration
struct UnstructuredGrid;

namespace Opm {

template <typename T>
class RunLenView {
protected:
    int num_of_cols;
    int* pos;

    T* data;

public:
    RunLenView (int num_cols, int* pos_ptr, T* values)
        // store them locally for later use
        : num_of_cols (num_cols)
        , pos (pos_ptr)
        , data (values) {
    }

    RunLenView (const RunLenView& rhs)
        // copy all fields verbatim
        : num_of_cols (rhs.num_of_cols)
        , pos (rhs.pos)
        , data (rhs.data) {
    }

    T* operator [] (int col) const {
        return &data [pos [col]];
    }

    int cols () const {
        return num_of_cols;
    }

    int size (int col) const {
        return pos [col + 1] - pos [col];
    }

    T& last (int col) const {
        return data [pos [col + 1] - 1];
    }
};

template <typename T>
struct RunLenData : public RunLenView <T> {
    RunLenData (int number, int* pos_ptr)
        // allocate a new vector for the data, containing the needed
        // number of elements. note that there is only one new
        // operation is the parameter list, so there is no leakage if
        // an out-of-memory exception is thrown.
        : RunLenView <T> (number, pos_ptr, new T [pos_ptr [number]]) {
    }

    ~RunLenData () {
        // this member is initialized with data allocated in our ctor
        delete [] RunLenView <T>::data;
    }
};

// shorthands for most used types
typedef const RunLenView <int> rlw_int;
typedef const RunLenView <double> rlw_double;

// access common run-length encoded matrices in a grid structure
rlw_int grid_cell_facetag (const UnstructuredGrid& g);
rlw_int grid_cell_faces (const UnstructuredGrid& g);

} /* namespace Opm */

#endif /* OPM_VERTEQ_RUNLEN_HPP_INCLUDED */