src/timetable: Add new_from_timestamp function

[etc/awesome.git] / src / timetable.lua

diff --git a/src/timetable.lua b/src/timetable.lua
index b58ad0f531cc8386904e5aca72a8d21445126e44..887ee94d689344d48e869d9036e2e40cd7078c0a 100644 (file)
--- a/src/timetable.lua
+++ b/src/timetable.lua
@@ -102,12 +102,16 @@ local function normalise ( year , month , day , hour , min , sec )
        return year , month , day , hour , min , sec
 end
 
+local leap_years_since_1970 = leap_years_since ( 1970 )
 local function timestamp ( year , month , day , hour , min , sec )
-       return 60*60*24*(
-                       year * year_length ( year )
-                       + month * month_length ( month , year )
-                       + day
-               )
+       year , month , day , hour , min , sec = normalise ( year , month , day , hour , min , sec )
+
+       local days_since_epoch = day_of_year ( day , month , year )
+               + 365 * ( year - 1970 )
+               -- Each leap year adds one day
+               + ( leap_years_since ( year - 1 ) - leap_years_since_1970 ) - 1
+
+       return days_since_epoch * 60*60*24
                + hour  * (60*60)
                + min   * 60
                + sec
@@ -147,6 +151,9 @@ local timetable_mt = {
        __index    = timetable_methods ;
        __tostring = timetable_methods.rfc_3339 ;
        __eq = function ( a , b )
+               return a:timestamp() == b:timestamp()
+       end ;
+       __lt = function ( a , b )
                return a:timestamp() < b:timestamp()
        end ;
 }
@@ -166,12 +173,17 @@ local function new_timetable ( year , month , day , hour , min , sec )
        }
 end
 
+local function new_from_timestamp ( ts )
+       return new_timetable ( 1970 , 1 , 1 , 0 , 0 , ts )
+end
+
 return {
        doomsday  = doomsday ;
        normalise = normalise ;
        timestamp = timestamp ;
 
        new = new_timetable ;
+       new_from_timestamp = new_from_timestamp ;
        cast = cast_timetable ;
        timetable_mt = timetable_mt ;
 }