scripting/12-binary-event-sequences.main.kts at main · dtsql-oss/scripting · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
@file:DependsOn("libs/client-0.2.1.jar")
@file:Import("scripting-engine.main.kts")

import org.tsdl.client.api.builder.Range.IntervalType
import org.tsdl.client.api.builder.TsdlQueryBuilder.`as`
import org.tsdl.client.impl.builder.EchoSpecificationImpl.echo
import org.tsdl.client.impl.builder.EventConnectiveSpecificationImpl.and
import org.tsdl.client.impl.builder.EventSpecificationImpl.event
import org.tsdl.client.impl.builder.RangeImpl.for_
import org.tsdl.client.impl.builder.SelectSpecificationImpl.precedes
import org.tsdl.client.impl.builder.ThresholdFilterSpecificationImpl.gt
import org.tsdl.client.impl.builder.ThresholdFilterSpecificationImpl.lt
import org.tsdl.client.impl.builder.ValueSampleSpecificationImpl.average
import org.tsdl.client.impl.builder.YieldSpecificationImpl.allPeriods
import org.tsdl.infrastructure.common.TsdlTimeUnit
import org.tsdl.infrastructure.model.TsdlPeriodSet
import java.io.File
import java.time.temporal.ChronoUnit

val queryEndpoint = "http://localhost:8080/query"
val storageName = "csv"
val serviceConfiguration = mapOf(
    "targetFile" to File("./evaluation-logs/12-binary-event-sequences.log").canonicalPath
)
val lookupConfiguration = mapOf(
    "filePath" to File(args[0]).canonicalPath,
    "skipHeaders" to 1,
    "fieldSeparator" to ','
)
val transformationConfiguration = mapOf(
    "valueColumn" to 1,
    "timeColumn" to 0,
    "timeFormat" to "yyyy-MM-dd'T'HH:mm:ssXXX"
)

invokeAnalysisSession { engine, client ->
    val query = engine.tsdl {
        it
            .valueSamples(average("globalAvg", echo("4")))
            .events(
                event(and(lt("globalAvg")), `for`(20L, TsdlTimeUnit.DAYS, IntervalType.CLOSED), `as`("low")),
                event(and(gt("globalAvg")), `for`(20L, TsdlTimeUnit.DAYS, IntervalType.CLOSED), `as`("high"))
            )
            .selection(
                precedes("high", "low")
            )
            .yield(allPeriods())
    }

    println("UC12: Binary Event Sequences\n")
    println(query)

    val (periods, _) = engine.query<TsdlPeriodSet>(
        client,
        queryEndpoint,
        storageName,
        serviceConfiguration,
        lookupConfiguration,
        transformationConfiguration,
        query
    )

    println(
        "\n" + periods.periods().joinToString(separator = "\n") {
            "${it.index()}: ${it.start()}--${it.end()} (${
                ChronoUnit.DAYS.between(
                    it.start(),
                    it.end()
                )
            } days)"
        })
    println("\nlog messages:\n  ${engine.joinLogMessagesToString(periods.logs())}")
    println("-------------------------------")
}

fun `for`(lowerBound: Long, upperBound: Long, unit: TsdlTimeUnit, type: IntervalType) =
    for_(lowerBound, upperBound, unit, type)

fun `for`(lowerBound: Long, unit: TsdlTimeUnit, type: IntervalType) =
    for_(lowerBound, type, unit)