182 lines
7.6 KiB
JavaScript
182 lines
7.6 KiB
JavaScript
/**
|
|
* LogarithmPlotter - 2D plotter software to make BODE plots, sequences and distribution functions.
|
|
* Copyright (C) 2022 Ad5001
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
.pragma library
|
|
|
|
.import "common.js" as Common
|
|
.import "../parameters.js" as P
|
|
|
|
class RepartitionFunction extends Common.ExecutableObject {
|
|
static type(){return 'Repartition'}
|
|
static displayType(){return qsTr('Repartition')}
|
|
static displayTypeMultiple(){return qsTr('Repartition functions')}
|
|
/*static properties() {return {
|
|
'beginIncluded': 'boolean',
|
|
'drawLineEnds': 'boolean',
|
|
'comment1': 'Note: Specify the properties for each potential result.',
|
|
'probabilities': new P.Dictionary('string', 'float', /^-?[\d.,]+$/, /^-?[\d\.,]+$/, 'P({name} = ', ') = '),
|
|
'labelPosition': new P.Enum('above', 'below', 'left', 'right', 'above-left', 'above-right', 'below-left', 'below-right'),
|
|
'labelX': 'number'
|
|
}}*/
|
|
static properties() {return {
|
|
[QT_TRANSLATE_NOOP('prop','labelPosition')]: P.Enum.Position,
|
|
[QT_TRANSLATE_NOOP('prop','labelX')]: 'number',
|
|
'comment1': QT_TRANSLATE_NOOP(
|
|
'comment',
|
|
'Note: Specify the probability for each value.'
|
|
),
|
|
[QT_TRANSLATE_NOOP('prop','probabilities')]: new P.Dictionary('string', 'float', /^-?[\d.,]+$/, /^-?[\d\.,]+$/, 'P({name} = ', ') = '),
|
|
}}
|
|
|
|
constructor(name = null, visible = true, color = null, labelContent = 'name + value',
|
|
beginIncluded = true, drawLineEnds = true, probabilities = {'0': '0'}, labelPosition = 'above', labelX = 1) {
|
|
if(name == null) name = Common.getNewName('XYZUVW')
|
|
super(name, visible, color, labelContent)
|
|
this.beginIncluded = beginIncluded
|
|
this.drawLineEnds = drawLineEnds
|
|
this.probabilities = probabilities
|
|
this.labelPosition = labelPosition
|
|
this.labelX = labelX
|
|
this.update()
|
|
}
|
|
|
|
export() {
|
|
return [this.name, this.visible, this.color.toString(), this.labelContent,
|
|
this.beginIncluded, this.drawLineEnds, this.probabilities, this.labelPosition, this.labelX]
|
|
}
|
|
|
|
|
|
getReadableString() {
|
|
var keys = Object.keys(this.probabilities).sort((a,b) => a-b);
|
|
return `F_${this.name}(x) = P(${this.name} ≤ x)\n` + keys.map(idx => `P(${this.name}=${idx})=${this.probabilities[idx]}`).join("; ")
|
|
}
|
|
|
|
execute(x = 1) {
|
|
var ret = 0;
|
|
Object.keys(this.probabilities).sort((a,b) => a-b).forEach(idx => {
|
|
if(x >= idx) ret += parseFloat(this.probabilities[idx].replace(/,/g, '.'))
|
|
})
|
|
return ret
|
|
}
|
|
|
|
canExecute(x = 1) {return true}
|
|
// Simplify returns the simplified string of the expression.
|
|
simplify(x = 1) {
|
|
return this.execute(x)
|
|
}
|
|
|
|
getLabel() {
|
|
switch(this.labelContent) {
|
|
case 'name':
|
|
return `P(${this.name} ≤ x)`
|
|
case 'name + value':
|
|
return this.getReadableString()
|
|
case 'null':
|
|
return ''
|
|
}
|
|
}
|
|
|
|
draw(canvas, ctx) {
|
|
var currentY = 0;
|
|
var keys = Object.keys(this.probabilities).map(idx => parseInt(idx)).sort((a,b) => a-b)
|
|
if(canvas.visible(keys[0],this.probabilities[keys[0]].replace(/,/g, '.'))) {
|
|
canvas.drawLine(ctx,
|
|
0,
|
|
canvas.y2px(0),
|
|
canvas.x2px(keys[0]),
|
|
canvas.y2px(0)
|
|
)
|
|
if(canvas.visible(keys[0],0)) {
|
|
ctx.beginPath();
|
|
ctx.arc(canvas.x2px(keys[0])+4,canvas.y2px(0), 4, Math.PI / 2, 3 * Math.PI / 2);
|
|
ctx.stroke();
|
|
}
|
|
}
|
|
for(var i = 0; i < keys.length-1; i++) {
|
|
var idx = keys[i];
|
|
currentY += parseFloat(this.probabilities[idx].replace(/,/g, '.'));
|
|
if(canvas.visible(idx,currentY) || canvas.visible(keys[i+1],currentY)) {
|
|
canvas.drawLine(ctx,
|
|
Math.max(0,canvas.x2px(idx)),
|
|
canvas.y2px(currentY),
|
|
Math.min(canvas.canvasSize.width,canvas.x2px(keys[i+1])),
|
|
canvas.y2px(currentY)
|
|
)
|
|
if(canvas.visible(idx,currentY)) {
|
|
ctx.beginPath();
|
|
ctx.arc(canvas.x2px(idx),canvas.y2px(currentY), 4, 0, 2 * Math.PI);
|
|
ctx.fill();
|
|
}
|
|
if(canvas.visible(keys[i+1],currentY)) {
|
|
ctx.beginPath();
|
|
ctx.arc(canvas.x2px(keys[i+1])+4,canvas.y2px(currentY), 4, Math.PI / 2, 3 * Math.PI / 2);
|
|
ctx.stroke();
|
|
}
|
|
}
|
|
}
|
|
if(canvas.visible(keys[keys.length-1],currentY+parseFloat(this.probabilities[keys[keys.length-1]]))) {
|
|
canvas.drawLine(ctx,
|
|
Math.max(0,canvas.x2px(keys[keys.length-1])),
|
|
canvas.y2px(currentY+parseFloat(this.probabilities[keys[keys.length-1]].replace(/,/g, '.'))),
|
|
canvas.canvasSize.width,
|
|
canvas.y2px(currentY+parseFloat(this.probabilities[keys[keys.length-1]].replace(/,/g, '.')))
|
|
)
|
|
ctx.beginPath();
|
|
ctx.arc(
|
|
canvas.x2px(keys[keys.length-1]),
|
|
canvas.y2px(currentY+parseFloat(this.probabilities[keys[keys.length-1]].replace(/,/g, '.'))),
|
|
4, 0, 2 * Math.PI);
|
|
ctx.fill();
|
|
}
|
|
|
|
// Label
|
|
var text = this.getLabel()
|
|
ctx.font = `${canvas.textsize}px sans-serif`
|
|
var textSize = canvas.measureText(ctx, text)
|
|
var posX = canvas.x2px(this.labelX)
|
|
var posY = canvas.y2px(this.execute(this.labelX))
|
|
switch(this.labelPosition) {
|
|
case 'above':
|
|
canvas.drawVisibleText(ctx, text, posX-textSize.width/2, posY-textSize.height)
|
|
break;
|
|
case 'below':
|
|
canvas.drawVisibleText(ctx, text, posX-textSize.width/2, posY+textSize.height)
|
|
break;
|
|
case 'left':
|
|
canvas.drawVisibleText(ctx, text, posX-textSize.width, posY-textSize.height/2)
|
|
break;
|
|
case 'right':
|
|
canvas.drawVisibleText(ctx, text, posX, posY-textSize.height/2)
|
|
break;
|
|
case 'above-left':
|
|
canvas.drawVisibleText(ctx, text, posX-textSize.width, posY-textSize.height)
|
|
break;
|
|
case 'above-right':
|
|
canvas.drawVisibleText(ctx, text, posX, posY-textSize.height)
|
|
break;
|
|
case 'below-left':
|
|
canvas.drawVisibleText(ctx, text, posX-textSize.width, posY+textSize.height)
|
|
break;
|
|
case 'below-right':
|
|
canvas.drawVisibleText(ctx, text, posX, posY+textSize.height)
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|