import { AnalyticUnitId } from './analytic_unit_model'; import { Collection, makeDBQ } from '../services/data_service'; import * as _ from 'lodash'; let db = makeDBQ(Collection.SEGMENTS); export type SegmentId = string; export class Segment { constructor( public analyticUnitId: AnalyticUnitId, public from: number, public to: number, public labeled: boolean = false, public deleted: boolean = false, public id?: SegmentId, public params?: any ) { if(analyticUnitId === undefined) { throw new Error('AnalyticUnitId is undefined'); } if(from === undefined) { throw new Error('from is undefined'); } if(isNaN(from)) { throw new Error('from is NaN'); } if(to === undefined) { throw new Error('to is undefined'); } if(isNaN(to)) { throw new Error('to is NaN'); } } public toObject() { return { _id: this.id, analyticUnitId: this.analyticUnitId, from: this.from, to: this.to, labeled: this.labeled, deleted: this.deleted }; } static fromObject(obj: any): Segment { if(obj === undefined) { throw new Error('obj is undefined'); } return new Segment( obj.analyticUnitId, +obj.from, +obj.to, obj.labeled, obj.deleted, obj._id ); } public equals(obj: Segment) : boolean { return this.analyticUnitId === obj.analyticUnitId && this.from === obj.from && this.to === obj.to && this.labeled === this.labeled && this.deleted === this.deleted; } } export type FindManyQuery = { timeFromGTE?: number, timeToLTE?: number, intexGT?: number, labeled?: boolean, deleted?: boolean } export async function findMany(id: AnalyticUnitId, query: FindManyQuery): Promise { var dbQuery: any = { analyticUnitId: id }; if(query.timeFromGTE !== undefined) { dbQuery.from = { $gte: query.timeFromGTE }; } if(query.timeToLTE !== undefined) { dbQuery.to = { $lte: query.timeToLTE }; } if(query.labeled !== undefined) { dbQuery.labeled = query.labeled; } if(query.deleted !== undefined) { dbQuery.deleted = query.deleted; } let segs = await db.findMany(dbQuery); if(segs === null) { return []; } return segs.map(Segment.fromObject); } export async function insertSegments(segments: Segment[]) { if(_.isEmpty(segments)) { return []; } let analyticUnitId: AnalyticUnitId = segments[0].analyticUnitId; let segmentIdsToRemove: SegmentId[] = []; let segmentsToInsert: Segment[] = []; let learningSegments: Segment[] = await db.findMany({ analyticUnitId, labeled: true, deleted: false }); for(let segment of segments) { let intersectedLearning = learningSegments.filter(s => { return segment.from <= s.to && segment.to >= s.from; }); if(intersectedLearning.length > 0) { continue; } let intersectedSegments = await db.findMany({ analyticUnitId, to: { $gte: segment.from }, from: { $lte: segment.to }, labeled: segment.labeled, deleted: segment.deleted }); if(intersectedSegments.length > 0) { let from = _.minBy(intersectedSegments, 'from').from; let to = _.maxBy(intersectedSegments, 'to').to; let newSegment = Segment.fromObject(segment.toObject()); newSegment.from = from; newSegment.to = to; segmentIdsToRemove = segmentIdsToRemove.concat(intersectedSegments.map(s => s._id)); segmentsToInsert.push(newSegment); } else { segmentsToInsert.push(segment); } } await db.removeMany(segmentIdsToRemove); return db.insertMany(segmentsToInsert.map(s => s.toObject())); } export async function setSegmentsDeleted(ids: SegmentId[]) { return db.updateMany(ids, { deleted: true, labeled: false }); } export function removeSegments(idsToRemove: SegmentId[]) { return db.removeMany(idsToRemove); }