Rcpp Version 1.0.9
cumsum.h
Go to the documentation of this file.
1 // -*- mode: C++; c-indent-level: 4; c-basic-offset: 4; tab-width: 8 -*-
2 //
3 // cumsum.h: Rcpp R/C++ interface class library -- cumsum
4 //
5 // Copyright (C) 2010 - 2011 Dirk Eddelbuettel and Romain Francois
6 //
7 // This file is part of Rcpp.
8 //
9 // Rcpp is free software: you can redistribute it and/or modify it
10 // under the terms of the GNU General Public License as published by
11 // the Free Software Foundation, either version 2 of the License, or
12 // (at your option) any later version.
13 //
14 // Rcpp is distributed in the hope that it will be useful, but
15 // WITHOUT ANY WARRANTY; without even the implied warranty of
16 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17 // GNU General Public License for more details.
18 //
19 // You should have received a copy of the GNU General Public License
20 // along with Rcpp. If not, see <http://www.gnu.org/licenses/>.
21 
22 #ifndef Rcpp__sugar__cumsum_h
23 #define Rcpp__sugar__cumsum_h
24 
25 namespace Rcpp{
26 namespace sugar{
27 
28 template <int RTYPE, bool NA, typename T>
29 class Cumsum : public Lazy< Rcpp::Vector<RTYPE> , Cumsum<RTYPE,NA,T> > {
30 public:
34 
35  Cumsum( const VEC_TYPE& object_ ) : object(object_){}
36 
37  VECTOR get() const {
38  R_xlen_t n = object.size() ;
39  VECTOR result( n, Rcpp::traits::get_na<RTYPE>() ) ;
40  STORAGE current = object[0] ;
41  if( Rcpp::traits::is_na<RTYPE>(current) )
42  return result ;
43  result[0] = current ;
44  for( R_xlen_t i=1; i<n; i++){
45  current = object[i] ;
46  if( Rcpp::traits::is_na<RTYPE>(current) )
47  return result ;
48  result[i] = result[i-1] + current ;
49  }
50  return result ;
51  }
52 private:
53  const VEC_TYPE& object ;
54 } ;
55 
56 } // sugar
57 
58 template <bool NA, typename T>
60  return sugar::Cumsum<INTSXP,NA,T>( t ) ;
61 }
62 
63 template <bool NA, typename T>
65  return sugar::Cumsum<REALSXP,NA,T>( t ) ;
66 }
67 
68 
69 } // Rcpp
70 #endif
71 
R_xlen_t size() const
Definition: Vector.h:276
Cumsum(const VEC_TYPE &object_)
Definition: cumsum.h:35
VECTOR get() const
Definition: cumsum.h:37
Rcpp::traits::storage_type< RTYPE >::type STORAGE
Definition: cumsum.h:32
const VEC_TYPE & object
Definition: cumsum.h:53
Rcpp::VectorBase< RTYPE, NA, T > VEC_TYPE
Definition: cumsum.h:31
Rcpp::Vector< RTYPE > VECTOR
Definition: cumsum.h:33
Rcpp API.
Definition: algo.h:28
sugar::Cumsum< INTSXP, NA, T > cumsum(const VectorBase< INTSXP, NA, T > &t)
Definition: cumsum.h:59