datasets.py

from scipy.spatial.transform import Rotation as R

from esimclasses.anomalies import ContractileRing, Protrusion


class Datasets:
    zygo01 = [
        # *[[f"zygo/01/zygo01{str(n).zfill(2)}", "wild", 1] for n in range(1, 41)],
        ["zygo/01/zygo0101", "wild", 1.1],
        ["zygo/01/zygo0102", "wild", 1.1],  # contractile star pattern
        ["zygo/01/zygo0103", "wild", 1.2],
        ["zygo/01/zygo0104", "wild", 1.2],
        ["zygo/01/zygo0105", "wild", 1.2],
        ["zygo/01/zygo0106", "wild", 1.3],
        ["zygo/01/zygo0107", "wild", 1.3],
        ["zygo/01/zygo0108", "wild", 1.3],
        ["zygo/01/zygo0109", "wild", 1.3],
        ["zygo/01/zygo0110", "wild", 1.4],
        ["zygo/01/zygo0111", "wild", 1.4],
        ["zygo/01/zygo0112", "wild", 1.4],
        ["zygo/01/zygo0113", "wild", 1.4],
        ["zygo/01/zygo0114", "wild", 1.5],
        ["zygo/01/zygo0115", "wild", 1.5],
        ["zygo/01/zygo0116", "wild", 1.5],
        ["zygo/01/zygo0117", "wild", 1.5],
        ["zygo/01/zygo0118", "wild", 1.6],
        ["zygo/01/zygo0119", "wild", 1.6],
        ["zygo/01/zygo0120", "wild", 1.6],
        ["zygo/01/zygo0121", "wild", 1.7],  # round
        ["zygo/01/zygo0122", "wild", 1.8],
        ["zygo/01/zygo0123", "wild", 1.8],
        ["zygo/01/zygo0124", "wild", 1.9],
        ["zygo/01/zygo0125", "wild", 2.0],
        ["zygo/01/zygo0126", "wild", 2.1],
        ["zygo/01/zygo0127", "wild", 2.1],
        ["zygo/01/zygo0128", "wild", 2.2],
        ["zygo/01/zygo0129", "wild", 2.2],
        ["zygo/01/zygo0130", "wild", 2.3],
        ["zygo/01/zygo0131", "wild", 2.3],
        ["zygo/01/zygo0132", "wild", 2.4],
        ["zygo/01/zygo0133", "wild", 2.4],
        ["zygo/01/zygo0134", "wild", 2.5],
        ["zygo/01/zygo0135", "wild", 2.5],
        ["zygo/01/zygo0136", "wild", 2.5],
        ["zygo/01/zygo0137", "wild", 2.6],
        ["zygo/01/zygo0138", "wild", 2.6],
        ["zygo/01/zygo0139", "wild", 2.6],
        ["zygo/01/zygo0140", "wild", 2.7],
    ]

    zygo02 = [
        ["zygo/02/zygo0201", "wild", 2.0],
        ["zygo/02/zygo0202", "wild", 2.0],
        ["zygo/02/zygo0203", "wild", 2.1],
        ["zygo/02/zygo0204", "wild", 2.1],
        ["zygo/02/zygo0205", "wild", 2.1],
        ["zygo/02/zygo0206", "wild", 2.2],
        ["zygo/02/zygo0207", "wild", 2.2],
        ["zygo/02/zygo0208", "wild", 2.2],
        ["zygo/02/zygo0209", "wild", 2.3],
        ["zygo/02/zygo0210", "wild", 2.3],
        ["zygo/02/zygo0211", "wild", 2.4],
        ["zygo/02/zygo0212", "wild", 2.4],
        ["zygo/02/zygo0213", "wild", 2.4],
        ["zygo/02/zygo0214", "wild", 2.5],
        ["zygo/02/zygo0215", "wild", 2.5],
        ["zygo/02/zygo0216", "wild", 2.6],
        ["zygo/02/zygo0217", "wild", 2.6],
        ["zygo/02/zygo0218", "wild", 2.7],
        ["zygo/02/zygo0219", "wild", 2.8],
        ["zygo/02/zygo0220", "wild", 2.9],
    ]

    zygo03 = [
        ["zygo/03/zygo0301", "wild", 2.0],
        ["zygo/03/zygo0302", "wild", 2.1],
        ["zygo/03/zygo0303", "wild", 2.2],
        ["zygo/03/zygo0304", "wild", 2.3],
        ["zygo/03/zygo0305", "wild", 2.4],
        ["zygo/03/zygo0306", "wild", 2.5],
        ["zygo/03/zygo0307", "wild", 2.6],
        ["zygo/03/zygo0308", "wild", 2.6],
        ["zygo/03/zygo0309", "wild", 2.7],
        ["zygo/03/zygo0310", "wild", 2.8],
        ["zygo/03/zygo0311", "wild", 3.0],
        ["zygo/03/zygo0312", "wild", 3.5],
        ["zygo/03/zygo0313", "wild", 4.0],
        ["zygo/03/zygo0314", "wild", 4.1],
        ["zygo/03/zygo0315", "wild", 4.2],
        ["zygo/03/zygo0316", "wild", 4.3],
        ["zygo/03/zygo0317", "wild", 4.4],
        ["zygo/03/zygo0318", "wild", 4.5],
        ["zygo/03/zygo0319", "wild", 4.6],
        ["zygo/03/zygo0320", "wild", 4.7],
        ["zygo/03/zygo0321", "wild", 4.9],
        ["zygo/03/zygo0322", "wild", 6.0],
        ["zygo/03/zygo0323", "wild", 6.5],
        ["zygo/03/zygo0324", "wild", 6.9],
        ["zygo/03/zygo0325", "wild", 7.0],
        ["zygo/03/zygo0326", "wild", 7.4],
        ["zygo/03/zygo0327", "wild", 8.0],
        ["zygo/03/zygo0328", "wild", 8.2],
        ["zygo/03/zygo0329", "wild", 8.3],
        ["zygo/03/zygo0330", "wild", 8.4],
        ["zygo/03/zygo0331", "wild", 8.5],
    ]

    old2to4 = [
        ["2to4/2to4_0102", "wild", 2.4],  # AB  round
        ["2to4/2to4_0103", "wild", 2.7],
        ["2to4/2to4_0104", "wild", 3.0],
        ["2to4/2to4_0105", "wild", 3.5],
        ["2to4/2to4_0106", "wild", 4.0],
        ["2to4/2to4_0202", "wild", 2.5],
        ["2to4/2to4_0203", "wild", 2.8],
        ["2to4/2to4_0204", "wild", 3.5],
        ["2to4/2to4_0205", "wild", 4.0],
        ["2to4/2to4_0206", "wild", 4.2],  # Looks pretty bad
    ]

    old7cellseries = [
        ["7cellseries/7cell0102", "wild", 7.2],  # ABpr abit random, ABpl ventral > anterior
        ["7cellseries/7cell0103", "wild", 7.6],  # all good, ABpl ventral > anterior
        ["7cellseries/7cell0104", "wild", 8.0],  # protr looks good, 0105 omitted cause crashing
        ["7cellseries/7cell0106", "wild", 8.5],

        ["7cellseries/7cell0202", "wild", 7.4],  # all ok, ABpl ventral > anterior  (!! rotation seems over, P2 behind)
        ["7cellseries/7cell0203", "wild", 8.0],  # pushing against wall
        ["7cellseries/7cell0204", "wild", 8.4],
        ["7cellseries/7cell0205", "wild", 8.7],

        ["7cellseries/7cell0302", "wild", 7.1],  # all kinda bad, sample feels passive
        ["7cellseries/7cell0303", "wild", 7.5],  # ABpl ventral bad

        ["7cellseries/7cell0402", "wild", 7.4],  # all perfect, 3x ABpl
        ["7cellseries/7cell0403", "wild", 7.8],  # all good, 3x ABpl
        ["7cellseries/7cell0405", "wild", 8.5],
        ["7cellseries/7cell0406", "wild", 8.7],

        ["7cellseries/7cell0502", "wild", 7.4],  # protrusions good, ring angle bad
        ["7cellseries/7cell0503", "wild", 7.8],  # protrusions weak, rotation happened
        ["7cellseries/7cell0504", "wild", 8.2],
        ["7cellseries/7cell0505", "wild", 8.5],

        ["7cellseries/7cell0602", "wild", 7.3],  # all good no ABpl anterior protrusion
        ["7cellseries/7cell0603", "wild", 7.6],  # all good rotation happened
        ["7cellseries/7cell0604", "wild", 8.1],
        ["7cellseries/7cell0605", "wild", 8.4],
        ["7cellseries/7cell0606", "wild", 8.6],

        ["7cellseries/7cell0702", "wild", 7.2],  # all good except ABpr small & wrong
        ["7cellseries/7cell0703", "wild", 7.6],  # looks good but rotation happened and ABpl weak forces
        ["7cellseries/7cell0704", "wild", 8.2],
    ]

    olddshmig = [  # DSH mutation (needs pics redo)
        ["dshmig/dshmig0137", "dsh", 7.1],  # ABpl protrusion good, sim good
        ["dshmig/dshmig0138", "dsh", 7.4],  # E MS contact small, sim ring bad but np
        ["dshmig/dshmig0139", "dsh", 8.0],  # sim good
        # ["dshmig/dshmig0140", "dsh", 8.3],  # pics failure

        ["dshmig/dshmig0209", "dsh", 7.0],  # sim anos eh but small
        ["dshmig/dshmig0210", "dsh", 7.2],  # Nice protrusions, sim excellent
        ["dshmig/dshmig0211", "dsh", 7.7],  # sim ok, ring could be better
        ["dshmig/dshmig0212", "dsh", 8.1],  # sim ok
        ["dshmig/dshmig0213", "dsh", 8.6],  # Rotated & C shifting, sim ok

        ["dshmig/dshmig0310", "dsh", 7.1],  # sim good
        ["dshmig/dshmig0311", "dsh", 7.4],  # Elongated, sim excellent
        ["dshmig/dshmig0312", "dsh", 8.0],  # sim ok, still found ABpl (313 crash, 314 overlap)

        ["dshmig/dshmig0425", "dsh", 7.0],  # ventral protrusion bad, dorsal good
        ["dshmig/dshmig0426", "dsh", 7.2],  # Nice protrusion, ring bad but non present
        ["dshmig/dshmig0427", "dsh", 7.4],  # Strong anos
        ["dshmig/dshmig0428", "dsh", 8.2],

        ["dshmig/dshmig0511", "dsh", 7.0],  # ok
        ["dshmig/dshmig0512", "dsh", 7.2],  # ok
        ["dshmig/dshmig0513", "dsh", 7.6],  # Good ring
        ["dshmig/dshmig0514", "dsh", 8.4],  #
        ["dshmig/dshmig0515", "dsh", 8.6],  # light pics failure here too
    ]

    wt02 = [
        ["wt/02/wt0201", "wild", 2.4],  # round
        ["wt/02/wt0202", "wild", 2.5],
        ["wt/02/wt0203", "wild", 2.6],
        ["wt/02/wt0204", "wild", 2.8],  # elongated, ring
        ["wt/02/wt0205", "wild", 3.0],
        ["wt/02/wt0206", "wild", 4.0],
        ["wt/02/wt0207", "wild", 4.1],
        ["wt/02/wt0208", "wild", 4.2],
        ["wt/02/wt0209", "wild", 4.2],
        ["wt/02/wt0210", "wild", 4.3],
        ["wt/02/wt0211", "wild", 4.4],
        ["wt/02/wt0212", "wild", 4.4],
        ["wt/02/wt0213", "wild", 4.5],  # E is already pretty long
        ["wt/02/wt0214", "wild", 4.6],
        ["wt/02/wt0215", "wild", 4.6],
        ["wt/02/wt0216", "wild", 4.7],
        ["wt/02/wt0217", "wild", 4.8],  # Both AB descendants show ring & shift
        ["wt/02/wt0218", "wild", 4.9],
        ["wt/02/wt0219", "wild", 6.0],
        ["wt/02/wt0220", "wild", 6.7],  # EMS ring
        ["wt/02/wt0221", "wild", 7.0],  # ABpl protrusion poor
        ["wt/02/wt0222", "wild", 7.4],
        ["wt/02/wt0223", "wild", 7.8],  # P2 concave
        ["wt/02/wt0224", "wild", 8.0],
        ["wt/02/wt0225", "wild", 8.1],
        ["wt/02/wt0226", "wild", 8.2],
        ["wt/02/wt0227", "wild", 8.3],
        ["wt/02/wt0228", "wild", 8.4],
        ["wt/02/wt0229", "wild", 8.5],
        ["wt/02/wt0230", "wild", 8.6],  # C shifted far to side
    ]

    wt07 = [
        ["wt/07/wt0701", "wild", 2.4],
        ["wt/07/wt0702", "wild", 2.5],
        ["wt/07/wt0703", "wild", 2.6],
        ["wt/07/wt0704", "wild", 2.7],
        ["wt/07/wt0705", "wild", 2.9],
        ["wt/07/wt0706", "wild", 3.1],
        ["wt/07/wt0707", "wild", 4.0],
        ["wt/07/wt0708", "wild", 4.1],
        ["wt/07/wt0709", "wild", 4.2],
        ["wt/07/wt0710", "wild", 4.3],
        ["wt/07/wt0711", "wild", 4.3],
        ["wt/07/wt0712", "wild", 4.4],
        ["wt/07/wt0713", "wild", 4.5],
        ["wt/07/wt0714", "wild", 4.6],
        ["wt/07/wt0715", "wild", 4.6],
        ["wt/07/wt0716", "wild", 4.7],
        ["wt/07/wt0717", "wild", 4.9],
        ["wt/07/wt0718", "wild", 5.0],  # ABp divided, ABa about to  !! pics messed up, need to redo for names
        ["wt/07/wt0719", "wild", 6.0],
        ["wt/07/wt0720", "wild", 7.0],
        ["wt/07/wt0721", "wild", 7.3],  # P2 round
        ["wt/07/wt0722", "wild", 7.6],
        ["wt/07/wt0723", "wild", 8.0],
        ["wt/07/wt0724", "wild", 8.1],
        ["wt/07/wt0725", "wild", 8.2],
        ["wt/07/wt0726", "wild", 8.3],  # 8c rotation
        ["wt/07/wt0727", "wild", 8.4],
    ]

    wt08 = [
        ["wt/08/wt0801", "wild", 2.5],  # round
        ["wt/08/wt0802", "wild", 2.6],
        ["wt/08/wt0803", "wild", 2.7],
        ["wt/08/wt0804", "wild", 2.8],
        ["wt/08/wt0805", "wild", 3.0],
        ["wt/08/wt0806", "wild", 4.0],
        ["wt/08/wt0807", "wild", 4.1],
        ["wt/08/wt0808", "wild", 4.2],
        ["wt/08/wt0809", "wild", 4.2],
        ["wt/08/wt0810", "wild", 4.3],
        ["wt/08/wt0811", "wild", 4.4],
        ["wt/08/wt0812", "wild", 4.4],
        ["wt/08/wt0813", "wild", 4.5],
        ["wt/08/wt0814", "wild", 4.6],
        ["wt/08/wt0815", "wild", 4.6],  # round
        ["wt/08/wt0816", "wild", 4.7],
        ["wt/08/wt0817", "wild", 4.8],
        ["wt/08/wt0818", "wild", 4.8],
        ["wt/08/wt0819", "wild", 4.9],
        ["wt/08/wt0820", "wild", 6.0],
        ["wt/08/wt0821", "wild", 6.4],
        ["wt/08/wt0822", "wild", 6.8],  # concave
        ["wt/08/wt0823", "wild", 7.0],
        ["wt/08/wt0824", "wild", 7.2],  # round
        ["wt/08/wt0825", "wild", 7.5],  # elongated
        ["wt/08/wt0826", "wild", 8.0],
        ["wt/08/wt0827", "wild", 8.1],
        ["wt/08/wt0828", "wild", 8.2],
        ["wt/08/wt0829", "wild", 8.3],
        ["wt/08/wt0830", "wild", 8.4],
        ["wt/08/wt0831", "wild", 8.5],
        ["wt/08/wt0832", "wild", 8.6],
        ["wt/08/wt0833", "wild", 8.7],  # rings for most ABxx
    ]

    wt08b = [  # pics should be good, still needs xu stuff
        ["wt/08/wt0834", "wild", 0],
        ["wt/08/wt0835", "wild", 0],
        ["wt/08/wt0836", "wild", 0],
        ["wt/08/wt0837", "wild", 0],
        ["wt/08/wt0838", "wild", 0],
        ["wt/08/wt0839", "wild", 0],
    ]

    # wt 10-13 added to folder but need names

    crm = [
        *[[f"crm/04/crm04{str(n).zfill(2)}", "gastru", 0] for n in range(17, 50) if n != 21],
        *[[f"crm/05/crm05{str(n).zfill(2)}", "gastru", 0] for n in range(25, 58) if n != 27],
        *[[f"crm/06/crm06{str(n).zfill(2)}", "gastru", 0] for n in range(25, 58) if n not in [29, 40]],
        *[[f"crm/07/crm07{str(n).zfill(2)}", "gastru", 0] for n in range(33, 66) if n not in [61, 65]],
        *[[f"crm/08/crm08{str(n).zfill(2)}", "gastru", 0] for n in range(20, 53) if n not in [46, 50]],
        *[[f"crm/09/crm09{str(n).zfill(2)}", "gastru", 0] for n in range(20, 53) if n not in [41, 50]],
        *[[f"crm/10/crm10{str(n).zfill(2)}", "gastru", 0] for n in range(22, 55) if n != 23],
        *[[f"crm/11/crm11{str(n).zfill(2)}", "gastru", 0] for n in range(24, 57) if n != 41],
        *[[f"crm/12/crm12{str(n).zfill(2)}", "gastru", 0] for n in range(22, 55) if n != 22],
        *[[f"crm/13/crm13{str(n).zfill(2)}", "gastru", 0] for n in range(18, 51) if n != 39],
        *[[f"crm/14/crm14{str(n).zfill(2)}", "gastru", 0] for n in range(20, 53)],
        *[[f"crm/15/crm15{str(n).zfill(2)}", "gastru", 0] for n in range(32, 65)],
        *[[f"crm/16/crm16{str(n).zfill(2)}", "gastru", 0] for n in range(23, 56) if n != 39],
        *[[f"crm/17/crm17{str(n).zfill(2)}", "gastru", 0] for n in range(23, 61)if n != 52],
        *[[f"crm/18/crm18{str(n).zfill(2)}", "gastru", 0] for n in range(21, 54)],
        *[[f"crm/19/crm19{str(n).zfill(2)}", "gastru", 0] for n in range(20, 53)],
        *[[f"crm/20/crm20{str(n).zfill(2)}", "gastru", 0] for n in range(25, 58)],
    ]

    crmb = [
        *[[f"crmb/04/crmb04{str(n).zfill(2)}", "gastru", 0] for n in range(1, 17)],
        *[[f"crmb/05/crmb05{str(n).zfill(2)}", "gastru", 0] for n in range(1, 25)],
        *[[f"crmb/06/crmb06{str(n).zfill(2)}", "gastru", 0] for n in range(1, 25)],
        *[[f"crmb/07/crmb07{str(n).zfill(2)}", "gastru", 0] for n in range(1, 33)],
        *[[f"crmb/08/crmb08{str(n).zfill(2)}", "gastru", 0] for n in range(1, 20)],
        *[[f"crmb/09/crmb09{str(n).zfill(2)}", "gastru", 0] for n in range(1, 20)],
        *[[f"crmb/10/crmb10{str(n).zfill(2)}", "gastru", 0] for n in range(1, 22)],
        *[[f"crmb/11/crmb11{str(n).zfill(2)}", "gastru", 0] for n in range(1, 24) if n != 16],
        *[[f"crmb/12/crmb12{str(n).zfill(2)}", "gastru", 0] for n in range(1, 22)],
        *[[f"crmb/13/crmb13{str(n).zfill(2)}", "gastru", 0] for n in range(1, 18)],
        *[[f"crmb/14/crmb14{str(n).zfill(2)}", "gastru", 0] for n in range(1, 20)],
        *[[f"crmb/15/crmb15{str(n).zfill(2)}", "gastru", 0] for n in range(1, 32)],
        *[[f"crmb/16/crmb16{str(n).zfill(2)}", "gastru", 0] for n in range(1, 23)],
        *[[f"crmb/17/crmb17{str(n).zfill(2)}", "gastru", 0] for n in range(1, 23)],
        *[[f"crmb/18/crmb18{str(n).zfill(2)}", "gastru", 0] for n in range(1, 21)],
        *[[f"crmb/19/crmb19{str(n).zfill(2)}", "gastru", 0] for n in range(1, 20)],
        *[[f"crmb/20/crmb20{str(n).zfill(2)}", "gastru", 0] for n in range(1, 20)],  # missing some frames ?
    ]


def default_anomalies(stage):
    ano = []
    if 1.8 <= stage < 2.0:
        ano = [ContractileRing("P0", [0, 1, 0], cut=3e-6)]  # Zygote ring
    elif 2.5 <= stage < 2.7:
        ano = [ContractileRing("AB", [-1, .2, 0])]  # Early AB ring
    elif 2.7 <= stage < 2.9:
        ano = [ContractileRing("AB", [-1, .5, 0])]  # Middle AB ring, rotating
    elif 2.9 <= stage < 3:
        ano = [ContractileRing("AB", [-1, .9, 0])]  # Late AB ring, rotated
    elif 3 <= stage < 4:
        ano = [ContractileRing("P1", [-.2, 1, 0])]  # P1 ring
    elif 4.7 <= stage < 6:
        ano = [ContractileRing("ABa", [-.5, .2, .8]), ContractileRing("ABp", [-.5, .2, .8])]  # ABa and ABp ring
    elif 6.7 <= stage < 7:
        ano = [ContractileRing("EMS", [0, 1, 0])]  # EMS ring
    elif 7 <= stage < 8:
        ano = [ContractileRing("P2", [1, .2, .2]),  # P2 Division
               Protrusion("ABpl", [1, 0, 0]),  # ABpl ventral protrusion -> anteroventral migration
               Protrusion("ABpl", [-.3, -1, 0]),  # ABpl anterior (real?)
               Protrusion("ABpl", [-1, -.2, 0]),  # ABpl dorsal lamellipodium
               Protrusion("ABpr", [1, -.3, 0])]  # ABpr ventral protrusion
    elif 8 <= stage < 8.4:
        ano = [Protrusion("ABpl", [1, 0, 0])]  # ABpl ventral]
    return ano


def align_data(xs, tris, loc):  # Rotate points in numpy array depending on data source
    # Alignment to make x,y,z mean ventral,posterior,right
    if "zygo01" in loc:
        xs = R.from_euler('z', 100, degrees=True).apply(xs)
        xs = R.from_euler('y', 180, degrees=True).apply(xs)
    elif "zygo02" in loc:
        xs = R.from_euler('z', 15, degrees=True).apply(xs)
        xs = R.from_euler('x', 180, degrees=True).apply(xs)
    elif "zygo03" in loc:
        xs = R.from_euler('z', 30, degrees=True).apply(xs)
        xs = R.from_euler('y', 180, degrees=True).apply(xs)
    elif "zygo06" in loc:
        xs = R.from_euler('z', 100, degrees=True).apply(xs)
    elif "zygo07" in loc:
        xs = R.from_euler('z', -120, degrees=True).apply(xs)
    elif "2to4_01" in loc:
        xs = R.from_euler('z', 50, degrees=True).apply(xs)
    elif "2to4_02" in loc:
        xs = R.from_euler('z', 180, degrees=True).apply(xs)
    elif "7cell010" in loc:
        xs = R.from_euler('z', 10, degrees=True).apply(xs)
    elif "7cell020" in loc:
        xs = R.from_euler('y', 180, degrees=True).apply(xs)
    elif "7cell030" in loc:
        xs = R.from_euler('z', 62, degrees=True).apply(xs)
    elif "7cell040" in loc:
        xs = R.from_euler('z', 144, degrees=True).apply(xs)
        xs = R.from_euler('y', 180, degrees=True).apply(xs)
    elif "7cell050" in loc:
        xs = R.from_euler('z', -25, degrees=True).apply(xs)
    elif "7cell060" in loc:
        xs = R.from_euler('z', -155, degrees=True).apply(xs)
        xs = R.from_euler('y', 180, degrees=True).apply(xs)
    elif "7cell070" in loc:
        xs = R.from_euler('z', -28, degrees=True).apply(xs)
        xs = R.from_euler('y', 180, degrees=True).apply(xs)
    elif "dshmig01" in loc:
        xs = R.from_euler('z', 175, degrees=True).apply(xs)
    elif "dshmig02" in loc:
        xs = R.from_euler('z', -110, degrees=True).apply(xs)
    elif "dshmig03" in loc:
        xs = R.from_euler('z', 95, degrees=True).apply(xs)
    elif "dshmig04" in loc:
        xs = R.from_euler('z', 115, degrees=True).apply(xs)
        xs = R.from_euler('y', 180, degrees=True).apply(xs)
    elif "dshmig05" in loc:
        xs = R.from_euler('z', 100, degrees=True).apply(xs)
    elif "wt02" in loc:
        xs = R.from_euler('z', 165, degrees=True).apply(xs)
    elif "wt07" in loc:
        xs = R.from_euler('z', 80, degrees=True).apply(xs)
        xs = R.from_euler('y', 180, degrees=True).apply(xs)
        xs[:, 2] = -xs[:, 2]  # Flip z messes with normals
        tris = [line[::-1] for line in tris]
    elif "wt08" in loc:
        xs = R.from_euler('z', -60, degrees=True).apply(xs)
    return xs, tris