light9 Changeset - ec816fd31c83

Changeset - ec816fd31c83

Parent rev.

Child rev.

[Not reviewed]

default

0 2 0

Drew Perttula - 6 years ago 2019-06-06 02:59:26
drewp@bigasterisk.com

reformat
Ignore-this: efe8ed938af5fbba3b1acdb8bb1efecd

2 files changed with 30 insertions and 26 deletions:

bin/vidref

light9/vidref/videorecorder.py

0 comments (0 inline, 0 general)

bin/vidref

➞

Show inline comments

@@ @@ -13,39 +13,40 @@ light9/vidref/setup.html web ui for setu @@
 light9/web/light9-vidref-live.js LitElement for live video frames
 light9/web/light9-vidref-playback.js LitElement for video playback
 """
 from run_local import log
 from twisted.internet import reactor, defer
 from typing import cast
 import logging, optparse, json, base64, os, glob
 import cyclone.web, cyclone.httpclient, cyclone.websocket
 from light9 import networking, showconfig
 from light9.vidref import videorecorder
 from rdflib import URIRef
 from light9.newtypes import Song
 from rdfdb.syncedgraph import SyncedGraph
 from cycloneerr import PrettyErrorHandler
 from typing import cast
 from greplin import scales
 from greplin.scales.cyclonehandler import StatsHandler
 from rdflib import URIRef
 from twisted.internet import reactor, defer
 import cyclone.web, cyclone.httpclient, cyclone.websocket
 from cycloneerr import PrettyErrorHandler
 from light9 import networking, showconfig
 from light9.newtypes import Song
 from light9.vidref import videorecorder
 from rdfdb.syncedgraph import SyncedGraph
 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)
 stats = scales.collection(
     '/webServer',
     scales.RecentFpsStat('liveWebsocketFrameFps'),
     scales.IntStat('liveClients'),
+)
 class Snapshot(cyclone.web.RequestHandler):
     @defer.inlineCallbacks
     def post(self):
         # save next pic
         # return /snapshot/path
@@ @@ -105,26 +106,26 @@ class Time(cyclone.web.RequestHandler): @@
         t = body['t']
         source = body['source']
         self.settings.gui.incomingTime(t, source)
         self.set_status(202)
 class Clips(PrettyErrorHandler, cyclone.web.RequestHandler):
     def delete(self):
         clip = URIRef(self.get_argument('uri'))
         videorecorder.deleteClip(clip)
 class ReplayMap(PrettyErrorHandler, cyclone.web.RequestHandler):
     def get(self):
         song = Song(self.get_argument('song'))
         clips = []
         videoPaths = glob.glob(os.path.join(videorecorder.songDir(song),
                                             b'*.mp4'))
         videoPaths = glob.glob(
             os.path.join(videorecorder.songDir(song), b'*.mp4'))
         for vid in videoPaths:
             pts = []
             for line in open(vid.replace(b'.mp4', b'.timing'), 'rb'):
                 _v, vt, _eq, _song, st = line.split()
                 pts.append([float(st), float(vt)])
@@ @@ -172,13 +173,12 @@ reactor.listenTCP( @@
                 "path": 'todo',
             }),
             (r'/time', Time),
             (r'/stats/(.*)', StatsHandler, {
                 'serverName': 'vidref'
             }),
         ],
         debug=True,
     ))
 log.info("serving on %s" % port)
 reactor.run()

light9/vidref/videorecorder.py

➞

Show inline comments

 from dataclasses import dataclass
 from io import BytesIO
 from typing import Optional
 import time, logging, os, traceback
 from io import BytesIO
 import gi
 gi.require_version('Gst', '1.0')
 gi.require_version('GstBase', '1.0')
 import PIL.Image
 from gi.repository import Gst
 from greplin import scales
 from rdflib import URIRef
 from rx.subject import BehaviorSubject
 from twisted.internet import threads
-from rdflib import URIRef
+import PIL.Image
 import moviepy.editor
 import numpy
 from greplin import scales
 from light9 import showconfig
 from light9.ascoltami.musictime_client import MusicTime
 from light9.newtypes import Song
 from light9 import showconfig
 log = logging.getLogger()
 stats = scales.collection(
     '/recorder',
     scales.PmfStat('jpegEncode', recalcPeriod=1),
     scales.IntStat('deletes'),
     scales.PmfStat('waitForNextImg', recalcPeriod=1),
     scales.PmfStat('crop', recalcPeriod=1),
     scales.RecentFpsStat('encodeFrameFps'),
     scales.RecentFpsStat('queueGstFrameFps'),
+)
 @dataclass
 class CaptureFrame:
     img: PIL.Image
     song: Song
     t: float
     isPlaying: bool
@@ @@ -52,30 +52,35 @@ class CaptureFrame: @@
 def songDir(song: Song) -> bytes:
     return os.path.join(
         showconfig.root(), b'video',
         song.replace('http://', '').replace('/', '_').encode('ascii'))
 def takeUri(songPath: bytes) -> URIRef:
     p = songPath.decode('ascii').split('/')
     take = p[-1].replace('.mp4', '')
     song = p[-2].split('_')
     return URIRef('/'.join(
         ['http://light9.bigasterisk.com/show', song[-2], song[-1], take]))
 def deleteClip(uri: URIRef):
     # uri http://light9.bigasterisk.com/show/dance2019/song6/take_155
     # path show/dance2019/video/light9.bigasterisk.com_show_dance2019_song6/take_155.*
     w = uri.split('/')[-4:]
     path = '/'.join([w[0], w[1], 'video',
                      f'light9.bigasterisk.com_{w[0]}_{w[1]}_{w[2]}', w[3]])
     path = '/'.join([
         w[0], w[1], 'video', f'light9.bigasterisk.com_{w[0]}_{w[1]}_{w[2]}',
         w[3]
     ])
     log.info(f'deleting {uri} {path}')
     stats.deletes += 1
     for fn in [path + '.mp4', path + '.timing']:
         os.remove(fn)
 class FramesToVideoFiles:
     """
     nextWriteAction: 'ignore'
     currentOutputClip: None
@@ @@ -163,13 +168,12 @@ class FramesToVideoFiles: @@
         log.info('write_videofile done')
         self.currentOutputClip = None
         if self.currentClipFrameCount < 400:
             log.info('too small- deleting')
             deleteClip(takeUri(self.outMp4.encode('ascii')))
     def _bg_make_frame(self, video_time_secs):
         stats.encodeFrameFps.mark()
         if self.nextWriteAction == 'close':
             raise StopIteration  # the one in write_videofile
         elif self.nextWriteAction == 'notWritingClip':
@@ @@ -253,13 +257,13 @@ class GstSource: @@
             traceback.print_exc()
         return Gst.FlowReturn.OK
     @stats.crop.time()
     def crop(self, img):
         return img.crop((0, 100, 640, 380))
     def setupPipelineError(self, pipe, cb):
         bus = pipe.get_bus()
         def onBusMessage(bus, msg):
             print('nusmsg', msg)

0 comments (0 inline, 0 general)