MiscPeriphs/CC Fireworks Show 1.4

-- Fireworks Show 1.4.0
-- by Dog
--
-- Requires:  Minecraft 1.6.4 (possibly 1.5.2), Misc. Peripherals Fireworks Launcher, ComputerCraft Computer, this script
-- Optional:  Monitor
--
-- Place your computer next to (touching) the launcher (I usually put the launcher on top of the computer) or connect via wired modems & network cable - connecting side will be autodetected
--
-- Place your monitor next to (touching) the computer or connect via wired modems and network cable - connecting side will be autodetected
--
-- Into your Fireworks Launcher, place the following:
-- gun powder (I use slots 1-9 - edit the values below if you do it differently)
-- colors (non-ceramic/non-molten dyes (also not indigo), squid ink sacks, lapis, rose red, dandelion yellow, cactus green)
-- effects (bone-meal, feather, glowstone dust, gold nugget, fire charge, diamond, skeleton skull, wither skeleton skull - assign an empty space or two for random, 'standard' fireworks without special effects)
--
-- Don't forget to edit the slot assignments below so this script tells the launcher to combine the right stuff
   -- Slots are counted by row, left to right, starting at the top left
--
-- Run this script and answer the two questions it asks (how many fireworks to launch and how long to wait before starting the show)
--
-- Fireworks Show does the following:
-- Mixes up to (3) gun powders, (3) colors, (3) effects, and adds a 30% chance for a fade effect/bonus color (at 20% it's a bit too rare for my liking)
-- Launches the firework
-- Repeats the process for the number of fireworks you selected
--
-- Edit these . . .
--
local powderFirst = 1                -- First gunpowder slot
local powderLast = 9                 -- Last gunpowder slot
local colorFirst = powderLast + 1    -- First color slot - I don't leave space between groups, so the immediately following slot is my first color slot - that way I don't have to edit the value when I move things around so long as colors begin immediately after gun powder
local colorLast = 24                 -- Last color slot
local effectFirst = colorLast + 1    -- First effect slot - for me, the slot immediately following colorLast
local effectLast = 35                -- Last effect slot
--
-- Don't edit these . . .
--
local allSlots                       -- array holding ALL slot values (gunpowder, color, effect)
local slotList                       -- comma separated list of generated slot values
local fwl = "none"                   -- variable that will be used for wrapping the fireworks launcher
local mon = "none"                   -- variable that will be used for wrapping the monitor, if there is one
local boomCount = 0                  -- number of fireworks to launch - don't change this value, it will be set by the program
local pA                             -- string holding Powder values for display
local pB                             -- string holding Color values for display
local pC                             -- string holding Effect values for display
local pD                             -- string holding Fade Effect value for display
local a,b,c,d,e,f                    -- strings for parsing tables/arrays
--
-- Program Begins
--
local function makeFireworks()
  local iCounter = 0                 -- tracks slot values in fireworks() loops to ensure allSlots is properly fed
  local slotA                        -- first slot for item being processed
  local slotB                        -- last slot for item being processed
  local fadeEffect                   -- fade effect
  local i,j                          -- counters
  allSlots = { }                     -- array holding ALL slot values (gunpowder, color, effect)
  slotList = ""                      -- comma separated list of generated slot values
  pA = ""                            -- string holding Powder values for display
  pB = ""                            -- string holding Color values for display
  pC = ""                            -- string holding Effect values for display
  pD = ""                            -- string holding Fade Effect value for display
  local fadeEffectChance = math.random(1,10)       -- fade effect chance
  if fadeEffectChance >= 7 then                    -- if fadeEffectChance is 7+ then fadeEffect is set to a color slot value and subsequently added to the allSlots array - see comment 3 lines below
    iCounter = iCounter + 1                        -- increment iCounter outside of normal loop to account for Fade Effect and feed allSlots properly
    fadeEffect = math.random(colorFirst,colorLast) -- grab fade effect from random color slot
    allSlots[iCounter] = fadeEffect                -- assign value to allSlots array for fade effect - this probably only adds an additional color since the launcher doesn't 'make' stars as far as I know
  end
  for j = 1,3,1 do                   -- start loop once for each set of items
    if j == 1 then                   -- first iteration is for gun powder
      slotA = powderFirst            -- assign first slot for gun powder
      slotB = powderLast             -- assign last slot for gun powder
    elseif
      j == 2 then                    -- second iteration is for color
      slotA = colorFirst             -- assign first slot for color
      slotB = colorLast              -- assign last slot for color
    elseif
      j == 3 then                    -- third iteration is for effect
      slotA = effectFirst            -- assign first slot for effect
      slotB = effectLast             -- assign last slot for effect
    end
    local numItems = math.random(1,3)-- how many of the current item (gun powder, color, effect)
    for i = 1,numItems,1 do
      iCounter = iCounter + 1
      allSlots[iCounter] = math.random(slotA,slotB)-- assign item from random slot in item slot range
      if j == 1 then                               -- start building display string for gun powder
        if pA == "" then                           -- if pA has no values yet, set the first gun powder slot, otherwise set the remaining slots
          pA = allSlots[iCounter]                  -- first slot is always gun powder ( could use allSlots[1] )
        else
          pA = pA .. ',' .. allSlots[iCounter]     -- remaining gun powder slots
        end
      elseif j == 2 then                           -- start building display string for color
        if pB == "" then                           -- if pB has no values yet, set the first color slot, otherwise set the remaining slots
          pB = allSlots[iCounter]                  -- first color slot
        else
          pB = pB .. ',' .. allSlots[iCounter]     -- remaining color slots
        end
      elseif j == 3 then                           -- start building display string for effect
        if pC == "" then                           -- if pC has no values yet, set the first effect slot, otherwise set the remaining slots
          pC = allSlots[iCounter]                  -- first effect slot
        else
          pC = pC .. ',' .. allSlots[iCounter]     -- remaining effect slots
        end
      end
    end
  end
  if fadeEffectChance >= 7 then                    -- if fadeEffectChance is 7+ then fadeEffect is set to a color slot value and subsequently added to the allSlots array - see comment 3 lines below
    pD = fadeEffect                                -- add effect to string
    slotList = pD .. ',' .. pA .. ',' .. pB .. ',' .. pC -- concatenate all arrays (w/fade effect) into a human-readable CSV string for display
  else
    fadeEffect = 0
    slotList = pA .. ',' .. pB .. ',' .. pC        -- concatenate all arrays (no fade effect) into a human-readable CSV string for display
  end
  --iCounter = 0
end

local function displayIt()           -- mon.* = external monitor, term/term.* & print = console
  if mon ~= "none" then
    mon.clear()
    mon.setCursorPos(4,1)
    mon.write("Fireworks")           -- Monitor display 'header'
    mon.setCursorPos(5,2)
    mon.write("Status")
    mon.setCursorPos(2,4)
    mon.write("Po: " .. pA)          -- Display Powder values on monitor
    mon.setCursorPos(2,5)
    mon.write("Co: " .. pB)          -- Display Color values on monitor
    mon.setCursorPos(2,6)
    mon.write("Fx: " .. pC)          -- Display Effect values on monitor
    mon.setCursorPos(2,7)
    mon.write("Fe: " .. pD)          -- Display Fade value on monitor
  end
  term.clear()
  term.setCursorPos(2,2)
  term.write("Powder:  " .. pA)      -- Display Powder values on terminal
  term.setCursorPos(2,4)
  term.write("Colors:  " .. pB)      -- Display Color values on terminal
  term.setCursorPos(2,6)
  term.write("Effects: " .. pC)      -- Display Effect values on terminal
  term.setCursorPos(2,8)
  term.write("Fade:    " .. pD)      -- Display Fade value on terminal
  term.setCursorPos(2,12)
  term.write("csv: " .. slotList)    -- Display comma separated, single-line list of values generated on terminal
  term.setCursorPos(2,14)
  term.write("raw:")
  term.setCursorPos(7,14)
  term.write(tostring(unpack(allSlots))) -- Display 'unpack' on terminal - as the machine sees it - don't ask me, I just try to make the machine do what I want it to do
  term.write("")
end

local function launchIt()
  fwl.launch(unpack(allSlots))       -- launch firework
  sleep(math.random(0.5,2))          -- random break time (1/2 second to 2 seconds) before repeating the process
end

local function initShow()
  if mon ~= "none" then              -- While we answer the questions, show idle message on monitor
    mon.clear()
    mon.setCursorPos(4,1)
    mon.write("Fireworks")
    mon.setCursorPos(5,2)
    mon.write("Status")
    mon.setCursorPos(6,6)
    mon.write("IDLE")
  end
  term.clear()
  term.setCursorPos(2,1)
  term.write("How many fireworks would you like to see?")
  term.setCursorPos(2,2)
  term.write("(0 = exit / 999 = infinite)")
  term.setCursorPos(2,4)
  boomCount = tonumber(read())
  if boomCount == nil or boomCount < 1 then boomCount = 0 return end
  if boomCount > 998 then boomCount = 999 end
  term.setCursorPos(2,6)
  term.write("How long before starting the show?")
  term.setCursorPos(2,7)
  term.write("(enter number of seconds to delay start of show)")
  term.setCursorPos(2,8)
  term.write("(0 = no delay)")
  term.setCursorPos(2,10)
  local delayTime = tonumber(read())
  if delayTime == nil or delayTime < 1 then delayTime = 0 end
  sleep(delayTime)
end

local function initMe(a,b,c,d,e,f)   -- This will localize these variables within the scope of the function, whether information is passed or not
  for a,b in pairs(rs.getSides()) do -- Auto-detect launcher
    if peripheral.getType(b) == 'launcher' or peripheral.getType(b) == 'fireworks' then
      fwl = peripheral.wrap(b)       -- this will 'wrap' the fireworks launcher, allowing API calls via 'fwl'
      break                          -- we've found the launcher - exit the loop
    elseif peripheral.getType(b) == 'modem' then          -- The launcher is not directly attached - looking for modems
      if not peripheral.call(b,"isWireless") then         -- but not wireless modems
        local mdm = peripheral.wrap(b)
        for e,f in pairs(mdm.getNamesRemote()) do         -- look for a network attached launcher
          if string.sub(f,1,9) == 'fireworks' then        -- if we find one
            fwl = peripheral.wrap(f)                      -- wrap it
            break                                         -- and exit the loop
          end
        end
      end
    end
    if fwl == "none" then                                 -- We didn't find a launcher
      term.clear()
      term.setCursorPos(1,1)
      term.write("Unable to identify fireworks launcher") -- Notify user
      term.setCursorPos(1,3)
      return
    end
  end
  for c,d in pairs(rs.getSides()) do -- Auto-detect monitor
    if peripheral.getType(d) == 'monitor' then
      mon = peripheral.wrap(d)       -- this will 'wrap' the monitor, allowing API calls via 'mon'
      break
    end
    local mon = "none"
  end
  if mon ~= "none" then
    mon.setTextScale(0.5)            -- getSize() reports the size of the monitor based on the TextScale (default is 1) - setTextScale() before doing a getSize() - valid TextScales are 0.5 through 5
    --monW,monH = mon.getSize()      -- gets monitor size - not necessary - this script is formatted for a single screen at TextScale(0.5) - monSize *can* help you determine how many monitors are connected if you know your current textScale
    mon.setBackgroundColor(colors.black)
    mon.setTextColor(colors.white)
    mon.clear()
    mon.setCursorPos(1,1)
  end
end

initMe()
if fwl == "none" then                 -- When the launcher isn't detected
  term.clear()
  term.setCursorPos(2,2)
  term.write("Fireworks Control Station")
  term.setCursorPos(1,5)
  if mon ~= "none" then
    mon.clear()                        -- Clear the monitor
    mon.setCursorPos(4,1)
    mon.write("Fireworks")
    mon.setCursorPos(5,2)
    mon.write("Status")
    mon.setCursorPos(6,6)
    mon.write("IDLE")
  end
  return
end

while true do
  initShow()
  if boomCount > 0 then
    if boomCount == 999 then
      while true do                    -- this will generate an 'infinite' number of fireworks (limited by supplies)
        makeFireworks()                -- generate a firework
        displayIt()                    -- display the info
        launchIt()                     -- fire that sucker!
      end
    else
      local z
      for z = 1,boomCount,1 do         -- this will generate and fire a fixed number of fireworks
        makeFireworks()                -- generate a firework
        displayIt()                    -- display the info
        launchIt()                     -- fire that sucker!
      end
    end
  end
  sleep(2)
end