def save_used_cuts(collage, timestamps): db = util.get_db() cur = db.cursor() for movie_id, time_slice, _ in timestamps: cur.execute('INSERT INTO used_cuts (collage, movie_id, start, end) VALUES (?, ?, ?, ?)', (collage, movie_id, time_slice[0], time_slice[1])) db.commit()
def test_bearing(): cal_id = 3 dep_id = 105 t_start = 1407452400 t_end = 1407455985 #- (50 * 60) db_con = util.get_db('reader') sv = position.steering_vectors(db_con, cal_id) signal = Signal(db_con, dep_id, t_start, t_end) bearings = signal.get_bearings(sv, 3) p = GeneralizedVonMises.mle(bearings) fig, ax = pp.subplots(1, 1) # Plot bearing distribution. N = 100 n, bins, patches = ax.hist(bearings, bins = [ (i * 2 * np.pi) / N for i in range(N) ], normed = 1.0, facecolor='blue', alpha=0.25) # Plot fitted vonMises distribution. x = np.arange(0, 2*np.pi, np.pi / 180) print np.sum(p(x) * (np.pi / 180)) pp.xlim([0,2*np.pi]) ax.plot(x, p(x), 'k-', lw=2, label='$\mu_1=%.2f$, $\mu_2=%.2f$, $\kappa_1=%.2f$, $\kappa_2=%.2f$' % ( p.mu1, p.mu2, p.kappa1, p.kappa2)) pp.xlim([0,2*np.pi]) ax.legend(loc='best', frameon=False) pp.show()
def user(username=None): # User creation if request.method == 'POST': # We are using a serializer to validate data data_validator = UserSchema().load(request.json) # If we have errors if data_validator.errors: # We send back the errors with a 400 return jsonify(data_validator.errors), 400 else: # Are getting the cursor to the database cursor = get_db() try: to_commit = """ INSERT INTO users (username, age, occupation) VALUES ('{username}', '{age}', '{occupation}') """.format(**data_validator.data).strip() cursor.execute(to_commit) cursor.commit() except sqlite3.DatabaseError as e: return jsonify("Error in creation"), 400 return jsonify("We created your goddam user. Be happy now."), 201 # Getting the user elif request.method == 'GET': # Guess who has to implement this!!!!!!!????? return jsonify("Not implemented yet"), 400
def make_screens(movie_id, filename, duration, orientation, screens_tmp): # Originally tried using the FPS filter, but I'm not really sure # where the first thumbnail starts and so that makes it hard to # match thumbnails with timestamps. This way is slower, but I # have more control. screens_tmp.mkdir(exist_ok=False) rng = range(C.SCREENSHOT_START, int(duration), C.SCREENSHOT_FREQUENCY) video_filters = 'stereo3d={}l:ml'.format(orientation) db = util.get_db() cursor = db.cursor() for i, ts in enumerate(rng): thumbnail = screens_tmp.joinpath('thumb{:04d}.jpg'.format(i)) cmd = 'ffmpeg -ss {} -i "{}" -vf "{}" -vframes 1 "{}"'.format( ts, filename, video_filters, thumbnail) print(cmd) subprocess.run(shlex.split(cmd), check=True) cursor.execute( 'INSERT INTO thumbnail (movie_id, filename) VALUES (?, ?)', (movie_id, thumbnail.name)) db.commit()
def main(loc, opt): files = glob(os.path.join(loc, "*jpg")) d = tempfile.mkdtemp() # copy all files in loc to /tmp directory for f in files: dest = os.path.join(d, os.path.basename(f)) copyfile(f, dest) if opt.verbose: print "Copying %s" % f files = glob(os.path.join(d, "*jpg")) db = get_db() for f in files: im = Image(f) if opt.verbose: print "Inserting into the database %s" % im.filename pk = insert_image(im, im.filename.replace("-", " "), None, True) for i in xrange(random.randint(0, 100)): curs = db.execute( """insert into instaclone_likes(image) values(%s)""", [pk])
def main(loc, opt): files = glob(os.path.join(loc, "*jpg")) d = tempfile.mkdtemp() # copy all files in loc to /tmp directory for f in files: dest = os.path.join(d, os.path.basename(f)) copyfile(f, dest) if opt.verbose: print "Copying %s" % f files = glob(os.path.join(d, "*jpg")) db = get_db() for f in files: im = Image(f) if opt.verbose: print "Inserting into the database %s" % im.filename pk = insert_image(im, im.filename.replace("-", " "), None, True) for i in xrange(random.randint(0, 100)): curs = db.execute( """insert into instaclone_likes(image) values(%s)""", [pk], )
def test1(): import time cal_id = 3 dep_id = 105 t_start = 1407452400 t_end = 1407455985 - (59 * 60) db_con = util.get_db('reader') sv = signal1.SteeringVectors(db_con, cal_id) signal = signal1.Signal(db_con, dep_id, t_start, t_end) sites = util.get_sites(db_con) (center, zone) = util.get_center(db_con) assert zone == util.get_utm_zone(db_con) start = time.time() pos = PositionEstimator(dep_id, sites, center, signal, sv, method=signal1.Signal.MLE) print "Finished in {0:.2f} seconds.".format(time.time() - start) print compute_conf(pos.p, pos.num_sites, sites, pos.splines)