FIR: Rework FirRegularTowerDataContexts

- Get rid of SPECIAL mode (just left REGULAR and class-related instead)
- Clear naming
- Restore contexts after lambda/callable reference are processed

The test has been failing before this change because after
callable reference is resolved, its tower data context has been left
erroneously in the SPECIAL-related entry

analysis/low-level-api-fir/tests/org/jetbrains/kotlin/analysis/low/level/api/fir/diagnostic/compiler/based/DiagnosisCompilerTestFE10TestdataTestGenerated.java

@@ -7164,6 +7164,12 @@ public class DiagnosisCompilerTestFE10TestdataTestGenerated extends AbstractDiag
                 runTest("compiler/testData/diagnostics/tests/dataFlow/IsExpression.kt");
             }
 
+            @Test
+            @TestMetadata("smartCastWithLambdaAndCallableReference.kt")
+            public void testSmartCastWithLambdaAndCallableReference() throws Exception {
+                runTest("compiler/testData/diagnostics/tests/dataFlow/smartCastWithLambdaAndCallableReference.kt");
+            }
+
             @Test
             @TestMetadata("WhenSubject.kt")
             public void testWhenSubject() throws Exception {