@d8x/perpetuals-sdk 2.0.12-alpha → 2.1.0-alpha2

package/dist/cjs/marketData.js

@@ -286,7 +286,8 @@ class MarketData extends perpetualDataHandler_1.default {
      */
     async _positionRiskForTraderInPerpetual(traderAddr, symbol, provider, overrides) {
         let obj = await this.priceFeedGetter.fetchPricesForPerpetual(symbol);
-        let mgnAcct = await perpetualDataHandler_1.default.getMarginAccount(traderAddr, symbol, this.symbolToPerpStaticInfo, new ethers_1.Contract(this.proxyAddr, this.config.proxyABI, provider), [obj.idxPrices[0], obj.idxPrices[1]], overrides);
+        const isPred = this.isPredictionMarket(symbol);
+        let mgnAcct = await perpetualDataHandler_1.default.getMarginAccount(traderAddr, symbol, this.symbolToPerpStaticInfo, new ethers_1.Contract(this.proxyAddr, this.config.proxyABI, provider), obj, isPred, overrides);
         return mgnAcct;
     }
     /**
@@ -298,122 +299,67 @@ class MarketData extends perpetualDataHandler_1.default {
      */
     async _positionRiskForTraderInPerpetuals(traderAddr, symbols, provider, overrides) {
         const MAX_SYMBOLS_PER_CALL = 10;
-        const S2S3 = new Array();
+        const pxInfo = new Array();
         for (let i = 0; i < symbols.length; i++) {
             let obj = await this.priceFeedGetter.fetchPricesForPerpetual(symbols[i]);
-            S2S3.push([obj.idxPrices[0], obj.idxPrices[1]]);
+            pxInfo.push(obj);
         }
         let mgnAcct = [];
         let callSymbols = symbols.slice(0, MAX_SYMBOLS_PER_CALL);
-        let pxS2S3 = S2S3.slice(0, MAX_SYMBOLS_PER_CALL);
+        let _px = pxInfo.slice(0, MAX_SYMBOLS_PER_CALL);
         while (callSymbols.length > 0) {
-            let acc = await perpetualDataHandler_1.default.getMarginAccounts(Array(callSymbols.length).fill(traderAddr), callSymbols, this.symbolToPerpStaticInfo, contracts_1.Multicall3__factory.connect(this.config.multicall ?? constants_1.MULTICALL_ADDRESS, provider), new ethers_1.Contract(this.proxyAddr, this.config.proxyABI, provider), pxS2S3, overrides);
+            const isPred = callSymbols.map((_sym) => this.isPredictionMarket(_sym));
+            let acc = await perpetualDataHandler_1.default.getMarginAccounts(Array(callSymbols.length).fill(traderAddr), callSymbols, this.symbolToPerpStaticInfo, contracts_1.Multicall3__factory.connect(this.config.multicall ?? constants_1.MULTICALL_ADDRESS, provider), new ethers_1.Contract(this.proxyAddr, this.config.proxyABI, provider), _px, isPred, overrides);
             mgnAcct = mgnAcct.concat(acc);
             callSymbols = symbols.slice(mgnAcct.length, mgnAcct.length + MAX_SYMBOLS_PER_CALL);
-            pxS2S3 = S2S3.slice(mgnAcct.length, mgnAcct.length + MAX_SYMBOLS_PER_CALL);
+            _px = pxInfo.slice(mgnAcct.length, mgnAcct.length + MAX_SYMBOLS_PER_CALL);
         }
         return mgnAcct;
     }
-    /**
-     * Estimates what the position risk will be if a given order is executed.
-     * @param traderAddr Address of trader
-     * @param order Order to be submitted
-     * @param account Position risk before trade. Defaults to current position if not given.
-     * @param indexPriceInfo Index prices and market status (open/closed). Defaults to current market status if not given.
-     * @returns Position risk after trade, including order cost and maximal trade sizes for position
-     * @example
-     * import { MarketData, PerpetualDataHandler } from '@d8x/perpetuals-sdk';
-     * async function main() {
-     *   console.log(MarketData);
-     *   // setup
-     *   const config = PerpetualDataHandler.readSDKConfig("cardona");
-     *   const mktData = new MarketData(config);
-     *   await mktData.createProxyInstance();
-     *   const order: Order = {
-     *        symbol: "MATIC-USD-MATIC",
-     *        side: "BUY",
-     *        type: "MARKET",
-     *        quantity: 100,
-     *        leverage: 2,
-     *        executionTimestamp: Date.now()/1000,
-     *    };
-     *   // Get position risk conditional on this order being executed
-     *   const posRisk = await mktData.positionRiskOnTrade("0xAb5801a7D398351b8bE11C439e05C5B3259aeC9B", order);
-     *   console.log(posRisk);
-     * }
-     * main();
-     */
-    async positionRiskOnTrade(traderAddr, order, account, indexPriceInfo, overrides) {
+    async dataForPositionRiskOnTrade(symbol, traderAddr, tradeAmountBC, indexPriceInfo, signedPositionNotionalBaseCCY, overrides) {
         if (this.proxyContract == null || this.multicall == null) {
             throw Error("no proxy contract initialized. Use createProxyInstance().");
         }
-        // fetch prices
-        if (indexPriceInfo == undefined) {
-            let obj = await this.priceFeedGetter.fetchPricesForPerpetual(order.symbol);
-            indexPriceInfo = [obj.idxPrices[0], obj.idxPrices[1], obj.mktClosed[0], obj.mktClosed[1]];
-        }
-        // override total fee
-        let tradingFeeTbps;
-        if (overrides) {
-            ({ tradingFeeTbps, ...overrides } = overrides);
-        }
-        // signed trade amount
-        let tradeAmountBC = Math.abs(order.quantity) * (order.side == constants_1.BUY_SIDE ? 1 : -1);
-        const accountGiven = account !== undefined;
+        const isPredMkt = this.isPredictionMarket(symbol);
         // create all calls
-        const poolId = perpetualDataHandler_1.default._getPoolIdFromSymbol(order.symbol, this.poolStaticInfos);
-        const perpId = perpetualDataHandler_1.default.symbolToPerpetualId(order.symbol, this.symbolToPerpStaticInfo);
-        const fS2S3 = [indexPriceInfo[0], indexPriceInfo[1]].map((x) => (0, d8XMath_1.floatToABK64x64)(x));
+        const perpId = perpetualDataHandler_1.default.symbolToPerpetualId(symbol, this.symbolToPerpStaticInfo);
+        const [fS2, fS3, fEma] = [indexPriceInfo.s2, indexPriceInfo.s3, indexPriceInfo.ema].map((x) => (0, d8XMath_1.floatToABK64x64)(x));
         const proxyCalls = [
             // 0: traderState
             {
                 target: this.proxyContract.target,
                 allowFailure: true,
-                callData: this.proxyContract.interface.encodeFunctionData("getTraderState", [perpId, traderAddr, fS2S3]),
-            },
-            // 1: ammState
-            {
-                target: this.proxyContract.target,
-                allowFailure: true,
-                callData: this.proxyContract.interface.encodeFunctionData("getAMMState", [perpId, fS2S3]),
-            },
-            // 2: exchangeFee
-            {
-                target: this.proxyContract.target,
-                allowFailure: false,
-                callData: this.proxyContract.interface.encodeFunctionData("queryExchangeFee", [
-                    poolId,
-                    traderAddr,
-                    order.brokerAddr ?? constants_1.ZERO_ADDRESS,
-                ]),
+                callData: this.proxyContract.interface.encodeFunctionData("getTraderState", [perpId, traderAddr, [fEma, fS3]]),
             },
-            // 3: perpetual price
+            // 1: perpetual price
             {
                 target: this.proxyContract.target,
                 allowFailure: true,
                 callData: this.proxyContract.interface.encodeFunctionData("queryPerpetualPrice", [
                     perpId,
                     (0, d8XMath_1.floatToABK64x64)(tradeAmountBC),
-                    fS2S3,
+                    [fS2, fS3],
+                    indexPriceInfo.conf,
+                    indexPriceInfo.predMktCLOBParams,
                 ]),
             },
-            // 4: max long pos
+            // 2: max long pos
             {
                 target: this.proxyContract.target,
                 allowFailure: false,
                 callData: this.proxyContract.interface.encodeFunctionData("getMaxSignedOpenTradeSizeForPos", [
                     perpId,
-                    account ? (0, d8XMath_1.floatToABK64x64)(account.positionNotionalBaseCCY * (account.side === constants_1.BUY_SIDE ? 1 : -1)) : 0n,
+                    (0, d8XMath_1.floatToABK64x64)(signedPositionNotionalBaseCCY),
                     true,
                 ]),
             },
-            // 5: max short pos
+            // 3: max short pos
             {
                 target: this.proxyContract.target,
                 allowFailure: false,
                 callData: this.proxyContract.interface.encodeFunctionData("getMaxSignedOpenTradeSizeForPos", [
                     perpId,
-                    account ? (0, d8XMath_1.floatToABK64x64)(account.positionNotionalBaseCCY * (account.side === constants_1.BUY_SIDE ? 1 : -1)) : 0n,
+                    (0, d8XMath_1.floatToABK64x64)(signedPositionNotionalBaseCCY),
                     false,
                 ]),
             },
@@ -421,44 +367,84 @@ class MarketData extends perpetualDataHandler_1.default {
         // multicall
         const encodedResults = await this.multicall.aggregate3.staticCall(proxyCalls, (overrides || {}));
         // positionRisk to apply this trade on: if not given, defaults to the current trader's position
-        if (!account) {
-            let traderState;
-            if (encodedResults[0].success) {
-                traderState = this.proxyContract.interface.decodeFunctionResult("getTraderState", encodedResults[0].returnData)[0];
-            }
-            else {
-                traderState = await this.proxyContract.getTraderState(perpId, traderAddr, fS2S3);
-            }
-            account = MarketData.buildMarginAccountFromState(order.symbol, traderState, this.symbolToPerpStaticInfo, [
-                indexPriceInfo[0],
-                indexPriceInfo[1],
-            ]);
-        }
-        // perpetualState, for prices
-        let ammState;
-        if (encodedResults[1].success) {
-            ammState = this.proxyContract.interface.decodeFunctionResult("getAMMState", encodedResults[1].returnData)[0];
+        let traderState;
+        if (encodedResults[0].success) {
+            traderState = this.proxyContract.interface.decodeFunctionResult("getTraderState", encodedResults[0].returnData)[0];
         }
         else {
-            ammState = await this.proxyContract.getAMMState(perpId, fS2S3);
+            traderState = await this.proxyContract.getTraderState(perpId, traderAddr, [fEma, fS3]);
         }
-        const perpetualState = perpetualDataHandler_1.default._parseAMMState(order.symbol, ammState, [0n, 0n], // not used below
-        indexPriceInfo, this.symbolToPerpStaticInfo);
-        let [S2, S3, Sm] = [perpetualState.indexPrice, perpetualState.collToQuoteIndexPrice, perpetualState.markPrice];
-        // exchange fee based on this trader's address (volume, token holding, etc) and his broker address (if any)
-        const exchangeFeeTbps = this.proxyContract.interface.decodeFunctionResult("queryExchangeFee", encodedResults[2].returnData)[0];
+        const account = MarketData.buildMarginAccountFromState(symbol, traderState, this.symbolToPerpStaticInfo, indexPriceInfo, isPredMkt);
         // amm price for this trade amount
         let ammPrice;
         {
             let fPrice;
-            if (encodedResults[3].success) {
-                fPrice = this.proxyContract.interface.decodeFunctionResult("queryPerpetualPrice", encodedResults[3].returnData)[0];
+            if (encodedResults[1].success) {
+                fPrice = this.proxyContract.interface.decodeFunctionResult("queryPerpetualPrice", encodedResults[1].returnData)[0];
             }
             else {
-                fPrice = await this.proxyContract.queryPerpetualPrice(perpId, (0, d8XMath_1.floatToABK64x64)(tradeAmountBC), fS2S3);
+                fPrice = await this.proxyContract.queryPerpetualPrice(perpId, (0, d8XMath_1.floatToABK64x64)(tradeAmountBC), [indexPriceInfo.s2, indexPriceInfo.s3], indexPriceInfo.conf, indexPriceInfo.predMktCLOBParams);
             }
             ammPrice = (0, d8XMath_1.ABK64x64ToFloat)(fPrice);
         }
+        // max buy
+        const fMaxLong = this.proxyContract.interface.decodeFunctionResult("getMaxSignedOpenTradeSizeForPos", encodedResults[2].returnData)[0];
+        const maxLongTrade = Math.max(0, (0, d8XMath_1.ABK64x64ToFloat)(fMaxLong) - signedPositionNotionalBaseCCY);
+        // max sell
+        const fMaxShort = this.proxyContract.interface.decodeFunctionResult("getMaxSignedOpenTradeSizeForPos", encodedResults[3].returnData)[0];
+        const maxShortTrade = Math.max(0, Math.abs((0, d8XMath_1.ABK64x64ToFloat)(fMaxShort)) - signedPositionNotionalBaseCCY);
+        return { account: account, ammPrice: ammPrice, maxShortTrade: maxShortTrade, maxLongTrade: maxLongTrade };
+    }
+    /**
+     * Estimates what the position risk will be if a given order is executed.
+     * @param traderAddr Address of trader
+     * @param order Order to be submitted
+     * @param signedPositionNotionalBaseCCY signed position notional of current position (before trade)
+     * @param tradingFeeTbps trading fee in tenth of basis points (exchange fee and broker fee)
+     * @param indexPriceInfo Index prices and market status (open/closed). Defaults to current market status if not given.
+     * @returns Position risk after trade, including order cost and maximal trade sizes for position
+     * @example
+     * import { MarketData, PerpetualDataHandler } from '@d8x/perpetuals-sdk';
+     * async function main() {
+     *   console.log(MarketData);
+     *   // setup
+     *   const config = PerpetualDataHandler.readSDKConfig("cardona");
+     *   const mktData = new MarketData(config);
+     *   await mktData.createProxyInstance();
+     *   const order: Order = {
+     *        symbol: "MATIC-USD-MATIC",
+     *        side: "BUY",
+     *        type: "MARKET",
+     *        quantity: 100,
+     *        leverage: 2,
+     *        executionTimestamp: Date.now()/1000,
+     *    };
+     *   // Get position risk conditional on this order being executed
+     *   const posRisk = await mktData.positionRiskOnTrade("0xAb5801a7D398351b8bE11C439e05C5B3259aeC9B", order, 0, 60);
+     *   console.log(posRisk);
+     * }
+     * main();
+     */
+    async positionRiskOnTrade(traderAddr, order, signedPositionNotionalBaseCCY, tradingFeeTbps, indexPriceInfo, overrides) {
+        if (this.proxyContract == null || this.multicall == null) {
+            throw Error("no proxy contract initialized. Use createProxyInstance().");
+        }
+        const isPredMkt = this.isPredictionMarket(order.symbol);
+        // fetch prices
+        if (indexPriceInfo == undefined) {
+            indexPriceInfo = await this.priceFeedGetter.fetchPricesForPerpetual(order.symbol);
+        }
+        // signed trade amount
+        let tradeAmountBC = Math.abs(order.quantity) * (order.side == constants_1.BUY_SIDE ? 1 : -1);
+        const symbol = order.symbol;
+        let obj = await this.dataForPositionRiskOnTrade(symbol, traderAddr, tradeAmountBC, indexPriceInfo, signedPositionNotionalBaseCCY, overrides);
+        const account = obj.account;
+        const maxLongTrade = obj.maxLongTrade;
+        const maxShortTrade = obj.maxShortTrade;
+        const ammPrice = obj.ammPrice;
+        let Sm = account.markPrice;
+        let S2 = indexPriceInfo.s2;
+        let S3 = account.collToQuoteConversion;
         // price for this order = amm price if no limit given, else conservatively adjusted
         let tradePrice;
         if (order.limitPrice == undefined) {
@@ -494,16 +480,6 @@ class MarketData extends perpetualDataHandler_1.default {
                 }
             }
         }
-        // max buy
-        const fMaxLong = this.proxyContract.interface.decodeFunctionResult("getMaxSignedOpenTradeSizeForPos", encodedResults[4].returnData)[0];
-        const maxLongTrade = account.side == constants_1.BUY_SIDE
-            ? Math.max(0, (0, d8XMath_1.ABK64x64ToFloat)(fMaxLong) - (accountGiven ? 0 : account.positionNotionalBaseCCY))
-            : (0, d8XMath_1.ABK64x64ToFloat)(fMaxLong) + account.positionNotionalBaseCCY;
-        // max sell
-        const fMaxShort = this.proxyContract.interface.decodeFunctionResult("getMaxSignedOpenTradeSizeForPos", encodedResults[5].returnData)[0];
-        const maxShortTrade = account.side == constants_1.SELL_SIDE
-            ? Math.max(0, Math.abs((0, d8XMath_1.ABK64x64ToFloat)(fMaxShort)) - (accountGiven ? 0 : Math.abs(account.positionNotionalBaseCCY)))
-            : Math.abs((0, d8XMath_1.ABK64x64ToFloat)(fMaxShort)) + Math.abs(account.positionNotionalBaseCCY);
         // Current state:
         let lotSizeBC = MarketData._getLotSize(order.symbol, this.symbolToPerpStaticInfo);
         // Too small, no change to account
@@ -525,10 +501,6 @@ class MarketData extends perpetualDataHandler_1.default {
             newPositionBC = 0;
         }
         let newSide = newPositionBC > 0 ? constants_1.BUY_SIDE : newPositionBC < 0 ? constants_1.SELL_SIDE : constants_1.CLOSED_SIDE;
-        if (tradingFeeTbps === undefined) {
-            // use usual input if not overriden
-            tradingFeeTbps = Number(exchangeFeeTbps) + (order.brokerFeeTbps ?? 0);
-        }
         let tradingFeeCC = (Math.abs(tradeAmountBC) * tradingFeeTbps * 1e-5 * S2) / S3;
         let referralFeeCC = this.symbolToPerpStaticInfo.get(account.symbol).referralRebate;
         // Trade type:
@@ -553,7 +525,7 @@ class MarketData extends perpetualDataHandler_1.default {
             let initialMarginRate = this.symbolToPerpStaticInfo.get(account.symbol).initialMarginRate;
             targetLvg = isFlip || isOpen ? order.leverage ?? 1 / initialMarginRate : 0;
             let [b0, pos0] = isOpen ? [0, 0] : [account.collateralCC, currentPositionBC];
-            traderDepositCC = (0, d8XMath_1.getDepositAmountForLvgTrade)(pos0, b0, tradeAmountBC, targetLvg, tradePrice, S3, Sm);
+            traderDepositCC = (0, d8XMath_1.getDepositAmountForLvgTrade)(pos0, b0, tradeAmountBC, targetLvg, tradePrice, S3, Sm, isPredMkt);
             // fees are paid from wallet in this case
             traderDepositCC += tradingFeeCC + referralFeeCC;
         }
@@ -578,7 +550,7 @@ class MarketData extends perpetualDataHandler_1.default {
                 ? Infinity
                 : (Math.abs(newPositionBC) * Sm) / S3 / newMarginBalanceCC;
         // Liquidation params
-        let [S2Liq, S3Liq, tau] = MarketData._getLiquidationParams(account.symbol, newLockedInValueQC, newPositionBC, newMarginCashCC, Sm, S3, this.symbolToPerpStaticInfo);
+        let [S2Liq, S3Liq, tau] = MarketData._getLiquidationParams(account.symbol, newLockedInValueQC, newPositionBC, newMarginCashCC, Sm, S3, S2, this.symbolToPerpStaticInfo);
         // New position risk
         let newPositionRisk = {
             symbol: account.symbol,
@@ -601,6 +573,18 @@ class MarketData extends perpetualDataHandler_1.default {
             maxShortTrade: maxShortTrade,
         };
     }
+    /**
+     * Fee is relative to base-currency amount (=trade amount)
+     * @param state current perpetual state (need longBC and shortBC)
+     * @param maxMaintMgnRate maintenance margin rate param for pred mkts
+     * @param Sm Mark price
+     * @param tradeAmtBC signed trade amount
+     * @param tradeMgnRate margin rate param from perpetual
+     * @returns relative exchange fee in decimals
+     */
+    static exchangeFeePrdMkts(state, maxMaintMgnRate, Sm, tradeAmtBC, tradeMgnRate) {
+        return (0, d8XMath_1.pmExchangeFee)(Sm - 1, maxMaintMgnRate, state.shortBC, state.longBC, tradeAmtBC, tradeMgnRate);
+    }
     /**
      * Estimates what the position risk will be if given amount of collateral is added/removed from the account.
      * @param {number} deltaCollateral Amount of collateral to add or remove (signed)
@@ -630,11 +614,17 @@ class MarketData extends perpetualDataHandler_1.default {
             throw new Error("not enough margin to remove");
         }
         if (indexPriceInfo == undefined) {
-            let obj = await this.priceFeedGetter.fetchPricesForPerpetual(account.symbol);
-            indexPriceInfo = [obj.idxPrices[0], obj.idxPrices[1], obj.mktClosed[0], obj.mktClosed[1]];
+            indexPriceInfo = await this.priceFeedGetter.fetchPricesForPerpetual(account.symbol);
         }
         let perpetualState = await this.getPerpetualState(account.symbol, indexPriceInfo, overrides);
-        let [S2, S3, Sm] = [perpetualState.indexPrice, perpetualState.collToQuoteIndexPrice, perpetualState.markPrice];
+        let Sm; //mark price
+        if (this.isPredictionMarket(account.symbol)) {
+            Sm = indexPriceInfo.ema + perpetualState.markPremium;
+        }
+        else {
+            Sm = perpetualState.indexPrice * (1 + perpetualState.markPremium);
+        }
+        let [S2, S3] = [perpetualState.indexPrice, perpetualState.collToQuoteIndexPrice];
         // no position: just increase collateral and kill liquidation vars
         if (account.positionNotionalBaseCCY == 0) {
             return {
@@ -674,7 +664,7 @@ class MarketData extends perpetualDataHandler_1.default {
         }
         let newLeverage = (Math.abs(positionBC) * Sm) / S3 / newMarginBalanceCC;
         // Liquidation params
-        let [S2Liq, S3Liq, tau] = MarketData._getLiquidationParams(account.symbol, lockedInQC, positionBC, newMarginCashCC, Sm, S3, this.symbolToPerpStaticInfo);
+        let [S2Liq, S3Liq, tau] = MarketData._getLiquidationParams(account.symbol, lockedInQC, positionBC, newMarginCashCC, Sm, S3, S2, this.symbolToPerpStaticInfo);
         // New position risk
         let newPositionRisk = {
             symbol: account.symbol,
@@ -693,35 +683,39 @@ class MarketData extends perpetualDataHandler_1.default {
         return newPositionRisk;
     }
     /**
-     * Calculates liquidation prices for a given position
+     * Calculates liquidation prices for a position
+     * constructed in positionRiskOnTrade/positionRiskOnCollateralAction
      * @param symbol Perpetual symbol
      * @param lockedInQC Locked in value
      * @param signedPositionBC Signed position size
-     * @param marginCashCC Cash in margin account
+     * @param marginCashCC Available cash in margin account (includes unpaid funding)
      * @param markPrice Mark price
-     * @param collToQuoteConversion Collateral index price
+     * @param collToQuoteConversion Collateral index price (S3)
+     * @param S2 index price
      * @param symbolToPerpStaticInfo Symbol-to-perp static info mapping
      * @returns [Base index price, Collateral index price, Maintenance margin rate]
      * @ignore
      */
-    static _getLiquidationParams(symbol, lockedInQC, signedPositionBC, marginCashCC, markPrice, collToQuoteConversion, symbolToPerpStaticInfo) {
+    static _getLiquidationParams(symbol, lockedInQC, signedPositionBC, marginCashCC, markPrice, collToQuoteConversion, S2, symbolToPerpStaticInfo) {
         let S2Liq, S3Liq;
-        let tau = symbolToPerpStaticInfo.get(symbol).maintenanceMarginRate;
-        let ccyType = symbolToPerpStaticInfo.get(symbol).collateralCurrencyType;
-        if (ccyType == constants_1.CollaterlCCY.BASE) {
-            S2Liq = (0, d8XMath_1.calculateLiquidationPriceCollateralBase)(lockedInQC, signedPositionBC, marginCashCC, tau);
-            S3Liq = S2Liq;
-        }
-        else if (ccyType == constants_1.CollaterlCCY.QUANTO) {
-            S3Liq = collToQuoteConversion;
-            S2Liq = (0, d8XMath_1.calculateLiquidationPriceCollateralQuanto)(lockedInQC, signedPositionBC, marginCashCC, tau, collToQuoteConversion, markPrice);
-        }
-        else {
-            S2Liq = (0, d8XMath_1.calculateLiquidationPriceCollateralQuote)(lockedInQC, signedPositionBC, marginCashCC, tau);
-        }
-        // floor at 0
-        S2Liq = S2Liq < 0 ? 0 : S2Liq;
-        S3Liq = S3Liq && S3Liq < 0 ? 0 : S3Liq;
+        const staticInfo = symbolToPerpStaticInfo.get(symbol);
+        let tau = staticInfo.maintenanceMarginRate;
+        let ccyType = staticInfo.collateralCurrencyType;
+        const isPred = MarketData.isPredictionMarketStatic(staticInfo);
+        const idx_availableCashCC = 2;
+        const idx_cash = 3;
+        const idx_notional = 4;
+        const idx_locked_in = 5;
+        const idx_mark_price = 8;
+        const idx_s3 = 9;
+        let traderState = new Array(10);
+        traderState[idx_availableCashCC] = (0, d8XMath_1.floatToABK64x64)(marginCashCC);
+        traderState[idx_cash] = traderState[idx_availableCashCC];
+        traderState[idx_notional] = (0, d8XMath_1.floatToABK64x64)(signedPositionBC);
+        traderState[idx_locked_in] = (0, d8XMath_1.floatToABK64x64)(lockedInQC);
+        traderState[idx_mark_price] = (0, d8XMath_1.floatToABK64x64)(markPrice);
+        traderState[idx_s3] = (0, d8XMath_1.floatToABK64x64)(collToQuoteConversion);
+        [S2Liq, S3Liq, tau, ,] = MarketData._calculateLiquidationPrice(symbol, traderState, S2, symbolToPerpStaticInfo, isPred);
         return [S2Liq, S3Liq, tau];
     }
     /**
@@ -889,22 +883,28 @@ class MarketData extends perpetualDataHandler_1.default {
         if (!this.proxyContract || !this.multicall) {
             throw new Error("proxy contract not initialized");
         }
+        if (this.isPredictionMarket(symbol)) {
+            // prediction markets
+            return this.pmMaxOrderSizeForTrader(traderAddr, symbol, overrides);
+        }
+        // regular markets
+        return this.rmMaxOrderSizeForTrader(traderAddr, symbol, overrides);
+    }
+    async pmMaxOrderSizeForTrader(traderAddr, symbol, overrides) {
+        if (!this.proxyContract || !this.multicall) {
+            throw new Error("proxy contract not initialized");
+        }
+        const IERC20 = new ethers_1.Interface(constants_1.ERC20_ABI);
         const perpId = perpetualDataHandler_1.default.symbolToPerpetualId(symbol, this.symbolToPerpStaticInfo);
-        const poolId = this.getPoolIdFromSymbol(symbol);
         const poolInfo = this.poolStaticInfos[this.getPoolStaticInfoIndexFromSymbol(symbol)];
-        const perpInfo = this.getPerpetualStaticInfo(symbol);
-        const IERC20 = new ethers_1.Interface(constants_1.ERC20_ABI);
-        const indexPriceInfo = await this.priceFeedGetter
-            .fetchPricesForPerpetual(symbol)
-            .then((obj) => [obj.idxPrices[0], obj.idxPrices[1], obj.mktClosed[0], obj.mktClosed[1]]);
-        const fS2S3 = [indexPriceInfo[0], indexPriceInfo[1]].map((x) => (0, d8XMath_1.floatToABK64x64)(x));
-        let coll2SettlePromise = this.fetchCollateralToSettlementConversion(symbol);
+        const indexPriceInfo = await this.priceFeedGetter.fetchPricesForPerpetual(symbol);
+        const [fS2, fS3, fEma] = [indexPriceInfo.s2, indexPriceInfo.s3, indexPriceInfo.ema].map((x) => (0, d8XMath_1.floatToABK64x64)(x));
         const proxyCalls = [
             // 0: traderState
             {
                 target: this.proxyContract.target,
                 allowFailure: false,
-                callData: this.proxyContract.interface.encodeFunctionData("getTraderState", [perpId, traderAddr, fS2S3]),
+                callData: this.proxyContract.interface.encodeFunctionData("getTraderState", [perpId, traderAddr, [fEma, fS3]]),
             },
             // 1: wallet balance
             {
@@ -912,31 +912,53 @@ class MarketData extends perpetualDataHandler_1.default {
                 allowFailure: false,
                 callData: IERC20.encodeFunctionData("balanceOf", [traderAddr]),
             },
-            // 2: exchange fee
+            // 2: amm state
             {
                 target: this.proxyContract.target,
                 allowFailure: false,
-                callData: this.proxyContract.interface.encodeFunctionData("queryExchangeFee", [
-                    poolId,
-                    traderAddr,
-                    constants_1.ZERO_ADDRESS,
-                ]),
+                callData: this.proxyContract.interface.encodeFunctionData("getAMMState", [perpId, [fS2, fS3]]),
             },
         ];
         // multicall
         const encodedResults = await this.multicall.aggregate3.staticCall(proxyCalls, overrides || {});
-        // position risk
-        const idxNotional = 4;
         const traderState = this.proxyContract.interface.decodeFunctionResult("getTraderState", encodedResults[0].returnData)[0];
-        const account = MarketData.buildMarginAccountFromState(symbol, traderState, this.symbolToPerpStaticInfo, [
-            indexPriceInfo[0],
-            indexPriceInfo[1],
-        ]);
-        // fee rate
-        const feeRateTbps = this.proxyContract.interface.decodeFunctionResult("queryExchangeFee", encodedResults[2].returnData)[0];
-        const feeRate = 1e-5 * Number(feeRateTbps);
-        // Max based on margin requirements:
         const walletBalance = (0, d8XMath_1.decNToFloat)(IERC20.decodeFunctionResult("balanceOf", encodedResults[1].returnData)[0], poolInfo.poolSettleTokenDecimals);
+        const ammState = this.proxyContract.interface.decodeFunctionResult("getAMMState", encodedResults[2].returnData)[0];
+        const account = MarketData.buildMarginAccountFromState(symbol, traderState, this.symbolToPerpStaticInfo, indexPriceInfo, true //isPredMkt
+        );
+        const openInterestBC = (0, d8XMath_1.ABK64x64ToFloat)(ammState[11]);
+        const net = -(0, d8XMath_1.ABK64x64ToFloat)(ammState[1]);
+        let totLong, totShort;
+        if (net < 0) {
+            totLong = openInterestBC;
+            totShort = openInterestBC - Math.abs(net);
+        }
+        else {
+            totLong = openInterestBC - net;
+            totShort = openInterestBC;
+        }
+        let currentPositionBC = (account.side == constants_1.BUY_SIDE ? 1 : -1) * account.positionNotionalBaseCCY;
+        const Sm = (0, d8XMath_1.ABK64x64ToFloat)(traderState[8]);
+        // settlement token must be equal to collateral token for walletBalance to be correct
+        const availCashCC = account.collateralCC + walletBalance + account.unrealizedFundingCollateralCCY;
+        const idxNotional = 4;
+        const [maxShortPosPerp, maxLongPosPerp] = await this.getMaxShortLongPos(perpId, traderState[idxNotional], overrides);
+        const maxShort = (0, d8XMath_1.pmFindMaxTradeSize)(-1, currentPositionBC, availCashCC, account.entryPrice * currentPositionBC, Sm, Sm, indexPriceInfo.s3, totLong, totShort, maxShortPosPerp, maxLongPosPerp);
+        const maxLong = (0, d8XMath_1.pmFindMaxTradeSize)(-1, currentPositionBC, availCashCC, account.entryPrice * currentPositionBC, Sm, Sm, indexPriceInfo.s3, totLong, totShort, maxShortPosPerp, maxLongPosPerp);
+        return { buy: maxLong, sell: maxShort };
+    }
+    /**
+     * Returns the maximal allowed short pos and long pos (signed) for a trader
+     * with given notional (in ABDK format) in the perpetual, ignoring the traders wallet balance
+     * @param perpId
+     * @param currentTraderPos ABDK64x64 notional position of trader
+     * @param overrides
+     * @returns [maxShortPos, maxLongPos] signed maximal position sizes
+     */
+    async getMaxShortLongPos(perpId, currentTraderPos, overrides) {
+        if (!this.proxyContract || !this.multicall) {
+            throw new Error("proxy contract not initialized");
+        }
         const proxyCalls2 = [
             // 0: max long
             {
@@ -944,7 +966,7 @@ class MarketData extends perpetualDataHandler_1.default {
                 allowFailure: false,
                 callData: this.proxyContract.interface.encodeFunctionData("getMaxSignedOpenTradeSizeForPos", [
                     perpId,
-                    traderState[idxNotional],
+                    currentTraderPos,
                     true,
                 ]),
             },
@@ -954,7 +976,7 @@ class MarketData extends perpetualDataHandler_1.default {
                 allowFailure: false,
                 callData: this.proxyContract.interface.encodeFunctionData("getMaxSignedOpenTradeSizeForPos", [
                     perpId,
-                    traderState[idxNotional],
+                    currentTraderPos,
                     false,
                 ]),
             },
@@ -964,15 +986,66 @@ class MarketData extends perpetualDataHandler_1.default {
         // Max based on perp:
         // max buy
         const maxLongOrderPerp = (0, d8XMath_1.ABK64x64ToFloat)(this.proxyContract.interface.decodeFunctionResult("getMaxSignedOpenTradeSizeForPos", encodedResults2[0].returnData)[0]);
-        const maxLongPosPerp = maxLongOrderPerp + (0, d8XMath_1.ABK64x64ToFloat)(traderState[idxNotional]);
+        const maxLongPosPerp = maxLongOrderPerp + (0, d8XMath_1.ABK64x64ToFloat)(currentTraderPos);
         // max short
         const maxShortOrderPerp = (0, d8XMath_1.ABK64x64ToFloat)(this.proxyContract.interface.decodeFunctionResult("getMaxSignedOpenTradeSizeForPos", encodedResults2[1].returnData)[0]);
-        const maxShortPosPerp = maxShortOrderPerp + (0, d8XMath_1.ABK64x64ToFloat)(traderState[idxNotional]);
+        const maxShortPosPerp = maxShortOrderPerp + (0, d8XMath_1.ABK64x64ToFloat)(currentTraderPos);
+        return [maxShortPosPerp, maxLongPosPerp];
+    }
+    async rmMaxOrderSizeForTrader(traderAddr, symbol, overrides) {
+        const perpId = perpetualDataHandler_1.default.symbolToPerpetualId(symbol, this.symbolToPerpStaticInfo);
+        const poolId = this.getPoolIdFromSymbol(symbol);
+        const poolInfo = this.poolStaticInfos[this.getPoolStaticInfoIndexFromSymbol(symbol)];
+        const perpInfo = this.getPerpetualStaticInfo(symbol);
+        const IERC20 = new ethers_1.Interface(constants_1.ERC20_ABI);
+        const isPredMkt = false;
+        const indexPriceInfo = await this.priceFeedGetter.fetchPricesForPerpetual(symbol);
+        let coll2SettlePromise = this.fetchCollateralToSettlementConversion(symbol);
+        const [fS2, fS3, fEma] = [indexPriceInfo.s2, indexPriceInfo.s3, indexPriceInfo.ema].map((x) => (0, d8XMath_1.floatToABK64x64)(x));
+        if (!this.proxyContract || !this.multicall) {
+            throw new Error("proxy contract not initialized");
+        }
+        const proxyCalls = [
+            // 0: traderState
+            {
+                target: this.proxyContract.target,
+                allowFailure: false,
+                callData: this.proxyContract.interface.encodeFunctionData("getTraderState", [perpId, traderAddr, [fEma, fS3]]),
+            },
+            // 1: wallet balance
+            {
+                target: poolInfo.poolSettleTokenAddr,
+                allowFailure: false,
+                callData: IERC20.encodeFunctionData("balanceOf", [traderAddr]),
+            },
+            // 2: exchange fee
+            {
+                target: this.proxyContract.target,
+                allowFailure: false,
+                callData: this.proxyContract.interface.encodeFunctionData("queryExchangeFee", [
+                    poolId,
+                    traderAddr,
+                    constants_1.ZERO_ADDRESS,
+                ]),
+            },
+        ];
+        // multicall
+        const encodedResults = await this.multicall.aggregate3.staticCall(proxyCalls, overrides || {});
+        // position risk
+        const idxNotional = 4;
+        const traderState = this.proxyContract.interface.decodeFunctionResult("getTraderState", encodedResults[0].returnData)[0];
+        const account = MarketData.buildMarginAccountFromState(symbol, traderState, this.symbolToPerpStaticInfo, indexPriceInfo, isPredMkt);
+        // fee rate
+        const feeRateTbps = this.proxyContract.interface.decodeFunctionResult("queryExchangeFee", encodedResults[2].returnData)[0];
+        const feeRate = 1e-5 * Number(feeRateTbps);
+        // Max based on margin requirements:
+        const walletBalance = (0, d8XMath_1.decNToFloat)(IERC20.decodeFunctionResult("balanceOf", encodedResults[1].returnData)[0], poolInfo.poolSettleTokenDecimals);
+        const [maxShortPosPerp, maxLongPosPerp] = await this.getMaxShortLongPos(perpId, traderState[idxNotional], overrides);
         const curPos = (account.side == constants_1.BUY_SIDE ? 1 : -1) * account.positionNotionalBaseCCY;
         const px = await coll2SettlePromise;
         const walletBalanceInMgnToken = walletBalance / px;
-        const maxLongPosAccount = (0, d8XMath_1.getMaxSignedPositionSize)(account.collateralCC + walletBalanceInMgnToken + account.unrealizedFundingCollateralCCY, curPos, account.entryPrice * curPos, 1, account.markPrice, perpInfo.initialMarginRate, feeRate, account.markPrice, indexPriceInfo[0], account.collToQuoteConversion);
-        const maxShortPosAccount = (0, d8XMath_1.getMaxSignedPositionSize)(account.collateralCC + walletBalanceInMgnToken + account.unrealizedFundingCollateralCCY, curPos, account.entryPrice * curPos, -1, account.markPrice, perpInfo.initialMarginRate, feeRate, account.markPrice, indexPriceInfo[0], account.collToQuoteConversion);
+        const maxLongPosAccount = (0, d8XMath_1.getMaxSignedPositionSize)(account.collateralCC + walletBalanceInMgnToken + account.unrealizedFundingCollateralCCY, curPos, account.entryPrice * curPos, 1, account.markPrice, perpInfo.initialMarginRate, feeRate, account.markPrice, indexPriceInfo.s2, account.collToQuoteConversion);
+        const maxShortPosAccount = (0, d8XMath_1.getMaxSignedPositionSize)(account.collateralCC + walletBalanceInMgnToken + account.unrealizedFundingCollateralCCY, curPos, account.entryPrice * curPos, -1, account.markPrice, perpInfo.initialMarginRate, feeRate, account.markPrice, indexPriceInfo.s2, account.collToQuoteConversion);
         // max long/short all accounted for
         const maxLong = Math.min(Math.abs(maxLongPosPerp), Math.abs(maxLongPosAccount));
         const maxShort = Math.min(Math.abs(maxShortPosPerp), Math.abs(maxShortPosAccount));
@@ -1105,6 +1178,7 @@ class MarketData extends perpetualDataHandler_1.default {
     /**
      * Get the current mark price
      * @param symbol symbol of the form ETH-USD-MATIC
+     * @param indexPrices optional. IdxPriceInfo
      * @example
      * import { MarketData, PerpetualDataHandler } from '@d8x/perpetuals-sdk';
      * async function main() {
@@ -1126,15 +1200,15 @@ class MarketData extends perpetualDataHandler_1.default {
             throw Error("no proxy contract initialized. Use createProxyInstance().");
         }
         if (indexPrices == undefined) {
-            let obj = await this.priceFeedGetter.fetchPricesForPerpetual(symbol);
-            indexPrices = [obj.idxPrices[0], obj.idxPrices[1]];
+            indexPrices = await this.priceFeedGetter.fetchPricesForPerpetual(symbol);
         }
-        return await perpetualDataHandler_1.default._queryPerpetualMarkPrice(symbol, this.symbolToPerpStaticInfo, this.proxyContract, indexPrices);
+        return await perpetualDataHandler_1.default._queryPerpetualMarkPrice(symbol, this.symbolToPerpStaticInfo, this.proxyContract, indexPrices, this.isPredictionMarket(symbol));
     }
     /**
      * get the current price for a given quantity
      * @param symbol symbol of the form ETH-USD-MATIC
      * @param quantity quantity to be traded, negative if short
+     * @param priceInfo [s2, s3, conf, params]; for non-prediction markets conf/params can be 0
      * @example
      * import { MarketData, PerpetualDataHandler } from '@d8x/perpetuals-sdk';
      * async function main() {
@@ -1151,16 +1225,18 @@ class MarketData extends perpetualDataHandler_1.default {
      *
      * @returns {number} price
      */
-    async getPerpetualPrice(symbol, quantity, indexPrices, overrides) {
+    async getPerpetualPrice(symbol, quantity, priceInfo, overrides) {
         if (this.proxyContract == null) {
             throw Error("no proxy contract initialized. Use createProxyInstance().");
         }
-        if (indexPrices == undefined) {
+        if (priceInfo == undefined) {
             // fetch from API
-            let obj = await this.priceFeedGetter.fetchPricesForPerpetual(symbol);
-            indexPrices = [obj.idxPrices[0], obj.idxPrices[1]];
+            priceInfo = await this.priceFeedGetter.fetchPricesForPerpetual(symbol);
         }
-        return await perpetualDataHandler_1.default._queryPerpetualPrice(symbol, quantity, this.symbolToPerpStaticInfo, this.proxyContract, indexPrices, overrides);
+        return await perpetualDataHandler_1.default._queryPerpetualPrice(symbol, quantity, this.symbolToPerpStaticInfo, this.proxyContract, [priceInfo.s2, priceInfo.s3], //s2,s3
+        priceInfo.conf, //conf
+        priceInfo.predMktCLOBParams, //params
+        overrides);
     }
     /**
      * Query recent perpetual state from blockchain
@@ -1172,8 +1248,7 @@ class MarketData extends perpetualDataHandler_1.default {
             throw Error("no proxy contract initialized. Use createProxyInstance().");
         }
         if (indexPriceInfo == undefined) {
-            let obj = await this.priceFeedGetter.fetchPricesForPerpetual(symbol);
-            indexPriceInfo = [obj.idxPrices[0], obj.idxPrices[1], obj.mktClosed[0], obj.mktClosed[1]];
+            indexPriceInfo = await this.priceFeedGetter.fetchPricesForPerpetual(symbol);
         }
         let state = await perpetualDataHandler_1.default._queryPerpetualState(symbol, this.symbolToPerpStaticInfo, this.proxyContract, this.multicall, indexPriceInfo, overrides);
         return state;
@@ -1369,7 +1444,7 @@ class MarketData extends perpetualDataHandler_1.default {
      * Result is in collateral currency
      * @param {string} traderAddr address of the trader
      * @param {string} symbol perpetual symbol of the form BTC-USD-MATIC
-     * @param indexPrices optional index prices, will otherwise fetch from REST API
+     * @param indexPrices optional indexPriceInfo
      * @returns available margin in collateral currency
      * @example
      * import { MarketData, PerpetualDataHandler } from '@d8x/perpetuals-sdk';
@@ -1391,11 +1466,10 @@ class MarketData extends perpetualDataHandler_1.default {
         }
         if (indexPrices == undefined) {
             // fetch from API
-            let obj = await this.priceFeedGetter.fetchPricesForPerpetual(symbol);
-            indexPrices = [obj.idxPrices[0], obj.idxPrices[1]];
+            indexPrices = await this.priceFeedGetter.fetchPricesForPerpetual(symbol);
         }
         let perpID = perpetualDataHandler_1.default.symbolToPerpetualId(symbol, this.symbolToPerpStaticInfo);
-        let traderState = await this.proxyContract.getTraderState(perpID, traderAddr, indexPrices.map((x) => (0, d8XMath_1.floatToABK64x64)(x == undefined || Number.isNaN(x) ? 0 : x)), overrides || {});
+        let traderState = await this.proxyContract.getTraderState(perpID, traderAddr, [indexPrices.s2, indexPrices.s3].map((x) => (0, d8XMath_1.floatToABK64x64)(x == undefined || Number.isNaN(x) ? 0 : x)), overrides || {});
         const idx_availableMargin = 1;
         let mgn = (0, d8XMath_1.ABK64x64ToFloat)(traderState[idx_availableMargin]);
         return mgn;
@@ -1662,7 +1736,7 @@ class MarketData extends perpetualDataHandler_1.default {
                 quoteCurrency: (0, utils_1.contractSymbolToSymbol)(perp.S2QuoteCCY, _symbolList),
                 indexPrice: 0,
                 collToQuoteIndexPrice: 0,
-                markPrice: (0, d8XMath_1.ABK64x64ToFloat)(perp.currentMarkPremiumRate.fPrice),
+                markPremium: (0, d8XMath_1.ABK64x64ToFloat)(perp.currentMarkPremiumRate.fPrice),
                 midPrice: 0,
                 currentFundingRateBps: 1e4 * (0, d8XMath_1.ABK64x64ToFloat)(perp.fCurrentFundingRate),
                 openInterestBC: (0, d8XMath_1.ABK64x64ToFloat)(perp.fOpenInterest),
@@ -1711,7 +1785,6 @@ class MarketData extends perpetualDataHandler_1.default {
                 perp.isMarketClosed = perp.isMarketClosed || idxPriceS3Pair[1];
             }
             perp.indexPrice = idxPriceS2Pair[0];
-            perp.markPrice = idxPriceS2Pair[0] * (1 + perp.markPrice); // currently filled with mark premium rate
             let indexS3 = 1;
             if (info.collateralCurrencyType == constants_1.COLLATERAL_CURRENCY_BASE) {
                 indexS3 = idxPriceS2Pair[0];