R/GetFullWeeksPerMonth.R

In Mthrun/TSAT: Time Series Analysis Tools (TSAT)

# GetFullWeeksPerMonth = GetFullWeeksPerMonth(Time, DaysOrWeeks=TRUE, IgnoreWeekend=TRUE)
#
# Description:
# 
#
# INPUT
# Time              Vector of Time values.
#
# OPTIONAL
# DaysOrWeeks       Boolean, TRUE if Days, FALSE if Weeks. Default is TRUE.
# IgnoreWeekend     Boolean, if FALSE removes weekends. Default is TRUE.
#
# OUTPUT
# month             
#
# Author: 

GetFullWeeksPerMonth = function(Time, DaysOrWeeks=TRUE, IgnoreWeekend=TRUE,tz="UTC"){
  
  requireNamespace('lubridate')
  
  if(is.character(Time)) Time=as.Date(Time,tz=tz)

  if(isTRUE(DaysOrWeeks)){#Days
    temp=rep(1,length(Time))
    DF=data.frame(Time=Time,Data=temp)
    WeekTime=aggregateDays2Weeks(DF$Time,DF$Data,sum)$Time
  }else{
    WeekTime=Time
  }
  FullyInMonth=rep(FALSE,length(WeekTime))
  for(day in 1:length(WeekTime)){
    TimeCur=seq(from=WeekTime[day],by='1 days',length.out=7)

    if(IgnoreWeekend){
      monthNoCur=lubridate::month(TimeCur)
      if(length(unique(monthNoCur))==1) FullyInMonth[day]=TRUE
    }else{
      tmp=lubridate::wday(TimeCur)
      ind=which(tmp!=7&tmp!=1)
      monthNoCur=lubridate::month(TimeCur)[ind]
      if(length(unique(monthNoCur))==1) FullyInMonth[day]=TRUE
    }
  }
  month=lubridate::month(WeekTime)
  month[!FullyInMonth]=0
  return(month)
}