R/size-methods.R

In intervals: Tools for Working with Points and Intervals

# For R, size is Lebesgue measure, so closure is irrelevant. Note that we don't
# use the close_intervals method here, which is important since this method
# requires empty, and empty currently uses size. We need to avoid circular
# dependency. 

setGeneric( "size", def = function( x, ... ) standardGeneric( "size" ) )

setMethod(
          "size",
          signature( "Intervals" ),
          function( x, as = type(x) ) {
            result <- x[,2] - x[,1]            
            if ( as == "Z" ) {
              ties <- x[,2] == x[,1]
              ties[ is.na( ties ) ] <- FALSE
              result[ ties ] <- ifelse( all( closed(x) ), 1, 0 )
              result[ !ties ] <- result[ !ties ] + sum( closed(x) ) - 1 # NAs just stay NA
            }
            return( result )
          }
          )

setMethod(
          "size",
          signature( "Intervals_full" ),
          function( x, as = type(x) ) {
            result <- x[,2] - x[,1]            
            if ( as == "Z" ) {
              ties <- x[,2] == x[,1]
              ties[ is.na( ties ) ] <- FALSE
              rs <- rowSums( closed(x) )
              result[ ties ] <- ifelse( rs[ ties ] == 2, 1, 0 )
              result[ !ties ] <- result[ !ties ] + rs[ !ties ] - 1
            }
            return( result )
          }
          )