wip
This commit is contained in:
parent
afd78c3b3e
commit
29d20b2273
25 changed files with 369 additions and 469 deletions
|
|
@ -1,16 +1,5 @@
|
|||
import { lsType } from "../structures/list_common.js";
|
||||
import { fnType } from "../metacircular.js";
|
||||
import { deepEqual, DefaultMap, pretty } from "../util.js";
|
||||
import { numDictType } from "../typeclasses/num_type.js";
|
||||
|
||||
const genericTypeRegistry = new DefaultMap(underlyingType => ({ generic: underlyingType }));
|
||||
|
||||
// type constructor for generic kinds,
|
||||
// for instance:
|
||||
// a -> a -> Bool
|
||||
// is typed by
|
||||
// genericType(Function)
|
||||
export const genericType = underlyingType => genericTypeRegistry.getdefault(underlyingType, true);
|
||||
import { eqType } from "../metacircular.js";
|
||||
import { pretty, zip } from "../util.js";
|
||||
|
||||
// constructor for generic types
|
||||
// for instance, the type:
|
||||
|
|
@ -19,14 +8,13 @@ export const genericType = underlyingType => genericTypeRegistry.getdefault(unde
|
|||
// makeGeneric(a => fnType({in: a, out: fnType({in: a, out: Bool})}))
|
||||
export const makeGeneric = callback => {
|
||||
// type variables to make available:
|
||||
const a = Symbol('a');
|
||||
const b = Symbol('b');
|
||||
const c = Symbol('c');
|
||||
const d = Symbol('d');
|
||||
const e = Symbol('e');
|
||||
const type = callback(a, b, c, d, e);
|
||||
const typeVars = ['a', 'b', 'c', 'd', 'e'].map(letter => ({
|
||||
symbol: Symbol(letter),
|
||||
params: [],
|
||||
}));
|
||||
const type = callback(...typeVars);
|
||||
return {
|
||||
typeVars: occurring(type, new Set([a, b, c, d, e])),
|
||||
typeVars: occurring(type, new Set(typeVars)),
|
||||
type,
|
||||
};
|
||||
};
|
||||
|
|
@ -34,60 +22,10 @@ export const makeGeneric = callback => {
|
|||
// From the given set of type variables, return only those that occur in the given type.
|
||||
export const occurring = (type, typeVars) => {
|
||||
if (typeVars.has(type)) {
|
||||
// type IS a type variable:
|
||||
return new Set([type]);
|
||||
}
|
||||
if (type.in !== undefined) {
|
||||
// function type
|
||||
return new Set([
|
||||
...occurring(type.in, typeVars),
|
||||
...occurring(type.out, typeVars)]);
|
||||
}
|
||||
if (type.listOf !== undefined) {
|
||||
return occurring(type.listOf, typeVars);
|
||||
}
|
||||
return new Set();
|
||||
}
|
||||
|
||||
export const properUnify = eqDict => (
|
||||
{typeVars: formalTypeVars, type: formalType},
|
||||
{typeVars: actualTypeVars, type: actualType},
|
||||
) => {
|
||||
if (getEq(eqDict)(formalType)(actualType)) {
|
||||
return {
|
||||
substitutions: new Map(),
|
||||
typeVars: new Set([
|
||||
...actualTypeVars,
|
||||
// ...formalTypeVars, // <- i don't think we need these?
|
||||
]),
|
||||
type: actualType,
|
||||
}
|
||||
}
|
||||
|
||||
if (formalTypeVars.has(formalType)) {
|
||||
// formalType is type variable -> substitute it by actualType
|
||||
return {
|
||||
substitutions: new Map([[formalType, actualType]]),
|
||||
typeVars: new Set([
|
||||
...actualTypeVars,
|
||||
...formalTypeVars,
|
||||
].filter(a => a !== formalType)),
|
||||
type: actualType,
|
||||
}
|
||||
}
|
||||
if (actualTypeVars.has(actualType)) {
|
||||
// same as above, but in opposite direction:
|
||||
// actualType is type variable -> substitute it by formalType
|
||||
return {
|
||||
substitutions: new Map([[actualType, formalType]]),
|
||||
typeVars: new Set([
|
||||
...actualTypeVars,
|
||||
...formalTypeVars,
|
||||
].filter(a => a !== actualType)),
|
||||
type: formalType,
|
||||
}
|
||||
}
|
||||
|
||||
// WIP...
|
||||
return new Set(type.params.flatMap(p => [...occurring(p, typeVars)]));
|
||||
}
|
||||
|
||||
const mergeOneWay = (m1, m2) => {
|
||||
|
|
@ -103,7 +41,17 @@ const mergeOneWay = (m1, m2) => {
|
|||
return [true, m1copy, m2copy, new Set()]; // stable
|
||||
}
|
||||
|
||||
export const mergeSubstitutions = (m1, m2) => {
|
||||
export const mergeTwoWay = (m1, m2) => {
|
||||
// check for conflicts:
|
||||
for (const [typeVar, actual] of m1) {
|
||||
if (m2.has(typeVar)) {
|
||||
const other = m2.get(typeVar);
|
||||
if (!eqType(actual, other)) {
|
||||
throw new Error(`conflicting substitution: ${pretty(actual)}vs. ${pretty(other)}`);
|
||||
}
|
||||
}
|
||||
}
|
||||
// actually merge
|
||||
let stable = false;
|
||||
let deletedTypeVars = new Set();
|
||||
while (!stable) {
|
||||
|
|
@ -115,150 +63,80 @@ export const mergeSubstitutions = (m1, m2) => {
|
|||
return [new Map([...m1, ...m2]), deletedTypeVars];
|
||||
}
|
||||
|
||||
// Currently very ad-hoc.
|
||||
|
||||
// Thanks to Hans for pointing out that this algorithm exactly like "Unification" in Prolog (hence the function name):
|
||||
// https://www.dai.ed.ac.uk/groups/ssp/bookpages/quickprolog/node12.html
|
||||
export const unify = (
|
||||
{typeVars: formalTypeVars, type: formalType},
|
||||
{typeVars: actualTypeVars, type: actualType},
|
||||
) => {
|
||||
if (deepEqual(formalType, actualType)) {
|
||||
return {
|
||||
substitutions: new Map(),
|
||||
typeVars: new Set([
|
||||
...actualTypeVars,
|
||||
...formalTypeVars]),
|
||||
type: actualType,
|
||||
}
|
||||
}
|
||||
|
||||
// console.log("unify", {formalTypeVars, formalType, actualTypeVars, actualType});
|
||||
if (formalTypeVars.has(formalType)) {
|
||||
// simplest case: substitute formal type param
|
||||
// simplest case: formalType is a type paramater
|
||||
// => substitute with actualType
|
||||
return {
|
||||
substitutions: new Map([[formalType, actualType]]),
|
||||
typeVars: new Set([
|
||||
...actualTypeVars,
|
||||
...formalTypeVars].filter(a => a !== formalType)),
|
||||
...[...formalTypeVars].filter(a => a !== formalType),
|
||||
]),
|
||||
type: actualType,
|
||||
};
|
||||
}
|
||||
|
||||
if (actualTypeVars.has(actualType)) {
|
||||
// same as above, but in the other direction
|
||||
return {
|
||||
substitutions: new Map([[actualType, formalType]]),
|
||||
typeVars: new Set([
|
||||
...actualTypeVars,
|
||||
...formalTypeVars].filter(a => a !== actualType)),
|
||||
...[...actualTypeVars].filter(a => a !== actualType),
|
||||
...formalTypeVars,
|
||||
]),
|
||||
type: formalType,
|
||||
};
|
||||
}
|
||||
|
||||
if (formalType.in !== undefined) {
|
||||
// function type
|
||||
if (actualType.in === undefined) {
|
||||
throw new Error(`cannot assign ${pretty(actualType)} to ${pretty(formalType)}`);
|
||||
}
|
||||
else {
|
||||
// both are function type
|
||||
const inType = unify({typeVars: formalTypeVars, type: formalType.in}, {typeVars: actualTypeVars, type: actualType.in});
|
||||
const outType = unify({typeVars: formalTypeVars, type: formalType.out}, {typeVars: actualTypeVars, type: actualType.out});
|
||||
// check for conflicts between 'in' and 'out' subsitutions
|
||||
for (const [typeVar, actual] of inType.substitutions) {
|
||||
if (outType.substitutions.has(typeVar)) {
|
||||
if (!deepEqual(actual, outType.substitutions.get(typeVar))) {
|
||||
throw new Error(`conflicting assignment for ${pretty(typeVar)}: ${pretty(a)}`);
|
||||
}
|
||||
}
|
||||
}
|
||||
// merge substitutions
|
||||
const [newSubstitutions, deletedTypeVars] = mergeSubstitutions(
|
||||
inType.substitutions, outType.substitutions);
|
||||
// const newSubstitutions = new Map([
|
||||
// ...inType.substitutions,
|
||||
// ...outType.substitutions,
|
||||
// ]);
|
||||
const newTypeVars = new Set([
|
||||
...actualTypeVars,
|
||||
...formalTypeVars].filter(a => !newSubstitutions.has(a) && !deletedTypeVars.has(a)));
|
||||
return {
|
||||
substitutions: newSubstitutions,
|
||||
typeVars: newTypeVars,
|
||||
type: fnType({in: inType.type, out: outType.type}),
|
||||
};
|
||||
}
|
||||
// recursively unify
|
||||
if (formalType.symbol !== actualType.symbol) {
|
||||
throw new Error(`cannot unify ${pretty(formalType.symbol)} and ${pretty(actualType.symbol)}`);
|
||||
}
|
||||
|
||||
if (formalType.listOf !== undefined) {
|
||||
// list type
|
||||
if (actualType.listOf === undefined) {
|
||||
throw new Error(`cannot assign ${pretty(actualType)} to ${pretty(formalType)}`);
|
||||
}
|
||||
else {
|
||||
// both are list type
|
||||
const elementType = unify(
|
||||
{typeVars: formalTypeVars, type: formalType.listOf},
|
||||
{typeVars: actualTypeVars, type: actualType.listOf});
|
||||
return {
|
||||
substitutions: elementType.substitutions,
|
||||
typeVars: new Set([
|
||||
...actualTypeVars,
|
||||
...formalTypeVars].filter(a => !elementType.substitutions.has(a))),
|
||||
type: lsType(elementType.type),
|
||||
};
|
||||
}
|
||||
else {
|
||||
const unifiedParams = zip(formalType.params, actualType.params).map(([formalParam, actualParam]) => unify({typeVars: formalTypeVars, type: formalParam}, {typeVars: actualTypeVars, type: actualParam}));
|
||||
const [unifiedSubstitusions, deleted] = unifiedParams.reduce(([substitutionsSoFar, deletedSoFar], cur) => {
|
||||
// console.log('merging', substitutionsSoFar, cur.substitutions);
|
||||
const [newSubstitutions, deleted] = mergeTwoWay(substitutionsSoFar, cur.substitutions);
|
||||
return [newSubstitutions, deletedSoFar.union(deleted)];
|
||||
}, [new Map(), new Set()]);
|
||||
const unifiedTypeVars = new Set([
|
||||
...actualTypeVars,
|
||||
...formalTypeVars,
|
||||
].filter(a => !unifiedSubstitusions.has(a) && !deleted.has(a)));
|
||||
return {
|
||||
substitutions: unifiedSubstitusions,
|
||||
typeVars: unifiedTypeVars,
|
||||
type: {
|
||||
symbol: formalType.symbol,
|
||||
params: unifiedParams.map(p => p.type),
|
||||
},
|
||||
};
|
||||
}
|
||||
|
||||
if (formalType.numDict !== undefined) {
|
||||
if (actualType.numDict === undefined) {
|
||||
throw new Error(`cannot assign ${pretty(actualType)} to ${pretty(formalType)}`);
|
||||
}
|
||||
else {
|
||||
// both are NumDict type
|
||||
const underlyingType = unify(
|
||||
{typeVars: formalTypeVars, type: formalType.numDict},
|
||||
{typeVars: actualTypeVars, type: actualType.numDict});
|
||||
return {
|
||||
substitutions: underlyingType.substitutions,
|
||||
typeVars: new Set([
|
||||
...actualTypeVars,
|
||||
...formalTypeVars].filter(a => !underlyingType.substitutions.has(a))),
|
||||
type: numDictType(underlyingType.type),
|
||||
};
|
||||
}
|
||||
}
|
||||
throw new Error("i don't know what to do :(");
|
||||
};
|
||||
|
||||
// export const matchConcrete = ({typeVars, type: formalType}, actualType) => {
|
||||
// return unify({typeVars, type: formalType}, {typeVars: new Set(), type: actualType});
|
||||
// };
|
||||
|
||||
export const substitute = (type, substitutions) => {
|
||||
if (substitutions.has(type)) {
|
||||
// type IS a type var to be substituted:
|
||||
return substitutions.get(type);
|
||||
}
|
||||
if (type.listOf !== undefined) {
|
||||
// list type
|
||||
return lsType(substitute(type.listOf, substitutions));
|
||||
}
|
||||
if (type.in !== undefined) {
|
||||
// function type
|
||||
return fnType({
|
||||
in: substitute(type.in, substitutions),
|
||||
out: substitute(type.out, substitutions),
|
||||
})
|
||||
}
|
||||
return type;
|
||||
return {
|
||||
symbol: type.symbol,
|
||||
params: type.params.map(p => substitute(p, substitutions)),
|
||||
};
|
||||
}
|
||||
|
||||
export const assign = (genFnType, paramType) => {
|
||||
const [inType, outType] = genFnType.type.params;
|
||||
const matchedInType = unify({
|
||||
typeVars: genFnType.typeVars,
|
||||
type: genFnType.type.in,
|
||||
type: inType,
|
||||
}, paramType);
|
||||
const substitutedOutType = substitute(genFnType.type.out, matchedInType.substitutions);
|
||||
const substitutedOutType = substitute(outType, matchedInType.substitutions);
|
||||
return {
|
||||
typeVars: matchedInType.typeVars,
|
||||
type: substitutedOutType,
|
||||
|
|
|
|||
|
|
@ -1,21 +1,22 @@
|
|||
import { Bool, Int } from "../primitives/symbols.js";
|
||||
import { lsType } from "../structures/list_common.js";
|
||||
import { fnType } from "../metacircular.js";
|
||||
import { assign, makeGeneric, mergeSubstitutions, unify } from "./generics.js";
|
||||
import { lsType } from "../structures/list.js";
|
||||
import { fnType } from "../structures/function.js";
|
||||
import { assign, makeGeneric, unify } from "./generics.js";
|
||||
import { pretty } from "../util.js";
|
||||
|
||||
// a -> Int
|
||||
const a_to_Int = makeGeneric(a => fnType({in: a, out: Int}));
|
||||
// Bool -> Int
|
||||
const Bool_to_Int = makeGeneric(() => fnType({in: lsType(Bool), out: Int}));
|
||||
console.log("should be: Bool -> Int")
|
||||
console.log(unify(a_to_Int, Bool_to_Int));
|
||||
console.log("should be: [Bool] -> Int")
|
||||
console.log(pretty(unify(a_to_Int, Bool_to_Int)));
|
||||
|
||||
// (a -> a) -> b
|
||||
const fnType2 = makeGeneric((a,b) => fnType({in: fnType({in: a, out: a}), out: b}));
|
||||
// (Bool -> Bool) -> a
|
||||
const fnType3 = makeGeneric(a => fnType({in: fnType({in: Bool, out: Bool}), out: a}));
|
||||
console.log("should be: (Bool -> Bool) -> a");
|
||||
console.log(unify(fnType2, fnType3));
|
||||
console.log(pretty(unify(fnType2, fnType3)));
|
||||
|
||||
// (a -> b) -> [a] -> [b]
|
||||
const mapFnType = makeGeneric((a,b) =>
|
||||
|
|
@ -27,7 +28,7 @@ const mapFnType = makeGeneric((a,b) =>
|
|||
const idFnType = makeGeneric(a =>
|
||||
fnType({in: a, out: a}));
|
||||
console.log("should be: [a] -> [a]");
|
||||
console.log(assign(mapFnType, idFnType));
|
||||
console.log(pretty(assign(mapFnType, idFnType)));
|
||||
|
||||
// (a -> Int) -> [a] -> a
|
||||
const weirdFnType = makeGeneric(a =>
|
||||
|
|
@ -36,4 +37,4 @@ const weirdFnType = makeGeneric(a =>
|
|||
out: fnType({in: lsType(a), out: a}),
|
||||
}));
|
||||
console.log("should be: [Int] -> Int");
|
||||
console.log(assign(weirdFnType, idFnType));
|
||||
console.log(pretty(assign(weirdFnType, idFnType)));
|
||||
|
|
@ -1,2 +0,0 @@
|
|||
|
||||
|
||||
Loading…
Add table
Add a link
Reference in a new issue