statistics.hpp

Go to the documentation of this file.

/*
 * PoPS model - Various statistics computation
 *
 * Copyright (C) 2015-2020 by the authors.
 *
 * Authors: Anna Petrasova <akratoc gmail com>
 *
 * The code contained herein is licensed under the GNU General Public
 * License. You may obtain a copy of the GNU General Public License
 * Version 2 or later at the following locations:
 *
 * http://www.opensource.org/licenses/gpl-license.html
 * http://www.gnu.org/copyleft/gpl.html
 */
 
#ifndef POPS_STATISTICS_HPP
#define POPS_STATISTICS_HPP
 
namespace pops {
 
template<typename IntegerRaster>
unsigned sum_of_infected(
    const IntegerRaster& infected, const std::vector<std::vector<int>>& suitable_cells)
{
    unsigned sum = 0;
    for (auto indices : suitable_cells) {
        int i = indices[0];
        int j = indices[1];
        sum += infected(i, j);
    }
    return sum;
}
template<typename IntegerRaster>
double area_of_infected(
    const IntegerRaster& infected,
    double ew_res,
    double ns_res,
    const std::vector<std::vector<int>>& suitable_cells)
{
    unsigned cells = 0;
    for (auto indices : suitable_cells) {
        int i = indices[0];
        int j = indices[1];
        if (infected(i, j) > 0)
            cells++;
    }
    return cells * ew_res * ns_res;
}
 
}  // namespace pops
#endif  // POPS_STATISTICS_HPP