<reponame>gabrielvbauer/nft_generation
from TopHeadAcessory import TopHeadAcessory
from PIL import Image
import numpy as np
import os


class Hat(TopHeadAcessory):
  index = 0
  bg = (255, 255, 255)
  ol = (0, 0, 0)
  ff = (255, 255, 255)
  mn = (255, 255, 255)

  def __init__(self, index):
    self.index = index
    self.generateImage()
    
  
  def generateImage(self):
    self.mn = TopHeadAcessory.generateRgb(self.index)
    if self.mn == (255, 255, 255):
      print("255, 255, 255 foi gerado!")
      self.mn = TopHeadAcessory.generateRgb(self.index)

    template = [
      [self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg],
      [self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg],
      [self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.ol, self.ol, self.ol, self.ol, self.bg, self.bg, self.bg],
      [self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.ol, self.mn, self.mn, self.mn, self.mn, self.ol, self.bg, self.bg],
      [self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.ol, self.mn, self.mn, self.mn, self.mn, self.mn, self.ol, self.bg, self.bg],
      [self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.ol, self.mn, self.mn, self.mn, self.ol, self.mn, self.ol, self.bg, self.bg],
      [self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.ol, self.ol, self.ol, self.bg, self.ol, self.bg, self.bg, self.bg],
      [self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg],
      [self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg],
      [self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg],
      [self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg],
      [self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg],
      [self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg],
      [self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg],
      [self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg],
      [self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg],
      [self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg],
      [self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg],
      [self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg],
    ]

    arrImg = np.array(template, dtype=np.uint8)
    new_image = Image.fromarray(arrImg)
    new_image = new_image.resize(TopHeadAcessory.dimensions, resample=0)
    imgName = os.path.dirname(__file__) + '/traits/' +  str(self.index) + '.png'
    new_image.save(imgName)

    TopHeadAcessory.removeBackground(self.index)

from TopHeadAcessory import TopHeadAcessory
from PIL import Image
import numpy as np
import os


class Hat(TopHeadAcessory):
  index = 0
  bg = (255, 255, 255)
  ol = (0, 0, 0)
  ff = (255, 255, 255)
  mn = (255, 255, 255)

  def __init__(self, index):
    self.index = index
    self.generateImage()
    
  
  def generateImage(self):
    self.mn = TopHeadAcessory.generateRgb(self.index)
    if self.mn == (255, 255, 255):
      print("255, 255, 255 foi gerado!")
      self.mn = TopHeadAcessory.generateRgb(self.index)

    template = [
      [self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg],
      [self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg],
      [self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.ol, self.ol, self.ol, self.ol, self.bg, self.bg, self.bg],
      [self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.ol, self.mn, self.mn, self.mn, self.mn, self.ol, self.bg, self.bg],
      [self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.ol, self.mn, self.mn, self.mn, self.mn, self.mn, self.ol, self.bg, self.bg],
      [self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.ol, self.mn, self.mn, self.mn, self.ol, self.mn, self.ol, self.bg, self.bg],
      [self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.ol, self.ol, self.ol, self.bg, self.ol, self.bg, self.bg, self.bg],
      [self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg],
      [self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg],
      [self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg],
      [self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg],
      [self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg],
      [self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg],
      [self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg],
      [self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg],
      [self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg],
      [self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg],
      [self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg],
      [self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg, self.bg],
    ]

    arrImg = np.array(template, dtype=np.uint8)
    new_image = Image.fromarray(arrImg)
    new_image = new_image.resize(TopHeadAcessory.dimensions, resample=0)
    imgName = os.path.dirname(__file__) + '/traits/' +  str(self.index) + '.png'
    new_image.save(imgName)

    TopHeadAcessory.removeBackground(self.index)