Issue 2: dummy pdbtmDescriptor theme parameter

src/extensions/tmdet/behavior.ts

@@ -29,7 +29,7 @@ import { applyTransformations, createMembraneOrientation, getAtomGroupExpression
 import { PDBTMDescriptor, PDBTMRegion, PMS } from './types';
 import { registerTmDetSymmetry } from './symmetry';
 import { StripedResidues } from './coloring';
-import { CustomColorThemeProvider } from './tmdet-color-theme';
+import { TmDetColorThemeProvider } from './tmdet-color-theme';
 
 type StructureComponentType = StateObjectSelector<
         PMS,
@@ -58,7 +58,7 @@ export const TMDETMembraneOrientation = PluginBehavior.create<{ autoAttach: bool
             this.ctx.query.structure.registry.add(isTransmembrane);
 
             this.ctx.representation.structure.themes.colorThemeRegistry.add(StripedResidues.colorThemeProvider!);
-            this.ctx.representation.structure.themes.colorThemeRegistry.add(CustomColorThemeProvider);
+            this.ctx.representation.structure.themes.colorThemeRegistry.add(TmDetColorThemeProvider);
             this.ctx.managers.lociLabels.addProvider(StripedResidues.labelProvider!);
             this.ctx.customModelProperties.register(StripedResidues.propertyProvider, true);
 
@@ -158,7 +158,8 @@ export async function customTheme(plugin: PluginUIContext) {
             await plugin.managers.structure.component.updateRepresentationsTheme(
                 s.components,
                 {
-                    color: CustomColorThemeProvider.name as any
+                    color: TmDetColorThemeProvider.name as any,
+                    colorParams: { pdbtmDescriptor: 3 }
                 }
             );
         }

src/extensions/tmdet/tmdet-color-theme.ts

@@ -7,19 +7,19 @@ import { ParamDefinition as PD } from '../../mol-util/param-definition';
 import { Location } from '../../mol-model/location';
 import { PDBTMDescriptor } from './types';
 
-export type ResidueNameColorThemeParams = {
+export type TmDetColorThemeParams = {
     pdbtmDescriptor: any
 };
 
-export function CustomColorTheme(
+export function TmDetColorTheme(
     ctx: ThemeDataContext,
-    props: PD.Values<ResidueNameColorThemeParams>
-): ColorTheme<ResidueNameColorThemeParams> {
+    props: PD.Values<TmDetColorThemeParams>
+): ColorTheme<TmDetColorThemeParams> {
     console.log('ColorTheme CTX', ctx);
     console.log('ColorTheme PROPS', props);
 
     return {
-        factory: CustomColorTheme,
+        factory: TmDetColorTheme,
         granularity: 'group', //'residue' as any,
         color: getColor,
         props: props,
@@ -44,11 +44,11 @@ function getColor(location: Location): Color {
     return color;
 }
 
-export const CustomColorThemeProvider: ColorTheme.Provider<ResidueNameColorThemeParams, 'tmdet-custom-color-theme'> = {
+export const TmDetColorThemeProvider: ColorTheme.Provider<TmDetColorThemeParams, 'tmdet-custom-color-theme'> = {
     name: 'tmdet-custom-color-theme',
     label: 'TMDet Topology Theme',
     category: 'TMDet',
-    factory: CustomColorTheme,
+    factory: TmDetColorTheme,
     getParams: () => ({ pdbtmDescriptor: { isOptional: true } }),
     defaultValues: { pdbtmDescriptor: undefined },
     isApplicable: () => true,