src.php

<?
	require_once "db/connection.php";

	ini_set('max_execution_time', '3600'); //3600 seconds - 1 hour

	$CONST = (object)[
		'editionSize' => 324,
		'rootPath' => str_replace(array("\\"), "/", __DIR__),
		'dnaDelimiter' => '-',
		'rarityDelimiter' => '#',
		'collectionName' => 'Beer Mages',
		'basePath' => 'https://storage.googleapis.com/nfkingdom.appspot.com',
		'imgW' => 512,
		'imgH' => 512,
		'collectionLayersFolder' => 'mages',
		'collectionBasePrice' => 1000,
	];

	$LAYERS = [
		'Background',
		'Weapon',
		'Body',
		'Face',
		'Hood',
		'Beard',
		'Power',
	];
	// $LAYERS = [
	// 	"Background",
	// 	"Eyeball",
	// 	"Eye color",
	// 	"Iris",
	// 	"Shine",
	// 	"Bottom lid",
	// 	"Top lid",
	// ];

	$DNA_LIST = [];

	$IMG_W = $CONST->imgW;
	$IMG_H = $CONST->imgH;
	$IMG_BASE = imagecreatetruecolor($IMG_W, $IMG_H);

	function array_find($xs, $f) {
		foreach ($xs as $x) {
			if ($f($x) === true)
				return $x;
		}
		return null;
	}

	function isDnaUnique($dna) {
		global $DNA_LIST;
		$isUnique = !in_array($dna, $DNA_LIST);
		if ($isUnique) {
			array_push($DNA_LIST, $dna);
		}
		return $isUnique;
	}

	function cleanName($name) {
		global $CONST;
		$prettyName = substr($name, 0, -4);
  		$prettyName = explode($CONST->rarityDelimiter, $prettyName)[0];
		return $prettyName;
	}

	function getRarityWeight($picname) {
		global $CONST;
		$nameWithoutExt = substr($picname, 0, -4);
		$rarity = explode($CONST->rarityDelimiter, $nameWithoutExt)[1];
		$rarity = is_null($rarity) ? 100 : $rarity;
		var_dump($rarity);
		return $rarity;
	}

	function getLayerPictures($layers_dir_path) {
		$unfiltered = scandir($layers_dir_path, SCANDIR_SORT_NONE);
		$images = [];
		foreach ($unfiltered as $k => $v) {
			if (substr($v, -4) === ".png") {
				array_push($images, $v);
			}
		}
		$res = array_map(function($pic, $i) use ($layers_dir_path) {
			return [
				'id' => $i,
				'pic_name' => cleanName($pic),
				'filename' => $pic,
				'path' => "$layers_dir_path/$pic",
				'weight' => getRarityWeight($pic),
			];
		}, $images, array_keys($images));
		// print_r($res);
		return $res;
	}

	function setupLayers() {
		global $LAYERS, $CONST;
		$layers = array_map(function($l, $i) use ($CONST) {
			return [
				"order" => $i,
				"pictures" => getLayerPictures("$CONST->rootPath/$CONST->collectionLayersFolder/$l"),
				"name" => $l,
			];
		}, $LAYERS, array_keys($LAYERS));
		// print_r($layers);
		return $layers;
	}

	function generateDna($layers) {
		global $CONST;
		$dnaNodes = [];
		$rarityRates = [];
		$collectionTotalWeight = 0;
		foreach ($layers as $k => $v) {
			$totalWeight = 0;
			$layer_pics = $v['pictures'];
			$layer_size = count($layer_pics);

			// echo "<b>Layer {$v['name']}</b><br>";

			// summarize weights of all pictures of a layers
			foreach ($layer_pics as $k => $pic) {
				$totalWeight += (int)$pic['weight'];
				// echo "layer_fragment rarity: {$pic['weight']}<br>";
			}
		

			$accumulator = rand(0, $totalWeight);
			for ($i=0; $i < count($layer_pics); $i++) {
				$accumulator -= $layer_pics[$i]['weight'];
				if ($accumulator <= 0) {
					$pic = $layer_pics[$i];
					$pic_id = $pic['id'];
					$pic_filename = $pic['filename'];
					if ($layer_size > 1) {
						$rarity = $pic['weight'] / $totalWeight * 100;
						array_push($rarityRates, $rarity);
						$collectionTotalWeight += $totalWeight;
					}
					array_push($dnaNodes, "$pic_id:$pic_filename");
					break;
				}
			}
			// if ($layer_size > 1) {
			// 	$r = $pic['weight'] / $totalWeight * 100;
			// 	echo "item rarity: $r<br>";
			// } else {
			// 	echo "one item layer<br>";
			// }
		}
		$dna = implode($CONST->dnaDelimiter, $dnaNodes);
		$t = array_sum($rarityRates);
		$rarity = array_sum($rarityRates) / count($rarityRates);
		// echo "<b>Collection total weight:</b> $collectionTotalWeight, item total weight: $t, <b>Rarity:</b> $rarity<br><br>";
		return ["dna" => $dna, "rarity" => $rarity];
	}

	function getDnaNodeId($dna_node) {
		return explode(':', $dna_node)[0];
	}

	function getDnaLayers($dna, $layers) {
		global $CONST;
		$mapping = array_map(function($l, $i) use ($CONST, $dna) {
			$pics = $l['pictures'];
			// get dna node associated with layer
			$dna_node = explode($CONST->dnaDelimiter, $dna)[$i];
			$dna_node_id = getDnaNodeId($dna_node);
			$pic = array_find($pics, function($p) use ($dna_node_id) {
				// echo $p['id']." - ".$dna_node_id."<br>";
				return $p['id'] == $dna_node_id;
			});
			return [
				"layer_name" => $l['name'],
				"layer_pic" => $pic,
			];
		}, $layers, array_keys($layers));
		return $mapping;
	}

	function create() {
		global $CONST, $IMG_BASE;

		// configurate all layers and pictures
		$layers = setupLayers();

		//open connection
		$c = connect();
		$type = "creatures";
		$qCreateCollection = "insert into collections (name, type) values ('$CONST->collectionName', '$type')";
		$qGetId = "select LAST_INSERT_ID()";
		$new_collection = $c->query($qCreateCollection);
		$new_collection_id = $c->query($qGetId)->fetch_assoc()['LAST_INSERT_ID()'];
		// var_dump($new_collection);
		// var_dump($new_collection_id);
		if (!$new_collection) {
			disconnect($c);
			return;
		}

		// create unique nft picture & write to database
		$i = 1;
		while ($i <= $CONST->editionSize) { 
			// get unique nft dna
			$gen = generateDna($layers);
			$dna = $gen['dna'];
			$rarity = $gen['rarity'];
			$price = 200 + $CONST->collectionBasePrice * (1 - $rarity / 100) * rand(99, 101) / 100;

			// if dna exists, repeat iteration
			if (!isDnaUnique($dna)) {
				continue;
			}

			$layersToDraw = getDnaLayers($dna, $layers);

			// clear 'canvas' by setting empty image
			$output_image = $IMG_BASE;
			
			foreach ($layersToDraw as $k => $l) {
				$pic = $l['layer_pic'];
				$src = $pic['path'];

				$img = imagecreatefrompng($src);

				// header('Content-type: image/png');
				imagecopy($output_image, $img, 0, 0, 0, 0, $CONST->imgW, $CONST->imgH);
			}
			$filename = "$i.png";

			// format collection name
			$col_name_formatted = implode("-", explode(" ", $CONST->collectionName));
			$image_url = "$CONST->basePath/$col_name_formatted/$filename";
			
			// echo "$rarity, ";
			// echo $new_collection_id."<br>";
			$qInsertItem = "insert into items (collection_id, rarity, price, image_url) values ('$new_collection_id', '$rarity', '$price', '$image_url')";
			$res = $c->query($qInsertItem);
			if (!$res) {
				var_dump('ERROR: Insertion token data in database.');
				disconnect($c);
				return;
			}

			imagepng($output_image, "$CONST->rootPath/results/$CONST->collectionName/$filename");
			// imagedestroy($output_image);

			$i += 1;
		}

		// disconnect($c);
	}

	create();
	echo "All done";
	print_r($DNA_LIST)
?>