Mercurial > code > home > repos > light9
view bin/captureDevice @ 1883:17bee25a20cb
reformat
Ignore-this: 9392ba9c775fe3c5997496ddfe598f09
author | Drew Perttula <drewp@bigasterisk.com> |
---|---|
date | Tue, 28 May 2019 06:46:08 +0000 |
parents | 3c523c71da29 |
children | 1a7e5b07bf17 |
line wrap: on
line source
#!bin/python from rdflib import URIRef from twisted.internet import reactor from twisted.internet.defer import inlineCallbacks, Deferred import logging import optparse import os import time import treq import cyclone.web, cyclone.websocket, cyclone.httpclient from greplin import scales from run_local import log from cycloneerr import PrettyErrorHandler from light9.namespaces import L9, RDF from light9 import networking, showconfig from rdfdb.syncedgraph import SyncedGraph from light9.paint.capture import writeCaptureDescription from light9.greplin_cyclone import StatsForCyclone from light9.effect.settings import DeviceSettings from light9.collector.collector_client import sendToCollector from rdfdb.patch import Patch from light9.zmqtransport import parseJsonMessage stats = scales.collection('/webServer', scales.PmfStat('setAttr')) class Camera(object): def __init__(self, imageUrl): self.imageUrl = imageUrl def takePic(self, uri, writePath): log.info('takePic %s', uri) return treq.get( self.imageUrl).addCallbacks(lambda r: self._done(writePath, r), log.error) @inlineCallbacks def _done(self, writePath, response): jpg = yield response.content() try: os.makedirs(os.path.dirname(writePath)) except OSError: pass with open(writePath, 'w') as out: out.write(jpg) log.info('wrote %s', writePath) def deferSleep(sec): d = Deferred() reactor.callLater(sec, d.callback, None) return d class Capture(object): firstMoveTime = 3 settleTime = .5 def __init__(self, graph, dev): self.graph = graph self.dev = dev def steps(a, b, n): return [round(a + (b - a) * i / n, 5) for i in range(n)] startTime = time.time() self.captureId = 'cap%s' % (int(startTime) - 1495170000) self.toGather = [] #quantum rxSteps = steps(.06, .952, 10) rySteps = steps(0.1, .77, 5) zoomSteps = steps(.12, .85, 3) # aura rxSteps = steps(0.15, .95, 10) rySteps = steps(0, .9, 5) zoomSteps = steps(.6, .9, 3) row = 0 for ry in rySteps: xSteps = rxSteps[:] if row % 2: xSteps.reverse() row += 1 for rx in xSteps: for zoom in zoomSteps: self.toGather.append( DeviceSettings( graph, [ (dev, L9['rx'], rx), (dev, L9['ry'], ry), (dev, L9['color'], '#ffffff'), (dev, L9['zoom'], zoom), #(dev, L9['focus'], 0.13), ])) self.devTail = dev.rsplit('/')[-1] self.session = URIRef('/'.join( [showconfig.showUri(), 'capture', self.devTail, self.captureId])) self.ctx = URIRef(self.session + '/index') self.graph.patch( Patch(addQuads=[ (self.session, RDF.type, L9['CaptureSession'], self.ctx), ])) self.numPics = 0 self.settingsCache = set() self.step().addErrback(log.error) def off(self): return sendToCollector(client='captureDevice', session='main', settings=DeviceSettings(self.graph, [])) @inlineCallbacks def step(self): if not self.toGather: yield self.off() yield deferSleep(1) reactor.stop() return settings = self.toGather.pop() log.info('[%s left] move to %r', len(self.toGather), settings) yield sendToCollector(client='captureDevice', session='main', settings=settings) yield deferSleep(self.firstMoveTime if self.numPics == 0 else self.settleTime) picId = 'pic%s' % self.numPics path = '/'.join(['capture', self.devTail, self.captureId, picId ]) + '.jpg' uri = URIRef(self.session + '/' + picId) yield camera.takePic(uri, os.path.join(showconfig.root(), path)) self.numPics += 1 writeCaptureDescription(self.graph, self.ctx, self.session, uri, self.dev, path, self.settingsCache, settings) reactor.callLater(0, self.step) camera = Camera( 'http://plus:8200/picamserve/pic?res=1080&resize=800&iso=800&redgain=1.6&bluegain=1.6&shutter=60000&x=0&w=1&y=0&h=.952' ) class Attrs(PrettyErrorHandler, cyclone.web.RequestHandler): def put(self): with stats.setAttr.time(): client, clientSession, settings, sendTime = parseJsonMessage( self.request.body) self.set_status(202) def launch(graph): cap = Capture(graph, dev=L9['device/aura5']) reactor.listenTCP(networking.captureDevice.port, cyclone.web.Application(handlers=[ (r'/()', cyclone.web.StaticFileHandler, { "path": "light9/web", "default_filename": "captureDevice.html" }), (r'/stats', StatsForCyclone), ]), interface='::', cap=cap) log.info('serving http on %s', networking.captureDevice.port) def main(): parser = optparse.OptionParser() parser.add_option("-v", "--verbose", action="store_true", help="logging.DEBUG") (options, args) = parser.parse_args() log.setLevel(logging.DEBUG if options.verbose else logging.INFO) graph = SyncedGraph(networking.rdfdb.url, "captureDevice") graph.initiallySynced.addCallback(lambda _: launch(graph)).addErrback( log.error) reactor.run() if __name__ == '__main__': main()