class UI(QMainWindow):
"""
Main class. UI + graphics
"""
def __init__(self):
QMainWindow.__init__(self)
self.setWindowTitle("Darknessers")
self.setWindowIcon(QIcon('static/bulb.png'))
self.resize(QSize(800, 600))
scene = QGraphicsScene()
self.scene = scene
widget = QGraphicsView()
# restart = QPushButton('Restart')
# restart.clicked.connect(self.start)
# widget.setLayout(QHBoxLayout())
# widget.layout().addWidget(restart)
widget.setScene(scene)
self.setCentralWidget(widget)
self.start()
def start(self): # Implement restart button with ai reloading
self.scene.clear()
self.drawDots() # Coordinate helper
self.loadAI() # Get ais from plugins
self.stream = Stream(self.ai) # Main stream of events
self.world = World(stream=self.stream) # Main container of objects
self.stream.start()
self.world.ai = self.ai
self.initWorld() # Init barriers, and other stuff
self.api = API(
self.scene,
self.world) # TODO: remove singleton, make personal apis
self.api.addIconsFolder('static')
self.api.addIconsFolder('static/emblems')
self.initAI() # Create AI, place in World
def initAI(self): # Move not-gui logic to World
for ai in self.ai:
ai.world = World(ai, self.world)
### # Start stats. Move to World
self.world.stats[ai] = {
'pos': QPointF(randint(-50, 50), randint(-50, 50)),
'speed': 20,
'skillpoints': 5,
'hp': 50,
'ac': 10,
'light': 150,
'light_angle': 90,
'angle': 0
}
###
if ai.color == 'orange':
self.world.stats[ai]['speed'] = 40
ai.world.stream = self.stream
ai.api = self.api
### # Init start position and graphics logic
cont = QGraphicsPolygonItem()
cont.setPos(self.world.stats[ai]['pos']) # Implement start areas
ai.object = cont
cont.ai = ai
### # Draw light circle and color dot
lc = QGraphicsEllipseItem(
QRectF(QPointF(-ai.lightr, -ai.lightr),
QSizeF(ai.lightr * 2, ai.lightr * 2)), cont)
lc.setStartAngle(
(90 - self.world.stats[ai]['light_angle'] / 2) * 16)
lc.setSpanAngle(self.world.stats[ai]['light_angle'] * 16)
lc.setTransformOriginPoint(QPointF(0, 0))
yl = QColor('yellow')
yl.setAlpha(100)
lc.setPen(QPen(yl))
lc.setBrush(QBrush(yl))
lc.setZValue(-50)
cont.lc = lc
em = QGraphicsPixmapItem(QPixmap(self.api.icons[ai.color]), cont)
em.setZValue(50)
em.setOffset(-10, -10)
cont.em = em
# ra = QGraphicsLineItem(QLineF(QPointF(0, 40), QPointF(0, 0)), cont)
# cont.ra = ra
self.scene.addItem(cont)
self.connect(ai, SIGNAL('moved'), self.moveAI)
self.stream.addEvent(ai.init) # Start AI init method
def initWorld(self): # Move not-gui logic to World
proto = QPolygonF(
[QPointF(0, 0),
QPointF(0, 50),
QPointF(50, 50),
QPointF(50, 0)])
for i in range(0, 5):
b = Barrier(proto)
b.translate(QPointF(randint(-300, 300), randint(-300, 300)))
self.world.barriers.append(b)
item = self.scene.addPolygon(b)
item.setBrush(QBrush(QColor('black')))
def moveAI(self, ai): # Drow changed position
ai.object.setPos(ai.pos)
ai.object.lc.setRotation(ai.angle)
# ai.object.ra.setRotation(ai.angle)
self.scene.update(self.scene.sceneRect())
def drawDots(self): # Helper
tl = QPoint(-300, -300)
br = QPoint(300, 300)
step = 20
for x in range(tl.x(), br.x(), step):
for y in range(tl.y(), br.y(), step):
e = self.scene.addEllipse(
QRectF(QPointF(x, y), QPointF(x + 0.5, y + 0.5)),
QPen(QColor('#555')))
e.setZValue(-60)
def loadAI(self): # Load ais files
self.loader = Loader(AI, 'ai')
self.ai = self.loader.modules
class UI(QMainWindow):
"""
Main class. UI + graphics
"""
def __init__(self):
QMainWindow.__init__(self)
self.setWindowTitle("Darknessers")
self.setWindowIcon(QIcon('static/bulb.png'))
self.resize(QSize(800, 600))
scene = QGraphicsScene()
self.scene = scene
widget = QGraphicsView()
# restart = QPushButton('Restart')
# restart.clicked.connect(self.start)
# widget.setLayout(QHBoxLayout())
# widget.layout().addWidget(restart)
widget.setScene(scene)
self.setCentralWidget(widget)
self.start()
def start(self): # Implement restart button with ai reloading
self.scene.clear()
self.drawDots() # Coordinate helper
self.loadAI() # Get ais from plugins
self.stream = Stream(self.ai) # Main stream of events
self.world = World(stream=self.stream) # Main container of objects
self.stream.start()
self.world.ai = self.ai
self.initWorld() # Init barriers, and other stuff
self.api = API(self.scene, self.world) # TODO: remove singleton, make personal apis
self.api.addIconsFolder('static')
self.api.addIconsFolder('static/emblems')
self.initAI() # Create AI, place in World
def initAI(self): # Move not-gui logic to World
for ai in self.ai:
ai.world = World(ai, self.world)
### # Start stats. Move to World
self.world.stats[ai] = {
'pos': QPointF(randint(-50, 50), randint(-50, 50)),
'speed': 20,
'skillpoints': 5,
'hp': 50,
'ac': 10,
'light': 150,
'light_angle': 90,
'angle': 0
}
###
if ai.color == 'orange':
self.world.stats[ai]['speed'] = 40
ai.world.stream = self.stream
ai.api = self.api
### # Init start position and graphics logic
cont = QGraphicsPolygonItem()
cont.setPos(self.world.stats[ai]['pos']) # Implement start areas
ai.object = cont
cont.ai = ai
### # Draw light circle and color dot
lc = QGraphicsEllipseItem(QRectF(QPointF(-ai.lightr, -ai.lightr), QSizeF(ai.lightr * 2, ai.lightr * 2)), cont)
lc.setStartAngle((90 - self.world.stats[ai]['light_angle'] / 2) * 16)
lc.setSpanAngle(self.world.stats[ai]['light_angle'] * 16)
lc.setTransformOriginPoint(QPointF(0, 0))
yl = QColor('yellow')
yl.setAlpha(100)
lc.setPen(QPen(yl))
lc.setBrush(QBrush(yl))
lc.setZValue(-50)
cont.lc = lc
em = QGraphicsPixmapItem(QPixmap(self.api.icons[ai.color]), cont)
em.setZValue(50)
em.setOffset(-10, -10)
cont.em = em
# ra = QGraphicsLineItem(QLineF(QPointF(0, 40), QPointF(0, 0)), cont)
# cont.ra = ra
self.scene.addItem(cont)
self.connect(ai, SIGNAL('moved'), self.moveAI)
self.stream.addEvent(ai.init) # Start AI init method
def initWorld(self): # Move not-gui logic to World
proto = QPolygonF([QPointF(0, 0), QPointF(0, 50), QPointF(50, 50), QPointF(50, 0)])
for i in range(0, 5):
b = Barrier(proto)
b.translate(QPointF(randint(-300, 300), randint(-300, 300)))
self.world.barriers.append(b)
item = self.scene.addPolygon(b)
item.setBrush(QBrush(QColor('black')))
def moveAI(self, ai): # Drow changed position
ai.object.setPos(ai.pos)
ai.object.lc.setRotation(ai.angle)
# ai.object.ra.setRotation(ai.angle)
self.scene.update(self.scene.sceneRect())
def drawDots(self): # Helper
tl = QPoint(-300, -300)
br = QPoint(300, 300)
step = 20
for x in range(tl.x(), br.x(), step):
for y in range(tl.y(), br.y(), step):
e = self.scene.addEllipse(QRectF(QPointF(x, y), QPointF(x + 0.5, y + 0.5)), QPen(QColor('#555')))
e.setZValue(-60)
def loadAI(self): # Load ais files
self.loader = Loader(AI, 'ai')
self.ai = self.loader.modules
