[ML4SE-239] added tracking, inspection handler marked to be rewritten.

JetBrains-Research · Dec 4, 2023 · e4aab25 · e4aab25
1 parent 0cfaf9a
commit e4aab25
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 17 deletions.
diff --git a/...lugin/src/main/kotlin/org/jetbrains/research/tasktracker/handler/ide/InspectionHandler.kt b/...lugin/src/main/kotlin/org/jetbrains/research/tasktracker/handler/ide/InspectionHandler.kt
@@ -60,9 +60,10 @@ class InspectionHandler(override val config: InspectionConfig, override val proj
     }
 
     override fun destroy() {
-        initialProfile?.let {
-            ProjectInspectionProfileManager.getInstance(project).setCurrentProfile(it)
-        }
+        // TODO rewrite inspection handler, now it raises an error.
+//        initialProfile?.let {
+//            ProjectInspectionProfileManager.getInstance(project).setCurrentProfile(it)
+//        }
 //        inspectionDisposable?.let { Disposer.dispose(it) }
     }
 

diff --git a/...rc/main/kotlin/org/jetbrains/research/tasktracker/ui/main/panel/MainPluginPanelFactory.kt b/...rc/main/kotlin/org/jetbrains/research/tasktracker/ui/main/panel/MainPluginPanelFactory.kt
@@ -15,15 +15,12 @@ import com.intellij.ui.jcef.JBCefApp
 import com.intellij.util.ui.JBUI
 import kotlinx.serialization.json.Json
 import org.jetbrains.concurrency.Promise
-import org.jetbrains.research.tasktracker.TaskTrackerPlugin
 import org.jetbrains.research.tasktracker.config.content.task.base.Task
-import org.jetbrains.research.tasktracker.modelInference.model.EmoModel
 import org.jetbrains.research.tasktracker.tracking.BaseTracker
 import org.jetbrains.research.tasktracker.tracking.TaskFileHandler
 import org.jetbrains.research.tasktracker.tracking.activity.ActivityTracker
 import org.jetbrains.research.tasktracker.tracking.fileEditor.FileEditorTracker
 import org.jetbrains.research.tasktracker.tracking.toolWindow.ToolWindowTracker
-import org.jetbrains.research.tasktracker.tracking.webcam.WebCamTracker
 import org.jetbrains.research.tasktracker.tracking.webcam.collectAllDevices
 import org.jetbrains.research.tasktracker.ui.main.panel.models.AgreementChecker
 import org.jetbrains.research.tasktracker.ui.main.panel.models.ButtonState
@@ -79,23 +76,22 @@ class MainPluginPanelFactory : ToolWindowFactory {
         if (trackers.isNotEmpty()) { // Otherwise we can lose data
             return
         }
-        TaskTrackerPlugin.mainConfig.emotionConfig?.let {
-            GlobalPluginStorage.emoPredictor = EmoModel(it)
-        } ?: error("emotion config must exist by this moment")
-
         trackers.addAll(
             listOf(
                 ActivityTracker(project),
                 ToolWindowTracker(project),
                 FileEditorTracker(project)
             )
         )
-        GlobalPluginStorage.emoPredictor?.let {
-            trackers.add(WebCamTracker(project, it))
-        }
         trackers.forEach { it.startTracking() }
     }
 
+    fun stopTracking() {
+        trackers.forEach {
+            it.stopTracking()
+        }
+    }
+
     fun loadBasePage(
         template: HtmlTemplate,
         buttonTextKey: String? = null,

diff --git a/...main/kotlin/org/jetbrains/research/tasktracker/ui/main/panel/panelStates/DefaultStates.kt b/...main/kotlin/org/jetbrains/research/tasktracker/ui/main/panel/panelStates/DefaultStates.kt
@@ -9,7 +9,6 @@ import org.jetbrains.research.tasktracker.config.content.task.base.Task
 import org.jetbrains.research.tasktracker.config.content.task.base.TaskWithFiles
 import org.jetbrains.research.tasktracker.config.scenario.models.*
 import org.jetbrains.research.tasktracker.tracking.TaskFileHandler
-import org.jetbrains.research.tasktracker.tracking.activity.ActivityTracker
 import org.jetbrains.research.tasktracker.ui.main.panel.MainPluginPanelFactory
 import org.jetbrains.research.tasktracker.ui.main.panel.runOnSuccess
 import org.jetbrains.research.tasktracker.ui.main.panel.storage.MainPanelStorage
@@ -43,6 +42,7 @@ fun Panel.welcomePage() {
     loadBasePage(MainPageTemplate.loadCurrentTemplate(), "ui.button.next", false)
     setNextAction {
         TaskTrackerPlugin.initializationHandler.setupEnvironment(project)
+        startTracking()
         processScenario()
     }
 }
@@ -65,7 +65,6 @@ private fun Panel.selectTask(taskIds: List<String>, allRequired: Boolean = true)
  * Loads configs by selected task and language
  */
 fun Panel.processTask(id: String): Task {
-    startTracking() // TODO
     val task =
         MainPanelStorage.taskIdTask.values.find { it.id == id } ?: error("Can't find task with id '$id'")
     ApplicationManager.getApplication().invokeAndWait {
@@ -83,8 +82,6 @@ fun Panel.processTask(id: String): Task {
  */
 private fun Panel.solveTask(id: String, nextTasks: List<String> = emptyList()) {
     val task = processTask(id)
-    val activityTracker = ActivityTracker(project)
-    activityTracker.startTracking() // TODO start tracking for all trackers instead of this one
     loadBasePage(SolvePageTemplate(task))
     setNextAction {
         TaskFileHandler.disposeTask(project, task)
@@ -164,6 +161,7 @@ fun Panel.processScenario() {
 
         null -> {
             scenario.reset()
+            stopTracking()
             finalPage()
         }
     }