Ts/1871 task stratify a model workflow

packages/client/hmi-client/src/components/stratification/tera-mira-notebook.vue

@@ -0,0 +1,50 @@
+<template>
+	<h4>Code Editor</h4>
+	<Textarea class="codeBox" v-model="codeAsText" />
+	<Button label="Run" @click="hitMira" />
+</template>
+
+<script setup lang="ts">
+import { ref, onUnmounted } from 'vue';
+import Textarea from 'primevue/textarea';
+import Button from 'primevue/button';
+import { SessionContext } from '@jupyterlab/apputils/lib/sessioncontext';
+import { newSession, JupyterMessage } from '@/services/jupyter';
+import { createMessage } from '@jupyterlab/services/lib/kernel/messages';
+
+const INITIAL_CODE =
+	"from mira.metamodel import ControlledConversion, NaturalConversion, Concept, Template \n infected = Concept(name='infected population', identifiers={'ido': '0000511'}) \n susceptible = Concept(name='susceptible population', identifiers={'ido': '0000514'}) \n immune = Concept(name='immune population', identifiers={'ido': '0000592'}) \n \n t1 = ControlledConversion(controller=infected, subject=susceptible, outcome=infected,) \n t2 = NaturalConversion(subject=infected, outcome=immune) \n t1.dict() \n ";
+
+const codeAsText = ref(INITIAL_CODE);
+const jupyterSession: SessionContext = await newSession('beaker', 'Beaker');
+
+async function hitMira() {
+	if (!jupyterSession) return;
+
+	console.log('Trying to hit mira');
+	console.log(codeAsText);
+	const messageBody = {
+		session: jupyterSession.session?.name || '',
+		channel: 'shell',
+		content: codeAsText as any,
+		msgType: 'context_setup_request',
+		msgId: 'tgpt-context_setup_request'
+	};
+	const message: JupyterMessage = createMessage(messageBody);
+	console.log(jupyterSession.session);
+	jupyterSession.session?.kernel?.sendJupyterMessage(message);
+	console.log(jupyterSession.session);
+}
+
+onUnmounted(() => {
+	if (!jupyterSession) return;
+	jupyterSession.shutdown();
+});
+</script>
+
+<style>
+.codeBox {
+	width: 100%;
+	height: 50%;
+}
+</style>

packages/client/hmi-client/src/components/stratification/tera-stratification-group-form.vue

@@ -0,0 +1,121 @@
+<template>
+	<div class="strata-group" :style="`border-left: 9px solid ${props.config.borderColour}`">
+		<div class="sub-header">
+			<label for="strata-name">Cartesian product</label>
+			<InputSwitch v-model="cartesianProduct" />
+			<i class="trash-button pi pi-trash" @click="emit('delete-self', { index: props.index })" />
+		</div>
+		<div class="first-row">
+			<div class="age-group">
+				<label for="strata-name">Name of strata</label>
+				<InputText
+					id="strata-name"
+					v-model="strataName"
+					placeholder="Age group"
+					@focusout="emit('update-self', { index: props.index, updatedConfig: updatedConfig })"
+				/>
+			</div>
+			<div class="select-variables">
+				<label for="variables-select">Select variables and parameters to stratify</label>
+				<MultiSelect
+					id="variables-select"
+					v-model="selectedVariables"
+					:options="props.modelNodeOptions"
+					placeholder="Model states"
+					display="chip"
+					@update:model-value="
+						emit('update-self', { index: props.index, updatedConfig: updatedConfig })
+					"
+				></MultiSelect>
+			</div>
+		</div>
+		<div class="flex flex-column">
+			<label for="group-labels"
+				>Enter a comma separated list of labels for each group.
+				<span class="subdued-text">(Max 100)</span></label
+			>
+			<InputText
+				id="group-labels"
+				v-model="labels"
+				placeholder="Young, Old"
+				@focusout="emit('update-self', { index: props.index, updatedConfig: updatedConfig })"
+			/>
+		</div>
+	</div>
+</template>
+
+<script setup lang="ts">
+import { ref, computed } from 'vue';
+import InputText from 'primevue/inputtext';
+import MultiSelect from 'primevue/multiselect';
+import InputSwitch from 'primevue/inputswitch';
+import { StratifyGroup } from '@/workflow/ops/stratify-mira/stratify-mira-operation';
+
+const props = defineProps<{
+	modelNodeOptions: string[];
+	config: StratifyGroup;
+	index: number;
+}>();
+
+const emit = defineEmits(['delete-self', 'update-self']);
+
+const strataName = ref(props.config.name);
+const selectedVariables = ref<string[]>(props.config.selectedVariables);
+const labels = ref(props.config.groupLabels);
+const cartesianProduct = ref<boolean>(props.config.cartesianProduct);
+
+const updatedConfig = computed<StratifyGroup>(() => ({
+	borderColour: props.config.borderColour,
+	name: strataName.value,
+	selectedVariables: selectedVariables.value,
+	groupLabels: labels.value,
+	cartesianProduct: cartesianProduct.value
+}));
+</script>
+
+<style>
+.strata-group {
+	margin: 1rem;
+	display: flex;
+	padding: 1rem 1rem 1rem 1.5rem;
+	flex-direction: column;
+	justify-content: center;
+	align-items: flex-start;
+	border-radius: 0.375rem;
+	background: #fff;
+	/* Shadow/medium */
+	box-shadow: 0px 2px 4px -1px rgba(0, 0, 0, 0.06), 0px 4px 6px -1px rgba(0, 0, 0, 0.08);
+}
+
+.sub-header {
+	display: flex;
+	padding-bottom: 0px;
+	justify-content: flex-end;
+	align-items: center;
+	gap: 1rem;
+	align-self: stretch;
+}
+
+.first-row {
+	display: flex;
+}
+.age-group {
+	display: flex;
+	flex-direction: column;
+	padding-right: 2rem;
+	padding-bottom: 1rem;
+}
+
+.select-variables {
+	display: flex;
+	flex-direction: column;
+}
+
+.subdued-text {
+	color: var(--text-color-subdued);
+}
+
+.trash-button {
+	cursor: pointer;
+}
+</style>

packages/client/hmi-client/src/page/WorkflowNode.vue

@@ -1,7 +1,8 @@
 <template>
 	<template v-if="node">
 		<tera-model-workflow-wrapper v-if="isNodeType(OperationType.MODEL)" :node="node" />
-		<tera-stratify-julia v-else-if="isNodeType(OperationType.STRATIFY)" :node="node" />
+		<tera-stratify-julia v-else-if="isNodeType(OperationType.STRATIFY_JULIA)" :node="node" />
+		<tera-stratify-mira v-else-if="isNodeType(OperationType.STRATIFY_MIRA)" :node="node" />
 		<tera-dataset-transformer
 			v-else-if="isNodeType(OperationType.DATASET_TRANSFORMER)"
 			:node="node"
@@ -38,6 +39,7 @@ import TeraCalibrateEnsembleCiemss from '@/workflow/ops/calibrate-ensemble-ciems
 import TeraSimulateCiemss from '@/workflow/ops/simulate-ciemss/tera-simulate-ciemss.vue';
 import TeraSimulateEnsembleCiemss from '@/workflow/ops/simulate-ensemble-ciemss/tera-simulate-ensemble-ciemss.vue';
 import TeraStratifyJulia from '@/workflow/ops/stratify-julia/tera-stratify.vue';
+import teraStratifyMira from '@/workflow/ops/stratify-mira/tera-stratify-mira.vue';
 
 const props = defineProps<{ nodeId: string; workflowId: string }>();
 

packages/client/hmi-client/src/page/project/tera-project.vue

@@ -68,8 +68,14 @@
 					"
 					:node="workflowNode"
 				/>
+				<tera-stratify-mira
+					v-if="workflowNode && workflowNode.operationType === WorkflowOperationTypes.STRATIFY_MIRA"
+					:node="workflowNode"
+				/>
 				<tera-stratify
-					v-if="workflowNode && workflowNode.operationType === WorkflowOperationTypes.STRATIFY"
+					v-if="
+						workflowNode && workflowNode.operationType === WorkflowOperationTypes.STRATIFY_JULIA
+					"
 					:node="workflowNode"
 					:key="workflowNode.id"
 					@open-asset="openAsset"
@@ -123,6 +129,7 @@ import TeraDatasetWorkflowWrapper from '@//workflow/ops/dataset/tera-dataset-wor
 import TeraCalibrateJulia from '@/workflow/ops/calibrate-julia/tera-calibrate-julia.vue';
 import TeraCalibrateCiemss from '@/workflow/ops/calibrate-ciemss/tera-calibrate-ciemss.vue';
 import TeraSimulateJulia from '@/workflow/ops/simulate-julia/tera-simulate-julia.vue';
+import TeraStratifyMira from '@/workflow/ops/stratify-mira/tera-stratify-mira.vue';
 import TeraStratify from '@/workflow/ops/stratify-julia/tera-stratify.vue';
 import TeraSimulateCiemss from '@/workflow/ops/simulate-ciemss/tera-simulate-ciemss.vue';
 import teraSimulateEnsembleCiemss from '@/workflow/ops/simulate-ensemble-ciemss/tera-simulate-ensemble-ciemss.vue';

packages/client/hmi-client/src/types/workflow.ts

@@ -7,7 +7,8 @@ export enum WorkflowOperationTypes {
 	MODEL = 'ModelOperation',
 	SIMULATE_JULIA = 'SimulateJuliaOperation',
 	SIMULATE_CIEMSS = 'SimulateCiemssOperation',
-	STRATIFY = 'Stratify',
+	STRATIFY_JULIA = 'StratifyJulia',
+	STRATIFY_MIRA = 'StratifyMira',
 	SIMULATE_ENSEMBLE_CIEMSS = 'SimulateEnsembleCiemms',
 	CALIBRATE_ENSEMBLE_CIEMSS = 'CalibrateEnsembleCiemms',
 	DATASET_TRANSFORMER = 'DatasetTransformer',

packages/client/hmi-client/src/workflow/ops/stratify-julia/stratify-operation.ts

@@ -1,7 +1,7 @@
 import { Operation, WorkflowOperationTypes } from '@/types/workflow';
 
 export const StratifyOperation: Operation = {
-	name: WorkflowOperationTypes.STRATIFY,
+	name: WorkflowOperationTypes.STRATIFY_JULIA,
 	displayName: 'Stratify',
 	description: 'Stratify a model',
 	inputs: [{ type: 'modelConfigId', label: 'Model configuration', acceptMultiple: false }],

packages/client/hmi-client/src/workflow/ops/stratify-mira/mod.ts

@@ -0,0 +1,3 @@
+export * from './stratify-mira-operation';
+export { default as TeraStratifyMira } from './tera-stratify-mira.vue';
+export { default as TeraStratifyNodeMira } from './tera-stratify-node-mira.vue';

packages/client/hmi-client/src/workflow/ops/stratify-mira/stratify-mira-operation.ts

@@ -0,0 +1,37 @@
+import { Operation, WorkflowOperationTypes } from '@/types/workflow';
+
+export interface StratifyGroup {
+	borderColour: string;
+	name: string;
+	selectedVariables: string[];
+	groupLabels: string;
+	cartesianProduct: boolean;
+}
+
+export interface StratifyOperationStateMira {
+	strataGroups: StratifyGroup[];
+}
+
+export const StratifyMiraOperation: Operation = {
+	name: WorkflowOperationTypes.STRATIFY_MIRA,
+	displayName: 'Stratify MIRA',
+	description: 'Stratify a model',
+	inputs: [{ type: 'modelConfigId', label: 'Model configuration', acceptMultiple: false }],
+	outputs: [{ type: 'model' }],
+	isRunnable: false,
+	action: () => {},
+	initState: () => {
+		const init: StratifyOperationStateMira = {
+			strataGroups: [
+				{
+					borderColour: '#c300a6',
+					name: '',
+					selectedVariables: [],
+					groupLabels: '',
+					cartesianProduct: true
+				}
+			]
+		};
+		return init;
+	}
+};