force graph input

This commit is contained in:
marshennikovaolga
2024-08-31 22:13:59 +03:00
parent 32352ca5f4
commit 2a8675e3d9
3 changed files with 148 additions and 107 deletions
+68 -18
View File
@@ -1,8 +1,9 @@
"use client" "use client"
import * as react from "react" import * as react from "react"
import type * as force_graph from "./force-graph-client" import type * as force_graph from "./force-graph-client"
import { useCoState } from "@/lib/providers/jazz-provider"
import { PublicGlobalGroup } from "@/lib/schema/master/public-group"
import { ID } from "jazz-tools"
let graph_data_promise = import("./graph-data.json").then(a => a.default) let graph_data_promise = import("./graph-data.json").then(a => a.default)
let ForceGraphClient = react.lazy(() => import("./force-graph-client-lazy")) let ForceGraphClient = react.lazy(() => import("./force-graph-client-lazy"))
@@ -10,23 +11,72 @@ let ForceGraphClient = react.lazy(() => import("./force-graph-client-lazy"))
export function PublicHomeRoute() { export function PublicHomeRoute() {
let raw_graph_data = react.use(graph_data_promise) let raw_graph_data = react.use(graph_data_promise)
let graph_items = react.useMemo(() => { const [placeholder, setPlaceholder] = react.useState("Search something...")
return raw_graph_data.map( const [currentTopicIndex, setCurrentTopicIndex] = react.useState(0)
(item): force_graph.ConnectionItem => ({ const [currentCharIndex, setCurrentCharIndex] = react.useState(0)
key: item.name, const globalGroup = useCoState(
title: item.prettyName, PublicGlobalGroup,
connections: item.connectedTopics process.env.NEXT_PUBLIC_JAZZ_GLOBAL_GROUP as ID<PublicGlobalGroup>,
}) {
) root: {
}, [raw_graph_data]) topics: []
}
}
)
const topics = globalGroup?.root.topics?.map(topic => topic?.prettyName) || []
// let graph_items = react.useMemo(() => {
// return raw_graph_data.map(
// (item): force_graph.ConnectionItem => ({
// key: item.name,
// title: item.prettyName,
// connections: item.connectedTopics
// })
// )
// }, [raw_graph_data])
react.useEffect(() => {
if (topics.length === 0) return
const typingInterval = setInterval(() => {
const currentTopic = topics[currentTopicIndex]
if (currentTopic && currentCharIndex < currentTopic.length) {
setPlaceholder(`${currentTopic.slice(0, currentCharIndex + 1)}`)
setCurrentCharIndex(currentCharIndex + 1)
} else {
clearInterval(typingInterval)
setTimeout(() => {
setCurrentTopicIndex(prevIndex => (prevIndex + 1) % topics.length)
setCurrentCharIndex(0)
}, 1000)
}
}, 200)
return () => clearInterval(typingInterval)
}, [currentTopicIndex, currentCharIndex, topics])
return ( return (
<ForceGraphClient <div className="relative h-full w-screen">
raw_nodes={raw_graph_data} <ForceGraphClient
onNodeClick={val => { raw_nodes={raw_graph_data}
console.log("clicked", val) onNodeClick={val => {
}} console.log("clicked", val)
filter_query="" }}
/> filter_query=""
/>
<div className="absolute left-0 top-0 z-20 p-4">
<h2 className="text-xl font-bold text-black dark:text-white">Learn Anything</h2>
</div>
<div className="absolute left-1/2 top-1/2 z-10 w-[60%] -translate-x-1/2 -translate-y-1/2 transform">
<div className="flex flex-col items-center justify-center gap-6">
<h1 className="text-center text-5xl font-bold uppercase text-black dark:text-white">i want to learn</h1>
<input
type="text"
placeholder={placeholder}
className="bg-result w-[70%] rounded-md border px-6 py-3 text-lg shadow-lg placeholder:text-black/40 focus:outline-none active:outline-none dark:placeholder:text-white/40"
/>
</div>
</div>
</div>
) )
} }
@@ -1,16 +1,16 @@
"use client" "use client"
import * as react from "react" import * as react from "react"
import * as fg from "@nothing-but/force-graph" import * as fg from "@nothing-but/force-graph"
import {ease, trig, raf} from "@nothing-but/utils" import { ease, trig, raf } from "@nothing-but/utils"
import * as schedule from "@/lib/utils/schedule" import * as schedule from "@/lib/utils/schedule"
import * as canvas from "@/lib/utils/canvas" import * as canvas from "@/lib/utils/canvas"
export type RawGraphNode = { export type RawGraphNode = {
name: string, name: string
prettyName: string, prettyName: string
connectedTopics: string[], connectedTopics: string[]
} }
type HSL = [hue: number, saturation: number, lightness: number] type HSL = [hue: number, saturation: number, lightness: number]
@@ -57,8 +57,7 @@ const visitColorNode = (
} }
} }
function generateColorMap(nodes: readonly fg.graph.Node[]): ColorMap function generateColorMap(nodes: readonly fg.graph.Node[]): ColorMap {
{
const hls_map: HSLMap = new Map() const hls_map: HSLMap = new Map()
for (let i = 0; i < nodes.length; i++) { for (let i = 0; i < nodes.length; i++) {
@@ -75,8 +74,7 @@ function generateColorMap(nodes: readonly fg.graph.Node[]): ColorMap
return color_map return color_map
} }
function generateNodesFromRawData(raw_data: RawGraphNode[]): [fg.graph.Node[], fg.graph.Edge[]] function generateNodesFromRawData(raw_data: RawGraphNode[]): [fg.graph.Node[], fg.graph.Edge[]] {
{
const nodes_map = new Map<string, fg.graph.Node>() const nodes_map = new Map<string, fg.graph.Node>()
const edges: fg.graph.Edge[] = [] const edges: fg.graph.Edge[] = []
@@ -120,22 +118,20 @@ function filterNodes(
// regex matching all letters of the filter (out of order) // regex matching all letters of the filter (out of order)
const regex = new RegExp(filter.split("").join(".*"), "i") const regex = new RegExp(filter.split("").join(".*"), "i")
graph.nodes = nodes.filter((node) => regex.test(node.label)) graph.nodes = nodes.filter(node => regex.test(node.label))
graph.edges = edges.filter( graph.edges = edges.filter(edge => regex.test(edge.a.label) && regex.test(edge.b.label))
(edge) => regex.test(edge.a.label) && regex.test(edge.b.label)
)
fg.graph.resetGraphGrid(graph.grid, graph.nodes) fg.graph.resetGraphGrid(graph.grid, graph.nodes)
} }
const GRAPH_OPTIONS: fg.graph.Options = { const GRAPH_OPTIONS: fg.graph.Options = {
min_move: 0.001, min_move: 0.001,
inertia_strength: 0.3, inertia_strength: 0.3,
origin_strength: 0.01, origin_strength: 0.01,
repel_distance: 40, repel_distance: 40,
repel_strength: 2, repel_strength: 2,
link_strength: 0.015, link_strength: 0.015,
grid_size: 500, grid_size: 500
} }
const TITLE_SIZE_PX = 400 const TITLE_SIZE_PX = 400
@@ -155,18 +151,17 @@ const simulateGraph = (
Push nodes away from the center (the title) Push nodes away from the center (the title)
*/ */
let grid_radius = graph.grid.size / 2 let grid_radius = graph.grid.size / 2
let origin_x = grid_radius + canvas.translate.x let origin_x = grid_radius + canvas.translate.x
let origin_y = grid_radius + canvas.translate.y let origin_y = grid_radius + canvas.translate.y
let vmax = Math.max(vw, vh) let vmax = Math.max(vw, vh)
let push_radius = let push_radius =
(Math.min(TITLE_SIZE_PX, vw / 2, vh / 2) / vmax) * (Math.min(TITLE_SIZE_PX, vw / 2, vh / 2) / vmax) * (graph.grid.size / canvas.scale) +
(graph.grid.size / canvas.scale) +
80 /* additional margin for when scrolled in */ 80 /* additional margin for when scrolled in */
for (let node of graph.nodes) { for (let node of graph.nodes) {
let dist_x = node.position.x - origin_x let dist_x = node.position.x - origin_x
let dist_y = (node.position.y - origin_y) * 2 let dist_y = (node.position.y - origin_y) * 2
let dist = Math.sqrt(dist_x * dist_x + dist_y * dist_y) let dist = Math.sqrt(dist_x * dist_x + dist_y * dist_y)
if (dist > push_radius) continue if (dist > push_radius) continue
let strength = ease.in_expo((push_radius - dist) / push_radius) let strength = ease.in_expo((push_radius - dist) / push_radius)
@@ -176,45 +171,44 @@ const simulateGraph = (
} }
} }
const drawGraph = ( const drawGraph = (canvas: fg.canvas.CanvasState, color_map: ColorMap): void => {
canvas: fg.canvas.CanvasState,
color_map: ColorMap
): void => {
fg.canvas.resetFrame(canvas) fg.canvas.resetFrame(canvas)
fg.canvas.drawEdges(canvas) fg.canvas.drawEdges(canvas)
/* /*
Draw text nodes Draw text nodes
*/ */
let {ctx, graph} = canvas let { ctx, graph } = canvas
let {width, height} = canvas.ctx.canvas let { width, height } = canvas.ctx.canvas
let max_size = Math.max(width, height) let max_size = Math.max(width, height)
ctx.textAlign = "center" ctx.textAlign = "center"
ctx.textBaseline = "middle" ctx.textBaseline = "middle"
for (let node of graph.nodes) { for (let node of graph.nodes) {
let opacity = 0.6 + ((node.mass - 1) / 50) * 4
let opacity = 0.6 + ((node.mass-1) / 50) * 4 ctx.font = `${max_size / 200 + (((node.mass - 1) / 5) * (max_size / 100)) / canvas.scale}px sans-serif`
ctx.font = `${max_size/200 + (((node.mass-1) / 5) * (max_size/100)) / canvas.scale}px sans-serif` ctx.fillStyle =
node.anchor || canvas.hovered_node === node
? `rgba(129, 140, 248, ${opacity})`
: `hsl(${color_map[node.key as string]} / ${opacity})`
ctx.fillStyle = node.anchor || canvas.hovered_node === node ctx.fillText(
? `rgba(129, 140, 248, ${opacity})` node.label,
: `hsl(${color_map[node.key as string]} / ${opacity})`
ctx.fillText(node.label,
(node.position.x / graph.grid.size) * max_size, (node.position.x / graph.grid.size) * max_size,
(node.position.y / graph.grid.size) * max_size) (node.position.y / graph.grid.size) * max_size
)
} }
} }
class State { class State {
ctx: CanvasRenderingContext2D | null = null ctx: CanvasRenderingContext2D | null = null
nodes: fg.graph.Node[] = [] nodes: fg.graph.Node[] = []
edges: fg.graph.Edge[] = [] edges: fg.graph.Edge[] = []
graph: fg.graph.Graph = fg.graph.makeGraph(GRAPH_OPTIONS, [], []) graph: fg.graph.Graph = fg.graph.makeGraph(GRAPH_OPTIONS, [], [])
gestures: fg.canvas.CanvasGestures | null = null gestures: fg.canvas.CanvasGestures | null = null
loop: raf.AnimationLoop | null = null loop: raf.AnimationLoop | null = null
@@ -226,29 +220,30 @@ class State {
} }
function init( function init(
s : State, s: State,
props: { props: {
onNodeClick: (name: string) => void onNodeClick: (name: string) => void
raw_nodes: RawGraphNode[] raw_nodes: RawGraphNode[]
canvas_el: HTMLCanvasElement | null canvas_el: HTMLCanvasElement | null
}) { }
let {canvas_el, raw_nodes} = props ) {
let { canvas_el, raw_nodes } = props
if (canvas_el == null) return if (canvas_el == null) return
s.ctx = canvas_el.getContext("2d") s.ctx = canvas_el.getContext("2d")
if (s.ctx == null) return if (s.ctx == null) return
[s.nodes, s.edges] = generateNodesFromRawData(raw_nodes) ;[s.nodes, s.edges] = generateNodesFromRawData(raw_nodes)
let color_map = generateColorMap(s.nodes) let color_map = generateColorMap(s.nodes)
s.graph = fg.graph.makeGraph(GRAPH_OPTIONS, s.nodes.slice(), s.edges.slice()) s.graph = fg.graph.makeGraph(GRAPH_OPTIONS, s.nodes.slice(), s.edges.slice())
let canvas_state = fg.canvas.canvasState({ let canvas_state = fg.canvas.canvasState({
ctx: s.ctx, ctx: s.ctx,
graph: s.graph, graph: s.graph,
max_scale: 3, max_scale: 3,
init_scale: 1.7, init_scale: 1.7,
init_grid_pos: trig.ZERO init_grid_pos: trig.ZERO
}) })
@@ -259,7 +254,7 @@ function init(
}) })
s.ro.observe(canvas_el) s.ro.observe(canvas_el)
let loop = s.loop = raf.makeAnimationLoop((time) => { let loop = (s.loop = raf.makeAnimationLoop(time => {
let is_active = gestures.mode.type === fg.canvas.Mode.DraggingNode let is_active = gestures.mode.type === fg.canvas.Mode.DraggingNode
let iterations = raf.calcIterations(s.frame_iter_limit, time) let iterations = raf.calcIterations(s.frame_iter_limit, time)
@@ -268,30 +263,25 @@ function init(
simulateGraph(s.alpha, s.graph, canvas_state, window.innerWidth, window.innerHeight) simulateGraph(s.alpha, s.graph, canvas_state, window.innerWidth, window.innerHeight)
} }
drawGraph(canvas_state, color_map) drawGraph(canvas_state, color_map)
}) }))
raf.loopStart(loop) raf.loopStart(loop)
let gestures = s.gestures = fg.canvas.canvasGestures({ let gestures = (s.gestures = fg.canvas.canvasGestures({
canvas: canvas_state, canvas: canvas_state,
onGesture: (e) => { onGesture: e => {
switch (e.type) { switch (e.type) {
case fg.canvas.GestureEventType.Translate: case fg.canvas.GestureEventType.Translate:
s.bump_end = raf.bump(s.bump_end) s.bump_end = raf.bump(s.bump_end)
break break
case fg.canvas.GestureEventType.NodeClick: case fg.canvas.GestureEventType.NodeClick:
props.onNodeClick(e.node.key as string) props.onNodeClick(e.node.key as string)
break break
case fg.canvas.GestureEventType.NodeDrag: case fg.canvas.GestureEventType.NodeDrag:
fg.graph.changeNodePosition( fg.graph.changeNodePosition(canvas_state.graph.grid, e.node, e.pos.x, e.pos.y)
canvas_state.graph.grid, break
e.node,
e.pos.x,
e.pos.y
)
break
} }
} }
}) }))
} }
function updateQuery(s: State, filter_query: string) { function updateQuery(s: State, filter_query: string) {
@@ -318,16 +308,15 @@ export type ForceGraphProps = {
} }
export default function ForceGraphClient(props: ForceGraphProps): react.JSX.Element { export default function ForceGraphClient(props: ForceGraphProps): react.JSX.Element {
const [canvas_el, setCanvasEl] = react.useState<HTMLCanvasElement | null>(null) const [canvas_el, setCanvasEl] = react.useState<HTMLCanvasElement | null>(null)
const state = react.useRef(new State()) const state = react.useRef(new State())
react.useEffect(() => { react.useEffect(() => {
init(state.current, { init(state.current, {
canvas_el: canvas_el, canvas_el: canvas_el,
onNodeClick: props.onNodeClick, onNodeClick: props.onNodeClick,
raw_nodes: props.raw_nodes, raw_nodes: props.raw_nodes
}) })
}, [canvas_el]) }, [canvas_el])
@@ -339,16 +328,18 @@ export default function ForceGraphClient(props: ForceGraphProps): react.JSX.Elem
return () => cleanup(state.current) return () => cleanup(state.current)
}, []) }, [])
return <div className="absolute inset-0 overflow-hidden"> return (
<canvas <div className="absolute inset-0 overflow-hidden">
ref={setCanvasEl} <canvas
style={{ ref={setCanvasEl}
position: "absolute", style={{
top: "-10%", position: "absolute",
left: "-10%", top: "-10%",
width: "120%", left: "-10%",
height: "120%", width: "120%",
}} height: "120%"
/> }}
</div> />
</div>
)
} }
+1 -1
View File
@@ -328,7 +328,7 @@ export const createForceGraph = (props: ForceGraphProps): react.JSX.Element => {
top: "-10%", top: "-10%",
left: "-10%", left: "-10%",
width: "120%", width: "120%",
height: "120%" height: "100%"
}} }}
/> />
</div> </div>