forked from kscience/visionforge
99 lines
3.0 KiB
Kotlin
99 lines
3.0 KiB
Kotlin
package playground
|
|
|
|
import kotlinx.html.div
|
|
import kotlinx.html.id
|
|
import kotlinx.html.script
|
|
import kotlinx.html.stream.createHTML
|
|
import kotlinx.html.unsafe
|
|
import org.jetbrains.kotlinx.jupyter.api.HTML
|
|
import org.jetbrains.kotlinx.jupyter.api.annotations.JupyterLibrary
|
|
import org.jetbrains.kotlinx.jupyter.api.libraries.*
|
|
import space.kscience.dataforge.context.Context
|
|
import space.kscience.dataforge.misc.DFExperimental
|
|
import space.kscience.gdml.Gdml
|
|
import space.kscience.plotly.Plot
|
|
import space.kscience.visionforge.Vision
|
|
import space.kscience.visionforge.gdml.toVision
|
|
import space.kscience.visionforge.html.HtmlVisionFragment
|
|
import space.kscience.visionforge.html.Page
|
|
import space.kscience.visionforge.html.embedVisionFragment
|
|
import space.kscience.visionforge.plotly.PlotlyPlugin
|
|
import space.kscience.visionforge.plotly.asVision
|
|
import space.kscience.visionforge.solid.Solids
|
|
import space.kscience.visionforge.visionManager
|
|
|
|
@JupyterLibrary
|
|
@DFExperimental
|
|
public class VisionForgePlayGroundForJupyter : JupyterIntegration() {
|
|
|
|
private val context = Context("VisionForge") {
|
|
plugin(Solids)
|
|
plugin(PlotlyPlugin)
|
|
}
|
|
|
|
private val jsBundle = ResourceFallbacksBundle(listOf(
|
|
ResourceLocation("js/visionforge-playground.js", ResourcePathType.CLASSPATH_PATH))
|
|
)
|
|
private val jsResource = LibraryResource(name = "VisionForge", type = ResourceType.JS, bundles = listOf(jsBundle))
|
|
|
|
private var counter = 0
|
|
|
|
private fun produceHtmlVisionString(fragment: HtmlVisionFragment) = createHTML().div {
|
|
val id = "visionforge.vision[${counter++}]"
|
|
div {
|
|
this.id = id
|
|
embedVisionFragment(context.visionManager, fragment = fragment)
|
|
}
|
|
script {
|
|
type = "text/javascript"
|
|
unsafe { +"window.renderAllVisionsById(\"$id\");" }
|
|
}
|
|
}
|
|
|
|
override fun Builder.onLoaded() {
|
|
resource(jsResource)
|
|
|
|
import(
|
|
"space.kscience.gdml.*",
|
|
"space.kscience.plotly.*",
|
|
"space.kscience.plotly.models.*",
|
|
"kotlinx.html.*",
|
|
"space.kscience.visionforge.solid.*",
|
|
"space.kscience.visionforge.html.Page",
|
|
"space.kscience.visionforge.html.page"
|
|
)
|
|
|
|
render<Gdml> { gdmlModel ->
|
|
val fragment = HtmlVisionFragment {
|
|
vision(gdmlModel.toVision())
|
|
}
|
|
HTML(produceHtmlVisionString(fragment))
|
|
}
|
|
|
|
render<Vision> { vision ->
|
|
val fragment = HtmlVisionFragment {
|
|
vision(vision)
|
|
}
|
|
|
|
HTML(produceHtmlVisionString(fragment))
|
|
}
|
|
|
|
render<Plot> { plot ->
|
|
val fragment = HtmlVisionFragment {
|
|
vision(plot.asVision())
|
|
}
|
|
|
|
HTML(produceHtmlVisionString(fragment))
|
|
}
|
|
|
|
render<space.kscience.plotly.PlotlyHtmlFragment> { fragment ->
|
|
HTML(createHTML().apply(fragment.visit).finalize())
|
|
}
|
|
|
|
render<Page> { page ->
|
|
HTML(page.render(createHTML()), true)
|
|
}
|
|
}
|
|
|
|
}
|