local util = require("data-util");

-- TECHS
util.add_prerequisite("steel-processing", "graphite-processing")

util.add_prerequisite("low-density-structure", "diamond-processing")
util.add_prerequisite("processing-unit", "diamond-processing")
util.add_prerequisite("laser", "diamond-processing")

-- update carbon black alternate recipe if possible
util.replace_ingredient("carbon-black", "coal", "crushed-coal")


if not mods["space-age"] then -- more uses for graphene in vanilla
  util.add_prerequisite("power-armor-mk2", "graphene")
  util.add_ingredient("power-armor-mk2", "graphene", 30)

  util.add_prerequisite("rocket-control-unit", "graphene")
  util.add_ingredient("rocket-control-unit", "graphene", 1)
end

util.add_prerequisite("electronics", "graphite-processing")

if util.k2() then
  util.add_prerequisite("graphite-processing", "automation-science-pack")
  util.add_prerequisite("graphite-processing", "basic-fluid-handling")
end

-- furnace recipe rework w/ crucible, more in data-final-fixes
util.add_effect("advanced-material-processing-2", {type="unlock-recipe", recipe="crucible"})
for i, ingredient in pairs(util.me.crucible_ingredients) do
  util.add_ingredient("crucible", ingredient, 5)
  if data.raw.item[ingredient] then
    util.add_to_product("crucible", "crucible", 1)
  end
end
for i, furnace in pairs(util.me.furnaces) do
  util.add_ingredient(furnace, "crucible", 1)
  for j, ingredient in pairs(util.me.crucible_ingredients)  do
    util.remove_ingredient(furnace, ingredient)
  end
end

-- space age
if mods["space-age"] then
  util.add_ingredient("crusher", "diamond", 10)
  util.add_ingredient("big-mining-drill", "diamond", 10)
  util.add_prerequisite("big-mining-drill", "diamond-processing-2")
  util.replace_ingredient("quantum-processor", "tungsten-carbide", "diamond", 2)
  if data.raw.fluid["vacuum"] then
    util.add_ingredient("graphite-synthesis", "vacuum", 4)
  end
  util.remove_product("scrap-recycling", "solder")
  util.add_product("scrap-recycling", util.item("rough-diamond", 1, .01))
  util.set_product_probability("scrap-recycling", "processing-unit", 0.03)
  util.add_ingredient("cryogenic-plant", "graphene", 20)
  util.add_ingredient("fluoroketone", "graphene", 1)

  -- Crushing industries
  util.add_ingredient("crusher-2", "diamond", 10)
  util.add_ingredient("big-crusher", "diamond", 10)
  util.add_ingredient("big-electric-crusher", "diamond", 10)

  if mods["planet-muluna"] then
    util.add_unlock("greenhouses", "diamond-processing")
  end
end

-- fullerenes & nanotubes
if util.me.use_fullerenes() then
  util.add_prerequisite("graphene", "fullerenes")
  util.add_prerequisite("military-3", "fullerenes")
  util.replace_some_ingredient("poison-capsule", "coal", 9, "fullerenes", 90)
  if not util.k2() then
    util.replace_some_ingredient("slowdown-capsule", "coal", 4, "fullerenes", 40)
  end
  util.add_ingredient("capture-bot-rocket", "fullerenes", 10)
  util.add_ingredient("biolab", "fullerenes", 50)
  util.add_ingredient("overgrowth-yumako-soil", "fullerenes", 50)
  util.add_ingredient("overgrowth-jellynut-soil", "fullerenes", 50)

  -- K2
  util.add_ingredient("kr-biter-virus", "fullerenes", 100)
  util.add_ingredient("kr-creep-virus", "fullerenes", 100)
  util.replace_some_ingredient("kr-biomass-growing", "petroleum-gas", 25, "fullerenes", 25)

  -- SE
  util.add_ingredient("se-vitalic-reagent", "fullerenes", 4)
  util.add_ingredient("se-medpack-2", "fullerenes", 100)
  util.add_ingredient("se-nutrient-gel", "fullerenes", 5)
  util.add_ingredient("se-nutrient-gel-coal", "fullerenes", 5)
end


-- carbon black
if util.me.use_carbon_black() then

  -- Plastic from carbon black, carbon black from oil processing
  util.replace_ingredient("plastic-bar", "coal", "carbon-black")
  util.replace_ingredient("plastic-bar", "crushed-coal", "carbon-black", 1)
  util.add_product("basic-oil-processing", util.item("carbon-black", 1))
  util.add_product("advanced-oil-processing", util.item("carbon-black", 1))
  util.add_product("light-oil-cracking", util.item("carbon-black", 1))
  util.add_product("heavy-oil-cracking", util.item("carbon-black", 1))
  -- More outlets for carbon black now, so can make it here too
  util.add_product("simple-coal-liquefaction", util.item("carbon-black", 1))
  util.add_product("coal-liquefaction", util.item("carbon-black", 1))

  --K2
  if mods.bztitanium then
    util.add_ingredient("kr-black-reinforced-plate", "carbon-black", 5)
    util.add_ingredient("kr-white-reinforced-plate", "titanium-plate", 1)
  end

  util.add_effect("plastics", {type="unlock-recipe", recipe="carbon-black"})
  util.add_effect("oil-processing", {type="unlock-recipe", recipe="graphite-carbon-black"})

end


util.add_ingredient("rocket-engine-nozzle", "graphite", 1)
util.add_to_product("rocket-engine-nozzle", "rocket-engine-nozzle", 1)
util.add_time("rocket-engine-nozzle", 5)
util.add_ingredient("rocket-engine-nozzle-basic", "graphite", 1)
util.add_to_product("rocket-engine-nozzle-basic", "rocket-engine-nozzle", 1)
util.add_time("rocket-engine-nozzle-basic", 5)

util.add_ingredient("battery", "graphite", 1)
util.add_ingredient("pump", "graphite", 2)


if data.raw.recipe["speed-module-5"] then
  util.add_ingredient("speed-module-5", "diamond", 5)
  util.add_ingredient("effectivity-module-5", "diamond", 5)
  util.add_ingredient("productivity-module-5", "diamond", 5)
elseif data.raw.recipe["speed-module-4"] then
  util.add_ingredient("speed-module-4", "diamond", 6)
  util.add_ingredient("effectivity-module-4", "diamond", 6)
  util.add_ingredient("productivity-module-4", "diamond", 6)
else
  util.add_ingredient("speed-module-2", "diamond", 1)
  util.add_ingredient("efficiency-module-2", "diamond", 1)
  util.add_ingredient("quality-module-2", "diamond", 1)
  util.add_ingredient("productivity-module-2", "diamond", 1)
end
util.add_ingredient("laser-turret", "diamond", 1)
util.add_ingredient("assembling-machine-3", "diamond", 4)

util.add_ingredient("low-density-structure", "diamond", 1)
util.add_ingredient("se-low-density-structure-beryllium", "diamond", 1)
if util.me.reuse() then
  util.add_product("low-density-structure", {type="item", name="diamond", amount=1, probability = 0.5, ignored_by_stats=1, ignored_by_productivity=1})
  util.add_product("se-low-density-structure-beryllium", {type="item", name="diamond", amount=1, probability = 0.5, ignored_by_stats=1, ignored_by_productivity=1})
  if data.raw.recipe["low-density-structure"] and 
    not data.raw.recipe["low-density-structure"].icon and
    not data.raw.recipe["low-density-structure"].icons
    then
    util.set_subgroup("low-density-structure", data.raw.item["low-density-structure"].subgroup)
    util.set_icons("low-density-structure", {
      { icon = "__base__/graphics/icons/low-density-structure.png", icon_size = 64, icon_mipmaps = 4},
    })
    util.set_subgroup("se-low-density-structure-beryllium", data.raw.item["low-density-structure"].subgroup)
    util.set_icons("se-low-density-structure-beryllium", {
      { icon = "__base__/graphics/icons/low-density-structure.png", icon_size = 64, icon_mipmaps = 4},
    })
    util.set_subgroup("low-density-structure-RS", data.raw.item["low-density-structure"].subgroup)
    util.set_icons("low-density-structure-RS", {
      { icon = "__base__/graphics/icons/low-density-structure.png", icon_size = 64, icon_mipmaps = 4},
    })
  end
end

if util.k2() then
  util.replace_some_ingredient("steel-plate", "iron-plate", 2, "graphite", 4)
  util.add_ingredient("tungsten-carbide", "graphite", 4)
  util.replace_some_ingredient("kr-s-c-steel-beam", "iron-plate", 4, "graphite", 8)
  util.replace_some_ingredient("kr-s-c-steel-gear-wheel", "iron-plate", 2, "graphite", 4)
elseif mods.bzfoundry and util.me.foundry_enable() then
  util.replace_some_ingredient("steel-plate", "iron-plate", 1, "graphite", 1)
  util.replace_some_ingredient("tungsten-carbide", "tungsten-plate", 1, "graphite", 1)
end

util.add_ingredient("kr-energy-control-unit", "activated-carbon", 3)

if not util.k2() then
  util.multiply_recipe("lubricant", 2)
end
util.replace_some_ingredient("lubricant", "heavy-oil", 10, "graphite", 1)

-- SE Lubricant
util.multiply_recipe("se-cryonite-lubricant", 2)
util.replace_some_ingredient("se-cryonite-lubricant", "se-cryonite-slush", 10, "graphite", 1)

util.add_ingredient("nuclear-reactor", "graphite", 500)

util.add_ingredient("satellite", "graphene", 100)

-- K2
util.add_ingredient("empty-dt-fuel", "graphite", 1)
util.add_ingredient("kr-steel-pump", "graphite", 2)

util.add_ingredient("kr-electrolysis-plant", "graphite", 10)
util.add_ingredient("kr-fusion-reactor", "graphite", 350)
util.add_ingredient("kr-antimatter-reactor", "graphite", 350)
util.add_ingredient("kr-electric-mining-drill-mk3", "diamond", 5)
util.add_ingredient("kr-quarry-drill", "diamond", 20)
util.add_ingredient("advanced-exoskeleton-equipment", "diamond", 10)

util.add_ingredient("improved-pollution-filter", "graphene", 1)
util.add_ingredient("lithium-sulfur-battery", "graphene", 1)
util.add_ingredient("matter-research-data", "graphene", 5)
util.add_ingredient("kr-advanced-solar-panel", "graphene", 5)
util.add_ingredient("imersite-solar-panel-equipment", "graphene", 5)
util.add_ingredient("kr-advanced-chemical-plant", "graphene", 20)
util.add_ingredient("empty-antimatter-fuel-cell", "graphene", 10)

util.add_prerequisite("kr-lithium-sulfur-battery", "graphene")
util.set_tech_recipe("kr-lithium-sulfur-battery", 
          {
            {"automation-science-pack", 1},
            {"logistic-science-pack", 1},
            {"chemical-science-pack", 1},
            {"production-science-pack", 1},
            {"utility-science-pack", 1},
          })
util.set_tech_recipe("kr-robot-battery", 
          {
            {"automation-science-pack", 1},
            {"logistic-science-pack", 1},
            {"chemical-science-pack", 1},
            {"production-science-pack", 1},
            {"utility-science-pack", 1},
          })
util.set_tech_recipe("kr-battery-mk3-equipment",
          {
            {"automation-science-pack", 1},
            {"logistic-science-pack", 1},
            {"chemical-science-pack", 1},
            {"production-science-pack", 1},
            {"utility-science-pack", 1},
          })



-- AAI Industry
util.add_ingredient("area-mining-drill", "diamond", 8)
util.add_ingredient("vehicle-miner-mk4", "diamond", 4)

-- Space Exploration
util.add_ingredient("se-material-testing-pack", "graphite", 1)
util.add_ingredient("se-space-particle-accelerator", "graphite", 10)
util.add_ingredient("se-space-particle-collider", "graphite", 10)
util.add_ingredient("se-canister", "graphite", 1)
util.add_ingredient("se-naquium-ingot", "graphite", 1)
util.add_ingredient("se-antimatter-reactor", "graphite", 500)
util.replace_some_ingredient("se-naquium-powder", "se-vitalic-acid", 2, "graphene", 2)

util.add_ingredient("se-quantum-processor", "graphene", 1)
util.add_ingredient("se-space-solar-panel-2", "graphene", 4)
util.add_ingredient("se-biochemical-resistance-data", "graphene", 1)
util.add_ingredient("se-tensile-strength-data", "graphene", 1)
util.add_ingredient("se-space-biochemical-laboratory", "graphene", 4)
util.add_ingredient("se-space-decontamination-facility", "graphene", 10)
util.add_ingredient("se-space-genetics-facility", "graphene", 20)
util.add_ingredient("se-space-growth-facility", "graphene", 4)
util.add_ingredient("se-nanomaterial", "graphene", 1)
util.add_ingredient("se-void-probe", "graphene", 100)
util.add_ingredient("se-adaptive-armour-equipment-4", "graphene", 20)
util.add_ingredient("se-thuster-suit-2", "graphene", 50)

util.add_ingredient_raw("se-ballistic-shielding-data", {type="item", name="graphene", amount=10, ignored_by_stats=9, ignored_by_productivity=9})
util.add_product("se-ballistic-shielding-data", {type="item", name="graphene", amount_min=9, amount_max=10, ignored_by_stats=10, ignored_by_productivity=10})

util.add_ingredient("se-space-radiator-2", "diamond", 16)

-- Add diamonds to some SE Data, and boost output
util.add_ingredient("se-rigidity-data", "diamond", 1)
util.add_to_product("se-rigidity-data", "se-rigidity-data", 1)
util.add_to_ingredient("se-rigidity-data", "se-empty-data", 1)
util.add_ingredient("se-hot-thermodynamics-data", "diamond", 1)
util.add_to_product("se-hot-thermodynamics-data", "se-hot-thermodynamics-data", 1)
util.add_to_ingredient("se-hot-thermodynamics-data", "se-empty-data", 1)
util.add_ingredient("se-compressive-strength-data", "diamond", 1)
util.add_to_product("se-compressive-strength-data", "se-compressive-strength-data", 1)
util.add_to_ingredient("se-compressive-strength-data", "se-empty-data", 1)


-- Realistic Reactors
util.add_ingredient("realistic-reactor", "graphite", 100)
util.add_ingredient("breeder-reactor", "graphite", 100)

-- Plutonium Energy
util.add_ingredient("MOX-reactor", "graphite", 400)

-- RFP
util.add_ingredient("rfp-reactor", "graphite", 1000)
util.add_ingredient("rfp-aneutronic-reactor", "graphite", 5000)


-- vortik's deep core mining
util.add_ingredient("vtk-deepcore-mining-moho", "diamond", 10)
util.add_ingredient("vtk-deepcore-mining-drone", "diamond", 1)
util.add_ingredient("vtk-deepcore-mining-drill", "diamond", 20)
util.add_ingredient("vtk-deepcore-mining-drill-advanced", "diamond", 50)

-- Advanced Mining Drills
util.add_ingredient("elite-mining-drill", "diamond", 5)
util.add_ingredient("ultimate-mining-drill", "diamond", 10)

-- Skandragon
util.add_ingredient("skan-elite-mining-drill", "diamond", 4)
util.add_ingredient("skan-ultimate-mining-drill", "diamond", 4)

-- Hiladdar
util.add_ingredient("hsmd-electric-mining-drill-mk3", "diamond", 3)
util.add_ingredient("hsmd-electric-mining-drill-mk4", "diamond", 3)
util.add_ingredient("hsmd-electric-mining-drill-mk5", "diamond", 3)
util.add_ingredient("hsmd-electric-mining-drill-mk6", "diamond", 3)

-- Zombies
util.add_ingredient("electric-mining-drill-mk2", "diamond", 2)
util.add_ingredient("edrill-1to2", "diamond", 2)
util.add_ingredient("electric-mining-drill-mk3", "diamond", 2)
util.add_ingredient("electric-mining-drill-mk4", "diamond", 2)
util.add_ingredient("electric-mining-drill-mk5", "diamond", 2)
util.add_ingredient("electric-mining-drill-mk6", "diamond", 2)
if not mods["zombiesextended-core"] then
  util.add_ingredient("electric-mining-drill-mk1", "tungsten-carbide", 5)
end

-- Tiny
util.add_ingredient("tiny-electric-mining-drill-3", "diamond", 2)
util.add_ingredient("tiny-electric-mining-drill-5", "diamond", 3)

-- TA Miners
util.add_ingredient("moho_extractor", "diamond", 50)
util.add_ingredient("mantle_extractor", "diamond", 100)

-- other mining drills
util.add_ingredient("electric-mining-drill-lv2", "diamond", 5)
util.add_ingredient("advanced-electric-mining-drill", "diamond", 5)
util.add_ingredient("omega-drill", "diamond", 15)
util.add_ingredient("omega-drill-mk2", "diamond", 30)
util.add_ingredient("passive-furnace", "diamond", 2)
util.add_ingredient("quarry-mk3", "diamond", 10)

-- FastFurnaces
util.add_ingredient("reika-fast-electric-mining-drill", "diamond", 1)

-- Power Armor Mk3
util.add_ingredient("pamk3-pamk3", "graphene", 60)

-- Orbital ion cannon
util.add_ingredient("orbital-ion-cannon", "graphene", 100)

-- FE+
util.add_ingredient("pump-mk2", "graphite", 4)
util.add_ingredient("offshore-pump-mk2", "graphite", 2)
util.add_ingredient("pumpjack-mk2", "graphite", 2)
util.add_ingredient("fusion-reactor-mk2-equipment", "graphene", 2)

util.add_shiftite_recipe("graphite", {"alpha"}, 1)
util.add_shiftite_recipe("diamond", {"beta", "epsilon"}, 1)