irie 0.0.0__py3-none-any.whl

irie/apps/networks/tests.py

@@ -0,0 +1 @@
+from django.test import TestCase

irie/apps/networks/urls.py

@@ -0,0 +1,18 @@
+#===----------------------------------------------------------------------===#
+#
+#         STAIRLab -- STructural Artificial Intelligence Laboratory
+#
+#===----------------------------------------------------------------------===#
+#
+#   Fall 2024, BRACE2 Team
+#
+#   Berkeley, CA
+#
+#----------------------------------------------------------------------------#
+from django.urls import path
+from apps.networks import views
+
+urlpatterns = [
+    path("networks/",     views.network_maps,     name="networks"),
+    path("api/networks/", views.load_network_map, name='load_network_map'),
+]

irie/apps/networks/views.py

@@ -0,0 +1,89 @@
+#===----------------------------------------------------------------------===#
+#
+#         STAIRLab -- STructural Artificial Intelligence Laboratory
+#
+#===----------------------------------------------------------------------===#
+#
+#   Fall 2024, BRACE2 Team
+#   Berkeley, CA
+#
+#----------------------------------------------------------------------------#
+
+import os
+import json
+from django.template import loader
+from django.contrib.auth.decorators import login_required
+from django.http import HttpRequest, HttpResponse
+import logging
+from .models import UserNetworkPreferences
+
+logger = logging.getLogger(__name__)
+
+from .networks import BridgeNetworks_method_a, BridgeNetworks_method_b, BridgeNetworks_method_b_alt
+
+from .forms import CorridorsForm
+
+
+@login_required(login_url="/login/")
+def load_network_map(request: HttpRequest)->HttpResponse:
+    
+    preferences, created = UserNetworkPreferences.objects.get_or_create(user=request.user)
+
+    tab = request.GET.get('tab', None)
+
+    form = CorridorsForm(request.GET or None)
+    #
+    #
+    if not form.is_valid():
+        return HttpResponse(json.dumps({"map_html": '<p>Invalid inputs.</p>'}))
+
+    method = form.cleaned_data["method"]
+    weights = {k: v for k,v in form.cleaned_data.items() if k.endswith("_weight")}
+
+
+    if method == 0:
+        Analysis = BridgeNetworks_method_a
+    elif method == 1:
+        Analysis = BridgeNetworks_method_b
+    elif method == 2:
+        Analysis = BridgeNetworks_method_b_alt
+    else:
+        return HttpResponse(json.dumps({"map_html": '<p>Map unavailable.</p>'}))
+    
+
+    data = {}
+    if tab != "tab1":
+        analysis = Analysis(preferences, weights, False, True)
+        map = analysis.create_map(corridor=form.cleaned_data["corridor_input"])
+
+    else:
+        cp = form.cleaned_data["consider_population"]
+        analysis = Analysis(preferences, weights, cp, False)
+        map = analysis.create_map()
+        data["table_html"] = loader.get_template("networks/corridor_table.html").render({
+            "corridors": (i for _, i in analysis.ranked_corridors().iterrows())
+        })
+
+    data["map_html"] = map._repr_html_()
+    return HttpResponse(json.dumps(data))
+
+
+@login_required(login_url="/login/")
+def network_maps(request: HttpRequest)->HttpResponse:
+
+    context = {}
+    context["segment"] = "networks"
+
+    context["maps"] = {
+        "tab1": {"name": "All", "form": CorridorsForm()},
+    }
+
+    try:
+        html_template = loader.get_template("networks/networks.html")
+        return HttpResponse(html_template.render(context, request))
+
+    except Exception as e:
+        if "DEBUG" in os.environ and os.environ["DEBUG"]:
+            raise e
+        html_template = loader.get_template("site/page-500.html")
+        return HttpResponse(html_template.render(context, request))

irie/apps/prediction/admin.py

@@ -0,0 +1,9 @@
+#===----------------------------------------------------------------------===#
+#
+#         STAIRLab -- STructural Artificial Intelligence Laboratory
+#
+#===----------------------------------------------------------------------===#
+from django.contrib import admin
+from .models import PredictorModel
+
+admin.site.register(PredictorModel)

irie/apps/prediction/apps.py

@@ -0,0 +1,12 @@
+#===----------------------------------------------------------------------===#
+#
+#         STAIRLab -- STructural Artificial Intelligence Laboratory
+#
+#===----------------------------------------------------------------------===#
+from django.apps import AppConfig
+
+class PredictionConfig(AppConfig):
+    default_auto_field = 'django.db.models.BigAutoField'
+    name  = 'apps.prediction'
+    label = 'apps_prediction'
+

irie/apps/prediction/forms.py

@@ -0,0 +1,20 @@
+#===----------------------------------------------------------------------===#
+#
+#         STAIRLab -- STructural Artificial Intelligence Laboratory
+#
+#===----------------------------------------------------------------------===#
+from django import forms
+from apps.prediction.models import PredictorModel
+
+
+# class CsiForm(forms.Form):
+#     option1 = forms.CharField(label='Option 1', max_length=100)
+#     option2 = forms.CharField(label='Option 2', max_length=100)
+#     file_upload = forms.FileField(label='Upload a file', required=True)
+
+
+class PredictorForm(forms.ModelForm):
+    class Meta:
+        model = PredictorModel
+        fields = '__all__'
+        exclude = ['render_file', 'metrics', 'active', 'entry_point', 'config', 'protocol']

irie/apps/prediction/metrics.py

@@ -0,0 +1,61 @@
+#===----------------------------------------------------------------------===#
+#
+#         STAIRLab -- STructural Artificial Intelligence Laboratory
+#
+#===----------------------------------------------------------------------===#
+import orjson
+
+def _serialize(data):
+    return orjson.loads(orjson.dumps(data,
+                        option=orjson.OPT_SERIALIZE_NUMPY))
+
+class HealthMetric:
+    metric_tag: str
+    def __init__(self, predictor: str, data = None):
+        self.predictor = predictor
+        self.data = data
+
+
+class PeriodShiftMetric(HealthMetric):
+    metric_tag = "SPECTRAL_SHIFT_IDENTIFICATION"
+    def buildDetails(self):
+        return self.data
+    def getSummary(self):
+        return self.data
+
+
+
+class ColumnStrainStateMetric01(HealthMetric):
+    metric_tag = "COLUMN_STRAIN_STATES"
+
+    def format_html(self):
+        pass
+
+class PeakAccelMetric01(HealthMetric):
+    metric_tag = "PEAK_ACCEL"
+
+
+class PeakDriftMetric01(HealthMetric):
+    metric_tag = "PEAK_DRIFT"
+
+
+class AccelRHMetric01(HealthMetric):
+    metric_tag = "ACC_RESPONSE_HISTORY"
+
+    def getSummary(self)->dict:
+        return _serialize(self.data)
+
+    def buildDetails(self)->str:
+        return _serialize(self.data)
+
+
+METRIC_CLASSES = {
+    c.metric_tag: c for c in  [
+        PeriodShiftMetric,
+        PeakAccelMetric01,
+        PeakDriftMetric01,
+        AccelRHMetric01,
+        ColumnStrainStateMetric01
+    ]
+}
+

irie/apps/prediction/migrations/0001_initial.py

@@ -0,0 +1,32 @@
+# Generated by Django 3.2.13 on 2023-05-20 20:30
+
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    initial = True
+
+    dependencies = [
+        ('apps_events', '0003_event_cesmd'),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='AssetModel',
+            fields=[
+                ('id', models.BigAutoField(primary_key=True, serialize=False)),
+                ('cesmd', models.CharField(max_length=7)),
+            ],
+        ),
+        migrations.CreateModel(
+            name='AssetEvaluationModel',
+            fields=[
+                ('id', models.BigAutoField(primary_key=True, serialize=False)),
+                ('cesmd', models.CharField(max_length=7)),
+                ('evaluation_data', models.JSONField(default=dict)),
+                ('event', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='apps_events.event')),
+            ],
+        ),
+    ]

irie/apps/prediction/migrations/0002_auto_20230731_1801.py

@@ -0,0 +1,27 @@
+# Generated by Django 3.2.13 on 2023-07-31 18:01
+
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('apps_inventory', '0004_asset'),
+        ('apps_prediction', '0001_initial'),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='PredictorModel',
+            fields=[
+                ('id', models.BigAutoField(primary_key=True, serialize=False)),
+                ('config', models.JSONField(default=dict)),
+                ('name', models.CharField(max_length=35)),
+                ('asset', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='apps_inventory.asset')),
+            ],
+        ),
+        migrations.DeleteModel(
+            name='AssetModel',
+        ),
+    ]

irie/apps/prediction/migrations/0003_rename_assetevaluationmodel_evaluation.py

@@ -0,0 +1,18 @@
+# Generated by Django 3.2.13 on 2023-07-31 18:06
+
+from django.db import migrations
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('apps_events', '0004_event_record_identifier'),
+        ('apps_prediction', '0002_auto_20230731_1801'),
+    ]
+
+    operations = [
+        migrations.RenameModel(
+            old_name='AssetEvaluationModel',
+            new_name='Evaluation',
+        ),
+    ]

irie/apps/prediction/migrations/0004_delete_evaluation.py

@@ -0,0 +1,16 @@
+# Generated by Django 3.2.13 on 2023-08-07 19:20
+
+from django.db import migrations
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('apps_prediction', '0003_rename_assetevaluationmodel_evaluation'),
+    ]
+
+    operations = [
+        migrations.DeleteModel(
+            name='Evaluation',
+        ),
+    ]

irie/apps/prediction/migrations/0005_predictormodel_protocol.py

@@ -0,0 +1,18 @@
+# Generated by Django 3.2.13 on 2023-12-17 22:39
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('apps_prediction', '0004_delete_evaluation'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='predictormodel',
+            name='protocol',
+            field=models.CharField(choices=[('T1', 'Type1'), ('T2', 'Type2')], default='T2', max_length=2),
+        ),
+    ]

irie/apps/prediction/migrations/0006_alter_predictormodel_protocol.py

@@ -0,0 +1,18 @@
+# Generated by Django 3.2.13 on 2023-12-18 00:12
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('apps_prediction', '0005_predictormodel_protocol'),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name='predictormodel',
+            name='protocol',
+            field=models.CharField(choices=[('BRACE2_CLI_PREDICTOR_V1', 'Type1'), ('BRACE2_CLI_PREDICTOR_T2', 'Type2'), ('BRACE2_CLI_PREDICTOR_T3', 'Type3')], default='BRACE2_CLI_PREDICTOR_T2', max_length=25),
+        ),
+    ]

irie/apps/prediction/migrations/0007_predictormodel_active.py

@@ -0,0 +1,19 @@
+# Generated by Django 3.2.13 on 2023-12-18 00:17
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('apps_prediction', '0006_alter_predictormodel_protocol'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='predictormodel',
+            name='active',
+            field=models.BooleanField(default=True),
+            preserve_default=False,
+        ),
+    ]

irie/apps/prediction/migrations/0008_predictormodel_description.py

@@ -0,0 +1,18 @@
+# Generated by Django 3.2.13 on 2023-12-18 00:55
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('apps_prediction', '0007_predictormodel_active'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='predictormodel',
+            name='description',
+            field=models.TextField(default=''),
+        ),
+    ]

irie/apps/prediction/migrations/0009_predictormodel_entry_point.py

@@ -0,0 +1,19 @@
+# Generated by Django 3.2.13 on 2023-12-18 01:04
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('apps_prediction', '0008_predictormodel_description'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='predictormodel',
+            name='entry_point',
+            field=models.CharField(default='', max_length=100),
+            preserve_default=False,
+        ),
+    ]

irie/apps/prediction/migrations/0010_alter_predictormodel_entry_point.py

@@ -0,0 +1,18 @@
+# Generated by Django 3.2.13 on 2023-12-18 01:10
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('apps_prediction', '0009_predictormodel_entry_point'),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name='predictormodel',
+            name='entry_point',
+            field=models.JSONField(default=list),
+        ),
+    ]

irie/apps/prediction/migrations/0011_remove_predictormodel_entry_point.py

@@ -0,0 +1,17 @@
+# Generated by Django 3.2.13 on 2023-12-18 01:11
+
+from django.db import migrations
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('apps_prediction', '0010_alter_predictormodel_entry_point'),
+    ]
+
+    operations = [
+        migrations.RemoveField(
+            model_name='predictormodel',
+            name='entry_point',
+        ),
+    ]

irie/apps/prediction/migrations/0012_predictormodel_entry_point.py

@@ -0,0 +1,18 @@
+# Generated by Django 3.2.13 on 2023-12-18 01:12
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('apps_prediction', '0011_remove_predictormodel_entry_point'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='predictormodel',
+            name='entry_point',
+            field=models.JSONField(default=list),
+        ),
+    ]

irie/apps/prediction/migrations/0013_predictormodel_metrics.py

@@ -0,0 +1,18 @@
+# Generated by Django 3.2.13 on 2023-12-18 01:20
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('apps_prediction', '0012_predictormodel_entry_point'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='predictormodel',
+            name='metrics',
+            field=models.JSONField(default=list),
+        ),
+    ]

irie/apps/prediction/migrations/0014_auto_20240930_0004.py

@@ -0,0 +1,28 @@
+# Generated by Django 3.2.13 on 2024-09-30 00:04
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('apps_prediction', '0013_predictormodel_metrics'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='predictormodel',
+            name='config_file',
+            field=models.FileField(blank=True, null=True, upload_to='predictor_configs/'),
+        ),
+        migrations.AddField(
+            model_name='predictormodel',
+            name='render_file',
+            field=models.FileField(blank=True, null=True, upload_to='predictor_configs/'),
+        ),
+        migrations.AlterField(
+            model_name='predictormodel',
+            name='protocol',
+            field=models.CharField(choices=[('BRACE2_CLI_PREDICTOR_V1', 'Type1'), ('BRACE2_CLI_PREDICTOR_T2', 'Type2'), ('BRACE2_CLI_PREDICTOR_T3', 'Type3'), ('BRACE2_CLI_PREDICTOR_T4', 'Type4')], default='BRACE2_CLI_PREDICTOR_T2', max_length=25),
+        ),
+    ]

irie/apps/prediction/migrations/0015_alter_predictormodel_render_file.py

@@ -0,0 +1,18 @@
+# Generated by Django 5.1 on 2024-11-03 00:27
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('apps_prediction', '0014_auto_20240930_0004'),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name='predictormodel',
+            name='render_file',
+            field=models.FileField(blank=True, null=True, upload_to='renderings/'),
+        ),
+    ]

irie/apps/prediction/models.py

@@ -0,0 +1,37 @@
+#===----------------------------------------------------------------------===#
+#
+#         STAIRLab -- STructural Artificial Intelligence Laboratory
+#
+#===----------------------------------------------------------------------===#
+from django.db import models
+
+from apps.inventory.models import Asset
+
+class PredictorModel(models.Model):
+    # https://docs.djangoproject.com/en/4.2/ref/models/fields/
+    class Protocol(models.TextChoices):
+        TYPE1 = "BRACE2_CLI_PREDICTOR_V1"
+        TYPE2 = "BRACE2_CLI_PREDICTOR_T2"
+        TYPE3 = "BRACE2_CLI_PREDICTOR_T3"
+        TYPE4 = "BRACE2_CLI_PREDICTOR_T4"
+
+    id          = models.BigAutoField(primary_key=True)
+    name        = models.CharField(max_length=35)
+    asset       = models.ForeignKey(Asset, on_delete=models.CASCADE)
+    description = models.TextField(default="")
+
+    protocol    = models.CharField(max_length=25, 
+                                   choices=Protocol.choices, 
+                                   default=Protocol.TYPE2)
+
+    entry_point = models.JSONField(default=list)
+    config      = models.JSONField(default=dict)
+    config_file = models.FileField(upload_to='predictor_configs/', null=True, blank=True)
+    render_file = models.FileField(upload_to='renderings/', null=True, blank=True)
+    metrics     = models.JSONField(default=list)
+
+    active      = models.BooleanField()
+
+    def __str__(self):
+        return f"{self.asset.calid} - {self.name} : {self.description}"
+