通用輸入演算法

wiki

Lua 中的通用函數和演算法

有兩個特殊的輸入迭代器，numbers() 和 words()，它們的工作方式類似於非常有用的 io.lines() 迭代器。要列印標準輸入中找到的所有字詞

-- words.lua
require 'func'
for w in words() do
  print(w)
end

$ lua words.lua < test.txt

printall(words(),'\n')

require 'func'
local s 
for x in numbers() do
  s = s + x
end
print(s)

local s,n = sum(numbers())
print('average value =',s/n)

f = io.open(arg[1])
print(sum(numbers(f)))
f:close()

function copy(iter)
  local res = {}
  local k = 1
  for v in iter do
     res[k] = v
     k = k + 1
  end
  return res
end

t = copy(numbers '10 20 30') -- will be {10,20,30}
s = sum(list(t))             -- will be 60

t = copy(io.lines())
table.sort(t)
printall(t,'\n')   -- try table.foreach(t,print) and see!

printall(sort(io.lines()),'\n')

printall(slice(io.lines(),1,10),'\n')

require 'func'
print(count(words()))

require 'func'
print(count(io.lines(),matching '^%s*function'))

function matching(s)
  local strfind = string.find
  return function(v)
    return strfind(v,s)
  end
end

當然您可以在這些操作中使用任何序列。如果您載入了非常有用的 lfs（Lua 文件系統）函式庫，則 t = copy_if(lfs.dir(path),matching '%.cpp$') 將使用擴展名為 .cpp 的所有文件填滿一個列表。

修改 count() 輸入的另一種有用的方法是使用 unique() 函數

-- number of unique words in a file
print(count(unique(words())))

function unique(iter)
  local t = count_map(iter)
  return keys(t)
end

table.foreach(count_map(words()),print)

for x,y in join(numbers(f1),numbers(f2)) do
  print(x-y)
end

Lua 的 AWK 編程樣式

{ print $1/$4, $3/$4 }

for x,y,z in fields{1,3,4} do
   print(x/z,y/z)
end

print(count(fields{7},greater_than(44000)))

{ if ($7 > 44000) k++ } END { print(k) }

for x,y in fields({1,2},',',f) do ...

for x,y in fields(2,',',f) do ...  --equivalent--

效能及運算式

> tt = copy(random(10000))
> = sum(tt)
5039.542771691  10000

local t = {}
local random = math.random
for i = 1,1e6 do
   t[i] = random()
end

第二個反對意見是，這會導致奇怪且不自然的程式碼。對於 C++ 來說，這確實有可能發生，這是因為（讓我們面對現實吧）C++ 實際上並未適合函式風格；沒有閉包，而且嚴格的靜態輸入會不斷造成阻礙，導致所有內容都必需為範本。此風格更適合 Lua - 使用 Boost Lambda 函式庫時，用 C++ 執行此程式並不會好讀一半

-- sum of squares of input data using an internal iterator
for_each(numbers(),function(v)
    s = s + v*v
end)

-- sum of squares of input data using an external iterator
for v in numbers() do
    s = s + v*v
end

撰寫自訂輸入物件

Files = {}
 
function Files.create(list)
   local files = {}
   files.list = {}
   local n = table.getn(list)
   for i = 1,n do
      files.list[i] = list[i]
   end
   files.open_next = Files.open_next
   files.read = Files.read
   files:open_next()
   return files
end
 
function Files:open_next()
   if self.f then self.f:close() end
   local nf = table.remove(self.list,1)
   if nf then
      self.f = io.open(nf)
      return true
   else
      self.f = nil
      return false
   end
end
 
function Files:read()
  local ret = self.f:read()
  if not ret then
     if not self:open_next() then return nil end
     return self.f:read()
  else
     return ret
  end
end

files = Files.create(arg)
printall(words(files))

關於實作和進一步開發的注意事項

'%s*([^,]+),[^,]+,([^,])'

cur = con:execute 'SELECT * FROM [Event Summaries]'
mag = -9
row = cur:fetch({},'a')
while row do
  if row.Magnitude > mag then 
     mag = row.Magnitude
  end
  row = cur:fetch(row,'n')
end
cur:close()

我只要建立一個能持續追蹤 row 的迭代器，就能讓這個過程變得簡單

function rows(cursor)
  local row = {}
  return function()
    return cursor:fetch(row,'a')
  end
end

for row in rows(cur) do
   if row.Magnitude > mag then 
      mag = row.Magnitude    
   end
end

function column(fieldname,cursor)
  local row = {}
  return function()
    row = cur:fetch(row,'a')
    if not row then return nil 
    else return row[fieldname]
    end
  end
end

local minm,maxm = minmax(column('Magnitude',cur))

print(count(column('Magnitude',cur),greater_than(2)))

-- SteveDonovan