hu.bme.mit.theta.xsts.cli.optiongroup.InputOptions.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of theta-xsts-cli Show documentation

Xsts Cli subproject in the Theta model checking framework

There is a newer version: 6.8.5

/*
 *  Copyright 2024 Budapest University of Technology and Economics
 *
 *  Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 */

package hu.bme.mit.theta.xsts.cli.optiongroup

import com.github.ajalt.clikt.parameters.groups.OptionGroup
import com.github.ajalt.clikt.parameters.options.default
import com.github.ajalt.clikt.parameters.options.option
import com.github.ajalt.clikt.parameters.options.required
import com.github.ajalt.clikt.parameters.types.file
import com.github.ajalt.clikt.parameters.types.inputStream
import hu.bme.mit.theta.frontend.petrinet.model.PetriNet
import hu.bme.mit.theta.frontend.petrinet.pnml.PetriNetParser
import hu.bme.mit.theta.frontend.petrinet.pnml.XMLPnmlToPetrinet
import hu.bme.mit.theta.frontend.petrinet.xsts.PetriNetToXSTS
import hu.bme.mit.theta.xsts.XSTS
import hu.bme.mit.theta.xsts.dsl.XstsDslManager
import java.io.*

class InputOptions : OptionGroup(
    name = "Input options",
    help = "Options related to model and property input"
) {

    val model: File by option(
        help = "Path of the input model (XSTS or Pnml). Extension should be .pnml to be handled as petri-net input"
    ).file(mustExist = true, canBeDir = false).required()
    private val property: InputStream? by option(
        help = "Path of the property file. Has priority over --inlineProperty"
    ).inputStream()
    private val inlineProperty: String? by option(help = "Input property as a string. Ignored if --property is defined")
    private val initialmarking: String by option(help = "Initial marking of the pnml model").default("")

    fun isPnml() = model.path.endsWith("pnml")

    fun loadXsts(): XSTS {
        val propertyStream = if (property != null) property else (if (inlineProperty != null) ByteArrayInputStream(
            "prop { $inlineProperty }".toByteArray()
        ) else null)
        if (isPnml()) {
            val petriNet = XMLPnmlToPetrinet.parse(model.absolutePath, initialmarking)
            return PetriNetToXSTS.createXSTS(petriNet, propertyStream)
        }
        return XstsDslManager.createXsts(
            SequenceInputStream(FileInputStream(model), propertyStream ?: InputStream.nullInputStream())
        )
    }

    fun loadPetriNet(): MutableList = PetriNetParser.loadPnml(model).parsePTNet()
}