]> git.madduck.net Git - etc/awesome.git/commitdiff

madduck's git repository

Every one of the projects in this repository is available at the canonical URL git://git.madduck.net/madduck/pub/<projectpath> — see each project's metadata for the exact URL.

All patches and comments are welcome. Please squash your changes to logical commits before using git-format-patch and git-send-email to patches@git.madduck.net. If you'd read over the Git project's submission guidelines and adhered to them, I'd be especially grateful.

SSH access, as well as push access can be individually arranged.

If you use my repositories frequently, consider adding the following snippet to ~/.gitconfig and using the third clone URL listed for each project:

[url "git://git.madduck.net/madduck/"]
  insteadOf = madduck:

luatz/timetable: Add support to `normalise` for fractional columns
authordaurnimator <quae@daurnimator.com>
Fri, 29 Aug 2014 20:03:47 +0000 (16:03 -0400)
committerdaurnimator <quae@daurnimator.com>
Fri, 29 Aug 2014 20:03:47 +0000 (16:03 -0400)
luatz/timetable.lua
spec/timetable_spec.lua

index 879f0a226fe6839b8c5424112d0945a91899063c..3e802428310822aae1162df923aede765174f094 100644 (file)
@@ -56,6 +56,13 @@ local function day_of_week ( day , month , year )
        return ( year + leap_years_since ( year ) + sakamoto[month] + day ) % 7 + 1
 end
 
        return ( year + leap_years_since ( year ) + sakamoto[month] + day ) % 7 + 1
 end
 
+local function borrow ( tens , units , base )
+       local frac = tens % 1
+       units = units + frac * base
+       tens = tens - frac
+       return tens , units
+end
+
 local function carry ( tens , units , base )
        if units >= base then
                tens  = tens + idiv ( units , base )
 local function carry ( tens , units , base )
        if units >= base then
                tens  = tens + idiv ( units , base )
@@ -69,34 +76,50 @@ end
 
 -- Modify parameters so they all fit within the "normal" range
 local function normalise ( year , month , day , hour , min , sec )
 
 -- Modify parameters so they all fit within the "normal" range
 local function normalise ( year , month , day , hour , min , sec )
+       -- `month` and `day` start from 1, need -1 and +1 so it works modulo
+       month , day = month - 1 , day - 1
+
+       -- Convert everything (except seconds) to an integer
+       -- by propagating fractional components down.
+       year  , month = borrow ( year  , month , 12 )
+       -- Carry from month to year first, so we get month length correct in next line around leap years
+       year  , month = carry ( year , month , 12 )
+       month , day   = borrow ( month , day   , month_length ( floor ( month + 1 ) , year ) )
+       day   , hour  = borrow ( day   , hour  , 24 )
+       hour  , min   = borrow ( hour  , min   , 60 )
+       min   , sec   = borrow ( min   , sec   , 60 )
+
        -- Propagate out of range values up
        -- e.g. if `min` is 70, `hour` increments by 1 and `min` becomes 10
        -- Propagate out of range values up
        -- e.g. if `min` is 70, `hour` increments by 1 and `min` becomes 10
+       -- This has to happen for all columns after borrowing, as lower radixes may be pushed out of range
        min   , sec   = carry ( min   , sec   , 60 ) -- TODO: consider leap seconds?
        hour  , min   = carry ( hour  , min   , 60 )
        day   , hour  = carry ( day   , hour  , 24 )
        min   , sec   = carry ( min   , sec   , 60 ) -- TODO: consider leap seconds?
        hour  , min   = carry ( hour  , min   , 60 )
        day   , hour  = carry ( day   , hour  , 24 )
-
-       while day <= 0 do
+       -- Ensure `day` is not underflowed
+       -- Add a whole year of days at a time, this is later resolved by adding months
+       -- TODO[OPTIMIZE]: This could be slow if `day` is far out of range
+       while day < 0 do
                year = year - 1
                day  = day + year_length ( year )
        end
                year = year - 1
                day  = day + year_length ( year )
        end
-
-       -- Lua months start from 1, need -1 and +1 around this increment
-       month = month - 1
        year , month = carry ( year , month , 12 )
        year , month = carry ( year , month , 12 )
-       month = month + 1
 
 
-       -- This could potentially be slow if `day` is very large
+       -- TODO[OPTIMIZE]: This could potentially be slow if `day` is very large
        while true do
        while true do
-               local i = month_length ( month , year )
-               if day <= i then break end
+               local i = month_length ( month + 1 , year )
+               if day < i then break end
                day = day - i
                month = month + 1
                day = day - i
                month = month + 1
-               if month > 12 then
-                       month = 1
+               if month >= 12 then
+                       month = 0
                        year = year + 1
                end
        end
 
                        year = year + 1
                end
        end
 
+       -- Now we can place `day` and `month` back in their normal ranges
+       -- e.g. month as 1-12 instead of 0-11
+       month , day = month + 1 , day + 1
+
        return year , month , day , hour , min , sec
 end
 
        return year , month , day , hour , min , sec
 end
 
index 34c05295992d0e2dba03768a5e5acce52de62f84..569864fc8beba43faeee83a8a036fcce84c9bc86 100644 (file)
@@ -84,4 +84,24 @@ describe ( "Timetable library" , function ( )
                assert.same ( { timetable.normalise(2013,42,52,0,0,0) } , { 2016,7,22,0,0,0 } )
                assert.same ( { timetable.normalise(2013,42,52,50,0,0) } , { 2016,7,24,2,0,0 } )
        end )
                assert.same ( { timetable.normalise(2013,42,52,0,0,0) } , { 2016,7,22,0,0,0 } )
                assert.same ( { timetable.normalise(2013,42,52,50,0,0) } , { 2016,7,24,2,0,0 } )
        end )
+
+       it ( "#normalise handles fractional #month" , function ( )
+               assert.same ( { timetable.normalise(2014,14.5,1,0,0,0) } , { 2015,2,15,0,0,0 } )
+               assert.same ( { timetable.normalise(2015,14.5,1,0,0,0) } , { 2016,2,15,12,0,0 } ) -- leap year, so hours is 12
+               assert.same ( { timetable.normalise(2016,14.5,1,0,0,0) } , { 2017,2,15,0,0,0 } )
+       end )
+
+       local function round_trip_add(t, field, x)
+               local before = t:clone()
+               t[field]=t[field]+x;
+               t:normalise();
+               t[field]=t[field]-x;
+               t:normalise();
+               assert.same(0, t-before)
+       end
+       it ( "#normalise round trips" , function ( )
+               round_trip_add(timetable.new(2000,2,28,0,0,0), "month", 0.5)
+               round_trip_add(timetable.new(2014,8,28,19,23,0), "month", 0.4)
+               round_trip_add(timetable.new(2014,14.5,28,0,0,0), "month", 0.4)
+       end )
 end )
 end )